xref: /openbmc/linux/fs/btrfs/ctree.c (revision e5e1c1741b3de3f8d06fe4b700d83709a7da0610)
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 
1704c8c42864SChris Mason /*
17054271eceaSNikolay Borisov  * btrfs_search_slot - look for a key in a tree and perform necessary
17064271eceaSNikolay Borisov  * modifications to preserve tree invariants.
170774123bd7SChris Mason  *
17084271eceaSNikolay Borisov  * @trans:	Handle of transaction, used when modifying the tree
17094271eceaSNikolay Borisov  * @p:		Holds all btree nodes along the search path
17104271eceaSNikolay Borisov  * @root:	The root node of the tree
17114271eceaSNikolay Borisov  * @key:	The key we are looking for
17129a664971Sethanwu  * @ins_len:	Indicates purpose of search:
17139a664971Sethanwu  *              >0  for inserts it's size of item inserted (*)
17149a664971Sethanwu  *              <0  for deletions
17159a664971Sethanwu  *               0  for plain searches, not modifying the tree
17169a664971Sethanwu  *
17179a664971Sethanwu  *              (*) If size of item inserted doesn't include
17189a664971Sethanwu  *              sizeof(struct btrfs_item), then p->search_for_extension must
17199a664971Sethanwu  *              be set.
17204271eceaSNikolay Borisov  * @cow:	boolean should CoW operations be performed. Must always be 1
17214271eceaSNikolay Borisov  *		when modifying the tree.
172297571fd0SChris Mason  *
17234271eceaSNikolay Borisov  * If @ins_len > 0, nodes and leaves will be split as we walk down the tree.
17244271eceaSNikolay Borisov  * If @ins_len < 0, nodes will be merged as we walk down the tree (if possible)
17254271eceaSNikolay Borisov  *
17264271eceaSNikolay Borisov  * If @key is found, 0 is returned and you can find the item in the leaf level
17274271eceaSNikolay Borisov  * of the path (level 0)
17284271eceaSNikolay Borisov  *
17294271eceaSNikolay Borisov  * If @key isn't found, 1 is returned and the leaf level of the path (level 0)
17304271eceaSNikolay Borisov  * points to the slot where it should be inserted
17314271eceaSNikolay Borisov  *
17324271eceaSNikolay Borisov  * If an error is encountered while searching the tree a negative error number
17334271eceaSNikolay Borisov  * is returned
173474123bd7SChris Mason  */
1735310712b2SOmar Sandoval int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root *root,
1736310712b2SOmar Sandoval 		      const struct btrfs_key *key, struct btrfs_path *p,
1737310712b2SOmar Sandoval 		      int ins_len, int cow)
1738be0e5c09SChris Mason {
1739d96b3424SFilipe Manana 	struct btrfs_fs_info *fs_info = root->fs_info;
17405f39d397SChris Mason 	struct extent_buffer *b;
1741be0e5c09SChris Mason 	int slot;
1742be0e5c09SChris Mason 	int ret;
174333c66f43SYan Zheng 	int err;
1744be0e5c09SChris Mason 	int level;
1745925baeddSChris Mason 	int lowest_unlock = 1;
1746bd681513SChris Mason 	/* everything at write_lock_level or lower must be write locked */
1747bd681513SChris Mason 	int write_lock_level = 0;
17489f3a7427SChris Mason 	u8 lowest_level = 0;
1749f7c79f30SChris Mason 	int min_write_lock_level;
1750d7396f07SFilipe David Borba Manana 	int prev_cmp;
17519f3a7427SChris Mason 
17526702ed49SChris Mason 	lowest_level = p->lowest_level;
1753323ac95bSChris Mason 	WARN_ON(lowest_level && ins_len > 0);
175422b0ebdaSChris Mason 	WARN_ON(p->nodes[0] != NULL);
1755eb653de1SFilipe David Borba Manana 	BUG_ON(!cow && ins_len);
175625179201SJosef Bacik 
1757bd681513SChris Mason 	if (ins_len < 0) {
1758925baeddSChris Mason 		lowest_unlock = 2;
175965b51a00SChris Mason 
1760bd681513SChris Mason 		/* when we are removing items, we might have to go up to level
1761bd681513SChris Mason 		 * two as we update tree pointers  Make sure we keep write
1762bd681513SChris Mason 		 * for those levels as well
1763bd681513SChris Mason 		 */
1764bd681513SChris Mason 		write_lock_level = 2;
1765bd681513SChris Mason 	} else if (ins_len > 0) {
1766bd681513SChris Mason 		/*
1767bd681513SChris Mason 		 * for inserting items, make sure we have a write lock on
1768bd681513SChris Mason 		 * level 1 so we can update keys
1769bd681513SChris Mason 		 */
1770bd681513SChris Mason 		write_lock_level = 1;
1771bd681513SChris Mason 	}
1772bd681513SChris Mason 
1773bd681513SChris Mason 	if (!cow)
1774bd681513SChris Mason 		write_lock_level = -1;
1775bd681513SChris Mason 
177609a2a8f9SJosef Bacik 	if (cow && (p->keep_locks || p->lowest_level))
1777bd681513SChris Mason 		write_lock_level = BTRFS_MAX_LEVEL;
1778bd681513SChris Mason 
1779f7c79f30SChris Mason 	min_write_lock_level = write_lock_level;
1780f7c79f30SChris Mason 
1781d96b3424SFilipe Manana 	if (p->need_commit_sem) {
1782d96b3424SFilipe Manana 		ASSERT(p->search_commit_root);
1783d96b3424SFilipe Manana 		down_read(&fs_info->commit_root_sem);
1784d96b3424SFilipe Manana 	}
1785d96b3424SFilipe Manana 
1786bb803951SChris Mason again:
1787d7396f07SFilipe David Borba Manana 	prev_cmp = -1;
17881fc28d8eSLiu Bo 	b = btrfs_search_slot_get_root(root, p, write_lock_level);
1789be6821f8SFilipe Manana 	if (IS_ERR(b)) {
1790be6821f8SFilipe Manana 		ret = PTR_ERR(b);
1791be6821f8SFilipe Manana 		goto done;
1792be6821f8SFilipe Manana 	}
1793925baeddSChris Mason 
1794eb60ceacSChris Mason 	while (b) {
1795f624d976SQu Wenruo 		int dec = 0;
1796f624d976SQu Wenruo 
17975f39d397SChris Mason 		level = btrfs_header_level(b);
179865b51a00SChris Mason 
179902217ed2SChris Mason 		if (cow) {
18009ea2c7c9SNikolay Borisov 			bool last_level = (level == (BTRFS_MAX_LEVEL - 1));
18019ea2c7c9SNikolay Borisov 
1802c8c42864SChris Mason 			/*
1803c8c42864SChris Mason 			 * if we don't really need to cow this block
1804c8c42864SChris Mason 			 * then we don't want to set the path blocking,
1805c8c42864SChris Mason 			 * so we test it here
1806c8c42864SChris Mason 			 */
18075963ffcaSJosef Bacik 			if (!should_cow_block(trans, root, b))
180865b51a00SChris Mason 				goto cow_done;
18095d4f98a2SYan Zheng 
1810bd681513SChris Mason 			/*
1811bd681513SChris Mason 			 * must have write locks on this node and the
1812bd681513SChris Mason 			 * parent
1813bd681513SChris Mason 			 */
18145124e00eSJosef Bacik 			if (level > write_lock_level ||
18155124e00eSJosef Bacik 			    (level + 1 > write_lock_level &&
18165124e00eSJosef Bacik 			    level + 1 < BTRFS_MAX_LEVEL &&
18175124e00eSJosef Bacik 			    p->nodes[level + 1])) {
1818bd681513SChris Mason 				write_lock_level = level + 1;
1819bd681513SChris Mason 				btrfs_release_path(p);
1820bd681513SChris Mason 				goto again;
1821bd681513SChris Mason 			}
1822bd681513SChris Mason 
18239ea2c7c9SNikolay Borisov 			if (last_level)
18249ea2c7c9SNikolay Borisov 				err = btrfs_cow_block(trans, root, b, NULL, 0,
18259631e4ccSJosef Bacik 						      &b,
18269631e4ccSJosef Bacik 						      BTRFS_NESTING_COW);
18279ea2c7c9SNikolay Borisov 			else
182833c66f43SYan Zheng 				err = btrfs_cow_block(trans, root, b,
1829e20d96d6SChris Mason 						      p->nodes[level + 1],
18309631e4ccSJosef Bacik 						      p->slots[level + 1], &b,
18319631e4ccSJosef Bacik 						      BTRFS_NESTING_COW);
183233c66f43SYan Zheng 			if (err) {
183333c66f43SYan Zheng 				ret = err;
183465b51a00SChris Mason 				goto done;
183554aa1f4dSChris Mason 			}
183602217ed2SChris Mason 		}
183765b51a00SChris Mason cow_done:
1838eb60ceacSChris Mason 		p->nodes[level] = b;
183952398340SLiu Bo 		/*
184052398340SLiu Bo 		 * Leave path with blocking locks to avoid massive
184152398340SLiu Bo 		 * lock context switch, this is made on purpose.
184252398340SLiu Bo 		 */
1843b4ce94deSChris Mason 
1844b4ce94deSChris Mason 		/*
1845b4ce94deSChris Mason 		 * we have a lock on b and as long as we aren't changing
1846b4ce94deSChris Mason 		 * the tree, there is no way to for the items in b to change.
1847b4ce94deSChris Mason 		 * It is safe to drop the lock on our parent before we
1848b4ce94deSChris Mason 		 * go through the expensive btree search on b.
1849b4ce94deSChris Mason 		 *
1850eb653de1SFilipe David Borba Manana 		 * If we're inserting or deleting (ins_len != 0), then we might
1851eb653de1SFilipe David Borba Manana 		 * be changing slot zero, which may require changing the parent.
1852eb653de1SFilipe David Borba Manana 		 * So, we can't drop the lock until after we know which slot
1853eb653de1SFilipe David Borba Manana 		 * we're operating on.
1854b4ce94deSChris Mason 		 */
1855eb653de1SFilipe David Borba Manana 		if (!ins_len && !p->keep_locks) {
1856eb653de1SFilipe David Borba Manana 			int u = level + 1;
1857eb653de1SFilipe David Borba Manana 
1858eb653de1SFilipe David Borba Manana 			if (u < BTRFS_MAX_LEVEL && p->locks[u]) {
1859eb653de1SFilipe David Borba Manana 				btrfs_tree_unlock_rw(p->nodes[u], p->locks[u]);
1860eb653de1SFilipe David Borba Manana 				p->locks[u] = 0;
1861eb653de1SFilipe David Borba Manana 			}
1862eb653de1SFilipe David Borba Manana 		}
1863b4ce94deSChris Mason 
1864e2e58d0fSFilipe Manana 		if (level == 0) {
1865e2e58d0fSFilipe Manana 			int leaf_free_space = 0;
1866e2e58d0fSFilipe Manana 			int search_low_slot = 0;
1867e2e58d0fSFilipe Manana 
1868995e9a16SNikolay Borisov 			/*
1869e2e58d0fSFilipe Manana 			 * If we are doing an insertion, the leaf has enough free
1870e2e58d0fSFilipe Manana 			 * space and the destination slot for the key is not slot
1871e2e58d0fSFilipe Manana 			 * 0, then we can unlock our write lock on the parent, and
1872e2e58d0fSFilipe Manana 			 * any other upper nodes, before doing the binary search
1873e2e58d0fSFilipe Manana 			 * on the leaf (with search_for_key_slot()), allowing other
1874e2e58d0fSFilipe Manana 			 * tasks to lock the parent and any other upper nodes.
1875995e9a16SNikolay Borisov 			 */
1876e2e58d0fSFilipe Manana 			if (ins_len > 0) {
1877e2e58d0fSFilipe Manana 				struct btrfs_disk_key first_key;
1878e2e58d0fSFilipe Manana 
1879e2e58d0fSFilipe Manana 				/*
1880e2e58d0fSFilipe Manana 				 * Cache the leaf free space, since we will need it
1881e2e58d0fSFilipe Manana 				 * later and it will not change until then.
1882e2e58d0fSFilipe Manana 				 */
1883e2e58d0fSFilipe Manana 				leaf_free_space = btrfs_leaf_free_space(b);
1884e2e58d0fSFilipe Manana 
1885e2e58d0fSFilipe Manana 				/*
1886e2e58d0fSFilipe Manana 				 * !p->locks[1] means we have a single node tree,
1887e2e58d0fSFilipe Manana 				 * the leaf is the root of the tree.
1888e2e58d0fSFilipe Manana 				 */
1889e2e58d0fSFilipe Manana 				if (!p->locks[1] || leaf_free_space < ins_len)
1890e2e58d0fSFilipe Manana 					goto leaf_search;
1891e2e58d0fSFilipe Manana 
1892e2e58d0fSFilipe Manana 				ASSERT(btrfs_header_nritems(b) > 0);
1893e2e58d0fSFilipe Manana 				btrfs_item_key(b, &first_key, 0);
1894e2e58d0fSFilipe Manana 
1895e2e58d0fSFilipe Manana 				/*
1896e2e58d0fSFilipe Manana 				 * Doing the extra comparison with the first key
1897e2e58d0fSFilipe Manana 				 * is cheap, taking into account that the first
1898e2e58d0fSFilipe Manana 				 * key is very likely already in a cache line
1899e2e58d0fSFilipe Manana 				 * because it immediately follows the extent
1900e2e58d0fSFilipe Manana 				 * buffer's header and we have recently accessed
1901e2e58d0fSFilipe Manana 				 * the header's level field.
1902e2e58d0fSFilipe Manana 				 */
1903e2e58d0fSFilipe Manana 				ret = comp_keys(&first_key, key);
1904e2e58d0fSFilipe Manana 				if (ret < 0) {
1905e2e58d0fSFilipe Manana 					/*
1906e2e58d0fSFilipe Manana 					 * The first key is smaller than the key
1907e2e58d0fSFilipe Manana 					 * we want to insert, so we are safe to
1908e2e58d0fSFilipe Manana 					 * unlock all upper nodes and we have to
1909e2e58d0fSFilipe Manana 					 * do the binary search.
1910e2e58d0fSFilipe Manana 					 *
1911e2e58d0fSFilipe Manana 					 * We do use btrfs_unlock_up_safe() and
1912e2e58d0fSFilipe Manana 					 * not unlock_up() because the later does
1913e2e58d0fSFilipe Manana 					 * not unlock nodes with a slot of 0.
1914e2e58d0fSFilipe Manana 					 * We can safely unlock any node even if
1915e2e58d0fSFilipe Manana 					 * its slot is 0 since in this case the
1916e2e58d0fSFilipe Manana 					 * key does not end up at slot 0 of the
1917e2e58d0fSFilipe Manana 					 * leaf and there's also no need to split
1918e2e58d0fSFilipe Manana 					 * the leaf.
1919e2e58d0fSFilipe Manana 					 */
1920e2e58d0fSFilipe Manana 					btrfs_unlock_up_safe(p, 1);
1921e2e58d0fSFilipe Manana 					search_low_slot = 1;
1922995e9a16SNikolay Borisov 				} else {
1923e2e58d0fSFilipe Manana 					/*
1924e2e58d0fSFilipe Manana 					 * The first key is >= then the key we
1925e2e58d0fSFilipe Manana 					 * want to insert, so we can skip the
1926e2e58d0fSFilipe Manana 					 * binary search as the target key will
1927e2e58d0fSFilipe Manana 					 * be at slot 0.
1928e2e58d0fSFilipe Manana 					 *
1929e2e58d0fSFilipe Manana 					 * We can not unlock upper nodes when
1930e2e58d0fSFilipe Manana 					 * the key is less than the first key,
1931e2e58d0fSFilipe Manana 					 * because we will need to update the key
1932e2e58d0fSFilipe Manana 					 * at slot 0 of the parent node and
1933e2e58d0fSFilipe Manana 					 * possibly of other upper nodes too.
1934e2e58d0fSFilipe Manana 					 * If the key matches the first key, then
1935e2e58d0fSFilipe Manana 					 * we can unlock all the upper nodes,
1936e2e58d0fSFilipe Manana 					 * using btrfs_unlock_up_safe() instead
1937e2e58d0fSFilipe Manana 					 * of unlock_up() as stated above.
1938e2e58d0fSFilipe Manana 					 */
1939e2e58d0fSFilipe Manana 					if (ret == 0)
1940e2e58d0fSFilipe Manana 						btrfs_unlock_up_safe(p, 1);
1941e2e58d0fSFilipe Manana 					slot = 0;
1942e2e58d0fSFilipe Manana 					/*
1943e2e58d0fSFilipe Manana 					 * ret is already 0 or 1, matching the
1944e2e58d0fSFilipe Manana 					 * result of a btrfs_bin_search() call,
1945e2e58d0fSFilipe Manana 					 * so there is no need to adjust it.
1946e2e58d0fSFilipe Manana 					 */
1947e2e58d0fSFilipe Manana 					goto skip_leaf_search;
1948e2e58d0fSFilipe Manana 				}
1949e2e58d0fSFilipe Manana 			}
1950e2e58d0fSFilipe Manana leaf_search:
1951e2e58d0fSFilipe Manana 			ret = search_for_key_slot(b, search_low_slot, key,
1952e2e58d0fSFilipe Manana 						  prev_cmp, &slot);
1953415b35a5SLiu Bo 			if (ret < 0)
1954415b35a5SLiu Bo 				goto done;
1955e2e58d0fSFilipe Manana skip_leaf_search:
1956be0e5c09SChris Mason 			p->slots[level] = slot;
19579a664971Sethanwu 			/*
19589a664971Sethanwu 			 * Item key already exists. In this case, if we are
19599a664971Sethanwu 			 * allowed to insert the item (for example, in dir_item
19609a664971Sethanwu 			 * case, item key collision is allowed), it will be
19619a664971Sethanwu 			 * merged with the original item. Only the item size
19629a664971Sethanwu 			 * grows, no new btrfs item will be added. If
19639a664971Sethanwu 			 * search_for_extension is not set, ins_len already
19649a664971Sethanwu 			 * accounts the size btrfs_item, deduct it here so leaf
19659a664971Sethanwu 			 * space check will be correct.
19669a664971Sethanwu 			 */
19679a664971Sethanwu 			if (ret == 0 && ins_len > 0 && !p->search_for_extension) {
19689a664971Sethanwu 				ASSERT(ins_len >= sizeof(struct btrfs_item));
19699a664971Sethanwu 				ins_len -= sizeof(struct btrfs_item);
19709a664971Sethanwu 			}
1971e2e58d0fSFilipe Manana 			if (ins_len > 0 && leaf_free_space < ins_len) {
1972*e5e1c174SFilipe Manana 				ASSERT(write_lock_level >= 1);
1973bd681513SChris Mason 
197433c66f43SYan Zheng 				err = split_leaf(trans, root, key,
1975cc0c5538SChris Mason 						 p, ins_len, ret == 0);
1976b4ce94deSChris Mason 
197733c66f43SYan Zheng 				BUG_ON(err > 0);
197833c66f43SYan Zheng 				if (err) {
197933c66f43SYan Zheng 					ret = err;
198065b51a00SChris Mason 					goto done;
198165b51a00SChris Mason 				}
19825c680ed6SChris Mason 			}
1983459931ecSChris Mason 			if (!p->search_for_split)
1984f7c79f30SChris Mason 				unlock_up(p, level, lowest_unlock,
19854b6f8e96SLiu Bo 					  min_write_lock_level, NULL);
198665b51a00SChris Mason 			goto done;
198765b51a00SChris Mason 		}
1988e2e58d0fSFilipe Manana 
1989e2e58d0fSFilipe Manana 		ret = search_for_key_slot(b, 0, key, prev_cmp, &slot);
1990e2e58d0fSFilipe Manana 		if (ret < 0)
1991e2e58d0fSFilipe Manana 			goto done;
1992e2e58d0fSFilipe Manana 		prev_cmp = ret;
1993e2e58d0fSFilipe Manana 
1994f624d976SQu Wenruo 		if (ret && slot > 0) {
1995f624d976SQu Wenruo 			dec = 1;
1996f624d976SQu Wenruo 			slot--;
1997f624d976SQu Wenruo 		}
1998f624d976SQu Wenruo 		p->slots[level] = slot;
1999f624d976SQu Wenruo 		err = setup_nodes_for_search(trans, root, p, b, level, ins_len,
2000f624d976SQu Wenruo 					     &write_lock_level);
2001f624d976SQu Wenruo 		if (err == -EAGAIN)
2002f624d976SQu Wenruo 			goto again;
2003f624d976SQu Wenruo 		if (err) {
2004f624d976SQu Wenruo 			ret = err;
2005f624d976SQu Wenruo 			goto done;
2006f624d976SQu Wenruo 		}
2007f624d976SQu Wenruo 		b = p->nodes[level];
2008f624d976SQu Wenruo 		slot = p->slots[level];
2009f624d976SQu Wenruo 
2010f624d976SQu Wenruo 		/*
2011f624d976SQu Wenruo 		 * Slot 0 is special, if we change the key we have to update
2012f624d976SQu Wenruo 		 * the parent pointer which means we must have a write lock on
2013f624d976SQu Wenruo 		 * the parent
2014f624d976SQu Wenruo 		 */
2015f624d976SQu Wenruo 		if (slot == 0 && ins_len && write_lock_level < level + 1) {
2016f624d976SQu Wenruo 			write_lock_level = level + 1;
2017f624d976SQu Wenruo 			btrfs_release_path(p);
2018f624d976SQu Wenruo 			goto again;
2019f624d976SQu Wenruo 		}
2020f624d976SQu Wenruo 
2021f624d976SQu Wenruo 		unlock_up(p, level, lowest_unlock, min_write_lock_level,
2022f624d976SQu Wenruo 			  &write_lock_level);
2023f624d976SQu Wenruo 
2024f624d976SQu Wenruo 		if (level == lowest_level) {
2025f624d976SQu Wenruo 			if (dec)
2026f624d976SQu Wenruo 				p->slots[level]++;
2027f624d976SQu Wenruo 			goto done;
2028f624d976SQu Wenruo 		}
2029f624d976SQu Wenruo 
2030f624d976SQu Wenruo 		err = read_block_for_search(root, p, &b, level, slot, key);
2031f624d976SQu Wenruo 		if (err == -EAGAIN)
2032f624d976SQu Wenruo 			goto again;
2033f624d976SQu Wenruo 		if (err) {
2034f624d976SQu Wenruo 			ret = err;
2035f624d976SQu Wenruo 			goto done;
2036f624d976SQu Wenruo 		}
2037f624d976SQu Wenruo 
2038f624d976SQu Wenruo 		if (!p->skip_locking) {
2039f624d976SQu Wenruo 			level = btrfs_header_level(b);
2040f624d976SQu Wenruo 			if (level <= write_lock_level) {
2041f624d976SQu Wenruo 				btrfs_tree_lock(b);
2042f624d976SQu Wenruo 				p->locks[level] = BTRFS_WRITE_LOCK;
2043f624d976SQu Wenruo 			} else {
2044fe596ca3SJosef Bacik 				btrfs_tree_read_lock(b);
2045f624d976SQu Wenruo 				p->locks[level] = BTRFS_READ_LOCK;
2046f624d976SQu Wenruo 			}
2047f624d976SQu Wenruo 			p->nodes[level] = b;
2048f624d976SQu Wenruo 		}
204965b51a00SChris Mason 	}
205065b51a00SChris Mason 	ret = 1;
205165b51a00SChris Mason done:
20525f5bc6b1SFilipe Manana 	if (ret < 0 && !p->skip_release_on_error)
2053b3b4aa74SDavid Sterba 		btrfs_release_path(p);
2054d96b3424SFilipe Manana 
2055d96b3424SFilipe Manana 	if (p->need_commit_sem) {
2056d96b3424SFilipe Manana 		int ret2;
2057d96b3424SFilipe Manana 
2058d96b3424SFilipe Manana 		ret2 = finish_need_commit_sem_search(p);
2059d96b3424SFilipe Manana 		up_read(&fs_info->commit_root_sem);
2060d96b3424SFilipe Manana 		if (ret2)
2061d96b3424SFilipe Manana 			ret = ret2;
2062d96b3424SFilipe Manana 	}
2063d96b3424SFilipe Manana 
2064be0e5c09SChris Mason 	return ret;
2065be0e5c09SChris Mason }
2066f75e2b79SJosef Bacik ALLOW_ERROR_INJECTION(btrfs_search_slot, ERRNO);
2067be0e5c09SChris Mason 
206874123bd7SChris Mason /*
20695d9e75c4SJan Schmidt  * Like btrfs_search_slot, this looks for a key in the given tree. It uses the
20705d9e75c4SJan Schmidt  * current state of the tree together with the operations recorded in the tree
20715d9e75c4SJan Schmidt  * modification log to search for the key in a previous version of this tree, as
20725d9e75c4SJan Schmidt  * denoted by the time_seq parameter.
20735d9e75c4SJan Schmidt  *
20745d9e75c4SJan Schmidt  * Naturally, there is no support for insert, delete or cow operations.
20755d9e75c4SJan Schmidt  *
20765d9e75c4SJan Schmidt  * The resulting path and return value will be set up as if we called
20775d9e75c4SJan Schmidt  * btrfs_search_slot at that point in time with ins_len and cow both set to 0.
20785d9e75c4SJan Schmidt  */
2079310712b2SOmar Sandoval int btrfs_search_old_slot(struct btrfs_root *root, const struct btrfs_key *key,
20805d9e75c4SJan Schmidt 			  struct btrfs_path *p, u64 time_seq)
20815d9e75c4SJan Schmidt {
20820b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
20835d9e75c4SJan Schmidt 	struct extent_buffer *b;
20845d9e75c4SJan Schmidt 	int slot;
20855d9e75c4SJan Schmidt 	int ret;
20865d9e75c4SJan Schmidt 	int err;
20875d9e75c4SJan Schmidt 	int level;
20885d9e75c4SJan Schmidt 	int lowest_unlock = 1;
20895d9e75c4SJan Schmidt 	u8 lowest_level = 0;
20905d9e75c4SJan Schmidt 
20915d9e75c4SJan Schmidt 	lowest_level = p->lowest_level;
20925d9e75c4SJan Schmidt 	WARN_ON(p->nodes[0] != NULL);
20935d9e75c4SJan Schmidt 
20945d9e75c4SJan Schmidt 	if (p->search_commit_root) {
20955d9e75c4SJan Schmidt 		BUG_ON(time_seq);
20965d9e75c4SJan Schmidt 		return btrfs_search_slot(NULL, root, key, p, 0, 0);
20975d9e75c4SJan Schmidt 	}
20985d9e75c4SJan Schmidt 
20995d9e75c4SJan Schmidt again:
2100f3a84ccdSFilipe Manana 	b = btrfs_get_old_root(root, time_seq);
2101315bed43SNikolay Borisov 	if (!b) {
2102315bed43SNikolay Borisov 		ret = -EIO;
2103315bed43SNikolay Borisov 		goto done;
2104315bed43SNikolay Borisov 	}
21055d9e75c4SJan Schmidt 	level = btrfs_header_level(b);
21065d9e75c4SJan Schmidt 	p->locks[level] = BTRFS_READ_LOCK;
21075d9e75c4SJan Schmidt 
21085d9e75c4SJan Schmidt 	while (b) {
2109abe9339dSQu Wenruo 		int dec = 0;
2110abe9339dSQu Wenruo 
21115d9e75c4SJan Schmidt 		level = btrfs_header_level(b);
21125d9e75c4SJan Schmidt 		p->nodes[level] = b;
21135d9e75c4SJan Schmidt 
21145d9e75c4SJan Schmidt 		/*
21155d9e75c4SJan Schmidt 		 * we have a lock on b and as long as we aren't changing
21165d9e75c4SJan Schmidt 		 * the tree, there is no way to for the items in b to change.
21175d9e75c4SJan Schmidt 		 * It is safe to drop the lock on our parent before we
21185d9e75c4SJan Schmidt 		 * go through the expensive btree search on b.
21195d9e75c4SJan Schmidt 		 */
21205d9e75c4SJan Schmidt 		btrfs_unlock_up_safe(p, level + 1);
21215d9e75c4SJan Schmidt 
2122995e9a16SNikolay Borisov 		ret = btrfs_bin_search(b, key, &slot);
2123cbca7d59SFilipe Manana 		if (ret < 0)
2124cbca7d59SFilipe Manana 			goto done;
21255d9e75c4SJan Schmidt 
2126abe9339dSQu Wenruo 		if (level == 0) {
2127abe9339dSQu Wenruo 			p->slots[level] = slot;
2128abe9339dSQu Wenruo 			unlock_up(p, level, lowest_unlock, 0, NULL);
2129abe9339dSQu Wenruo 			goto done;
2130abe9339dSQu Wenruo 		}
2131abe9339dSQu Wenruo 
21325d9e75c4SJan Schmidt 		if (ret && slot > 0) {
21335d9e75c4SJan Schmidt 			dec = 1;
2134abe9339dSQu Wenruo 			slot--;
21355d9e75c4SJan Schmidt 		}
21365d9e75c4SJan Schmidt 		p->slots[level] = slot;
21375d9e75c4SJan Schmidt 		unlock_up(p, level, lowest_unlock, 0, NULL);
21385d9e75c4SJan Schmidt 
21395d9e75c4SJan Schmidt 		if (level == lowest_level) {
21405d9e75c4SJan Schmidt 			if (dec)
21415d9e75c4SJan Schmidt 				p->slots[level]++;
21425d9e75c4SJan Schmidt 			goto done;
21435d9e75c4SJan Schmidt 		}
21445d9e75c4SJan Schmidt 
2145abe9339dSQu Wenruo 		err = read_block_for_search(root, p, &b, level, slot, key);
21465d9e75c4SJan Schmidt 		if (err == -EAGAIN)
21475d9e75c4SJan Schmidt 			goto again;
21485d9e75c4SJan Schmidt 		if (err) {
21495d9e75c4SJan Schmidt 			ret = err;
21505d9e75c4SJan Schmidt 			goto done;
21515d9e75c4SJan Schmidt 		}
21525d9e75c4SJan Schmidt 
21535d9e75c4SJan Schmidt 		level = btrfs_header_level(b);
21545d9e75c4SJan Schmidt 		btrfs_tree_read_lock(b);
2155f3a84ccdSFilipe Manana 		b = btrfs_tree_mod_log_rewind(fs_info, p, b, time_seq);
2156db7f3436SJosef Bacik 		if (!b) {
2157db7f3436SJosef Bacik 			ret = -ENOMEM;
2158db7f3436SJosef Bacik 			goto done;
2159db7f3436SJosef Bacik 		}
21605d9e75c4SJan Schmidt 		p->locks[level] = BTRFS_READ_LOCK;
21615d9e75c4SJan Schmidt 		p->nodes[level] = b;
21625d9e75c4SJan Schmidt 	}
21635d9e75c4SJan Schmidt 	ret = 1;
21645d9e75c4SJan Schmidt done:
21655d9e75c4SJan Schmidt 	if (ret < 0)
21665d9e75c4SJan Schmidt 		btrfs_release_path(p);
21675d9e75c4SJan Schmidt 
21685d9e75c4SJan Schmidt 	return ret;
21695d9e75c4SJan Schmidt }
21705d9e75c4SJan Schmidt 
21715d9e75c4SJan Schmidt /*
21722f38b3e1SArne Jansen  * helper to use instead of search slot if no exact match is needed but
21732f38b3e1SArne Jansen  * instead the next or previous item should be returned.
21742f38b3e1SArne Jansen  * When find_higher is true, the next higher item is returned, the next lower
21752f38b3e1SArne Jansen  * otherwise.
21762f38b3e1SArne Jansen  * When return_any and find_higher are both true, and no higher item is found,
21772f38b3e1SArne Jansen  * return the next lower instead.
21782f38b3e1SArne Jansen  * When return_any is true and find_higher is false, and no lower item is found,
21792f38b3e1SArne Jansen  * return the next higher instead.
21802f38b3e1SArne Jansen  * It returns 0 if any item is found, 1 if none is found (tree empty), and
21812f38b3e1SArne Jansen  * < 0 on error
21822f38b3e1SArne Jansen  */
21832f38b3e1SArne Jansen int btrfs_search_slot_for_read(struct btrfs_root *root,
2184310712b2SOmar Sandoval 			       const struct btrfs_key *key,
2185310712b2SOmar Sandoval 			       struct btrfs_path *p, int find_higher,
2186310712b2SOmar Sandoval 			       int return_any)
21872f38b3e1SArne Jansen {
21882f38b3e1SArne Jansen 	int ret;
21892f38b3e1SArne Jansen 	struct extent_buffer *leaf;
21902f38b3e1SArne Jansen 
21912f38b3e1SArne Jansen again:
21922f38b3e1SArne Jansen 	ret = btrfs_search_slot(NULL, root, key, p, 0, 0);
21932f38b3e1SArne Jansen 	if (ret <= 0)
21942f38b3e1SArne Jansen 		return ret;
21952f38b3e1SArne Jansen 	/*
21962f38b3e1SArne Jansen 	 * a return value of 1 means the path is at the position where the
21972f38b3e1SArne Jansen 	 * item should be inserted. Normally this is the next bigger item,
21982f38b3e1SArne Jansen 	 * but in case the previous item is the last in a leaf, path points
21992f38b3e1SArne Jansen 	 * to the first free slot in the previous leaf, i.e. at an invalid
22002f38b3e1SArne Jansen 	 * item.
22012f38b3e1SArne Jansen 	 */
22022f38b3e1SArne Jansen 	leaf = p->nodes[0];
22032f38b3e1SArne Jansen 
22042f38b3e1SArne Jansen 	if (find_higher) {
22052f38b3e1SArne Jansen 		if (p->slots[0] >= btrfs_header_nritems(leaf)) {
22062f38b3e1SArne Jansen 			ret = btrfs_next_leaf(root, p);
22072f38b3e1SArne Jansen 			if (ret <= 0)
22082f38b3e1SArne Jansen 				return ret;
22092f38b3e1SArne Jansen 			if (!return_any)
22102f38b3e1SArne Jansen 				return 1;
22112f38b3e1SArne Jansen 			/*
22122f38b3e1SArne Jansen 			 * no higher item found, return the next
22132f38b3e1SArne Jansen 			 * lower instead
22142f38b3e1SArne Jansen 			 */
22152f38b3e1SArne Jansen 			return_any = 0;
22162f38b3e1SArne Jansen 			find_higher = 0;
22172f38b3e1SArne Jansen 			btrfs_release_path(p);
22182f38b3e1SArne Jansen 			goto again;
22192f38b3e1SArne Jansen 		}
22202f38b3e1SArne Jansen 	} else {
22212f38b3e1SArne Jansen 		if (p->slots[0] == 0) {
22222f38b3e1SArne Jansen 			ret = btrfs_prev_leaf(root, p);
2223e6793769SArne Jansen 			if (ret < 0)
22242f38b3e1SArne Jansen 				return ret;
2225e6793769SArne Jansen 			if (!ret) {
222623c6bf6aSFilipe David Borba Manana 				leaf = p->nodes[0];
222723c6bf6aSFilipe David Borba Manana 				if (p->slots[0] == btrfs_header_nritems(leaf))
222823c6bf6aSFilipe David Borba Manana 					p->slots[0]--;
2229e6793769SArne Jansen 				return 0;
2230e6793769SArne Jansen 			}
22312f38b3e1SArne Jansen 			if (!return_any)
22322f38b3e1SArne Jansen 				return 1;
22332f38b3e1SArne Jansen 			/*
22342f38b3e1SArne Jansen 			 * no lower item found, return the next
22352f38b3e1SArne Jansen 			 * higher instead
22362f38b3e1SArne Jansen 			 */
22372f38b3e1SArne Jansen 			return_any = 0;
22382f38b3e1SArne Jansen 			find_higher = 1;
22392f38b3e1SArne Jansen 			btrfs_release_path(p);
22402f38b3e1SArne Jansen 			goto again;
2241e6793769SArne Jansen 		} else {
22422f38b3e1SArne Jansen 			--p->slots[0];
22432f38b3e1SArne Jansen 		}
22442f38b3e1SArne Jansen 	}
22452f38b3e1SArne Jansen 	return 0;
22462f38b3e1SArne Jansen }
22472f38b3e1SArne Jansen 
22482f38b3e1SArne Jansen /*
22490ff40a91SMarcos Paulo de Souza  * Execute search and call btrfs_previous_item to traverse backwards if the item
22500ff40a91SMarcos Paulo de Souza  * was not found.
22510ff40a91SMarcos Paulo de Souza  *
22520ff40a91SMarcos Paulo de Souza  * Return 0 if found, 1 if not found and < 0 if error.
22530ff40a91SMarcos Paulo de Souza  */
22540ff40a91SMarcos Paulo de Souza int btrfs_search_backwards(struct btrfs_root *root, struct btrfs_key *key,
22550ff40a91SMarcos Paulo de Souza 			   struct btrfs_path *path)
22560ff40a91SMarcos Paulo de Souza {
22570ff40a91SMarcos Paulo de Souza 	int ret;
22580ff40a91SMarcos Paulo de Souza 
22590ff40a91SMarcos Paulo de Souza 	ret = btrfs_search_slot(NULL, root, key, path, 0, 0);
22600ff40a91SMarcos Paulo de Souza 	if (ret > 0)
22610ff40a91SMarcos Paulo de Souza 		ret = btrfs_previous_item(root, path, key->objectid, key->type);
22620ff40a91SMarcos Paulo de Souza 
22630ff40a91SMarcos Paulo de Souza 	if (ret == 0)
22640ff40a91SMarcos Paulo de Souza 		btrfs_item_key_to_cpu(path->nodes[0], key, path->slots[0]);
22650ff40a91SMarcos Paulo de Souza 
22660ff40a91SMarcos Paulo de Souza 	return ret;
22670ff40a91SMarcos Paulo de Souza }
22680ff40a91SMarcos Paulo de Souza 
22690ff40a91SMarcos Paulo de Souza /*
227074123bd7SChris Mason  * adjust the pointers going up the tree, starting at level
227174123bd7SChris Mason  * making sure the right key of each node is points to 'key'.
227274123bd7SChris Mason  * This is used after shifting pointers to the left, so it stops
227374123bd7SChris Mason  * fixing up pointers when a given leaf/node is not in slot 0 of the
227474123bd7SChris Mason  * higher levels
2275aa5d6bedSChris Mason  *
227674123bd7SChris Mason  */
2277b167fa91SNikolay Borisov static void fixup_low_keys(struct btrfs_path *path,
22785f39d397SChris Mason 			   struct btrfs_disk_key *key, int level)
2279be0e5c09SChris Mason {
2280be0e5c09SChris Mason 	int i;
22815f39d397SChris Mason 	struct extent_buffer *t;
22820e82bcfeSDavid Sterba 	int ret;
22835f39d397SChris Mason 
2284234b63a0SChris Mason 	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
2285be0e5c09SChris Mason 		int tslot = path->slots[i];
22860e82bcfeSDavid Sterba 
2287eb60ceacSChris Mason 		if (!path->nodes[i])
2288be0e5c09SChris Mason 			break;
22895f39d397SChris Mason 		t = path->nodes[i];
2290f3a84ccdSFilipe Manana 		ret = btrfs_tree_mod_log_insert_key(t, tslot,
2291f3a84ccdSFilipe Manana 				BTRFS_MOD_LOG_KEY_REPLACE, GFP_ATOMIC);
22920e82bcfeSDavid Sterba 		BUG_ON(ret < 0);
22935f39d397SChris Mason 		btrfs_set_node_key(t, key, tslot);
2294d6025579SChris Mason 		btrfs_mark_buffer_dirty(path->nodes[i]);
2295be0e5c09SChris Mason 		if (tslot != 0)
2296be0e5c09SChris Mason 			break;
2297be0e5c09SChris Mason 	}
2298be0e5c09SChris Mason }
2299be0e5c09SChris Mason 
230074123bd7SChris Mason /*
230131840ae1SZheng Yan  * update item key.
230231840ae1SZheng Yan  *
230331840ae1SZheng Yan  * This function isn't completely safe. It's the caller's responsibility
230431840ae1SZheng Yan  * that the new key won't break the order
230531840ae1SZheng Yan  */
2306b7a0365eSDaniel Dressler void btrfs_set_item_key_safe(struct btrfs_fs_info *fs_info,
2307b7a0365eSDaniel Dressler 			     struct btrfs_path *path,
2308310712b2SOmar Sandoval 			     const struct btrfs_key *new_key)
230931840ae1SZheng Yan {
231031840ae1SZheng Yan 	struct btrfs_disk_key disk_key;
231131840ae1SZheng Yan 	struct extent_buffer *eb;
231231840ae1SZheng Yan 	int slot;
231331840ae1SZheng Yan 
231431840ae1SZheng Yan 	eb = path->nodes[0];
231531840ae1SZheng Yan 	slot = path->slots[0];
231631840ae1SZheng Yan 	if (slot > 0) {
231731840ae1SZheng Yan 		btrfs_item_key(eb, &disk_key, slot - 1);
23187c15d410SQu Wenruo 		if (unlikely(comp_keys(&disk_key, new_key) >= 0)) {
23197c15d410SQu Wenruo 			btrfs_crit(fs_info,
23207c15d410SQu Wenruo 		"slot %u key (%llu %u %llu) new key (%llu %u %llu)",
23217c15d410SQu Wenruo 				   slot, btrfs_disk_key_objectid(&disk_key),
23227c15d410SQu Wenruo 				   btrfs_disk_key_type(&disk_key),
23237c15d410SQu Wenruo 				   btrfs_disk_key_offset(&disk_key),
23247c15d410SQu Wenruo 				   new_key->objectid, new_key->type,
23257c15d410SQu Wenruo 				   new_key->offset);
23267c15d410SQu Wenruo 			btrfs_print_leaf(eb);
23277c15d410SQu Wenruo 			BUG();
23287c15d410SQu Wenruo 		}
232931840ae1SZheng Yan 	}
233031840ae1SZheng Yan 	if (slot < btrfs_header_nritems(eb) - 1) {
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 
234531840ae1SZheng Yan 	btrfs_cpu_key_to_disk(&disk_key, new_key);
234631840ae1SZheng Yan 	btrfs_set_item_key(eb, &disk_key, slot);
234731840ae1SZheng Yan 	btrfs_mark_buffer_dirty(eb);
234831840ae1SZheng Yan 	if (slot == 0)
2349b167fa91SNikolay Borisov 		fixup_low_keys(path, &disk_key, 1);
235031840ae1SZheng Yan }
235131840ae1SZheng Yan 
235231840ae1SZheng Yan /*
2353d16c702fSQu Wenruo  * Check key order of two sibling extent buffers.
2354d16c702fSQu Wenruo  *
2355d16c702fSQu Wenruo  * Return true if something is wrong.
2356d16c702fSQu Wenruo  * Return false if everything is fine.
2357d16c702fSQu Wenruo  *
2358d16c702fSQu Wenruo  * Tree-checker only works inside one tree block, thus the following
2359d16c702fSQu Wenruo  * corruption can not be detected by tree-checker:
2360d16c702fSQu Wenruo  *
2361d16c702fSQu Wenruo  * Leaf @left			| Leaf @right
2362d16c702fSQu Wenruo  * --------------------------------------------------------------
2363d16c702fSQu Wenruo  * | 1 | 2 | 3 | 4 | 5 | f6 |   | 7 | 8 |
2364d16c702fSQu Wenruo  *
2365d16c702fSQu Wenruo  * Key f6 in leaf @left itself is valid, but not valid when the next
2366d16c702fSQu Wenruo  * key in leaf @right is 7.
2367d16c702fSQu Wenruo  * This can only be checked at tree block merge time.
2368d16c702fSQu Wenruo  * And since tree checker has ensured all key order in each tree block
2369d16c702fSQu Wenruo  * is correct, we only need to bother the last key of @left and the first
2370d16c702fSQu Wenruo  * key of @right.
2371d16c702fSQu Wenruo  */
2372d16c702fSQu Wenruo static bool check_sibling_keys(struct extent_buffer *left,
2373d16c702fSQu Wenruo 			       struct extent_buffer *right)
2374d16c702fSQu Wenruo {
2375d16c702fSQu Wenruo 	struct btrfs_key left_last;
2376d16c702fSQu Wenruo 	struct btrfs_key right_first;
2377d16c702fSQu Wenruo 	int level = btrfs_header_level(left);
2378d16c702fSQu Wenruo 	int nr_left = btrfs_header_nritems(left);
2379d16c702fSQu Wenruo 	int nr_right = btrfs_header_nritems(right);
2380d16c702fSQu Wenruo 
2381d16c702fSQu Wenruo 	/* No key to check in one of the tree blocks */
2382d16c702fSQu Wenruo 	if (!nr_left || !nr_right)
2383d16c702fSQu Wenruo 		return false;
2384d16c702fSQu Wenruo 
2385d16c702fSQu Wenruo 	if (level) {
2386d16c702fSQu Wenruo 		btrfs_node_key_to_cpu(left, &left_last, nr_left - 1);
2387d16c702fSQu Wenruo 		btrfs_node_key_to_cpu(right, &right_first, 0);
2388d16c702fSQu Wenruo 	} else {
2389d16c702fSQu Wenruo 		btrfs_item_key_to_cpu(left, &left_last, nr_left - 1);
2390d16c702fSQu Wenruo 		btrfs_item_key_to_cpu(right, &right_first, 0);
2391d16c702fSQu Wenruo 	}
2392d16c702fSQu Wenruo 
2393d16c702fSQu Wenruo 	if (btrfs_comp_cpu_keys(&left_last, &right_first) >= 0) {
2394d16c702fSQu Wenruo 		btrfs_crit(left->fs_info,
2395d16c702fSQu Wenruo "bad key order, sibling blocks, left last (%llu %u %llu) right first (%llu %u %llu)",
2396d16c702fSQu Wenruo 			   left_last.objectid, left_last.type,
2397d16c702fSQu Wenruo 			   left_last.offset, right_first.objectid,
2398d16c702fSQu Wenruo 			   right_first.type, right_first.offset);
2399d16c702fSQu Wenruo 		return true;
2400d16c702fSQu Wenruo 	}
2401d16c702fSQu Wenruo 	return false;
2402d16c702fSQu Wenruo }
2403d16c702fSQu Wenruo 
2404d16c702fSQu Wenruo /*
240574123bd7SChris Mason  * try to push data from one node into the next node left in the
240679f95c82SChris Mason  * tree.
2407aa5d6bedSChris Mason  *
2408aa5d6bedSChris Mason  * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
2409aa5d6bedSChris Mason  * error, and > 0 if there was no room in the left hand block.
241074123bd7SChris Mason  */
241198ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans,
24122ff7e61eSJeff Mahoney 			  struct extent_buffer *dst,
2413971a1f66SChris Mason 			  struct extent_buffer *src, int empty)
2414be0e5c09SChris Mason {
2415d30a668fSDavid Sterba 	struct btrfs_fs_info *fs_info = trans->fs_info;
2416be0e5c09SChris Mason 	int push_items = 0;
2417bb803951SChris Mason 	int src_nritems;
2418bb803951SChris Mason 	int dst_nritems;
2419aa5d6bedSChris Mason 	int ret = 0;
2420be0e5c09SChris Mason 
24215f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
24225f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
24230b246afaSJeff Mahoney 	push_items = BTRFS_NODEPTRS_PER_BLOCK(fs_info) - dst_nritems;
24247bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
24257bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
242654aa1f4dSChris Mason 
2427bce4eae9SChris Mason 	if (!empty && src_nritems <= 8)
2428971a1f66SChris Mason 		return 1;
2429971a1f66SChris Mason 
2430d397712bSChris Mason 	if (push_items <= 0)
2431be0e5c09SChris Mason 		return 1;
2432be0e5c09SChris Mason 
2433bce4eae9SChris Mason 	if (empty) {
2434971a1f66SChris Mason 		push_items = min(src_nritems, push_items);
2435bce4eae9SChris Mason 		if (push_items < src_nritems) {
2436bce4eae9SChris Mason 			/* leave at least 8 pointers in the node if
2437bce4eae9SChris Mason 			 * we aren't going to empty it
2438bce4eae9SChris Mason 			 */
2439bce4eae9SChris Mason 			if (src_nritems - push_items < 8) {
2440bce4eae9SChris Mason 				if (push_items <= 8)
2441bce4eae9SChris Mason 					return 1;
2442bce4eae9SChris Mason 				push_items -= 8;
2443bce4eae9SChris Mason 			}
2444bce4eae9SChris Mason 		}
2445bce4eae9SChris Mason 	} else
2446bce4eae9SChris Mason 		push_items = min(src_nritems - 8, push_items);
244779f95c82SChris Mason 
2448d16c702fSQu Wenruo 	/* dst is the left eb, src is the middle eb */
2449d16c702fSQu Wenruo 	if (check_sibling_keys(dst, src)) {
2450d16c702fSQu Wenruo 		ret = -EUCLEAN;
2451d16c702fSQu Wenruo 		btrfs_abort_transaction(trans, ret);
2452d16c702fSQu Wenruo 		return ret;
2453d16c702fSQu Wenruo 	}
2454f3a84ccdSFilipe Manana 	ret = btrfs_tree_mod_log_eb_copy(dst, src, dst_nritems, 0, push_items);
24555de865eeSFilipe David Borba Manana 	if (ret) {
245666642832SJeff Mahoney 		btrfs_abort_transaction(trans, ret);
24575de865eeSFilipe David Borba Manana 		return ret;
24585de865eeSFilipe David Borba Manana 	}
24595f39d397SChris Mason 	copy_extent_buffer(dst, src,
24605f39d397SChris Mason 			   btrfs_node_key_ptr_offset(dst_nritems),
24615f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
2462123abc88SChris Mason 			   push_items * sizeof(struct btrfs_key_ptr));
24635f39d397SChris Mason 
2464bb803951SChris Mason 	if (push_items < src_nritems) {
246557911b8bSJan Schmidt 		/*
2466f3a84ccdSFilipe Manana 		 * Don't call btrfs_tree_mod_log_insert_move() here, key removal
2467f3a84ccdSFilipe Manana 		 * was already fully logged by btrfs_tree_mod_log_eb_copy() above.
246857911b8bSJan Schmidt 		 */
24695f39d397SChris Mason 		memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
24705f39d397SChris Mason 				      btrfs_node_key_ptr_offset(push_items),
2471e2fa7227SChris Mason 				      (src_nritems - push_items) *
2472123abc88SChris Mason 				      sizeof(struct btrfs_key_ptr));
2473bb803951SChris Mason 	}
24745f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
24755f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
24765f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
24775f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
247831840ae1SZheng Yan 
2479bb803951SChris Mason 	return ret;
2480be0e5c09SChris Mason }
2481be0e5c09SChris Mason 
248297571fd0SChris Mason /*
248379f95c82SChris Mason  * try to push data from one node into the next node right in the
248479f95c82SChris Mason  * tree.
248579f95c82SChris Mason  *
248679f95c82SChris Mason  * returns 0 if some ptrs were pushed, < 0 if there was some horrible
248779f95c82SChris Mason  * error, and > 0 if there was no room in the right hand block.
248879f95c82SChris Mason  *
248979f95c82SChris Mason  * this will  only push up to 1/2 the contents of the left node over
249079f95c82SChris Mason  */
24915f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans,
24925f39d397SChris Mason 			      struct extent_buffer *dst,
24935f39d397SChris Mason 			      struct extent_buffer *src)
249479f95c82SChris Mason {
249555d32ed8SDavid Sterba 	struct btrfs_fs_info *fs_info = trans->fs_info;
249679f95c82SChris Mason 	int push_items = 0;
249779f95c82SChris Mason 	int max_push;
249879f95c82SChris Mason 	int src_nritems;
249979f95c82SChris Mason 	int dst_nritems;
250079f95c82SChris Mason 	int ret = 0;
250179f95c82SChris Mason 
25027bb86316SChris Mason 	WARN_ON(btrfs_header_generation(src) != trans->transid);
25037bb86316SChris Mason 	WARN_ON(btrfs_header_generation(dst) != trans->transid);
25047bb86316SChris Mason 
25055f39d397SChris Mason 	src_nritems = btrfs_header_nritems(src);
25065f39d397SChris Mason 	dst_nritems = btrfs_header_nritems(dst);
25070b246afaSJeff Mahoney 	push_items = BTRFS_NODEPTRS_PER_BLOCK(fs_info) - dst_nritems;
2508d397712bSChris Mason 	if (push_items <= 0)
250979f95c82SChris Mason 		return 1;
2510bce4eae9SChris Mason 
2511d397712bSChris Mason 	if (src_nritems < 4)
2512bce4eae9SChris Mason 		return 1;
251379f95c82SChris Mason 
251479f95c82SChris Mason 	max_push = src_nritems / 2 + 1;
251579f95c82SChris Mason 	/* don't try to empty the node */
2516d397712bSChris Mason 	if (max_push >= src_nritems)
251779f95c82SChris Mason 		return 1;
2518252c38f0SYan 
251979f95c82SChris Mason 	if (max_push < push_items)
252079f95c82SChris Mason 		push_items = max_push;
252179f95c82SChris Mason 
2522d16c702fSQu Wenruo 	/* dst is the right eb, src is the middle eb */
2523d16c702fSQu Wenruo 	if (check_sibling_keys(src, dst)) {
2524d16c702fSQu Wenruo 		ret = -EUCLEAN;
2525d16c702fSQu Wenruo 		btrfs_abort_transaction(trans, ret);
2526d16c702fSQu Wenruo 		return ret;
2527d16c702fSQu Wenruo 	}
2528f3a84ccdSFilipe Manana 	ret = btrfs_tree_mod_log_insert_move(dst, push_items, 0, dst_nritems);
2529bf1d3425SDavid Sterba 	BUG_ON(ret < 0);
25305f39d397SChris Mason 	memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
25315f39d397SChris Mason 				      btrfs_node_key_ptr_offset(0),
25325f39d397SChris Mason 				      (dst_nritems) *
25335f39d397SChris Mason 				      sizeof(struct btrfs_key_ptr));
2534d6025579SChris Mason 
2535f3a84ccdSFilipe Manana 	ret = btrfs_tree_mod_log_eb_copy(dst, src, 0, src_nritems - push_items,
2536ed874f0dSDavid Sterba 					 push_items);
25375de865eeSFilipe David Borba Manana 	if (ret) {
253866642832SJeff Mahoney 		btrfs_abort_transaction(trans, ret);
25395de865eeSFilipe David Borba Manana 		return ret;
25405de865eeSFilipe David Borba Manana 	}
25415f39d397SChris Mason 	copy_extent_buffer(dst, src,
25425f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
25435f39d397SChris Mason 			   btrfs_node_key_ptr_offset(src_nritems - push_items),
2544123abc88SChris Mason 			   push_items * sizeof(struct btrfs_key_ptr));
254579f95c82SChris Mason 
25465f39d397SChris Mason 	btrfs_set_header_nritems(src, src_nritems - push_items);
25475f39d397SChris Mason 	btrfs_set_header_nritems(dst, dst_nritems + push_items);
254879f95c82SChris Mason 
25495f39d397SChris Mason 	btrfs_mark_buffer_dirty(src);
25505f39d397SChris Mason 	btrfs_mark_buffer_dirty(dst);
255131840ae1SZheng Yan 
255279f95c82SChris Mason 	return ret;
255379f95c82SChris Mason }
255479f95c82SChris Mason 
255579f95c82SChris Mason /*
255697571fd0SChris Mason  * helper function to insert a new root level in the tree.
255797571fd0SChris Mason  * A new node is allocated, and a single item is inserted to
255897571fd0SChris Mason  * point to the existing root
2559aa5d6bedSChris Mason  *
2560aa5d6bedSChris Mason  * returns zero on success or < 0 on failure.
256197571fd0SChris Mason  */
2562d397712bSChris Mason static noinline int insert_new_root(struct btrfs_trans_handle *trans,
25635f39d397SChris Mason 			   struct btrfs_root *root,
2564fdd99c72SLiu Bo 			   struct btrfs_path *path, int level)
256574123bd7SChris Mason {
25660b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
25677bb86316SChris Mason 	u64 lower_gen;
25685f39d397SChris Mason 	struct extent_buffer *lower;
25695f39d397SChris Mason 	struct extent_buffer *c;
2570925baeddSChris Mason 	struct extent_buffer *old;
25715f39d397SChris Mason 	struct btrfs_disk_key lower_key;
2572d9d19a01SDavid Sterba 	int ret;
25735c680ed6SChris Mason 
25745c680ed6SChris Mason 	BUG_ON(path->nodes[level]);
25755c680ed6SChris Mason 	BUG_ON(path->nodes[level-1] != root->node);
25765c680ed6SChris Mason 
25777bb86316SChris Mason 	lower = path->nodes[level-1];
25787bb86316SChris Mason 	if (level == 1)
25797bb86316SChris Mason 		btrfs_item_key(lower, &lower_key, 0);
25807bb86316SChris Mason 	else
25817bb86316SChris Mason 		btrfs_node_key(lower, &lower_key, 0);
25827bb86316SChris Mason 
258379bd3712SFilipe Manana 	c = btrfs_alloc_tree_block(trans, root, 0, root->root_key.objectid,
258479bd3712SFilipe Manana 				   &lower_key, level, root->node->start, 0,
2585cf6f34aaSJosef Bacik 				   BTRFS_NESTING_NEW_ROOT);
25865f39d397SChris Mason 	if (IS_ERR(c))
25875f39d397SChris Mason 		return PTR_ERR(c);
2588925baeddSChris Mason 
25890b246afaSJeff Mahoney 	root_add_used(root, fs_info->nodesize);
2590f0486c68SYan, Zheng 
25915f39d397SChris Mason 	btrfs_set_header_nritems(c, 1);
25925f39d397SChris Mason 	btrfs_set_node_key(c, &lower_key, 0);
2593db94535dSChris Mason 	btrfs_set_node_blockptr(c, 0, lower->start);
25947bb86316SChris Mason 	lower_gen = btrfs_header_generation(lower);
259531840ae1SZheng Yan 	WARN_ON(lower_gen != trans->transid);
25967bb86316SChris Mason 
25977bb86316SChris Mason 	btrfs_set_node_ptr_generation(c, 0, lower_gen);
25985f39d397SChris Mason 
25995f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
2600d5719762SChris Mason 
2601925baeddSChris Mason 	old = root->node;
2602406808abSFilipe Manana 	ret = btrfs_tree_mod_log_insert_root(root->node, c, false);
2603d9d19a01SDavid Sterba 	BUG_ON(ret < 0);
2604240f62c8SChris Mason 	rcu_assign_pointer(root->node, c);
2605925baeddSChris Mason 
2606925baeddSChris Mason 	/* the super has an extra ref to root->node */
2607925baeddSChris Mason 	free_extent_buffer(old);
2608925baeddSChris Mason 
26090b86a832SChris Mason 	add_root_to_dirty_list(root);
261067439dadSDavid Sterba 	atomic_inc(&c->refs);
26115f39d397SChris Mason 	path->nodes[level] = c;
2612ac5887c8SJosef Bacik 	path->locks[level] = BTRFS_WRITE_LOCK;
261374123bd7SChris Mason 	path->slots[level] = 0;
261474123bd7SChris Mason 	return 0;
261574123bd7SChris Mason }
26165c680ed6SChris Mason 
26175c680ed6SChris Mason /*
26185c680ed6SChris Mason  * worker function to insert a single pointer in a node.
26195c680ed6SChris Mason  * the node should have enough room for the pointer already
262097571fd0SChris Mason  *
26215c680ed6SChris Mason  * slot and level indicate where you want the key to go, and
26225c680ed6SChris Mason  * blocknr is the block the key points to.
26235c680ed6SChris Mason  */
2624143bede5SJeff Mahoney static void insert_ptr(struct btrfs_trans_handle *trans,
26256ad3cf6dSDavid Sterba 		       struct btrfs_path *path,
2626143bede5SJeff Mahoney 		       struct btrfs_disk_key *key, u64 bytenr,
2627c3e06965SJan Schmidt 		       int slot, int level)
26285c680ed6SChris Mason {
26295f39d397SChris Mason 	struct extent_buffer *lower;
26305c680ed6SChris Mason 	int nritems;
2631f3ea38daSJan Schmidt 	int ret;
26325c680ed6SChris Mason 
26335c680ed6SChris Mason 	BUG_ON(!path->nodes[level]);
263449d0c642SFilipe Manana 	btrfs_assert_tree_write_locked(path->nodes[level]);
26355f39d397SChris Mason 	lower = path->nodes[level];
26365f39d397SChris Mason 	nritems = btrfs_header_nritems(lower);
2637c293498bSStoyan Gaydarov 	BUG_ON(slot > nritems);
26386ad3cf6dSDavid Sterba 	BUG_ON(nritems == BTRFS_NODEPTRS_PER_BLOCK(trans->fs_info));
263974123bd7SChris Mason 	if (slot != nritems) {
2640bf1d3425SDavid Sterba 		if (level) {
2641f3a84ccdSFilipe Manana 			ret = btrfs_tree_mod_log_insert_move(lower, slot + 1,
2642f3a84ccdSFilipe Manana 					slot, nritems - slot);
2643bf1d3425SDavid Sterba 			BUG_ON(ret < 0);
2644bf1d3425SDavid Sterba 		}
26455f39d397SChris Mason 		memmove_extent_buffer(lower,
26465f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
26475f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
2648123abc88SChris Mason 			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
264974123bd7SChris Mason 	}
2650c3e06965SJan Schmidt 	if (level) {
2651f3a84ccdSFilipe Manana 		ret = btrfs_tree_mod_log_insert_key(lower, slot,
2652f3a84ccdSFilipe Manana 					    BTRFS_MOD_LOG_KEY_ADD, GFP_NOFS);
2653f3ea38daSJan Schmidt 		BUG_ON(ret < 0);
2654f3ea38daSJan Schmidt 	}
26555f39d397SChris Mason 	btrfs_set_node_key(lower, key, slot);
2656db94535dSChris Mason 	btrfs_set_node_blockptr(lower, slot, bytenr);
265774493f7aSChris Mason 	WARN_ON(trans->transid == 0);
265874493f7aSChris Mason 	btrfs_set_node_ptr_generation(lower, slot, trans->transid);
26595f39d397SChris Mason 	btrfs_set_header_nritems(lower, nritems + 1);
26605f39d397SChris Mason 	btrfs_mark_buffer_dirty(lower);
266174123bd7SChris Mason }
266274123bd7SChris Mason 
266397571fd0SChris Mason /*
266497571fd0SChris Mason  * split the node at the specified level in path in two.
266597571fd0SChris Mason  * The path is corrected to point to the appropriate node after the split
266697571fd0SChris Mason  *
266797571fd0SChris Mason  * Before splitting this tries to make some room in the node by pushing
266897571fd0SChris Mason  * left and right, if either one works, it returns right away.
2669aa5d6bedSChris Mason  *
2670aa5d6bedSChris Mason  * returns 0 on success and < 0 on failure
267197571fd0SChris Mason  */
2672e02119d5SChris Mason static noinline int split_node(struct btrfs_trans_handle *trans,
2673e02119d5SChris Mason 			       struct btrfs_root *root,
2674e02119d5SChris Mason 			       struct btrfs_path *path, int level)
2675be0e5c09SChris Mason {
26760b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
26775f39d397SChris Mason 	struct extent_buffer *c;
26785f39d397SChris Mason 	struct extent_buffer *split;
26795f39d397SChris Mason 	struct btrfs_disk_key disk_key;
2680be0e5c09SChris Mason 	int mid;
26815c680ed6SChris Mason 	int ret;
26827518a238SChris Mason 	u32 c_nritems;
2683be0e5c09SChris Mason 
26845f39d397SChris Mason 	c = path->nodes[level];
26857bb86316SChris Mason 	WARN_ON(btrfs_header_generation(c) != trans->transid);
26865f39d397SChris Mason 	if (c == root->node) {
2687d9abbf1cSJan Schmidt 		/*
268890f8d62eSJan Schmidt 		 * trying to split the root, lets make a new one
268990f8d62eSJan Schmidt 		 *
2690fdd99c72SLiu Bo 		 * tree mod log: We don't log_removal old root in
269190f8d62eSJan Schmidt 		 * insert_new_root, because that root buffer will be kept as a
269290f8d62eSJan Schmidt 		 * normal node. We are going to log removal of half of the
2693f3a84ccdSFilipe Manana 		 * elements below with btrfs_tree_mod_log_eb_copy(). We're
2694f3a84ccdSFilipe Manana 		 * holding a tree lock on the buffer, which is why we cannot
2695f3a84ccdSFilipe Manana 		 * race with other tree_mod_log users.
2696d9abbf1cSJan Schmidt 		 */
2697fdd99c72SLiu Bo 		ret = insert_new_root(trans, root, path, level + 1);
26985c680ed6SChris Mason 		if (ret)
26995c680ed6SChris Mason 			return ret;
2700b3612421SChris Mason 	} else {
2701e66f709bSChris Mason 		ret = push_nodes_for_insert(trans, root, path, level);
27025f39d397SChris Mason 		c = path->nodes[level];
27035f39d397SChris Mason 		if (!ret && btrfs_header_nritems(c) <
27040b246afaSJeff Mahoney 		    BTRFS_NODEPTRS_PER_BLOCK(fs_info) - 3)
2705e66f709bSChris Mason 			return 0;
270654aa1f4dSChris Mason 		if (ret < 0)
270754aa1f4dSChris Mason 			return ret;
27085c680ed6SChris Mason 	}
2709e66f709bSChris Mason 
27105f39d397SChris Mason 	c_nritems = btrfs_header_nritems(c);
27115d4f98a2SYan Zheng 	mid = (c_nritems + 1) / 2;
27125d4f98a2SYan Zheng 	btrfs_node_key(c, &disk_key, mid);
27137bb86316SChris Mason 
271479bd3712SFilipe Manana 	split = btrfs_alloc_tree_block(trans, root, 0, root->root_key.objectid,
271579bd3712SFilipe Manana 				       &disk_key, level, c->start, 0,
271679bd3712SFilipe Manana 				       BTRFS_NESTING_SPLIT);
27175f39d397SChris Mason 	if (IS_ERR(split))
27185f39d397SChris Mason 		return PTR_ERR(split);
271954aa1f4dSChris Mason 
27200b246afaSJeff Mahoney 	root_add_used(root, fs_info->nodesize);
2721bc877d28SNikolay Borisov 	ASSERT(btrfs_header_level(c) == level);
27225f39d397SChris Mason 
2723f3a84ccdSFilipe Manana 	ret = btrfs_tree_mod_log_eb_copy(split, c, 0, mid, c_nritems - mid);
27245de865eeSFilipe David Borba Manana 	if (ret) {
272566642832SJeff Mahoney 		btrfs_abort_transaction(trans, ret);
27265de865eeSFilipe David Borba Manana 		return ret;
27275de865eeSFilipe David Borba Manana 	}
27285f39d397SChris Mason 	copy_extent_buffer(split, c,
27295f39d397SChris Mason 			   btrfs_node_key_ptr_offset(0),
27305f39d397SChris Mason 			   btrfs_node_key_ptr_offset(mid),
2731123abc88SChris Mason 			   (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
27325f39d397SChris Mason 	btrfs_set_header_nritems(split, c_nritems - mid);
27335f39d397SChris Mason 	btrfs_set_header_nritems(c, mid);
2734aa5d6bedSChris Mason 
27355f39d397SChris Mason 	btrfs_mark_buffer_dirty(c);
27365f39d397SChris Mason 	btrfs_mark_buffer_dirty(split);
27375f39d397SChris Mason 
27386ad3cf6dSDavid Sterba 	insert_ptr(trans, path, &disk_key, split->start,
2739c3e06965SJan Schmidt 		   path->slots[level + 1] + 1, level + 1);
2740aa5d6bedSChris Mason 
27415de08d7dSChris Mason 	if (path->slots[level] >= mid) {
27425c680ed6SChris Mason 		path->slots[level] -= mid;
2743925baeddSChris Mason 		btrfs_tree_unlock(c);
27445f39d397SChris Mason 		free_extent_buffer(c);
27455f39d397SChris Mason 		path->nodes[level] = split;
27465c680ed6SChris Mason 		path->slots[level + 1] += 1;
2747eb60ceacSChris Mason 	} else {
2748925baeddSChris Mason 		btrfs_tree_unlock(split);
27495f39d397SChris Mason 		free_extent_buffer(split);
2750be0e5c09SChris Mason 	}
2751d5286a92SNikolay Borisov 	return 0;
2752be0e5c09SChris Mason }
2753be0e5c09SChris Mason 
275474123bd7SChris Mason /*
275574123bd7SChris Mason  * how many bytes are required to store the items in a leaf.  start
275674123bd7SChris Mason  * and nr indicate which items in the leaf to check.  This totals up the
275774123bd7SChris Mason  * space used both by the item structs and the item data
275874123bd7SChris Mason  */
27595f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr)
2760be0e5c09SChris Mason {
2761be0e5c09SChris Mason 	int data_len;
27625f39d397SChris Mason 	int nritems = btrfs_header_nritems(l);
2763d4dbff95SChris Mason 	int end = min(nritems, start + nr) - 1;
2764be0e5c09SChris Mason 
2765be0e5c09SChris Mason 	if (!nr)
2766be0e5c09SChris Mason 		return 0;
27673212fa14SJosef Bacik 	data_len = btrfs_item_offset(l, start) + btrfs_item_size(l, start);
27683212fa14SJosef Bacik 	data_len = data_len - btrfs_item_offset(l, end);
27690783fcfcSChris Mason 	data_len += sizeof(struct btrfs_item) * nr;
2770d4dbff95SChris Mason 	WARN_ON(data_len < 0);
2771be0e5c09SChris Mason 	return data_len;
2772be0e5c09SChris Mason }
2773be0e5c09SChris Mason 
277474123bd7SChris Mason /*
2775d4dbff95SChris Mason  * The space between the end of the leaf items and
2776d4dbff95SChris Mason  * the start of the leaf data.  IOW, how much room
2777d4dbff95SChris Mason  * the leaf has left for both items and data
2778d4dbff95SChris Mason  */
2779e902baacSDavid Sterba noinline int btrfs_leaf_free_space(struct extent_buffer *leaf)
2780d4dbff95SChris Mason {
2781e902baacSDavid Sterba 	struct btrfs_fs_info *fs_info = leaf->fs_info;
27825f39d397SChris Mason 	int nritems = btrfs_header_nritems(leaf);
27835f39d397SChris Mason 	int ret;
27840b246afaSJeff Mahoney 
27850b246afaSJeff Mahoney 	ret = BTRFS_LEAF_DATA_SIZE(fs_info) - leaf_space_used(leaf, 0, nritems);
27865f39d397SChris Mason 	if (ret < 0) {
27870b246afaSJeff Mahoney 		btrfs_crit(fs_info,
2788efe120a0SFrank Holton 			   "leaf free space ret %d, leaf data size %lu, used %d nritems %d",
2789da17066cSJeff Mahoney 			   ret,
27900b246afaSJeff Mahoney 			   (unsigned long) BTRFS_LEAF_DATA_SIZE(fs_info),
27915f39d397SChris Mason 			   leaf_space_used(leaf, 0, nritems), nritems);
27925f39d397SChris Mason 	}
27935f39d397SChris Mason 	return ret;
2794d4dbff95SChris Mason }
2795d4dbff95SChris Mason 
279699d8f83cSChris Mason /*
279799d8f83cSChris Mason  * min slot controls the lowest index we're willing to push to the
279899d8f83cSChris Mason  * right.  We'll push up to and including min_slot, but no lower
279999d8f83cSChris Mason  */
2800f72f0010SDavid Sterba static noinline int __push_leaf_right(struct btrfs_path *path,
280144871b1bSChris Mason 				      int data_size, int empty,
280244871b1bSChris Mason 				      struct extent_buffer *right,
280399d8f83cSChris Mason 				      int free_space, u32 left_nritems,
280499d8f83cSChris Mason 				      u32 min_slot)
280500ec4c51SChris Mason {
2806f72f0010SDavid Sterba 	struct btrfs_fs_info *fs_info = right->fs_info;
28075f39d397SChris Mason 	struct extent_buffer *left = path->nodes[0];
280844871b1bSChris Mason 	struct extent_buffer *upper = path->nodes[1];
2809cfed81a0SChris Mason 	struct btrfs_map_token token;
28105f39d397SChris Mason 	struct btrfs_disk_key disk_key;
281100ec4c51SChris Mason 	int slot;
281234a38218SChris Mason 	u32 i;
281300ec4c51SChris Mason 	int push_space = 0;
281400ec4c51SChris Mason 	int push_items = 0;
281534a38218SChris Mason 	u32 nr;
28167518a238SChris Mason 	u32 right_nritems;
28175f39d397SChris Mason 	u32 data_end;
2818db94535dSChris Mason 	u32 this_item_size;
281900ec4c51SChris Mason 
282034a38218SChris Mason 	if (empty)
282134a38218SChris Mason 		nr = 0;
282234a38218SChris Mason 	else
282399d8f83cSChris Mason 		nr = max_t(u32, 1, min_slot);
282434a38218SChris Mason 
282531840ae1SZheng Yan 	if (path->slots[0] >= left_nritems)
282687b29b20SYan Zheng 		push_space += data_size;
282731840ae1SZheng Yan 
282844871b1bSChris Mason 	slot = path->slots[1];
282934a38218SChris Mason 	i = left_nritems - 1;
283034a38218SChris Mason 	while (i >= nr) {
283131840ae1SZheng Yan 		if (!empty && push_items > 0) {
283231840ae1SZheng Yan 			if (path->slots[0] > i)
283331840ae1SZheng Yan 				break;
283431840ae1SZheng Yan 			if (path->slots[0] == i) {
2835e902baacSDavid Sterba 				int space = btrfs_leaf_free_space(left);
2836e902baacSDavid Sterba 
283731840ae1SZheng Yan 				if (space + push_space * 2 > free_space)
283831840ae1SZheng Yan 					break;
283931840ae1SZheng Yan 			}
284031840ae1SZheng Yan 		}
284131840ae1SZheng Yan 
284200ec4c51SChris Mason 		if (path->slots[0] == i)
284387b29b20SYan Zheng 			push_space += data_size;
2844db94535dSChris Mason 
28453212fa14SJosef Bacik 		this_item_size = btrfs_item_size(left, i);
284674794207SJosef Bacik 		if (this_item_size + sizeof(struct btrfs_item) +
284774794207SJosef Bacik 		    push_space > free_space)
284800ec4c51SChris Mason 			break;
284931840ae1SZheng Yan 
285000ec4c51SChris Mason 		push_items++;
285174794207SJosef Bacik 		push_space += this_item_size + sizeof(struct btrfs_item);
285234a38218SChris Mason 		if (i == 0)
285334a38218SChris Mason 			break;
285434a38218SChris Mason 		i--;
2855db94535dSChris Mason 	}
28565f39d397SChris Mason 
2857925baeddSChris Mason 	if (push_items == 0)
2858925baeddSChris Mason 		goto out_unlock;
28595f39d397SChris Mason 
28606c1500f2SJulia Lawall 	WARN_ON(!empty && push_items == left_nritems);
28615f39d397SChris Mason 
286200ec4c51SChris Mason 	/* push left to right */
28635f39d397SChris Mason 	right_nritems = btrfs_header_nritems(right);
286434a38218SChris Mason 
2865dc2e724eSJosef Bacik 	push_space = btrfs_item_data_end(left, left_nritems - push_items);
28668f881e8cSDavid Sterba 	push_space -= leaf_data_end(left);
28675f39d397SChris Mason 
286800ec4c51SChris Mason 	/* make room in the right data area */
28698f881e8cSDavid Sterba 	data_end = leaf_data_end(right);
28705f39d397SChris Mason 	memmove_extent_buffer(right,
28713d9ec8c4SNikolay Borisov 			      BTRFS_LEAF_DATA_OFFSET + data_end - push_space,
28723d9ec8c4SNikolay Borisov 			      BTRFS_LEAF_DATA_OFFSET + data_end,
28730b246afaSJeff Mahoney 			      BTRFS_LEAF_DATA_SIZE(fs_info) - data_end);
28745f39d397SChris Mason 
287500ec4c51SChris Mason 	/* copy from the left data area */
28763d9ec8c4SNikolay Borisov 	copy_extent_buffer(right, left, BTRFS_LEAF_DATA_OFFSET +
28770b246afaSJeff Mahoney 		     BTRFS_LEAF_DATA_SIZE(fs_info) - push_space,
28788f881e8cSDavid Sterba 		     BTRFS_LEAF_DATA_OFFSET + leaf_data_end(left),
2879d6025579SChris Mason 		     push_space);
28805f39d397SChris Mason 
28815f39d397SChris Mason 	memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
28825f39d397SChris Mason 			      btrfs_item_nr_offset(0),
28830783fcfcSChris Mason 			      right_nritems * sizeof(struct btrfs_item));
28845f39d397SChris Mason 
288500ec4c51SChris Mason 	/* copy the items from left to right */
28865f39d397SChris Mason 	copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
28875f39d397SChris Mason 		   btrfs_item_nr_offset(left_nritems - push_items),
28880783fcfcSChris Mason 		   push_items * sizeof(struct btrfs_item));
288900ec4c51SChris Mason 
289000ec4c51SChris Mason 	/* update the item pointers */
2891c82f823cSDavid Sterba 	btrfs_init_map_token(&token, right);
28927518a238SChris Mason 	right_nritems += push_items;
28935f39d397SChris Mason 	btrfs_set_header_nritems(right, right_nritems);
28940b246afaSJeff Mahoney 	push_space = BTRFS_LEAF_DATA_SIZE(fs_info);
28957518a238SChris Mason 	for (i = 0; i < right_nritems; i++) {
28963212fa14SJosef Bacik 		push_space -= btrfs_token_item_size(&token, i);
28973212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, i, push_space);
2898db94535dSChris Mason 	}
2899db94535dSChris Mason 
29007518a238SChris Mason 	left_nritems -= push_items;
29015f39d397SChris Mason 	btrfs_set_header_nritems(left, left_nritems);
290200ec4c51SChris Mason 
290334a38218SChris Mason 	if (left_nritems)
29045f39d397SChris Mason 		btrfs_mark_buffer_dirty(left);
2905f0486c68SYan, Zheng 	else
29066a884d7dSDavid Sterba 		btrfs_clean_tree_block(left);
2907f0486c68SYan, Zheng 
29085f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
2909a429e513SChris Mason 
29105f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
29115f39d397SChris Mason 	btrfs_set_node_key(upper, &disk_key, slot + 1);
2912d6025579SChris Mason 	btrfs_mark_buffer_dirty(upper);
291302217ed2SChris Mason 
291400ec4c51SChris Mason 	/* then fixup the leaf pointer in the path */
29157518a238SChris Mason 	if (path->slots[0] >= left_nritems) {
29167518a238SChris Mason 		path->slots[0] -= left_nritems;
2917925baeddSChris Mason 		if (btrfs_header_nritems(path->nodes[0]) == 0)
29186a884d7dSDavid Sterba 			btrfs_clean_tree_block(path->nodes[0]);
2919925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
29205f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
29215f39d397SChris Mason 		path->nodes[0] = right;
292200ec4c51SChris Mason 		path->slots[1] += 1;
292300ec4c51SChris Mason 	} else {
2924925baeddSChris Mason 		btrfs_tree_unlock(right);
29255f39d397SChris Mason 		free_extent_buffer(right);
292600ec4c51SChris Mason 	}
292700ec4c51SChris Mason 	return 0;
2928925baeddSChris Mason 
2929925baeddSChris Mason out_unlock:
2930925baeddSChris Mason 	btrfs_tree_unlock(right);
2931925baeddSChris Mason 	free_extent_buffer(right);
2932925baeddSChris Mason 	return 1;
293300ec4c51SChris Mason }
2934925baeddSChris Mason 
293500ec4c51SChris Mason /*
293644871b1bSChris Mason  * push some data in the path leaf to the right, trying to free up at
293774123bd7SChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
293844871b1bSChris Mason  *
293944871b1bSChris Mason  * returns 1 if the push failed because the other node didn't have enough
294044871b1bSChris Mason  * room, 0 if everything worked out and < 0 if there were major errors.
294199d8f83cSChris Mason  *
294299d8f83cSChris Mason  * this will push starting from min_slot to the end of the leaf.  It won't
294399d8f83cSChris Mason  * push any slot lower than min_slot
294474123bd7SChris Mason  */
294544871b1bSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
294699d8f83cSChris Mason 			   *root, struct btrfs_path *path,
294799d8f83cSChris Mason 			   int min_data_size, int data_size,
294899d8f83cSChris Mason 			   int empty, u32 min_slot)
2949be0e5c09SChris Mason {
295044871b1bSChris Mason 	struct extent_buffer *left = path->nodes[0];
295144871b1bSChris Mason 	struct extent_buffer *right;
295244871b1bSChris Mason 	struct extent_buffer *upper;
295344871b1bSChris Mason 	int slot;
295444871b1bSChris Mason 	int free_space;
295544871b1bSChris Mason 	u32 left_nritems;
295644871b1bSChris Mason 	int ret;
295744871b1bSChris Mason 
295844871b1bSChris Mason 	if (!path->nodes[1])
295944871b1bSChris Mason 		return 1;
296044871b1bSChris Mason 
296144871b1bSChris Mason 	slot = path->slots[1];
296244871b1bSChris Mason 	upper = path->nodes[1];
296344871b1bSChris Mason 	if (slot >= btrfs_header_nritems(upper) - 1)
296444871b1bSChris Mason 		return 1;
296544871b1bSChris Mason 
296649d0c642SFilipe Manana 	btrfs_assert_tree_write_locked(path->nodes[1]);
296744871b1bSChris Mason 
29684b231ae4SDavid Sterba 	right = btrfs_read_node_slot(upper, slot + 1);
2969fb770ae4SLiu Bo 	/*
2970fb770ae4SLiu Bo 	 * slot + 1 is not valid or we fail to read the right node,
2971fb770ae4SLiu Bo 	 * no big deal, just return.
2972fb770ae4SLiu Bo 	 */
2973fb770ae4SLiu Bo 	if (IS_ERR(right))
297491ca338dSTsutomu Itoh 		return 1;
297591ca338dSTsutomu Itoh 
2976bf77467aSJosef Bacik 	__btrfs_tree_lock(right, BTRFS_NESTING_RIGHT);
297744871b1bSChris Mason 
2978e902baacSDavid Sterba 	free_space = btrfs_leaf_free_space(right);
297944871b1bSChris Mason 	if (free_space < data_size)
298044871b1bSChris Mason 		goto out_unlock;
298144871b1bSChris Mason 
298244871b1bSChris Mason 	/* cow and double check */
298344871b1bSChris Mason 	ret = btrfs_cow_block(trans, root, right, upper,
2984bf59a5a2SJosef Bacik 			      slot + 1, &right, BTRFS_NESTING_RIGHT_COW);
298544871b1bSChris Mason 	if (ret)
298644871b1bSChris Mason 		goto out_unlock;
298744871b1bSChris Mason 
2988e902baacSDavid Sterba 	free_space = btrfs_leaf_free_space(right);
298944871b1bSChris Mason 	if (free_space < data_size)
299044871b1bSChris Mason 		goto out_unlock;
299144871b1bSChris Mason 
299244871b1bSChris Mason 	left_nritems = btrfs_header_nritems(left);
299344871b1bSChris Mason 	if (left_nritems == 0)
299444871b1bSChris Mason 		goto out_unlock;
299544871b1bSChris Mason 
2996d16c702fSQu Wenruo 	if (check_sibling_keys(left, right)) {
2997d16c702fSQu Wenruo 		ret = -EUCLEAN;
2998d16c702fSQu Wenruo 		btrfs_tree_unlock(right);
2999d16c702fSQu Wenruo 		free_extent_buffer(right);
3000d16c702fSQu Wenruo 		return ret;
3001d16c702fSQu Wenruo 	}
30022ef1fed2SFilipe David Borba Manana 	if (path->slots[0] == left_nritems && !empty) {
30032ef1fed2SFilipe David Borba Manana 		/* Key greater than all keys in the leaf, right neighbor has
30042ef1fed2SFilipe David Borba Manana 		 * enough room for it and we're not emptying our leaf to delete
30052ef1fed2SFilipe David Borba Manana 		 * it, therefore use right neighbor to insert the new item and
300652042d8eSAndrea Gelmini 		 * no need to touch/dirty our left leaf. */
30072ef1fed2SFilipe David Borba Manana 		btrfs_tree_unlock(left);
30082ef1fed2SFilipe David Borba Manana 		free_extent_buffer(left);
30092ef1fed2SFilipe David Borba Manana 		path->nodes[0] = right;
30102ef1fed2SFilipe David Borba Manana 		path->slots[0] = 0;
30112ef1fed2SFilipe David Borba Manana 		path->slots[1]++;
30122ef1fed2SFilipe David Borba Manana 		return 0;
30132ef1fed2SFilipe David Borba Manana 	}
30142ef1fed2SFilipe David Borba Manana 
3015f72f0010SDavid Sterba 	return __push_leaf_right(path, min_data_size, empty,
301699d8f83cSChris Mason 				right, free_space, left_nritems, min_slot);
301744871b1bSChris Mason out_unlock:
301844871b1bSChris Mason 	btrfs_tree_unlock(right);
301944871b1bSChris Mason 	free_extent_buffer(right);
302044871b1bSChris Mason 	return 1;
302144871b1bSChris Mason }
302244871b1bSChris Mason 
302344871b1bSChris Mason /*
302444871b1bSChris Mason  * push some data in the path leaf to the left, trying to free up at
302544871b1bSChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
302699d8f83cSChris Mason  *
302799d8f83cSChris Mason  * max_slot can put a limit on how far into the leaf we'll push items.  The
302899d8f83cSChris Mason  * item at 'max_slot' won't be touched.  Use (u32)-1 to make us do all the
302999d8f83cSChris Mason  * items
303044871b1bSChris Mason  */
30318087c193SDavid Sterba static noinline int __push_leaf_left(struct btrfs_path *path, int data_size,
303244871b1bSChris Mason 				     int empty, struct extent_buffer *left,
303399d8f83cSChris Mason 				     int free_space, u32 right_nritems,
303499d8f83cSChris Mason 				     u32 max_slot)
303544871b1bSChris Mason {
30368087c193SDavid Sterba 	struct btrfs_fs_info *fs_info = left->fs_info;
30375f39d397SChris Mason 	struct btrfs_disk_key disk_key;
30385f39d397SChris Mason 	struct extent_buffer *right = path->nodes[0];
3039be0e5c09SChris Mason 	int i;
3040be0e5c09SChris Mason 	int push_space = 0;
3041be0e5c09SChris Mason 	int push_items = 0;
30427518a238SChris Mason 	u32 old_left_nritems;
304334a38218SChris Mason 	u32 nr;
3044aa5d6bedSChris Mason 	int ret = 0;
3045db94535dSChris Mason 	u32 this_item_size;
3046db94535dSChris Mason 	u32 old_left_item_size;
3047cfed81a0SChris Mason 	struct btrfs_map_token token;
3048cfed81a0SChris Mason 
304934a38218SChris Mason 	if (empty)
305099d8f83cSChris Mason 		nr = min(right_nritems, max_slot);
305134a38218SChris Mason 	else
305299d8f83cSChris Mason 		nr = min(right_nritems - 1, max_slot);
305334a38218SChris Mason 
305434a38218SChris Mason 	for (i = 0; i < nr; i++) {
305531840ae1SZheng Yan 		if (!empty && push_items > 0) {
305631840ae1SZheng Yan 			if (path->slots[0] < i)
305731840ae1SZheng Yan 				break;
305831840ae1SZheng Yan 			if (path->slots[0] == i) {
3059e902baacSDavid Sterba 				int space = btrfs_leaf_free_space(right);
3060e902baacSDavid Sterba 
306131840ae1SZheng Yan 				if (space + push_space * 2 > free_space)
306231840ae1SZheng Yan 					break;
306331840ae1SZheng Yan 			}
306431840ae1SZheng Yan 		}
306531840ae1SZheng Yan 
3066be0e5c09SChris Mason 		if (path->slots[0] == i)
306787b29b20SYan Zheng 			push_space += data_size;
3068db94535dSChris Mason 
30693212fa14SJosef Bacik 		this_item_size = btrfs_item_size(right, i);
307074794207SJosef Bacik 		if (this_item_size + sizeof(struct btrfs_item) + push_space >
307174794207SJosef Bacik 		    free_space)
3072be0e5c09SChris Mason 			break;
3073db94535dSChris Mason 
3074be0e5c09SChris Mason 		push_items++;
307574794207SJosef Bacik 		push_space += this_item_size + sizeof(struct btrfs_item);
3076be0e5c09SChris Mason 	}
3077db94535dSChris Mason 
3078be0e5c09SChris Mason 	if (push_items == 0) {
3079925baeddSChris Mason 		ret = 1;
3080925baeddSChris Mason 		goto out;
3081be0e5c09SChris Mason 	}
3082fae7f21cSDulshani Gunawardhana 	WARN_ON(!empty && push_items == btrfs_header_nritems(right));
30835f39d397SChris Mason 
3084be0e5c09SChris Mason 	/* push data from right to left */
30855f39d397SChris Mason 	copy_extent_buffer(left, right,
30865f39d397SChris Mason 			   btrfs_item_nr_offset(btrfs_header_nritems(left)),
30875f39d397SChris Mason 			   btrfs_item_nr_offset(0),
30885f39d397SChris Mason 			   push_items * sizeof(struct btrfs_item));
30895f39d397SChris Mason 
30900b246afaSJeff Mahoney 	push_space = BTRFS_LEAF_DATA_SIZE(fs_info) -
30913212fa14SJosef Bacik 		     btrfs_item_offset(right, push_items - 1);
30925f39d397SChris Mason 
30933d9ec8c4SNikolay Borisov 	copy_extent_buffer(left, right, BTRFS_LEAF_DATA_OFFSET +
30948f881e8cSDavid Sterba 		     leaf_data_end(left) - push_space,
30953d9ec8c4SNikolay Borisov 		     BTRFS_LEAF_DATA_OFFSET +
30963212fa14SJosef Bacik 		     btrfs_item_offset(right, push_items - 1),
3097be0e5c09SChris Mason 		     push_space);
30985f39d397SChris Mason 	old_left_nritems = btrfs_header_nritems(left);
309987b29b20SYan Zheng 	BUG_ON(old_left_nritems <= 0);
3100eb60ceacSChris Mason 
3101c82f823cSDavid Sterba 	btrfs_init_map_token(&token, left);
31023212fa14SJosef Bacik 	old_left_item_size = btrfs_item_offset(left, old_left_nritems - 1);
3103be0e5c09SChris Mason 	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
31045f39d397SChris Mason 		u32 ioff;
3105db94535dSChris Mason 
31063212fa14SJosef Bacik 		ioff = btrfs_token_item_offset(&token, i);
31073212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, i,
3108cc4c13d5SDavid Sterba 		      ioff - (BTRFS_LEAF_DATA_SIZE(fs_info) - old_left_item_size));
3109be0e5c09SChris Mason 	}
31105f39d397SChris Mason 	btrfs_set_header_nritems(left, old_left_nritems + push_items);
3111be0e5c09SChris Mason 
3112be0e5c09SChris Mason 	/* fixup right node */
311331b1a2bdSJulia Lawall 	if (push_items > right_nritems)
311431b1a2bdSJulia Lawall 		WARN(1, KERN_CRIT "push items %d nr %u\n", push_items,
3115d397712bSChris Mason 		       right_nritems);
311634a38218SChris Mason 
311734a38218SChris Mason 	if (push_items < right_nritems) {
31183212fa14SJosef Bacik 		push_space = btrfs_item_offset(right, push_items - 1) -
31198f881e8cSDavid Sterba 						  leaf_data_end(right);
31203d9ec8c4SNikolay Borisov 		memmove_extent_buffer(right, BTRFS_LEAF_DATA_OFFSET +
31210b246afaSJeff Mahoney 				      BTRFS_LEAF_DATA_SIZE(fs_info) - push_space,
31223d9ec8c4SNikolay Borisov 				      BTRFS_LEAF_DATA_OFFSET +
31238f881e8cSDavid Sterba 				      leaf_data_end(right), push_space);
31245f39d397SChris Mason 
31255f39d397SChris Mason 		memmove_extent_buffer(right, btrfs_item_nr_offset(0),
31265f39d397SChris Mason 			      btrfs_item_nr_offset(push_items),
31275f39d397SChris Mason 			     (btrfs_header_nritems(right) - push_items) *
31280783fcfcSChris Mason 			     sizeof(struct btrfs_item));
312934a38218SChris Mason 	}
3130c82f823cSDavid Sterba 
3131c82f823cSDavid Sterba 	btrfs_init_map_token(&token, right);
3132eef1c494SYan 	right_nritems -= push_items;
3133eef1c494SYan 	btrfs_set_header_nritems(right, right_nritems);
31340b246afaSJeff Mahoney 	push_space = BTRFS_LEAF_DATA_SIZE(fs_info);
31355f39d397SChris Mason 	for (i = 0; i < right_nritems; i++) {
31363212fa14SJosef Bacik 		push_space = push_space - btrfs_token_item_size(&token, i);
31373212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, i, push_space);
3138db94535dSChris Mason 	}
3139eb60ceacSChris Mason 
31405f39d397SChris Mason 	btrfs_mark_buffer_dirty(left);
314134a38218SChris Mason 	if (right_nritems)
31425f39d397SChris Mason 		btrfs_mark_buffer_dirty(right);
3143f0486c68SYan, Zheng 	else
31446a884d7dSDavid Sterba 		btrfs_clean_tree_block(right);
3145098f59c2SChris Mason 
31465f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
3147b167fa91SNikolay Borisov 	fixup_low_keys(path, &disk_key, 1);
3148be0e5c09SChris Mason 
3149be0e5c09SChris Mason 	/* then fixup the leaf pointer in the path */
3150be0e5c09SChris Mason 	if (path->slots[0] < push_items) {
3151be0e5c09SChris Mason 		path->slots[0] += old_left_nritems;
3152925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
31535f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
31545f39d397SChris Mason 		path->nodes[0] = left;
3155be0e5c09SChris Mason 		path->slots[1] -= 1;
3156be0e5c09SChris Mason 	} else {
3157925baeddSChris Mason 		btrfs_tree_unlock(left);
31585f39d397SChris Mason 		free_extent_buffer(left);
3159be0e5c09SChris Mason 		path->slots[0] -= push_items;
3160be0e5c09SChris Mason 	}
3161eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
3162aa5d6bedSChris Mason 	return ret;
3163925baeddSChris Mason out:
3164925baeddSChris Mason 	btrfs_tree_unlock(left);
3165925baeddSChris Mason 	free_extent_buffer(left);
3166925baeddSChris Mason 	return ret;
3167be0e5c09SChris Mason }
3168be0e5c09SChris Mason 
316974123bd7SChris Mason /*
317044871b1bSChris Mason  * push some data in the path leaf to the left, trying to free up at
317144871b1bSChris Mason  * least data_size bytes.  returns zero if the push worked, nonzero otherwise
317299d8f83cSChris Mason  *
317399d8f83cSChris Mason  * max_slot can put a limit on how far into the leaf we'll push items.  The
317499d8f83cSChris Mason  * item at 'max_slot' won't be touched.  Use (u32)-1 to make us push all the
317599d8f83cSChris Mason  * items
317644871b1bSChris Mason  */
317744871b1bSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
317899d8f83cSChris Mason 			  *root, struct btrfs_path *path, int min_data_size,
317999d8f83cSChris Mason 			  int data_size, int empty, u32 max_slot)
318044871b1bSChris Mason {
318144871b1bSChris Mason 	struct extent_buffer *right = path->nodes[0];
318244871b1bSChris Mason 	struct extent_buffer *left;
318344871b1bSChris Mason 	int slot;
318444871b1bSChris Mason 	int free_space;
318544871b1bSChris Mason 	u32 right_nritems;
318644871b1bSChris Mason 	int ret = 0;
318744871b1bSChris Mason 
318844871b1bSChris Mason 	slot = path->slots[1];
318944871b1bSChris Mason 	if (slot == 0)
319044871b1bSChris Mason 		return 1;
319144871b1bSChris Mason 	if (!path->nodes[1])
319244871b1bSChris Mason 		return 1;
319344871b1bSChris Mason 
319444871b1bSChris Mason 	right_nritems = btrfs_header_nritems(right);
319544871b1bSChris Mason 	if (right_nritems == 0)
319644871b1bSChris Mason 		return 1;
319744871b1bSChris Mason 
319849d0c642SFilipe Manana 	btrfs_assert_tree_write_locked(path->nodes[1]);
319944871b1bSChris Mason 
32004b231ae4SDavid Sterba 	left = btrfs_read_node_slot(path->nodes[1], slot - 1);
3201fb770ae4SLiu Bo 	/*
3202fb770ae4SLiu Bo 	 * slot - 1 is not valid or we fail to read the left node,
3203fb770ae4SLiu Bo 	 * no big deal, just return.
3204fb770ae4SLiu Bo 	 */
3205fb770ae4SLiu Bo 	if (IS_ERR(left))
320691ca338dSTsutomu Itoh 		return 1;
320791ca338dSTsutomu Itoh 
3208bf77467aSJosef Bacik 	__btrfs_tree_lock(left, BTRFS_NESTING_LEFT);
320944871b1bSChris Mason 
3210e902baacSDavid Sterba 	free_space = btrfs_leaf_free_space(left);
321144871b1bSChris Mason 	if (free_space < data_size) {
321244871b1bSChris Mason 		ret = 1;
321344871b1bSChris Mason 		goto out;
321444871b1bSChris Mason 	}
321544871b1bSChris Mason 
321644871b1bSChris Mason 	/* cow and double check */
321744871b1bSChris Mason 	ret = btrfs_cow_block(trans, root, left,
32189631e4ccSJosef Bacik 			      path->nodes[1], slot - 1, &left,
3219bf59a5a2SJosef Bacik 			      BTRFS_NESTING_LEFT_COW);
322044871b1bSChris Mason 	if (ret) {
322144871b1bSChris Mason 		/* we hit -ENOSPC, but it isn't fatal here */
322279787eaaSJeff Mahoney 		if (ret == -ENOSPC)
322344871b1bSChris Mason 			ret = 1;
322444871b1bSChris Mason 		goto out;
322544871b1bSChris Mason 	}
322644871b1bSChris Mason 
3227e902baacSDavid Sterba 	free_space = btrfs_leaf_free_space(left);
322844871b1bSChris Mason 	if (free_space < data_size) {
322944871b1bSChris Mason 		ret = 1;
323044871b1bSChris Mason 		goto out;
323144871b1bSChris Mason 	}
323244871b1bSChris Mason 
3233d16c702fSQu Wenruo 	if (check_sibling_keys(left, right)) {
3234d16c702fSQu Wenruo 		ret = -EUCLEAN;
3235d16c702fSQu Wenruo 		goto out;
3236d16c702fSQu Wenruo 	}
32378087c193SDavid Sterba 	return __push_leaf_left(path, min_data_size,
323899d8f83cSChris Mason 			       empty, left, free_space, right_nritems,
323999d8f83cSChris Mason 			       max_slot);
324044871b1bSChris Mason out:
324144871b1bSChris Mason 	btrfs_tree_unlock(left);
324244871b1bSChris Mason 	free_extent_buffer(left);
324344871b1bSChris Mason 	return ret;
324444871b1bSChris Mason }
324544871b1bSChris Mason 
324644871b1bSChris Mason /*
324774123bd7SChris Mason  * split the path's leaf in two, making sure there is at least data_size
324874123bd7SChris Mason  * available for the resulting leaf level of the path.
324974123bd7SChris Mason  */
3250143bede5SJeff Mahoney static noinline void copy_for_split(struct btrfs_trans_handle *trans,
325144871b1bSChris Mason 				    struct btrfs_path *path,
325244871b1bSChris Mason 				    struct extent_buffer *l,
325344871b1bSChris Mason 				    struct extent_buffer *right,
325444871b1bSChris Mason 				    int slot, int mid, int nritems)
3255be0e5c09SChris Mason {
325694f94ad9SDavid Sterba 	struct btrfs_fs_info *fs_info = trans->fs_info;
3257be0e5c09SChris Mason 	int data_copy_size;
3258be0e5c09SChris Mason 	int rt_data_off;
3259be0e5c09SChris Mason 	int i;
3260d4dbff95SChris Mason 	struct btrfs_disk_key disk_key;
3261cfed81a0SChris Mason 	struct btrfs_map_token token;
3262cfed81a0SChris Mason 
32635f39d397SChris Mason 	nritems = nritems - mid;
32645f39d397SChris Mason 	btrfs_set_header_nritems(right, nritems);
3265dc2e724eSJosef Bacik 	data_copy_size = btrfs_item_data_end(l, mid) - leaf_data_end(l);
32665f39d397SChris Mason 
32675f39d397SChris Mason 	copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
32685f39d397SChris Mason 			   btrfs_item_nr_offset(mid),
32695f39d397SChris Mason 			   nritems * sizeof(struct btrfs_item));
32705f39d397SChris Mason 
32715f39d397SChris Mason 	copy_extent_buffer(right, l,
32723d9ec8c4SNikolay Borisov 		     BTRFS_LEAF_DATA_OFFSET + BTRFS_LEAF_DATA_SIZE(fs_info) -
32733d9ec8c4SNikolay Borisov 		     data_copy_size, BTRFS_LEAF_DATA_OFFSET +
32748f881e8cSDavid Sterba 		     leaf_data_end(l), data_copy_size);
327574123bd7SChris Mason 
3276dc2e724eSJosef Bacik 	rt_data_off = BTRFS_LEAF_DATA_SIZE(fs_info) - btrfs_item_data_end(l, mid);
32775f39d397SChris Mason 
3278c82f823cSDavid Sterba 	btrfs_init_map_token(&token, right);
32795f39d397SChris Mason 	for (i = 0; i < nritems; i++) {
3280db94535dSChris Mason 		u32 ioff;
3281db94535dSChris Mason 
32823212fa14SJosef Bacik 		ioff = btrfs_token_item_offset(&token, i);
32833212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, i, ioff + rt_data_off);
32840783fcfcSChris Mason 	}
328574123bd7SChris Mason 
32865f39d397SChris Mason 	btrfs_set_header_nritems(l, mid);
32875f39d397SChris Mason 	btrfs_item_key(right, &disk_key, 0);
32886ad3cf6dSDavid Sterba 	insert_ptr(trans, path, &disk_key, right->start, path->slots[1] + 1, 1);
32895f39d397SChris Mason 
32905f39d397SChris Mason 	btrfs_mark_buffer_dirty(right);
32915f39d397SChris Mason 	btrfs_mark_buffer_dirty(l);
3292eb60ceacSChris Mason 	BUG_ON(path->slots[0] != slot);
32935f39d397SChris Mason 
3294be0e5c09SChris Mason 	if (mid <= slot) {
3295925baeddSChris Mason 		btrfs_tree_unlock(path->nodes[0]);
32965f39d397SChris Mason 		free_extent_buffer(path->nodes[0]);
32975f39d397SChris Mason 		path->nodes[0] = right;
3298be0e5c09SChris Mason 		path->slots[0] -= mid;
3299be0e5c09SChris Mason 		path->slots[1] += 1;
3300925baeddSChris Mason 	} else {
3301925baeddSChris Mason 		btrfs_tree_unlock(right);
33025f39d397SChris Mason 		free_extent_buffer(right);
3303925baeddSChris Mason 	}
33045f39d397SChris Mason 
3305eb60ceacSChris Mason 	BUG_ON(path->slots[0] < 0);
330644871b1bSChris Mason }
330744871b1bSChris Mason 
330844871b1bSChris Mason /*
330999d8f83cSChris Mason  * double splits happen when we need to insert a big item in the middle
331099d8f83cSChris Mason  * of a leaf.  A double split can leave us with 3 mostly empty leaves:
331199d8f83cSChris Mason  * leaf: [ slots 0 - N] [ our target ] [ N + 1 - total in leaf ]
331299d8f83cSChris Mason  *          A                 B                 C
331399d8f83cSChris Mason  *
331499d8f83cSChris Mason  * We avoid this by trying to push the items on either side of our target
331599d8f83cSChris Mason  * into the adjacent leaves.  If all goes well we can avoid the double split
331699d8f83cSChris Mason  * completely.
331799d8f83cSChris Mason  */
331899d8f83cSChris Mason static noinline int push_for_double_split(struct btrfs_trans_handle *trans,
331999d8f83cSChris Mason 					  struct btrfs_root *root,
332099d8f83cSChris Mason 					  struct btrfs_path *path,
332199d8f83cSChris Mason 					  int data_size)
332299d8f83cSChris Mason {
332399d8f83cSChris Mason 	int ret;
332499d8f83cSChris Mason 	int progress = 0;
332599d8f83cSChris Mason 	int slot;
332699d8f83cSChris Mason 	u32 nritems;
33275a4267caSFilipe David Borba Manana 	int space_needed = data_size;
332899d8f83cSChris Mason 
332999d8f83cSChris Mason 	slot = path->slots[0];
33305a4267caSFilipe David Borba Manana 	if (slot < btrfs_header_nritems(path->nodes[0]))
3331e902baacSDavid Sterba 		space_needed -= btrfs_leaf_free_space(path->nodes[0]);
333299d8f83cSChris Mason 
333399d8f83cSChris Mason 	/*
333499d8f83cSChris Mason 	 * try to push all the items after our slot into the
333599d8f83cSChris Mason 	 * right leaf
333699d8f83cSChris Mason 	 */
33375a4267caSFilipe David Borba Manana 	ret = push_leaf_right(trans, root, path, 1, space_needed, 0, slot);
333899d8f83cSChris Mason 	if (ret < 0)
333999d8f83cSChris Mason 		return ret;
334099d8f83cSChris Mason 
334199d8f83cSChris Mason 	if (ret == 0)
334299d8f83cSChris Mason 		progress++;
334399d8f83cSChris Mason 
334499d8f83cSChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
334599d8f83cSChris Mason 	/*
334699d8f83cSChris Mason 	 * our goal is to get our slot at the start or end of a leaf.  If
334799d8f83cSChris Mason 	 * we've done so we're done
334899d8f83cSChris Mason 	 */
334999d8f83cSChris Mason 	if (path->slots[0] == 0 || path->slots[0] == nritems)
335099d8f83cSChris Mason 		return 0;
335199d8f83cSChris Mason 
3352e902baacSDavid Sterba 	if (btrfs_leaf_free_space(path->nodes[0]) >= data_size)
335399d8f83cSChris Mason 		return 0;
335499d8f83cSChris Mason 
335599d8f83cSChris Mason 	/* try to push all the items before our slot into the next leaf */
335699d8f83cSChris Mason 	slot = path->slots[0];
3357263d3995SFilipe Manana 	space_needed = data_size;
3358263d3995SFilipe Manana 	if (slot > 0)
3359e902baacSDavid Sterba 		space_needed -= btrfs_leaf_free_space(path->nodes[0]);
33605a4267caSFilipe David Borba Manana 	ret = push_leaf_left(trans, root, path, 1, space_needed, 0, slot);
336199d8f83cSChris Mason 	if (ret < 0)
336299d8f83cSChris Mason 		return ret;
336399d8f83cSChris Mason 
336499d8f83cSChris Mason 	if (ret == 0)
336599d8f83cSChris Mason 		progress++;
336699d8f83cSChris Mason 
336799d8f83cSChris Mason 	if (progress)
336899d8f83cSChris Mason 		return 0;
336999d8f83cSChris Mason 	return 1;
337099d8f83cSChris Mason }
337199d8f83cSChris Mason 
337299d8f83cSChris Mason /*
337344871b1bSChris Mason  * split the path's leaf in two, making sure there is at least data_size
337444871b1bSChris Mason  * available for the resulting leaf level of the path.
337544871b1bSChris Mason  *
337644871b1bSChris Mason  * returns 0 if all went well and < 0 on failure.
337744871b1bSChris Mason  */
337844871b1bSChris Mason static noinline int split_leaf(struct btrfs_trans_handle *trans,
337944871b1bSChris Mason 			       struct btrfs_root *root,
3380310712b2SOmar Sandoval 			       const struct btrfs_key *ins_key,
338144871b1bSChris Mason 			       struct btrfs_path *path, int data_size,
338244871b1bSChris Mason 			       int extend)
338344871b1bSChris Mason {
33845d4f98a2SYan Zheng 	struct btrfs_disk_key disk_key;
338544871b1bSChris Mason 	struct extent_buffer *l;
338644871b1bSChris Mason 	u32 nritems;
338744871b1bSChris Mason 	int mid;
338844871b1bSChris Mason 	int slot;
338944871b1bSChris Mason 	struct extent_buffer *right;
3390b7a0365eSDaniel Dressler 	struct btrfs_fs_info *fs_info = root->fs_info;
339144871b1bSChris Mason 	int ret = 0;
339244871b1bSChris Mason 	int wret;
33935d4f98a2SYan Zheng 	int split;
339444871b1bSChris Mason 	int num_doubles = 0;
339599d8f83cSChris Mason 	int tried_avoid_double = 0;
339644871b1bSChris Mason 
3397a5719521SYan, Zheng 	l = path->nodes[0];
3398a5719521SYan, Zheng 	slot = path->slots[0];
33993212fa14SJosef Bacik 	if (extend && data_size + btrfs_item_size(l, slot) +
34000b246afaSJeff Mahoney 	    sizeof(struct btrfs_item) > BTRFS_LEAF_DATA_SIZE(fs_info))
3401a5719521SYan, Zheng 		return -EOVERFLOW;
3402a5719521SYan, Zheng 
340344871b1bSChris Mason 	/* first try to make some room by pushing left and right */
340433157e05SLiu Bo 	if (data_size && path->nodes[1]) {
34055a4267caSFilipe David Borba Manana 		int space_needed = data_size;
34065a4267caSFilipe David Borba Manana 
34075a4267caSFilipe David Borba Manana 		if (slot < btrfs_header_nritems(l))
3408e902baacSDavid Sterba 			space_needed -= btrfs_leaf_free_space(l);
34095a4267caSFilipe David Borba Manana 
34105a4267caSFilipe David Borba Manana 		wret = push_leaf_right(trans, root, path, space_needed,
34115a4267caSFilipe David Borba Manana 				       space_needed, 0, 0);
341244871b1bSChris Mason 		if (wret < 0)
341344871b1bSChris Mason 			return wret;
341444871b1bSChris Mason 		if (wret) {
3415263d3995SFilipe Manana 			space_needed = data_size;
3416263d3995SFilipe Manana 			if (slot > 0)
3417e902baacSDavid Sterba 				space_needed -= btrfs_leaf_free_space(l);
34185a4267caSFilipe David Borba Manana 			wret = push_leaf_left(trans, root, path, space_needed,
34195a4267caSFilipe David Borba Manana 					      space_needed, 0, (u32)-1);
342044871b1bSChris Mason 			if (wret < 0)
342144871b1bSChris Mason 				return wret;
342244871b1bSChris Mason 		}
342344871b1bSChris Mason 		l = path->nodes[0];
342444871b1bSChris Mason 
342544871b1bSChris Mason 		/* did the pushes work? */
3426e902baacSDavid Sterba 		if (btrfs_leaf_free_space(l) >= data_size)
342744871b1bSChris Mason 			return 0;
342844871b1bSChris Mason 	}
342944871b1bSChris Mason 
343044871b1bSChris Mason 	if (!path->nodes[1]) {
3431fdd99c72SLiu Bo 		ret = insert_new_root(trans, root, path, 1);
343244871b1bSChris Mason 		if (ret)
343344871b1bSChris Mason 			return ret;
343444871b1bSChris Mason 	}
343544871b1bSChris Mason again:
34365d4f98a2SYan Zheng 	split = 1;
343744871b1bSChris Mason 	l = path->nodes[0];
343844871b1bSChris Mason 	slot = path->slots[0];
343944871b1bSChris Mason 	nritems = btrfs_header_nritems(l);
344044871b1bSChris Mason 	mid = (nritems + 1) / 2;
344144871b1bSChris Mason 
34425d4f98a2SYan Zheng 	if (mid <= slot) {
34435d4f98a2SYan Zheng 		if (nritems == 1 ||
34445d4f98a2SYan Zheng 		    leaf_space_used(l, mid, nritems - mid) + data_size >
34450b246afaSJeff Mahoney 			BTRFS_LEAF_DATA_SIZE(fs_info)) {
34465d4f98a2SYan Zheng 			if (slot >= nritems) {
34475d4f98a2SYan Zheng 				split = 0;
34485d4f98a2SYan Zheng 			} else {
34495d4f98a2SYan Zheng 				mid = slot;
34505d4f98a2SYan Zheng 				if (mid != nritems &&
34515d4f98a2SYan Zheng 				    leaf_space_used(l, mid, nritems - mid) +
34520b246afaSJeff Mahoney 				    data_size > BTRFS_LEAF_DATA_SIZE(fs_info)) {
345399d8f83cSChris Mason 					if (data_size && !tried_avoid_double)
345499d8f83cSChris Mason 						goto push_for_double;
34555d4f98a2SYan Zheng 					split = 2;
34565d4f98a2SYan Zheng 				}
34575d4f98a2SYan Zheng 			}
34585d4f98a2SYan Zheng 		}
34595d4f98a2SYan Zheng 	} else {
34605d4f98a2SYan Zheng 		if (leaf_space_used(l, 0, mid) + data_size >
34610b246afaSJeff Mahoney 			BTRFS_LEAF_DATA_SIZE(fs_info)) {
34625d4f98a2SYan Zheng 			if (!extend && data_size && slot == 0) {
34635d4f98a2SYan Zheng 				split = 0;
34645d4f98a2SYan Zheng 			} else if ((extend || !data_size) && slot == 0) {
34655d4f98a2SYan Zheng 				mid = 1;
34665d4f98a2SYan Zheng 			} else {
34675d4f98a2SYan Zheng 				mid = slot;
34685d4f98a2SYan Zheng 				if (mid != nritems &&
34695d4f98a2SYan Zheng 				    leaf_space_used(l, mid, nritems - mid) +
34700b246afaSJeff Mahoney 				    data_size > BTRFS_LEAF_DATA_SIZE(fs_info)) {
347199d8f83cSChris Mason 					if (data_size && !tried_avoid_double)
347299d8f83cSChris Mason 						goto push_for_double;
34735d4f98a2SYan Zheng 					split = 2;
34745d4f98a2SYan Zheng 				}
34755d4f98a2SYan Zheng 			}
34765d4f98a2SYan Zheng 		}
34775d4f98a2SYan Zheng 	}
34785d4f98a2SYan Zheng 
34795d4f98a2SYan Zheng 	if (split == 0)
34805d4f98a2SYan Zheng 		btrfs_cpu_key_to_disk(&disk_key, ins_key);
34815d4f98a2SYan Zheng 	else
34825d4f98a2SYan Zheng 		btrfs_item_key(l, &disk_key, mid);
34835d4f98a2SYan Zheng 
3484ca9d473aSJosef Bacik 	/*
3485ca9d473aSJosef Bacik 	 * We have to about BTRFS_NESTING_NEW_ROOT here if we've done a double
3486ca9d473aSJosef Bacik 	 * split, because we're only allowed to have MAX_LOCKDEP_SUBCLASSES
3487ca9d473aSJosef Bacik 	 * subclasses, which is 8 at the time of this patch, and we've maxed it
3488ca9d473aSJosef Bacik 	 * out.  In the future we could add a
3489ca9d473aSJosef Bacik 	 * BTRFS_NESTING_SPLIT_THE_SPLITTENING if we need to, but for now just
3490ca9d473aSJosef Bacik 	 * use BTRFS_NESTING_NEW_ROOT.
3491ca9d473aSJosef Bacik 	 */
349279bd3712SFilipe Manana 	right = btrfs_alloc_tree_block(trans, root, 0, root->root_key.objectid,
349379bd3712SFilipe Manana 				       &disk_key, 0, l->start, 0,
349479bd3712SFilipe Manana 				       num_doubles ? BTRFS_NESTING_NEW_ROOT :
3495ca9d473aSJosef Bacik 				       BTRFS_NESTING_SPLIT);
3496f0486c68SYan, Zheng 	if (IS_ERR(right))
349744871b1bSChris Mason 		return PTR_ERR(right);
3498f0486c68SYan, Zheng 
34990b246afaSJeff Mahoney 	root_add_used(root, fs_info->nodesize);
350044871b1bSChris Mason 
35015d4f98a2SYan Zheng 	if (split == 0) {
350244871b1bSChris Mason 		if (mid <= slot) {
350344871b1bSChris Mason 			btrfs_set_header_nritems(right, 0);
35046ad3cf6dSDavid Sterba 			insert_ptr(trans, path, &disk_key,
35052ff7e61eSJeff Mahoney 				   right->start, path->slots[1] + 1, 1);
350644871b1bSChris Mason 			btrfs_tree_unlock(path->nodes[0]);
350744871b1bSChris Mason 			free_extent_buffer(path->nodes[0]);
350844871b1bSChris Mason 			path->nodes[0] = right;
350944871b1bSChris Mason 			path->slots[0] = 0;
351044871b1bSChris Mason 			path->slots[1] += 1;
351144871b1bSChris Mason 		} else {
351244871b1bSChris Mason 			btrfs_set_header_nritems(right, 0);
35136ad3cf6dSDavid Sterba 			insert_ptr(trans, path, &disk_key,
35142ff7e61eSJeff Mahoney 				   right->start, path->slots[1], 1);
351544871b1bSChris Mason 			btrfs_tree_unlock(path->nodes[0]);
351644871b1bSChris Mason 			free_extent_buffer(path->nodes[0]);
351744871b1bSChris Mason 			path->nodes[0] = right;
351844871b1bSChris Mason 			path->slots[0] = 0;
3519143bede5SJeff Mahoney 			if (path->slots[1] == 0)
3520b167fa91SNikolay Borisov 				fixup_low_keys(path, &disk_key, 1);
35215d4f98a2SYan Zheng 		}
3522196e0249SLiu Bo 		/*
3523196e0249SLiu Bo 		 * We create a new leaf 'right' for the required ins_len and
3524196e0249SLiu Bo 		 * we'll do btrfs_mark_buffer_dirty() on this leaf after copying
3525196e0249SLiu Bo 		 * the content of ins_len to 'right'.
3526196e0249SLiu Bo 		 */
352744871b1bSChris Mason 		return ret;
352844871b1bSChris Mason 	}
352944871b1bSChris Mason 
353094f94ad9SDavid Sterba 	copy_for_split(trans, path, l, right, slot, mid, nritems);
353144871b1bSChris Mason 
35325d4f98a2SYan Zheng 	if (split == 2) {
3533cc0c5538SChris Mason 		BUG_ON(num_doubles != 0);
3534cc0c5538SChris Mason 		num_doubles++;
3535cc0c5538SChris Mason 		goto again;
35363326d1b0SChris Mason 	}
353744871b1bSChris Mason 
3538143bede5SJeff Mahoney 	return 0;
353999d8f83cSChris Mason 
354099d8f83cSChris Mason push_for_double:
354199d8f83cSChris Mason 	push_for_double_split(trans, root, path, data_size);
354299d8f83cSChris Mason 	tried_avoid_double = 1;
3543e902baacSDavid Sterba 	if (btrfs_leaf_free_space(path->nodes[0]) >= data_size)
354499d8f83cSChris Mason 		return 0;
354599d8f83cSChris Mason 	goto again;
3546be0e5c09SChris Mason }
3547be0e5c09SChris Mason 
3548ad48fd75SYan, Zheng static noinline int setup_leaf_for_split(struct btrfs_trans_handle *trans,
3549ad48fd75SYan, Zheng 					 struct btrfs_root *root,
3550ad48fd75SYan, Zheng 					 struct btrfs_path *path, int ins_len)
3551ad48fd75SYan, Zheng {
3552ad48fd75SYan, Zheng 	struct btrfs_key key;
3553ad48fd75SYan, Zheng 	struct extent_buffer *leaf;
3554ad48fd75SYan, Zheng 	struct btrfs_file_extent_item *fi;
3555ad48fd75SYan, Zheng 	u64 extent_len = 0;
3556ad48fd75SYan, Zheng 	u32 item_size;
3557ad48fd75SYan, Zheng 	int ret;
3558ad48fd75SYan, Zheng 
3559ad48fd75SYan, Zheng 	leaf = path->nodes[0];
3560ad48fd75SYan, Zheng 	btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
3561ad48fd75SYan, Zheng 
3562ad48fd75SYan, Zheng 	BUG_ON(key.type != BTRFS_EXTENT_DATA_KEY &&
3563ad48fd75SYan, Zheng 	       key.type != BTRFS_EXTENT_CSUM_KEY);
3564ad48fd75SYan, Zheng 
3565e902baacSDavid Sterba 	if (btrfs_leaf_free_space(leaf) >= ins_len)
3566ad48fd75SYan, Zheng 		return 0;
3567ad48fd75SYan, Zheng 
35683212fa14SJosef Bacik 	item_size = btrfs_item_size(leaf, path->slots[0]);
3569ad48fd75SYan, Zheng 	if (key.type == BTRFS_EXTENT_DATA_KEY) {
3570ad48fd75SYan, Zheng 		fi = btrfs_item_ptr(leaf, path->slots[0],
3571ad48fd75SYan, Zheng 				    struct btrfs_file_extent_item);
3572ad48fd75SYan, Zheng 		extent_len = btrfs_file_extent_num_bytes(leaf, fi);
3573ad48fd75SYan, Zheng 	}
3574b3b4aa74SDavid Sterba 	btrfs_release_path(path);
3575ad48fd75SYan, Zheng 
3576ad48fd75SYan, Zheng 	path->keep_locks = 1;
3577ad48fd75SYan, Zheng 	path->search_for_split = 1;
3578ad48fd75SYan, Zheng 	ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
3579ad48fd75SYan, Zheng 	path->search_for_split = 0;
3580a8df6fe6SFilipe Manana 	if (ret > 0)
3581a8df6fe6SFilipe Manana 		ret = -EAGAIN;
3582ad48fd75SYan, Zheng 	if (ret < 0)
3583ad48fd75SYan, Zheng 		goto err;
3584ad48fd75SYan, Zheng 
3585ad48fd75SYan, Zheng 	ret = -EAGAIN;
3586ad48fd75SYan, Zheng 	leaf = path->nodes[0];
3587a8df6fe6SFilipe Manana 	/* if our item isn't there, return now */
35883212fa14SJosef Bacik 	if (item_size != btrfs_item_size(leaf, path->slots[0]))
3589ad48fd75SYan, Zheng 		goto err;
3590ad48fd75SYan, Zheng 
3591109f6aefSChris Mason 	/* the leaf has  changed, it now has room.  return now */
3592e902baacSDavid Sterba 	if (btrfs_leaf_free_space(path->nodes[0]) >= ins_len)
3593109f6aefSChris Mason 		goto err;
3594109f6aefSChris Mason 
3595ad48fd75SYan, Zheng 	if (key.type == BTRFS_EXTENT_DATA_KEY) {
3596ad48fd75SYan, Zheng 		fi = btrfs_item_ptr(leaf, path->slots[0],
3597ad48fd75SYan, Zheng 				    struct btrfs_file_extent_item);
3598ad48fd75SYan, Zheng 		if (extent_len != btrfs_file_extent_num_bytes(leaf, fi))
3599ad48fd75SYan, Zheng 			goto err;
3600ad48fd75SYan, Zheng 	}
3601ad48fd75SYan, Zheng 
3602ad48fd75SYan, Zheng 	ret = split_leaf(trans, root, &key, path, ins_len, 1);
3603f0486c68SYan, Zheng 	if (ret)
3604f0486c68SYan, Zheng 		goto err;
3605ad48fd75SYan, Zheng 
3606ad48fd75SYan, Zheng 	path->keep_locks = 0;
3607ad48fd75SYan, Zheng 	btrfs_unlock_up_safe(path, 1);
3608ad48fd75SYan, Zheng 	return 0;
3609ad48fd75SYan, Zheng err:
3610ad48fd75SYan, Zheng 	path->keep_locks = 0;
3611ad48fd75SYan, Zheng 	return ret;
3612ad48fd75SYan, Zheng }
3613ad48fd75SYan, Zheng 
361425263cd7SDavid Sterba static noinline int split_item(struct btrfs_path *path,
3615310712b2SOmar Sandoval 			       const struct btrfs_key *new_key,
3616459931ecSChris Mason 			       unsigned long split_offset)
3617459931ecSChris Mason {
3618459931ecSChris Mason 	struct extent_buffer *leaf;
3619c91666b1SJosef Bacik 	int orig_slot, slot;
3620ad48fd75SYan, Zheng 	char *buf;
3621459931ecSChris Mason 	u32 nritems;
3622ad48fd75SYan, Zheng 	u32 item_size;
3623459931ecSChris Mason 	u32 orig_offset;
3624459931ecSChris Mason 	struct btrfs_disk_key disk_key;
3625459931ecSChris Mason 
3626459931ecSChris Mason 	leaf = path->nodes[0];
3627e902baacSDavid Sterba 	BUG_ON(btrfs_leaf_free_space(leaf) < sizeof(struct btrfs_item));
3628b9473439SChris Mason 
3629c91666b1SJosef Bacik 	orig_slot = path->slots[0];
36303212fa14SJosef Bacik 	orig_offset = btrfs_item_offset(leaf, path->slots[0]);
36313212fa14SJosef Bacik 	item_size = btrfs_item_size(leaf, path->slots[0]);
3632459931ecSChris Mason 
3633459931ecSChris Mason 	buf = kmalloc(item_size, GFP_NOFS);
3634ad48fd75SYan, Zheng 	if (!buf)
3635ad48fd75SYan, Zheng 		return -ENOMEM;
3636ad48fd75SYan, Zheng 
3637459931ecSChris Mason 	read_extent_buffer(leaf, buf, btrfs_item_ptr_offset(leaf,
3638459931ecSChris Mason 			    path->slots[0]), item_size);
3639ad48fd75SYan, Zheng 
3640459931ecSChris Mason 	slot = path->slots[0] + 1;
3641459931ecSChris Mason 	nritems = btrfs_header_nritems(leaf);
3642459931ecSChris Mason 	if (slot != nritems) {
3643459931ecSChris Mason 		/* shift the items */
3644459931ecSChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1),
3645459931ecSChris Mason 				btrfs_item_nr_offset(slot),
3646459931ecSChris Mason 				(nritems - slot) * sizeof(struct btrfs_item));
3647459931ecSChris Mason 	}
3648459931ecSChris Mason 
3649459931ecSChris Mason 	btrfs_cpu_key_to_disk(&disk_key, new_key);
3650459931ecSChris Mason 	btrfs_set_item_key(leaf, &disk_key, slot);
3651459931ecSChris Mason 
36523212fa14SJosef Bacik 	btrfs_set_item_offset(leaf, slot, orig_offset);
36533212fa14SJosef Bacik 	btrfs_set_item_size(leaf, slot, item_size - split_offset);
3654459931ecSChris Mason 
36553212fa14SJosef Bacik 	btrfs_set_item_offset(leaf, orig_slot,
3656459931ecSChris Mason 				 orig_offset + item_size - split_offset);
36573212fa14SJosef Bacik 	btrfs_set_item_size(leaf, orig_slot, split_offset);
3658459931ecSChris Mason 
3659459931ecSChris Mason 	btrfs_set_header_nritems(leaf, nritems + 1);
3660459931ecSChris Mason 
3661459931ecSChris Mason 	/* write the data for the start of the original item */
3662459931ecSChris Mason 	write_extent_buffer(leaf, buf,
3663459931ecSChris Mason 			    btrfs_item_ptr_offset(leaf, path->slots[0]),
3664459931ecSChris Mason 			    split_offset);
3665459931ecSChris Mason 
3666459931ecSChris Mason 	/* write the data for the new item */
3667459931ecSChris Mason 	write_extent_buffer(leaf, buf + split_offset,
3668459931ecSChris Mason 			    btrfs_item_ptr_offset(leaf, slot),
3669459931ecSChris Mason 			    item_size - split_offset);
3670459931ecSChris Mason 	btrfs_mark_buffer_dirty(leaf);
3671459931ecSChris Mason 
3672e902baacSDavid Sterba 	BUG_ON(btrfs_leaf_free_space(leaf) < 0);
3673459931ecSChris Mason 	kfree(buf);
3674ad48fd75SYan, Zheng 	return 0;
3675ad48fd75SYan, Zheng }
3676ad48fd75SYan, Zheng 
3677ad48fd75SYan, Zheng /*
3678ad48fd75SYan, Zheng  * This function splits a single item into two items,
3679ad48fd75SYan, Zheng  * giving 'new_key' to the new item and splitting the
3680ad48fd75SYan, Zheng  * old one at split_offset (from the start of the item).
3681ad48fd75SYan, Zheng  *
3682ad48fd75SYan, Zheng  * The path may be released by this operation.  After
3683ad48fd75SYan, Zheng  * the split, the path is pointing to the old item.  The
3684ad48fd75SYan, Zheng  * new item is going to be in the same node as the old one.
3685ad48fd75SYan, Zheng  *
3686ad48fd75SYan, Zheng  * Note, the item being split must be smaller enough to live alone on
3687ad48fd75SYan, Zheng  * a tree block with room for one extra struct btrfs_item
3688ad48fd75SYan, Zheng  *
3689ad48fd75SYan, Zheng  * This allows us to split the item in place, keeping a lock on the
3690ad48fd75SYan, Zheng  * leaf the entire time.
3691ad48fd75SYan, Zheng  */
3692ad48fd75SYan, Zheng int btrfs_split_item(struct btrfs_trans_handle *trans,
3693ad48fd75SYan, Zheng 		     struct btrfs_root *root,
3694ad48fd75SYan, Zheng 		     struct btrfs_path *path,
3695310712b2SOmar Sandoval 		     const struct btrfs_key *new_key,
3696ad48fd75SYan, Zheng 		     unsigned long split_offset)
3697ad48fd75SYan, Zheng {
3698ad48fd75SYan, Zheng 	int ret;
3699ad48fd75SYan, Zheng 	ret = setup_leaf_for_split(trans, root, path,
3700ad48fd75SYan, Zheng 				   sizeof(struct btrfs_item));
3701ad48fd75SYan, Zheng 	if (ret)
3702459931ecSChris Mason 		return ret;
3703ad48fd75SYan, Zheng 
370425263cd7SDavid Sterba 	ret = split_item(path, new_key, split_offset);
3705ad48fd75SYan, Zheng 	return ret;
3706ad48fd75SYan, Zheng }
3707ad48fd75SYan, Zheng 
3708ad48fd75SYan, Zheng /*
3709d352ac68SChris Mason  * make the item pointed to by the path smaller.  new_size indicates
3710d352ac68SChris Mason  * how small to make it, and from_end tells us if we just chop bytes
3711d352ac68SChris Mason  * off the end of the item or if we shift the item to chop bytes off
3712d352ac68SChris Mason  * the front.
3713d352ac68SChris Mason  */
371478ac4f9eSDavid Sterba void btrfs_truncate_item(struct btrfs_path *path, u32 new_size, int from_end)
3715b18c6685SChris Mason {
3716b18c6685SChris Mason 	int slot;
37175f39d397SChris Mason 	struct extent_buffer *leaf;
3718b18c6685SChris Mason 	u32 nritems;
3719b18c6685SChris Mason 	unsigned int data_end;
3720b18c6685SChris Mason 	unsigned int old_data_start;
3721b18c6685SChris Mason 	unsigned int old_size;
3722b18c6685SChris Mason 	unsigned int size_diff;
3723b18c6685SChris Mason 	int i;
3724cfed81a0SChris Mason 	struct btrfs_map_token token;
3725cfed81a0SChris Mason 
37265f39d397SChris Mason 	leaf = path->nodes[0];
3727179e29e4SChris Mason 	slot = path->slots[0];
3728179e29e4SChris Mason 
37293212fa14SJosef Bacik 	old_size = btrfs_item_size(leaf, slot);
3730179e29e4SChris Mason 	if (old_size == new_size)
3731143bede5SJeff Mahoney 		return;
3732b18c6685SChris Mason 
37335f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
37348f881e8cSDavid Sterba 	data_end = leaf_data_end(leaf);
3735b18c6685SChris Mason 
37363212fa14SJosef Bacik 	old_data_start = btrfs_item_offset(leaf, slot);
3737179e29e4SChris Mason 
3738b18c6685SChris Mason 	size_diff = old_size - new_size;
3739b18c6685SChris Mason 
3740b18c6685SChris Mason 	BUG_ON(slot < 0);
3741b18c6685SChris Mason 	BUG_ON(slot >= nritems);
3742b18c6685SChris Mason 
3743b18c6685SChris Mason 	/*
3744b18c6685SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
3745b18c6685SChris Mason 	 */
3746b18c6685SChris Mason 	/* first correct the data pointers */
3747c82f823cSDavid Sterba 	btrfs_init_map_token(&token, leaf);
3748b18c6685SChris Mason 	for (i = slot; i < nritems; i++) {
37495f39d397SChris Mason 		u32 ioff;
3750db94535dSChris Mason 
37513212fa14SJosef Bacik 		ioff = btrfs_token_item_offset(&token, i);
37523212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, i, ioff + size_diff);
3753b18c6685SChris Mason 	}
3754db94535dSChris Mason 
3755b18c6685SChris Mason 	/* shift the data */
3756179e29e4SChris Mason 	if (from_end) {
37573d9ec8c4SNikolay Borisov 		memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET +
37583d9ec8c4SNikolay Borisov 			      data_end + size_diff, BTRFS_LEAF_DATA_OFFSET +
3759b18c6685SChris Mason 			      data_end, old_data_start + new_size - data_end);
3760179e29e4SChris Mason 	} else {
3761179e29e4SChris Mason 		struct btrfs_disk_key disk_key;
3762179e29e4SChris Mason 		u64 offset;
3763179e29e4SChris Mason 
3764179e29e4SChris Mason 		btrfs_item_key(leaf, &disk_key, slot);
3765179e29e4SChris Mason 
3766179e29e4SChris Mason 		if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) {
3767179e29e4SChris Mason 			unsigned long ptr;
3768179e29e4SChris Mason 			struct btrfs_file_extent_item *fi;
3769179e29e4SChris Mason 
3770179e29e4SChris Mason 			fi = btrfs_item_ptr(leaf, slot,
3771179e29e4SChris Mason 					    struct btrfs_file_extent_item);
3772179e29e4SChris Mason 			fi = (struct btrfs_file_extent_item *)(
3773179e29e4SChris Mason 			     (unsigned long)fi - size_diff);
3774179e29e4SChris Mason 
3775179e29e4SChris Mason 			if (btrfs_file_extent_type(leaf, fi) ==
3776179e29e4SChris Mason 			    BTRFS_FILE_EXTENT_INLINE) {
3777179e29e4SChris Mason 				ptr = btrfs_item_ptr_offset(leaf, slot);
3778179e29e4SChris Mason 				memmove_extent_buffer(leaf, ptr,
3779179e29e4SChris Mason 				      (unsigned long)fi,
37807ec20afbSDavid Sterba 				      BTRFS_FILE_EXTENT_INLINE_DATA_START);
3781179e29e4SChris Mason 			}
3782179e29e4SChris Mason 		}
3783179e29e4SChris Mason 
37843d9ec8c4SNikolay Borisov 		memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET +
37853d9ec8c4SNikolay Borisov 			      data_end + size_diff, BTRFS_LEAF_DATA_OFFSET +
3786179e29e4SChris Mason 			      data_end, old_data_start - data_end);
3787179e29e4SChris Mason 
3788179e29e4SChris Mason 		offset = btrfs_disk_key_offset(&disk_key);
3789179e29e4SChris Mason 		btrfs_set_disk_key_offset(&disk_key, offset + size_diff);
3790179e29e4SChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot);
3791179e29e4SChris Mason 		if (slot == 0)
3792b167fa91SNikolay Borisov 			fixup_low_keys(path, &disk_key, 1);
3793179e29e4SChris Mason 	}
37945f39d397SChris Mason 
37953212fa14SJosef Bacik 	btrfs_set_item_size(leaf, slot, new_size);
37965f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
3797b18c6685SChris Mason 
3798e902baacSDavid Sterba 	if (btrfs_leaf_free_space(leaf) < 0) {
3799a4f78750SDavid Sterba 		btrfs_print_leaf(leaf);
3800b18c6685SChris Mason 		BUG();
38015f39d397SChris Mason 	}
3802b18c6685SChris Mason }
3803b18c6685SChris Mason 
3804d352ac68SChris Mason /*
38058f69dbd2SStefan Behrens  * make the item pointed to by the path bigger, data_size is the added size.
3806d352ac68SChris Mason  */
3807c71dd880SDavid Sterba void btrfs_extend_item(struct btrfs_path *path, u32 data_size)
38086567e837SChris Mason {
38096567e837SChris Mason 	int slot;
38105f39d397SChris Mason 	struct extent_buffer *leaf;
38116567e837SChris Mason 	u32 nritems;
38126567e837SChris Mason 	unsigned int data_end;
38136567e837SChris Mason 	unsigned int old_data;
38146567e837SChris Mason 	unsigned int old_size;
38156567e837SChris Mason 	int i;
3816cfed81a0SChris Mason 	struct btrfs_map_token token;
3817cfed81a0SChris Mason 
38185f39d397SChris Mason 	leaf = path->nodes[0];
38196567e837SChris Mason 
38205f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
38218f881e8cSDavid Sterba 	data_end = leaf_data_end(leaf);
38226567e837SChris Mason 
3823e902baacSDavid Sterba 	if (btrfs_leaf_free_space(leaf) < data_size) {
3824a4f78750SDavid Sterba 		btrfs_print_leaf(leaf);
38256567e837SChris Mason 		BUG();
38265f39d397SChris Mason 	}
38276567e837SChris Mason 	slot = path->slots[0];
3828dc2e724eSJosef Bacik 	old_data = btrfs_item_data_end(leaf, slot);
38296567e837SChris Mason 
38306567e837SChris Mason 	BUG_ON(slot < 0);
38313326d1b0SChris Mason 	if (slot >= nritems) {
3832a4f78750SDavid Sterba 		btrfs_print_leaf(leaf);
3833c71dd880SDavid Sterba 		btrfs_crit(leaf->fs_info, "slot %d too large, nritems %d",
3834d397712bSChris Mason 			   slot, nritems);
3835290342f6SArnd Bergmann 		BUG();
38363326d1b0SChris Mason 	}
38376567e837SChris Mason 
38386567e837SChris Mason 	/*
38396567e837SChris Mason 	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
38406567e837SChris Mason 	 */
38416567e837SChris Mason 	/* first correct the data pointers */
3842c82f823cSDavid Sterba 	btrfs_init_map_token(&token, leaf);
38436567e837SChris Mason 	for (i = slot; i < nritems; i++) {
38445f39d397SChris Mason 		u32 ioff;
3845db94535dSChris Mason 
38463212fa14SJosef Bacik 		ioff = btrfs_token_item_offset(&token, i);
38473212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, i, ioff - data_size);
38486567e837SChris Mason 	}
38495f39d397SChris Mason 
38506567e837SChris Mason 	/* shift the data */
38513d9ec8c4SNikolay Borisov 	memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET +
38523d9ec8c4SNikolay Borisov 		      data_end - data_size, BTRFS_LEAF_DATA_OFFSET +
38536567e837SChris Mason 		      data_end, old_data - data_end);
38545f39d397SChris Mason 
38556567e837SChris Mason 	data_end = old_data;
38563212fa14SJosef Bacik 	old_size = btrfs_item_size(leaf, slot);
38573212fa14SJosef Bacik 	btrfs_set_item_size(leaf, slot, old_size + data_size);
38585f39d397SChris Mason 	btrfs_mark_buffer_dirty(leaf);
38596567e837SChris Mason 
3860e902baacSDavid Sterba 	if (btrfs_leaf_free_space(leaf) < 0) {
3861a4f78750SDavid Sterba 		btrfs_print_leaf(leaf);
38626567e837SChris Mason 		BUG();
38635f39d397SChris Mason 	}
38646567e837SChris Mason }
38656567e837SChris Mason 
3866da9ffb24SNikolay Borisov /**
3867da9ffb24SNikolay Borisov  * setup_items_for_insert - Helper called before inserting one or more items
3868da9ffb24SNikolay Borisov  * to a leaf. Main purpose is to save stack depth by doing the bulk of the work
3869da9ffb24SNikolay Borisov  * in a function that doesn't call btrfs_search_slot
3870da9ffb24SNikolay Borisov  *
3871da9ffb24SNikolay Borisov  * @root:	root we are inserting items to
3872da9ffb24SNikolay Borisov  * @path:	points to the leaf/slot where we are going to insert new items
3873b7ef5f3aSFilipe Manana  * @batch:      information about the batch of items to insert
387474123bd7SChris Mason  */
3875f0641656SFilipe Manana static void setup_items_for_insert(struct btrfs_root *root, struct btrfs_path *path,
3876b7ef5f3aSFilipe Manana 				   const struct btrfs_item_batch *batch)
3877be0e5c09SChris Mason {
38780b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
38799c58309dSChris Mason 	int i;
38807518a238SChris Mason 	u32 nritems;
3881be0e5c09SChris Mason 	unsigned int data_end;
3882e2fa7227SChris Mason 	struct btrfs_disk_key disk_key;
388344871b1bSChris Mason 	struct extent_buffer *leaf;
388444871b1bSChris Mason 	int slot;
3885cfed81a0SChris Mason 	struct btrfs_map_token token;
3886fc0d82e1SNikolay Borisov 	u32 total_size;
3887fc0d82e1SNikolay Borisov 
3888b7ef5f3aSFilipe Manana 	/*
3889b7ef5f3aSFilipe Manana 	 * Before anything else, update keys in the parent and other ancestors
3890b7ef5f3aSFilipe Manana 	 * if needed, then release the write locks on them, so that other tasks
3891b7ef5f3aSFilipe Manana 	 * can use them while we modify the leaf.
3892b7ef5f3aSFilipe Manana 	 */
389324cdc847SFilipe Manana 	if (path->slots[0] == 0) {
3894b7ef5f3aSFilipe Manana 		btrfs_cpu_key_to_disk(&disk_key, &batch->keys[0]);
3895b167fa91SNikolay Borisov 		fixup_low_keys(path, &disk_key, 1);
389624cdc847SFilipe Manana 	}
389724cdc847SFilipe Manana 	btrfs_unlock_up_safe(path, 1);
389824cdc847SFilipe Manana 
38995f39d397SChris Mason 	leaf = path->nodes[0];
390044871b1bSChris Mason 	slot = path->slots[0];
390174123bd7SChris Mason 
39025f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
39038f881e8cSDavid Sterba 	data_end = leaf_data_end(leaf);
3904b7ef5f3aSFilipe Manana 	total_size = batch->total_data_size + (batch->nr * sizeof(struct btrfs_item));
3905eb60ceacSChris Mason 
3906e902baacSDavid Sterba 	if (btrfs_leaf_free_space(leaf) < total_size) {
3907a4f78750SDavid Sterba 		btrfs_print_leaf(leaf);
39080b246afaSJeff Mahoney 		btrfs_crit(fs_info, "not enough freespace need %u have %d",
3909e902baacSDavid Sterba 			   total_size, btrfs_leaf_free_space(leaf));
3910be0e5c09SChris Mason 		BUG();
3911d4dbff95SChris Mason 	}
39125f39d397SChris Mason 
3913c82f823cSDavid Sterba 	btrfs_init_map_token(&token, leaf);
3914be0e5c09SChris Mason 	if (slot != nritems) {
3915dc2e724eSJosef Bacik 		unsigned int old_data = btrfs_item_data_end(leaf, slot);
3916be0e5c09SChris Mason 
39175f39d397SChris Mason 		if (old_data < data_end) {
3918a4f78750SDavid Sterba 			btrfs_print_leaf(leaf);
39197269ddd2SNikolay Borisov 			btrfs_crit(fs_info,
39207269ddd2SNikolay Borisov 		"item at slot %d with data offset %u beyond data end of leaf %u",
39215f39d397SChris Mason 				   slot, old_data, data_end);
3922290342f6SArnd Bergmann 			BUG();
39235f39d397SChris Mason 		}
3924be0e5c09SChris Mason 		/*
3925be0e5c09SChris Mason 		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
3926be0e5c09SChris Mason 		 */
3927be0e5c09SChris Mason 		/* first correct the data pointers */
39280783fcfcSChris Mason 		for (i = slot; i < nritems; i++) {
39295f39d397SChris Mason 			u32 ioff;
3930db94535dSChris Mason 
39313212fa14SJosef Bacik 			ioff = btrfs_token_item_offset(&token, i);
39323212fa14SJosef Bacik 			btrfs_set_token_item_offset(&token, i,
3933b7ef5f3aSFilipe Manana 						       ioff - batch->total_data_size);
39340783fcfcSChris Mason 		}
3935be0e5c09SChris Mason 		/* shift the items */
3936b7ef5f3aSFilipe Manana 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + batch->nr),
39375f39d397SChris Mason 			      btrfs_item_nr_offset(slot),
39380783fcfcSChris Mason 			      (nritems - slot) * sizeof(struct btrfs_item));
3939be0e5c09SChris Mason 
3940be0e5c09SChris Mason 		/* shift the data */
39413d9ec8c4SNikolay Borisov 		memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET +
3942b7ef5f3aSFilipe Manana 				      data_end - batch->total_data_size,
3943b7ef5f3aSFilipe Manana 				      BTRFS_LEAF_DATA_OFFSET + data_end,
3944b7ef5f3aSFilipe Manana 				      old_data - data_end);
3945be0e5c09SChris Mason 		data_end = old_data;
3946be0e5c09SChris Mason 	}
39475f39d397SChris Mason 
394862e2749eSChris Mason 	/* setup the item for the new data */
3949b7ef5f3aSFilipe Manana 	for (i = 0; i < batch->nr; i++) {
3950b7ef5f3aSFilipe Manana 		btrfs_cpu_key_to_disk(&disk_key, &batch->keys[i]);
39519c58309dSChris Mason 		btrfs_set_item_key(leaf, &disk_key, slot + i);
3952b7ef5f3aSFilipe Manana 		data_end -= batch->data_sizes[i];
39533212fa14SJosef Bacik 		btrfs_set_token_item_offset(&token, slot + i, data_end);
39543212fa14SJosef Bacik 		btrfs_set_token_item_size(&token, slot + i, batch->data_sizes[i]);
39559c58309dSChris Mason 	}
395644871b1bSChris Mason 
3957b7ef5f3aSFilipe Manana 	btrfs_set_header_nritems(leaf, nritems + batch->nr);
3958b9473439SChris Mason 	btrfs_mark_buffer_dirty(leaf);
3959aa5d6bedSChris Mason 
3960e902baacSDavid Sterba 	if (btrfs_leaf_free_space(leaf) < 0) {
3961a4f78750SDavid Sterba 		btrfs_print_leaf(leaf);
3962be0e5c09SChris Mason 		BUG();
39635f39d397SChris Mason 	}
396444871b1bSChris Mason }
396544871b1bSChris Mason 
396644871b1bSChris Mason /*
3967f0641656SFilipe Manana  * Insert a new item into a leaf.
3968f0641656SFilipe Manana  *
3969f0641656SFilipe Manana  * @root:      The root of the btree.
3970f0641656SFilipe Manana  * @path:      A path pointing to the target leaf and slot.
3971f0641656SFilipe Manana  * @key:       The key of the new item.
3972f0641656SFilipe Manana  * @data_size: The size of the data associated with the new key.
3973f0641656SFilipe Manana  */
3974f0641656SFilipe Manana void btrfs_setup_item_for_insert(struct btrfs_root *root,
3975f0641656SFilipe Manana 				 struct btrfs_path *path,
3976f0641656SFilipe Manana 				 const struct btrfs_key *key,
3977f0641656SFilipe Manana 				 u32 data_size)
3978f0641656SFilipe Manana {
3979f0641656SFilipe Manana 	struct btrfs_item_batch batch;
3980f0641656SFilipe Manana 
3981f0641656SFilipe Manana 	batch.keys = key;
3982f0641656SFilipe Manana 	batch.data_sizes = &data_size;
3983f0641656SFilipe Manana 	batch.total_data_size = data_size;
3984f0641656SFilipe Manana 	batch.nr = 1;
3985f0641656SFilipe Manana 
3986f0641656SFilipe Manana 	setup_items_for_insert(root, path, &batch);
3987f0641656SFilipe Manana }
3988f0641656SFilipe Manana 
3989f0641656SFilipe Manana /*
399044871b1bSChris Mason  * Given a key and some data, insert items into the tree.
399144871b1bSChris Mason  * This does all the path init required, making room in the tree if needed.
399244871b1bSChris Mason  */
399344871b1bSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans,
399444871b1bSChris Mason 			    struct btrfs_root *root,
399544871b1bSChris Mason 			    struct btrfs_path *path,
3996b7ef5f3aSFilipe Manana 			    const struct btrfs_item_batch *batch)
399744871b1bSChris Mason {
399844871b1bSChris Mason 	int ret = 0;
399944871b1bSChris Mason 	int slot;
4000b7ef5f3aSFilipe Manana 	u32 total_size;
400144871b1bSChris Mason 
4002b7ef5f3aSFilipe Manana 	total_size = batch->total_data_size + (batch->nr * sizeof(struct btrfs_item));
4003b7ef5f3aSFilipe Manana 	ret = btrfs_search_slot(trans, root, &batch->keys[0], path, total_size, 1);
400444871b1bSChris Mason 	if (ret == 0)
400544871b1bSChris Mason 		return -EEXIST;
400644871b1bSChris Mason 	if (ret < 0)
4007143bede5SJeff Mahoney 		return ret;
400844871b1bSChris Mason 
400944871b1bSChris Mason 	slot = path->slots[0];
401044871b1bSChris Mason 	BUG_ON(slot < 0);
401144871b1bSChris Mason 
4012b7ef5f3aSFilipe Manana 	setup_items_for_insert(root, path, batch);
4013143bede5SJeff Mahoney 	return 0;
401462e2749eSChris Mason }
401562e2749eSChris Mason 
401662e2749eSChris Mason /*
401762e2749eSChris Mason  * Given a key and some data, insert an item into the tree.
401862e2749eSChris Mason  * This does all the path init required, making room in the tree if needed.
401962e2749eSChris Mason  */
4020310712b2SOmar Sandoval int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root *root,
4021310712b2SOmar Sandoval 		      const struct btrfs_key *cpu_key, void *data,
4022310712b2SOmar Sandoval 		      u32 data_size)
402362e2749eSChris Mason {
402462e2749eSChris Mason 	int ret = 0;
40252c90e5d6SChris Mason 	struct btrfs_path *path;
40265f39d397SChris Mason 	struct extent_buffer *leaf;
40275f39d397SChris Mason 	unsigned long ptr;
402862e2749eSChris Mason 
40292c90e5d6SChris Mason 	path = btrfs_alloc_path();
4030db5b493aSTsutomu Itoh 	if (!path)
4031db5b493aSTsutomu Itoh 		return -ENOMEM;
40322c90e5d6SChris Mason 	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
403362e2749eSChris Mason 	if (!ret) {
40345f39d397SChris Mason 		leaf = path->nodes[0];
40355f39d397SChris Mason 		ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
40365f39d397SChris Mason 		write_extent_buffer(leaf, data, ptr, data_size);
40375f39d397SChris Mason 		btrfs_mark_buffer_dirty(leaf);
403862e2749eSChris Mason 	}
40392c90e5d6SChris Mason 	btrfs_free_path(path);
4040aa5d6bedSChris Mason 	return ret;
4041be0e5c09SChris Mason }
4042be0e5c09SChris Mason 
404374123bd7SChris Mason /*
4044f0641656SFilipe Manana  * This function duplicates an item, giving 'new_key' to the new item.
4045f0641656SFilipe Manana  * It guarantees both items live in the same tree leaf and the new item is
4046f0641656SFilipe Manana  * contiguous with the original item.
4047f0641656SFilipe Manana  *
4048f0641656SFilipe Manana  * This allows us to split a file extent in place, keeping a lock on the leaf
4049f0641656SFilipe Manana  * the entire time.
4050f0641656SFilipe Manana  */
4051f0641656SFilipe Manana int btrfs_duplicate_item(struct btrfs_trans_handle *trans,
4052f0641656SFilipe Manana 			 struct btrfs_root *root,
4053f0641656SFilipe Manana 			 struct btrfs_path *path,
4054f0641656SFilipe Manana 			 const struct btrfs_key *new_key)
4055f0641656SFilipe Manana {
4056f0641656SFilipe Manana 	struct extent_buffer *leaf;
4057f0641656SFilipe Manana 	int ret;
4058f0641656SFilipe Manana 	u32 item_size;
4059f0641656SFilipe Manana 
4060f0641656SFilipe Manana 	leaf = path->nodes[0];
40613212fa14SJosef Bacik 	item_size = btrfs_item_size(leaf, path->slots[0]);
4062f0641656SFilipe Manana 	ret = setup_leaf_for_split(trans, root, path,
4063f0641656SFilipe Manana 				   item_size + sizeof(struct btrfs_item));
4064f0641656SFilipe Manana 	if (ret)
4065f0641656SFilipe Manana 		return ret;
4066f0641656SFilipe Manana 
4067f0641656SFilipe Manana 	path->slots[0]++;
4068f0641656SFilipe Manana 	btrfs_setup_item_for_insert(root, path, new_key, item_size);
4069f0641656SFilipe Manana 	leaf = path->nodes[0];
4070f0641656SFilipe Manana 	memcpy_extent_buffer(leaf,
4071f0641656SFilipe Manana 			     btrfs_item_ptr_offset(leaf, path->slots[0]),
4072f0641656SFilipe Manana 			     btrfs_item_ptr_offset(leaf, path->slots[0] - 1),
4073f0641656SFilipe Manana 			     item_size);
4074f0641656SFilipe Manana 	return 0;
4075f0641656SFilipe Manana }
4076f0641656SFilipe Manana 
4077f0641656SFilipe Manana /*
40785de08d7dSChris Mason  * delete the pointer from a given node.
407974123bd7SChris Mason  *
4080d352ac68SChris Mason  * the tree should have been previously balanced so the deletion does not
4081d352ac68SChris Mason  * empty a node.
408274123bd7SChris Mason  */
4083afe5fea7STsutomu Itoh static void del_ptr(struct btrfs_root *root, struct btrfs_path *path,
4084afe5fea7STsutomu Itoh 		    int level, int slot)
4085be0e5c09SChris Mason {
40865f39d397SChris Mason 	struct extent_buffer *parent = path->nodes[level];
40877518a238SChris Mason 	u32 nritems;
4088f3ea38daSJan Schmidt 	int ret;
4089be0e5c09SChris Mason 
40905f39d397SChris Mason 	nritems = btrfs_header_nritems(parent);
4091be0e5c09SChris Mason 	if (slot != nritems - 1) {
4092bf1d3425SDavid Sterba 		if (level) {
4093f3a84ccdSFilipe Manana 			ret = btrfs_tree_mod_log_insert_move(parent, slot,
4094f3a84ccdSFilipe Manana 					slot + 1, nritems - slot - 1);
4095bf1d3425SDavid Sterba 			BUG_ON(ret < 0);
4096bf1d3425SDavid Sterba 		}
40975f39d397SChris Mason 		memmove_extent_buffer(parent,
40985f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot),
40995f39d397SChris Mason 			      btrfs_node_key_ptr_offset(slot + 1),
4100d6025579SChris Mason 			      sizeof(struct btrfs_key_ptr) *
4101d6025579SChris Mason 			      (nritems - slot - 1));
410257ba86c0SChris Mason 	} else if (level) {
4103f3a84ccdSFilipe Manana 		ret = btrfs_tree_mod_log_insert_key(parent, slot,
4104f3a84ccdSFilipe Manana 				BTRFS_MOD_LOG_KEY_REMOVE, GFP_NOFS);
410557ba86c0SChris Mason 		BUG_ON(ret < 0);
4106be0e5c09SChris Mason 	}
4107f3ea38daSJan Schmidt 
41087518a238SChris Mason 	nritems--;
41095f39d397SChris Mason 	btrfs_set_header_nritems(parent, nritems);
41107518a238SChris Mason 	if (nritems == 0 && parent == root->node) {
41115f39d397SChris Mason 		BUG_ON(btrfs_header_level(root->node) != 1);
4112eb60ceacSChris Mason 		/* just turn the root into a leaf and break */
41135f39d397SChris Mason 		btrfs_set_header_level(root->node, 0);
4114bb803951SChris Mason 	} else if (slot == 0) {
41155f39d397SChris Mason 		struct btrfs_disk_key disk_key;
41165f39d397SChris Mason 
41175f39d397SChris Mason 		btrfs_node_key(parent, &disk_key, 0);
4118b167fa91SNikolay Borisov 		fixup_low_keys(path, &disk_key, level + 1);
4119be0e5c09SChris Mason 	}
4120d6025579SChris Mason 	btrfs_mark_buffer_dirty(parent);
4121be0e5c09SChris Mason }
4122be0e5c09SChris Mason 
412374123bd7SChris Mason /*
4124323ac95bSChris Mason  * a helper function to delete the leaf pointed to by path->slots[1] and
41255d4f98a2SYan Zheng  * path->nodes[1].
4126323ac95bSChris Mason  *
4127323ac95bSChris Mason  * This deletes the pointer in path->nodes[1] and frees the leaf
4128323ac95bSChris Mason  * block extent.  zero is returned if it all worked out, < 0 otherwise.
4129323ac95bSChris Mason  *
4130323ac95bSChris Mason  * The path must have already been setup for deleting the leaf, including
4131323ac95bSChris Mason  * all the proper balancing.  path->nodes[1] must be locked.
4132323ac95bSChris Mason  */
4133143bede5SJeff Mahoney static noinline void btrfs_del_leaf(struct btrfs_trans_handle *trans,
4134323ac95bSChris Mason 				    struct btrfs_root *root,
41355d4f98a2SYan Zheng 				    struct btrfs_path *path,
41365d4f98a2SYan Zheng 				    struct extent_buffer *leaf)
4137323ac95bSChris Mason {
41385d4f98a2SYan Zheng 	WARN_ON(btrfs_header_generation(leaf) != trans->transid);
4139afe5fea7STsutomu Itoh 	del_ptr(root, path, 1, path->slots[1]);
4140323ac95bSChris Mason 
41414d081c41SChris Mason 	/*
41424d081c41SChris Mason 	 * btrfs_free_extent is expensive, we want to make sure we
41434d081c41SChris Mason 	 * aren't holding any locks when we call it
41444d081c41SChris Mason 	 */
41454d081c41SChris Mason 	btrfs_unlock_up_safe(path, 0);
41464d081c41SChris Mason 
4147f0486c68SYan, Zheng 	root_sub_used(root, leaf->len);
4148f0486c68SYan, Zheng 
414967439dadSDavid Sterba 	atomic_inc(&leaf->refs);
41507a163608SFilipe Manana 	btrfs_free_tree_block(trans, btrfs_root_id(root), leaf, 0, 1);
41513083ee2eSJosef Bacik 	free_extent_buffer_stale(leaf);
4152323ac95bSChris Mason }
4153323ac95bSChris Mason /*
415474123bd7SChris Mason  * delete the item at the leaf level in path.  If that empties
415574123bd7SChris Mason  * the leaf, remove it from the tree
415674123bd7SChris Mason  */
415785e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root,
415885e21bacSChris Mason 		    struct btrfs_path *path, int slot, int nr)
4159be0e5c09SChris Mason {
41600b246afaSJeff Mahoney 	struct btrfs_fs_info *fs_info = root->fs_info;
41615f39d397SChris Mason 	struct extent_buffer *leaf;
4162ce0eac2aSAlexandru Moise 	u32 last_off;
4163ce0eac2aSAlexandru Moise 	u32 dsize = 0;
4164aa5d6bedSChris Mason 	int ret = 0;
4165aa5d6bedSChris Mason 	int wret;
416685e21bacSChris Mason 	int i;
41677518a238SChris Mason 	u32 nritems;
4168be0e5c09SChris Mason 
41695f39d397SChris Mason 	leaf = path->nodes[0];
41703212fa14SJosef Bacik 	last_off = btrfs_item_offset(leaf, slot + nr - 1);
417185e21bacSChris Mason 
417285e21bacSChris Mason 	for (i = 0; i < nr; i++)
41733212fa14SJosef Bacik 		dsize += btrfs_item_size(leaf, slot + i);
417485e21bacSChris Mason 
41755f39d397SChris Mason 	nritems = btrfs_header_nritems(leaf);
4176be0e5c09SChris Mason 
417785e21bacSChris Mason 	if (slot + nr != nritems) {
41788f881e8cSDavid Sterba 		int data_end = leaf_data_end(leaf);
4179c82f823cSDavid Sterba 		struct btrfs_map_token token;
41805f39d397SChris Mason 
41813d9ec8c4SNikolay Borisov 		memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET +
4182d6025579SChris Mason 			      data_end + dsize,
41833d9ec8c4SNikolay Borisov 			      BTRFS_LEAF_DATA_OFFSET + data_end,
418485e21bacSChris Mason 			      last_off - data_end);
41855f39d397SChris Mason 
4186c82f823cSDavid Sterba 		btrfs_init_map_token(&token, leaf);
418785e21bacSChris Mason 		for (i = slot + nr; i < nritems; i++) {
41885f39d397SChris Mason 			u32 ioff;
4189db94535dSChris Mason 
41903212fa14SJosef Bacik 			ioff = btrfs_token_item_offset(&token, i);
41913212fa14SJosef Bacik 			btrfs_set_token_item_offset(&token, i, ioff + dsize);
41920783fcfcSChris Mason 		}
4193db94535dSChris Mason 
41945f39d397SChris Mason 		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
419585e21bacSChris Mason 			      btrfs_item_nr_offset(slot + nr),
41960783fcfcSChris Mason 			      sizeof(struct btrfs_item) *
419785e21bacSChris Mason 			      (nritems - slot - nr));
4198be0e5c09SChris Mason 	}
419985e21bacSChris Mason 	btrfs_set_header_nritems(leaf, nritems - nr);
420085e21bacSChris Mason 	nritems -= nr;
42015f39d397SChris Mason 
420274123bd7SChris Mason 	/* delete the leaf if we've emptied it */
42037518a238SChris Mason 	if (nritems == 0) {
42045f39d397SChris Mason 		if (leaf == root->node) {
42055f39d397SChris Mason 			btrfs_set_header_level(leaf, 0);
42069a8dd150SChris Mason 		} else {
42076a884d7dSDavid Sterba 			btrfs_clean_tree_block(leaf);
4208143bede5SJeff Mahoney 			btrfs_del_leaf(trans, root, path, leaf);
42099a8dd150SChris Mason 		}
4210be0e5c09SChris Mason 	} else {
42117518a238SChris Mason 		int used = leaf_space_used(leaf, 0, nritems);
4212aa5d6bedSChris Mason 		if (slot == 0) {
42135f39d397SChris Mason 			struct btrfs_disk_key disk_key;
42145f39d397SChris Mason 
42155f39d397SChris Mason 			btrfs_item_key(leaf, &disk_key, 0);
4216b167fa91SNikolay Borisov 			fixup_low_keys(path, &disk_key, 1);
4217aa5d6bedSChris Mason 		}
4218aa5d6bedSChris Mason 
421974123bd7SChris Mason 		/* delete the leaf if it is mostly empty */
42200b246afaSJeff Mahoney 		if (used < BTRFS_LEAF_DATA_SIZE(fs_info) / 3) {
4221be0e5c09SChris Mason 			/* push_leaf_left fixes the path.
4222be0e5c09SChris Mason 			 * make sure the path still points to our leaf
4223be0e5c09SChris Mason 			 * for possible call to del_ptr below
4224be0e5c09SChris Mason 			 */
42254920c9acSChris Mason 			slot = path->slots[1];
422667439dadSDavid Sterba 			atomic_inc(&leaf->refs);
42275f39d397SChris Mason 
422899d8f83cSChris Mason 			wret = push_leaf_left(trans, root, path, 1, 1,
422999d8f83cSChris Mason 					      1, (u32)-1);
423054aa1f4dSChris Mason 			if (wret < 0 && wret != -ENOSPC)
4231aa5d6bedSChris Mason 				ret = wret;
42325f39d397SChris Mason 
42335f39d397SChris Mason 			if (path->nodes[0] == leaf &&
42345f39d397SChris Mason 			    btrfs_header_nritems(leaf)) {
423599d8f83cSChris Mason 				wret = push_leaf_right(trans, root, path, 1,
423699d8f83cSChris Mason 						       1, 1, 0);
423754aa1f4dSChris Mason 				if (wret < 0 && wret != -ENOSPC)
4238aa5d6bedSChris Mason 					ret = wret;
4239aa5d6bedSChris Mason 			}
42405f39d397SChris Mason 
42415f39d397SChris Mason 			if (btrfs_header_nritems(leaf) == 0) {
4242323ac95bSChris Mason 				path->slots[1] = slot;
4243143bede5SJeff Mahoney 				btrfs_del_leaf(trans, root, path, leaf);
42445f39d397SChris Mason 				free_extent_buffer(leaf);
4245143bede5SJeff Mahoney 				ret = 0;
42465de08d7dSChris Mason 			} else {
4247925baeddSChris Mason 				/* if we're still in the path, make sure
4248925baeddSChris Mason 				 * we're dirty.  Otherwise, one of the
4249925baeddSChris Mason 				 * push_leaf functions must have already
4250925baeddSChris Mason 				 * dirtied this buffer
4251925baeddSChris Mason 				 */
4252925baeddSChris Mason 				if (path->nodes[0] == leaf)
42535f39d397SChris Mason 					btrfs_mark_buffer_dirty(leaf);
42545f39d397SChris Mason 				free_extent_buffer(leaf);
4255be0e5c09SChris Mason 			}
4256d5719762SChris Mason 		} else {
42575f39d397SChris Mason 			btrfs_mark_buffer_dirty(leaf);
4258be0e5c09SChris Mason 		}
42599a8dd150SChris Mason 	}
4260aa5d6bedSChris Mason 	return ret;
42619a8dd150SChris Mason }
42629a8dd150SChris Mason 
426397571fd0SChris Mason /*
4264925baeddSChris Mason  * search the tree again to find a leaf with lesser keys
42657bb86316SChris Mason  * returns 0 if it found something or 1 if there are no lesser leaves.
42667bb86316SChris Mason  * returns < 0 on io errors.
4267d352ac68SChris Mason  *
4268d352ac68SChris Mason  * This may release the path, and so you may lose any locks held at the
4269d352ac68SChris Mason  * time you call it.
42707bb86316SChris Mason  */
427116e7549fSJosef Bacik int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path)
42727bb86316SChris Mason {
4273925baeddSChris Mason 	struct btrfs_key key;
4274925baeddSChris Mason 	struct btrfs_disk_key found_key;
4275925baeddSChris Mason 	int ret;
42767bb86316SChris Mason 
4277925baeddSChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, 0);
4278925baeddSChris Mason 
4279e8b0d724SFilipe David Borba Manana 	if (key.offset > 0) {
4280925baeddSChris Mason 		key.offset--;
4281e8b0d724SFilipe David Borba Manana 	} else if (key.type > 0) {
4282925baeddSChris Mason 		key.type--;
4283e8b0d724SFilipe David Borba Manana 		key.offset = (u64)-1;
4284e8b0d724SFilipe David Borba Manana 	} else if (key.objectid > 0) {
4285925baeddSChris Mason 		key.objectid--;
4286e8b0d724SFilipe David Borba Manana 		key.type = (u8)-1;
4287e8b0d724SFilipe David Borba Manana 		key.offset = (u64)-1;
4288e8b0d724SFilipe David Borba Manana 	} else {
42897bb86316SChris Mason 		return 1;
4290e8b0d724SFilipe David Borba Manana 	}
42917bb86316SChris Mason 
4292b3b4aa74SDavid Sterba 	btrfs_release_path(path);
4293925baeddSChris Mason 	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
4294925baeddSChris Mason 	if (ret < 0)
4295925baeddSChris Mason 		return ret;
4296925baeddSChris Mason 	btrfs_item_key(path->nodes[0], &found_key, 0);
4297925baeddSChris Mason 	ret = comp_keys(&found_key, &key);
4298337c6f68SFilipe Manana 	/*
4299337c6f68SFilipe Manana 	 * We might have had an item with the previous key in the tree right
4300337c6f68SFilipe Manana 	 * before we released our path. And after we released our path, that
4301337c6f68SFilipe Manana 	 * item might have been pushed to the first slot (0) of the leaf we
4302337c6f68SFilipe Manana 	 * were holding due to a tree balance. Alternatively, an item with the
4303337c6f68SFilipe Manana 	 * previous key can exist as the only element of a leaf (big fat item).
4304337c6f68SFilipe Manana 	 * Therefore account for these 2 cases, so that our callers (like
4305337c6f68SFilipe Manana 	 * btrfs_previous_item) don't miss an existing item with a key matching
4306337c6f68SFilipe Manana 	 * the previous key we computed above.
4307337c6f68SFilipe Manana 	 */
4308337c6f68SFilipe Manana 	if (ret <= 0)
43097bb86316SChris Mason 		return 0;
4310925baeddSChris Mason 	return 1;
43117bb86316SChris Mason }
43127bb86316SChris Mason 
43133f157a2fSChris Mason /*
43143f157a2fSChris Mason  * A helper function to walk down the tree starting at min_key, and looking
4315de78b51aSEric Sandeen  * for nodes or leaves that are have a minimum transaction id.
4316de78b51aSEric Sandeen  * This is used by the btree defrag code, and tree logging
43173f157a2fSChris Mason  *
43183f157a2fSChris Mason  * This does not cow, but it does stuff the starting key it finds back
43193f157a2fSChris Mason  * into min_key, so you can call btrfs_search_slot with cow=1 on the
43203f157a2fSChris Mason  * key and get a writable path.
43213f157a2fSChris Mason  *
43223f157a2fSChris Mason  * This honors path->lowest_level to prevent descent past a given level
43233f157a2fSChris Mason  * of the tree.
43243f157a2fSChris Mason  *
4325d352ac68SChris Mason  * min_trans indicates the oldest transaction that you are interested
4326d352ac68SChris Mason  * in walking through.  Any nodes or leaves older than min_trans are
4327d352ac68SChris Mason  * skipped over (without reading them).
4328d352ac68SChris Mason  *
43293f157a2fSChris Mason  * returns zero if something useful was found, < 0 on error and 1 if there
43303f157a2fSChris Mason  * was nothing in the tree that matched the search criteria.
43313f157a2fSChris Mason  */
43323f157a2fSChris Mason int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key,
4333de78b51aSEric Sandeen 			 struct btrfs_path *path,
43343f157a2fSChris Mason 			 u64 min_trans)
43353f157a2fSChris Mason {
43363f157a2fSChris Mason 	struct extent_buffer *cur;
43373f157a2fSChris Mason 	struct btrfs_key found_key;
43383f157a2fSChris Mason 	int slot;
43399652480bSYan 	int sret;
43403f157a2fSChris Mason 	u32 nritems;
43413f157a2fSChris Mason 	int level;
43423f157a2fSChris Mason 	int ret = 1;
4343f98de9b9SFilipe Manana 	int keep_locks = path->keep_locks;
43443f157a2fSChris Mason 
4345f98de9b9SFilipe Manana 	path->keep_locks = 1;
43463f157a2fSChris Mason again:
4347bd681513SChris Mason 	cur = btrfs_read_lock_root_node(root);
43483f157a2fSChris Mason 	level = btrfs_header_level(cur);
4349e02119d5SChris Mason 	WARN_ON(path->nodes[level]);
43503f157a2fSChris Mason 	path->nodes[level] = cur;
4351bd681513SChris Mason 	path->locks[level] = BTRFS_READ_LOCK;
43523f157a2fSChris Mason 
43533f157a2fSChris Mason 	if (btrfs_header_generation(cur) < min_trans) {
43543f157a2fSChris Mason 		ret = 1;
43553f157a2fSChris Mason 		goto out;
43563f157a2fSChris Mason 	}
43573f157a2fSChris Mason 	while (1) {
43583f157a2fSChris Mason 		nritems = btrfs_header_nritems(cur);
43593f157a2fSChris Mason 		level = btrfs_header_level(cur);
4360e3b83361SQu Wenruo 		sret = btrfs_bin_search(cur, min_key, &slot);
4361cbca7d59SFilipe Manana 		if (sret < 0) {
4362cbca7d59SFilipe Manana 			ret = sret;
4363cbca7d59SFilipe Manana 			goto out;
4364cbca7d59SFilipe Manana 		}
43653f157a2fSChris Mason 
4366323ac95bSChris Mason 		/* at the lowest level, we're done, setup the path and exit */
4367323ac95bSChris Mason 		if (level == path->lowest_level) {
4368e02119d5SChris Mason 			if (slot >= nritems)
4369e02119d5SChris Mason 				goto find_next_key;
43703f157a2fSChris Mason 			ret = 0;
43713f157a2fSChris Mason 			path->slots[level] = slot;
43723f157a2fSChris Mason 			btrfs_item_key_to_cpu(cur, &found_key, slot);
43733f157a2fSChris Mason 			goto out;
43743f157a2fSChris Mason 		}
43759652480bSYan 		if (sret && slot > 0)
43769652480bSYan 			slot--;
43773f157a2fSChris Mason 		/*
4378de78b51aSEric Sandeen 		 * check this node pointer against the min_trans parameters.
4379260db43cSRandy Dunlap 		 * If it is too old, skip to the next one.
43803f157a2fSChris Mason 		 */
43813f157a2fSChris Mason 		while (slot < nritems) {
43823f157a2fSChris Mason 			u64 gen;
4383e02119d5SChris Mason 
43843f157a2fSChris Mason 			gen = btrfs_node_ptr_generation(cur, slot);
43853f157a2fSChris Mason 			if (gen < min_trans) {
43863f157a2fSChris Mason 				slot++;
43873f157a2fSChris Mason 				continue;
43883f157a2fSChris Mason 			}
43893f157a2fSChris Mason 			break;
43903f157a2fSChris Mason 		}
4391e02119d5SChris Mason find_next_key:
43923f157a2fSChris Mason 		/*
43933f157a2fSChris Mason 		 * we didn't find a candidate key in this node, walk forward
43943f157a2fSChris Mason 		 * and find another one
43953f157a2fSChris Mason 		 */
43963f157a2fSChris Mason 		if (slot >= nritems) {
4397e02119d5SChris Mason 			path->slots[level] = slot;
4398e02119d5SChris Mason 			sret = btrfs_find_next_key(root, path, min_key, level,
4399de78b51aSEric Sandeen 						  min_trans);
4400e02119d5SChris Mason 			if (sret == 0) {
4401b3b4aa74SDavid Sterba 				btrfs_release_path(path);
44023f157a2fSChris Mason 				goto again;
44033f157a2fSChris Mason 			} else {
44043f157a2fSChris Mason 				goto out;
44053f157a2fSChris Mason 			}
44063f157a2fSChris Mason 		}
44073f157a2fSChris Mason 		/* save our key for returning back */
44083f157a2fSChris Mason 		btrfs_node_key_to_cpu(cur, &found_key, slot);
44093f157a2fSChris Mason 		path->slots[level] = slot;
44103f157a2fSChris Mason 		if (level == path->lowest_level) {
44113f157a2fSChris Mason 			ret = 0;
44123f157a2fSChris Mason 			goto out;
44133f157a2fSChris Mason 		}
44144b231ae4SDavid Sterba 		cur = btrfs_read_node_slot(cur, slot);
4415fb770ae4SLiu Bo 		if (IS_ERR(cur)) {
4416fb770ae4SLiu Bo 			ret = PTR_ERR(cur);
4417fb770ae4SLiu Bo 			goto out;
4418fb770ae4SLiu Bo 		}
44193f157a2fSChris Mason 
4420bd681513SChris Mason 		btrfs_tree_read_lock(cur);
4421b4ce94deSChris Mason 
4422bd681513SChris Mason 		path->locks[level - 1] = BTRFS_READ_LOCK;
44233f157a2fSChris Mason 		path->nodes[level - 1] = cur;
4424f7c79f30SChris Mason 		unlock_up(path, level, 1, 0, NULL);
44253f157a2fSChris Mason 	}
44263f157a2fSChris Mason out:
4427f98de9b9SFilipe Manana 	path->keep_locks = keep_locks;
4428f98de9b9SFilipe Manana 	if (ret == 0) {
4429f98de9b9SFilipe Manana 		btrfs_unlock_up_safe(path, path->lowest_level + 1);
4430f98de9b9SFilipe Manana 		memcpy(min_key, &found_key, sizeof(found_key));
4431f98de9b9SFilipe Manana 	}
44323f157a2fSChris Mason 	return ret;
44333f157a2fSChris Mason }
44343f157a2fSChris Mason 
44353f157a2fSChris Mason /*
44363f157a2fSChris Mason  * this is similar to btrfs_next_leaf, but does not try to preserve
44373f157a2fSChris Mason  * and fixup the path.  It looks for and returns the next key in the
4438de78b51aSEric Sandeen  * tree based on the current path and the min_trans parameters.
44393f157a2fSChris Mason  *
44403f157a2fSChris Mason  * 0 is returned if another key is found, < 0 if there are any errors
44413f157a2fSChris Mason  * and 1 is returned if there are no higher keys in the tree
44423f157a2fSChris Mason  *
44433f157a2fSChris Mason  * path->keep_locks should be set to 1 on the search made before
44443f157a2fSChris Mason  * calling this function.
44453f157a2fSChris Mason  */
4446e7a84565SChris Mason int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path,
4447de78b51aSEric Sandeen 			struct btrfs_key *key, int level, u64 min_trans)
4448e7a84565SChris Mason {
4449e7a84565SChris Mason 	int slot;
4450e7a84565SChris Mason 	struct extent_buffer *c;
4451e7a84565SChris Mason 
44526a9fb468SJosef Bacik 	WARN_ON(!path->keep_locks && !path->skip_locking);
4453e7a84565SChris Mason 	while (level < BTRFS_MAX_LEVEL) {
4454e7a84565SChris Mason 		if (!path->nodes[level])
4455e7a84565SChris Mason 			return 1;
4456e7a84565SChris Mason 
4457e7a84565SChris Mason 		slot = path->slots[level] + 1;
4458e7a84565SChris Mason 		c = path->nodes[level];
44593f157a2fSChris Mason next:
4460e7a84565SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
446133c66f43SYan Zheng 			int ret;
446233c66f43SYan Zheng 			int orig_lowest;
446333c66f43SYan Zheng 			struct btrfs_key cur_key;
446433c66f43SYan Zheng 			if (level + 1 >= BTRFS_MAX_LEVEL ||
446533c66f43SYan Zheng 			    !path->nodes[level + 1])
4466e7a84565SChris Mason 				return 1;
446733c66f43SYan Zheng 
44686a9fb468SJosef Bacik 			if (path->locks[level + 1] || path->skip_locking) {
446933c66f43SYan Zheng 				level++;
4470e7a84565SChris Mason 				continue;
4471e7a84565SChris Mason 			}
447233c66f43SYan Zheng 
447333c66f43SYan Zheng 			slot = btrfs_header_nritems(c) - 1;
447433c66f43SYan Zheng 			if (level == 0)
447533c66f43SYan Zheng 				btrfs_item_key_to_cpu(c, &cur_key, slot);
447633c66f43SYan Zheng 			else
447733c66f43SYan Zheng 				btrfs_node_key_to_cpu(c, &cur_key, slot);
447833c66f43SYan Zheng 
447933c66f43SYan Zheng 			orig_lowest = path->lowest_level;
4480b3b4aa74SDavid Sterba 			btrfs_release_path(path);
448133c66f43SYan Zheng 			path->lowest_level = level;
448233c66f43SYan Zheng 			ret = btrfs_search_slot(NULL, root, &cur_key, path,
448333c66f43SYan Zheng 						0, 0);
448433c66f43SYan Zheng 			path->lowest_level = orig_lowest;
448533c66f43SYan Zheng 			if (ret < 0)
448633c66f43SYan Zheng 				return ret;
448733c66f43SYan Zheng 
448833c66f43SYan Zheng 			c = path->nodes[level];
448933c66f43SYan Zheng 			slot = path->slots[level];
449033c66f43SYan Zheng 			if (ret == 0)
449133c66f43SYan Zheng 				slot++;
449233c66f43SYan Zheng 			goto next;
449333c66f43SYan Zheng 		}
449433c66f43SYan Zheng 
4495e7a84565SChris Mason 		if (level == 0)
4496e7a84565SChris Mason 			btrfs_item_key_to_cpu(c, key, slot);
44973f157a2fSChris Mason 		else {
44983f157a2fSChris Mason 			u64 gen = btrfs_node_ptr_generation(c, slot);
44993f157a2fSChris Mason 
45003f157a2fSChris Mason 			if (gen < min_trans) {
45013f157a2fSChris Mason 				slot++;
45023f157a2fSChris Mason 				goto next;
45033f157a2fSChris Mason 			}
4504e7a84565SChris Mason 			btrfs_node_key_to_cpu(c, key, slot);
45053f157a2fSChris Mason 		}
4506e7a84565SChris Mason 		return 0;
4507e7a84565SChris Mason 	}
4508e7a84565SChris Mason 	return 1;
4509e7a84565SChris Mason }
4510e7a84565SChris Mason 
45113d7806ecSJan Schmidt int btrfs_next_old_leaf(struct btrfs_root *root, struct btrfs_path *path,
45123d7806ecSJan Schmidt 			u64 time_seq)
45133d7806ecSJan Schmidt {
4514d97e63b6SChris Mason 	int slot;
45158e73f275SChris Mason 	int level;
45165f39d397SChris Mason 	struct extent_buffer *c;
45178e73f275SChris Mason 	struct extent_buffer *next;
4518d96b3424SFilipe Manana 	struct btrfs_fs_info *fs_info = root->fs_info;
4519925baeddSChris Mason 	struct btrfs_key key;
4520d96b3424SFilipe Manana 	bool need_commit_sem = false;
4521925baeddSChris Mason 	u32 nritems;
4522925baeddSChris Mason 	int ret;
45230e46318dSJosef Bacik 	int i;
4524925baeddSChris Mason 
4525925baeddSChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
4526d397712bSChris Mason 	if (nritems == 0)
4527925baeddSChris Mason 		return 1;
4528925baeddSChris Mason 
45298e73f275SChris Mason 	btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1);
45308e73f275SChris Mason again:
45318e73f275SChris Mason 	level = 1;
45328e73f275SChris Mason 	next = NULL;
4533b3b4aa74SDavid Sterba 	btrfs_release_path(path);
45348e73f275SChris Mason 
4535a2135011SChris Mason 	path->keep_locks = 1;
45368e73f275SChris Mason 
4537d96b3424SFilipe Manana 	if (time_seq) {
45383d7806ecSJan Schmidt 		ret = btrfs_search_old_slot(root, &key, path, time_seq);
4539d96b3424SFilipe Manana 	} else {
4540d96b3424SFilipe Manana 		if (path->need_commit_sem) {
4541d96b3424SFilipe Manana 			path->need_commit_sem = 0;
4542d96b3424SFilipe Manana 			need_commit_sem = true;
4543d96b3424SFilipe Manana 			down_read(&fs_info->commit_root_sem);
4544d96b3424SFilipe Manana 		}
4545925baeddSChris Mason 		ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
4546d96b3424SFilipe Manana 	}
4547925baeddSChris Mason 	path->keep_locks = 0;
4548925baeddSChris Mason 
4549925baeddSChris Mason 	if (ret < 0)
4550d96b3424SFilipe Manana 		goto done;
4551925baeddSChris Mason 
4552a2135011SChris Mason 	nritems = btrfs_header_nritems(path->nodes[0]);
4553168fd7d2SChris Mason 	/*
4554168fd7d2SChris Mason 	 * by releasing the path above we dropped all our locks.  A balance
4555168fd7d2SChris Mason 	 * could have added more items next to the key that used to be
4556168fd7d2SChris Mason 	 * at the very end of the block.  So, check again here and
4557168fd7d2SChris Mason 	 * advance the path if there are now more items available.
4558168fd7d2SChris Mason 	 */
4559a2135011SChris Mason 	if (nritems > 0 && path->slots[0] < nritems - 1) {
4560e457afecSYan Zheng 		if (ret == 0)
4561168fd7d2SChris Mason 			path->slots[0]++;
45628e73f275SChris Mason 		ret = 0;
4563925baeddSChris Mason 		goto done;
4564925baeddSChris Mason 	}
45650b43e04fSLiu Bo 	/*
45660b43e04fSLiu Bo 	 * So the above check misses one case:
45670b43e04fSLiu Bo 	 * - after releasing the path above, someone has removed the item that
45680b43e04fSLiu Bo 	 *   used to be at the very end of the block, and balance between leafs
45690b43e04fSLiu Bo 	 *   gets another one with bigger key.offset to replace it.
45700b43e04fSLiu Bo 	 *
45710b43e04fSLiu Bo 	 * This one should be returned as well, or we can get leaf corruption
45720b43e04fSLiu Bo 	 * later(esp. in __btrfs_drop_extents()).
45730b43e04fSLiu Bo 	 *
45740b43e04fSLiu Bo 	 * And a bit more explanation about this check,
45750b43e04fSLiu Bo 	 * with ret > 0, the key isn't found, the path points to the slot
45760b43e04fSLiu Bo 	 * where it should be inserted, so the path->slots[0] item must be the
45770b43e04fSLiu Bo 	 * bigger one.
45780b43e04fSLiu Bo 	 */
45790b43e04fSLiu Bo 	if (nritems > 0 && ret > 0 && path->slots[0] == nritems - 1) {
45800b43e04fSLiu Bo 		ret = 0;
45810b43e04fSLiu Bo 		goto done;
45820b43e04fSLiu Bo 	}
4583d97e63b6SChris Mason 
4584234b63a0SChris Mason 	while (level < BTRFS_MAX_LEVEL) {
45858e73f275SChris Mason 		if (!path->nodes[level]) {
45868e73f275SChris Mason 			ret = 1;
45878e73f275SChris Mason 			goto done;
45888e73f275SChris Mason 		}
45895f39d397SChris Mason 
4590d97e63b6SChris Mason 		slot = path->slots[level] + 1;
4591d97e63b6SChris Mason 		c = path->nodes[level];
45925f39d397SChris Mason 		if (slot >= btrfs_header_nritems(c)) {
4593d97e63b6SChris Mason 			level++;
45948e73f275SChris Mason 			if (level == BTRFS_MAX_LEVEL) {
45958e73f275SChris Mason 				ret = 1;
45968e73f275SChris Mason 				goto done;
45978e73f275SChris Mason 			}
4598d97e63b6SChris Mason 			continue;
4599d97e63b6SChris Mason 		}
46005f39d397SChris Mason 
46010e46318dSJosef Bacik 
46020e46318dSJosef Bacik 		/*
46030e46318dSJosef Bacik 		 * Our current level is where we're going to start from, and to
46040e46318dSJosef Bacik 		 * make sure lockdep doesn't complain we need to drop our locks
46050e46318dSJosef Bacik 		 * and nodes from 0 to our current level.
46060e46318dSJosef Bacik 		 */
46070e46318dSJosef Bacik 		for (i = 0; i < level; i++) {
46080e46318dSJosef Bacik 			if (path->locks[level]) {
46090e46318dSJosef Bacik 				btrfs_tree_read_unlock(path->nodes[i]);
46100e46318dSJosef Bacik 				path->locks[i] = 0;
46110e46318dSJosef Bacik 			}
46120e46318dSJosef Bacik 			free_extent_buffer(path->nodes[i]);
46130e46318dSJosef Bacik 			path->nodes[i] = NULL;
4614925baeddSChris Mason 		}
46155f39d397SChris Mason 
46168e73f275SChris Mason 		next = c;
4617d07b8528SLiu Bo 		ret = read_block_for_search(root, path, &next, level,
4618cda79c54SDavid Sterba 					    slot, &key);
46198e73f275SChris Mason 		if (ret == -EAGAIN)
46208e73f275SChris Mason 			goto again;
46215f39d397SChris Mason 
462276a05b35SChris Mason 		if (ret < 0) {
4623b3b4aa74SDavid Sterba 			btrfs_release_path(path);
462476a05b35SChris Mason 			goto done;
462576a05b35SChris Mason 		}
462676a05b35SChris Mason 
46275cd57b2cSChris Mason 		if (!path->skip_locking) {
4628bd681513SChris Mason 			ret = btrfs_try_tree_read_lock(next);
4629d42244a0SJan Schmidt 			if (!ret && time_seq) {
4630d42244a0SJan Schmidt 				/*
4631d42244a0SJan Schmidt 				 * If we don't get the lock, we may be racing
4632d42244a0SJan Schmidt 				 * with push_leaf_left, holding that lock while
4633d42244a0SJan Schmidt 				 * itself waiting for the leaf we've currently
4634d42244a0SJan Schmidt 				 * locked. To solve this situation, we give up
4635d42244a0SJan Schmidt 				 * on our lock and cycle.
4636d42244a0SJan Schmidt 				 */
4637cf538830SJan Schmidt 				free_extent_buffer(next);
4638d42244a0SJan Schmidt 				btrfs_release_path(path);
4639d42244a0SJan Schmidt 				cond_resched();
4640d42244a0SJan Schmidt 				goto again;
4641d42244a0SJan Schmidt 			}
46420e46318dSJosef Bacik 			if (!ret)
46430e46318dSJosef Bacik 				btrfs_tree_read_lock(next);
4644bd681513SChris Mason 		}
4645d97e63b6SChris Mason 		break;
4646d97e63b6SChris Mason 	}
4647d97e63b6SChris Mason 	path->slots[level] = slot;
4648d97e63b6SChris Mason 	while (1) {
4649d97e63b6SChris Mason 		level--;
4650d97e63b6SChris Mason 		path->nodes[level] = next;
4651d97e63b6SChris Mason 		path->slots[level] = 0;
4652a74a4b97SChris Mason 		if (!path->skip_locking)
4653ffeb03cfSJosef Bacik 			path->locks[level] = BTRFS_READ_LOCK;
4654d97e63b6SChris Mason 		if (!level)
4655d97e63b6SChris Mason 			break;
4656b4ce94deSChris Mason 
4657d07b8528SLiu Bo 		ret = read_block_for_search(root, path, &next, level,
4658cda79c54SDavid Sterba 					    0, &key);
46598e73f275SChris Mason 		if (ret == -EAGAIN)
46608e73f275SChris Mason 			goto again;
46618e73f275SChris Mason 
466276a05b35SChris Mason 		if (ret < 0) {
4663b3b4aa74SDavid Sterba 			btrfs_release_path(path);
466476a05b35SChris Mason 			goto done;
466576a05b35SChris Mason 		}
466676a05b35SChris Mason 
4667ffeb03cfSJosef Bacik 		if (!path->skip_locking)
46680e46318dSJosef Bacik 			btrfs_tree_read_lock(next);
4669d97e63b6SChris Mason 	}
46708e73f275SChris Mason 	ret = 0;
4671925baeddSChris Mason done:
4672f7c79f30SChris Mason 	unlock_up(path, 0, 1, 0, NULL);
4673d96b3424SFilipe Manana 	if (need_commit_sem) {
4674d96b3424SFilipe Manana 		int ret2;
4675d96b3424SFilipe Manana 
4676d96b3424SFilipe Manana 		path->need_commit_sem = 1;
4677d96b3424SFilipe Manana 		ret2 = finish_need_commit_sem_search(path);
4678d96b3424SFilipe Manana 		up_read(&fs_info->commit_root_sem);
4679d96b3424SFilipe Manana 		if (ret2)
4680d96b3424SFilipe Manana 			ret = ret2;
4681d96b3424SFilipe Manana 	}
46828e73f275SChris Mason 
46838e73f275SChris Mason 	return ret;
4684d97e63b6SChris Mason }
46850b86a832SChris Mason 
46863f157a2fSChris Mason /*
46873f157a2fSChris Mason  * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps
46883f157a2fSChris Mason  * searching until it gets past min_objectid or finds an item of 'type'
46893f157a2fSChris Mason  *
46903f157a2fSChris Mason  * returns 0 if something is found, 1 if nothing was found and < 0 on error
46913f157a2fSChris Mason  */
46920b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root,
46930b86a832SChris Mason 			struct btrfs_path *path, u64 min_objectid,
46940b86a832SChris Mason 			int type)
46950b86a832SChris Mason {
46960b86a832SChris Mason 	struct btrfs_key found_key;
46970b86a832SChris Mason 	struct extent_buffer *leaf;
4698e02119d5SChris Mason 	u32 nritems;
46990b86a832SChris Mason 	int ret;
47000b86a832SChris Mason 
47010b86a832SChris Mason 	while (1) {
47020b86a832SChris Mason 		if (path->slots[0] == 0) {
47030b86a832SChris Mason 			ret = btrfs_prev_leaf(root, path);
47040b86a832SChris Mason 			if (ret != 0)
47050b86a832SChris Mason 				return ret;
47060b86a832SChris Mason 		} else {
47070b86a832SChris Mason 			path->slots[0]--;
47080b86a832SChris Mason 		}
47090b86a832SChris Mason 		leaf = path->nodes[0];
4710e02119d5SChris Mason 		nritems = btrfs_header_nritems(leaf);
4711e02119d5SChris Mason 		if (nritems == 0)
4712e02119d5SChris Mason 			return 1;
4713e02119d5SChris Mason 		if (path->slots[0] == nritems)
4714e02119d5SChris Mason 			path->slots[0]--;
4715e02119d5SChris Mason 
47160b86a832SChris Mason 		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
4717e02119d5SChris Mason 		if (found_key.objectid < min_objectid)
4718e02119d5SChris Mason 			break;
47190a4eefbbSYan Zheng 		if (found_key.type == type)
47200a4eefbbSYan Zheng 			return 0;
4721e02119d5SChris Mason 		if (found_key.objectid == min_objectid &&
4722e02119d5SChris Mason 		    found_key.type < type)
4723e02119d5SChris Mason 			break;
47240b86a832SChris Mason 	}
47250b86a832SChris Mason 	return 1;
47260b86a832SChris Mason }
4727ade2e0b3SWang Shilong 
4728ade2e0b3SWang Shilong /*
4729ade2e0b3SWang Shilong  * search in extent tree to find a previous Metadata/Data extent item with
4730ade2e0b3SWang Shilong  * min objecitd.
4731ade2e0b3SWang Shilong  *
4732ade2e0b3SWang Shilong  * returns 0 if something is found, 1 if nothing was found and < 0 on error
4733ade2e0b3SWang Shilong  */
4734ade2e0b3SWang Shilong int btrfs_previous_extent_item(struct btrfs_root *root,
4735ade2e0b3SWang Shilong 			struct btrfs_path *path, u64 min_objectid)
4736ade2e0b3SWang Shilong {
4737ade2e0b3SWang Shilong 	struct btrfs_key found_key;
4738ade2e0b3SWang Shilong 	struct extent_buffer *leaf;
4739ade2e0b3SWang Shilong 	u32 nritems;
4740ade2e0b3SWang Shilong 	int ret;
4741ade2e0b3SWang Shilong 
4742ade2e0b3SWang Shilong 	while (1) {
4743ade2e0b3SWang Shilong 		if (path->slots[0] == 0) {
4744ade2e0b3SWang Shilong 			ret = btrfs_prev_leaf(root, path);
4745ade2e0b3SWang Shilong 			if (ret != 0)
4746ade2e0b3SWang Shilong 				return ret;
4747ade2e0b3SWang Shilong 		} else {
4748ade2e0b3SWang Shilong 			path->slots[0]--;
4749ade2e0b3SWang Shilong 		}
4750ade2e0b3SWang Shilong 		leaf = path->nodes[0];
4751ade2e0b3SWang Shilong 		nritems = btrfs_header_nritems(leaf);
4752ade2e0b3SWang Shilong 		if (nritems == 0)
4753ade2e0b3SWang Shilong 			return 1;
4754ade2e0b3SWang Shilong 		if (path->slots[0] == nritems)
4755ade2e0b3SWang Shilong 			path->slots[0]--;
4756ade2e0b3SWang Shilong 
4757ade2e0b3SWang Shilong 		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
4758ade2e0b3SWang Shilong 		if (found_key.objectid < min_objectid)
4759ade2e0b3SWang Shilong 			break;
4760ade2e0b3SWang Shilong 		if (found_key.type == BTRFS_EXTENT_ITEM_KEY ||
4761ade2e0b3SWang Shilong 		    found_key.type == BTRFS_METADATA_ITEM_KEY)
4762ade2e0b3SWang Shilong 			return 0;
4763ade2e0b3SWang Shilong 		if (found_key.objectid == min_objectid &&
4764ade2e0b3SWang Shilong 		    found_key.type < BTRFS_EXTENT_ITEM_KEY)
4765ade2e0b3SWang Shilong 			break;
4766ade2e0b3SWang Shilong 	}
4767ade2e0b3SWang Shilong 	return 1;
4768ade2e0b3SWang Shilong }
4769