xref: /openbmc/linux/fs/btrfs/ctree.c (revision bb8e9a60)
1c1d7c514SDavid Sterba // SPDX-License-Identifier: GPL-2.0
26cbd5570SChris Mason /*
3d352ac68SChris Mason  * Copyright (C) 2007,2008 Oracle.  All rights reserved.
46cbd5570SChris Mason  */
56cbd5570SChris Mason 
6a6b6e75eSChris Mason #include <linux/sched.h>
75a0e3ad6STejun Heo #include <linux/slab.h>
8bd989ba3SJan Schmidt #include <linux/rbtree.h>
9adf02123SDavid Sterba #include <linux/mm.h>
10e41d12f5SChristoph Hellwig #include <linux/error-injection.h>
11eb60ceacSChris Mason #include "ctree.h"
12eb60ceacSChris Mason #include "disk-io.h"
137f5c1516SChris Mason #include "transaction.h"
145f39d397SChris Mason #include "print-tree.h"
15925baeddSChris Mason #include "locking.h"
16de37aa51SNikolay Borisov #include "volumes.h"
17f616f5cdSQu Wenruo #include "qgroup.h"
18f3a84ccdSFilipe Manana #include "tree-mod-log.h"
199a8dd150SChris Mason 
20e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
21e089f05cSChris Mason 		      *root, struct btrfs_path *path, int level);
22310712b2SOmar Sandoval static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root *root,
23310712b2SOmar Sandoval 		      const struct btrfs_key *ins_key, struct btrfs_path *path,
24310712b2SOmar Sandoval 		      int data_size, int extend);
255f39d397SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
262ff7e61eSJeff Mahoney 			  struct extent_buffer *dst,
27971a1f66SChris Mason 			  struct extent_buffer *src, int empty);
285f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
295f39d397SChris Mason 			      struct extent_buffer *dst_buf,
305f39d397SChris Mason 			      struct extent_buffer *src_buf);
31afe5fea7STsutomu Itoh static void del_ptr(struct btrfs_root *root, struct btrfs_path *path,
32afe5fea7STsutomu Itoh 		    int level, int slot);
33d97e63b6SChris Mason 
34af024ed2SJohannes Thumshirn static const struct btrfs_csums {
35af024ed2SJohannes Thumshirn 	u16		size;
3659a0fcdbSDavid Sterba 	const char	name[10];
3759a0fcdbSDavid Sterba 	const char	driver[12];
38af024ed2SJohannes Thumshirn } btrfs_csums[] = {
39af024ed2SJohannes Thumshirn 	[BTRFS_CSUM_TYPE_CRC32] = { .size = 4, .name = "crc32c" },
403951e7f0SJohannes Thumshirn 	[BTRFS_CSUM_TYPE_XXHASH] = { .size = 8, .name = "xxhash64" },
413831bf00SJohannes Thumshirn 	[BTRFS_CSUM_TYPE_SHA256] = { .size = 32, .name = "sha256" },
42352ae07bSDavid Sterba 	[BTRFS_CSUM_TYPE_BLAKE2] = { .size = 32, .name = "blake2b",
43352ae07bSDavid Sterba 				     .driver = "blake2b-256" },
44af024ed2SJohannes Thumshirn };
45af024ed2SJohannes Thumshirn 
46af024ed2SJohannes Thumshirn int btrfs_super_csum_size(const struct btrfs_super_block *s)
47af024ed2SJohannes Thumshirn {
48af024ed2SJohannes Thumshirn 	u16 t = btrfs_super_csum_type(s);
49af024ed2SJohannes Thumshirn 	/*
50af024ed2SJohannes Thumshirn 	 * csum type is validated at mount time
51af024ed2SJohannes Thumshirn 	 */
52af024ed2SJohannes Thumshirn 	return btrfs_csums[t].size;
53af024ed2SJohannes Thumshirn }
54af024ed2SJohannes Thumshirn 
55af024ed2SJohannes Thumshirn const char *btrfs_super_csum_name(u16 csum_type)
56af024ed2SJohannes Thumshirn {
57af024ed2SJohannes Thumshirn 	/* csum type is validated at mount time */
58af024ed2SJohannes Thumshirn 	return btrfs_csums[csum_type].name;
59af024ed2SJohannes Thumshirn }
60af024ed2SJohannes Thumshirn 
61b4e967beSDavid Sterba /*
62b4e967beSDavid Sterba  * Return driver name if defined, otherwise the name that's also a valid driver
63b4e967beSDavid Sterba  * name
64b4e967beSDavid Sterba  */
65b4e967beSDavid Sterba const char *btrfs_super_csum_driver(u16 csum_type)
66b4e967beSDavid Sterba {
67b4e967beSDavid Sterba 	/* csum type is validated at mount time */
6859a0fcdbSDavid Sterba 	return btrfs_csums[csum_type].driver[0] ?
6959a0fcdbSDavid Sterba 		btrfs_csums[csum_type].driver :
70b4e967beSDavid Sterba 		btrfs_csums[csum_type].name;
71b4e967beSDavid Sterba }
72b4e967beSDavid Sterba 
73604997b4SDavid Sterba size_t __attribute_const__ btrfs_get_num_csums(void)
74f7cea56cSDavid Sterba {
75f7cea56cSDavid Sterba 	return ARRAY_SIZE(btrfs_csums);
76f7cea56cSDavid Sterba }
77f7cea56cSDavid Sterba 
782c90e5d6SChris Mason struct btrfs_path *btrfs_alloc_path(void)
792c90e5d6SChris Mason {
80e2c89907SMasahiro Yamada 	return kmem_cache_zalloc(btrfs_path_cachep, GFP_NOFS);
812c90e5d6SChris Mason }
822c90e5d6SChris Mason 
83d352ac68SChris Mason /* this also releases the path */
842c90e5d6SChris Mason void btrfs_free_path(struct btrfs_path *p)
852c90e5d6SChris Mason {
86ff175d57SJesper Juhl 	if (!p)
87ff175d57SJesper Juhl 		return;
88b3b4aa74SDavid Sterba 	btrfs_release_path(p);
892c90e5d6SChris Mason 	kmem_cache_free(btrfs_path_cachep, p);
902c90e5d6SChris Mason }
912c90e5d6SChris Mason 
92d352ac68SChris Mason /*
93d352ac68SChris Mason  * path release drops references on the extent buffers in the path
94d352ac68SChris Mason  * and it drops any locks held by this path
95d352ac68SChris Mason  *
96d352ac68SChris Mason  * It is safe to call this on paths that no locks or extent buffers held.
97d352ac68SChris Mason  */
98b3b4aa74SDavid Sterba noinline void btrfs_release_path(struct btrfs_path *p)
99eb60ceacSChris Mason {
100eb60ceacSChris Mason 	int i;
101a2135011SChris Mason 
102234b63a0SChris Mason 	for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
1033f157a2fSChris Mason 		p->slots[i] = 0;
104eb60ceacSChris Mason 		if (!p->nodes[i])
105925baeddSChris Mason 			continue;
106925baeddSChris Mason 		if (p->locks[i]) {
107bd681513SChris Mason 			btrfs_tree_unlock_rw(p->nodes[i], p->locks[i]);
108925baeddSChris Mason 			p->locks[i] = 0;
109925baeddSChris Mason 		}
1105f39d397SChris Mason 		free_extent_buffer(p->nodes[i]);
1113f157a2fSChris Mason 		p->nodes[i] = NULL;
112eb60ceacSChris Mason 	}
113eb60ceacSChris Mason }
114eb60ceacSChris Mason 
115d352ac68SChris Mason /*
116d352ac68SChris Mason  * safely gets a reference on the root node of a tree.  A lock
117d352ac68SChris Mason  * is not taken, so a concurrent writer may put a different node
118d352ac68SChris Mason  * at the root of the tree.  See btrfs_lock_root_node for the
119d352ac68SChris Mason  * looping required.
120d352ac68SChris Mason  *
121d352ac68SChris Mason  * The extent buffer returned by this has a reference taken, so
122d352ac68SChris Mason  * it won't disappear.  It may stop being the root of the tree
123d352ac68SChris Mason  * at any time because there are no locks held.
124d352ac68SChris Mason  */
125925baeddSChris Mason struct extent_buffer *btrfs_root_node(struct btrfs_root *root)
126925baeddSChris Mason {
127925baeddSChris Mason 	struct extent_buffer *eb;
128240f62c8SChris Mason 
1293083ee2eSJosef Bacik 	while (1) {
130240f62c8SChris Mason 		rcu_read_lock();
131240f62c8SChris Mason 		eb = rcu_dereference(root->node);
1323083ee2eSJosef Bacik 
1333083ee2eSJosef Bacik 		/*
1343083ee2eSJosef Bacik 		 * RCU really hurts here, we could free up the root node because
13501327610SNicholas D Steeves 		 * it was COWed but we may not get the new root node yet so do
1363083ee2eSJosef Bacik 		 * the inc_not_zero dance and if it doesn't work then
1373083ee2eSJosef Bacik 		 * synchronize_rcu and try again.
1383083ee2eSJosef Bacik 		 */
1393083ee2eSJosef Bacik 		if (atomic_inc_not_zero(&eb->refs)) {
140240f62c8SChris Mason 			rcu_read_unlock();
1413083ee2eSJosef Bacik 			break;
1423083ee2eSJosef Bacik 		}
1433083ee2eSJosef Bacik 		rcu_read_unlock();
1443083ee2eSJosef Bacik 		synchronize_rcu();
1453083ee2eSJosef Bacik 	}
146925baeddSChris Mason 	return eb;
147925baeddSChris Mason }
148925baeddSChris Mason 
14992a7cc42SQu Wenruo /*
15092a7cc42SQu Wenruo  * Cowonly root (not-shareable trees, everything not subvolume or reloc roots),
15192a7cc42SQu Wenruo  * just get put onto a simple dirty list.  Transaction walks this list to make
15292a7cc42SQu Wenruo  * sure they get properly updated on disk.
153d352ac68SChris Mason  */
1540b86a832SChris Mason static void add_root_to_dirty_list(struct btrfs_root *root)
1550b86a832SChris Mason {
1560b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
1570b246afaSJeff Mahoney 
158e7070be1SJosef Bacik 	if (test_bit(BTRFS_ROOT_DIRTY, &root->state) ||
159e7070be1SJosef Bacik 	    !test_bit(BTRFS_ROOT_TRACK_DIRTY, &root->state))
160e7070be1SJosef Bacik 		return;
161e7070be1SJosef Bacik 
1620b246afaSJeff Mahoney 	spin_lock(&fs_info->trans_lock);
163e7070be1SJosef Bacik 	if (!test_and_set_bit(BTRFS_ROOT_DIRTY, &root->state)) {
164e7070be1SJosef Bacik 		/* Want the extent tree to be the last on the list */
1654fd786e6SMisono Tomohiro 		if (root->root_key.objectid == BTRFS_EXTENT_TREE_OBJECTID)
166e7070be1SJosef Bacik 			list_move_tail(&root->dirty_list,
1670b246afaSJeff Mahoney 				       &fs_info->dirty_cowonly_roots);
168e7070be1SJosef Bacik 		else
169e7070be1SJosef Bacik 			list_move(&root->dirty_list,
1700b246afaSJeff Mahoney 				  &fs_info->dirty_cowonly_roots);
1710b86a832SChris Mason 	}
1720b246afaSJeff Mahoney 	spin_unlock(&fs_info->trans_lock);
1730b86a832SChris Mason }
1740b86a832SChris Mason 
175d352ac68SChris Mason /*
176d352ac68SChris Mason  * used by snapshot creation to make a copy of a root for a tree with
177d352ac68SChris Mason  * a given objectid.  The buffer with the new root node is returned in
178d352ac68SChris Mason  * cow_ret, and this func returns zero on success or a negative error code.
179d352ac68SChris Mason  */
180be20aa9dSChris Mason int btrfs_copy_root(struct btrfs_trans_handle *trans,
181be20aa9dSChris Mason 		      struct btrfs_root *root,
182be20aa9dSChris Mason 		      struct extent_buffer *buf,
183be20aa9dSChris Mason 		      struct extent_buffer **cow_ret, u64 new_root_objectid)
184be20aa9dSChris Mason {
1850b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
186be20aa9dSChris Mason 	struct extent_buffer *cow;
187be20aa9dSChris Mason 	int ret = 0;
188be20aa9dSChris Mason 	int level;
1895d4f98a2SYan Zheng 	struct btrfs_disk_key disk_key;
190be20aa9dSChris Mason 
19192a7cc42SQu Wenruo 	WARN_ON(test_bit(BTRFS_ROOT_SHAREABLE, &root->state) &&
1920b246afaSJeff Mahoney 		trans->transid != fs_info->running_transaction->transid);
19392a7cc42SQu Wenruo 	WARN_ON(test_bit(BTRFS_ROOT_SHAREABLE, &root->state) &&
19427cdeb70SMiao Xie 		trans->transid != root->last_trans);
195be20aa9dSChris Mason 
196be20aa9dSChris Mason 	level = btrfs_header_level(buf);
1975d4f98a2SYan Zheng 	if (level == 0)
1985d4f98a2SYan Zheng 		btrfs_item_key(buf, &disk_key, 0);
1995d4f98a2SYan Zheng 	else
2005d4f98a2SYan Zheng 		btrfs_node_key(buf, &disk_key, 0);
20131840ae1SZheng Yan 
2024d75f8a9SDavid Sterba 	cow = btrfs_alloc_tree_block(trans, root, 0, new_root_objectid,
203cf6f34aaSJosef Bacik 				     &disk_key, level, buf->start, 0,
204cf6f34aaSJosef Bacik 				     BTRFS_NESTING_NEW_ROOT);
2055d4f98a2SYan Zheng 	if (IS_ERR(cow))
206be20aa9dSChris Mason 		return PTR_ERR(cow);
207be20aa9dSChris Mason 
20858e8012cSDavid Sterba 	copy_extent_buffer_full(cow, buf);
209be20aa9dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
210be20aa9dSChris Mason 	btrfs_set_header_generation(cow, trans->transid);
2115d4f98a2SYan Zheng 	btrfs_set_header_backref_rev(cow, BTRFS_MIXED_BACKREF_REV);
2125d4f98a2SYan Zheng 	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN |
2135d4f98a2SYan Zheng 				     BTRFS_HEADER_FLAG_RELOC);
2145d4f98a2SYan Zheng 	if (new_root_objectid == BTRFS_TREE_RELOC_OBJECTID)
2155d4f98a2SYan Zheng 		btrfs_set_header_flag(cow, BTRFS_HEADER_FLAG_RELOC);
2165d4f98a2SYan Zheng 	else
217be20aa9dSChris Mason 		btrfs_set_header_owner(cow, new_root_objectid);
218be20aa9dSChris Mason 
219de37aa51SNikolay Borisov 	write_extent_buffer_fsid(cow, fs_info->fs_devices->metadata_uuid);
2202b82032cSYan Zheng 
221be20aa9dSChris Mason 	WARN_ON(btrfs_header_generation(buf) > trans->transid);
2225d4f98a2SYan Zheng 	if (new_root_objectid == BTRFS_TREE_RELOC_OBJECTID)
223e339a6b0SJosef Bacik 		ret = btrfs_inc_ref(trans, root, cow, 1);
2245d4f98a2SYan Zheng 	else
225e339a6b0SJosef Bacik 		ret = btrfs_inc_ref(trans, root, cow, 0);
226867ed321SJosef Bacik 	if (ret) {
22772c9925fSFilipe Manana 		btrfs_tree_unlock(cow);
22872c9925fSFilipe Manana 		free_extent_buffer(cow);
229867ed321SJosef Bacik 		btrfs_abort_transaction(trans, ret);
230be20aa9dSChris Mason 		return ret;
231867ed321SJosef Bacik 	}
232be20aa9dSChris Mason 
233be20aa9dSChris Mason 	btrfs_mark_buffer_dirty(cow);
234be20aa9dSChris Mason 	*cow_ret = cow;
235be20aa9dSChris Mason 	return 0;
236be20aa9dSChris Mason }
237be20aa9dSChris Mason 
238d352ac68SChris Mason /*
2395d4f98a2SYan Zheng  * check if the tree block can be shared by multiple trees
2405d4f98a2SYan Zheng  */
2415d4f98a2SYan Zheng int btrfs_block_can_be_shared(struct btrfs_root *root,
2425d4f98a2SYan Zheng 			      struct extent_buffer *buf)
2435d4f98a2SYan Zheng {
2445d4f98a2SYan Zheng 	/*
24592a7cc42SQu Wenruo 	 * Tree blocks not in shareable trees and tree roots are never shared.
24692a7cc42SQu Wenruo 	 * If a block was allocated after the last snapshot and the block was
24792a7cc42SQu Wenruo 	 * not allocated by tree relocation, we know the block is not shared.
2485d4f98a2SYan Zheng 	 */
24992a7cc42SQu Wenruo 	if (test_bit(BTRFS_ROOT_SHAREABLE, &root->state) &&
2505d4f98a2SYan Zheng 	    buf != root->node && buf != root->commit_root &&
2515d4f98a2SYan Zheng 	    (btrfs_header_generation(buf) <=
2525d4f98a2SYan Zheng 	     btrfs_root_last_snapshot(&root->root_item) ||
2535d4f98a2SYan Zheng 	     btrfs_header_flag(buf, BTRFS_HEADER_FLAG_RELOC)))
2545d4f98a2SYan Zheng 		return 1;
255a79865c6SNikolay Borisov 
2565d4f98a2SYan Zheng 	return 0;
2575d4f98a2SYan Zheng }
2585d4f98a2SYan Zheng 
2595d4f98a2SYan Zheng static noinline int update_ref_for_cow(struct btrfs_trans_handle *trans,
2605d4f98a2SYan Zheng 				       struct btrfs_root *root,
2615d4f98a2SYan Zheng 				       struct extent_buffer *buf,
262f0486c68SYan, Zheng 				       struct extent_buffer *cow,
263f0486c68SYan, Zheng 				       int *last_ref)
2645d4f98a2SYan Zheng {
2650b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
2665d4f98a2SYan Zheng 	u64 refs;
2675d4f98a2SYan Zheng 	u64 owner;
2685d4f98a2SYan Zheng 	u64 flags;
2695d4f98a2SYan Zheng 	u64 new_flags = 0;
2705d4f98a2SYan Zheng 	int ret;
2715d4f98a2SYan Zheng 
2725d4f98a2SYan Zheng 	/*
2735d4f98a2SYan Zheng 	 * Backrefs update rules:
2745d4f98a2SYan Zheng 	 *
2755d4f98a2SYan Zheng 	 * Always use full backrefs for extent pointers in tree block
2765d4f98a2SYan Zheng 	 * allocated by tree relocation.
2775d4f98a2SYan Zheng 	 *
2785d4f98a2SYan Zheng 	 * If a shared tree block is no longer referenced by its owner
2795d4f98a2SYan Zheng 	 * tree (btrfs_header_owner(buf) == root->root_key.objectid),
2805d4f98a2SYan Zheng 	 * use full backrefs for extent pointers in tree block.
2815d4f98a2SYan Zheng 	 *
2825d4f98a2SYan Zheng 	 * If a tree block is been relocating
2835d4f98a2SYan Zheng 	 * (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID),
2845d4f98a2SYan Zheng 	 * use full backrefs for extent pointers in tree block.
2855d4f98a2SYan Zheng 	 * The reason for this is some operations (such as drop tree)
2865d4f98a2SYan Zheng 	 * are only allowed for blocks use full backrefs.
2875d4f98a2SYan Zheng 	 */
2885d4f98a2SYan Zheng 
2895d4f98a2SYan Zheng 	if (btrfs_block_can_be_shared(root, buf)) {
2902ff7e61eSJeff Mahoney 		ret = btrfs_lookup_extent_info(trans, fs_info, buf->start,
2913173a18fSJosef Bacik 					       btrfs_header_level(buf), 1,
2923173a18fSJosef Bacik 					       &refs, &flags);
293be1a5564SMark Fasheh 		if (ret)
294be1a5564SMark Fasheh 			return ret;
295e5df9573SMark Fasheh 		if (refs == 0) {
296e5df9573SMark Fasheh 			ret = -EROFS;
2970b246afaSJeff Mahoney 			btrfs_handle_fs_error(fs_info, ret, NULL);
298e5df9573SMark Fasheh 			return ret;
299e5df9573SMark Fasheh 		}
3005d4f98a2SYan Zheng 	} else {
3015d4f98a2SYan Zheng 		refs = 1;
3025d4f98a2SYan Zheng 		if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID ||
3035d4f98a2SYan Zheng 		    btrfs_header_backref_rev(buf) < BTRFS_MIXED_BACKREF_REV)
3045d4f98a2SYan Zheng 			flags = BTRFS_BLOCK_FLAG_FULL_BACKREF;
3055d4f98a2SYan Zheng 		else
3065d4f98a2SYan Zheng 			flags = 0;
3075d4f98a2SYan Zheng 	}
3085d4f98a2SYan Zheng 
3095d4f98a2SYan Zheng 	owner = btrfs_header_owner(buf);
3105d4f98a2SYan Zheng 	BUG_ON(owner == BTRFS_TREE_RELOC_OBJECTID &&
3115d4f98a2SYan Zheng 	       !(flags & BTRFS_BLOCK_FLAG_FULL_BACKREF));
3125d4f98a2SYan Zheng 
3135d4f98a2SYan Zheng 	if (refs > 1) {
3145d4f98a2SYan Zheng 		if ((owner == root->root_key.objectid ||
3155d4f98a2SYan Zheng 		     root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) &&
3165d4f98a2SYan Zheng 		    !(flags & BTRFS_BLOCK_FLAG_FULL_BACKREF)) {
317e339a6b0SJosef Bacik 			ret = btrfs_inc_ref(trans, root, buf, 1);
318692826b2SJeff Mahoney 			if (ret)
319692826b2SJeff Mahoney 				return ret;
3205d4f98a2SYan Zheng 
3215d4f98a2SYan Zheng 			if (root->root_key.objectid ==
3225d4f98a2SYan Zheng 			    BTRFS_TREE_RELOC_OBJECTID) {
323e339a6b0SJosef Bacik 				ret = btrfs_dec_ref(trans, root, buf, 0);
324692826b2SJeff Mahoney 				if (ret)
325692826b2SJeff Mahoney 					return ret;
326e339a6b0SJosef Bacik 				ret = btrfs_inc_ref(trans, root, cow, 1);
327692826b2SJeff Mahoney 				if (ret)
328692826b2SJeff Mahoney 					return ret;
3295d4f98a2SYan Zheng 			}
3305d4f98a2SYan Zheng 			new_flags |= BTRFS_BLOCK_FLAG_FULL_BACKREF;
3315d4f98a2SYan Zheng 		} else {
3325d4f98a2SYan Zheng 
3335d4f98a2SYan Zheng 			if (root->root_key.objectid ==
3345d4f98a2SYan Zheng 			    BTRFS_TREE_RELOC_OBJECTID)
335e339a6b0SJosef Bacik 				ret = btrfs_inc_ref(trans, root, cow, 1);
3365d4f98a2SYan Zheng 			else
337e339a6b0SJosef Bacik 				ret = btrfs_inc_ref(trans, root, cow, 0);
338692826b2SJeff Mahoney 			if (ret)
339692826b2SJeff Mahoney 				return ret;
3405d4f98a2SYan Zheng 		}
3415d4f98a2SYan Zheng 		if (new_flags != 0) {
342b1c79e09SJosef Bacik 			int level = btrfs_header_level(buf);
343b1c79e09SJosef Bacik 
34442c9d0b5SDavid Sterba 			ret = btrfs_set_disk_extent_flags(trans, buf,
345b1c79e09SJosef Bacik 							  new_flags, level, 0);
346be1a5564SMark Fasheh 			if (ret)
347be1a5564SMark Fasheh 				return ret;
3485d4f98a2SYan Zheng 		}
3495d4f98a2SYan Zheng 	} else {
3505d4f98a2SYan Zheng 		if (flags & BTRFS_BLOCK_FLAG_FULL_BACKREF) {
3515d4f98a2SYan Zheng 			if (root->root_key.objectid ==
3525d4f98a2SYan Zheng 			    BTRFS_TREE_RELOC_OBJECTID)
353e339a6b0SJosef Bacik 				ret = btrfs_inc_ref(trans, root, cow, 1);
3545d4f98a2SYan Zheng 			else
355e339a6b0SJosef Bacik 				ret = btrfs_inc_ref(trans, root, cow, 0);
356692826b2SJeff Mahoney 			if (ret)
357692826b2SJeff Mahoney 				return ret;
358e339a6b0SJosef Bacik 			ret = btrfs_dec_ref(trans, root, buf, 1);
359692826b2SJeff Mahoney 			if (ret)
360692826b2SJeff Mahoney 				return ret;
3615d4f98a2SYan Zheng 		}
3626a884d7dSDavid Sterba 		btrfs_clean_tree_block(buf);
363f0486c68SYan, Zheng 		*last_ref = 1;
3645d4f98a2SYan Zheng 	}
3655d4f98a2SYan Zheng 	return 0;
3665d4f98a2SYan Zheng }
3675d4f98a2SYan Zheng 
3685d4f98a2SYan Zheng /*
369d397712bSChris Mason  * does the dirty work in cow of a single block.  The parent block (if
370d397712bSChris Mason  * supplied) is updated to point to the new cow copy.  The new buffer is marked
371d397712bSChris Mason  * dirty and returned locked.  If you modify the block it needs to be marked
372d397712bSChris Mason  * dirty again.
373d352ac68SChris Mason  *
374d352ac68SChris Mason  * search_start -- an allocation hint for the new block
375d352ac68SChris Mason  *
376d397712bSChris Mason  * empty_size -- a hint that you plan on doing more cow.  This is the size in
377d397712bSChris Mason  * bytes the allocator should try to find free next to the block it returns.
378d397712bSChris Mason  * This is just a hint and may be ignored by the allocator.
379d352ac68SChris Mason  */
380d397712bSChris Mason static noinline int __btrfs_cow_block(struct btrfs_trans_handle *trans,
3815f39d397SChris Mason 			     struct btrfs_root *root,
3825f39d397SChris Mason 			     struct extent_buffer *buf,
3835f39d397SChris Mason 			     struct extent_buffer *parent, int parent_slot,
3845f39d397SChris Mason 			     struct extent_buffer **cow_ret,
3859631e4ccSJosef Bacik 			     u64 search_start, u64 empty_size,
3869631e4ccSJosef Bacik 			     enum btrfs_lock_nesting nest)
3876702ed49SChris Mason {
3880b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
3895d4f98a2SYan Zheng 	struct btrfs_disk_key disk_key;
3905f39d397SChris Mason 	struct extent_buffer *cow;
391be1a5564SMark Fasheh 	int level, ret;
392f0486c68SYan, Zheng 	int last_ref = 0;
393925baeddSChris Mason 	int unlock_orig = 0;
3940f5053ebSGoldwyn Rodrigues 	u64 parent_start = 0;
3956702ed49SChris Mason 
396925baeddSChris Mason 	if (*cow_ret == buf)
397925baeddSChris Mason 		unlock_orig = 1;
398925baeddSChris Mason 
39949d0c642SFilipe Manana 	btrfs_assert_tree_write_locked(buf);
400925baeddSChris Mason 
40192a7cc42SQu Wenruo 	WARN_ON(test_bit(BTRFS_ROOT_SHAREABLE, &root->state) &&
4020b246afaSJeff Mahoney 		trans->transid != fs_info->running_transaction->transid);
40392a7cc42SQu Wenruo 	WARN_ON(test_bit(BTRFS_ROOT_SHAREABLE, &root->state) &&
40427cdeb70SMiao Xie 		trans->transid != root->last_trans);
4055f39d397SChris Mason 
4067bb86316SChris Mason 	level = btrfs_header_level(buf);
40731840ae1SZheng Yan 
4085d4f98a2SYan Zheng 	if (level == 0)
4095d4f98a2SYan Zheng 		btrfs_item_key(buf, &disk_key, 0);
4105d4f98a2SYan Zheng 	else
4115d4f98a2SYan Zheng 		btrfs_node_key(buf, &disk_key, 0);
4125d4f98a2SYan Zheng 
4130f5053ebSGoldwyn Rodrigues 	if ((root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) && parent)
4145d4f98a2SYan Zheng 		parent_start = parent->start;
4155d4f98a2SYan Zheng 
41679bd3712SFilipe Manana 	cow = btrfs_alloc_tree_block(trans, root, parent_start,
41779bd3712SFilipe Manana 				     root->root_key.objectid, &disk_key, level,
41879bd3712SFilipe Manana 				     search_start, empty_size, nest);
4196702ed49SChris Mason 	if (IS_ERR(cow))
4206702ed49SChris Mason 		return PTR_ERR(cow);
4216702ed49SChris Mason 
422b4ce94deSChris Mason 	/* cow is set to blocking by btrfs_init_new_buffer */
423b4ce94deSChris Mason 
42458e8012cSDavid Sterba 	copy_extent_buffer_full(cow, buf);
425db94535dSChris Mason 	btrfs_set_header_bytenr(cow, cow->start);
4265f39d397SChris Mason 	btrfs_set_header_generation(cow, trans->transid);
4275d4f98a2SYan Zheng 	btrfs_set_header_backref_rev(cow, BTRFS_MIXED_BACKREF_REV);
4285d4f98a2SYan Zheng 	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN |
4295d4f98a2SYan Zheng 				     BTRFS_HEADER_FLAG_RELOC);
4305d4f98a2SYan Zheng 	if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID)
4315d4f98a2SYan Zheng 		btrfs_set_header_flag(cow, BTRFS_HEADER_FLAG_RELOC);
4325d4f98a2SYan Zheng 	else
4335f39d397SChris Mason 		btrfs_set_header_owner(cow, root->root_key.objectid);
4346702ed49SChris Mason 
435de37aa51SNikolay Borisov 	write_extent_buffer_fsid(cow, fs_info->fs_devices->metadata_uuid);
4362b82032cSYan Zheng 
437be1a5564SMark Fasheh 	ret = update_ref_for_cow(trans, root, buf, cow, &last_ref);
438b68dc2a9SMark Fasheh 	if (ret) {
439572c83acSJosef Bacik 		btrfs_tree_unlock(cow);
440572c83acSJosef Bacik 		free_extent_buffer(cow);
44166642832SJeff Mahoney 		btrfs_abort_transaction(trans, ret);
442b68dc2a9SMark Fasheh 		return ret;
443b68dc2a9SMark Fasheh 	}
4441a40e23bSZheng Yan 
44592a7cc42SQu Wenruo 	if (test_bit(BTRFS_ROOT_SHAREABLE, &root->state)) {
44683d4cfd4SJosef Bacik 		ret = btrfs_reloc_cow_block(trans, root, buf, cow);
44793314e3bSZhaolei 		if (ret) {
448572c83acSJosef Bacik 			btrfs_tree_unlock(cow);
449572c83acSJosef Bacik 			free_extent_buffer(cow);
45066642832SJeff Mahoney 			btrfs_abort_transaction(trans, ret);
45183d4cfd4SJosef Bacik 			return ret;
45283d4cfd4SJosef Bacik 		}
45393314e3bSZhaolei 	}
4543fd0a558SYan, Zheng 
4556702ed49SChris Mason 	if (buf == root->node) {
456925baeddSChris Mason 		WARN_ON(parent && parent != buf);
4575d4f98a2SYan Zheng 		if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID ||
4585d4f98a2SYan Zheng 		    btrfs_header_backref_rev(buf) < BTRFS_MIXED_BACKREF_REV)
4595d4f98a2SYan Zheng 			parent_start = buf->start;
460925baeddSChris Mason 
46167439dadSDavid Sterba 		atomic_inc(&cow->refs);
462406808abSFilipe Manana 		ret = btrfs_tree_mod_log_insert_root(root->node, cow, true);
463d9d19a01SDavid Sterba 		BUG_ON(ret < 0);
464240f62c8SChris Mason 		rcu_assign_pointer(root->node, cow);
465925baeddSChris Mason 
4667a163608SFilipe Manana 		btrfs_free_tree_block(trans, btrfs_root_id(root), buf,
4677a163608SFilipe Manana 				      parent_start, last_ref);
4685f39d397SChris Mason 		free_extent_buffer(buf);
4690b86a832SChris Mason 		add_root_to_dirty_list(root);
4706702ed49SChris Mason 	} else {
4715d4f98a2SYan Zheng 		WARN_ON(trans->transid != btrfs_header_generation(parent));
472f3a84ccdSFilipe Manana 		btrfs_tree_mod_log_insert_key(parent, parent_slot,
473f3a84ccdSFilipe Manana 					      BTRFS_MOD_LOG_KEY_REPLACE, GFP_NOFS);
4745f39d397SChris Mason 		btrfs_set_node_blockptr(parent, parent_slot,
475db94535dSChris Mason 					cow->start);
47674493f7aSChris Mason 		btrfs_set_node_ptr_generation(parent, parent_slot,
47774493f7aSChris Mason 					      trans->transid);
4786702ed49SChris Mason 		btrfs_mark_buffer_dirty(parent);
4795de865eeSFilipe David Borba Manana 		if (last_ref) {
480f3a84ccdSFilipe Manana 			ret = btrfs_tree_mod_log_free_eb(buf);
4815de865eeSFilipe David Borba Manana 			if (ret) {
482572c83acSJosef Bacik 				btrfs_tree_unlock(cow);
483572c83acSJosef Bacik 				free_extent_buffer(cow);
48466642832SJeff Mahoney 				btrfs_abort_transaction(trans, ret);
4855de865eeSFilipe David Borba Manana 				return ret;
4865de865eeSFilipe David Borba Manana 			}
4875de865eeSFilipe David Borba Manana 		}
4887a163608SFilipe Manana 		btrfs_free_tree_block(trans, btrfs_root_id(root), buf,
4897a163608SFilipe Manana 				      parent_start, last_ref);
4906702ed49SChris Mason 	}
491925baeddSChris Mason 	if (unlock_orig)
492925baeddSChris Mason 		btrfs_tree_unlock(buf);
4933083ee2eSJosef Bacik 	free_extent_buffer_stale(buf);
4946702ed49SChris Mason 	btrfs_mark_buffer_dirty(cow);
4956702ed49SChris Mason 	*cow_ret = cow;
4966702ed49SChris Mason 	return 0;
4976702ed49SChris Mason }
4986702ed49SChris Mason 
4995d4f98a2SYan Zheng static inline int should_cow_block(struct btrfs_trans_handle *trans,
5005d4f98a2SYan Zheng 				   struct btrfs_root *root,
5015d4f98a2SYan Zheng 				   struct extent_buffer *buf)
5025d4f98a2SYan Zheng {
503f5ee5c9aSJeff Mahoney 	if (btrfs_is_testing(root->fs_info))
504faa2dbf0SJosef Bacik 		return 0;
505fccb84c9SDavid Sterba 
506d1980131SDavid Sterba 	/* Ensure we can see the FORCE_COW bit */
507d1980131SDavid Sterba 	smp_mb__before_atomic();
508f1ebcc74SLiu Bo 
509f1ebcc74SLiu Bo 	/*
510f1ebcc74SLiu Bo 	 * We do not need to cow a block if
511f1ebcc74SLiu Bo 	 * 1) this block is not created or changed in this transaction;
512f1ebcc74SLiu Bo 	 * 2) this block does not belong to TREE_RELOC tree;
513f1ebcc74SLiu Bo 	 * 3) the root is not forced COW.
514f1ebcc74SLiu Bo 	 *
515f1ebcc74SLiu Bo 	 * What is forced COW:
51601327610SNicholas D Steeves 	 *    when we create snapshot during committing the transaction,
51752042d8eSAndrea Gelmini 	 *    after we've finished copying src root, we must COW the shared
518f1ebcc74SLiu Bo 	 *    block to ensure the metadata consistency.
519f1ebcc74SLiu Bo 	 */
5205d4f98a2SYan Zheng 	if (btrfs_header_generation(buf) == trans->transid &&
5215d4f98a2SYan Zheng 	    !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN) &&
5225d4f98a2SYan Zheng 	    !(root->root_key.objectid != BTRFS_TREE_RELOC_OBJECTID &&
523f1ebcc74SLiu Bo 	      btrfs_header_flag(buf, BTRFS_HEADER_FLAG_RELOC)) &&
52427cdeb70SMiao Xie 	    !test_bit(BTRFS_ROOT_FORCE_COW, &root->state))
5255d4f98a2SYan Zheng 		return 0;
5265d4f98a2SYan Zheng 	return 1;
5275d4f98a2SYan Zheng }
5285d4f98a2SYan Zheng 
529d352ac68SChris Mason /*
530d352ac68SChris Mason  * cows a single block, see __btrfs_cow_block for the real work.
53101327610SNicholas D Steeves  * This version of it has extra checks so that a block isn't COWed more than
532d352ac68SChris Mason  * once per transaction, as long as it hasn't been written yet
533d352ac68SChris Mason  */
534d397712bSChris Mason noinline int btrfs_cow_block(struct btrfs_trans_handle *trans,
5355f39d397SChris Mason 		    struct btrfs_root *root, struct extent_buffer *buf,
5365f39d397SChris Mason 		    struct extent_buffer *parent, int parent_slot,
5379631e4ccSJosef Bacik 		    struct extent_buffer **cow_ret,
5389631e4ccSJosef Bacik 		    enum btrfs_lock_nesting nest)
53902217ed2SChris Mason {
5400b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
5416702ed49SChris Mason 	u64 search_start;
542f510cfecSChris Mason 	int ret;
543dc17ff8fSChris Mason 
54483354f07SJosef Bacik 	if (test_bit(BTRFS_ROOT_DELETING, &root->state))
54583354f07SJosef Bacik 		btrfs_err(fs_info,
54683354f07SJosef Bacik 			"COW'ing blocks on a fs root that's being dropped");
54783354f07SJosef Bacik 
5480b246afaSJeff Mahoney 	if (trans->transaction != fs_info->running_transaction)
54931b1a2bdSJulia Lawall 		WARN(1, KERN_CRIT "trans %llu running %llu\n",
550c1c9ff7cSGeert Uytterhoeven 		       trans->transid,
5510b246afaSJeff Mahoney 		       fs_info->running_transaction->transid);
55231b1a2bdSJulia Lawall 
5530b246afaSJeff Mahoney 	if (trans->transid != fs_info->generation)
55431b1a2bdSJulia Lawall 		WARN(1, KERN_CRIT "trans %llu running %llu\n",
5550b246afaSJeff Mahoney 		       trans->transid, fs_info->generation);
556dc17ff8fSChris Mason 
5575d4f98a2SYan Zheng 	if (!should_cow_block(trans, root, buf)) {
55802217ed2SChris Mason 		*cow_ret = buf;
55902217ed2SChris Mason 		return 0;
56002217ed2SChris Mason 	}
561c487685dSChris Mason 
562ee22184bSByongho Lee 	search_start = buf->start & ~((u64)SZ_1G - 1);
563b4ce94deSChris Mason 
564f616f5cdSQu Wenruo 	/*
565f616f5cdSQu Wenruo 	 * Before CoWing this block for later modification, check if it's
566f616f5cdSQu Wenruo 	 * the subtree root and do the delayed subtree trace if needed.
567f616f5cdSQu Wenruo 	 *
568f616f5cdSQu Wenruo 	 * Also We don't care about the error, as it's handled internally.
569f616f5cdSQu Wenruo 	 */
570f616f5cdSQu Wenruo 	btrfs_qgroup_trace_subtree_after_cow(trans, root, buf);
571f510cfecSChris Mason 	ret = __btrfs_cow_block(trans, root, buf, parent,
5729631e4ccSJosef Bacik 				 parent_slot, cow_ret, search_start, 0, nest);
5731abe9b8aSliubo 
5741abe9b8aSliubo 	trace_btrfs_cow_block(root, buf, *cow_ret);
5751abe9b8aSliubo 
576f510cfecSChris Mason 	return ret;
5772c90e5d6SChris Mason }
578f75e2b79SJosef Bacik ALLOW_ERROR_INJECTION(btrfs_cow_block, ERRNO);
5796702ed49SChris Mason 
580d352ac68SChris Mason /*
581d352ac68SChris Mason  * helper function for defrag to decide if two blocks pointed to by a
582d352ac68SChris Mason  * node are actually close by
583d352ac68SChris Mason  */
5846b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
5856702ed49SChris Mason {
5866b80053dSChris Mason 	if (blocknr < other && other - (blocknr + blocksize) < 32768)
5876702ed49SChris Mason 		return 1;
5886b80053dSChris Mason 	if (blocknr > other && blocknr - (other + blocksize) < 32768)
5896702ed49SChris Mason 		return 1;
59002217ed2SChris Mason 	return 0;
59102217ed2SChris Mason }
59202217ed2SChris Mason 
593ce6ef5abSDavid Sterba #ifdef __LITTLE_ENDIAN
594ce6ef5abSDavid Sterba 
595ce6ef5abSDavid Sterba /*
596ce6ef5abSDavid Sterba  * Compare two keys, on little-endian the disk order is same as CPU order and
597ce6ef5abSDavid Sterba  * we can avoid the conversion.
598ce6ef5abSDavid Sterba  */
599ce6ef5abSDavid Sterba static int comp_keys(const struct btrfs_disk_key *disk_key,
600ce6ef5abSDavid Sterba 		     const struct btrfs_key *k2)
601ce6ef5abSDavid Sterba {
602ce6ef5abSDavid Sterba 	const struct btrfs_key *k1 = (const struct btrfs_key *)disk_key;
603ce6ef5abSDavid Sterba 
604ce6ef5abSDavid Sterba 	return btrfs_comp_cpu_keys(k1, k2);
605ce6ef5abSDavid Sterba }
606ce6ef5abSDavid Sterba 
607ce6ef5abSDavid Sterba #else
608ce6ef5abSDavid Sterba 
609081e9573SChris Mason /*
610081e9573SChris Mason  * compare two keys in a memcmp fashion
611081e9573SChris Mason  */
612310712b2SOmar Sandoval static int comp_keys(const struct btrfs_disk_key *disk,
613310712b2SOmar Sandoval 		     const struct btrfs_key *k2)
614081e9573SChris Mason {
615081e9573SChris Mason 	struct btrfs_key k1;
616081e9573SChris Mason 
617081e9573SChris Mason 	btrfs_disk_key_to_cpu(&k1, disk);
618081e9573SChris Mason 
61920736abaSDiego Calleja 	return btrfs_comp_cpu_keys(&k1, k2);
620081e9573SChris Mason }
621ce6ef5abSDavid Sterba #endif
622081e9573SChris Mason 
623f3465ca4SJosef Bacik /*
624f3465ca4SJosef Bacik  * same as comp_keys only with two btrfs_key's
625f3465ca4SJosef Bacik  */
626e1f60a65SDavid Sterba int __pure btrfs_comp_cpu_keys(const struct btrfs_key *k1, const struct btrfs_key *k2)
627f3465ca4SJosef Bacik {
628f3465ca4SJosef Bacik 	if (k1->objectid > k2->objectid)
629f3465ca4SJosef Bacik 		return 1;
630f3465ca4SJosef Bacik 	if (k1->objectid < k2->objectid)
631f3465ca4SJosef Bacik 		return -1;
632f3465ca4SJosef Bacik 	if (k1->type > k2->type)
633f3465ca4SJosef Bacik 		return 1;
634f3465ca4SJosef Bacik 	if (k1->type < k2->type)
635f3465ca4SJosef Bacik 		return -1;
636f3465ca4SJosef Bacik 	if (k1->offset > k2->offset)
637f3465ca4SJosef Bacik 		return 1;
638f3465ca4SJosef Bacik 	if (k1->offset < k2->offset)
639f3465ca4SJosef Bacik 		return -1;
640f3465ca4SJosef Bacik 	return 0;
641f3465ca4SJosef Bacik }
642081e9573SChris Mason 
643d352ac68SChris Mason /*
644d352ac68SChris Mason  * this is used by the defrag code to go through all the
645d352ac68SChris Mason  * leaves pointed to by a node and reallocate them so that
646d352ac68SChris Mason  * disk order is close to key order
647d352ac68SChris Mason  */
6486702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans,
6495f39d397SChris Mason 		       struct btrfs_root *root, struct extent_buffer *parent,
650de78b51aSEric Sandeen 		       int start_slot, u64 *last_ret,
651a6b6e75eSChris Mason 		       struct btrfs_key *progress)
6526702ed49SChris Mason {
6530b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
6546b80053dSChris Mason 	struct extent_buffer *cur;
6556702ed49SChris Mason 	u64 blocknr;
656e9d0b13bSChris Mason 	u64 search_start = *last_ret;
657e9d0b13bSChris Mason 	u64 last_block = 0;
6586702ed49SChris Mason 	u64 other;
6596702ed49SChris Mason 	u32 parent_nritems;
6606702ed49SChris Mason 	int end_slot;
6616702ed49SChris Mason 	int i;
6626702ed49SChris Mason 	int err = 0;
6636b80053dSChris Mason 	u32 blocksize;
664081e9573SChris Mason 	int progress_passed = 0;
665081e9573SChris Mason 	struct btrfs_disk_key disk_key;
6666702ed49SChris Mason 
6670b246afaSJeff Mahoney 	WARN_ON(trans->transaction != fs_info->running_transaction);
6680b246afaSJeff Mahoney 	WARN_ON(trans->transid != fs_info->generation);
66986479a04SChris Mason 
6706b80053dSChris Mason 	parent_nritems = btrfs_header_nritems(parent);
6710b246afaSJeff Mahoney 	blocksize = fs_info->nodesize;
6725dfe2be7SFilipe Manana 	end_slot = parent_nritems - 1;
6736702ed49SChris Mason 
6745dfe2be7SFilipe Manana 	if (parent_nritems <= 1)
6756702ed49SChris Mason 		return 0;
6766702ed49SChris Mason 
6775dfe2be7SFilipe Manana 	for (i = start_slot; i <= end_slot; i++) {
6786702ed49SChris Mason 		int close = 1;
679a6b6e75eSChris Mason 
680081e9573SChris Mason 		btrfs_node_key(parent, &disk_key, i);
681081e9573SChris Mason 		if (!progress_passed && comp_keys(&disk_key, progress) < 0)
682081e9573SChris Mason 			continue;
683081e9573SChris Mason 
684081e9573SChris Mason 		progress_passed = 1;
6856b80053dSChris Mason 		blocknr = btrfs_node_blockptr(parent, i);
686e9d0b13bSChris Mason 		if (last_block == 0)
687e9d0b13bSChris Mason 			last_block = blocknr;
6885708b959SChris Mason 
6896702ed49SChris Mason 		if (i > 0) {
6906b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i - 1);
6916b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
6926702ed49SChris Mason 		}
6935dfe2be7SFilipe Manana 		if (!close && i < end_slot) {
6946b80053dSChris Mason 			other = btrfs_node_blockptr(parent, i + 1);
6956b80053dSChris Mason 			close = close_blocks(blocknr, other, blocksize);
6966702ed49SChris Mason 		}
697e9d0b13bSChris Mason 		if (close) {
698e9d0b13bSChris Mason 			last_block = blocknr;
6996702ed49SChris Mason 			continue;
700e9d0b13bSChris Mason 		}
7016702ed49SChris Mason 
702206983b7SJosef Bacik 		cur = btrfs_read_node_slot(parent, i);
703206983b7SJosef Bacik 		if (IS_ERR(cur))
70464c043deSLiu Bo 			return PTR_ERR(cur);
705e9d0b13bSChris Mason 		if (search_start == 0)
7066b80053dSChris Mason 			search_start = last_block;
707e9d0b13bSChris Mason 
708e7a84565SChris Mason 		btrfs_tree_lock(cur);
7096b80053dSChris Mason 		err = __btrfs_cow_block(trans, root, cur, parent, i,
710e7a84565SChris Mason 					&cur, search_start,
7116b80053dSChris Mason 					min(16 * blocksize,
7129631e4ccSJosef Bacik 					    (end_slot - i) * blocksize),
7139631e4ccSJosef Bacik 					BTRFS_NESTING_COW);
714252c38f0SYan 		if (err) {
715e7a84565SChris Mason 			btrfs_tree_unlock(cur);
7166b80053dSChris Mason 			free_extent_buffer(cur);
7176702ed49SChris Mason 			break;
718252c38f0SYan 		}
719e7a84565SChris Mason 		search_start = cur->start;
720e7a84565SChris Mason 		last_block = cur->start;
721f2183bdeSChris Mason 		*last_ret = search_start;
722e7a84565SChris Mason 		btrfs_tree_unlock(cur);
723e7a84565SChris Mason 		free_extent_buffer(cur);
7246702ed49SChris Mason 	}
7256702ed49SChris Mason 	return err;
7266702ed49SChris Mason }
7276702ed49SChris Mason 
72874123bd7SChris Mason /*
729fb81212cSFilipe Manana  * Search for a key in the given extent_buffer.
7305f39d397SChris Mason  *
731fb81212cSFilipe Manana  * The lower boundary for the search is specified by the slot number @low. Use a
732fb81212cSFilipe Manana  * value of 0 to search over the whole extent buffer.
73374123bd7SChris Mason  *
734fb81212cSFilipe Manana  * The slot in the extent buffer is returned via @slot. If the key exists in the
735fb81212cSFilipe Manana  * extent buffer, then @slot will point to the slot where the key is, otherwise
736fb81212cSFilipe Manana  * it points to the slot where you would insert the key.
737fb81212cSFilipe Manana  *
738fb81212cSFilipe Manana  * Slot may point to the total number of items (i.e. one position beyond the last
739fb81212cSFilipe Manana  * key) if the key is bigger than the last key in the extent buffer.
74074123bd7SChris Mason  */
741fb81212cSFilipe Manana static noinline int generic_bin_search(struct extent_buffer *eb, int low,
74267d5e289SMarcos Paulo de Souza 				       const struct btrfs_key *key, int *slot)
743be0e5c09SChris Mason {
744fb81212cSFilipe Manana 	unsigned long p;
745fb81212cSFilipe Manana 	int item_size;
74667d5e289SMarcos Paulo de Souza 	int high = btrfs_header_nritems(eb);
747be0e5c09SChris Mason 	int ret;
7485cd17f34SDavid Sterba 	const int key_size = sizeof(struct btrfs_disk_key);
749be0e5c09SChris Mason 
7505e24e9afSLiu Bo 	if (low > high) {
7515e24e9afSLiu Bo 		btrfs_err(eb->fs_info,
7525e24e9afSLiu Bo 		 "%s: low (%d) > high (%d) eb %llu owner %llu level %d",
7535e24e9afSLiu Bo 			  __func__, low, high, eb->start,
7545e24e9afSLiu Bo 			  btrfs_header_owner(eb), btrfs_header_level(eb));
7555e24e9afSLiu Bo 		return -EINVAL;
7565e24e9afSLiu Bo 	}
7575e24e9afSLiu Bo 
758fb81212cSFilipe Manana 	if (btrfs_header_level(eb) == 0) {
759fb81212cSFilipe Manana 		p = offsetof(struct btrfs_leaf, items);
760fb81212cSFilipe Manana 		item_size = sizeof(struct btrfs_item);
761fb81212cSFilipe Manana 	} else {
762fb81212cSFilipe Manana 		p = offsetof(struct btrfs_node, ptrs);
763fb81212cSFilipe Manana 		item_size = sizeof(struct btrfs_key_ptr);
764fb81212cSFilipe Manana 	}
765fb81212cSFilipe Manana 
766be0e5c09SChris Mason 	while (low < high) {
7675cd17f34SDavid Sterba 		unsigned long oip;
7685cd17f34SDavid Sterba 		unsigned long offset;
7695cd17f34SDavid Sterba 		struct btrfs_disk_key *tmp;
7705cd17f34SDavid Sterba 		struct btrfs_disk_key unaligned;
7715cd17f34SDavid Sterba 		int mid;
7725cd17f34SDavid Sterba 
773be0e5c09SChris Mason 		mid = (low + high) / 2;
7745f39d397SChris Mason 		offset = p + mid * item_size;
7755cd17f34SDavid Sterba 		oip = offset_in_page(offset);
7765f39d397SChris Mason 
7775cd17f34SDavid Sterba 		if (oip + key_size <= PAGE_SIZE) {
778884b07d0SQu Wenruo 			const unsigned long idx = get_eb_page_index(offset);
7795cd17f34SDavid Sterba 			char *kaddr = page_address(eb->pages[idx]);
780934d375bSChris Mason 
781884b07d0SQu Wenruo 			oip = get_eb_offset_in_page(eb, offset);
7825cd17f34SDavid Sterba 			tmp = (struct btrfs_disk_key *)(kaddr + oip);
7835cd17f34SDavid Sterba 		} else {
7845cd17f34SDavid Sterba 			read_extent_buffer(eb, &unaligned, offset, key_size);
7855f39d397SChris Mason 			tmp = &unaligned;
786479965d6SChris Mason 		}
787479965d6SChris Mason 
788be0e5c09SChris Mason 		ret = comp_keys(tmp, key);
789be0e5c09SChris Mason 
790be0e5c09SChris Mason 		if (ret < 0)
791be0e5c09SChris Mason 			low = mid + 1;
792be0e5c09SChris Mason 		else if (ret > 0)
793be0e5c09SChris Mason 			high = mid;
794be0e5c09SChris Mason 		else {
795be0e5c09SChris Mason 			*slot = mid;
796be0e5c09SChris Mason 			return 0;
797be0e5c09SChris Mason 		}
798be0e5c09SChris Mason 	}
799be0e5c09SChris Mason 	*slot = low;
800be0e5c09SChris Mason 	return 1;
801be0e5c09SChris Mason }
802be0e5c09SChris Mason 
80397571fd0SChris Mason /*
804fb81212cSFilipe Manana  * Simple binary search on an extent buffer. Works for both leaves and nodes, and
805fb81212cSFilipe Manana  * always searches over the whole range of keys (slot 0 to slot 'nritems - 1').
80697571fd0SChris Mason  */
807a74b35ecSNikolay Borisov int btrfs_bin_search(struct extent_buffer *eb, const struct btrfs_key *key,
808e3b83361SQu Wenruo 		     int *slot)
809be0e5c09SChris Mason {
810fb81212cSFilipe Manana 	return generic_bin_search(eb, 0, key, slot);
811be0e5c09SChris Mason }
812be0e5c09SChris Mason 
813f0486c68SYan, Zheng static void root_add_used(struct btrfs_root *root, u32 size)
814f0486c68SYan, Zheng {
815f0486c68SYan, Zheng 	spin_lock(&root->accounting_lock);
816f0486c68SYan, Zheng 	btrfs_set_root_used(&root->root_item,
817f0486c68SYan, Zheng 			    btrfs_root_used(&root->root_item) + size);
818f0486c68SYan, Zheng 	spin_unlock(&root->accounting_lock);
819f0486c68SYan, Zheng }
820f0486c68SYan, Zheng 
821f0486c68SYan, Zheng static void root_sub_used(struct btrfs_root *root, u32 size)
822f0486c68SYan, Zheng {
823f0486c68SYan, Zheng 	spin_lock(&root->accounting_lock);
824f0486c68SYan, Zheng 	btrfs_set_root_used(&root->root_item,
825f0486c68SYan, Zheng 			    btrfs_root_used(&root->root_item) - size);
826f0486c68SYan, Zheng 	spin_unlock(&root->accounting_lock);
827f0486c68SYan, Zheng }
828f0486c68SYan, Zheng 
829d352ac68SChris Mason /* given a node and slot number, this reads the blocks it points to.  The
830d352ac68SChris Mason  * extent buffer is returned with a reference taken (but unlocked).
831d352ac68SChris Mason  */
8324b231ae4SDavid Sterba struct extent_buffer *btrfs_read_node_slot(struct extent_buffer *parent,
8334b231ae4SDavid Sterba 					   int slot)
834bb803951SChris Mason {
835ca7a79adSChris Mason 	int level = btrfs_header_level(parent);
836416bc658SJosef Bacik 	struct extent_buffer *eb;
837581c1760SQu Wenruo 	struct btrfs_key first_key;
838416bc658SJosef Bacik 
839fb770ae4SLiu Bo 	if (slot < 0 || slot >= btrfs_header_nritems(parent))
840fb770ae4SLiu Bo 		return ERR_PTR(-ENOENT);
841ca7a79adSChris Mason 
842ca7a79adSChris Mason 	BUG_ON(level == 0);
843ca7a79adSChris Mason 
844581c1760SQu Wenruo 	btrfs_node_key_to_cpu(parent, &first_key, slot);
845d0d20b0fSDavid Sterba 	eb = read_tree_block(parent->fs_info, btrfs_node_blockptr(parent, slot),
8461b7ec85eSJosef Bacik 			     btrfs_header_owner(parent),
847581c1760SQu Wenruo 			     btrfs_node_ptr_generation(parent, slot),
848581c1760SQu Wenruo 			     level - 1, &first_key);
849fb770ae4SLiu Bo 	if (!IS_ERR(eb) && !extent_buffer_uptodate(eb)) {
850416bc658SJosef Bacik 		free_extent_buffer(eb);
851fb770ae4SLiu Bo 		eb = ERR_PTR(-EIO);
852416bc658SJosef Bacik 	}
853416bc658SJosef Bacik 
854416bc658SJosef Bacik 	return eb;
855bb803951SChris Mason }
856bb803951SChris Mason 
857d352ac68SChris Mason /*
858d352ac68SChris Mason  * node level balancing, used to make sure nodes are in proper order for
859d352ac68SChris Mason  * item deletion.  We balance from the top down, so we have to make sure
860d352ac68SChris Mason  * that a deletion won't leave an node completely empty later on.
861d352ac68SChris Mason  */
862e02119d5SChris Mason static noinline int balance_level(struct btrfs_trans_handle *trans,
86398ed5174SChris Mason 			 struct btrfs_root *root,
86498ed5174SChris Mason 			 struct btrfs_path *path, int level)
865bb803951SChris Mason {
8660b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
8675f39d397SChris Mason 	struct extent_buffer *right = NULL;
8685f39d397SChris Mason 	struct extent_buffer *mid;
8695f39d397SChris Mason 	struct extent_buffer *left = NULL;
8705f39d397SChris Mason 	struct extent_buffer *parent = NULL;
871bb803951SChris Mason 	int ret = 0;
872bb803951SChris Mason 	int wret;
873bb803951SChris Mason 	int pslot;
874bb803951SChris Mason 	int orig_slot = path->slots[level];
87579f95c82SChris Mason 	u64 orig_ptr;
876bb803951SChris Mason 
87798e6b1ebSLiu Bo 	ASSERT(level > 0);
878bb803951SChris Mason 
8795f39d397SChris Mason 	mid = path->nodes[level];
880b4ce94deSChris Mason 
881ac5887c8SJosef Bacik 	WARN_ON(path->locks[level] != BTRFS_WRITE_LOCK);
8827bb86316SChris Mason 	WARN_ON(btrfs_header_generation(mid) != trans->transid);
8837bb86316SChris Mason 
8841d4f8a0cSChris Mason 	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
88579f95c82SChris Mason 
886a05a9bb1SLi Zefan 	if (level < BTRFS_MAX_LEVEL - 1) {
8875f39d397SChris Mason 		parent = path->nodes[level + 1];
888bb803951SChris Mason 		pslot = path->slots[level + 1];
889a05a9bb1SLi Zefan 	}
890bb803951SChris Mason 
89140689478SChris Mason 	/*
89240689478SChris Mason 	 * deal with the case where there is only one pointer in the root
89340689478SChris Mason 	 * by promoting the node below to a root
89440689478SChris Mason 	 */
8955f39d397SChris Mason 	if (!parent) {
8965f39d397SChris Mason 		struct extent_buffer *child;
897bb803951SChris Mason 
8985f39d397SChris Mason 		if (btrfs_header_nritems(mid) != 1)
899bb803951SChris Mason 			return 0;
900bb803951SChris Mason 
901bb803951SChris Mason 		/* promote the child to a root */
9024b231ae4SDavid Sterba 		child = btrfs_read_node_slot(mid, 0);
903fb770ae4SLiu Bo 		if (IS_ERR(child)) {
904fb770ae4SLiu Bo 			ret = PTR_ERR(child);
9050b246afaSJeff Mahoney 			btrfs_handle_fs_error(fs_info, ret, NULL);
906305a26afSMark Fasheh 			goto enospc;
907305a26afSMark Fasheh 		}
908305a26afSMark Fasheh 
909925baeddSChris Mason 		btrfs_tree_lock(child);
9109631e4ccSJosef Bacik 		ret = btrfs_cow_block(trans, root, child, mid, 0, &child,
9119631e4ccSJosef Bacik 				      BTRFS_NESTING_COW);
912f0486c68SYan, Zheng 		if (ret) {
913f0486c68SYan, Zheng 			btrfs_tree_unlock(child);
914f0486c68SYan, Zheng 			free_extent_buffer(child);
915f0486c68SYan, Zheng 			goto enospc;
916f0486c68SYan, Zheng 		}
9172f375ab9SYan 
918406808abSFilipe Manana 		ret = btrfs_tree_mod_log_insert_root(root->node, child, true);
919d9d19a01SDavid Sterba 		BUG_ON(ret < 0);
920240f62c8SChris Mason 		rcu_assign_pointer(root->node, child);
921925baeddSChris Mason 
9220b86a832SChris Mason 		add_root_to_dirty_list(root);
923925baeddSChris Mason 		btrfs_tree_unlock(child);
924b4ce94deSChris Mason 
925925baeddSChris Mason 		path->locks[level] = 0;
926bb803951SChris Mason 		path->nodes[level] = NULL;
9276a884d7dSDavid Sterba 		btrfs_clean_tree_block(mid);
928925baeddSChris Mason 		btrfs_tree_unlock(mid);
929bb803951SChris Mason 		/* once for the path */
9305f39d397SChris Mason 		free_extent_buffer(mid);
931f0486c68SYan, Zheng 
932f0486c68SYan, Zheng 		root_sub_used(root, mid->len);
9337a163608SFilipe Manana 		btrfs_free_tree_block(trans, btrfs_root_id(root), mid, 0, 1);
934bb803951SChris Mason 		/* once for the root ptr */
9353083ee2eSJosef Bacik 		free_extent_buffer_stale(mid);
936f0486c68SYan, Zheng 		return 0;
937bb803951SChris Mason 	}
9385f39d397SChris Mason 	if (btrfs_header_nritems(mid) >
9390b246afaSJeff Mahoney 	    BTRFS_NODEPTRS_PER_BLOCK(fs_info) / 4)
940bb803951SChris Mason 		return 0;
941bb803951SChris Mason 
9424b231ae4SDavid Sterba 	left = btrfs_read_node_slot(parent, pslot - 1);
943fb770ae4SLiu Bo 	if (IS_ERR(left))
944fb770ae4SLiu Bo 		left = NULL;
945fb770ae4SLiu Bo 
9465f39d397SChris Mason 	if (left) {
947bf77467aSJosef Bacik 		__btrfs_tree_lock(left, BTRFS_NESTING_LEFT);
9485f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, left,
9499631e4ccSJosef Bacik 				       parent, pslot - 1, &left,
950bf59a5a2SJosef Bacik 				       BTRFS_NESTING_LEFT_COW);
95154aa1f4dSChris Mason 		if (wret) {
95254aa1f4dSChris Mason 			ret = wret;
95354aa1f4dSChris Mason 			goto enospc;
95454aa1f4dSChris Mason 		}
9552cc58cf2SChris Mason 	}
956fb770ae4SLiu Bo 
9574b231ae4SDavid Sterba 	right = btrfs_read_node_slot(parent, pslot + 1);
958fb770ae4SLiu Bo 	if (IS_ERR(right))
959fb770ae4SLiu Bo 		right = NULL;
960fb770ae4SLiu Bo 
9615f39d397SChris Mason 	if (right) {
962bf77467aSJosef Bacik 		__btrfs_tree_lock(right, BTRFS_NESTING_RIGHT);
9635f39d397SChris Mason 		wret = btrfs_cow_block(trans, root, right,
9649631e4ccSJosef Bacik 				       parent, pslot + 1, &right,
965bf59a5a2SJosef Bacik 				       BTRFS_NESTING_RIGHT_COW);
9662cc58cf2SChris Mason 		if (wret) {
9672cc58cf2SChris Mason 			ret = wret;
9682cc58cf2SChris Mason 			goto enospc;
9692cc58cf2SChris Mason 		}
9702cc58cf2SChris Mason 	}
9712cc58cf2SChris Mason 
9722cc58cf2SChris Mason 	/* first, try to make some room in the middle buffer */
9735f39d397SChris Mason 	if (left) {
9745f39d397SChris Mason 		orig_slot += btrfs_header_nritems(left);
975d30a668fSDavid Sterba 		wret = push_node_left(trans, left, mid, 1);
97679f95c82SChris Mason 		if (wret < 0)
97779f95c82SChris Mason 			ret = wret;
978bb803951SChris Mason 	}
97979f95c82SChris Mason 
98079f95c82SChris Mason 	/*
98179f95c82SChris Mason 	 * then try to empty the right most buffer into the middle
98279f95c82SChris Mason 	 */
9835f39d397SChris Mason 	if (right) {
984d30a668fSDavid Sterba 		wret = push_node_left(trans, mid, right, 1);
98554aa1f4dSChris Mason 		if (wret < 0 && wret != -ENOSPC)
98679f95c82SChris Mason 			ret = wret;
9875f39d397SChris Mason 		if (btrfs_header_nritems(right) == 0) {
9886a884d7dSDavid Sterba 			btrfs_clean_tree_block(right);
989925baeddSChris Mason 			btrfs_tree_unlock(right);
990afe5fea7STsutomu Itoh 			del_ptr(root, path, level + 1, pslot + 1);
991f0486c68SYan, Zheng 			root_sub_used(root, right->len);
9927a163608SFilipe Manana 			btrfs_free_tree_block(trans, btrfs_root_id(root), right,
9937a163608SFilipe Manana 					      0, 1);
9943083ee2eSJosef Bacik 			free_extent_buffer_stale(right);
995f0486c68SYan, Zheng 			right = NULL;
996bb803951SChris Mason 		} else {
9975f39d397SChris Mason 			struct btrfs_disk_key right_key;
9985f39d397SChris Mason 			btrfs_node_key(right, &right_key, 0);
999f3a84ccdSFilipe Manana 			ret = btrfs_tree_mod_log_insert_key(parent, pslot + 1,
1000f3a84ccdSFilipe Manana 					BTRFS_MOD_LOG_KEY_REPLACE, GFP_NOFS);
10010e82bcfeSDavid Sterba 			BUG_ON(ret < 0);
10025f39d397SChris Mason 			btrfs_set_node_key(parent, &right_key, pslot + 1);
10035f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
1004bb803951SChris Mason 		}
1005bb803951SChris Mason 	}
10065f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 1) {
100779f95c82SChris Mason 		/*
100879f95c82SChris Mason 		 * we're not allowed to leave a node with one item in the
100979f95c82SChris Mason 		 * tree during a delete.  A deletion from lower in the tree
101079f95c82SChris Mason 		 * could try to delete the only pointer in this node.
101179f95c82SChris Mason 		 * So, pull some keys from the left.
101279f95c82SChris Mason 		 * There has to be a left pointer at this point because
101379f95c82SChris Mason 		 * otherwise we would have pulled some pointers from the
101479f95c82SChris Mason 		 * right
101579f95c82SChris Mason 		 */
1016305a26afSMark Fasheh 		if (!left) {
1017305a26afSMark Fasheh 			ret = -EROFS;
10180b246afaSJeff Mahoney 			btrfs_handle_fs_error(fs_info, ret, NULL);
1019305a26afSMark Fasheh 			goto enospc;
1020305a26afSMark Fasheh 		}
102155d32ed8SDavid Sterba 		wret = balance_node_right(trans, mid, left);
102254aa1f4dSChris Mason 		if (wret < 0) {
102379f95c82SChris Mason 			ret = wret;
102454aa1f4dSChris Mason 			goto enospc;
102554aa1f4dSChris Mason 		}
1026bce4eae9SChris Mason 		if (wret == 1) {
1027d30a668fSDavid Sterba 			wret = push_node_left(trans, left, mid, 1);
1028bce4eae9SChris Mason 			if (wret < 0)
1029bce4eae9SChris Mason 				ret = wret;
1030bce4eae9SChris Mason 		}
103179f95c82SChris Mason 		BUG_ON(wret == 1);
103279f95c82SChris Mason 	}
10335f39d397SChris Mason 	if (btrfs_header_nritems(mid) == 0) {
10346a884d7dSDavid Sterba 		btrfs_clean_tree_block(mid);
1035925baeddSChris Mason 		btrfs_tree_unlock(mid);
1036afe5fea7STsutomu Itoh 		del_ptr(root, path, level + 1, pslot);
1037f0486c68SYan, Zheng 		root_sub_used(root, mid->len);
10387a163608SFilipe Manana 		btrfs_free_tree_block(trans, btrfs_root_id(root), mid, 0, 1);
10393083ee2eSJosef Bacik 		free_extent_buffer_stale(mid);
1040f0486c68SYan, Zheng 		mid = NULL;
104179f95c82SChris Mason 	} else {
104279f95c82SChris Mason 		/* update the parent key to reflect our changes */
10435f39d397SChris Mason 		struct btrfs_disk_key mid_key;
10445f39d397SChris Mason 		btrfs_node_key(mid, &mid_key, 0);
1045f3a84ccdSFilipe Manana 		ret = btrfs_tree_mod_log_insert_key(parent, pslot,
1046f3a84ccdSFilipe Manana 				BTRFS_MOD_LOG_KEY_REPLACE, GFP_NOFS);
10470e82bcfeSDavid Sterba 		BUG_ON(ret < 0);
10485f39d397SChris Mason 		btrfs_set_node_key(parent, &mid_key, pslot);
10495f39d397SChris Mason 		btrfs_mark_buffer_dirty(parent);
105079f95c82SChris Mason 	}
1051bb803951SChris Mason 
105279f95c82SChris Mason 	/* update the path */
10535f39d397SChris Mason 	if (left) {
10545f39d397SChris Mason 		if (btrfs_header_nritems(left) > orig_slot) {
105567439dadSDavid Sterba 			atomic_inc(&left->refs);
1056925baeddSChris Mason 			/* left was locked after cow */
10575f39d397SChris Mason 			path->nodes[level] = left;
1058bb803951SChris Mason 			path->slots[level + 1] -= 1;
1059bb803951SChris Mason 			path->slots[level] = orig_slot;
1060925baeddSChris Mason 			if (mid) {
1061925baeddSChris Mason 				btrfs_tree_unlock(mid);
10625f39d397SChris Mason 				free_extent_buffer(mid);
1063925baeddSChris Mason 			}
1064bb803951SChris Mason 		} else {
10655f39d397SChris Mason 			orig_slot -= btrfs_header_nritems(left);
1066bb803951SChris Mason 			path->slots[level] = orig_slot;
1067bb803951SChris Mason 		}
1068bb803951SChris Mason 	}
106979f95c82SChris Mason 	/* double check we haven't messed things up */
1070e20d96d6SChris Mason 	if (orig_ptr !=
10715f39d397SChris Mason 	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
107279f95c82SChris Mason 		BUG();
107354aa1f4dSChris Mason enospc:
1074925baeddSChris Mason 	if (right) {
1075925baeddSChris Mason 		btrfs_tree_unlock(right);
10765f39d397SChris Mason 		free_extent_buffer(right);
1077925baeddSChris Mason 	}
1078925baeddSChris Mason 	if (left) {
1079925baeddSChris Mason 		if (path->nodes[level] != left)
1080925baeddSChris Mason 			btrfs_tree_unlock(left);
10815f39d397SChris Mason 		free_extent_buffer(left);
1082925baeddSChris Mason 	}
1083bb803951SChris Mason 	return ret;
1084bb803951SChris Mason }
1085bb803951SChris Mason 
1086d352ac68SChris Mason /* Node balancing for insertion.  Here we only split or push nodes around
1087d352ac68SChris Mason  * when they are completely full.  This is also done top down, so we
1088d352ac68SChris Mason  * have to be pessimistic.
1089d352ac68SChris Mason  */
1090d397712bSChris Mason static noinline int push_nodes_for_insert(struct btrfs_trans_handle *trans,
1091e66f709bSChris Mason 					  struct btrfs_root *root,
1092e66f709bSChris Mason 					  struct btrfs_path *path, int level)
1093e66f709bSChris Mason {
10940b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
10955f39d397SChris Mason 	struct extent_buffer *right = NULL;
10965f39d397SChris Mason 	struct extent_buffer *mid;
10975f39d397SChris Mason 	struct extent_buffer *left = NULL;
10985f39d397SChris Mason 	struct extent_buffer *parent = NULL;
1099e66f709bSChris Mason 	int ret = 0;
1100e66f709bSChris Mason 	int wret;
1101e66f709bSChris Mason 	int pslot;
1102e66f709bSChris Mason 	int orig_slot = path->slots[level];
1103e66f709bSChris Mason 
1104e66f709bSChris Mason 	if (level == 0)
1105e66f709bSChris Mason 		return 1;
1106e66f709bSChris Mason 
11075f39d397SChris Mason 	mid = path->nodes[level];
11087bb86316SChris Mason 	WARN_ON(btrfs_header_generation(mid) != trans->transid);
1109e66f709bSChris Mason 
1110a05a9bb1SLi Zefan 	if (level < BTRFS_MAX_LEVEL - 1) {
11115f39d397SChris Mason 		parent = path->nodes[level + 1];
1112e66f709bSChris Mason 		pslot = path->slots[level + 1];
1113a05a9bb1SLi Zefan 	}
1114e66f709bSChris Mason 
11155f39d397SChris Mason 	if (!parent)
1116e66f709bSChris Mason 		return 1;
1117e66f709bSChris Mason 
11184b231ae4SDavid Sterba 	left = btrfs_read_node_slot(parent, pslot - 1);
1119fb770ae4SLiu Bo 	if (IS_ERR(left))
1120fb770ae4SLiu Bo 		left = NULL;
1121e66f709bSChris Mason 
1122e66f709bSChris Mason 	/* first, try to make some room in the middle buffer */
11235f39d397SChris Mason 	if (left) {
1124e66f709bSChris Mason 		u32 left_nr;
1125925baeddSChris Mason 
1126bf77467aSJosef Bacik 		__btrfs_tree_lock(left, BTRFS_NESTING_LEFT);
1127b4ce94deSChris Mason 
11285f39d397SChris Mason 		left_nr = btrfs_header_nritems(left);
11290b246afaSJeff Mahoney 		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(fs_info) - 1) {
113033ade1f8SChris Mason 			wret = 1;
113133ade1f8SChris Mason 		} else {
11325f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, left, parent,
11339631e4ccSJosef Bacik 					      pslot - 1, &left,
1134bf59a5a2SJosef Bacik 					      BTRFS_NESTING_LEFT_COW);
113554aa1f4dSChris Mason 			if (ret)
113654aa1f4dSChris Mason 				wret = 1;
113754aa1f4dSChris Mason 			else {
1138d30a668fSDavid Sterba 				wret = push_node_left(trans, left, mid, 0);
113954aa1f4dSChris Mason 			}
114033ade1f8SChris Mason 		}
1141e66f709bSChris Mason 		if (wret < 0)
1142e66f709bSChris Mason 			ret = wret;
1143e66f709bSChris Mason 		if (wret == 0) {
11445f39d397SChris Mason 			struct btrfs_disk_key disk_key;
1145e66f709bSChris Mason 			orig_slot += left_nr;
11465f39d397SChris Mason 			btrfs_node_key(mid, &disk_key, 0);
1147f3a84ccdSFilipe Manana 			ret = btrfs_tree_mod_log_insert_key(parent, pslot,
1148f3a84ccdSFilipe Manana 					BTRFS_MOD_LOG_KEY_REPLACE, GFP_NOFS);
11490e82bcfeSDavid Sterba 			BUG_ON(ret < 0);
11505f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot);
11515f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
11525f39d397SChris Mason 			if (btrfs_header_nritems(left) > orig_slot) {
11535f39d397SChris Mason 				path->nodes[level] = left;
1154e66f709bSChris Mason 				path->slots[level + 1] -= 1;
1155e66f709bSChris Mason 				path->slots[level] = orig_slot;
1156925baeddSChris Mason 				btrfs_tree_unlock(mid);
11575f39d397SChris Mason 				free_extent_buffer(mid);
1158e66f709bSChris Mason 			} else {
1159e66f709bSChris Mason 				orig_slot -=
11605f39d397SChris Mason 					btrfs_header_nritems(left);
1161e66f709bSChris Mason 				path->slots[level] = orig_slot;
1162925baeddSChris Mason 				btrfs_tree_unlock(left);
11635f39d397SChris Mason 				free_extent_buffer(left);
1164e66f709bSChris Mason 			}
1165e66f709bSChris Mason 			return 0;
1166e66f709bSChris Mason 		}
1167925baeddSChris Mason 		btrfs_tree_unlock(left);
11685f39d397SChris Mason 		free_extent_buffer(left);
1169e66f709bSChris Mason 	}
11704b231ae4SDavid Sterba 	right = btrfs_read_node_slot(parent, pslot + 1);
1171fb770ae4SLiu Bo 	if (IS_ERR(right))
1172fb770ae4SLiu Bo 		right = NULL;
1173e66f709bSChris Mason 
1174e66f709bSChris Mason 	/*
1175e66f709bSChris Mason 	 * then try to empty the right most buffer into the middle
1176e66f709bSChris Mason 	 */
11775f39d397SChris Mason 	if (right) {
117833ade1f8SChris Mason 		u32 right_nr;
1179b4ce94deSChris Mason 
1180bf77467aSJosef Bacik 		__btrfs_tree_lock(right, BTRFS_NESTING_RIGHT);
1181b4ce94deSChris Mason 
11825f39d397SChris Mason 		right_nr = btrfs_header_nritems(right);
11830b246afaSJeff Mahoney 		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(fs_info) - 1) {
118433ade1f8SChris Mason 			wret = 1;
118533ade1f8SChris Mason 		} else {
11865f39d397SChris Mason 			ret = btrfs_cow_block(trans, root, right,
11875f39d397SChris Mason 					      parent, pslot + 1,
1188bf59a5a2SJosef Bacik 					      &right, BTRFS_NESTING_RIGHT_COW);
118954aa1f4dSChris Mason 			if (ret)
119054aa1f4dSChris Mason 				wret = 1;
119154aa1f4dSChris Mason 			else {
119255d32ed8SDavid Sterba 				wret = balance_node_right(trans, right, mid);
119333ade1f8SChris Mason 			}
119454aa1f4dSChris Mason 		}
1195e66f709bSChris Mason 		if (wret < 0)
1196e66f709bSChris Mason 			ret = wret;
1197e66f709bSChris Mason 		if (wret == 0) {
11985f39d397SChris Mason 			struct btrfs_disk_key disk_key;
11995f39d397SChris Mason 
12005f39d397SChris Mason 			btrfs_node_key(right, &disk_key, 0);
1201f3a84ccdSFilipe Manana 			ret = btrfs_tree_mod_log_insert_key(parent, pslot + 1,
1202f3a84ccdSFilipe Manana 					BTRFS_MOD_LOG_KEY_REPLACE, GFP_NOFS);
12030e82bcfeSDavid Sterba 			BUG_ON(ret < 0);
12045f39d397SChris Mason 			btrfs_set_node_key(parent, &disk_key, pslot + 1);
12055f39d397SChris Mason 			btrfs_mark_buffer_dirty(parent);
12065f39d397SChris Mason 
12075f39d397SChris Mason 			if (btrfs_header_nritems(mid) <= orig_slot) {
12085f39d397SChris Mason 				path->nodes[level] = right;
1209e66f709bSChris Mason 				path->slots[level + 1] += 1;
1210e66f709bSChris Mason 				path->slots[level] = orig_slot -
12115f39d397SChris Mason 					btrfs_header_nritems(mid);
1212925baeddSChris Mason 				btrfs_tree_unlock(mid);
12135f39d397SChris Mason 				free_extent_buffer(mid);
1214e66f709bSChris Mason 			} else {
1215925baeddSChris Mason 				btrfs_tree_unlock(right);
12165f39d397SChris Mason 				free_extent_buffer(right);
1217e66f709bSChris Mason 			}
1218e66f709bSChris Mason 			return 0;
1219e66f709bSChris Mason 		}
1220925baeddSChris Mason 		btrfs_tree_unlock(right);
12215f39d397SChris Mason 		free_extent_buffer(right);
1222e66f709bSChris Mason 	}
1223e66f709bSChris Mason 	return 1;
1224e66f709bSChris Mason }
1225e66f709bSChris Mason 
122674123bd7SChris Mason /*
1227d352ac68SChris Mason  * readahead one full node of leaves, finding things that are close
1228d352ac68SChris Mason  * to the block in 'slot', and triggering ra on them.
12293c69faecSChris Mason  */
12302ff7e61eSJeff Mahoney static void reada_for_search(struct btrfs_fs_info *fs_info,
1231e02119d5SChris Mason 			     struct btrfs_path *path,
123201f46658SChris Mason 			     int level, int slot, u64 objectid)
12333c69faecSChris Mason {
12345f39d397SChris Mason 	struct extent_buffer *node;
123501f46658SChris Mason 	struct btrfs_disk_key disk_key;
12363c69faecSChris Mason 	u32 nritems;
12373c69faecSChris Mason 	u64 search;
1238a7175319SChris Mason 	u64 target;
12396b80053dSChris Mason 	u64 nread = 0;
1240ace75066SFilipe Manana 	u64 nread_max;
12416b80053dSChris Mason 	u32 nr;
12426b80053dSChris Mason 	u32 blocksize;
12436b80053dSChris Mason 	u32 nscan = 0;
1244db94535dSChris Mason 
1245ace75066SFilipe Manana 	if (level != 1 && path->reada != READA_FORWARD_ALWAYS)
12463c69faecSChris Mason 		return;
12473c69faecSChris Mason 
12486702ed49SChris Mason 	if (!path->nodes[level])
12496702ed49SChris Mason 		return;
12506702ed49SChris Mason 
12515f39d397SChris Mason 	node = path->nodes[level];
1252925baeddSChris Mason 
1253ace75066SFilipe Manana 	/*
1254ace75066SFilipe Manana 	 * Since the time between visiting leaves is much shorter than the time
1255ace75066SFilipe Manana 	 * between visiting nodes, limit read ahead of nodes to 1, to avoid too
1256ace75066SFilipe Manana 	 * much IO at once (possibly random).
1257ace75066SFilipe Manana 	 */
1258ace75066SFilipe Manana 	if (path->reada == READA_FORWARD_ALWAYS) {
1259ace75066SFilipe Manana 		if (level > 1)
1260ace75066SFilipe Manana 			nread_max = node->fs_info->nodesize;
1261ace75066SFilipe Manana 		else
1262ace75066SFilipe Manana 			nread_max = SZ_128K;
1263ace75066SFilipe Manana 	} else {
1264ace75066SFilipe Manana 		nread_max = SZ_64K;
1265ace75066SFilipe Manana 	}
1266ace75066SFilipe Manana 
12673c69faecSChris Mason 	search = btrfs_node_blockptr(node, slot);
12680b246afaSJeff Mahoney 	blocksize = fs_info->nodesize;
1269069a2e37SFilipe Manana 	if (path->reada != READA_FORWARD_ALWAYS) {
1270069a2e37SFilipe Manana 		struct extent_buffer *eb;
1271069a2e37SFilipe Manana 
12720b246afaSJeff Mahoney 		eb = find_extent_buffer(fs_info, search);
12735f39d397SChris Mason 		if (eb) {
12745f39d397SChris Mason 			free_extent_buffer(eb);
12753c69faecSChris Mason 			return;
12763c69faecSChris Mason 		}
1277069a2e37SFilipe Manana 	}
12783c69faecSChris Mason 
1279a7175319SChris Mason 	target = search;
12806b80053dSChris Mason 
12815f39d397SChris Mason 	nritems = btrfs_header_nritems(node);
12826b80053dSChris Mason 	nr = slot;
128325b8b936SJosef Bacik 
12843c69faecSChris Mason 	while (1) {
1285e4058b54SDavid Sterba 		if (path->reada == READA_BACK) {
12866b80053dSChris Mason 			if (nr == 0)
12873c69faecSChris Mason 				break;
12886b80053dSChris Mason 			nr--;
1289ace75066SFilipe Manana 		} else if (path->reada == READA_FORWARD ||
1290ace75066SFilipe Manana 			   path->reada == READA_FORWARD_ALWAYS) {
12916b80053dSChris Mason 			nr++;
12926b80053dSChris Mason 			if (nr >= nritems)
12936b80053dSChris Mason 				break;
12943c69faecSChris Mason 		}
1295e4058b54SDavid Sterba 		if (path->reada == READA_BACK && objectid) {
129601f46658SChris Mason 			btrfs_node_key(node, &disk_key, nr);
129701f46658SChris Mason 			if (btrfs_disk_key_objectid(&disk_key) != objectid)
129801f46658SChris Mason 				break;
129901f46658SChris Mason 		}
13006b80053dSChris Mason 		search = btrfs_node_blockptr(node, nr);
1301ace75066SFilipe Manana 		if (path->reada == READA_FORWARD_ALWAYS ||
1302ace75066SFilipe Manana 		    (search <= target && target - search <= 65536) ||
1303a7175319SChris Mason 		    (search > target && search - target <= 65536)) {
1304bfb484d9SJosef Bacik 			btrfs_readahead_node_child(node, nr);
13056b80053dSChris Mason 			nread += blocksize;
13063c69faecSChris Mason 		}
13076b80053dSChris Mason 		nscan++;
1308ace75066SFilipe Manana 		if (nread > nread_max || nscan > 32)
13096b80053dSChris Mason 			break;
13103c69faecSChris Mason 	}
13113c69faecSChris Mason }
1312925baeddSChris Mason 
1313bfb484d9SJosef Bacik static noinline void reada_for_balance(struct btrfs_path *path, int level)
1314b4ce94deSChris Mason {
1315bfb484d9SJosef Bacik 	struct extent_buffer *parent;
1316b4ce94deSChris Mason 	int slot;
1317b4ce94deSChris Mason 	int nritems;
1318b4ce94deSChris Mason 
13198c594ea8SChris Mason 	parent = path->nodes[level + 1];
1320b4ce94deSChris Mason 	if (!parent)
13210b08851fSJosef Bacik 		return;
1322b4ce94deSChris Mason 
1323b4ce94deSChris Mason 	nritems = btrfs_header_nritems(parent);
13248c594ea8SChris Mason 	slot = path->slots[level + 1];
1325b4ce94deSChris Mason 
1326bfb484d9SJosef Bacik 	if (slot > 0)
1327bfb484d9SJosef Bacik 		btrfs_readahead_node_child(parent, slot - 1);
1328bfb484d9SJosef Bacik 	if (slot + 1 < nritems)
1329bfb484d9SJosef Bacik 		btrfs_readahead_node_child(parent, slot + 1);
1330b4ce94deSChris Mason }
1331b4ce94deSChris Mason 
1332b4ce94deSChris Mason 
1333b4ce94deSChris Mason /*
1334d397712bSChris Mason  * when we walk down the tree, it is usually safe to unlock the higher layers
1335d397712bSChris Mason  * in the tree.  The exceptions are when our path goes through slot 0, because
1336d397712bSChris Mason  * operations on the tree might require changing key pointers higher up in the
1337d397712bSChris Mason  * tree.
1338d352ac68SChris Mason  *
1339d397712bSChris Mason  * callers might also have set path->keep_locks, which tells this code to keep
1340d397712bSChris Mason  * the lock if the path points to the last slot in the block.  This is part of
1341d397712bSChris Mason  * walking through the tree, and selecting the next slot in the higher block.
1342d352ac68SChris Mason  *
1343d397712bSChris Mason  * lowest_unlock sets the lowest level in the tree we're allowed to unlock.  so
1344d397712bSChris Mason  * if lowest_unlock is 1, level 0 won't be unlocked
1345d352ac68SChris Mason  */
1346e02119d5SChris Mason static noinline void unlock_up(struct btrfs_path *path, int level,
1347f7c79f30SChris Mason 			       int lowest_unlock, int min_write_lock_level,
1348f7c79f30SChris Mason 			       int *write_lock_level)
1349925baeddSChris Mason {
1350925baeddSChris Mason 	int i;
1351925baeddSChris Mason 	int skip_level = level;
1352051e1b9fSChris Mason 	int no_skips = 0;
1353925baeddSChris Mason 	struct extent_buffer *t;
1354925baeddSChris Mason 
1355925baeddSChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1356925baeddSChris Mason 		if (!path->nodes[i])
1357925baeddSChris Mason 			break;
1358925baeddSChris Mason 		if (!path->locks[i])
1359925baeddSChris Mason 			break;
1360051e1b9fSChris Mason 		if (!no_skips && path->slots[i] == 0) {
1361925baeddSChris Mason 			skip_level = i + 1;
1362925baeddSChris Mason 			continue;
1363925baeddSChris Mason 		}
1364051e1b9fSChris Mason 		if (!no_skips && path->keep_locks) {
1365925baeddSChris Mason 			u32 nritems;
1366925baeddSChris Mason 			t = path->nodes[i];
1367925baeddSChris Mason 			nritems = btrfs_header_nritems(t);
1368051e1b9fSChris Mason 			if (nritems < 1 || path->slots[i] >= nritems - 1) {
1369925baeddSChris Mason 				skip_level = i + 1;
1370925baeddSChris Mason 				continue;
1371925baeddSChris Mason 			}
1372925baeddSChris Mason 		}
1373051e1b9fSChris Mason 		if (skip_level < i && i >= lowest_unlock)
1374051e1b9fSChris Mason 			no_skips = 1;
1375051e1b9fSChris Mason 
1376925baeddSChris Mason 		t = path->nodes[i];
1377d80bb3f9SLiu Bo 		if (i >= lowest_unlock && i > skip_level) {
1378bd681513SChris Mason 			btrfs_tree_unlock_rw(t, path->locks[i]);
1379925baeddSChris Mason 			path->locks[i] = 0;
1380f7c79f30SChris Mason 			if (write_lock_level &&
1381f7c79f30SChris Mason 			    i > min_write_lock_level &&
1382f7c79f30SChris Mason 			    i <= *write_lock_level) {
1383f7c79f30SChris Mason 				*write_lock_level = i - 1;
1384f7c79f30SChris Mason 			}
1385925baeddSChris Mason 		}
1386925baeddSChris Mason 	}
1387925baeddSChris Mason }
1388925baeddSChris Mason 
13893c69faecSChris Mason /*
1390c8c42864SChris Mason  * helper function for btrfs_search_slot.  The goal is to find a block
1391c8c42864SChris Mason  * in cache without setting the path to blocking.  If we find the block
1392c8c42864SChris Mason  * we return zero and the path is unchanged.
1393c8c42864SChris Mason  *
1394c8c42864SChris Mason  * If we can't find the block, we set the path blocking and do some
1395c8c42864SChris Mason  * reada.  -EAGAIN is returned and the search must be repeated.
1396c8c42864SChris Mason  */
1397c8c42864SChris Mason static int
1398d07b8528SLiu Bo read_block_for_search(struct btrfs_root *root, struct btrfs_path *p,
1399c8c42864SChris Mason 		      struct extent_buffer **eb_ret, int level, int slot,
1400cda79c54SDavid Sterba 		      const struct btrfs_key *key)
1401c8c42864SChris Mason {
14020b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
1403c8c42864SChris Mason 	u64 blocknr;
1404c8c42864SChris Mason 	u64 gen;
1405c8c42864SChris Mason 	struct extent_buffer *tmp;
1406581c1760SQu Wenruo 	struct btrfs_key first_key;
140776a05b35SChris Mason 	int ret;
1408581c1760SQu Wenruo 	int parent_level;
1409c8c42864SChris Mason 
1410213ff4b7SNikolay Borisov 	blocknr = btrfs_node_blockptr(*eb_ret, slot);
1411213ff4b7SNikolay Borisov 	gen = btrfs_node_ptr_generation(*eb_ret, slot);
1412213ff4b7SNikolay Borisov 	parent_level = btrfs_header_level(*eb_ret);
1413213ff4b7SNikolay Borisov 	btrfs_node_key_to_cpu(*eb_ret, &first_key, slot);
1414c8c42864SChris Mason 
14150b246afaSJeff Mahoney 	tmp = find_extent_buffer(fs_info, blocknr);
1416cb44921aSChris Mason 	if (tmp) {
1417ace75066SFilipe Manana 		if (p->reada == READA_FORWARD_ALWAYS)
1418ace75066SFilipe Manana 			reada_for_search(fs_info, p, level, slot, key->objectid);
1419ace75066SFilipe Manana 
1420b9fab919SChris Mason 		/* first we do an atomic uptodate check */
1421b9fab919SChris Mason 		if (btrfs_buffer_uptodate(tmp, gen, 1) > 0) {
1422448de471SQu Wenruo 			/*
1423448de471SQu Wenruo 			 * Do extra check for first_key, eb can be stale due to
1424448de471SQu Wenruo 			 * being cached, read from scrub, or have multiple
1425448de471SQu Wenruo 			 * parents (shared tree blocks).
1426448de471SQu Wenruo 			 */
1427e064d5e9SDavid Sterba 			if (btrfs_verify_level_key(tmp,
1428448de471SQu Wenruo 					parent_level - 1, &first_key, gen)) {
1429448de471SQu Wenruo 				free_extent_buffer(tmp);
1430448de471SQu Wenruo 				return -EUCLEAN;
1431448de471SQu Wenruo 			}
1432c8c42864SChris Mason 			*eb_ret = tmp;
1433c8c42864SChris Mason 			return 0;
1434c8c42864SChris Mason 		}
1435bdf7c00eSJosef Bacik 
1436b9fab919SChris Mason 		/* now we're allowed to do a blocking uptodate check */
1437581c1760SQu Wenruo 		ret = btrfs_read_buffer(tmp, gen, parent_level - 1, &first_key);
1438bdf7c00eSJosef Bacik 		if (!ret) {
1439cb44921aSChris Mason 			*eb_ret = tmp;
1440cb44921aSChris Mason 			return 0;
1441cb44921aSChris Mason 		}
1442cb44921aSChris Mason 		free_extent_buffer(tmp);
1443b3b4aa74SDavid Sterba 		btrfs_release_path(p);
1444cb44921aSChris Mason 		return -EIO;
1445cb44921aSChris Mason 	}
1446c8c42864SChris Mason 
1447c8c42864SChris Mason 	/*
1448c8c42864SChris Mason 	 * reduce lock contention at high levels
1449c8c42864SChris Mason 	 * of the btree by dropping locks before
145076a05b35SChris Mason 	 * we read.  Don't release the lock on the current
145176a05b35SChris Mason 	 * level because we need to walk this node to figure
145276a05b35SChris Mason 	 * out which blocks to read.
1453c8c42864SChris Mason 	 */
14548c594ea8SChris Mason 	btrfs_unlock_up_safe(p, level + 1);
14558c594ea8SChris Mason 
1456e4058b54SDavid Sterba 	if (p->reada != READA_NONE)
14572ff7e61eSJeff Mahoney 		reada_for_search(fs_info, p, level, slot, key->objectid);
1458c8c42864SChris Mason 
145976a05b35SChris Mason 	ret = -EAGAIN;
14601b7ec85eSJosef Bacik 	tmp = read_tree_block(fs_info, blocknr, root->root_key.objectid,
14611b7ec85eSJosef Bacik 			      gen, parent_level - 1, &first_key);
146264c043deSLiu Bo 	if (!IS_ERR(tmp)) {
146376a05b35SChris Mason 		/*
146476a05b35SChris Mason 		 * If the read above didn't mark this buffer up to date,
146576a05b35SChris Mason 		 * it will never end up being up to date.  Set ret to EIO now
146676a05b35SChris Mason 		 * and give up so that our caller doesn't loop forever
146776a05b35SChris Mason 		 * on our EAGAINs.
146876a05b35SChris Mason 		 */
1469e6a1d6fdSLiu Bo 		if (!extent_buffer_uptodate(tmp))
147076a05b35SChris Mason 			ret = -EIO;
1471c8c42864SChris Mason 		free_extent_buffer(tmp);
1472c871b0f2SLiu Bo 	} else {
1473c871b0f2SLiu Bo 		ret = PTR_ERR(tmp);
147476a05b35SChris Mason 	}
147502a3307aSLiu Bo 
147602a3307aSLiu Bo 	btrfs_release_path(p);
147776a05b35SChris Mason 	return ret;
1478c8c42864SChris Mason }
1479c8c42864SChris Mason 
1480c8c42864SChris Mason /*
1481c8c42864SChris Mason  * helper function for btrfs_search_slot.  This does all of the checks
1482c8c42864SChris Mason  * for node-level blocks and does any balancing required based on
1483c8c42864SChris Mason  * the ins_len.
1484c8c42864SChris Mason  *
1485c8c42864SChris Mason  * If no extra work was required, zero is returned.  If we had to
1486c8c42864SChris Mason  * drop the path, -EAGAIN is returned and btrfs_search_slot must
1487c8c42864SChris Mason  * start over
1488c8c42864SChris Mason  */
1489c8c42864SChris Mason static int
1490c8c42864SChris Mason setup_nodes_for_search(struct btrfs_trans_handle *trans,
1491c8c42864SChris Mason 		       struct btrfs_root *root, struct btrfs_path *p,
1492bd681513SChris Mason 		       struct extent_buffer *b, int level, int ins_len,
1493bd681513SChris Mason 		       int *write_lock_level)
1494c8c42864SChris Mason {
14950b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
149695b982deSNikolay Borisov 	int ret = 0;
14970b246afaSJeff Mahoney 
1498c8c42864SChris Mason 	if ((p->search_for_split || ins_len > 0) && btrfs_header_nritems(b) >=
14990b246afaSJeff Mahoney 	    BTRFS_NODEPTRS_PER_BLOCK(fs_info) - 3) {
1500c8c42864SChris Mason 
1501bd681513SChris Mason 		if (*write_lock_level < level + 1) {
1502bd681513SChris Mason 			*write_lock_level = level + 1;
1503bd681513SChris Mason 			btrfs_release_path(p);
150495b982deSNikolay Borisov 			return -EAGAIN;
1505bd681513SChris Mason 		}
1506bd681513SChris Mason 
1507bfb484d9SJosef Bacik 		reada_for_balance(p, level);
150895b982deSNikolay Borisov 		ret = split_node(trans, root, p, level);
1509c8c42864SChris Mason 
1510c8c42864SChris Mason 		b = p->nodes[level];
1511c8c42864SChris Mason 	} else if (ins_len < 0 && btrfs_header_nritems(b) <
15120b246afaSJeff Mahoney 		   BTRFS_NODEPTRS_PER_BLOCK(fs_info) / 2) {
1513c8c42864SChris Mason 
1514bd681513SChris Mason 		if (*write_lock_level < level + 1) {
1515bd681513SChris Mason 			*write_lock_level = level + 1;
1516bd681513SChris Mason 			btrfs_release_path(p);
151795b982deSNikolay Borisov 			return -EAGAIN;
1518bd681513SChris Mason 		}
1519bd681513SChris Mason 
1520bfb484d9SJosef Bacik 		reada_for_balance(p, level);
152195b982deSNikolay Borisov 		ret = balance_level(trans, root, p, level);
152295b982deSNikolay Borisov 		if (ret)
152395b982deSNikolay Borisov 			return ret;
1524c8c42864SChris Mason 
1525c8c42864SChris Mason 		b = p->nodes[level];
1526c8c42864SChris Mason 		if (!b) {
1527b3b4aa74SDavid Sterba 			btrfs_release_path(p);
152895b982deSNikolay Borisov 			return -EAGAIN;
1529c8c42864SChris Mason 		}
1530c8c42864SChris Mason 		BUG_ON(btrfs_header_nritems(b) == 1);
1531c8c42864SChris Mason 	}
1532c8c42864SChris Mason 	return ret;
1533c8c42864SChris Mason }
1534c8c42864SChris Mason 
1535381cf658SDavid Sterba int btrfs_find_item(struct btrfs_root *fs_root, struct btrfs_path *path,
1536e33d5c3dSKelley Nielsen 		u64 iobjectid, u64 ioff, u8 key_type,
1537e33d5c3dSKelley Nielsen 		struct btrfs_key *found_key)
1538e33d5c3dSKelley Nielsen {
1539e33d5c3dSKelley Nielsen 	int ret;
1540e33d5c3dSKelley Nielsen 	struct btrfs_key key;
1541e33d5c3dSKelley Nielsen 	struct extent_buffer *eb;
1542381cf658SDavid Sterba 
1543381cf658SDavid Sterba 	ASSERT(path);
15441d4c08e0SDavid Sterba 	ASSERT(found_key);
1545e33d5c3dSKelley Nielsen 
1546e33d5c3dSKelley Nielsen 	key.type = key_type;
1547e33d5c3dSKelley Nielsen 	key.objectid = iobjectid;
1548e33d5c3dSKelley Nielsen 	key.offset = ioff;
1549e33d5c3dSKelley Nielsen 
1550e33d5c3dSKelley Nielsen 	ret = btrfs_search_slot(NULL, fs_root, &key, path, 0, 0);
15511d4c08e0SDavid Sterba 	if (ret < 0)
1552e33d5c3dSKelley Nielsen 		return ret;
1553e33d5c3dSKelley Nielsen 
1554e33d5c3dSKelley Nielsen 	eb = path->nodes[0];
1555e33d5c3dSKelley Nielsen 	if (ret && path->slots[0] >= btrfs_header_nritems(eb)) {
1556e33d5c3dSKelley Nielsen 		ret = btrfs_next_leaf(fs_root, path);
1557e33d5c3dSKelley Nielsen 		if (ret)
1558e33d5c3dSKelley Nielsen 			return ret;
1559e33d5c3dSKelley Nielsen 		eb = path->nodes[0];
1560e33d5c3dSKelley Nielsen 	}
1561e33d5c3dSKelley Nielsen 
1562e33d5c3dSKelley Nielsen 	btrfs_item_key_to_cpu(eb, found_key, path->slots[0]);
1563e33d5c3dSKelley Nielsen 	if (found_key->type != key.type ||
1564e33d5c3dSKelley Nielsen 			found_key->objectid != key.objectid)
1565e33d5c3dSKelley Nielsen 		return 1;
1566e33d5c3dSKelley Nielsen 
1567e33d5c3dSKelley Nielsen 	return 0;
1568e33d5c3dSKelley Nielsen }
1569e33d5c3dSKelley Nielsen 
15701fc28d8eSLiu Bo static struct extent_buffer *btrfs_search_slot_get_root(struct btrfs_root *root,
15711fc28d8eSLiu Bo 							struct btrfs_path *p,
15721fc28d8eSLiu Bo 							int write_lock_level)
15731fc28d8eSLiu Bo {
15741fc28d8eSLiu Bo 	struct extent_buffer *b;
1575120de408SJosef Bacik 	int root_lock = 0;
15761fc28d8eSLiu Bo 	int level = 0;
15771fc28d8eSLiu Bo 
15781fc28d8eSLiu Bo 	if (p->search_commit_root) {
15791fc28d8eSLiu Bo 		b = root->commit_root;
158067439dadSDavid Sterba 		atomic_inc(&b->refs);
15811fc28d8eSLiu Bo 		level = btrfs_header_level(b);
1582f9ddfd05SLiu Bo 		/*
1583f9ddfd05SLiu Bo 		 * Ensure that all callers have set skip_locking when
1584f9ddfd05SLiu Bo 		 * p->search_commit_root = 1.
1585f9ddfd05SLiu Bo 		 */
1586f9ddfd05SLiu Bo 		ASSERT(p->skip_locking == 1);
15871fc28d8eSLiu Bo 
15881fc28d8eSLiu Bo 		goto out;
15891fc28d8eSLiu Bo 	}
15901fc28d8eSLiu Bo 
15911fc28d8eSLiu Bo 	if (p->skip_locking) {
15921fc28d8eSLiu Bo 		b = btrfs_root_node(root);
15931fc28d8eSLiu Bo 		level = btrfs_header_level(b);
15941fc28d8eSLiu Bo 		goto out;
15951fc28d8eSLiu Bo 	}
15961fc28d8eSLiu Bo 
1597120de408SJosef Bacik 	/* We try very hard to do read locks on the root */
1598120de408SJosef Bacik 	root_lock = BTRFS_READ_LOCK;
1599120de408SJosef Bacik 
16001fc28d8eSLiu Bo 	/*
1601662c653bSLiu Bo 	 * If the level is set to maximum, we can skip trying to get the read
1602662c653bSLiu Bo 	 * lock.
1603662c653bSLiu Bo 	 */
1604662c653bSLiu Bo 	if (write_lock_level < BTRFS_MAX_LEVEL) {
1605662c653bSLiu Bo 		/*
1606662c653bSLiu Bo 		 * We don't know the level of the root node until we actually
1607662c653bSLiu Bo 		 * have it read locked
16081fc28d8eSLiu Bo 		 */
16091bb96598SJosef Bacik 		b = btrfs_read_lock_root_node(root);
16101fc28d8eSLiu Bo 		level = btrfs_header_level(b);
16111fc28d8eSLiu Bo 		if (level > write_lock_level)
16121fc28d8eSLiu Bo 			goto out;
16131fc28d8eSLiu Bo 
1614662c653bSLiu Bo 		/* Whoops, must trade for write lock */
16151fc28d8eSLiu Bo 		btrfs_tree_read_unlock(b);
16161fc28d8eSLiu Bo 		free_extent_buffer(b);
1617662c653bSLiu Bo 	}
1618662c653bSLiu Bo 
16191fc28d8eSLiu Bo 	b = btrfs_lock_root_node(root);
16201fc28d8eSLiu Bo 	root_lock = BTRFS_WRITE_LOCK;
16211fc28d8eSLiu Bo 
16221fc28d8eSLiu Bo 	/* The level might have changed, check again */
16231fc28d8eSLiu Bo 	level = btrfs_header_level(b);
16241fc28d8eSLiu Bo 
16251fc28d8eSLiu Bo out:
1626120de408SJosef Bacik 	/*
1627120de408SJosef Bacik 	 * The root may have failed to write out at some point, and thus is no
1628120de408SJosef Bacik 	 * longer valid, return an error in this case.
1629120de408SJosef Bacik 	 */
1630120de408SJosef Bacik 	if (!extent_buffer_uptodate(b)) {
1631120de408SJosef Bacik 		if (root_lock)
1632120de408SJosef Bacik 			btrfs_tree_unlock_rw(b, root_lock);
1633120de408SJosef Bacik 		free_extent_buffer(b);
1634120de408SJosef Bacik 		return ERR_PTR(-EIO);
1635120de408SJosef Bacik 	}
1636120de408SJosef Bacik 
16371fc28d8eSLiu Bo 	p->nodes[level] = b;
16381fc28d8eSLiu Bo 	if (!p->skip_locking)
16391fc28d8eSLiu Bo 		p->locks[level] = root_lock;
16401fc28d8eSLiu Bo 	/*
16411fc28d8eSLiu Bo 	 * Callers are responsible for dropping b's references.
16421fc28d8eSLiu Bo 	 */
16431fc28d8eSLiu Bo 	return b;
16441fc28d8eSLiu Bo }
16451fc28d8eSLiu Bo 
1646d96b3424SFilipe Manana /*
1647d96b3424SFilipe Manana  * Replace the extent buffer at the lowest level of the path with a cloned
1648d96b3424SFilipe Manana  * version. The purpose is to be able to use it safely, after releasing the
1649d96b3424SFilipe Manana  * commit root semaphore, even if relocation is happening in parallel, the
1650d96b3424SFilipe Manana  * transaction used for relocation is committed and the extent buffer is
1651d96b3424SFilipe Manana  * reallocated in the next transaction.
1652d96b3424SFilipe Manana  *
1653d96b3424SFilipe Manana  * This is used in a context where the caller does not prevent transaction
1654d96b3424SFilipe Manana  * commits from happening, either by holding a transaction handle or holding
1655d96b3424SFilipe Manana  * some lock, while it's doing searches through a commit root.
1656d96b3424SFilipe Manana  * At the moment it's only used for send operations.
1657d96b3424SFilipe Manana  */
1658d96b3424SFilipe Manana static int finish_need_commit_sem_search(struct btrfs_path *path)
1659d96b3424SFilipe Manana {
1660d96b3424SFilipe Manana 	const int i = path->lowest_level;
1661d96b3424SFilipe Manana 	const int slot = path->slots[i];
1662d96b3424SFilipe Manana 	struct extent_buffer *lowest = path->nodes[i];
1663d96b3424SFilipe Manana 	struct extent_buffer *clone;
1664d96b3424SFilipe Manana 
1665d96b3424SFilipe Manana 	ASSERT(path->need_commit_sem);
1666d96b3424SFilipe Manana 
1667d96b3424SFilipe Manana 	if (!lowest)
1668d96b3424SFilipe Manana 		return 0;
1669d96b3424SFilipe Manana 
1670d96b3424SFilipe Manana 	lockdep_assert_held_read(&lowest->fs_info->commit_root_sem);
1671d96b3424SFilipe Manana 
1672d96b3424SFilipe Manana 	clone = btrfs_clone_extent_buffer(lowest);
1673d96b3424SFilipe Manana 	if (!clone)
1674d96b3424SFilipe Manana 		return -ENOMEM;
1675d96b3424SFilipe Manana 
1676d96b3424SFilipe Manana 	btrfs_release_path(path);
1677d96b3424SFilipe Manana 	path->nodes[i] = clone;
1678d96b3424SFilipe Manana 	path->slots[i] = slot;
1679d96b3424SFilipe Manana 
1680d96b3424SFilipe Manana 	return 0;
1681d96b3424SFilipe Manana }
16821fc28d8eSLiu Bo 
1683e2e58d0fSFilipe Manana static inline int search_for_key_slot(struct extent_buffer *eb,
1684e2e58d0fSFilipe Manana 				      int search_low_slot,
1685e2e58d0fSFilipe Manana 				      const struct btrfs_key *key,
1686e2e58d0fSFilipe Manana 				      int prev_cmp,
1687e2e58d0fSFilipe Manana 				      int *slot)
1688e2e58d0fSFilipe Manana {
1689e2e58d0fSFilipe Manana 	/*
1690e2e58d0fSFilipe Manana 	 * If a previous call to btrfs_bin_search() on a parent node returned an
1691e2e58d0fSFilipe Manana 	 * exact match (prev_cmp == 0), we can safely assume the target key will
1692e2e58d0fSFilipe Manana 	 * always be at slot 0 on lower levels, since each key pointer
1693e2e58d0fSFilipe Manana 	 * (struct btrfs_key_ptr) refers to the lowest key accessible from the
1694e2e58d0fSFilipe Manana 	 * subtree it points to. Thus we can skip searching lower levels.
1695e2e58d0fSFilipe Manana 	 */
1696e2e58d0fSFilipe Manana 	if (prev_cmp == 0) {
1697e2e58d0fSFilipe Manana 		*slot = 0;
1698e2e58d0fSFilipe Manana 		return 0;
1699e2e58d0fSFilipe Manana 	}
1700e2e58d0fSFilipe Manana 
1701e2e58d0fSFilipe Manana 	return generic_bin_search(eb, search_low_slot, key, slot);
1702e2e58d0fSFilipe Manana }
1703e2e58d0fSFilipe Manana 
1704109324cfSFilipe Manana static int search_leaf(struct btrfs_trans_handle *trans,
1705109324cfSFilipe Manana 		       struct btrfs_root *root,
1706109324cfSFilipe Manana 		       const struct btrfs_key *key,
1707109324cfSFilipe Manana 		       struct btrfs_path *path,
1708109324cfSFilipe Manana 		       int ins_len,
1709109324cfSFilipe Manana 		       int prev_cmp)
1710109324cfSFilipe Manana {
1711109324cfSFilipe Manana 	struct extent_buffer *leaf = path->nodes[0];
1712109324cfSFilipe Manana 	int leaf_free_space = -1;
1713109324cfSFilipe Manana 	int search_low_slot = 0;
1714109324cfSFilipe Manana 	int ret;
1715109324cfSFilipe Manana 	bool do_bin_search = true;
1716109324cfSFilipe Manana 
1717109324cfSFilipe Manana 	/*
1718109324cfSFilipe Manana 	 * If we are doing an insertion, the leaf has enough free space and the
1719109324cfSFilipe Manana 	 * destination slot for the key is not slot 0, then we can unlock our
1720109324cfSFilipe Manana 	 * write lock on the parent, and any other upper nodes, before doing the
1721109324cfSFilipe Manana 	 * binary search on the leaf (with search_for_key_slot()), allowing other
1722109324cfSFilipe Manana 	 * tasks to lock the parent and any other upper nodes.
1723109324cfSFilipe Manana 	 */
1724109324cfSFilipe Manana 	if (ins_len > 0) {
1725109324cfSFilipe Manana 		/*
1726109324cfSFilipe Manana 		 * Cache the leaf free space, since we will need it later and it
1727109324cfSFilipe Manana 		 * will not change until then.
1728109324cfSFilipe Manana 		 */
1729109324cfSFilipe Manana 		leaf_free_space = btrfs_leaf_free_space(leaf);
1730109324cfSFilipe Manana 
1731109324cfSFilipe Manana 		/*
1732109324cfSFilipe Manana 		 * !path->locks[1] means we have a single node tree, the leaf is
1733109324cfSFilipe Manana 		 * the root of the tree.
1734109324cfSFilipe Manana 		 */
1735109324cfSFilipe Manana 		if (path->locks[1] && leaf_free_space >= ins_len) {
1736109324cfSFilipe Manana 			struct btrfs_disk_key first_key;
1737109324cfSFilipe Manana 
1738109324cfSFilipe Manana 			ASSERT(btrfs_header_nritems(leaf) > 0);
1739109324cfSFilipe Manana 			btrfs_item_key(leaf, &first_key, 0);
1740109324cfSFilipe Manana 
1741109324cfSFilipe Manana 			/*
1742109324cfSFilipe Manana 			 * Doing the extra comparison with the first key is cheap,
1743109324cfSFilipe Manana 			 * taking into account that the first key is very likely
1744109324cfSFilipe Manana 			 * already in a cache line because it immediately follows
1745109324cfSFilipe Manana 			 * the extent buffer's header and we have recently accessed
1746109324cfSFilipe Manana 			 * the header's level field.
1747109324cfSFilipe Manana 			 */
1748109324cfSFilipe Manana 			ret = comp_keys(&first_key, key);
1749109324cfSFilipe Manana 			if (ret < 0) {
1750109324cfSFilipe Manana 				/*
1751109324cfSFilipe Manana 				 * The first key is smaller than the key we want
1752109324cfSFilipe Manana 				 * to insert, so we are safe to unlock all upper
1753109324cfSFilipe Manana 				 * nodes and we have to do the binary search.
1754109324cfSFilipe Manana 				 *
1755109324cfSFilipe Manana 				 * We do use btrfs_unlock_up_safe() and not
1756109324cfSFilipe Manana 				 * unlock_up() because the later does not unlock
1757109324cfSFilipe Manana 				 * nodes with a slot of 0 - we can safely unlock
1758109324cfSFilipe Manana 				 * any node even if its slot is 0 since in this
1759109324cfSFilipe Manana 				 * case the key does not end up at slot 0 of the
1760109324cfSFilipe Manana 				 * leaf and there's no need to split the leaf.
1761109324cfSFilipe Manana 				 */
1762109324cfSFilipe Manana 				btrfs_unlock_up_safe(path, 1);
1763109324cfSFilipe Manana 				search_low_slot = 1;
1764109324cfSFilipe Manana 			} else {
1765109324cfSFilipe Manana 				/*
1766109324cfSFilipe Manana 				 * The first key is >= then the key we want to
1767109324cfSFilipe Manana 				 * insert, so we can skip the binary search as
1768109324cfSFilipe Manana 				 * the target key will be at slot 0.
1769109324cfSFilipe Manana 				 *
1770109324cfSFilipe Manana 				 * We can not unlock upper nodes when the key is
1771109324cfSFilipe Manana 				 * less than the first key, because we will need
1772109324cfSFilipe Manana 				 * to update the key at slot 0 of the parent node
1773109324cfSFilipe Manana 				 * and possibly of other upper nodes too.
1774109324cfSFilipe Manana 				 * If the key matches the first key, then we can
1775109324cfSFilipe Manana 				 * unlock all the upper nodes, using
1776109324cfSFilipe Manana 				 * btrfs_unlock_up_safe() instead of unlock_up()
1777109324cfSFilipe Manana 				 * as stated above.
1778109324cfSFilipe Manana 				 */
1779109324cfSFilipe Manana 				if (ret == 0)
1780109324cfSFilipe Manana 					btrfs_unlock_up_safe(path, 1);
1781109324cfSFilipe Manana 				/*
1782109324cfSFilipe Manana 				 * ret is already 0 or 1, matching the result of
1783109324cfSFilipe Manana 				 * a btrfs_bin_search() call, so there is no need
1784109324cfSFilipe Manana 				 * to adjust it.
1785109324cfSFilipe Manana 				 */
1786109324cfSFilipe Manana 				do_bin_search = false;
1787109324cfSFilipe Manana 				path->slots[0] = 0;
1788109324cfSFilipe Manana 			}
1789109324cfSFilipe Manana 		}
1790109324cfSFilipe Manana 	}
1791109324cfSFilipe Manana 
1792109324cfSFilipe Manana 	if (do_bin_search) {
1793109324cfSFilipe Manana 		ret = search_for_key_slot(leaf, search_low_slot, key,
1794109324cfSFilipe Manana 					  prev_cmp, &path->slots[0]);
1795109324cfSFilipe Manana 		if (ret < 0)
1796109324cfSFilipe Manana 			return ret;
1797109324cfSFilipe Manana 	}
1798109324cfSFilipe Manana 
1799109324cfSFilipe Manana 	if (ins_len > 0) {
1800109324cfSFilipe Manana 		/*
1801109324cfSFilipe Manana 		 * Item key already exists. In this case, if we are allowed to
1802109324cfSFilipe Manana 		 * insert the item (for example, in dir_item case, item key
1803109324cfSFilipe Manana 		 * collision is allowed), it will be merged with the original
1804109324cfSFilipe Manana 		 * item. Only the item size grows, no new btrfs item will be
1805109324cfSFilipe Manana 		 * added. If search_for_extension is not set, ins_len already
1806109324cfSFilipe Manana 		 * accounts the size btrfs_item, deduct it here so leaf space
1807109324cfSFilipe Manana 		 * check will be correct.
1808109324cfSFilipe Manana 		 */
1809109324cfSFilipe Manana 		if (ret == 0 && !path->search_for_extension) {
1810109324cfSFilipe Manana 			ASSERT(ins_len >= sizeof(struct btrfs_item));
1811109324cfSFilipe Manana 			ins_len -= sizeof(struct btrfs_item);
1812109324cfSFilipe Manana 		}
1813109324cfSFilipe Manana 
1814109324cfSFilipe Manana 		ASSERT(leaf_free_space >= 0);
1815109324cfSFilipe Manana 
1816109324cfSFilipe Manana 		if (leaf_free_space < ins_len) {
1817109324cfSFilipe Manana 			int err;
1818109324cfSFilipe Manana 
1819109324cfSFilipe Manana 			err = split_leaf(trans, root, key, path, ins_len,
1820109324cfSFilipe Manana 					 (ret == 0));
1821*bb8e9a60SFilipe Manana 			ASSERT(err <= 0);
1822*bb8e9a60SFilipe Manana 			if (WARN_ON(err > 0))
1823*bb8e9a60SFilipe Manana 				err = -EUCLEAN;
1824109324cfSFilipe Manana 			if (err)
1825109324cfSFilipe Manana 				ret = err;
1826109324cfSFilipe Manana 		}
1827109324cfSFilipe Manana 	}
1828109324cfSFilipe Manana 
1829109324cfSFilipe Manana 	return ret;
1830109324cfSFilipe Manana }
1831109324cfSFilipe Manana 
1832c8c42864SChris Mason /*
18334271eceaSNikolay Borisov  * btrfs_search_slot - look for a key in a tree and perform necessary
18344271eceaSNikolay Borisov  * modifications to preserve tree invariants.
183574123bd7SChris Mason  *
18364271eceaSNikolay Borisov  * @trans:	Handle of transaction, used when modifying the tree
18374271eceaSNikolay Borisov  * @p:		Holds all btree nodes along the search path
18384271eceaSNikolay Borisov  * @root:	The root node of the tree
18394271eceaSNikolay Borisov  * @key:	The key we are looking for
18409a664971Sethanwu  * @ins_len:	Indicates purpose of search:
18419a664971Sethanwu  *              >0  for inserts it's size of item inserted (*)
18429a664971Sethanwu  *              <0  for deletions
18439a664971Sethanwu  *               0  for plain searches, not modifying the tree
18449a664971Sethanwu  *
18459a664971Sethanwu  *              (*) If size of item inserted doesn't include
18469a664971Sethanwu  *              sizeof(struct btrfs_item), then p->search_for_extension must
18479a664971Sethanwu  *              be set.
18484271eceaSNikolay Borisov  * @cow:	boolean should CoW operations be performed. Must always be 1
18494271eceaSNikolay Borisov  *		when modifying the tree.
185097571fd0SChris Mason  *
18514271eceaSNikolay Borisov  * If @ins_len > 0, nodes and leaves will be split as we walk down the tree.
18524271eceaSNikolay Borisov  * If @ins_len < 0, nodes will be merged as we walk down the tree (if possible)
18534271eceaSNikolay Borisov  *
18544271eceaSNikolay Borisov  * If @key is found, 0 is returned and you can find the item in the leaf level
18554271eceaSNikolay Borisov  * of the path (level 0)
18564271eceaSNikolay Borisov  *
18574271eceaSNikolay Borisov  * If @key isn't found, 1 is returned and the leaf level of the path (level 0)
18584271eceaSNikolay Borisov  * points to the slot where it should be inserted
18594271eceaSNikolay Borisov  *
18604271eceaSNikolay Borisov  * If an error is encountered while searching the tree a negative error number
18614271eceaSNikolay Borisov  * is returned
186274123bd7SChris Mason  */
1863310712b2SOmar Sandoval int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root *root,
1864310712b2SOmar Sandoval 		      const struct btrfs_key *key, struct btrfs_path *p,
1865310712b2SOmar Sandoval 		      int ins_len, int cow)
1866be0e5c09SChris Mason {
1867d96b3424SFilipe Manana 	struct btrfs_fs_info *fs_info = root->fs_info;
18685f39d397SChris Mason 	struct extent_buffer *b;
1869be0e5c09SChris Mason 	int slot;
1870be0e5c09SChris Mason 	int ret;
187133c66f43SYan Zheng 	int err;
1872be0e5c09SChris Mason 	int level;
1873925baeddSChris Mason 	int lowest_unlock = 1;
1874bd681513SChris Mason 	/* everything at write_lock_level or lower must be write locked */
1875bd681513SChris Mason 	int write_lock_level = 0;
18769f3a7427SChris Mason 	u8 lowest_level = 0;
1877f7c79f30SChris Mason 	int min_write_lock_level;
1878d7396f07SFilipe David Borba Manana 	int prev_cmp;
18799f3a7427SChris Mason 
18806702ed49SChris Mason 	lowest_level = p->lowest_level;
1881323ac95bSChris Mason 	WARN_ON(lowest_level && ins_len > 0);
188222b0ebdaSChris Mason 	WARN_ON(p->nodes[0] != NULL);
1883eb653de1SFilipe David Borba Manana 	BUG_ON(!cow && ins_len);
188425179201SJosef Bacik 
1885bd681513SChris Mason 	if (ins_len < 0) {
1886925baeddSChris Mason 		lowest_unlock = 2;
188765b51a00SChris Mason 
1888bd681513SChris Mason 		/* when we are removing items, we might have to go up to level
1889bd681513SChris Mason 		 * two as we update tree pointers  Make sure we keep write
1890bd681513SChris Mason 		 * for those levels as well
1891bd681513SChris Mason 		 */
1892bd681513SChris Mason 		write_lock_level = 2;
1893bd681513SChris Mason 	} else if (ins_len > 0) {
1894bd681513SChris Mason 		/*
1895bd681513SChris Mason 		 * for inserting items, make sure we have a write lock on
1896bd681513SChris Mason 		 * level 1 so we can update keys
1897bd681513SChris Mason 		 */
1898bd681513SChris Mason 		write_lock_level = 1;
1899bd681513SChris Mason 	}
1900bd681513SChris Mason 
1901bd681513SChris Mason 	if (!cow)
1902bd681513SChris Mason 		write_lock_level = -1;
1903bd681513SChris Mason 
190409a2a8f9SJosef Bacik 	if (cow && (p->keep_locks || p->lowest_level))
1905bd681513SChris Mason 		write_lock_level = BTRFS_MAX_LEVEL;
1906bd681513SChris Mason 
1907f7c79f30SChris Mason 	min_write_lock_level = write_lock_level;
1908f7c79f30SChris Mason 
1909d96b3424SFilipe Manana 	if (p->need_commit_sem) {
1910d96b3424SFilipe Manana 		ASSERT(p->search_commit_root);
1911d96b3424SFilipe Manana 		down_read(&fs_info->commit_root_sem);
1912d96b3424SFilipe Manana 	}
1913d96b3424SFilipe Manana 
1914bb803951SChris Mason again:
1915d7396f07SFilipe David Borba Manana 	prev_cmp = -1;
19161fc28d8eSLiu Bo 	b = btrfs_search_slot_get_root(root, p, write_lock_level);
1917be6821f8SFilipe Manana 	if (IS_ERR(b)) {
1918be6821f8SFilipe Manana 		ret = PTR_ERR(b);
1919be6821f8SFilipe Manana 		goto done;
1920be6821f8SFilipe Manana 	}
1921925baeddSChris Mason 
1922eb60ceacSChris Mason 	while (b) {
1923f624d976SQu Wenruo 		int dec = 0;
1924f624d976SQu Wenruo 
19255f39d397SChris Mason 		level = btrfs_header_level(b);
192665b51a00SChris Mason 
192702217ed2SChris Mason 		if (cow) {
19289ea2c7c9SNikolay Borisov 			bool last_level = (level == (BTRFS_MAX_LEVEL - 1));
19299ea2c7c9SNikolay Borisov 
1930c8c42864SChris Mason 			/*
1931c8c42864SChris Mason 			 * if we don't really need to cow this block
1932c8c42864SChris Mason 			 * then we don't want to set the path blocking,
1933c8c42864SChris Mason 			 * so we test it here
1934c8c42864SChris Mason 			 */
19355963ffcaSJosef Bacik 			if (!should_cow_block(trans, root, b))
193665b51a00SChris Mason 				goto cow_done;
19375d4f98a2SYan Zheng 
1938bd681513SChris Mason 			/*
1939bd681513SChris Mason 			 * must have write locks on this node and the
1940bd681513SChris Mason 			 * parent
1941bd681513SChris Mason 			 */
19425124e00eSJosef Bacik 			if (level > write_lock_level ||
19435124e00eSJosef Bacik 			    (level + 1 > write_lock_level &&
19445124e00eSJosef Bacik 			    level + 1 < BTRFS_MAX_LEVEL &&
19455124e00eSJosef Bacik 			    p->nodes[level + 1])) {
1946bd681513SChris Mason 				write_lock_level = level + 1;
1947bd681513SChris Mason 				btrfs_release_path(p);
1948bd681513SChris Mason 				goto again;
1949bd681513SChris Mason 			}
1950bd681513SChris Mason 
19519ea2c7c9SNikolay Borisov 			if (last_level)
19529ea2c7c9SNikolay Borisov 				err = btrfs_cow_block(trans, root, b, NULL, 0,
19539631e4ccSJosef Bacik 						      &b,
19549631e4ccSJosef Bacik 						      BTRFS_NESTING_COW);
19559ea2c7c9SNikolay Borisov 			else
195633c66f43SYan Zheng 				err = btrfs_cow_block(trans, root, b,
1957e20d96d6SChris Mason 						      p->nodes[level + 1],
19589631e4ccSJosef Bacik 						      p->slots[level + 1], &b,
19599631e4ccSJosef Bacik 						      BTRFS_NESTING_COW);
196033c66f43SYan Zheng 			if (err) {
196133c66f43SYan Zheng 				ret = err;
196265b51a00SChris Mason 				goto done;
196354aa1f4dSChris Mason 			}
196402217ed2SChris Mason 		}
196565b51a00SChris Mason cow_done:
1966eb60ceacSChris Mason 		p->nodes[level] = b;
196752398340SLiu Bo 		/*
196852398340SLiu Bo 		 * Leave path with blocking locks to avoid massive
196952398340SLiu Bo 		 * lock context switch, this is made on purpose.
197052398340SLiu Bo 		 */
1971b4ce94deSChris Mason 
1972b4ce94deSChris Mason 		/*
1973b4ce94deSChris Mason 		 * we have a lock on b and as long as we aren't changing
1974b4ce94deSChris Mason 		 * the tree, there is no way to for the items in b to change.
1975b4ce94deSChris Mason 		 * It is safe to drop the lock on our parent before we
1976b4ce94deSChris Mason 		 * go through the expensive btree search on b.
1977b4ce94deSChris Mason 		 *
1978eb653de1SFilipe David Borba Manana 		 * If we're inserting or deleting (ins_len != 0), then we might
1979eb653de1SFilipe David Borba Manana 		 * be changing slot zero, which may require changing the parent.
1980eb653de1SFilipe David Borba Manana 		 * So, we can't drop the lock until after we know which slot
1981eb653de1SFilipe David Borba Manana 		 * we're operating on.
1982b4ce94deSChris Mason 		 */
1983eb653de1SFilipe David Borba Manana 		if (!ins_len && !p->keep_locks) {
1984eb653de1SFilipe David Borba Manana 			int u = level + 1;
1985eb653de1SFilipe David Borba Manana 
1986eb653de1SFilipe David Borba Manana 			if (u < BTRFS_MAX_LEVEL && p->locks[u]) {
1987eb653de1SFilipe David Borba Manana 				btrfs_tree_unlock_rw(p->nodes[u], p->locks[u]);
1988eb653de1SFilipe David Borba Manana 				p->locks[u] = 0;
1989eb653de1SFilipe David Borba Manana 			}
1990eb653de1SFilipe David Borba Manana 		}
1991b4ce94deSChris Mason 
1992e2e58d0fSFilipe Manana 		if (level == 0) {
1993109324cfSFilipe Manana 			if (ins_len > 0)
1994e5e1c174SFilipe Manana 				ASSERT(write_lock_level >= 1);
1995bd681513SChris Mason 
1996109324cfSFilipe Manana 			ret = search_leaf(trans, root, key, p, ins_len, prev_cmp);
1997459931ecSChris Mason 			if (!p->search_for_split)
1998f7c79f30SChris Mason 				unlock_up(p, level, lowest_unlock,
19994b6f8e96SLiu Bo 					  min_write_lock_level, NULL);
200065b51a00SChris Mason 			goto done;
200165b51a00SChris Mason 		}
2002e2e58d0fSFilipe Manana 
2003e2e58d0fSFilipe Manana 		ret = search_for_key_slot(b, 0, key, prev_cmp, &slot);
2004e2e58d0fSFilipe Manana 		if (ret < 0)
2005e2e58d0fSFilipe Manana 			goto done;
2006e2e58d0fSFilipe Manana 		prev_cmp = ret;
2007e2e58d0fSFilipe Manana 
2008f624d976SQu Wenruo 		if (ret && slot > 0) {
2009f624d976SQu Wenruo 			dec = 1;
2010f624d976SQu Wenruo 			slot--;
2011f624d976SQu Wenruo 		}
2012f624d976SQu Wenruo 		p->slots[level] = slot;
2013f624d976SQu Wenruo 		err = setup_nodes_for_search(trans, root, p, b, level, ins_len,
2014f624d976SQu Wenruo 					     &write_lock_level);
2015f624d976SQu Wenruo 		if (err == -EAGAIN)
2016f624d976SQu Wenruo 			goto again;
2017f624d976SQu Wenruo 		if (err) {
2018f624d976SQu Wenruo 			ret = err;
2019f624d976SQu Wenruo 			goto done;
2020f624d976SQu Wenruo 		}
2021f624d976SQu Wenruo 		b = p->nodes[level];
2022f624d976SQu Wenruo 		slot = p->slots[level];
2023f624d976SQu Wenruo 
2024f624d976SQu Wenruo 		/*
2025f624d976SQu Wenruo 		 * Slot 0 is special, if we change the key we have to update
2026f624d976SQu Wenruo 		 * the parent pointer which means we must have a write lock on
2027f624d976SQu Wenruo 		 * the parent
2028f624d976SQu Wenruo 		 */
2029f624d976SQu Wenruo 		if (slot == 0 && ins_len && write_lock_level < level + 1) {
2030f624d976SQu Wenruo 			write_lock_level = level + 1;
2031f624d976SQu Wenruo 			btrfs_release_path(p);
2032f624d976SQu Wenruo 			goto again;
2033f624d976SQu Wenruo 		}
2034f624d976SQu Wenruo 
2035f624d976SQu Wenruo 		unlock_up(p, level, lowest_unlock, min_write_lock_level,
2036f624d976SQu Wenruo 			  &write_lock_level);
2037f624d976SQu Wenruo 
2038f624d976SQu Wenruo 		if (level == lowest_level) {
2039f624d976SQu Wenruo 			if (dec)
2040f624d976SQu Wenruo 				p->slots[level]++;
2041f624d976SQu Wenruo 			goto done;
2042f624d976SQu Wenruo 		}
2043f624d976SQu Wenruo 
2044f624d976SQu Wenruo 		err = read_block_for_search(root, p, &b, level, slot, key);
2045f624d976SQu Wenruo 		if (err == -EAGAIN)
2046f624d976SQu Wenruo 			goto again;
2047f624d976SQu Wenruo 		if (err) {
2048f624d976SQu Wenruo 			ret = err;
2049f624d976SQu Wenruo 			goto done;
2050f624d976SQu Wenruo 		}
2051f624d976SQu Wenruo 
2052f624d976SQu Wenruo 		if (!p->skip_locking) {
2053f624d976SQu Wenruo 			level = btrfs_header_level(b);
2054f624d976SQu Wenruo 			if (level <= write_lock_level) {
2055f624d976SQu Wenruo 				btrfs_tree_lock(b);
2056f624d976SQu Wenruo 				p->locks[level] = BTRFS_WRITE_LOCK;
2057f624d976SQu Wenruo 			} else {
2058fe596ca3SJosef Bacik 				btrfs_tree_read_lock(b);
2059f624d976SQu Wenruo 				p->locks[level] = BTRFS_READ_LOCK;
2060f624d976SQu Wenruo 			}
2061f624d976SQu Wenruo 			p->nodes[level] = b;
2062f624d976SQu Wenruo 		}
206365b51a00SChris Mason 	}
206465b51a00SChris Mason 	ret = 1;
206565b51a00SChris Mason done:
20665f5bc6b1SFilipe Manana 	if (ret < 0 && !p->skip_release_on_error)
2067b3b4aa74SDavid Sterba 		btrfs_release_path(p);
2068d96b3424SFilipe Manana 
2069d96b3424SFilipe Manana 	if (p->need_commit_sem) {
2070d96b3424SFilipe Manana 		int ret2;
2071d96b3424SFilipe Manana 
2072d96b3424SFilipe Manana 		ret2 = finish_need_commit_sem_search(p);
2073d96b3424SFilipe Manana 		up_read(&fs_info->commit_root_sem);
2074d96b3424SFilipe Manana 		if (ret2)
2075d96b3424SFilipe Manana 			ret = ret2;
2076d96b3424SFilipe Manana 	}
2077d96b3424SFilipe Manana 
2078be0e5c09SChris Mason 	return ret;
2079be0e5c09SChris Mason }
2080f75e2b79SJosef Bacik ALLOW_ERROR_INJECTION(btrfs_search_slot, ERRNO);
2081be0e5c09SChris Mason 
208274123bd7SChris Mason /*
20835d9e75c4SJan Schmidt  * Like btrfs_search_slot, this looks for a key in the given tree. It uses the
20845d9e75c4SJan Schmidt  * current state of the tree together with the operations recorded in the tree
20855d9e75c4SJan Schmidt  * modification log to search for the key in a previous version of this tree, as
20865d9e75c4SJan Schmidt  * denoted by the time_seq parameter.
20875d9e75c4SJan Schmidt  *
20885d9e75c4SJan Schmidt  * Naturally, there is no support for insert, delete or cow operations.
20895d9e75c4SJan Schmidt  *
20905d9e75c4SJan Schmidt  * The resulting path and return value will be set up as if we called
20915d9e75c4SJan Schmidt  * btrfs_search_slot at that point in time with ins_len and cow both set to 0.
20925d9e75c4SJan Schmidt  */
2093310712b2SOmar Sandoval int btrfs_search_old_slot(struct btrfs_root *root, const struct btrfs_key *key,
20945d9e75c4SJan Schmidt 			  struct btrfs_path *p, u64 time_seq)
20955d9e75c4SJan Schmidt {
20960b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
20975d9e75c4SJan Schmidt 	struct extent_buffer *b;
20985d9e75c4SJan Schmidt 	int slot;
20995d9e75c4SJan Schmidt 	int ret;
21005d9e75c4SJan Schmidt 	int err;
21015d9e75c4SJan Schmidt 	int level;
21025d9e75c4SJan Schmidt 	int lowest_unlock = 1;
21035d9e75c4SJan Schmidt 	u8 lowest_level = 0;
21045d9e75c4SJan Schmidt 
21055d9e75c4SJan Schmidt 	lowest_level = p->lowest_level;
21065d9e75c4SJan Schmidt 	WARN_ON(p->nodes[0] != NULL);
21075d9e75c4SJan Schmidt 
21085d9e75c4SJan Schmidt 	if (p->search_commit_root) {
21095d9e75c4SJan Schmidt 		BUG_ON(time_seq);
21105d9e75c4SJan Schmidt 		return btrfs_search_slot(NULL, root, key, p, 0, 0);
21115d9e75c4SJan Schmidt 	}
21125d9e75c4SJan Schmidt 
21135d9e75c4SJan Schmidt again:
2114f3a84ccdSFilipe Manana 	b = btrfs_get_old_root(root, time_seq);
2115315bed43SNikolay Borisov 	if (!b) {
2116315bed43SNikolay Borisov 		ret = -EIO;
2117315bed43SNikolay Borisov 		goto done;
2118315bed43SNikolay Borisov 	}
21195d9e75c4SJan Schmidt 	level = btrfs_header_level(b);
21205d9e75c4SJan Schmidt 	p->locks[level] = BTRFS_READ_LOCK;
21215d9e75c4SJan Schmidt 
21225d9e75c4SJan Schmidt 	while (b) {
2123abe9339dSQu Wenruo 		int dec = 0;
2124abe9339dSQu Wenruo 
21255d9e75c4SJan Schmidt 		level = btrfs_header_level(b);
21265d9e75c4SJan Schmidt 		p->nodes[level] = b;
21275d9e75c4SJan Schmidt 
21285d9e75c4SJan Schmidt 		/*
21295d9e75c4SJan Schmidt 		 * we have a lock on b and as long as we aren't changing
21305d9e75c4SJan Schmidt 		 * the tree, there is no way to for the items in b to change.
21315d9e75c4SJan Schmidt 		 * It is safe to drop the lock on our parent before we
21325d9e75c4SJan Schmidt 		 * go through the expensive btree search on b.
21335d9e75c4SJan Schmidt 		 */
21345d9e75c4SJan Schmidt 		btrfs_unlock_up_safe(p, level + 1);
21355d9e75c4SJan Schmidt 
2136995e9a16SNikolay Borisov 		ret = btrfs_bin_search(b, key, &slot);
2137cbca7d59SFilipe Manana 		if (ret < 0)
2138cbca7d59SFilipe Manana 			goto done;
21395d9e75c4SJan Schmidt 
2140abe9339dSQu Wenruo 		if (level == 0) {
2141abe9339dSQu Wenruo 			p->slots[level] = slot;
2142abe9339dSQu Wenruo 			unlock_up(p, level, lowest_unlock, 0, NULL);
2143abe9339dSQu Wenruo 			goto done;
2144abe9339dSQu Wenruo 		}
2145abe9339dSQu Wenruo 
21465d9e75c4SJan Schmidt 		if (ret && slot > 0) {
21475d9e75c4SJan Schmidt 			dec = 1;
2148abe9339dSQu Wenruo 			slot--;
21495d9e75c4SJan Schmidt 		}
21505d9e75c4SJan Schmidt 		p->slots[level] = slot;
21515d9e75c4SJan Schmidt 		unlock_up(p, level, lowest_unlock, 0, NULL);
21525d9e75c4SJan Schmidt 
21535d9e75c4SJan Schmidt 		if (level == lowest_level) {
21545d9e75c4SJan Schmidt 			if (dec)
21555d9e75c4SJan Schmidt 				p->slots[level]++;
21565d9e75c4SJan Schmidt 			goto done;
21575d9e75c4SJan Schmidt 		}
21585d9e75c4SJan Schmidt 
2159abe9339dSQu Wenruo 		err = read_block_for_search(root, p, &b, level, slot, key);
21605d9e75c4SJan Schmidt 		if (err == -EAGAIN)
21615d9e75c4SJan Schmidt 			goto again;
21625d9e75c4SJan Schmidt 		if (err) {
21635d9e75c4SJan Schmidt 			ret = err;
21645d9e75c4SJan Schmidt 			goto done;
21655d9e75c4SJan Schmidt 		}
21665d9e75c4SJan Schmidt 
21675d9e75c4SJan Schmidt 		level = btrfs_header_level(b);
21685d9e75c4SJan Schmidt 		btrfs_tree_read_lock(b);
2169f3a84ccdSFilipe Manana 		b = btrfs_tree_mod_log_rewind(fs_info, p, b, time_seq);
2170db7f3436SJosef Bacik 		if (!b) {
2171db7f3436SJosef Bacik 			ret = -ENOMEM;
2172db7f3436SJosef Bacik 			goto done;
2173db7f3436SJosef Bacik 		}
21745d9e75c4SJan Schmidt 		p->locks[level] = BTRFS_READ_LOCK;
21755d9e75c4SJan Schmidt 		p->nodes[level] = b;
21765d9e75c4SJan Schmidt 	}
21775d9e75c4SJan Schmidt 	ret = 1;
21785d9e75c4SJan Schmidt done:
21795d9e75c4SJan Schmidt 	if (ret < 0)
21805d9e75c4SJan Schmidt 		btrfs_release_path(p);
21815d9e75c4SJan Schmidt 
21825d9e75c4SJan Schmidt 	return ret;
21835d9e75c4SJan Schmidt }
21845d9e75c4SJan Schmidt 
21855d9e75c4SJan Schmidt /*
21862f38b3e1SArne Jansen  * helper to use instead of search slot if no exact match is needed but
21872f38b3e1SArne Jansen  * instead the next or previous item should be returned.
21882f38b3e1SArne Jansen  * When find_higher is true, the next higher item is returned, the next lower
21892f38b3e1SArne Jansen  * otherwise.
21902f38b3e1SArne Jansen  * When return_any and find_higher are both true, and no higher item is found,
21912f38b3e1SArne Jansen  * return the next lower instead.
21922f38b3e1SArne Jansen  * When return_any is true and find_higher is false, and no lower item is found,
21932f38b3e1SArne Jansen  * return the next higher instead.
21942f38b3e1SArne Jansen  * It returns 0 if any item is found, 1 if none is found (tree empty), and
21952f38b3e1SArne Jansen  * < 0 on error
21962f38b3e1SArne Jansen  */
21972f38b3e1SArne Jansen int btrfs_search_slot_for_read(struct btrfs_root *root,
2198310712b2SOmar Sandoval 			       const struct btrfs_key *key,
2199310712b2SOmar Sandoval 			       struct btrfs_path *p, int find_higher,
2200310712b2SOmar Sandoval 			       int return_any)
22012f38b3e1SArne Jansen {
22022f38b3e1SArne Jansen 	int ret;
22032f38b3e1SArne Jansen 	struct extent_buffer *leaf;
22042f38b3e1SArne Jansen 
22052f38b3e1SArne Jansen again:
22062f38b3e1SArne Jansen 	ret = btrfs_search_slot(NULL, root, key, p, 0, 0);
22072f38b3e1SArne Jansen 	if (ret <= 0)
22082f38b3e1SArne Jansen 		return ret;
22092f38b3e1SArne Jansen 	/*
22102f38b3e1SArne Jansen 	 * a return value of 1 means the path is at the position where the
22112f38b3e1SArne Jansen 	 * item should be inserted. Normally this is the next bigger item,
22122f38b3e1SArne Jansen 	 * but in case the previous item is the last in a leaf, path points
22132f38b3e1SArne Jansen 	 * to the first free slot in the previous leaf, i.e. at an invalid
22142f38b3e1SArne Jansen 	 * item.
22152f38b3e1SArne Jansen 	 */
22162f38b3e1SArne Jansen 	leaf = p->nodes[0];
22172f38b3e1SArne Jansen 
22182f38b3e1SArne Jansen 	if (find_higher) {
22192f38b3e1SArne Jansen 		if (p->slots[0] >= btrfs_header_nritems(leaf)) {
22202f38b3e1SArne Jansen 			ret = btrfs_next_leaf(root, p);
22212f38b3e1SArne Jansen 			if (ret <= 0)
22222f38b3e1SArne Jansen 				return ret;
22232f38b3e1SArne Jansen 			if (!return_any)
22242f38b3e1SArne Jansen 				return 1;
22252f38b3e1SArne Jansen 			/*
22262f38b3e1SArne Jansen 			 * no higher item found, return the next
22272f38b3e1SArne Jansen 			 * lower instead
22282f38b3e1SArne Jansen 			 */
22292f38b3e1SArne Jansen 			return_any = 0;
22302f38b3e1SArne Jansen 			find_higher = 0;
22312f38b3e1SArne Jansen 			btrfs_release_path(p);
22322f38b3e1SArne Jansen 			goto again;
22332f38b3e1SArne Jansen 		}
22342f38b3e1SArne Jansen 	} else {
22352f38b3e1SArne Jansen 		if (p->slots[0] == 0) {
22362f38b3e1SArne Jansen 			ret = btrfs_prev_leaf(root, p);
2237e6793769SArne Jansen 			if (ret < 0)
22382f38b3e1SArne Jansen 				return ret;
2239e6793769SArne Jansen 			if (!ret) {
224023c6bf6aSFilipe David Borba Manana 				leaf = p->nodes[0];
224123c6bf6aSFilipe David Borba Manana 				if (p->slots[0] == btrfs_header_nritems(leaf))
224223c6bf6aSFilipe David Borba Manana 					p->slots[0]--;
2243e6793769SArne Jansen 				return 0;
2244e6793769SArne Jansen 			}
22452f38b3e1SArne Jansen 			if (!return_any)
22462f38b3e1SArne Jansen 				return 1;
22472f38b3e1SArne Jansen 			/*
22482f38b3e1SArne Jansen 			 * no lower item found, return the next
22492f38b3e1SArne Jansen 			 * higher instead
22502f38b3e1SArne Jansen 			 */
22512f38b3e1SArne Jansen 			return_any = 0;
22522f38b3e1SArne Jansen 			find_higher = 1;
22532f38b3e1SArne Jansen 			btrfs_release_path(p);
22542f38b3e1SArne Jansen 			goto again;
2255e6793769SArne Jansen 		} else {
22562f38b3e1SArne Jansen 			--p->slots[0];
22572f38b3e1SArne Jansen 		}
22582f38b3e1SArne Jansen 	}
22592f38b3e1SArne Jansen 	return 0;
22602f38b3e1SArne Jansen }
22612f38b3e1SArne Jansen 
22622f38b3e1SArne Jansen /*
22630ff40a91SMarcos Paulo de Souza  * Execute search and call btrfs_previous_item to traverse backwards if the item
22640ff40a91SMarcos Paulo de Souza  * was not found.
22650ff40a91SMarcos Paulo de Souza  *
22660ff40a91SMarcos Paulo de Souza  * Return 0 if found, 1 if not found and < 0 if error.
22670ff40a91SMarcos Paulo de Souza  */
22680ff40a91SMarcos Paulo de Souza int btrfs_search_backwards(struct btrfs_root *root, struct btrfs_key *key,
22690ff40a91SMarcos Paulo de Souza 			   struct btrfs_path *path)
22700ff40a91SMarcos Paulo de Souza {
22710ff40a91SMarcos Paulo de Souza 	int ret;
22720ff40a91SMarcos Paulo de Souza 
22730ff40a91SMarcos Paulo de Souza 	ret = btrfs_search_slot(NULL, root, key, path, 0, 0);
22740ff40a91SMarcos Paulo de Souza 	if (ret > 0)
22750ff40a91SMarcos Paulo de Souza 		ret = btrfs_previous_item(root, path, key->objectid, key->type);
22760ff40a91SMarcos Paulo de Souza 
22770ff40a91SMarcos Paulo de Souza 	if (ret == 0)
22780ff40a91SMarcos Paulo de Souza 		btrfs_item_key_to_cpu(path->nodes[0], key, path->slots[0]);
22790ff40a91SMarcos Paulo de Souza 
22800ff40a91SMarcos Paulo de Souza 	return ret;
22810ff40a91SMarcos Paulo de Souza }
22820ff40a91SMarcos Paulo de Souza 
22830ff40a91SMarcos Paulo de Souza /*
228474123bd7SChris Mason  * adjust the pointers going up the tree, starting at level
228574123bd7SChris Mason  * making sure the right key of each node is points to 'key'.
228674123bd7SChris Mason  * This is used after shifting pointers to the left, so it stops
228774123bd7SChris Mason  * fixing up pointers when a given leaf/node is not in slot 0 of the
228874123bd7SChris Mason  * higher levels
2289aa5d6bedSChris Mason  *
229074123bd7SChris Mason  */
2291b167fa91SNikolay Borisov static void fixup_low_keys(struct btrfs_path *path,
22925f39d397SChris Mason 			   struct btrfs_disk_key *key, int level)
2293be0e5c09SChris Mason {
2294be0e5c09SChris Mason 	int i;
22955f39d397SChris Mason 	struct extent_buffer *t;
22960e82bcfeSDavid Sterba 	int ret;
22975f39d397SChris Mason 
2298234b63a0SChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
2299be0e5c09SChris Mason 		int tslot = path->slots[i];
23000e82bcfeSDavid Sterba 
2301eb60ceacSChris Mason 		if (!path->nodes[i])
2302be0e5c09SChris Mason 			break;
23035f39d397SChris Mason 		t = path->nodes[i];
2304f3a84ccdSFilipe Manana 		ret = btrfs_tree_mod_log_insert_key(t, tslot,
2305f3a84ccdSFilipe Manana 				BTRFS_MOD_LOG_KEY_REPLACE, GFP_ATOMIC);
23060e82bcfeSDavid Sterba 		BUG_ON(ret < 0);
23075f39d397SChris Mason 		btrfs_set_node_key(t, key, tslot);
2308d6025579SChris Mason 		btrfs_mark_buffer_dirty(path->nodes[i]);
2309be0e5c09SChris Mason 		if (tslot != 0)
2310be0e5c09SChris Mason 			break;
2311be0e5c09SChris Mason 	}
2312be0e5c09SChris Mason }
2313be0e5c09SChris Mason 
231474123bd7SChris Mason /*
231531840ae1SZheng Yan  * update item key.
231631840ae1SZheng Yan  *
231731840ae1SZheng Yan  * This function isn't completely safe. It's the caller's responsibility
231831840ae1SZheng Yan  * that the new key won't break the order
231931840ae1SZheng Yan  */
2320b7a0365eSDaniel Dressler void btrfs_set_item_key_safe(struct btrfs_fs_info *fs_info,
2321b7a0365eSDaniel Dressler 			     struct btrfs_path *path,
2322310712b2SOmar Sandoval 			     const struct btrfs_key *new_key)
232331840ae1SZheng Yan {
232431840ae1SZheng Yan 	struct btrfs_disk_key disk_key;
232531840ae1SZheng Yan 	struct extent_buffer *eb;
232631840ae1SZheng Yan 	int slot;
232731840ae1SZheng Yan 
232831840ae1SZheng Yan 	eb = path->nodes[0];
232931840ae1SZheng Yan 	slot = path->slots[0];
233031840ae1SZheng Yan 	if (slot > 0) {
233131840ae1SZheng Yan 		btrfs_item_key(eb, &disk_key, slot - 1);
23327c15d410SQu Wenruo 		if (unlikely(comp_keys(&disk_key, new_key) >= 0)) {
23337c15d410SQu Wenruo 			btrfs_crit(fs_info,
23347c15d410SQu Wenruo 		"slot %u key (%llu %u %llu) new key (%llu %u %llu)",
23357c15d410SQu Wenruo 				   slot, btrfs_disk_key_objectid(&disk_key),
23367c15d410SQu Wenruo 				   btrfs_disk_key_type(&disk_key),
23377c15d410SQu Wenruo 				   btrfs_disk_key_offset(&disk_key),
23387c15d410SQu Wenruo 				   new_key->objectid, new_key->type,
23397c15d410SQu Wenruo 				   new_key->offset);
23407c15d410SQu Wenruo 			btrfs_print_leaf(eb);
23417c15d410SQu Wenruo 			BUG();
23427c15d410SQu Wenruo 		}
234331840ae1SZheng Yan 	}
234431840ae1SZheng Yan 	if (slot < btrfs_header_nritems(eb) - 1) {
234531840ae1SZheng Yan 		btrfs_item_key(eb, &disk_key, slot + 1);
23467c15d410SQu Wenruo 		if (unlikely(comp_keys(&disk_key, new_key) <= 0)) {
23477c15d410SQu Wenruo 			btrfs_crit(fs_info,
23487c15d410SQu Wenruo 		"slot %u key (%llu %u %llu) new key (%llu %u %llu)",
23497c15d410SQu Wenruo 				   slot, btrfs_disk_key_objectid(&disk_key),
23507c15d410SQu Wenruo 				   btrfs_disk_key_type(&disk_key),
23517c15d410SQu Wenruo 				   btrfs_disk_key_offset(&disk_key),
23527c15d410SQu Wenruo 				   new_key->objectid, new_key->type,
23537c15d410SQu Wenruo 				   new_key->offset);
23547c15d410SQu Wenruo 			btrfs_print_leaf(eb);
23557c15d410SQu Wenruo 			BUG();
23567c15d410SQu Wenruo 		}
235731840ae1SZheng Yan 	}
235831840ae1SZheng Yan 
235931840ae1SZheng Yan 	btrfs_cpu_key_to_disk(&disk_key, new_key);
236031840ae1SZheng Yan 	btrfs_set_item_key(eb, &disk_key, slot);
236131840ae1SZheng Yan 	btrfs_mark_buffer_dirty(eb);
236231840ae1SZheng Yan 	if (slot == 0)
2363b167fa91SNikolay Borisov 		fixup_low_keys(path, &disk_key, 1);
236431840ae1SZheng Yan }
236531840ae1SZheng Yan 
236631840ae1SZheng Yan /*
2367d16c702fSQu Wenruo  * Check key order of two sibling extent buffers.
2368d16c702fSQu Wenruo  *
2369d16c702fSQu Wenruo  * Return true if something is wrong.
2370d16c702fSQu Wenruo  * Return false if everything is fine.
2371d16c702fSQu Wenruo  *
2372d16c702fSQu Wenruo  * Tree-checker only works inside one tree block, thus the following
2373d16c702fSQu Wenruo  * corruption can not be detected by tree-checker:
2374d16c702fSQu Wenruo  *
2375d16c702fSQu Wenruo  * Leaf @left			| Leaf @right
2376d16c702fSQu Wenruo  * --------------------------------------------------------------
2377d16c702fSQu Wenruo  * | 1 | 2 | 3 | 4 | 5 | f6 |   | 7 | 8 |
2378d16c702fSQu Wenruo  *
2379d16c702fSQu Wenruo  * Key f6 in leaf @left itself is valid, but not valid when the next
2380d16c702fSQu Wenruo  * key in leaf @right is 7.
2381d16c702fSQu Wenruo  * This can only be checked at tree block merge time.
2382d16c702fSQu Wenruo  * And since tree checker has ensured all key order in each tree block
2383d16c702fSQu Wenruo  * is correct, we only need to bother the last key of @left and the first
2384d16c702fSQu Wenruo  * key of @right.
2385d16c702fSQu Wenruo  */
2386d16c702fSQu Wenruo static bool check_sibling_keys(struct extent_buffer *left,
2387d16c702fSQu Wenruo 			       struct extent_buffer *right)
2388d16c702fSQu Wenruo {
2389d16c702fSQu Wenruo 	struct btrfs_key left_last;
2390d16c702fSQu Wenruo 	struct btrfs_key right_first;
2391d16c702fSQu Wenruo 	int level = btrfs_header_level(left);
2392d16c702fSQu Wenruo 	int nr_left = btrfs_header_nritems(left);
2393d16c702fSQu Wenruo 	int nr_right = btrfs_header_nritems(right);
2394d16c702fSQu Wenruo 
2395d16c702fSQu Wenruo 	/* No key to check in one of the tree blocks */
2396d16c702fSQu Wenruo 	if (!nr_left || !nr_right)
2397d16c702fSQu Wenruo 		return false;
2398d16c702fSQu Wenruo 
2399d16c702fSQu Wenruo 	if (level) {
2400d16c702fSQu Wenruo 		btrfs_node_key_to_cpu(left, &left_last, nr_left - 1);
2401d16c702fSQu Wenruo 		btrfs_node_key_to_cpu(right, &right_first, 0);
2402d16c702fSQu Wenruo 	} else {
2403d16c702fSQu Wenruo 		btrfs_item_key_to_cpu(left, &left_last, nr_left - 1);
2404d16c702fSQu Wenruo 		btrfs_item_key_to_cpu(right, &right_first, 0);
2405d16c702fSQu Wenruo 	}
2406d16c702fSQu Wenruo 
2407d16c702fSQu Wenruo 	if (btrfs_comp_cpu_keys(&left_last, &right_first) >= 0) {
2408d16c702fSQu Wenruo 		btrfs_crit(left->fs_info,
2409d16c702fSQu Wenruo "bad key order, sibling blocks, left last (%llu %u %llu) right first (%llu %u %llu)",
2410d16c702fSQu Wenruo 			   left_last.objectid, left_last.type,
2411d16c702fSQu Wenruo 			   left_last.offset, right_first.objectid,
2412d16c702fSQu Wenruo 			   right_first.type, right_first.offset);
2413d16c702fSQu Wenruo 		return true;
2414d16c702fSQu Wenruo 	}
2415d16c702fSQu Wenruo 	return false;
2416d16c702fSQu Wenruo }
2417d16c702fSQu Wenruo 
2418d16c702fSQu Wenruo /*
241974123bd7SChris Mason  * try to push data from one node into the next node left in the
242079f95c82SChris Mason  * tree.
2421aa5d6bedSChris Mason  *
2422aa5d6bedSChris Mason  * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
2423aa5d6bedSChris Mason  * error, and > 0 if there was no room in the left hand block.
242474123bd7SChris Mason  */
242598ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
24262ff7e61eSJeff Mahoney 			  struct extent_buffer *dst,
2427971a1f66SChris Mason 			  struct extent_buffer *src, int empty)
2428be0e5c09SChris Mason {
2429d30a668fSDavid Sterba 	struct btrfs_fs_info *fs_info = trans->fs_info;
2430be0e5c09SChris Mason 	int push_items = 0;
2431bb803951SChris Mason 	int src_nritems;
2432bb803951SChris Mason 	int dst_nritems;
2433aa5d6bedSChris Mason 	int ret = 0;
2434be0e5c09SChris Mason 
24355f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
24365f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
24370b246afaSJeff Mahoney 	push_items = BTRFS_NODEPTRS_PER_BLOCK(fs_info) - dst_nritems;
24387bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
24397bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
244054aa1f4dSChris Mason 
2441bce4eae9SChris Mason 	if (!empty && src_nritems <= 8)
2442971a1f66SChris Mason 		return 1;
2443971a1f66SChris Mason 
2444d397712bSChris Mason 	if (push_items <= 0)
2445be0e5c09SChris Mason 		return 1;
2446be0e5c09SChris Mason 
2447bce4eae9SChris Mason 	if (empty) {
2448971a1f66SChris Mason 		push_items = min(src_nritems, push_items);
2449bce4eae9SChris Mason 		if (push_items < src_nritems) {
2450bce4eae9SChris Mason 			/* leave at least 8 pointers in the node if
2451bce4eae9SChris Mason 			 * we aren't going to empty it
2452bce4eae9SChris Mason 			 */
2453bce4eae9SChris Mason 			if (src_nritems - push_items < 8) {
2454bce4eae9SChris Mason 				if (push_items <= 8)
2455bce4eae9SChris Mason 					return 1;
2456bce4eae9SChris Mason 				push_items -= 8;
2457bce4eae9SChris Mason 			}
2458bce4eae9SChris Mason 		}
2459bce4eae9SChris Mason 	} else
2460bce4eae9SChris Mason 		push_items = min(src_nritems - 8, push_items);
246179f95c82SChris Mason 
2462d16c702fSQu Wenruo 	/* dst is the left eb, src is the middle eb */
2463d16c702fSQu Wenruo 	if (check_sibling_keys(dst, src)) {
2464d16c702fSQu Wenruo 		ret = -EUCLEAN;
2465d16c702fSQu Wenruo 		btrfs_abort_transaction(trans, ret);
2466d16c702fSQu Wenruo 		return ret;
2467d16c702fSQu Wenruo 	}
2468f3a84ccdSFilipe Manana 	ret = btrfs_tree_mod_log_eb_copy(dst, src, dst_nritems, 0, push_items);
24695de865eeSFilipe David Borba Manana 	if (ret) {
247066642832SJeff Mahoney 		btrfs_abort_transaction(trans, ret);
24715de865eeSFilipe David Borba Manana 		return ret;
24725de865eeSFilipe David Borba Manana 	}
24735f39d397SChris Mason 	copy_extent_buffer(dst, src,
24745f39d397SChris Mason 			   btrfs_node_key_ptr_offset(dst_nritems),
24755f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
2476123abc88SChris Mason 			   push_items * sizeof(struct btrfs_key_ptr));
24775f39d397SChris Mason 
2478bb803951SChris Mason 	if (push_items < src_nritems) {
247957911b8bSJan Schmidt 		/*
2480f3a84ccdSFilipe Manana 		 * Don't call btrfs_tree_mod_log_insert_move() here, key removal
2481f3a84ccdSFilipe Manana 		 * was already fully logged by btrfs_tree_mod_log_eb_copy() above.
248257911b8bSJan Schmidt 		 */
24835f39d397SChris Mason 		memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
24845f39d397SChris Mason 				      btrfs_node_key_ptr_offset(push_items),
2485e2fa7227SChris Mason 				      (src_nritems - push_items) *
2486123abc88SChris Mason 				      sizeof(struct btrfs_key_ptr));
2487bb803951SChris Mason 	}
24885f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
24895f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
24905f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
24915f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
249231840ae1SZheng Yan 
2493bb803951SChris Mason 	return ret;
2494be0e5c09SChris Mason }
2495be0e5c09SChris Mason 
249697571fd0SChris Mason /*
249779f95c82SChris Mason  * try to push data from one node into the next node right in the
249879f95c82SChris Mason  * tree.
249979f95c82SChris Mason  *
250079f95c82SChris Mason  * returns 0 if some ptrs were pushed, < 0 if there was some horrible
250179f95c82SChris Mason  * error, and > 0 if there was no room in the right hand block.
250279f95c82SChris Mason  *
250379f95c82SChris Mason  * this will  only push up to 1/2 the contents of the left node over
250479f95c82SChris Mason  */
25055f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
25065f39d397SChris Mason 			      struct extent_buffer *dst,
25075f39d397SChris Mason 			      struct extent_buffer *src)
250879f95c82SChris Mason {
250955d32ed8SDavid Sterba 	struct btrfs_fs_info *fs_info = trans->fs_info;
251079f95c82SChris Mason 	int push_items = 0;
251179f95c82SChris Mason 	int max_push;
251279f95c82SChris Mason 	int src_nritems;
251379f95c82SChris Mason 	int dst_nritems;
251479f95c82SChris Mason 	int ret = 0;
251579f95c82SChris Mason 
25167bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
25177bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
25187bb86316SChris Mason 
25195f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
25205f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
25210b246afaSJeff Mahoney 	push_items = BTRFS_NODEPTRS_PER_BLOCK(fs_info) - dst_nritems;
2522d397712bSChris Mason 	if (push_items <= 0)
252379f95c82SChris Mason 		return 1;
2524bce4eae9SChris Mason 
2525d397712bSChris Mason 	if (src_nritems < 4)
2526bce4eae9SChris Mason 		return 1;
252779f95c82SChris Mason 
252879f95c82SChris Mason 	max_push = src_nritems / 2 + 1;
252979f95c82SChris Mason 	/* don't try to empty the node */
2530d397712bSChris Mason 	if (max_push >= src_nritems)
253179f95c82SChris Mason 		return 1;
2532252c38f0SYan 
253379f95c82SChris Mason 	if (max_push < push_items)
253479f95c82SChris Mason 		push_items = max_push;
253579f95c82SChris Mason 
2536d16c702fSQu Wenruo 	/* dst is the right eb, src is the middle eb */
2537d16c702fSQu Wenruo 	if (check_sibling_keys(src, dst)) {
2538d16c702fSQu Wenruo 		ret = -EUCLEAN;
2539d16c702fSQu Wenruo 		btrfs_abort_transaction(trans, ret);
2540d16c702fSQu Wenruo 		return ret;
2541d16c702fSQu Wenruo 	}
2542f3a84ccdSFilipe Manana 	ret = btrfs_tree_mod_log_insert_move(dst, push_items, 0, dst_nritems);
2543bf1d3425SDavid Sterba 	BUG_ON(ret < 0);
25445f39d397SChris Mason 	memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
25455f39d397SChris Mason 				      btrfs_node_key_ptr_offset(0),
25465f39d397SChris Mason 				      (dst_nritems) *
25475f39d397SChris Mason 				      sizeof(struct btrfs_key_ptr));
2548d6025579SChris Mason 
2549f3a84ccdSFilipe Manana 	ret = btrfs_tree_mod_log_eb_copy(dst, src, 0, src_nritems - push_items,
2550ed874f0dSDavid Sterba 					 push_items);
25515de865eeSFilipe David Borba Manana 	if (ret) {
255266642832SJeff Mahoney 		btrfs_abort_transaction(trans, ret);
25535de865eeSFilipe David Borba Manana 		return ret;
25545de865eeSFilipe David Borba Manana 	}
25555f39d397SChris Mason 	copy_extent_buffer(dst, src,
25565f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
25575f39d397SChris Mason 			   btrfs_node_key_ptr_offset(src_nritems - push_items),
2558123abc88SChris Mason 			   push_items * sizeof(struct btrfs_key_ptr));
255979f95c82SChris Mason 
25605f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
25615f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
256279f95c82SChris Mason 
25635f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
25645f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
256531840ae1SZheng Yan 
256679f95c82SChris Mason 	return ret;
256779f95c82SChris Mason }
256879f95c82SChris Mason 
256979f95c82SChris Mason /*
257097571fd0SChris Mason  * helper function to insert a new root level in the tree.
257197571fd0SChris Mason  * A new node is allocated, and a single item is inserted to
257297571fd0SChris Mason  * point to the existing root
2573aa5d6bedSChris Mason  *
2574aa5d6bedSChris Mason  * returns zero on success or < 0 on failure.
257597571fd0SChris Mason  */
2576d397712bSChris Mason static noinline int insert_new_root(struct btrfs_trans_handle *trans,
25775f39d397SChris Mason 			   struct btrfs_root *root,
2578fdd99c72SLiu Bo 			   struct btrfs_path *path, int level)
257974123bd7SChris Mason {
25800b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
25817bb86316SChris Mason 	u64 lower_gen;
25825f39d397SChris Mason 	struct extent_buffer *lower;
25835f39d397SChris Mason 	struct extent_buffer *c;
2584925baeddSChris Mason 	struct extent_buffer *old;
25855f39d397SChris Mason 	struct btrfs_disk_key lower_key;
2586d9d19a01SDavid Sterba 	int ret;
25875c680ed6SChris Mason 
25885c680ed6SChris Mason 	BUG_ON(path->nodes[level]);
25895c680ed6SChris Mason 	BUG_ON(path->nodes[level-1] != root->node);
25905c680ed6SChris Mason 
25917bb86316SChris Mason 	lower = path->nodes[level-1];
25927bb86316SChris Mason 	if (level == 1)
25937bb86316SChris Mason 		btrfs_item_key(lower, &lower_key, 0);
25947bb86316SChris Mason 	else
25957bb86316SChris Mason 		btrfs_node_key(lower, &lower_key, 0);
25967bb86316SChris Mason 
259779bd3712SFilipe Manana 	c = btrfs_alloc_tree_block(trans, root, 0, root->root_key.objectid,
259879bd3712SFilipe Manana 				   &lower_key, level, root->node->start, 0,
2599cf6f34aaSJosef Bacik 				   BTRFS_NESTING_NEW_ROOT);
26005f39d397SChris Mason 	if (IS_ERR(c))
26015f39d397SChris Mason 		return PTR_ERR(c);
2602925baeddSChris Mason 
26030b246afaSJeff Mahoney 	root_add_used(root, fs_info->nodesize);
2604f0486c68SYan, Zheng 
26055f39d397SChris Mason 	btrfs_set_header_nritems(c, 1);
26065f39d397SChris Mason 	btrfs_set_node_key(c, &lower_key, 0);
2607db94535dSChris Mason 	btrfs_set_node_blockptr(c, 0, lower->start);
26087bb86316SChris Mason 	lower_gen = btrfs_header_generation(lower);
260931840ae1SZheng Yan 	WARN_ON(lower_gen != trans->transid);
26107bb86316SChris Mason 
26117bb86316SChris Mason 	btrfs_set_node_ptr_generation(c, 0, lower_gen);
26125f39d397SChris Mason 
26135f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
2614d5719762SChris Mason 
2615925baeddSChris Mason 	old = root->node;
2616406808abSFilipe Manana 	ret = btrfs_tree_mod_log_insert_root(root->node, c, false);
2617d9d19a01SDavid Sterba 	BUG_ON(ret < 0);
2618240f62c8SChris Mason 	rcu_assign_pointer(root->node, c);
2619925baeddSChris Mason 
2620925baeddSChris Mason 	/* the super has an extra ref to root->node */
2621925baeddSChris Mason 	free_extent_buffer(old);
2622925baeddSChris Mason 
26230b86a832SChris Mason 	add_root_to_dirty_list(root);
262467439dadSDavid Sterba 	atomic_inc(&c->refs);
26255f39d397SChris Mason 	path->nodes[level] = c;
2626ac5887c8SJosef Bacik 	path->locks[level] = BTRFS_WRITE_LOCK;
262774123bd7SChris Mason 	path->slots[level] = 0;
262874123bd7SChris Mason 	return 0;
262974123bd7SChris Mason }
26305c680ed6SChris Mason 
26315c680ed6SChris Mason /*
26325c680ed6SChris Mason  * worker function to insert a single pointer in a node.
26335c680ed6SChris Mason  * the node should have enough room for the pointer already
263497571fd0SChris Mason  *
26355c680ed6SChris Mason  * slot and level indicate where you want the key to go, and
26365c680ed6SChris Mason  * blocknr is the block the key points to.
26375c680ed6SChris Mason  */
2638143bede5SJeff Mahoney static void insert_ptr(struct btrfs_trans_handle *trans,
26396ad3cf6dSDavid Sterba 		       struct btrfs_path *path,
2640143bede5SJeff Mahoney 		       struct btrfs_disk_key *key, u64 bytenr,
2641c3e06965SJan Schmidt 		       int slot, int level)
26425c680ed6SChris Mason {
26435f39d397SChris Mason 	struct extent_buffer *lower;
26445c680ed6SChris Mason 	int nritems;
2645f3ea38daSJan Schmidt 	int ret;
26465c680ed6SChris Mason 
26475c680ed6SChris Mason 	BUG_ON(!path->nodes[level]);
264849d0c642SFilipe Manana 	btrfs_assert_tree_write_locked(path->nodes[level]);
26495f39d397SChris Mason 	lower = path->nodes[level];
26505f39d397SChris Mason 	nritems = btrfs_header_nritems(lower);
2651c293498bSStoyan Gaydarov 	BUG_ON(slot > nritems);
26526ad3cf6dSDavid Sterba 	BUG_ON(nritems == BTRFS_NODEPTRS_PER_BLOCK(trans->fs_info));
265374123bd7SChris Mason 	if (slot != nritems) {
2654bf1d3425SDavid Sterba 		if (level) {
2655f3a84ccdSFilipe Manana 			ret = btrfs_tree_mod_log_insert_move(lower, slot + 1,
2656f3a84ccdSFilipe Manana 					slot, nritems - slot);
2657bf1d3425SDavid Sterba 			BUG_ON(ret < 0);
2658bf1d3425SDavid Sterba 		}
26595f39d397SChris Mason 		memmove_extent_buffer(lower,
26605f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
26615f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
2662123abc88SChris Mason 			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
266374123bd7SChris Mason 	}
2664c3e06965SJan Schmidt 	if (level) {
2665f3a84ccdSFilipe Manana 		ret = btrfs_tree_mod_log_insert_key(lower, slot,
2666f3a84ccdSFilipe Manana 					    BTRFS_MOD_LOG_KEY_ADD, GFP_NOFS);
2667f3ea38daSJan Schmidt 		BUG_ON(ret < 0);
2668f3ea38daSJan Schmidt 	}
26695f39d397SChris Mason 	btrfs_set_node_key(lower, key, slot);
2670db94535dSChris Mason 	btrfs_set_node_blockptr(lower, slot, bytenr);
267174493f7aSChris Mason 	WARN_ON(trans->transid == 0);
267274493f7aSChris Mason 	btrfs_set_node_ptr_generation(lower, slot, trans->transid);
26735f39d397SChris Mason 	btrfs_set_header_nritems(lower, nritems + 1);
26745f39d397SChris Mason 	btrfs_mark_buffer_dirty(lower);
267574123bd7SChris Mason }
267674123bd7SChris Mason 
267797571fd0SChris Mason /*
267897571fd0SChris Mason  * split the node at the specified level in path in two.
267997571fd0SChris Mason  * The path is corrected to point to the appropriate node after the split
268097571fd0SChris Mason  *
268197571fd0SChris Mason  * Before splitting this tries to make some room in the node by pushing
268297571fd0SChris Mason  * left and right, if either one works, it returns right away.
2683aa5d6bedSChris Mason  *
2684aa5d6bedSChris Mason  * returns 0 on success and < 0 on failure
268597571fd0SChris Mason  */
2686e02119d5SChris Mason static noinline int split_node(struct btrfs_trans_handle *trans,
2687e02119d5SChris Mason 			       struct btrfs_root *root,
2688e02119d5SChris Mason 			       struct btrfs_path *path, int level)
2689be0e5c09SChris Mason {
26900b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
26915f39d397SChris Mason 	struct extent_buffer *c;
26925f39d397SChris Mason 	struct extent_buffer *split;
26935f39d397SChris Mason 	struct btrfs_disk_key disk_key;
2694be0e5c09SChris Mason 	int mid;
26955c680ed6SChris Mason 	int ret;
26967518a238SChris Mason 	u32 c_nritems;
2697be0e5c09SChris Mason 
26985f39d397SChris Mason 	c = path->nodes[level];
26997bb86316SChris Mason 	WARN_ON(btrfs_header_generation(c) != trans->transid);
27005f39d397SChris Mason 	if (c == root->node) {
2701d9abbf1cSJan Schmidt 		/*
270290f8d62eSJan Schmidt 		 * trying to split the root, lets make a new one
270390f8d62eSJan Schmidt 		 *
2704fdd99c72SLiu Bo 		 * tree mod log: We don't log_removal old root in
270590f8d62eSJan Schmidt 		 * insert_new_root, because that root buffer will be kept as a
270690f8d62eSJan Schmidt 		 * normal node. We are going to log removal of half of the
2707f3a84ccdSFilipe Manana 		 * elements below with btrfs_tree_mod_log_eb_copy(). We're
2708f3a84ccdSFilipe Manana 		 * holding a tree lock on the buffer, which is why we cannot
2709f3a84ccdSFilipe Manana 		 * race with other tree_mod_log users.
2710d9abbf1cSJan Schmidt 		 */
2711fdd99c72SLiu Bo 		ret = insert_new_root(trans, root, path, level + 1);
27125c680ed6SChris Mason 		if (ret)
27135c680ed6SChris Mason 			return ret;
2714b3612421SChris Mason 	} else {
2715e66f709bSChris Mason 		ret = push_nodes_for_insert(trans, root, path, level);
27165f39d397SChris Mason 		c = path->nodes[level];
27175f39d397SChris Mason 		if (!ret && btrfs_header_nritems(c) <
27180b246afaSJeff Mahoney 		    BTRFS_NODEPTRS_PER_BLOCK(fs_info) - 3)
2719e66f709bSChris Mason 			return 0;
272054aa1f4dSChris Mason 		if (ret < 0)
272154aa1f4dSChris Mason 			return ret;
27225c680ed6SChris Mason 	}
2723e66f709bSChris Mason 
27245f39d397SChris Mason 	c_nritems = btrfs_header_nritems(c);
27255d4f98a2SYan Zheng 	mid = (c_nritems + 1) / 2;
27265d4f98a2SYan Zheng 	btrfs_node_key(c, &disk_key, mid);
27277bb86316SChris Mason 
272879bd3712SFilipe Manana 	split = btrfs_alloc_tree_block(trans, root, 0, root->root_key.objectid,
272979bd3712SFilipe Manana 				       &disk_key, level, c->start, 0,
273079bd3712SFilipe Manana 				       BTRFS_NESTING_SPLIT);
27315f39d397SChris Mason 	if (IS_ERR(split))
27325f39d397SChris Mason 		return PTR_ERR(split);
273354aa1f4dSChris Mason 
27340b246afaSJeff Mahoney 	root_add_used(root, fs_info->nodesize);
2735bc877d28SNikolay Borisov 	ASSERT(btrfs_header_level(c) == level);
27365f39d397SChris Mason 
2737f3a84ccdSFilipe Manana 	ret = btrfs_tree_mod_log_eb_copy(split, c, 0, mid, c_nritems - mid);
27385de865eeSFilipe David Borba Manana 	if (ret) {
273966642832SJeff Mahoney 		btrfs_abort_transaction(trans, ret);
27405de865eeSFilipe David Borba Manana 		return ret;
27415de865eeSFilipe David Borba Manana 	}
27425f39d397SChris Mason 	copy_extent_buffer(split, c,
27435f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
27445f39d397SChris Mason 			   btrfs_node_key_ptr_offset(mid),
2745123abc88SChris Mason 			   (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
27465f39d397SChris Mason 	btrfs_set_header_nritems(split, c_nritems - mid);
27475f39d397SChris Mason 	btrfs_set_header_nritems(c, mid);
2748aa5d6bedSChris Mason 
27495f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
27505f39d397SChris Mason 	btrfs_mark_buffer_dirty(split);
27515f39d397SChris Mason 
27526ad3cf6dSDavid Sterba 	insert_ptr(trans, path, &disk_key, split->start,
2753c3e06965SJan Schmidt 		   path->slots[level + 1] + 1, level + 1);
2754aa5d6bedSChris Mason 
27555de08d7dSChris Mason 	if (path->slots[level] >= mid) {
27565c680ed6SChris Mason 		path->slots[level] -= mid;
2757925baeddSChris Mason 		btrfs_tree_unlock(c);
27585f39d397SChris Mason 		free_extent_buffer(c);
27595f39d397SChris Mason 		path->nodes[level] = split;
27605c680ed6SChris Mason 		path->slots[level + 1] += 1;
2761eb60ceacSChris Mason 	} else {
2762925baeddSChris Mason 		btrfs_tree_unlock(split);
27635f39d397SChris Mason 		free_extent_buffer(split);
2764be0e5c09SChris Mason 	}
2765d5286a92SNikolay Borisov 	return 0;
2766be0e5c09SChris Mason }
2767be0e5c09SChris Mason 
276874123bd7SChris Mason /*
276974123bd7SChris Mason  * how many bytes are required to store the items in a leaf.  start
277074123bd7SChris Mason  * and nr indicate which items in the leaf to check.  This totals up the
277174123bd7SChris Mason  * space used both by the item structs and the item data
277274123bd7SChris Mason  */
27735f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr)
2774be0e5c09SChris Mason {
2775be0e5c09SChris Mason 	int data_len;
27765f39d397SChris Mason 	int nritems = btrfs_header_nritems(l);
2777d4dbff95SChris Mason 	int end = min(nritems, start + nr) - 1;
2778be0e5c09SChris Mason 
2779be0e5c09SChris Mason 	if (!nr)
2780be0e5c09SChris Mason 		return 0;
27813212fa14SJosef Bacik 	data_len = btrfs_item_offset(l, start) + btrfs_item_size(l, start);
27823212fa14SJosef Bacik 	data_len = data_len - btrfs_item_offset(l, end);
27830783fcfcSChris Mason 	data_len += sizeof(struct btrfs_item) * nr;
2784d4dbff95SChris Mason 	WARN_ON(data_len < 0);
2785be0e5c09SChris Mason 	return data_len;
2786be0e5c09SChris Mason }
2787be0e5c09SChris Mason 
278874123bd7SChris Mason /*
2789d4dbff95SChris Mason  * The space between the end of the leaf items and
2790d4dbff95SChris Mason  * the start of the leaf data.  IOW, how much room
2791d4dbff95SChris Mason  * the leaf has left for both items and data
2792d4dbff95SChris Mason  */
2793e902baacSDavid Sterba noinline int btrfs_leaf_free_space(struct extent_buffer *leaf)
2794d4dbff95SChris Mason {
2795e902baacSDavid Sterba 	struct btrfs_fs_info *fs_info = leaf->fs_info;
27965f39d397SChris Mason 	int nritems = btrfs_header_nritems(leaf);
27975f39d397SChris Mason 	int ret;
27980b246afaSJeff Mahoney 
27990b246afaSJeff Mahoney 	ret = BTRFS_LEAF_DATA_SIZE(fs_info) - leaf_space_used(leaf, 0, nritems);
28005f39d397SChris Mason 	if (ret < 0) {
28010b246afaSJeff Mahoney 		btrfs_crit(fs_info,
2802efe120a0SFrank Holton 			   "leaf free space ret %d, leaf data size %lu, used %d nritems %d",
2803da17066cSJeff Mahoney 			   ret,
28040b246afaSJeff Mahoney 			   (unsigned long) BTRFS_LEAF_DATA_SIZE(fs_info),
28055f39d397SChris Mason 			   leaf_space_used(leaf, 0, nritems), nritems);
28065f39d397SChris Mason 	}
28075f39d397SChris Mason 	return ret;
2808d4dbff95SChris Mason }
2809d4dbff95SChris Mason 
281099d8f83cSChris Mason /*
281199d8f83cSChris Mason  * min slot controls the lowest index we're willing to push to the
281299d8f83cSChris Mason  * right.  We'll push up to and including min_slot, but no lower
281399d8f83cSChris Mason  */
2814f72f0010SDavid Sterba static noinline int __push_leaf_right(struct btrfs_path *path,
281544871b1bSChris Mason 				      int data_size, int empty,
281644871b1bSChris Mason 				      struct extent_buffer *right,
281799d8f83cSChris Mason 				      int free_space, u32 left_nritems,
281899d8f83cSChris Mason 				      u32 min_slot)
281900ec4c51SChris Mason {
2820f72f0010SDavid Sterba 	struct btrfs_fs_info *fs_info = right->fs_info;
28215f39d397SChris Mason 	struct extent_buffer *left = path->nodes[0];
282244871b1bSChris Mason 	struct extent_buffer *upper = path->nodes[1];
2823cfed81a0SChris Mason 	struct btrfs_map_token token;
28245f39d397SChris Mason 	struct btrfs_disk_key disk_key;
282500ec4c51SChris Mason 	int slot;
282634a38218SChris Mason 	u32 i;
282700ec4c51SChris Mason 	int push_space = 0;
282800ec4c51SChris Mason 	int push_items = 0;
282934a38218SChris Mason 	u32 nr;
28307518a238SChris Mason 	u32 right_nritems;
28315f39d397SChris Mason 	u32 data_end;
2832db94535dSChris Mason 	u32 this_item_size;
283300ec4c51SChris Mason 
283434a38218SChris Mason 	if (empty)
283534a38218SChris Mason 		nr = 0;
283634a38218SChris Mason 	else
283799d8f83cSChris Mason 		nr = max_t(u32, 1, min_slot);
283834a38218SChris Mason 
283931840ae1SZheng Yan 	if (path->slots[0] >= left_nritems)
284087b29b20SYan Zheng 		push_space += data_size;
284131840ae1SZheng Yan 
284244871b1bSChris Mason 	slot = path->slots[1];
284334a38218SChris Mason 	i = left_nritems - 1;
284434a38218SChris Mason 	while (i >= nr) {
284531840ae1SZheng Yan 		if (!empty && push_items > 0) {
284631840ae1SZheng Yan 			if (path->slots[0] > i)
284731840ae1SZheng Yan 				break;
284831840ae1SZheng Yan 			if (path->slots[0] == i) {
2849e902baacSDavid Sterba 				int space = btrfs_leaf_free_space(left);
2850e902baacSDavid Sterba 
285131840ae1SZheng Yan 				if (space + push_space * 2 > free_space)
285231840ae1SZheng Yan 					break;
285331840ae1SZheng Yan 			}
285431840ae1SZheng Yan 		}
285531840ae1SZheng Yan 
285600ec4c51SChris Mason 		if (path->slots[0] == i)
285787b29b20SYan Zheng 			push_space += data_size;
2858db94535dSChris Mason 
28593212fa14SJosef Bacik 		this_item_size = btrfs_item_size(left, i);
286074794207SJosef Bacik 		if (this_item_size + sizeof(struct btrfs_item) +
286174794207SJosef Bacik 		    push_space > free_space)
286200ec4c51SChris Mason 			break;
286331840ae1SZheng Yan 
286400ec4c51SChris Mason 		push_items++;
286574794207SJosef Bacik 		push_space += this_item_size + sizeof(struct btrfs_item);
286634a38218SChris Mason 		if (i == 0)
286734a38218SChris Mason 			break;
286834a38218SChris Mason 		i--;
2869db94535dSChris Mason 	}
28705f39d397SChris Mason 
2871925baeddSChris Mason 	if (push_items == 0)
2872925baeddSChris Mason 		goto out_unlock;
28735f39d397SChris Mason 
28746c1500f2SJulia Lawall 	WARN_ON(!empty && push_items == left_nritems);
28755f39d397SChris Mason 
287600ec4c51SChris Mason 	/* push left to right */
28775f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right);
287834a38218SChris Mason 
2879dc2e724eSJosef Bacik 	push_space = btrfs_item_data_end(left, left_nritems - push_items);
28808f881e8cSDavid Sterba 	push_space -= leaf_data_end(left);
28815f39d397SChris Mason 
288200ec4c51SChris Mason 	/* make room in the right data area */
28838f881e8cSDavid Sterba 	data_end = leaf_data_end(right);
28845f39d397SChris Mason 	memmove_extent_buffer(right,
28853d9ec8c4SNikolay Borisov 			      BTRFS_LEAF_DATA_OFFSET + data_end - push_space,
28863d9ec8c4SNikolay Borisov 			      BTRFS_LEAF_DATA_OFFSET + data_end,
28870b246afaSJeff Mahoney 			      BTRFS_LEAF_DATA_SIZE(fs_info) - data_end);
28885f39d397SChris Mason 
288900ec4c51SChris Mason 	/* copy from the left data area */
28903d9ec8c4SNikolay Borisov 	copy_extent_buffer(right, left, BTRFS_LEAF_DATA_OFFSET +
28910b246afaSJeff Mahoney 		     BTRFS_LEAF_DATA_SIZE(fs_info) - push_space,
28928f881e8cSDavid Sterba 		     BTRFS_LEAF_DATA_OFFSET + leaf_data_end(left),
2893d6025579SChris Mason 		     push_space);
28945f39d397SChris Mason 
28955f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
28965f39d397SChris Mason 			      btrfs_item_nr_offset(0),
28970783fcfcSChris Mason 			      right_nritems * sizeof(struct btrfs_item));
28985f39d397SChris Mason 
289900ec4c51SChris Mason 	/* copy the items from left to right */
29005f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
29015f39d397SChris Mason 		   btrfs_item_nr_offset(left_nritems - push_items),
29020783fcfcSChris Mason 		   push_items * sizeof(struct btrfs_item));
290300ec4c51SChris Mason 
290400ec4c51SChris Mason 	/* update the item pointers */
2905c82f823cSDavid Sterba 	btrfs_init_map_token(&token, right);
29067518a238SChris Mason 	right_nritems += push_items;
29075f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
29080b246afaSJeff Mahoney 	push_space = BTRFS_LEAF_DATA_SIZE(fs_info);
29097518a238SChris Mason 	for (i = 0; i < right_nritems; i++) {
29103212fa14SJosef Bacik 		push_space -= btrfs_token_item_size(&token, i);
29113212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, i, push_space);
2912db94535dSChris Mason 	}
2913db94535dSChris Mason 
29147518a238SChris Mason 	left_nritems -= push_items;
29155f39d397SChris Mason 	btrfs_set_header_nritems(left, left_nritems);
291600ec4c51SChris Mason 
291734a38218SChris Mason 	if (left_nritems)
29185f39d397SChris Mason 		btrfs_mark_buffer_dirty(left);
2919f0486c68SYan, Zheng 	else
29206a884d7dSDavid Sterba 		btrfs_clean_tree_block(left);
2921f0486c68SYan, Zheng 
29225f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
2923a429e513SChris Mason 
29245f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
29255f39d397SChris Mason 	btrfs_set_node_key(upper, &disk_key, slot + 1);
2926d6025579SChris Mason 	btrfs_mark_buffer_dirty(upper);
292702217ed2SChris Mason 
292800ec4c51SChris Mason 	/* then fixup the leaf pointer in the path */
29297518a238SChris Mason 	if (path->slots[0] >= left_nritems) {
29307518a238SChris Mason 		path->slots[0] -= left_nritems;
2931925baeddSChris Mason 		if (btrfs_header_nritems(path->nodes[0]) == 0)
29326a884d7dSDavid Sterba 			btrfs_clean_tree_block(path->nodes[0]);
2933925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
29345f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
29355f39d397SChris Mason 		path->nodes[0] = right;
293600ec4c51SChris Mason 		path->slots[1] += 1;
293700ec4c51SChris Mason 	} else {
2938925baeddSChris Mason 		btrfs_tree_unlock(right);
29395f39d397SChris Mason 		free_extent_buffer(right);
294000ec4c51SChris Mason 	}
294100ec4c51SChris Mason 	return 0;
2942925baeddSChris Mason 
2943925baeddSChris Mason out_unlock:
2944925baeddSChris Mason 	btrfs_tree_unlock(right);
2945925baeddSChris Mason 	free_extent_buffer(right);
2946925baeddSChris Mason 	return 1;
294700ec4c51SChris Mason }
2948925baeddSChris Mason 
294900ec4c51SChris Mason /*
295044871b1bSChris Mason  * push some data in the path leaf to the right, trying to free up at
295174123bd7SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
295244871b1bSChris Mason  *
295344871b1bSChris Mason  * returns 1 if the push failed because the other node didn't have enough
295444871b1bSChris Mason  * room, 0 if everything worked out and < 0 if there were major errors.
295599d8f83cSChris Mason  *
295699d8f83cSChris Mason  * this will push starting from min_slot to the end of the leaf.  It won't
295799d8f83cSChris Mason  * push any slot lower than min_slot
295874123bd7SChris Mason  */
295944871b1bSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
296099d8f83cSChris Mason 			   *root, struct btrfs_path *path,
296199d8f83cSChris Mason 			   int min_data_size, int data_size,
296299d8f83cSChris Mason 			   int empty, u32 min_slot)
2963be0e5c09SChris Mason {
296444871b1bSChris Mason 	struct extent_buffer *left = path->nodes[0];
296544871b1bSChris Mason 	struct extent_buffer *right;
296644871b1bSChris Mason 	struct extent_buffer *upper;
296744871b1bSChris Mason 	int slot;
296844871b1bSChris Mason 	int free_space;
296944871b1bSChris Mason 	u32 left_nritems;
297044871b1bSChris Mason 	int ret;
297144871b1bSChris Mason 
297244871b1bSChris Mason 	if (!path->nodes[1])
297344871b1bSChris Mason 		return 1;
297444871b1bSChris Mason 
297544871b1bSChris Mason 	slot = path->slots[1];
297644871b1bSChris Mason 	upper = path->nodes[1];
297744871b1bSChris Mason 	if (slot >= btrfs_header_nritems(upper) - 1)
297844871b1bSChris Mason 		return 1;
297944871b1bSChris Mason 
298049d0c642SFilipe Manana 	btrfs_assert_tree_write_locked(path->nodes[1]);
298144871b1bSChris Mason 
29824b231ae4SDavid Sterba 	right = btrfs_read_node_slot(upper, slot + 1);
2983fb770ae4SLiu Bo 	/*
2984fb770ae4SLiu Bo 	 * slot + 1 is not valid or we fail to read the right node,
2985fb770ae4SLiu Bo 	 * no big deal, just return.
2986fb770ae4SLiu Bo 	 */
2987fb770ae4SLiu Bo 	if (IS_ERR(right))
298891ca338dSTsutomu Itoh 		return 1;
298991ca338dSTsutomu Itoh 
2990bf77467aSJosef Bacik 	__btrfs_tree_lock(right, BTRFS_NESTING_RIGHT);
299144871b1bSChris Mason 
2992e902baacSDavid Sterba 	free_space = btrfs_leaf_free_space(right);
299344871b1bSChris Mason 	if (free_space < data_size)
299444871b1bSChris Mason 		goto out_unlock;
299544871b1bSChris Mason 
299644871b1bSChris Mason 	/* cow and double check */
299744871b1bSChris Mason 	ret = btrfs_cow_block(trans, root, right, upper,
2998bf59a5a2SJosef Bacik 			      slot + 1, &right, BTRFS_NESTING_RIGHT_COW);
299944871b1bSChris Mason 	if (ret)
300044871b1bSChris Mason 		goto out_unlock;
300144871b1bSChris Mason 
3002e902baacSDavid Sterba 	free_space = btrfs_leaf_free_space(right);
300344871b1bSChris Mason 	if (free_space < data_size)
300444871b1bSChris Mason 		goto out_unlock;
300544871b1bSChris Mason 
300644871b1bSChris Mason 	left_nritems = btrfs_header_nritems(left);
300744871b1bSChris Mason 	if (left_nritems == 0)
300844871b1bSChris Mason 		goto out_unlock;
300944871b1bSChris Mason 
3010d16c702fSQu Wenruo 	if (check_sibling_keys(left, right)) {
3011d16c702fSQu Wenruo 		ret = -EUCLEAN;
3012d16c702fSQu Wenruo 		btrfs_tree_unlock(right);
3013d16c702fSQu Wenruo 		free_extent_buffer(right);
3014d16c702fSQu Wenruo 		return ret;
3015d16c702fSQu Wenruo 	}
30162ef1fed2SFilipe David Borba Manana 	if (path->slots[0] == left_nritems && !empty) {
30172ef1fed2SFilipe David Borba Manana 		/* Key greater than all keys in the leaf, right neighbor has
30182ef1fed2SFilipe David Borba Manana 		 * enough room for it and we're not emptying our leaf to delete
30192ef1fed2SFilipe David Borba Manana 		 * it, therefore use right neighbor to insert the new item and
302052042d8eSAndrea Gelmini 		 * no need to touch/dirty our left leaf. */
30212ef1fed2SFilipe David Borba Manana 		btrfs_tree_unlock(left);
30222ef1fed2SFilipe David Borba Manana 		free_extent_buffer(left);
30232ef1fed2SFilipe David Borba Manana 		path->nodes[0] = right;
30242ef1fed2SFilipe David Borba Manana 		path->slots[0] = 0;
30252ef1fed2SFilipe David Borba Manana 		path->slots[1]++;
30262ef1fed2SFilipe David Borba Manana 		return 0;
30272ef1fed2SFilipe David Borba Manana 	}
30282ef1fed2SFilipe David Borba Manana 
3029f72f0010SDavid Sterba 	return __push_leaf_right(path, min_data_size, empty,
303099d8f83cSChris Mason 				right, free_space, left_nritems, min_slot);
303144871b1bSChris Mason out_unlock:
303244871b1bSChris Mason 	btrfs_tree_unlock(right);
303344871b1bSChris Mason 	free_extent_buffer(right);
303444871b1bSChris Mason 	return 1;
303544871b1bSChris Mason }
303644871b1bSChris Mason 
303744871b1bSChris Mason /*
303844871b1bSChris Mason  * push some data in the path leaf to the left, trying to free up at
303944871b1bSChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
304099d8f83cSChris Mason  *
304199d8f83cSChris Mason  * max_slot can put a limit on how far into the leaf we'll push items.  The
304299d8f83cSChris Mason  * item at 'max_slot' won't be touched.  Use (u32)-1 to make us do all the
304399d8f83cSChris Mason  * items
304444871b1bSChris Mason  */
30458087c193SDavid Sterba static noinline int __push_leaf_left(struct btrfs_path *path, int data_size,
304644871b1bSChris Mason 				     int empty, struct extent_buffer *left,
304799d8f83cSChris Mason 				     int free_space, u32 right_nritems,
304899d8f83cSChris Mason 				     u32 max_slot)
304944871b1bSChris Mason {
30508087c193SDavid Sterba 	struct btrfs_fs_info *fs_info = left->fs_info;
30515f39d397SChris Mason 	struct btrfs_disk_key disk_key;
30525f39d397SChris Mason 	struct extent_buffer *right = path->nodes[0];
3053be0e5c09SChris Mason 	int i;
3054be0e5c09SChris Mason 	int push_space = 0;
3055be0e5c09SChris Mason 	int push_items = 0;
30567518a238SChris Mason 	u32 old_left_nritems;
305734a38218SChris Mason 	u32 nr;
3058aa5d6bedSChris Mason 	int ret = 0;
3059db94535dSChris Mason 	u32 this_item_size;
3060db94535dSChris Mason 	u32 old_left_item_size;
3061cfed81a0SChris Mason 	struct btrfs_map_token token;
3062cfed81a0SChris Mason 
306334a38218SChris Mason 	if (empty)
306499d8f83cSChris Mason 		nr = min(right_nritems, max_slot);
306534a38218SChris Mason 	else
306699d8f83cSChris Mason 		nr = min(right_nritems - 1, max_slot);
306734a38218SChris Mason 
306834a38218SChris Mason 	for (i = 0; i < nr; i++) {
306931840ae1SZheng Yan 		if (!empty && push_items > 0) {
307031840ae1SZheng Yan 			if (path->slots[0] < i)
307131840ae1SZheng Yan 				break;
307231840ae1SZheng Yan 			if (path->slots[0] == i) {
3073e902baacSDavid Sterba 				int space = btrfs_leaf_free_space(right);
3074e902baacSDavid Sterba 
307531840ae1SZheng Yan 				if (space + push_space * 2 > free_space)
307631840ae1SZheng Yan 					break;
307731840ae1SZheng Yan 			}
307831840ae1SZheng Yan 		}
307931840ae1SZheng Yan 
3080be0e5c09SChris Mason 		if (path->slots[0] == i)
308187b29b20SYan Zheng 			push_space += data_size;
3082db94535dSChris Mason 
30833212fa14SJosef Bacik 		this_item_size = btrfs_item_size(right, i);
308474794207SJosef Bacik 		if (this_item_size + sizeof(struct btrfs_item) + push_space >
308574794207SJosef Bacik 		    free_space)
3086be0e5c09SChris Mason 			break;
3087db94535dSChris Mason 
3088be0e5c09SChris Mason 		push_items++;
308974794207SJosef Bacik 		push_space += this_item_size + sizeof(struct btrfs_item);
3090be0e5c09SChris Mason 	}
3091db94535dSChris Mason 
3092be0e5c09SChris Mason 	if (push_items == 0) {
3093925baeddSChris Mason 		ret = 1;
3094925baeddSChris Mason 		goto out;
3095be0e5c09SChris Mason 	}
3096fae7f21cSDulshani Gunawardhana 	WARN_ON(!empty && push_items == btrfs_header_nritems(right));
30975f39d397SChris Mason 
3098be0e5c09SChris Mason 	/* push data from right to left */
30995f39d397SChris Mason 	copy_extent_buffer(left, right,
31005f39d397SChris Mason 			   btrfs_item_nr_offset(btrfs_header_nritems(left)),
31015f39d397SChris Mason 			   btrfs_item_nr_offset(0),
31025f39d397SChris Mason 			   push_items * sizeof(struct btrfs_item));
31035f39d397SChris Mason 
31040b246afaSJeff Mahoney 	push_space = BTRFS_LEAF_DATA_SIZE(fs_info) -
31053212fa14SJosef Bacik 		     btrfs_item_offset(right, push_items - 1);
31065f39d397SChris Mason 
31073d9ec8c4SNikolay Borisov 	copy_extent_buffer(left, right, BTRFS_LEAF_DATA_OFFSET +
31088f881e8cSDavid Sterba 		     leaf_data_end(left) - push_space,
31093d9ec8c4SNikolay Borisov 		     BTRFS_LEAF_DATA_OFFSET +
31103212fa14SJosef Bacik 		     btrfs_item_offset(right, push_items - 1),
3111be0e5c09SChris Mason 		     push_space);
31125f39d397SChris Mason 	old_left_nritems = btrfs_header_nritems(left);
311387b29b20SYan Zheng 	BUG_ON(old_left_nritems <= 0);
3114eb60ceacSChris Mason 
3115c82f823cSDavid Sterba 	btrfs_init_map_token(&token, left);
31163212fa14SJosef Bacik 	old_left_item_size = btrfs_item_offset(left, old_left_nritems - 1);
3117be0e5c09SChris Mason 	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
31185f39d397SChris Mason 		u32 ioff;
3119db94535dSChris Mason 
31203212fa14SJosef Bacik 		ioff = btrfs_token_item_offset(&token, i);
31213212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, i,
3122cc4c13d5SDavid Sterba 		      ioff - (BTRFS_LEAF_DATA_SIZE(fs_info) - old_left_item_size));
3123be0e5c09SChris Mason 	}
31245f39d397SChris Mason 	btrfs_set_header_nritems(left, old_left_nritems + push_items);
3125be0e5c09SChris Mason 
3126be0e5c09SChris Mason 	/* fixup right node */
312731b1a2bdSJulia Lawall 	if (push_items > right_nritems)
312831b1a2bdSJulia Lawall 		WARN(1, KERN_CRIT "push items %d nr %u\n", push_items,
3129d397712bSChris Mason 		       right_nritems);
313034a38218SChris Mason 
313134a38218SChris Mason 	if (push_items < right_nritems) {
31323212fa14SJosef Bacik 		push_space = btrfs_item_offset(right, push_items - 1) -
31338f881e8cSDavid Sterba 						  leaf_data_end(right);
31343d9ec8c4SNikolay Borisov 		memmove_extent_buffer(right, BTRFS_LEAF_DATA_OFFSET +
31350b246afaSJeff Mahoney 				      BTRFS_LEAF_DATA_SIZE(fs_info) - push_space,
31363d9ec8c4SNikolay Borisov 				      BTRFS_LEAF_DATA_OFFSET +
31378f881e8cSDavid Sterba 				      leaf_data_end(right), push_space);
31385f39d397SChris Mason 
31395f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_item_nr_offset(0),
31405f39d397SChris Mason 			      btrfs_item_nr_offset(push_items),
31415f39d397SChris Mason 			     (btrfs_header_nritems(right) - push_items) *
31420783fcfcSChris Mason 			     sizeof(struct btrfs_item));
314334a38218SChris Mason 	}
3144c82f823cSDavid Sterba 
3145c82f823cSDavid Sterba 	btrfs_init_map_token(&token, right);
3146eef1c494SYan 	right_nritems -= push_items;
3147eef1c494SYan 	btrfs_set_header_nritems(right, right_nritems);
31480b246afaSJeff Mahoney 	push_space = BTRFS_LEAF_DATA_SIZE(fs_info);
31495f39d397SChris Mason 	for (i = 0; i < right_nritems; i++) {
31503212fa14SJosef Bacik 		push_space = push_space - btrfs_token_item_size(&token, i);
31513212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, i, push_space);
3152db94535dSChris Mason 	}
3153eb60ceacSChris Mason 
31545f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
315534a38218SChris Mason 	if (right_nritems)
31565f39d397SChris Mason 		btrfs_mark_buffer_dirty(right);
3157f0486c68SYan, Zheng 	else
31586a884d7dSDavid Sterba 		btrfs_clean_tree_block(right);
3159098f59c2SChris Mason 
31605f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
3161b167fa91SNikolay Borisov 	fixup_low_keys(path, &disk_key, 1);
3162be0e5c09SChris Mason 
3163be0e5c09SChris Mason 	/* then fixup the leaf pointer in the path */
3164be0e5c09SChris Mason 	if (path->slots[0] < push_items) {
3165be0e5c09SChris Mason 		path->slots[0] += old_left_nritems;
3166925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
31675f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
31685f39d397SChris Mason 		path->nodes[0] = left;
3169be0e5c09SChris Mason 		path->slots[1] -= 1;
3170be0e5c09SChris Mason 	} else {
3171925baeddSChris Mason 		btrfs_tree_unlock(left);
31725f39d397SChris Mason 		free_extent_buffer(left);
3173be0e5c09SChris Mason 		path->slots[0] -= push_items;
3174be0e5c09SChris Mason 	}
3175eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
3176aa5d6bedSChris Mason 	return ret;
3177925baeddSChris Mason out:
3178925baeddSChris Mason 	btrfs_tree_unlock(left);
3179925baeddSChris Mason 	free_extent_buffer(left);
3180925baeddSChris Mason 	return ret;
3181be0e5c09SChris Mason }
3182be0e5c09SChris Mason 
318374123bd7SChris Mason /*
318444871b1bSChris Mason  * push some data in the path leaf to the left, trying to free up at
318544871b1bSChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
318699d8f83cSChris Mason  *
318799d8f83cSChris Mason  * max_slot can put a limit on how far into the leaf we'll push items.  The
318899d8f83cSChris Mason  * item at 'max_slot' won't be touched.  Use (u32)-1 to make us push all the
318999d8f83cSChris Mason  * items
319044871b1bSChris Mason  */
319144871b1bSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
319299d8f83cSChris Mason 			  *root, struct btrfs_path *path, int min_data_size,
319399d8f83cSChris Mason 			  int data_size, int empty, u32 max_slot)
319444871b1bSChris Mason {
319544871b1bSChris Mason 	struct extent_buffer *right = path->nodes[0];
319644871b1bSChris Mason 	struct extent_buffer *left;
319744871b1bSChris Mason 	int slot;
319844871b1bSChris Mason 	int free_space;
319944871b1bSChris Mason 	u32 right_nritems;
320044871b1bSChris Mason 	int ret = 0;
320144871b1bSChris Mason 
320244871b1bSChris Mason 	slot = path->slots[1];
320344871b1bSChris Mason 	if (slot == 0)
320444871b1bSChris Mason 		return 1;
320544871b1bSChris Mason 	if (!path->nodes[1])
320644871b1bSChris Mason 		return 1;
320744871b1bSChris Mason 
320844871b1bSChris Mason 	right_nritems = btrfs_header_nritems(right);
320944871b1bSChris Mason 	if (right_nritems == 0)
321044871b1bSChris Mason 		return 1;
321144871b1bSChris Mason 
321249d0c642SFilipe Manana 	btrfs_assert_tree_write_locked(path->nodes[1]);
321344871b1bSChris Mason 
32144b231ae4SDavid Sterba 	left = btrfs_read_node_slot(path->nodes[1], slot - 1);
3215fb770ae4SLiu Bo 	/*
3216fb770ae4SLiu Bo 	 * slot - 1 is not valid or we fail to read the left node,
3217fb770ae4SLiu Bo 	 * no big deal, just return.
3218fb770ae4SLiu Bo 	 */
3219fb770ae4SLiu Bo 	if (IS_ERR(left))
322091ca338dSTsutomu Itoh 		return 1;
322191ca338dSTsutomu Itoh 
3222bf77467aSJosef Bacik 	__btrfs_tree_lock(left, BTRFS_NESTING_LEFT);
322344871b1bSChris Mason 
3224e902baacSDavid Sterba 	free_space = btrfs_leaf_free_space(left);
322544871b1bSChris Mason 	if (free_space < data_size) {
322644871b1bSChris Mason 		ret = 1;
322744871b1bSChris Mason 		goto out;
322844871b1bSChris Mason 	}
322944871b1bSChris Mason 
323044871b1bSChris Mason 	/* cow and double check */
323144871b1bSChris Mason 	ret = btrfs_cow_block(trans, root, left,
32329631e4ccSJosef Bacik 			      path->nodes[1], slot - 1, &left,
3233bf59a5a2SJosef Bacik 			      BTRFS_NESTING_LEFT_COW);
323444871b1bSChris Mason 	if (ret) {
323544871b1bSChris Mason 		/* we hit -ENOSPC, but it isn't fatal here */
323679787eaaSJeff Mahoney 		if (ret == -ENOSPC)
323744871b1bSChris Mason 			ret = 1;
323844871b1bSChris Mason 		goto out;
323944871b1bSChris Mason 	}
324044871b1bSChris Mason 
3241e902baacSDavid Sterba 	free_space = btrfs_leaf_free_space(left);
324244871b1bSChris Mason 	if (free_space < data_size) {
324344871b1bSChris Mason 		ret = 1;
324444871b1bSChris Mason 		goto out;
324544871b1bSChris Mason 	}
324644871b1bSChris Mason 
3247d16c702fSQu Wenruo 	if (check_sibling_keys(left, right)) {
3248d16c702fSQu Wenruo 		ret = -EUCLEAN;
3249d16c702fSQu Wenruo 		goto out;
3250d16c702fSQu Wenruo 	}
32518087c193SDavid Sterba 	return __push_leaf_left(path, min_data_size,
325299d8f83cSChris Mason 			       empty, left, free_space, right_nritems,
325399d8f83cSChris Mason 			       max_slot);
325444871b1bSChris Mason out:
325544871b1bSChris Mason 	btrfs_tree_unlock(left);
325644871b1bSChris Mason 	free_extent_buffer(left);
325744871b1bSChris Mason 	return ret;
325844871b1bSChris Mason }
325944871b1bSChris Mason 
326044871b1bSChris Mason /*
326174123bd7SChris Mason  * split the path's leaf in two, making sure there is at least data_size
326274123bd7SChris Mason  * available for the resulting leaf level of the path.
326374123bd7SChris Mason  */
3264143bede5SJeff Mahoney static noinline void copy_for_split(struct btrfs_trans_handle *trans,
326544871b1bSChris Mason 				    struct btrfs_path *path,
326644871b1bSChris Mason 				    struct extent_buffer *l,
326744871b1bSChris Mason 				    struct extent_buffer *right,
326844871b1bSChris Mason 				    int slot, int mid, int nritems)
3269be0e5c09SChris Mason {
327094f94ad9SDavid Sterba 	struct btrfs_fs_info *fs_info = trans->fs_info;
3271be0e5c09SChris Mason 	int data_copy_size;
3272be0e5c09SChris Mason 	int rt_data_off;
3273be0e5c09SChris Mason 	int i;
3274d4dbff95SChris Mason 	struct btrfs_disk_key disk_key;
3275cfed81a0SChris Mason 	struct btrfs_map_token token;
3276cfed81a0SChris Mason 
32775f39d397SChris Mason 	nritems = nritems - mid;
32785f39d397SChris Mason 	btrfs_set_header_nritems(right, nritems);
3279dc2e724eSJosef Bacik 	data_copy_size = btrfs_item_data_end(l, mid) - leaf_data_end(l);
32805f39d397SChris Mason 
32815f39d397SChris Mason 	copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
32825f39d397SChris Mason 			   btrfs_item_nr_offset(mid),
32835f39d397SChris Mason 			   nritems * sizeof(struct btrfs_item));
32845f39d397SChris Mason 
32855f39d397SChris Mason 	copy_extent_buffer(right, l,
32863d9ec8c4SNikolay Borisov 		     BTRFS_LEAF_DATA_OFFSET + BTRFS_LEAF_DATA_SIZE(fs_info) -
32873d9ec8c4SNikolay Borisov 		     data_copy_size, BTRFS_LEAF_DATA_OFFSET +
32888f881e8cSDavid Sterba 		     leaf_data_end(l), data_copy_size);
328974123bd7SChris Mason 
3290dc2e724eSJosef Bacik 	rt_data_off = BTRFS_LEAF_DATA_SIZE(fs_info) - btrfs_item_data_end(l, mid);
32915f39d397SChris Mason 
3292c82f823cSDavid Sterba 	btrfs_init_map_token(&token, right);
32935f39d397SChris Mason 	for (i = 0; i < nritems; i++) {
3294db94535dSChris Mason 		u32 ioff;
3295db94535dSChris Mason 
32963212fa14SJosef Bacik 		ioff = btrfs_token_item_offset(&token, i);
32973212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, i, ioff + rt_data_off);
32980783fcfcSChris Mason 	}
329974123bd7SChris Mason 
33005f39d397SChris Mason 	btrfs_set_header_nritems(l, mid);
33015f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
33026ad3cf6dSDavid Sterba 	insert_ptr(trans, path, &disk_key, right->start, path->slots[1] + 1, 1);
33035f39d397SChris Mason 
33045f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
33055f39d397SChris Mason 	btrfs_mark_buffer_dirty(l);
3306eb60ceacSChris Mason 	BUG_ON(path->slots[0] != slot);
33075f39d397SChris Mason 
3308be0e5c09SChris Mason 	if (mid <= slot) {
3309925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
33105f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
33115f39d397SChris Mason 		path->nodes[0] = right;
3312be0e5c09SChris Mason 		path->slots[0] -= mid;
3313be0e5c09SChris Mason 		path->slots[1] += 1;
3314925baeddSChris Mason 	} else {
3315925baeddSChris Mason 		btrfs_tree_unlock(right);
33165f39d397SChris Mason 		free_extent_buffer(right);
3317925baeddSChris Mason 	}
33185f39d397SChris Mason 
3319eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
332044871b1bSChris Mason }
332144871b1bSChris Mason 
332244871b1bSChris Mason /*
332399d8f83cSChris Mason  * double splits happen when we need to insert a big item in the middle
332499d8f83cSChris Mason  * of a leaf.  A double split can leave us with 3 mostly empty leaves:
332599d8f83cSChris Mason  * leaf: [ slots 0 - N] [ our target ] [ N + 1 - total in leaf ]
332699d8f83cSChris Mason  *          A                 B                 C
332799d8f83cSChris Mason  *
332899d8f83cSChris Mason  * We avoid this by trying to push the items on either side of our target
332999d8f83cSChris Mason  * into the adjacent leaves.  If all goes well we can avoid the double split
333099d8f83cSChris Mason  * completely.
333199d8f83cSChris Mason  */
333299d8f83cSChris Mason static noinline int push_for_double_split(struct btrfs_trans_handle *trans,
333399d8f83cSChris Mason 					  struct btrfs_root *root,
333499d8f83cSChris Mason 					  struct btrfs_path *path,
333599d8f83cSChris Mason 					  int data_size)
333699d8f83cSChris Mason {
333799d8f83cSChris Mason 	int ret;
333899d8f83cSChris Mason 	int progress = 0;
333999d8f83cSChris Mason 	int slot;
334099d8f83cSChris Mason 	u32 nritems;
33415a4267caSFilipe David Borba Manana 	int space_needed = data_size;
334299d8f83cSChris Mason 
334399d8f83cSChris Mason 	slot = path->slots[0];
33445a4267caSFilipe David Borba Manana 	if (slot < btrfs_header_nritems(path->nodes[0]))
3345e902baacSDavid Sterba 		space_needed -= btrfs_leaf_free_space(path->nodes[0]);
334699d8f83cSChris Mason 
334799d8f83cSChris Mason 	/*
334899d8f83cSChris Mason 	 * try to push all the items after our slot into the
334999d8f83cSChris Mason 	 * right leaf
335099d8f83cSChris Mason 	 */
33515a4267caSFilipe David Borba Manana 	ret = push_leaf_right(trans, root, path, 1, space_needed, 0, slot);
335299d8f83cSChris Mason 	if (ret < 0)
335399d8f83cSChris Mason 		return ret;
335499d8f83cSChris Mason 
335599d8f83cSChris Mason 	if (ret == 0)
335699d8f83cSChris Mason 		progress++;
335799d8f83cSChris Mason 
335899d8f83cSChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
335999d8f83cSChris Mason 	/*
336099d8f83cSChris Mason 	 * our goal is to get our slot at the start or end of a leaf.  If
336199d8f83cSChris Mason 	 * we've done so we're done
336299d8f83cSChris Mason 	 */
336399d8f83cSChris Mason 	if (path->slots[0] == 0 || path->slots[0] == nritems)
336499d8f83cSChris Mason 		return 0;
336599d8f83cSChris Mason 
3366e902baacSDavid Sterba 	if (btrfs_leaf_free_space(path->nodes[0]) >= data_size)
336799d8f83cSChris Mason 		return 0;
336899d8f83cSChris Mason 
336999d8f83cSChris Mason 	/* try to push all the items before our slot into the next leaf */
337099d8f83cSChris Mason 	slot = path->slots[0];
3371263d3995SFilipe Manana 	space_needed = data_size;
3372263d3995SFilipe Manana 	if (slot > 0)
3373e902baacSDavid Sterba 		space_needed -= btrfs_leaf_free_space(path->nodes[0]);
33745a4267caSFilipe David Borba Manana 	ret = push_leaf_left(trans, root, path, 1, space_needed, 0, slot);
337599d8f83cSChris Mason 	if (ret < 0)
337699d8f83cSChris Mason 		return ret;
337799d8f83cSChris Mason 
337899d8f83cSChris Mason 	if (ret == 0)
337999d8f83cSChris Mason 		progress++;
338099d8f83cSChris Mason 
338199d8f83cSChris Mason 	if (progress)
338299d8f83cSChris Mason 		return 0;
338399d8f83cSChris Mason 	return 1;
338499d8f83cSChris Mason }
338599d8f83cSChris Mason 
338699d8f83cSChris Mason /*
338744871b1bSChris Mason  * split the path's leaf in two, making sure there is at least data_size
338844871b1bSChris Mason  * available for the resulting leaf level of the path.
338944871b1bSChris Mason  *
339044871b1bSChris Mason  * returns 0 if all went well and < 0 on failure.
339144871b1bSChris Mason  */
339244871b1bSChris Mason static noinline int split_leaf(struct btrfs_trans_handle *trans,
339344871b1bSChris Mason 			       struct btrfs_root *root,
3394310712b2SOmar Sandoval 			       const struct btrfs_key *ins_key,
339544871b1bSChris Mason 			       struct btrfs_path *path, int data_size,
339644871b1bSChris Mason 			       int extend)
339744871b1bSChris Mason {
33985d4f98a2SYan Zheng 	struct btrfs_disk_key disk_key;
339944871b1bSChris Mason 	struct extent_buffer *l;
340044871b1bSChris Mason 	u32 nritems;
340144871b1bSChris Mason 	int mid;
340244871b1bSChris Mason 	int slot;
340344871b1bSChris Mason 	struct extent_buffer *right;
3404b7a0365eSDaniel Dressler 	struct btrfs_fs_info *fs_info = root->fs_info;
340544871b1bSChris Mason 	int ret = 0;
340644871b1bSChris Mason 	int wret;
34075d4f98a2SYan Zheng 	int split;
340844871b1bSChris Mason 	int num_doubles = 0;
340999d8f83cSChris Mason 	int tried_avoid_double = 0;
341044871b1bSChris Mason 
3411a5719521SYan, Zheng 	l = path->nodes[0];
3412a5719521SYan, Zheng 	slot = path->slots[0];
34133212fa14SJosef Bacik 	if (extend && data_size + btrfs_item_size(l, slot) +
34140b246afaSJeff Mahoney 	    sizeof(struct btrfs_item) > BTRFS_LEAF_DATA_SIZE(fs_info))
3415a5719521SYan, Zheng 		return -EOVERFLOW;
3416a5719521SYan, Zheng 
341744871b1bSChris Mason 	/* first try to make some room by pushing left and right */
341833157e05SLiu Bo 	if (data_size && path->nodes[1]) {
34195a4267caSFilipe David Borba Manana 		int space_needed = data_size;
34205a4267caSFilipe David Borba Manana 
34215a4267caSFilipe David Borba Manana 		if (slot < btrfs_header_nritems(l))
3422e902baacSDavid Sterba 			space_needed -= btrfs_leaf_free_space(l);
34235a4267caSFilipe David Borba Manana 
34245a4267caSFilipe David Borba Manana 		wret = push_leaf_right(trans, root, path, space_needed,
34255a4267caSFilipe David Borba Manana 				       space_needed, 0, 0);
342644871b1bSChris Mason 		if (wret < 0)
342744871b1bSChris Mason 			return wret;
342844871b1bSChris Mason 		if (wret) {
3429263d3995SFilipe Manana 			space_needed = data_size;
3430263d3995SFilipe Manana 			if (slot > 0)
3431e902baacSDavid Sterba 				space_needed -= btrfs_leaf_free_space(l);
34325a4267caSFilipe David Borba Manana 			wret = push_leaf_left(trans, root, path, space_needed,
34335a4267caSFilipe David Borba Manana 					      space_needed, 0, (u32)-1);
343444871b1bSChris Mason 			if (wret < 0)
343544871b1bSChris Mason 				return wret;
343644871b1bSChris Mason 		}
343744871b1bSChris Mason 		l = path->nodes[0];
343844871b1bSChris Mason 
343944871b1bSChris Mason 		/* did the pushes work? */
3440e902baacSDavid Sterba 		if (btrfs_leaf_free_space(l) >= data_size)
344144871b1bSChris Mason 			return 0;
344244871b1bSChris Mason 	}
344344871b1bSChris Mason 
344444871b1bSChris Mason 	if (!path->nodes[1]) {
3445fdd99c72SLiu Bo 		ret = insert_new_root(trans, root, path, 1);
344644871b1bSChris Mason 		if (ret)
344744871b1bSChris Mason 			return ret;
344844871b1bSChris Mason 	}
344944871b1bSChris Mason again:
34505d4f98a2SYan Zheng 	split = 1;
345144871b1bSChris Mason 	l = path->nodes[0];
345244871b1bSChris Mason 	slot = path->slots[0];
345344871b1bSChris Mason 	nritems = btrfs_header_nritems(l);
345444871b1bSChris Mason 	mid = (nritems + 1) / 2;
345544871b1bSChris Mason 
34565d4f98a2SYan Zheng 	if (mid <= slot) {
34575d4f98a2SYan Zheng 		if (nritems == 1 ||
34585d4f98a2SYan Zheng 		    leaf_space_used(l, mid, nritems - mid) + data_size >
34590b246afaSJeff Mahoney 			BTRFS_LEAF_DATA_SIZE(fs_info)) {
34605d4f98a2SYan Zheng 			if (slot >= nritems) {
34615d4f98a2SYan Zheng 				split = 0;
34625d4f98a2SYan Zheng 			} else {
34635d4f98a2SYan Zheng 				mid = slot;
34645d4f98a2SYan Zheng 				if (mid != nritems &&
34655d4f98a2SYan Zheng 				    leaf_space_used(l, mid, nritems - mid) +
34660b246afaSJeff Mahoney 				    data_size > BTRFS_LEAF_DATA_SIZE(fs_info)) {
346799d8f83cSChris Mason 					if (data_size && !tried_avoid_double)
346899d8f83cSChris Mason 						goto push_for_double;
34695d4f98a2SYan Zheng 					split = 2;
34705d4f98a2SYan Zheng 				}
34715d4f98a2SYan Zheng 			}
34725d4f98a2SYan Zheng 		}
34735d4f98a2SYan Zheng 	} else {
34745d4f98a2SYan Zheng 		if (leaf_space_used(l, 0, mid) + data_size >
34750b246afaSJeff Mahoney 			BTRFS_LEAF_DATA_SIZE(fs_info)) {
34765d4f98a2SYan Zheng 			if (!extend && data_size && slot == 0) {
34775d4f98a2SYan Zheng 				split = 0;
34785d4f98a2SYan Zheng 			} else if ((extend || !data_size) && slot == 0) {
34795d4f98a2SYan Zheng 				mid = 1;
34805d4f98a2SYan Zheng 			} else {
34815d4f98a2SYan Zheng 				mid = slot;
34825d4f98a2SYan Zheng 				if (mid != nritems &&
34835d4f98a2SYan Zheng 				    leaf_space_used(l, mid, nritems - mid) +
34840b246afaSJeff Mahoney 				    data_size > BTRFS_LEAF_DATA_SIZE(fs_info)) {
348599d8f83cSChris Mason 					if (data_size && !tried_avoid_double)
348699d8f83cSChris Mason 						goto push_for_double;
34875d4f98a2SYan Zheng 					split = 2;
34885d4f98a2SYan Zheng 				}
34895d4f98a2SYan Zheng 			}
34905d4f98a2SYan Zheng 		}
34915d4f98a2SYan Zheng 	}
34925d4f98a2SYan Zheng 
34935d4f98a2SYan Zheng 	if (split == 0)
34945d4f98a2SYan Zheng 		btrfs_cpu_key_to_disk(&disk_key, ins_key);
34955d4f98a2SYan Zheng 	else
34965d4f98a2SYan Zheng 		btrfs_item_key(l, &disk_key, mid);
34975d4f98a2SYan Zheng 
3498ca9d473aSJosef Bacik 	/*
3499ca9d473aSJosef Bacik 	 * We have to about BTRFS_NESTING_NEW_ROOT here if we've done a double
3500ca9d473aSJosef Bacik 	 * split, because we're only allowed to have MAX_LOCKDEP_SUBCLASSES
3501ca9d473aSJosef Bacik 	 * subclasses, which is 8 at the time of this patch, and we've maxed it
3502ca9d473aSJosef Bacik 	 * out.  In the future we could add a
3503ca9d473aSJosef Bacik 	 * BTRFS_NESTING_SPLIT_THE_SPLITTENING if we need to, but for now just
3504ca9d473aSJosef Bacik 	 * use BTRFS_NESTING_NEW_ROOT.
3505ca9d473aSJosef Bacik 	 */
350679bd3712SFilipe Manana 	right = btrfs_alloc_tree_block(trans, root, 0, root->root_key.objectid,
350779bd3712SFilipe Manana 				       &disk_key, 0, l->start, 0,
350879bd3712SFilipe Manana 				       num_doubles ? BTRFS_NESTING_NEW_ROOT :
3509ca9d473aSJosef Bacik 				       BTRFS_NESTING_SPLIT);
3510f0486c68SYan, Zheng 	if (IS_ERR(right))
351144871b1bSChris Mason 		return PTR_ERR(right);
3512f0486c68SYan, Zheng 
35130b246afaSJeff Mahoney 	root_add_used(root, fs_info->nodesize);
351444871b1bSChris Mason 
35155d4f98a2SYan Zheng 	if (split == 0) {
351644871b1bSChris Mason 		if (mid <= slot) {
351744871b1bSChris Mason 			btrfs_set_header_nritems(right, 0);
35186ad3cf6dSDavid Sterba 			insert_ptr(trans, path, &disk_key,
35192ff7e61eSJeff Mahoney 				   right->start, path->slots[1] + 1, 1);
352044871b1bSChris Mason 			btrfs_tree_unlock(path->nodes[0]);
352144871b1bSChris Mason 			free_extent_buffer(path->nodes[0]);
352244871b1bSChris Mason 			path->nodes[0] = right;
352344871b1bSChris Mason 			path->slots[0] = 0;
352444871b1bSChris Mason 			path->slots[1] += 1;
352544871b1bSChris Mason 		} else {
352644871b1bSChris Mason 			btrfs_set_header_nritems(right, 0);
35276ad3cf6dSDavid Sterba 			insert_ptr(trans, path, &disk_key,
35282ff7e61eSJeff Mahoney 				   right->start, path->slots[1], 1);
352944871b1bSChris Mason 			btrfs_tree_unlock(path->nodes[0]);
353044871b1bSChris Mason 			free_extent_buffer(path->nodes[0]);
353144871b1bSChris Mason 			path->nodes[0] = right;
353244871b1bSChris Mason 			path->slots[0] = 0;
3533143bede5SJeff Mahoney 			if (path->slots[1] == 0)
3534b167fa91SNikolay Borisov 				fixup_low_keys(path, &disk_key, 1);
35355d4f98a2SYan Zheng 		}
3536196e0249SLiu Bo 		/*
3537196e0249SLiu Bo 		 * We create a new leaf 'right' for the required ins_len and
3538196e0249SLiu Bo 		 * we'll do btrfs_mark_buffer_dirty() on this leaf after copying
3539196e0249SLiu Bo 		 * the content of ins_len to 'right'.
3540196e0249SLiu Bo 		 */
354144871b1bSChris Mason 		return ret;
354244871b1bSChris Mason 	}
354344871b1bSChris Mason 
354494f94ad9SDavid Sterba 	copy_for_split(trans, path, l, right, slot, mid, nritems);
354544871b1bSChris Mason 
35465d4f98a2SYan Zheng 	if (split == 2) {
3547cc0c5538SChris Mason 		BUG_ON(num_doubles != 0);
3548cc0c5538SChris Mason 		num_doubles++;
3549cc0c5538SChris Mason 		goto again;
35503326d1b0SChris Mason 	}
355144871b1bSChris Mason 
3552143bede5SJeff Mahoney 	return 0;
355399d8f83cSChris Mason 
355499d8f83cSChris Mason push_for_double:
355599d8f83cSChris Mason 	push_for_double_split(trans, root, path, data_size);
355699d8f83cSChris Mason 	tried_avoid_double = 1;
3557e902baacSDavid Sterba 	if (btrfs_leaf_free_space(path->nodes[0]) >= data_size)
355899d8f83cSChris Mason 		return 0;
355999d8f83cSChris Mason 	goto again;
3560be0e5c09SChris Mason }
3561be0e5c09SChris Mason 
3562ad48fd75SYan, Zheng static noinline int setup_leaf_for_split(struct btrfs_trans_handle *trans,
3563ad48fd75SYan, Zheng 					 struct btrfs_root *root,
3564ad48fd75SYan, Zheng 					 struct btrfs_path *path, int ins_len)
3565ad48fd75SYan, Zheng {
3566ad48fd75SYan, Zheng 	struct btrfs_key key;
3567ad48fd75SYan, Zheng 	struct extent_buffer *leaf;
3568ad48fd75SYan, Zheng 	struct btrfs_file_extent_item *fi;
3569ad48fd75SYan, Zheng 	u64 extent_len = 0;
3570ad48fd75SYan, Zheng 	u32 item_size;
3571ad48fd75SYan, Zheng 	int ret;
3572ad48fd75SYan, Zheng 
3573ad48fd75SYan, Zheng 	leaf = path->nodes[0];
3574ad48fd75SYan, Zheng 	btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
3575ad48fd75SYan, Zheng 
3576ad48fd75SYan, Zheng 	BUG_ON(key.type != BTRFS_EXTENT_DATA_KEY &&
3577ad48fd75SYan, Zheng 	       key.type != BTRFS_EXTENT_CSUM_KEY);
3578ad48fd75SYan, Zheng 
3579e902baacSDavid Sterba 	if (btrfs_leaf_free_space(leaf) >= ins_len)
3580ad48fd75SYan, Zheng 		return 0;
3581ad48fd75SYan, Zheng 
35823212fa14SJosef Bacik 	item_size = btrfs_item_size(leaf, path->slots[0]);
3583ad48fd75SYan, Zheng 	if (key.type == BTRFS_EXTENT_DATA_KEY) {
3584ad48fd75SYan, Zheng 		fi = btrfs_item_ptr(leaf, path->slots[0],
3585ad48fd75SYan, Zheng 				    struct btrfs_file_extent_item);
3586ad48fd75SYan, Zheng 		extent_len = btrfs_file_extent_num_bytes(leaf, fi);
3587ad48fd75SYan, Zheng 	}
3588b3b4aa74SDavid Sterba 	btrfs_release_path(path);
3589ad48fd75SYan, Zheng 
3590ad48fd75SYan, Zheng 	path->keep_locks = 1;
3591ad48fd75SYan, Zheng 	path->search_for_split = 1;
3592ad48fd75SYan, Zheng 	ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
3593ad48fd75SYan, Zheng 	path->search_for_split = 0;
3594a8df6fe6SFilipe Manana 	if (ret > 0)
3595a8df6fe6SFilipe Manana 		ret = -EAGAIN;
3596ad48fd75SYan, Zheng 	if (ret < 0)
3597ad48fd75SYan, Zheng 		goto err;
3598ad48fd75SYan, Zheng 
3599ad48fd75SYan, Zheng 	ret = -EAGAIN;
3600ad48fd75SYan, Zheng 	leaf = path->nodes[0];
3601a8df6fe6SFilipe Manana 	/* if our item isn't there, return now */
36023212fa14SJosef Bacik 	if (item_size != btrfs_item_size(leaf, path->slots[0]))
3603ad48fd75SYan, Zheng 		goto err;
3604ad48fd75SYan, Zheng 
3605109f6aefSChris Mason 	/* the leaf has  changed, it now has room.  return now */
3606e902baacSDavid Sterba 	if (btrfs_leaf_free_space(path->nodes[0]) >= ins_len)
3607109f6aefSChris Mason 		goto err;
3608109f6aefSChris Mason 
3609ad48fd75SYan, Zheng 	if (key.type == BTRFS_EXTENT_DATA_KEY) {
3610ad48fd75SYan, Zheng 		fi = btrfs_item_ptr(leaf, path->slots[0],
3611ad48fd75SYan, Zheng 				    struct btrfs_file_extent_item);
3612ad48fd75SYan, Zheng 		if (extent_len != btrfs_file_extent_num_bytes(leaf, fi))
3613ad48fd75SYan, Zheng 			goto err;
3614ad48fd75SYan, Zheng 	}
3615ad48fd75SYan, Zheng 
3616ad48fd75SYan, Zheng 	ret = split_leaf(trans, root, &key, path, ins_len, 1);
3617f0486c68SYan, Zheng 	if (ret)
3618f0486c68SYan, Zheng 		goto err;
3619ad48fd75SYan, Zheng 
3620ad48fd75SYan, Zheng 	path->keep_locks = 0;
3621ad48fd75SYan, Zheng 	btrfs_unlock_up_safe(path, 1);
3622ad48fd75SYan, Zheng 	return 0;
3623ad48fd75SYan, Zheng err:
3624ad48fd75SYan, Zheng 	path->keep_locks = 0;
3625ad48fd75SYan, Zheng 	return ret;
3626ad48fd75SYan, Zheng }
3627ad48fd75SYan, Zheng 
362825263cd7SDavid Sterba static noinline int split_item(struct btrfs_path *path,
3629310712b2SOmar Sandoval 			       const struct btrfs_key *new_key,
3630459931ecSChris Mason 			       unsigned long split_offset)
3631459931ecSChris Mason {
3632459931ecSChris Mason 	struct extent_buffer *leaf;
3633c91666b1SJosef Bacik 	int orig_slot, slot;
3634ad48fd75SYan, Zheng 	char *buf;
3635459931ecSChris Mason 	u32 nritems;
3636ad48fd75SYan, Zheng 	u32 item_size;
3637459931ecSChris Mason 	u32 orig_offset;
3638459931ecSChris Mason 	struct btrfs_disk_key disk_key;
3639459931ecSChris Mason 
3640459931ecSChris Mason 	leaf = path->nodes[0];
3641e902baacSDavid Sterba 	BUG_ON(btrfs_leaf_free_space(leaf) < sizeof(struct btrfs_item));
3642b9473439SChris Mason 
3643c91666b1SJosef Bacik 	orig_slot = path->slots[0];
36443212fa14SJosef Bacik 	orig_offset = btrfs_item_offset(leaf, path->slots[0]);
36453212fa14SJosef Bacik 	item_size = btrfs_item_size(leaf, path->slots[0]);
3646459931ecSChris Mason 
3647459931ecSChris Mason 	buf = kmalloc(item_size, GFP_NOFS);
3648ad48fd75SYan, Zheng 	if (!buf)
3649ad48fd75SYan, Zheng 		return -ENOMEM;
3650ad48fd75SYan, Zheng 
3651459931ecSChris Mason 	read_extent_buffer(leaf, buf, btrfs_item_ptr_offset(leaf,
3652459931ecSChris Mason 			    path->slots[0]), item_size);
3653ad48fd75SYan, Zheng 
3654459931ecSChris Mason 	slot = path->slots[0] + 1;
3655459931ecSChris Mason 	nritems = btrfs_header_nritems(leaf);
3656459931ecSChris Mason 	if (slot != nritems) {
3657459931ecSChris Mason 		/* shift the items */
3658459931ecSChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1),
3659459931ecSChris Mason 				btrfs_item_nr_offset(slot),
3660459931ecSChris Mason 				(nritems - slot) * sizeof(struct btrfs_item));
3661459931ecSChris Mason 	}
3662459931ecSChris Mason 
3663459931ecSChris Mason 	btrfs_cpu_key_to_disk(&disk_key, new_key);
3664459931ecSChris Mason 	btrfs_set_item_key(leaf, &disk_key, slot);
3665459931ecSChris Mason 
36663212fa14SJosef Bacik 	btrfs_set_item_offset(leaf, slot, orig_offset);
36673212fa14SJosef Bacik 	btrfs_set_item_size(leaf, slot, item_size - split_offset);
3668459931ecSChris Mason 
36693212fa14SJosef Bacik 	btrfs_set_item_offset(leaf, orig_slot,
3670459931ecSChris Mason 				 orig_offset + item_size - split_offset);
36713212fa14SJosef Bacik 	btrfs_set_item_size(leaf, orig_slot, split_offset);
3672459931ecSChris Mason 
3673459931ecSChris Mason 	btrfs_set_header_nritems(leaf, nritems + 1);
3674459931ecSChris Mason 
3675459931ecSChris Mason 	/* write the data for the start of the original item */
3676459931ecSChris Mason 	write_extent_buffer(leaf, buf,
3677459931ecSChris Mason 			    btrfs_item_ptr_offset(leaf, path->slots[0]),
3678459931ecSChris Mason 			    split_offset);
3679459931ecSChris Mason 
3680459931ecSChris Mason 	/* write the data for the new item */
3681459931ecSChris Mason 	write_extent_buffer(leaf, buf + split_offset,
3682459931ecSChris Mason 			    btrfs_item_ptr_offset(leaf, slot),
3683459931ecSChris Mason 			    item_size - split_offset);
3684459931ecSChris Mason 	btrfs_mark_buffer_dirty(leaf);
3685459931ecSChris Mason 
3686e902baacSDavid Sterba 	BUG_ON(btrfs_leaf_free_space(leaf) < 0);
3687459931ecSChris Mason 	kfree(buf);
3688ad48fd75SYan, Zheng 	return 0;
3689ad48fd75SYan, Zheng }
3690ad48fd75SYan, Zheng 
3691ad48fd75SYan, Zheng /*
3692ad48fd75SYan, Zheng  * This function splits a single item into two items,
3693ad48fd75SYan, Zheng  * giving 'new_key' to the new item and splitting the
3694ad48fd75SYan, Zheng  * old one at split_offset (from the start of the item).
3695ad48fd75SYan, Zheng  *
3696ad48fd75SYan, Zheng  * The path may be released by this operation.  After
3697ad48fd75SYan, Zheng  * the split, the path is pointing to the old item.  The
3698ad48fd75SYan, Zheng  * new item is going to be in the same node as the old one.
3699ad48fd75SYan, Zheng  *
3700ad48fd75SYan, Zheng  * Note, the item being split must be smaller enough to live alone on
3701ad48fd75SYan, Zheng  * a tree block with room for one extra struct btrfs_item
3702ad48fd75SYan, Zheng  *
3703ad48fd75SYan, Zheng  * This allows us to split the item in place, keeping a lock on the
3704ad48fd75SYan, Zheng  * leaf the entire time.
3705ad48fd75SYan, Zheng  */
3706ad48fd75SYan, Zheng int btrfs_split_item(struct btrfs_trans_handle *trans,
3707ad48fd75SYan, Zheng 		     struct btrfs_root *root,
3708ad48fd75SYan, Zheng 		     struct btrfs_path *path,
3709310712b2SOmar Sandoval 		     const struct btrfs_key *new_key,
3710ad48fd75SYan, Zheng 		     unsigned long split_offset)
3711ad48fd75SYan, Zheng {
3712ad48fd75SYan, Zheng 	int ret;
3713ad48fd75SYan, Zheng 	ret = setup_leaf_for_split(trans, root, path,
3714ad48fd75SYan, Zheng 				   sizeof(struct btrfs_item));
3715ad48fd75SYan, Zheng 	if (ret)
3716459931ecSChris Mason 		return ret;
3717ad48fd75SYan, Zheng 
371825263cd7SDavid Sterba 	ret = split_item(path, new_key, split_offset);
3719ad48fd75SYan, Zheng 	return ret;
3720ad48fd75SYan, Zheng }
3721ad48fd75SYan, Zheng 
3722ad48fd75SYan, Zheng /*
3723d352ac68SChris Mason  * make the item pointed to by the path smaller.  new_size indicates
3724d352ac68SChris Mason  * how small to make it, and from_end tells us if we just chop bytes
3725d352ac68SChris Mason  * off the end of the item or if we shift the item to chop bytes off
3726d352ac68SChris Mason  * the front.
3727d352ac68SChris Mason  */
372878ac4f9eSDavid Sterba void btrfs_truncate_item(struct btrfs_path *path, u32 new_size, int from_end)
3729b18c6685SChris Mason {
3730b18c6685SChris Mason 	int slot;
37315f39d397SChris Mason 	struct extent_buffer *leaf;
3732b18c6685SChris Mason 	u32 nritems;
3733b18c6685SChris Mason 	unsigned int data_end;
3734b18c6685SChris Mason 	unsigned int old_data_start;
3735b18c6685SChris Mason 	unsigned int old_size;
3736b18c6685SChris Mason 	unsigned int size_diff;
3737b18c6685SChris Mason 	int i;
3738cfed81a0SChris Mason 	struct btrfs_map_token token;
3739cfed81a0SChris Mason 
37405f39d397SChris Mason 	leaf = path->nodes[0];
3741179e29e4SChris Mason 	slot = path->slots[0];
3742179e29e4SChris Mason 
37433212fa14SJosef Bacik 	old_size = btrfs_item_size(leaf, slot);
3744179e29e4SChris Mason 	if (old_size == new_size)
3745143bede5SJeff Mahoney 		return;
3746b18c6685SChris Mason 
37475f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
37488f881e8cSDavid Sterba 	data_end = leaf_data_end(leaf);
3749b18c6685SChris Mason 
37503212fa14SJosef Bacik 	old_data_start = btrfs_item_offset(leaf, slot);
3751179e29e4SChris Mason 
3752b18c6685SChris Mason 	size_diff = old_size - new_size;
3753b18c6685SChris Mason 
3754b18c6685SChris Mason 	BUG_ON(slot < 0);
3755b18c6685SChris Mason 	BUG_ON(slot >= nritems);
3756b18c6685SChris Mason 
3757b18c6685SChris Mason 	/*
3758b18c6685SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
3759b18c6685SChris Mason 	 */
3760b18c6685SChris Mason 	/* first correct the data pointers */
3761c82f823cSDavid Sterba 	btrfs_init_map_token(&token, leaf);
3762b18c6685SChris Mason 	for (i = slot; i < nritems; i++) {
37635f39d397SChris Mason 		u32 ioff;
3764db94535dSChris Mason 
37653212fa14SJosef Bacik 		ioff = btrfs_token_item_offset(&token, i);
37663212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, i, ioff + size_diff);
3767b18c6685SChris Mason 	}
3768db94535dSChris Mason 
3769b18c6685SChris Mason 	/* shift the data */
3770179e29e4SChris Mason 	if (from_end) {
37713d9ec8c4SNikolay Borisov 		memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET +
37723d9ec8c4SNikolay Borisov 			      data_end + size_diff, BTRFS_LEAF_DATA_OFFSET +
3773b18c6685SChris Mason 			      data_end, old_data_start + new_size - data_end);
3774179e29e4SChris Mason 	} else {
3775179e29e4SChris Mason 		struct btrfs_disk_key disk_key;
3776179e29e4SChris Mason 		u64 offset;
3777179e29e4SChris Mason 
3778179e29e4SChris Mason 		btrfs_item_key(leaf, &disk_key, slot);
3779179e29e4SChris Mason 
3780179e29e4SChris Mason 		if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) {
3781179e29e4SChris Mason 			unsigned long ptr;
3782179e29e4SChris Mason 			struct btrfs_file_extent_item *fi;
3783179e29e4SChris Mason 
3784179e29e4SChris Mason 			fi = btrfs_item_ptr(leaf, slot,
3785179e29e4SChris Mason 					    struct btrfs_file_extent_item);
3786179e29e4SChris Mason 			fi = (struct btrfs_file_extent_item *)(
3787179e29e4SChris Mason 			     (unsigned long)fi - size_diff);
3788179e29e4SChris Mason 
3789179e29e4SChris Mason 			if (btrfs_file_extent_type(leaf, fi) ==
3790179e29e4SChris Mason 			    BTRFS_FILE_EXTENT_INLINE) {
3791179e29e4SChris Mason 				ptr = btrfs_item_ptr_offset(leaf, slot);
3792179e29e4SChris Mason 				memmove_extent_buffer(leaf, ptr,
3793179e29e4SChris Mason 				      (unsigned long)fi,
37947ec20afbSDavid Sterba 				      BTRFS_FILE_EXTENT_INLINE_DATA_START);
3795179e29e4SChris Mason 			}
3796179e29e4SChris Mason 		}
3797179e29e4SChris Mason 
37983d9ec8c4SNikolay Borisov 		memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET +
37993d9ec8c4SNikolay Borisov 			      data_end + size_diff, BTRFS_LEAF_DATA_OFFSET +
3800179e29e4SChris Mason 			      data_end, old_data_start - data_end);
3801179e29e4SChris Mason 
3802179e29e4SChris Mason 		offset = btrfs_disk_key_offset(&disk_key);
3803179e29e4SChris Mason 		btrfs_set_disk_key_offset(&disk_key, offset + size_diff);
3804179e29e4SChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot);
3805179e29e4SChris Mason 		if (slot == 0)
3806b167fa91SNikolay Borisov 			fixup_low_keys(path, &disk_key, 1);
3807179e29e4SChris Mason 	}
38085f39d397SChris Mason 
38093212fa14SJosef Bacik 	btrfs_set_item_size(leaf, slot, new_size);
38105f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
3811b18c6685SChris Mason 
3812e902baacSDavid Sterba 	if (btrfs_leaf_free_space(leaf) < 0) {
3813a4f78750SDavid Sterba 		btrfs_print_leaf(leaf);
3814b18c6685SChris Mason 		BUG();
38155f39d397SChris Mason 	}
3816b18c6685SChris Mason }
3817b18c6685SChris Mason 
3818d352ac68SChris Mason /*
38198f69dbd2SStefan Behrens  * make the item pointed to by the path bigger, data_size is the added size.
3820d352ac68SChris Mason  */
3821c71dd880SDavid Sterba void btrfs_extend_item(struct btrfs_path *path, u32 data_size)
38226567e837SChris Mason {
38236567e837SChris Mason 	int slot;
38245f39d397SChris Mason 	struct extent_buffer *leaf;
38256567e837SChris Mason 	u32 nritems;
38266567e837SChris Mason 	unsigned int data_end;
38276567e837SChris Mason 	unsigned int old_data;
38286567e837SChris Mason 	unsigned int old_size;
38296567e837SChris Mason 	int i;
3830cfed81a0SChris Mason 	struct btrfs_map_token token;
3831cfed81a0SChris Mason 
38325f39d397SChris Mason 	leaf = path->nodes[0];
38336567e837SChris Mason 
38345f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
38358f881e8cSDavid Sterba 	data_end = leaf_data_end(leaf);
38366567e837SChris Mason 
3837e902baacSDavid Sterba 	if (btrfs_leaf_free_space(leaf) < data_size) {
3838a4f78750SDavid Sterba 		btrfs_print_leaf(leaf);
38396567e837SChris Mason 		BUG();
38405f39d397SChris Mason 	}
38416567e837SChris Mason 	slot = path->slots[0];
3842dc2e724eSJosef Bacik 	old_data = btrfs_item_data_end(leaf, slot);
38436567e837SChris Mason 
38446567e837SChris Mason 	BUG_ON(slot < 0);
38453326d1b0SChris Mason 	if (slot >= nritems) {
3846a4f78750SDavid Sterba 		btrfs_print_leaf(leaf);
3847c71dd880SDavid Sterba 		btrfs_crit(leaf->fs_info, "slot %d too large, nritems %d",
3848d397712bSChris Mason 			   slot, nritems);
3849290342f6SArnd Bergmann 		BUG();
38503326d1b0SChris Mason 	}
38516567e837SChris Mason 
38526567e837SChris Mason 	/*
38536567e837SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
38546567e837SChris Mason 	 */
38556567e837SChris Mason 	/* first correct the data pointers */
3856c82f823cSDavid Sterba 	btrfs_init_map_token(&token, leaf);
38576567e837SChris Mason 	for (i = slot; i < nritems; i++) {
38585f39d397SChris Mason 		u32 ioff;
3859db94535dSChris Mason 
38603212fa14SJosef Bacik 		ioff = btrfs_token_item_offset(&token, i);
38613212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, i, ioff - data_size);
38626567e837SChris Mason 	}
38635f39d397SChris Mason 
38646567e837SChris Mason 	/* shift the data */
38653d9ec8c4SNikolay Borisov 	memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET +
38663d9ec8c4SNikolay Borisov 		      data_end - data_size, BTRFS_LEAF_DATA_OFFSET +
38676567e837SChris Mason 		      data_end, old_data - data_end);
38685f39d397SChris Mason 
38696567e837SChris Mason 	data_end = old_data;
38703212fa14SJosef Bacik 	old_size = btrfs_item_size(leaf, slot);
38713212fa14SJosef Bacik 	btrfs_set_item_size(leaf, slot, old_size + data_size);
38725f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
38736567e837SChris Mason 
3874e902baacSDavid Sterba 	if (btrfs_leaf_free_space(leaf) < 0) {
3875a4f78750SDavid Sterba 		btrfs_print_leaf(leaf);
38766567e837SChris Mason 		BUG();
38775f39d397SChris Mason 	}
38786567e837SChris Mason }
38796567e837SChris Mason 
3880da9ffb24SNikolay Borisov /**
3881da9ffb24SNikolay Borisov  * setup_items_for_insert - Helper called before inserting one or more items
3882da9ffb24SNikolay Borisov  * to a leaf. Main purpose is to save stack depth by doing the bulk of the work
3883da9ffb24SNikolay Borisov  * in a function that doesn't call btrfs_search_slot
3884da9ffb24SNikolay Borisov  *
3885da9ffb24SNikolay Borisov  * @root:	root we are inserting items to
3886da9ffb24SNikolay Borisov  * @path:	points to the leaf/slot where we are going to insert new items
3887b7ef5f3aSFilipe Manana  * @batch:      information about the batch of items to insert
388874123bd7SChris Mason  */
3889f0641656SFilipe Manana static void setup_items_for_insert(struct btrfs_root *root, struct btrfs_path *path,
3890b7ef5f3aSFilipe Manana 				   const struct btrfs_item_batch *batch)
3891be0e5c09SChris Mason {
38920b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
38939c58309dSChris Mason 	int i;
38947518a238SChris Mason 	u32 nritems;
3895be0e5c09SChris Mason 	unsigned int data_end;
3896e2fa7227SChris Mason 	struct btrfs_disk_key disk_key;
389744871b1bSChris Mason 	struct extent_buffer *leaf;
389844871b1bSChris Mason 	int slot;
3899cfed81a0SChris Mason 	struct btrfs_map_token token;
3900fc0d82e1SNikolay Borisov 	u32 total_size;
3901fc0d82e1SNikolay Borisov 
3902b7ef5f3aSFilipe Manana 	/*
3903b7ef5f3aSFilipe Manana 	 * Before anything else, update keys in the parent and other ancestors
3904b7ef5f3aSFilipe Manana 	 * if needed, then release the write locks on them, so that other tasks
3905b7ef5f3aSFilipe Manana 	 * can use them while we modify the leaf.
3906b7ef5f3aSFilipe Manana 	 */
390724cdc847SFilipe Manana 	if (path->slots[0] == 0) {
3908b7ef5f3aSFilipe Manana 		btrfs_cpu_key_to_disk(&disk_key, &batch->keys[0]);
3909b167fa91SNikolay Borisov 		fixup_low_keys(path, &disk_key, 1);
391024cdc847SFilipe Manana 	}
391124cdc847SFilipe Manana 	btrfs_unlock_up_safe(path, 1);
391224cdc847SFilipe Manana 
39135f39d397SChris Mason 	leaf = path->nodes[0];
391444871b1bSChris Mason 	slot = path->slots[0];
391574123bd7SChris Mason 
39165f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
39178f881e8cSDavid Sterba 	data_end = leaf_data_end(leaf);
3918b7ef5f3aSFilipe Manana 	total_size = batch->total_data_size + (batch->nr * sizeof(struct btrfs_item));
3919eb60ceacSChris Mason 
3920e902baacSDavid Sterba 	if (btrfs_leaf_free_space(leaf) < total_size) {
3921a4f78750SDavid Sterba 		btrfs_print_leaf(leaf);
39220b246afaSJeff Mahoney 		btrfs_crit(fs_info, "not enough freespace need %u have %d",
3923e902baacSDavid Sterba 			   total_size, btrfs_leaf_free_space(leaf));
3924be0e5c09SChris Mason 		BUG();
3925d4dbff95SChris Mason 	}
39265f39d397SChris Mason 
3927c82f823cSDavid Sterba 	btrfs_init_map_token(&token, leaf);
3928be0e5c09SChris Mason 	if (slot != nritems) {
3929dc2e724eSJosef Bacik 		unsigned int old_data = btrfs_item_data_end(leaf, slot);
3930be0e5c09SChris Mason 
39315f39d397SChris Mason 		if (old_data < data_end) {
3932a4f78750SDavid Sterba 			btrfs_print_leaf(leaf);
39337269ddd2SNikolay Borisov 			btrfs_crit(fs_info,
39347269ddd2SNikolay Borisov 		"item at slot %d with data offset %u beyond data end of leaf %u",
39355f39d397SChris Mason 				   slot, old_data, data_end);
3936290342f6SArnd Bergmann 			BUG();
39375f39d397SChris Mason 		}
3938be0e5c09SChris Mason 		/*
3939be0e5c09SChris Mason 		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
3940be0e5c09SChris Mason 		 */
3941be0e5c09SChris Mason 		/* first correct the data pointers */
39420783fcfcSChris Mason 		for (i = slot; i < nritems; i++) {
39435f39d397SChris Mason 			u32 ioff;
3944db94535dSChris Mason 
39453212fa14SJosef Bacik 			ioff = btrfs_token_item_offset(&token, i);
39463212fa14SJosef Bacik 			btrfs_set_token_item_offset(&token, i,
3947b7ef5f3aSFilipe Manana 						       ioff - batch->total_data_size);
39480783fcfcSChris Mason 		}
3949be0e5c09SChris Mason 		/* shift the items */
3950b7ef5f3aSFilipe Manana 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + batch->nr),
39515f39d397SChris Mason 			      btrfs_item_nr_offset(slot),
39520783fcfcSChris Mason 			      (nritems - slot) * sizeof(struct btrfs_item));
3953be0e5c09SChris Mason 
3954be0e5c09SChris Mason 		/* shift the data */
39553d9ec8c4SNikolay Borisov 		memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET +
3956b7ef5f3aSFilipe Manana 				      data_end - batch->total_data_size,
3957b7ef5f3aSFilipe Manana 				      BTRFS_LEAF_DATA_OFFSET + data_end,
3958b7ef5f3aSFilipe Manana 				      old_data - data_end);
3959be0e5c09SChris Mason 		data_end = old_data;
3960be0e5c09SChris Mason 	}
39615f39d397SChris Mason 
396262e2749eSChris Mason 	/* setup the item for the new data */
3963b7ef5f3aSFilipe Manana 	for (i = 0; i < batch->nr; i++) {
3964b7ef5f3aSFilipe Manana 		btrfs_cpu_key_to_disk(&disk_key, &batch->keys[i]);
39659c58309dSChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot + i);
3966b7ef5f3aSFilipe Manana 		data_end -= batch->data_sizes[i];
39673212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, slot + i, data_end);
39683212fa14SJosef Bacik 		btrfs_set_token_item_size(&token, slot + i, batch->data_sizes[i]);
39699c58309dSChris Mason 	}
397044871b1bSChris Mason 
3971b7ef5f3aSFilipe Manana 	btrfs_set_header_nritems(leaf, nritems + batch->nr);
3972b9473439SChris Mason 	btrfs_mark_buffer_dirty(leaf);
3973aa5d6bedSChris Mason 
3974e902baacSDavid Sterba 	if (btrfs_leaf_free_space(leaf) < 0) {
3975a4f78750SDavid Sterba 		btrfs_print_leaf(leaf);
3976be0e5c09SChris Mason 		BUG();
39775f39d397SChris Mason 	}
397844871b1bSChris Mason }
397944871b1bSChris Mason 
398044871b1bSChris Mason /*
3981f0641656SFilipe Manana  * Insert a new item into a leaf.
3982f0641656SFilipe Manana  *
3983f0641656SFilipe Manana  * @root:      The root of the btree.
3984f0641656SFilipe Manana  * @path:      A path pointing to the target leaf and slot.
3985f0641656SFilipe Manana  * @key:       The key of the new item.
3986f0641656SFilipe Manana  * @data_size: The size of the data associated with the new key.
3987f0641656SFilipe Manana  */
3988f0641656SFilipe Manana void btrfs_setup_item_for_insert(struct btrfs_root *root,
3989f0641656SFilipe Manana 				 struct btrfs_path *path,
3990f0641656SFilipe Manana 				 const struct btrfs_key *key,
3991f0641656SFilipe Manana 				 u32 data_size)
3992f0641656SFilipe Manana {
3993f0641656SFilipe Manana 	struct btrfs_item_batch batch;
3994f0641656SFilipe Manana 
3995f0641656SFilipe Manana 	batch.keys = key;
3996f0641656SFilipe Manana 	batch.data_sizes = &data_size;
3997f0641656SFilipe Manana 	batch.total_data_size = data_size;
3998f0641656SFilipe Manana 	batch.nr = 1;
3999f0641656SFilipe Manana 
4000f0641656SFilipe Manana 	setup_items_for_insert(root, path, &batch);
4001f0641656SFilipe Manana }
4002f0641656SFilipe Manana 
4003f0641656SFilipe Manana /*
400444871b1bSChris Mason  * Given a key and some data, insert items into the tree.
400544871b1bSChris Mason  * This does all the path init required, making room in the tree if needed.
400644871b1bSChris Mason  */
400744871b1bSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans,
400844871b1bSChris Mason 			    struct btrfs_root *root,
400944871b1bSChris Mason 			    struct btrfs_path *path,
4010b7ef5f3aSFilipe Manana 			    const struct btrfs_item_batch *batch)
401144871b1bSChris Mason {
401244871b1bSChris Mason 	int ret = 0;
401344871b1bSChris Mason 	int slot;
4014b7ef5f3aSFilipe Manana 	u32 total_size;
401544871b1bSChris Mason 
4016b7ef5f3aSFilipe Manana 	total_size = batch->total_data_size + (batch->nr * sizeof(struct btrfs_item));
4017b7ef5f3aSFilipe Manana 	ret = btrfs_search_slot(trans, root, &batch->keys[0], path, total_size, 1);
401844871b1bSChris Mason 	if (ret == 0)
401944871b1bSChris Mason 		return -EEXIST;
402044871b1bSChris Mason 	if (ret < 0)
4021143bede5SJeff Mahoney 		return ret;
402244871b1bSChris Mason 
402344871b1bSChris Mason 	slot = path->slots[0];
402444871b1bSChris Mason 	BUG_ON(slot < 0);
402544871b1bSChris Mason 
4026b7ef5f3aSFilipe Manana 	setup_items_for_insert(root, path, batch);
4027143bede5SJeff Mahoney 	return 0;
402862e2749eSChris Mason }
402962e2749eSChris Mason 
403062e2749eSChris Mason /*
403162e2749eSChris Mason  * Given a key and some data, insert an item into the tree.
403262e2749eSChris Mason  * This does all the path init required, making room in the tree if needed.
403362e2749eSChris Mason  */
4034310712b2SOmar Sandoval int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root *root,
4035310712b2SOmar Sandoval 		      const struct btrfs_key *cpu_key, void *data,
4036310712b2SOmar Sandoval 		      u32 data_size)
403762e2749eSChris Mason {
403862e2749eSChris Mason 	int ret = 0;
40392c90e5d6SChris Mason 	struct btrfs_path *path;
40405f39d397SChris Mason 	struct extent_buffer *leaf;
40415f39d397SChris Mason 	unsigned long ptr;
404262e2749eSChris Mason 
40432c90e5d6SChris Mason 	path = btrfs_alloc_path();
4044db5b493aSTsutomu Itoh 	if (!path)
4045db5b493aSTsutomu Itoh 		return -ENOMEM;
40462c90e5d6SChris Mason 	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
404762e2749eSChris Mason 	if (!ret) {
40485f39d397SChris Mason 		leaf = path->nodes[0];
40495f39d397SChris Mason 		ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
40505f39d397SChris Mason 		write_extent_buffer(leaf, data, ptr, data_size);
40515f39d397SChris Mason 		btrfs_mark_buffer_dirty(leaf);
405262e2749eSChris Mason 	}
40532c90e5d6SChris Mason 	btrfs_free_path(path);
4054aa5d6bedSChris Mason 	return ret;
4055be0e5c09SChris Mason }
4056be0e5c09SChris Mason 
405774123bd7SChris Mason /*
4058f0641656SFilipe Manana  * This function duplicates an item, giving 'new_key' to the new item.
4059f0641656SFilipe Manana  * It guarantees both items live in the same tree leaf and the new item is
4060f0641656SFilipe Manana  * contiguous with the original item.
4061f0641656SFilipe Manana  *
4062f0641656SFilipe Manana  * This allows us to split a file extent in place, keeping a lock on the leaf
4063f0641656SFilipe Manana  * the entire time.
4064f0641656SFilipe Manana  */
4065f0641656SFilipe Manana int btrfs_duplicate_item(struct btrfs_trans_handle *trans,
4066f0641656SFilipe Manana 			 struct btrfs_root *root,
4067f0641656SFilipe Manana 			 struct btrfs_path *path,
4068f0641656SFilipe Manana 			 const struct btrfs_key *new_key)
4069f0641656SFilipe Manana {
4070f0641656SFilipe Manana 	struct extent_buffer *leaf;
4071f0641656SFilipe Manana 	int ret;
4072f0641656SFilipe Manana 	u32 item_size;
4073f0641656SFilipe Manana 
4074f0641656SFilipe Manana 	leaf = path->nodes[0];
40753212fa14SJosef Bacik 	item_size = btrfs_item_size(leaf, path->slots[0]);
4076f0641656SFilipe Manana 	ret = setup_leaf_for_split(trans, root, path,
4077f0641656SFilipe Manana 				   item_size + sizeof(struct btrfs_item));
4078f0641656SFilipe Manana 	if (ret)
4079f0641656SFilipe Manana 		return ret;
4080f0641656SFilipe Manana 
4081f0641656SFilipe Manana 	path->slots[0]++;
4082f0641656SFilipe Manana 	btrfs_setup_item_for_insert(root, path, new_key, item_size);
4083f0641656SFilipe Manana 	leaf = path->nodes[0];
4084f0641656SFilipe Manana 	memcpy_extent_buffer(leaf,
4085f0641656SFilipe Manana 			     btrfs_item_ptr_offset(leaf, path->slots[0]),
4086f0641656SFilipe Manana 			     btrfs_item_ptr_offset(leaf, path->slots[0] - 1),
4087f0641656SFilipe Manana 			     item_size);
4088f0641656SFilipe Manana 	return 0;
4089f0641656SFilipe Manana }
4090f0641656SFilipe Manana 
4091f0641656SFilipe Manana /*
40925de08d7dSChris Mason  * delete the pointer from a given node.
409374123bd7SChris Mason  *
4094d352ac68SChris Mason  * the tree should have been previously balanced so the deletion does not
4095d352ac68SChris Mason  * empty a node.
409674123bd7SChris Mason  */
4097afe5fea7STsutomu Itoh static void del_ptr(struct btrfs_root *root, struct btrfs_path *path,
4098afe5fea7STsutomu Itoh 		    int level, int slot)
4099be0e5c09SChris Mason {
41005f39d397SChris Mason 	struct extent_buffer *parent = path->nodes[level];
41017518a238SChris Mason 	u32 nritems;
4102f3ea38daSJan Schmidt 	int ret;
4103be0e5c09SChris Mason 
41045f39d397SChris Mason 	nritems = btrfs_header_nritems(parent);
4105be0e5c09SChris Mason 	if (slot != nritems - 1) {
4106bf1d3425SDavid Sterba 		if (level) {
4107f3a84ccdSFilipe Manana 			ret = btrfs_tree_mod_log_insert_move(parent, slot,
4108f3a84ccdSFilipe Manana 					slot + 1, nritems - slot - 1);
4109bf1d3425SDavid Sterba 			BUG_ON(ret < 0);
4110bf1d3425SDavid Sterba 		}
41115f39d397SChris Mason 		memmove_extent_buffer(parent,
41125f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
41135f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
4114d6025579SChris Mason 			      sizeof(struct btrfs_key_ptr) *
4115d6025579SChris Mason 			      (nritems - slot - 1));
411657ba86c0SChris Mason 	} else if (level) {
4117f3a84ccdSFilipe Manana 		ret = btrfs_tree_mod_log_insert_key(parent, slot,
4118f3a84ccdSFilipe Manana 				BTRFS_MOD_LOG_KEY_REMOVE, GFP_NOFS);
411957ba86c0SChris Mason 		BUG_ON(ret < 0);
4120be0e5c09SChris Mason 	}
4121f3ea38daSJan Schmidt 
41227518a238SChris Mason 	nritems--;
41235f39d397SChris Mason 	btrfs_set_header_nritems(parent, nritems);
41247518a238SChris Mason 	if (nritems == 0 && parent == root->node) {
41255f39d397SChris Mason 		BUG_ON(btrfs_header_level(root->node) != 1);
4126eb60ceacSChris Mason 		/* just turn the root into a leaf and break */
41275f39d397SChris Mason 		btrfs_set_header_level(root->node, 0);
4128bb803951SChris Mason 	} else if (slot == 0) {
41295f39d397SChris Mason 		struct btrfs_disk_key disk_key;
41305f39d397SChris Mason 
41315f39d397SChris Mason 		btrfs_node_key(parent, &disk_key, 0);
4132b167fa91SNikolay Borisov 		fixup_low_keys(path, &disk_key, level + 1);
4133be0e5c09SChris Mason 	}
4134d6025579SChris Mason 	btrfs_mark_buffer_dirty(parent);
4135be0e5c09SChris Mason }
4136be0e5c09SChris Mason 
413774123bd7SChris Mason /*
4138323ac95bSChris Mason  * a helper function to delete the leaf pointed to by path->slots[1] and
41395d4f98a2SYan Zheng  * path->nodes[1].
4140323ac95bSChris Mason  *
4141323ac95bSChris Mason  * This deletes the pointer in path->nodes[1] and frees the leaf
4142323ac95bSChris Mason  * block extent.  zero is returned if it all worked out, < 0 otherwise.
4143323ac95bSChris Mason  *
4144323ac95bSChris Mason  * The path must have already been setup for deleting the leaf, including
4145323ac95bSChris Mason  * all the proper balancing.  path->nodes[1] must be locked.
4146323ac95bSChris Mason  */
4147143bede5SJeff Mahoney static noinline void btrfs_del_leaf(struct btrfs_trans_handle *trans,
4148323ac95bSChris Mason 				    struct btrfs_root *root,
41495d4f98a2SYan Zheng 				    struct btrfs_path *path,
41505d4f98a2SYan Zheng 				    struct extent_buffer *leaf)
4151323ac95bSChris Mason {
41525d4f98a2SYan Zheng 	WARN_ON(btrfs_header_generation(leaf) != trans->transid);
4153afe5fea7STsutomu Itoh 	del_ptr(root, path, 1, path->slots[1]);
4154323ac95bSChris Mason 
41554d081c41SChris Mason 	/*
41564d081c41SChris Mason 	 * btrfs_free_extent is expensive, we want to make sure we
41574d081c41SChris Mason 	 * aren't holding any locks when we call it
41584d081c41SChris Mason 	 */
41594d081c41SChris Mason 	btrfs_unlock_up_safe(path, 0);
41604d081c41SChris Mason 
4161f0486c68SYan, Zheng 	root_sub_used(root, leaf->len);
4162f0486c68SYan, Zheng 
416367439dadSDavid Sterba 	atomic_inc(&leaf->refs);
41647a163608SFilipe Manana 	btrfs_free_tree_block(trans, btrfs_root_id(root), leaf, 0, 1);
41653083ee2eSJosef Bacik 	free_extent_buffer_stale(leaf);
4166323ac95bSChris Mason }
4167323ac95bSChris Mason /*
416874123bd7SChris Mason  * delete the item at the leaf level in path.  If that empties
416974123bd7SChris Mason  * the leaf, remove it from the tree
417074123bd7SChris Mason  */
417185e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root,
417285e21bacSChris Mason 		    struct btrfs_path *path, int slot, int nr)
4173be0e5c09SChris Mason {
41740b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
41755f39d397SChris Mason 	struct extent_buffer *leaf;
4176ce0eac2aSAlexandru Moise 	u32 last_off;
4177ce0eac2aSAlexandru Moise 	u32 dsize = 0;
4178aa5d6bedSChris Mason 	int ret = 0;
4179aa5d6bedSChris Mason 	int wret;
418085e21bacSChris Mason 	int i;
41817518a238SChris Mason 	u32 nritems;
4182be0e5c09SChris Mason 
41835f39d397SChris Mason 	leaf = path->nodes[0];
41843212fa14SJosef Bacik 	last_off = btrfs_item_offset(leaf, slot + nr - 1);
418585e21bacSChris Mason 
418685e21bacSChris Mason 	for (i = 0; i < nr; i++)
41873212fa14SJosef Bacik 		dsize += btrfs_item_size(leaf, slot + i);
418885e21bacSChris Mason 
41895f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
4190be0e5c09SChris Mason 
419185e21bacSChris Mason 	if (slot + nr != nritems) {
41928f881e8cSDavid Sterba 		int data_end = leaf_data_end(leaf);
4193c82f823cSDavid Sterba 		struct btrfs_map_token token;
41945f39d397SChris Mason 
41953d9ec8c4SNikolay Borisov 		memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET +
4196d6025579SChris Mason 			      data_end + dsize,
41973d9ec8c4SNikolay Borisov 			      BTRFS_LEAF_DATA_OFFSET + data_end,
419885e21bacSChris Mason 			      last_off - data_end);
41995f39d397SChris Mason 
4200c82f823cSDavid Sterba 		btrfs_init_map_token(&token, leaf);
420185e21bacSChris Mason 		for (i = slot + nr; i < nritems; i++) {
42025f39d397SChris Mason 			u32 ioff;
4203db94535dSChris Mason 
42043212fa14SJosef Bacik 			ioff = btrfs_token_item_offset(&token, i);
42053212fa14SJosef Bacik 			btrfs_set_token_item_offset(&token, i, ioff + dsize);
42060783fcfcSChris Mason 		}
4207db94535dSChris Mason 
42085f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
420985e21bacSChris Mason 			      btrfs_item_nr_offset(slot + nr),
42100783fcfcSChris Mason 			      sizeof(struct btrfs_item) *
421185e21bacSChris Mason 			      (nritems - slot - nr));
4212be0e5c09SChris Mason 	}
421385e21bacSChris Mason 	btrfs_set_header_nritems(leaf, nritems - nr);
421485e21bacSChris Mason 	nritems -= nr;
42155f39d397SChris Mason 
421674123bd7SChris Mason 	/* delete the leaf if we've emptied it */
42177518a238SChris Mason 	if (nritems == 0) {
42185f39d397SChris Mason 		if (leaf == root->node) {
42195f39d397SChris Mason 			btrfs_set_header_level(leaf, 0);
42209a8dd150SChris Mason 		} else {
42216a884d7dSDavid Sterba 			btrfs_clean_tree_block(leaf);
4222143bede5SJeff Mahoney 			btrfs_del_leaf(trans, root, path, leaf);
42239a8dd150SChris Mason 		}
4224be0e5c09SChris Mason 	} else {
42257518a238SChris Mason 		int used = leaf_space_used(leaf, 0, nritems);
4226aa5d6bedSChris Mason 		if (slot == 0) {
42275f39d397SChris Mason 			struct btrfs_disk_key disk_key;
42285f39d397SChris Mason 
42295f39d397SChris Mason 			btrfs_item_key(leaf, &disk_key, 0);
4230b167fa91SNikolay Borisov 			fixup_low_keys(path, &disk_key, 1);
4231aa5d6bedSChris Mason 		}
4232aa5d6bedSChris Mason 
423374123bd7SChris Mason 		/* delete the leaf if it is mostly empty */
42340b246afaSJeff Mahoney 		if (used < BTRFS_LEAF_DATA_SIZE(fs_info) / 3) {
4235be0e5c09SChris Mason 			/* push_leaf_left fixes the path.
4236be0e5c09SChris Mason 			 * make sure the path still points to our leaf
4237be0e5c09SChris Mason 			 * for possible call to del_ptr below
4238be0e5c09SChris Mason 			 */
42394920c9acSChris Mason 			slot = path->slots[1];
424067439dadSDavid Sterba 			atomic_inc(&leaf->refs);
42415f39d397SChris Mason 
424299d8f83cSChris Mason 			wret = push_leaf_left(trans, root, path, 1, 1,
424399d8f83cSChris Mason 					      1, (u32)-1);
424454aa1f4dSChris Mason 			if (wret < 0 && wret != -ENOSPC)
4245aa5d6bedSChris Mason 				ret = wret;
42465f39d397SChris Mason 
42475f39d397SChris Mason 			if (path->nodes[0] == leaf &&
42485f39d397SChris Mason 			    btrfs_header_nritems(leaf)) {
424999d8f83cSChris Mason 				wret = push_leaf_right(trans, root, path, 1,
425099d8f83cSChris Mason 						       1, 1, 0);
425154aa1f4dSChris Mason 				if (wret < 0 && wret != -ENOSPC)
4252aa5d6bedSChris Mason 					ret = wret;
4253aa5d6bedSChris Mason 			}
42545f39d397SChris Mason 
42555f39d397SChris Mason 			if (btrfs_header_nritems(leaf) == 0) {
4256323ac95bSChris Mason 				path->slots[1] = slot;
4257143bede5SJeff Mahoney 				btrfs_del_leaf(trans, root, path, leaf);
42585f39d397SChris Mason 				free_extent_buffer(leaf);
4259143bede5SJeff Mahoney 				ret = 0;
42605de08d7dSChris Mason 			} else {
4261925baeddSChris Mason 				/* if we're still in the path, make sure
4262925baeddSChris Mason 				 * we're dirty.  Otherwise, one of the
4263925baeddSChris Mason 				 * push_leaf functions must have already
4264925baeddSChris Mason 				 * dirtied this buffer
4265925baeddSChris Mason 				 */
4266925baeddSChris Mason 				if (path->nodes[0] == leaf)
42675f39d397SChris Mason 					btrfs_mark_buffer_dirty(leaf);
42685f39d397SChris Mason 				free_extent_buffer(leaf);
4269be0e5c09SChris Mason 			}
4270d5719762SChris Mason 		} else {
42715f39d397SChris Mason 			btrfs_mark_buffer_dirty(leaf);
4272be0e5c09SChris Mason 		}
42739a8dd150SChris Mason 	}
4274aa5d6bedSChris Mason 	return ret;
42759a8dd150SChris Mason }
42769a8dd150SChris Mason 
427797571fd0SChris Mason /*
4278925baeddSChris Mason  * search the tree again to find a leaf with lesser keys
42797bb86316SChris Mason  * returns 0 if it found something or 1 if there are no lesser leaves.
42807bb86316SChris Mason  * returns < 0 on io errors.
4281d352ac68SChris Mason  *
4282d352ac68SChris Mason  * This may release the path, and so you may lose any locks held at the
4283d352ac68SChris Mason  * time you call it.
42847bb86316SChris Mason  */
428516e7549fSJosef Bacik int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path)
42867bb86316SChris Mason {
4287925baeddSChris Mason 	struct btrfs_key key;
4288925baeddSChris Mason 	struct btrfs_disk_key found_key;
4289925baeddSChris Mason 	int ret;
42907bb86316SChris Mason 
4291925baeddSChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, 0);
4292925baeddSChris Mason 
4293e8b0d724SFilipe David Borba Manana 	if (key.offset > 0) {
4294925baeddSChris Mason 		key.offset--;
4295e8b0d724SFilipe David Borba Manana 	} else if (key.type > 0) {
4296925baeddSChris Mason 		key.type--;
4297e8b0d724SFilipe David Borba Manana 		key.offset = (u64)-1;
4298e8b0d724SFilipe David Borba Manana 	} else if (key.objectid > 0) {
4299925baeddSChris Mason 		key.objectid--;
4300e8b0d724SFilipe David Borba Manana 		key.type = (u8)-1;
4301e8b0d724SFilipe David Borba Manana 		key.offset = (u64)-1;
4302e8b0d724SFilipe David Borba Manana 	} else {
43037bb86316SChris Mason 		return 1;
4304e8b0d724SFilipe David Borba Manana 	}
43057bb86316SChris Mason 
4306b3b4aa74SDavid Sterba 	btrfs_release_path(path);
4307925baeddSChris Mason 	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
4308925baeddSChris Mason 	if (ret < 0)
4309925baeddSChris Mason 		return ret;
4310925baeddSChris Mason 	btrfs_item_key(path->nodes[0], &found_key, 0);
4311925baeddSChris Mason 	ret = comp_keys(&found_key, &key);
4312337c6f68SFilipe Manana 	/*
4313337c6f68SFilipe Manana 	 * We might have had an item with the previous key in the tree right
4314337c6f68SFilipe Manana 	 * before we released our path. And after we released our path, that
4315337c6f68SFilipe Manana 	 * item might have been pushed to the first slot (0) of the leaf we
4316337c6f68SFilipe Manana 	 * were holding due to a tree balance. Alternatively, an item with the
4317337c6f68SFilipe Manana 	 * previous key can exist as the only element of a leaf (big fat item).
4318337c6f68SFilipe Manana 	 * Therefore account for these 2 cases, so that our callers (like
4319337c6f68SFilipe Manana 	 * btrfs_previous_item) don't miss an existing item with a key matching
4320337c6f68SFilipe Manana 	 * the previous key we computed above.
4321337c6f68SFilipe Manana 	 */
4322337c6f68SFilipe Manana 	if (ret <= 0)
43237bb86316SChris Mason 		return 0;
4324925baeddSChris Mason 	return 1;
43257bb86316SChris Mason }
43267bb86316SChris Mason 
43273f157a2fSChris Mason /*
43283f157a2fSChris Mason  * A helper function to walk down the tree starting at min_key, and looking
4329de78b51aSEric Sandeen  * for nodes or leaves that are have a minimum transaction id.
4330de78b51aSEric Sandeen  * This is used by the btree defrag code, and tree logging
43313f157a2fSChris Mason  *
43323f157a2fSChris Mason  * This does not cow, but it does stuff the starting key it finds back
43333f157a2fSChris Mason  * into min_key, so you can call btrfs_search_slot with cow=1 on the
43343f157a2fSChris Mason  * key and get a writable path.
43353f157a2fSChris Mason  *
43363f157a2fSChris Mason  * This honors path->lowest_level to prevent descent past a given level
43373f157a2fSChris Mason  * of the tree.
43383f157a2fSChris Mason  *
4339d352ac68SChris Mason  * min_trans indicates the oldest transaction that you are interested
4340d352ac68SChris Mason  * in walking through.  Any nodes or leaves older than min_trans are
4341d352ac68SChris Mason  * skipped over (without reading them).
4342d352ac68SChris Mason  *
43433f157a2fSChris Mason  * returns zero if something useful was found, < 0 on error and 1 if there
43443f157a2fSChris Mason  * was nothing in the tree that matched the search criteria.
43453f157a2fSChris Mason  */
43463f157a2fSChris Mason int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key,
4347de78b51aSEric Sandeen 			 struct btrfs_path *path,
43483f157a2fSChris Mason 			 u64 min_trans)
43493f157a2fSChris Mason {
43503f157a2fSChris Mason 	struct extent_buffer *cur;
43513f157a2fSChris Mason 	struct btrfs_key found_key;
43523f157a2fSChris Mason 	int slot;
43539652480bSYan 	int sret;
43543f157a2fSChris Mason 	u32 nritems;
43553f157a2fSChris Mason 	int level;
43563f157a2fSChris Mason 	int ret = 1;
4357f98de9b9SFilipe Manana 	int keep_locks = path->keep_locks;
43583f157a2fSChris Mason 
4359f98de9b9SFilipe Manana 	path->keep_locks = 1;
43603f157a2fSChris Mason again:
4361bd681513SChris Mason 	cur = btrfs_read_lock_root_node(root);
43623f157a2fSChris Mason 	level = btrfs_header_level(cur);
4363e02119d5SChris Mason 	WARN_ON(path->nodes[level]);
43643f157a2fSChris Mason 	path->nodes[level] = cur;
4365bd681513SChris Mason 	path->locks[level] = BTRFS_READ_LOCK;
43663f157a2fSChris Mason 
43673f157a2fSChris Mason 	if (btrfs_header_generation(cur) < min_trans) {
43683f157a2fSChris Mason 		ret = 1;
43693f157a2fSChris Mason 		goto out;
43703f157a2fSChris Mason 	}
43713f157a2fSChris Mason 	while (1) {
43723f157a2fSChris Mason 		nritems = btrfs_header_nritems(cur);
43733f157a2fSChris Mason 		level = btrfs_header_level(cur);
4374e3b83361SQu Wenruo 		sret = btrfs_bin_search(cur, min_key, &slot);
4375cbca7d59SFilipe Manana 		if (sret < 0) {
4376cbca7d59SFilipe Manana 			ret = sret;
4377cbca7d59SFilipe Manana 			goto out;
4378cbca7d59SFilipe Manana 		}
43793f157a2fSChris Mason 
4380323ac95bSChris Mason 		/* at the lowest level, we're done, setup the path and exit */
4381323ac95bSChris Mason 		if (level == path->lowest_level) {
4382e02119d5SChris Mason 			if (slot >= nritems)
4383e02119d5SChris Mason 				goto find_next_key;
43843f157a2fSChris Mason 			ret = 0;
43853f157a2fSChris Mason 			path->slots[level] = slot;
43863f157a2fSChris Mason 			btrfs_item_key_to_cpu(cur, &found_key, slot);
43873f157a2fSChris Mason 			goto out;
43883f157a2fSChris Mason 		}
43899652480bSYan 		if (sret && slot > 0)
43909652480bSYan 			slot--;
43913f157a2fSChris Mason 		/*
4392de78b51aSEric Sandeen 		 * check this node pointer against the min_trans parameters.
4393260db43cSRandy Dunlap 		 * If it is too old, skip to the next one.
43943f157a2fSChris Mason 		 */
43953f157a2fSChris Mason 		while (slot < nritems) {
43963f157a2fSChris Mason 			u64 gen;
4397e02119d5SChris Mason 
43983f157a2fSChris Mason 			gen = btrfs_node_ptr_generation(cur, slot);
43993f157a2fSChris Mason 			if (gen < min_trans) {
44003f157a2fSChris Mason 				slot++;
44013f157a2fSChris Mason 				continue;
44023f157a2fSChris Mason 			}
44033f157a2fSChris Mason 			break;
44043f157a2fSChris Mason 		}
4405e02119d5SChris Mason find_next_key:
44063f157a2fSChris Mason 		/*
44073f157a2fSChris Mason 		 * we didn't find a candidate key in this node, walk forward
44083f157a2fSChris Mason 		 * and find another one
44093f157a2fSChris Mason 		 */
44103f157a2fSChris Mason 		if (slot >= nritems) {
4411e02119d5SChris Mason 			path->slots[level] = slot;
4412e02119d5SChris Mason 			sret = btrfs_find_next_key(root, path, min_key, level,
4413de78b51aSEric Sandeen 						  min_trans);
4414e02119d5SChris Mason 			if (sret == 0) {
4415b3b4aa74SDavid Sterba 				btrfs_release_path(path);
44163f157a2fSChris Mason 				goto again;
44173f157a2fSChris Mason 			} else {
44183f157a2fSChris Mason 				goto out;
44193f157a2fSChris Mason 			}
44203f157a2fSChris Mason 		}
44213f157a2fSChris Mason 		/* save our key for returning back */
44223f157a2fSChris Mason 		btrfs_node_key_to_cpu(cur, &found_key, slot);
44233f157a2fSChris Mason 		path->slots[level] = slot;
44243f157a2fSChris Mason 		if (level == path->lowest_level) {
44253f157a2fSChris Mason 			ret = 0;
44263f157a2fSChris Mason 			goto out;
44273f157a2fSChris Mason 		}
44284b231ae4SDavid Sterba 		cur = btrfs_read_node_slot(cur, slot);
4429fb770ae4SLiu Bo 		if (IS_ERR(cur)) {
4430fb770ae4SLiu Bo 			ret = PTR_ERR(cur);
4431fb770ae4SLiu Bo 			goto out;
4432fb770ae4SLiu Bo 		}
44333f157a2fSChris Mason 
4434bd681513SChris Mason 		btrfs_tree_read_lock(cur);
4435b4ce94deSChris Mason 
4436bd681513SChris Mason 		path->locks[level - 1] = BTRFS_READ_LOCK;
44373f157a2fSChris Mason 		path->nodes[level - 1] = cur;
4438f7c79f30SChris Mason 		unlock_up(path, level, 1, 0, NULL);
44393f157a2fSChris Mason 	}
44403f157a2fSChris Mason out:
4441f98de9b9SFilipe Manana 	path->keep_locks = keep_locks;
4442f98de9b9SFilipe Manana 	if (ret == 0) {
4443f98de9b9SFilipe Manana 		btrfs_unlock_up_safe(path, path->lowest_level + 1);
4444f98de9b9SFilipe Manana 		memcpy(min_key, &found_key, sizeof(found_key));
4445f98de9b9SFilipe Manana 	}
44463f157a2fSChris Mason 	return ret;
44473f157a2fSChris Mason }
44483f157a2fSChris Mason 
44493f157a2fSChris Mason /*
44503f157a2fSChris Mason  * this is similar to btrfs_next_leaf, but does not try to preserve
44513f157a2fSChris Mason  * and fixup the path.  It looks for and returns the next key in the
4452de78b51aSEric Sandeen  * tree based on the current path and the min_trans parameters.
44533f157a2fSChris Mason  *
44543f157a2fSChris Mason  * 0 is returned if another key is found, < 0 if there are any errors
44553f157a2fSChris Mason  * and 1 is returned if there are no higher keys in the tree
44563f157a2fSChris Mason  *
44573f157a2fSChris Mason  * path->keep_locks should be set to 1 on the search made before
44583f157a2fSChris Mason  * calling this function.
44593f157a2fSChris Mason  */
4460e7a84565SChris Mason int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path,
4461de78b51aSEric Sandeen 			struct btrfs_key *key, int level, u64 min_trans)
4462e7a84565SChris Mason {
4463e7a84565SChris Mason 	int slot;
4464e7a84565SChris Mason 	struct extent_buffer *c;
4465e7a84565SChris Mason 
44666a9fb468SJosef Bacik 	WARN_ON(!path->keep_locks && !path->skip_locking);
4467e7a84565SChris Mason 	while (level < BTRFS_MAX_LEVEL) {
4468e7a84565SChris Mason 		if (!path->nodes[level])
4469e7a84565SChris Mason 			return 1;
4470e7a84565SChris Mason 
4471e7a84565SChris Mason 		slot = path->slots[level] + 1;
4472e7a84565SChris Mason 		c = path->nodes[level];
44733f157a2fSChris Mason next:
4474e7a84565SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
447533c66f43SYan Zheng 			int ret;
447633c66f43SYan Zheng 			int orig_lowest;
447733c66f43SYan Zheng 			struct btrfs_key cur_key;
447833c66f43SYan Zheng 			if (level + 1 >= BTRFS_MAX_LEVEL ||
447933c66f43SYan Zheng 			    !path->nodes[level + 1])
4480e7a84565SChris Mason 				return 1;
448133c66f43SYan Zheng 
44826a9fb468SJosef Bacik 			if (path->locks[level + 1] || path->skip_locking) {
448333c66f43SYan Zheng 				level++;
4484e7a84565SChris Mason 				continue;
4485e7a84565SChris Mason 			}
448633c66f43SYan Zheng 
448733c66f43SYan Zheng 			slot = btrfs_header_nritems(c) - 1;
448833c66f43SYan Zheng 			if (level == 0)
448933c66f43SYan Zheng 				btrfs_item_key_to_cpu(c, &cur_key, slot);
449033c66f43SYan Zheng 			else
449133c66f43SYan Zheng 				btrfs_node_key_to_cpu(c, &cur_key, slot);
449233c66f43SYan Zheng 
449333c66f43SYan Zheng 			orig_lowest = path->lowest_level;
4494b3b4aa74SDavid Sterba 			btrfs_release_path(path);
449533c66f43SYan Zheng 			path->lowest_level = level;
449633c66f43SYan Zheng 			ret = btrfs_search_slot(NULL, root, &cur_key, path,
449733c66f43SYan Zheng 						0, 0);
449833c66f43SYan Zheng 			path->lowest_level = orig_lowest;
449933c66f43SYan Zheng 			if (ret < 0)
450033c66f43SYan Zheng 				return ret;
450133c66f43SYan Zheng 
450233c66f43SYan Zheng 			c = path->nodes[level];
450333c66f43SYan Zheng 			slot = path->slots[level];
450433c66f43SYan Zheng 			if (ret == 0)
450533c66f43SYan Zheng 				slot++;
450633c66f43SYan Zheng 			goto next;
450733c66f43SYan Zheng 		}
450833c66f43SYan Zheng 
4509e7a84565SChris Mason 		if (level == 0)
4510e7a84565SChris Mason 			btrfs_item_key_to_cpu(c, key, slot);
45113f157a2fSChris Mason 		else {
45123f157a2fSChris Mason 			u64 gen = btrfs_node_ptr_generation(c, slot);
45133f157a2fSChris Mason 
45143f157a2fSChris Mason 			if (gen < min_trans) {
45153f157a2fSChris Mason 				slot++;
45163f157a2fSChris Mason 				goto next;
45173f157a2fSChris Mason 			}
4518e7a84565SChris Mason 			btrfs_node_key_to_cpu(c, key, slot);
45193f157a2fSChris Mason 		}
4520e7a84565SChris Mason 		return 0;
4521e7a84565SChris Mason 	}
4522e7a84565SChris Mason 	return 1;
4523e7a84565SChris Mason }
4524e7a84565SChris Mason 
45253d7806ecSJan Schmidt int btrfs_next_old_leaf(struct btrfs_root *root, struct btrfs_path *path,
45263d7806ecSJan Schmidt 			u64 time_seq)
45273d7806ecSJan Schmidt {
4528d97e63b6SChris Mason 	int slot;
45298e73f275SChris Mason 	int level;
45305f39d397SChris Mason 	struct extent_buffer *c;
45318e73f275SChris Mason 	struct extent_buffer *next;
4532d96b3424SFilipe Manana 	struct btrfs_fs_info *fs_info = root->fs_info;
4533925baeddSChris Mason 	struct btrfs_key key;
4534d96b3424SFilipe Manana 	bool need_commit_sem = false;
4535925baeddSChris Mason 	u32 nritems;
4536925baeddSChris Mason 	int ret;
45370e46318dSJosef Bacik 	int i;
4538925baeddSChris Mason 
4539925baeddSChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
4540d397712bSChris Mason 	if (nritems == 0)
4541925baeddSChris Mason 		return 1;
4542925baeddSChris Mason 
45438e73f275SChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1);
45448e73f275SChris Mason again:
45458e73f275SChris Mason 	level = 1;
45468e73f275SChris Mason 	next = NULL;
4547b3b4aa74SDavid Sterba 	btrfs_release_path(path);
45488e73f275SChris Mason 
4549a2135011SChris Mason 	path->keep_locks = 1;
45508e73f275SChris Mason 
4551d96b3424SFilipe Manana 	if (time_seq) {
45523d7806ecSJan Schmidt 		ret = btrfs_search_old_slot(root, &key, path, time_seq);
4553d96b3424SFilipe Manana 	} else {
4554d96b3424SFilipe Manana 		if (path->need_commit_sem) {
4555d96b3424SFilipe Manana 			path->need_commit_sem = 0;
4556d96b3424SFilipe Manana 			need_commit_sem = true;
4557d96b3424SFilipe Manana 			down_read(&fs_info->commit_root_sem);
4558d96b3424SFilipe Manana 		}
4559925baeddSChris Mason 		ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
4560d96b3424SFilipe Manana 	}
4561925baeddSChris Mason 	path->keep_locks = 0;
4562925baeddSChris Mason 
4563925baeddSChris Mason 	if (ret < 0)
4564d96b3424SFilipe Manana 		goto done;
4565925baeddSChris Mason 
4566a2135011SChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
4567168fd7d2SChris Mason 	/*
4568168fd7d2SChris Mason 	 * by releasing the path above we dropped all our locks.  A balance
4569168fd7d2SChris Mason 	 * could have added more items next to the key that used to be
4570168fd7d2SChris Mason 	 * at the very end of the block.  So, check again here and
4571168fd7d2SChris Mason 	 * advance the path if there are now more items available.
4572168fd7d2SChris Mason 	 */
4573a2135011SChris Mason 	if (nritems > 0 && path->slots[0] < nritems - 1) {
4574e457afecSYan Zheng 		if (ret == 0)
4575168fd7d2SChris Mason 			path->slots[0]++;
45768e73f275SChris Mason 		ret = 0;
4577925baeddSChris Mason 		goto done;
4578925baeddSChris Mason 	}
45790b43e04fSLiu Bo 	/*
45800b43e04fSLiu Bo 	 * So the above check misses one case:
45810b43e04fSLiu Bo 	 * - after releasing the path above, someone has removed the item that
45820b43e04fSLiu Bo 	 *   used to be at the very end of the block, and balance between leafs
45830b43e04fSLiu Bo 	 *   gets another one with bigger key.offset to replace it.
45840b43e04fSLiu Bo 	 *
45850b43e04fSLiu Bo 	 * This one should be returned as well, or we can get leaf corruption
45860b43e04fSLiu Bo 	 * later(esp. in __btrfs_drop_extents()).
45870b43e04fSLiu Bo 	 *
45880b43e04fSLiu Bo 	 * And a bit more explanation about this check,
45890b43e04fSLiu Bo 	 * with ret > 0, the key isn't found, the path points to the slot
45900b43e04fSLiu Bo 	 * where it should be inserted, so the path->slots[0] item must be the
45910b43e04fSLiu Bo 	 * bigger one.
45920b43e04fSLiu Bo 	 */
45930b43e04fSLiu Bo 	if (nritems > 0 && ret > 0 && path->slots[0] == nritems - 1) {
45940b43e04fSLiu Bo 		ret = 0;
45950b43e04fSLiu Bo 		goto done;
45960b43e04fSLiu Bo 	}
4597d97e63b6SChris Mason 
4598234b63a0SChris Mason 	while (level < BTRFS_MAX_LEVEL) {
45998e73f275SChris Mason 		if (!path->nodes[level]) {
46008e73f275SChris Mason 			ret = 1;
46018e73f275SChris Mason 			goto done;
46028e73f275SChris Mason 		}
46035f39d397SChris Mason 
4604d97e63b6SChris Mason 		slot = path->slots[level] + 1;
4605d97e63b6SChris Mason 		c = path->nodes[level];
46065f39d397SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
4607d97e63b6SChris Mason 			level++;
46088e73f275SChris Mason 			if (level == BTRFS_MAX_LEVEL) {
46098e73f275SChris Mason 				ret = 1;
46108e73f275SChris Mason 				goto done;
46118e73f275SChris Mason 			}
4612d97e63b6SChris Mason 			continue;
4613d97e63b6SChris Mason 		}
46145f39d397SChris Mason 
46150e46318dSJosef Bacik 
46160e46318dSJosef Bacik 		/*
46170e46318dSJosef Bacik 		 * Our current level is where we're going to start from, and to
46180e46318dSJosef Bacik 		 * make sure lockdep doesn't complain we need to drop our locks
46190e46318dSJosef Bacik 		 * and nodes from 0 to our current level.
46200e46318dSJosef Bacik 		 */
46210e46318dSJosef Bacik 		for (i = 0; i < level; i++) {
46220e46318dSJosef Bacik 			if (path->locks[level]) {
46230e46318dSJosef Bacik 				btrfs_tree_read_unlock(path->nodes[i]);
46240e46318dSJosef Bacik 				path->locks[i] = 0;
46250e46318dSJosef Bacik 			}
46260e46318dSJosef Bacik 			free_extent_buffer(path->nodes[i]);
46270e46318dSJosef Bacik 			path->nodes[i] = NULL;
4628925baeddSChris Mason 		}
46295f39d397SChris Mason 
46308e73f275SChris Mason 		next = c;
4631d07b8528SLiu Bo 		ret = read_block_for_search(root, path, &next, level,
4632cda79c54SDavid Sterba 					    slot, &key);
46338e73f275SChris Mason 		if (ret == -EAGAIN)
46348e73f275SChris Mason 			goto again;
46355f39d397SChris Mason 
463676a05b35SChris Mason 		if (ret < 0) {
4637b3b4aa74SDavid Sterba 			btrfs_release_path(path);
463876a05b35SChris Mason 			goto done;
463976a05b35SChris Mason 		}
464076a05b35SChris Mason 
46415cd57b2cSChris Mason 		if (!path->skip_locking) {
4642bd681513SChris Mason 			ret = btrfs_try_tree_read_lock(next);
4643d42244a0SJan Schmidt 			if (!ret && time_seq) {
4644d42244a0SJan Schmidt 				/*
4645d42244a0SJan Schmidt 				 * If we don't get the lock, we may be racing
4646d42244a0SJan Schmidt 				 * with push_leaf_left, holding that lock while
4647d42244a0SJan Schmidt 				 * itself waiting for the leaf we've currently
4648d42244a0SJan Schmidt 				 * locked. To solve this situation, we give up
4649d42244a0SJan Schmidt 				 * on our lock and cycle.
4650d42244a0SJan Schmidt 				 */
4651cf538830SJan Schmidt 				free_extent_buffer(next);
4652d42244a0SJan Schmidt 				btrfs_release_path(path);
4653d42244a0SJan Schmidt 				cond_resched();
4654d42244a0SJan Schmidt 				goto again;
4655d42244a0SJan Schmidt 			}
46560e46318dSJosef Bacik 			if (!ret)
46570e46318dSJosef Bacik 				btrfs_tree_read_lock(next);
4658bd681513SChris Mason 		}
4659d97e63b6SChris Mason 		break;
4660d97e63b6SChris Mason 	}
4661d97e63b6SChris Mason 	path->slots[level] = slot;
4662d97e63b6SChris Mason 	while (1) {
4663d97e63b6SChris Mason 		level--;
4664d97e63b6SChris Mason 		path->nodes[level] = next;
4665d97e63b6SChris Mason 		path->slots[level] = 0;
4666a74a4b97SChris Mason 		if (!path->skip_locking)
4667ffeb03cfSJosef Bacik 			path->locks[level] = BTRFS_READ_LOCK;
4668d97e63b6SChris Mason 		if (!level)
4669d97e63b6SChris Mason 			break;
4670b4ce94deSChris Mason 
4671d07b8528SLiu Bo 		ret = read_block_for_search(root, path, &next, level,
4672cda79c54SDavid Sterba 					    0, &key);
46738e73f275SChris Mason 		if (ret == -EAGAIN)
46748e73f275SChris Mason 			goto again;
46758e73f275SChris Mason 
467676a05b35SChris Mason 		if (ret < 0) {
4677b3b4aa74SDavid Sterba 			btrfs_release_path(path);
467876a05b35SChris Mason 			goto done;
467976a05b35SChris Mason 		}
468076a05b35SChris Mason 
4681ffeb03cfSJosef Bacik 		if (!path->skip_locking)
46820e46318dSJosef Bacik 			btrfs_tree_read_lock(next);
4683d97e63b6SChris Mason 	}
46848e73f275SChris Mason 	ret = 0;
4685925baeddSChris Mason done:
4686f7c79f30SChris Mason 	unlock_up(path, 0, 1, 0, NULL);
4687d96b3424SFilipe Manana 	if (need_commit_sem) {
4688d96b3424SFilipe Manana 		int ret2;
4689d96b3424SFilipe Manana 
4690d96b3424SFilipe Manana 		path->need_commit_sem = 1;
4691d96b3424SFilipe Manana 		ret2 = finish_need_commit_sem_search(path);
4692d96b3424SFilipe Manana 		up_read(&fs_info->commit_root_sem);
4693d96b3424SFilipe Manana 		if (ret2)
4694d96b3424SFilipe Manana 			ret = ret2;
4695d96b3424SFilipe Manana 	}
46968e73f275SChris Mason 
46978e73f275SChris Mason 	return ret;
4698d97e63b6SChris Mason }
46990b86a832SChris Mason 
47003f157a2fSChris Mason /*
47013f157a2fSChris Mason  * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps
47023f157a2fSChris Mason  * searching until it gets past min_objectid or finds an item of 'type'
47033f157a2fSChris Mason  *
47043f157a2fSChris Mason  * returns 0 if something is found, 1 if nothing was found and < 0 on error
47053f157a2fSChris Mason  */
47060b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root,
47070b86a832SChris Mason 			struct btrfs_path *path, u64 min_objectid,
47080b86a832SChris Mason 			int type)
47090b86a832SChris Mason {
47100b86a832SChris Mason 	struct btrfs_key found_key;
47110b86a832SChris Mason 	struct extent_buffer *leaf;
4712e02119d5SChris Mason 	u32 nritems;
47130b86a832SChris Mason 	int ret;
47140b86a832SChris Mason 
47150b86a832SChris Mason 	while (1) {
47160b86a832SChris Mason 		if (path->slots[0] == 0) {
47170b86a832SChris Mason 			ret = btrfs_prev_leaf(root, path);
47180b86a832SChris Mason 			if (ret != 0)
47190b86a832SChris Mason 				return ret;
47200b86a832SChris Mason 		} else {
47210b86a832SChris Mason 			path->slots[0]--;
47220b86a832SChris Mason 		}
47230b86a832SChris Mason 		leaf = path->nodes[0];
4724e02119d5SChris Mason 		nritems = btrfs_header_nritems(leaf);
4725e02119d5SChris Mason 		if (nritems == 0)
4726e02119d5SChris Mason 			return 1;
4727e02119d5SChris Mason 		if (path->slots[0] == nritems)
4728e02119d5SChris Mason 			path->slots[0]--;
4729e02119d5SChris Mason 
47300b86a832SChris Mason 		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
4731e02119d5SChris Mason 		if (found_key.objectid < min_objectid)
4732e02119d5SChris Mason 			break;
47330a4eefbbSYan Zheng 		if (found_key.type == type)
47340a4eefbbSYan Zheng 			return 0;
4735e02119d5SChris Mason 		if (found_key.objectid == min_objectid &&
4736e02119d5SChris Mason 		    found_key.type < type)
4737e02119d5SChris Mason 			break;
47380b86a832SChris Mason 	}
47390b86a832SChris Mason 	return 1;
47400b86a832SChris Mason }
4741ade2e0b3SWang Shilong 
4742ade2e0b3SWang Shilong /*
4743ade2e0b3SWang Shilong  * search in extent tree to find a previous Metadata/Data extent item with
4744ade2e0b3SWang Shilong  * min objecitd.
4745ade2e0b3SWang Shilong  *
4746ade2e0b3SWang Shilong  * returns 0 if something is found, 1 if nothing was found and < 0 on error
4747ade2e0b3SWang Shilong  */
4748ade2e0b3SWang Shilong int btrfs_previous_extent_item(struct btrfs_root *root,
4749ade2e0b3SWang Shilong 			struct btrfs_path *path, u64 min_objectid)
4750ade2e0b3SWang Shilong {
4751ade2e0b3SWang Shilong 	struct btrfs_key found_key;
4752ade2e0b3SWang Shilong 	struct extent_buffer *leaf;
4753ade2e0b3SWang Shilong 	u32 nritems;
4754ade2e0b3SWang Shilong 	int ret;
4755ade2e0b3SWang Shilong 
4756ade2e0b3SWang Shilong 	while (1) {
4757ade2e0b3SWang Shilong 		if (path->slots[0] == 0) {
4758ade2e0b3SWang Shilong 			ret = btrfs_prev_leaf(root, path);
4759ade2e0b3SWang Shilong 			if (ret != 0)
4760ade2e0b3SWang Shilong 				return ret;
4761ade2e0b3SWang Shilong 		} else {
4762ade2e0b3SWang Shilong 			path->slots[0]--;
4763ade2e0b3SWang Shilong 		}
4764ade2e0b3SWang Shilong 		leaf = path->nodes[0];
4765ade2e0b3SWang Shilong 		nritems = btrfs_header_nritems(leaf);
4766ade2e0b3SWang Shilong 		if (nritems == 0)
4767ade2e0b3SWang Shilong 			return 1;
4768ade2e0b3SWang Shilong 		if (path->slots[0] == nritems)
4769ade2e0b3SWang Shilong 			path->slots[0]--;
4770ade2e0b3SWang Shilong 
4771ade2e0b3SWang Shilong 		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
4772ade2e0b3SWang Shilong 		if (found_key.objectid < min_objectid)
4773ade2e0b3SWang Shilong 			break;
4774ade2e0b3SWang Shilong 		if (found_key.type == BTRFS_EXTENT_ITEM_KEY ||
4775ade2e0b3SWang Shilong 		    found_key.type == BTRFS_METADATA_ITEM_KEY)
4776ade2e0b3SWang Shilong 			return 0;
4777ade2e0b3SWang Shilong 		if (found_key.objectid == min_objectid &&
4778ade2e0b3SWang Shilong 		    found_key.type < BTRFS_EXTENT_ITEM_KEY)
4779ade2e0b3SWang Shilong 			break;
4780ade2e0b3SWang Shilong 	}
4781ade2e0b3SWang Shilong 	return 1;
4782ade2e0b3SWang Shilong }
4783