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); 8494eb150d6SQu Wenruo if (IS_ERR(eb)) 8504eb150d6SQu Wenruo return eb; 8514eb150d6SQu Wenruo if (!extent_buffer_uptodate(eb)) { 852416bc658SJosef Bacik free_extent_buffer(eb); 8534eb150d6SQu Wenruo return ERR_PTR(-EIO); 854416bc658SJosef Bacik } 855416bc658SJosef Bacik 856416bc658SJosef Bacik return eb; 857bb803951SChris Mason } 858bb803951SChris Mason 859d352ac68SChris Mason /* 860d352ac68SChris Mason * node level balancing, used to make sure nodes are in proper order for 861d352ac68SChris Mason * item deletion. We balance from the top down, so we have to make sure 862d352ac68SChris Mason * that a deletion won't leave an node completely empty later on. 863d352ac68SChris Mason */ 864e02119d5SChris Mason static noinline int balance_level(struct btrfs_trans_handle *trans, 86598ed5174SChris Mason struct btrfs_root *root, 86698ed5174SChris Mason struct btrfs_path *path, int level) 867bb803951SChris Mason { 8680b246afaSJeff Mahoney struct btrfs_fs_info *fs_info = root->fs_info; 8695f39d397SChris Mason struct extent_buffer *right = NULL; 8705f39d397SChris Mason struct extent_buffer *mid; 8715f39d397SChris Mason struct extent_buffer *left = NULL; 8725f39d397SChris Mason struct extent_buffer *parent = NULL; 873bb803951SChris Mason int ret = 0; 874bb803951SChris Mason int wret; 875bb803951SChris Mason int pslot; 876bb803951SChris Mason int orig_slot = path->slots[level]; 87779f95c82SChris Mason u64 orig_ptr; 878bb803951SChris Mason 87998e6b1ebSLiu Bo ASSERT(level > 0); 880bb803951SChris Mason 8815f39d397SChris Mason mid = path->nodes[level]; 882b4ce94deSChris Mason 883ac5887c8SJosef Bacik WARN_ON(path->locks[level] != BTRFS_WRITE_LOCK); 8847bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 8857bb86316SChris Mason 8861d4f8a0cSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 88779f95c82SChris Mason 888a05a9bb1SLi Zefan if (level < BTRFS_MAX_LEVEL - 1) { 8895f39d397SChris Mason parent = path->nodes[level + 1]; 890bb803951SChris Mason pslot = path->slots[level + 1]; 891a05a9bb1SLi Zefan } 892bb803951SChris Mason 89340689478SChris Mason /* 89440689478SChris Mason * deal with the case where there is only one pointer in the root 89540689478SChris Mason * by promoting the node below to a root 89640689478SChris Mason */ 8975f39d397SChris Mason if (!parent) { 8985f39d397SChris Mason struct extent_buffer *child; 899bb803951SChris Mason 9005f39d397SChris Mason if (btrfs_header_nritems(mid) != 1) 901bb803951SChris Mason return 0; 902bb803951SChris Mason 903bb803951SChris Mason /* promote the child to a root */ 9044b231ae4SDavid Sterba child = btrfs_read_node_slot(mid, 0); 905fb770ae4SLiu Bo if (IS_ERR(child)) { 906fb770ae4SLiu Bo ret = PTR_ERR(child); 9070b246afaSJeff Mahoney btrfs_handle_fs_error(fs_info, ret, NULL); 908305a26afSMark Fasheh goto enospc; 909305a26afSMark Fasheh } 910305a26afSMark Fasheh 911925baeddSChris Mason btrfs_tree_lock(child); 9129631e4ccSJosef Bacik ret = btrfs_cow_block(trans, root, child, mid, 0, &child, 9139631e4ccSJosef Bacik BTRFS_NESTING_COW); 914f0486c68SYan, Zheng if (ret) { 915f0486c68SYan, Zheng btrfs_tree_unlock(child); 916f0486c68SYan, Zheng free_extent_buffer(child); 917f0486c68SYan, Zheng goto enospc; 918f0486c68SYan, Zheng } 9192f375ab9SYan 920406808abSFilipe Manana ret = btrfs_tree_mod_log_insert_root(root->node, child, true); 921d9d19a01SDavid Sterba BUG_ON(ret < 0); 922240f62c8SChris Mason rcu_assign_pointer(root->node, child); 923925baeddSChris Mason 9240b86a832SChris Mason add_root_to_dirty_list(root); 925925baeddSChris Mason btrfs_tree_unlock(child); 926b4ce94deSChris Mason 927925baeddSChris Mason path->locks[level] = 0; 928bb803951SChris Mason path->nodes[level] = NULL; 9296a884d7dSDavid Sterba btrfs_clean_tree_block(mid); 930925baeddSChris Mason btrfs_tree_unlock(mid); 931bb803951SChris Mason /* once for the path */ 9325f39d397SChris Mason free_extent_buffer(mid); 933f0486c68SYan, Zheng 934f0486c68SYan, Zheng root_sub_used(root, mid->len); 9357a163608SFilipe Manana btrfs_free_tree_block(trans, btrfs_root_id(root), mid, 0, 1); 936bb803951SChris Mason /* once for the root ptr */ 9373083ee2eSJosef Bacik free_extent_buffer_stale(mid); 938f0486c68SYan, Zheng return 0; 939bb803951SChris Mason } 9405f39d397SChris Mason if (btrfs_header_nritems(mid) > 9410b246afaSJeff Mahoney BTRFS_NODEPTRS_PER_BLOCK(fs_info) / 4) 942bb803951SChris Mason return 0; 943bb803951SChris Mason 9444b231ae4SDavid Sterba left = btrfs_read_node_slot(parent, pslot - 1); 945fb770ae4SLiu Bo if (IS_ERR(left)) 946fb770ae4SLiu Bo left = NULL; 947fb770ae4SLiu Bo 9485f39d397SChris Mason if (left) { 949bf77467aSJosef Bacik __btrfs_tree_lock(left, BTRFS_NESTING_LEFT); 9505f39d397SChris Mason wret = btrfs_cow_block(trans, root, left, 9519631e4ccSJosef Bacik parent, pslot - 1, &left, 952bf59a5a2SJosef Bacik BTRFS_NESTING_LEFT_COW); 95354aa1f4dSChris Mason if (wret) { 95454aa1f4dSChris Mason ret = wret; 95554aa1f4dSChris Mason goto enospc; 95654aa1f4dSChris Mason } 9572cc58cf2SChris Mason } 958fb770ae4SLiu Bo 9594b231ae4SDavid Sterba right = btrfs_read_node_slot(parent, pslot + 1); 960fb770ae4SLiu Bo if (IS_ERR(right)) 961fb770ae4SLiu Bo right = NULL; 962fb770ae4SLiu Bo 9635f39d397SChris Mason if (right) { 964bf77467aSJosef Bacik __btrfs_tree_lock(right, BTRFS_NESTING_RIGHT); 9655f39d397SChris Mason wret = btrfs_cow_block(trans, root, right, 9669631e4ccSJosef Bacik parent, pslot + 1, &right, 967bf59a5a2SJosef Bacik BTRFS_NESTING_RIGHT_COW); 9682cc58cf2SChris Mason if (wret) { 9692cc58cf2SChris Mason ret = wret; 9702cc58cf2SChris Mason goto enospc; 9712cc58cf2SChris Mason } 9722cc58cf2SChris Mason } 9732cc58cf2SChris Mason 9742cc58cf2SChris Mason /* first, try to make some room in the middle buffer */ 9755f39d397SChris Mason if (left) { 9765f39d397SChris Mason orig_slot += btrfs_header_nritems(left); 977d30a668fSDavid Sterba wret = push_node_left(trans, left, mid, 1); 97879f95c82SChris Mason if (wret < 0) 97979f95c82SChris Mason ret = wret; 980bb803951SChris Mason } 98179f95c82SChris Mason 98279f95c82SChris Mason /* 98379f95c82SChris Mason * then try to empty the right most buffer into the middle 98479f95c82SChris Mason */ 9855f39d397SChris Mason if (right) { 986d30a668fSDavid Sterba wret = push_node_left(trans, mid, right, 1); 98754aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 98879f95c82SChris Mason ret = wret; 9895f39d397SChris Mason if (btrfs_header_nritems(right) == 0) { 9906a884d7dSDavid Sterba btrfs_clean_tree_block(right); 991925baeddSChris Mason btrfs_tree_unlock(right); 992afe5fea7STsutomu Itoh del_ptr(root, path, level + 1, pslot + 1); 993f0486c68SYan, Zheng root_sub_used(root, right->len); 9947a163608SFilipe Manana btrfs_free_tree_block(trans, btrfs_root_id(root), right, 9957a163608SFilipe Manana 0, 1); 9963083ee2eSJosef Bacik free_extent_buffer_stale(right); 997f0486c68SYan, Zheng right = NULL; 998bb803951SChris Mason } else { 9995f39d397SChris Mason struct btrfs_disk_key right_key; 10005f39d397SChris Mason btrfs_node_key(right, &right_key, 0); 1001f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_insert_key(parent, pslot + 1, 1002f3a84ccdSFilipe Manana BTRFS_MOD_LOG_KEY_REPLACE, GFP_NOFS); 10030e82bcfeSDavid Sterba BUG_ON(ret < 0); 10045f39d397SChris Mason btrfs_set_node_key(parent, &right_key, pslot + 1); 10055f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 1006bb803951SChris Mason } 1007bb803951SChris Mason } 10085f39d397SChris Mason if (btrfs_header_nritems(mid) == 1) { 100979f95c82SChris Mason /* 101079f95c82SChris Mason * we're not allowed to leave a node with one item in the 101179f95c82SChris Mason * tree during a delete. A deletion from lower in the tree 101279f95c82SChris Mason * could try to delete the only pointer in this node. 101379f95c82SChris Mason * So, pull some keys from the left. 101479f95c82SChris Mason * There has to be a left pointer at this point because 101579f95c82SChris Mason * otherwise we would have pulled some pointers from the 101679f95c82SChris Mason * right 101779f95c82SChris Mason */ 1018305a26afSMark Fasheh if (!left) { 1019305a26afSMark Fasheh ret = -EROFS; 10200b246afaSJeff Mahoney btrfs_handle_fs_error(fs_info, ret, NULL); 1021305a26afSMark Fasheh goto enospc; 1022305a26afSMark Fasheh } 102355d32ed8SDavid Sterba wret = balance_node_right(trans, mid, left); 102454aa1f4dSChris Mason if (wret < 0) { 102579f95c82SChris Mason ret = wret; 102654aa1f4dSChris Mason goto enospc; 102754aa1f4dSChris Mason } 1028bce4eae9SChris Mason if (wret == 1) { 1029d30a668fSDavid Sterba wret = push_node_left(trans, left, mid, 1); 1030bce4eae9SChris Mason if (wret < 0) 1031bce4eae9SChris Mason ret = wret; 1032bce4eae9SChris Mason } 103379f95c82SChris Mason BUG_ON(wret == 1); 103479f95c82SChris Mason } 10355f39d397SChris Mason if (btrfs_header_nritems(mid) == 0) { 10366a884d7dSDavid Sterba btrfs_clean_tree_block(mid); 1037925baeddSChris Mason btrfs_tree_unlock(mid); 1038afe5fea7STsutomu Itoh del_ptr(root, path, level + 1, pslot); 1039f0486c68SYan, Zheng root_sub_used(root, mid->len); 10407a163608SFilipe Manana btrfs_free_tree_block(trans, btrfs_root_id(root), mid, 0, 1); 10413083ee2eSJosef Bacik free_extent_buffer_stale(mid); 1042f0486c68SYan, Zheng mid = NULL; 104379f95c82SChris Mason } else { 104479f95c82SChris Mason /* update the parent key to reflect our changes */ 10455f39d397SChris Mason struct btrfs_disk_key mid_key; 10465f39d397SChris Mason btrfs_node_key(mid, &mid_key, 0); 1047f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_insert_key(parent, pslot, 1048f3a84ccdSFilipe Manana BTRFS_MOD_LOG_KEY_REPLACE, GFP_NOFS); 10490e82bcfeSDavid Sterba BUG_ON(ret < 0); 10505f39d397SChris Mason btrfs_set_node_key(parent, &mid_key, pslot); 10515f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 105279f95c82SChris Mason } 1053bb803951SChris Mason 105479f95c82SChris Mason /* update the path */ 10555f39d397SChris Mason if (left) { 10565f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 105767439dadSDavid Sterba atomic_inc(&left->refs); 1058925baeddSChris Mason /* left was locked after cow */ 10595f39d397SChris Mason path->nodes[level] = left; 1060bb803951SChris Mason path->slots[level + 1] -= 1; 1061bb803951SChris Mason path->slots[level] = orig_slot; 1062925baeddSChris Mason if (mid) { 1063925baeddSChris Mason btrfs_tree_unlock(mid); 10645f39d397SChris Mason free_extent_buffer(mid); 1065925baeddSChris Mason } 1066bb803951SChris Mason } else { 10675f39d397SChris Mason orig_slot -= btrfs_header_nritems(left); 1068bb803951SChris Mason path->slots[level] = orig_slot; 1069bb803951SChris Mason } 1070bb803951SChris Mason } 107179f95c82SChris Mason /* double check we haven't messed things up */ 1072e20d96d6SChris Mason if (orig_ptr != 10735f39d397SChris Mason btrfs_node_blockptr(path->nodes[level], path->slots[level])) 107479f95c82SChris Mason BUG(); 107554aa1f4dSChris Mason enospc: 1076925baeddSChris Mason if (right) { 1077925baeddSChris Mason btrfs_tree_unlock(right); 10785f39d397SChris Mason free_extent_buffer(right); 1079925baeddSChris Mason } 1080925baeddSChris Mason if (left) { 1081925baeddSChris Mason if (path->nodes[level] != left) 1082925baeddSChris Mason btrfs_tree_unlock(left); 10835f39d397SChris Mason free_extent_buffer(left); 1084925baeddSChris Mason } 1085bb803951SChris Mason return ret; 1086bb803951SChris Mason } 1087bb803951SChris Mason 1088d352ac68SChris Mason /* Node balancing for insertion. Here we only split or push nodes around 1089d352ac68SChris Mason * when they are completely full. This is also done top down, so we 1090d352ac68SChris Mason * have to be pessimistic. 1091d352ac68SChris Mason */ 1092d397712bSChris Mason static noinline int push_nodes_for_insert(struct btrfs_trans_handle *trans, 1093e66f709bSChris Mason struct btrfs_root *root, 1094e66f709bSChris Mason struct btrfs_path *path, int level) 1095e66f709bSChris Mason { 10960b246afaSJeff Mahoney struct btrfs_fs_info *fs_info = root->fs_info; 10975f39d397SChris Mason struct extent_buffer *right = NULL; 10985f39d397SChris Mason struct extent_buffer *mid; 10995f39d397SChris Mason struct extent_buffer *left = NULL; 11005f39d397SChris Mason struct extent_buffer *parent = NULL; 1101e66f709bSChris Mason int ret = 0; 1102e66f709bSChris Mason int wret; 1103e66f709bSChris Mason int pslot; 1104e66f709bSChris Mason int orig_slot = path->slots[level]; 1105e66f709bSChris Mason 1106e66f709bSChris Mason if (level == 0) 1107e66f709bSChris Mason return 1; 1108e66f709bSChris Mason 11095f39d397SChris Mason mid = path->nodes[level]; 11107bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 1111e66f709bSChris Mason 1112a05a9bb1SLi Zefan if (level < BTRFS_MAX_LEVEL - 1) { 11135f39d397SChris Mason parent = path->nodes[level + 1]; 1114e66f709bSChris Mason pslot = path->slots[level + 1]; 1115a05a9bb1SLi Zefan } 1116e66f709bSChris Mason 11175f39d397SChris Mason if (!parent) 1118e66f709bSChris Mason return 1; 1119e66f709bSChris Mason 11204b231ae4SDavid Sterba left = btrfs_read_node_slot(parent, pslot - 1); 1121fb770ae4SLiu Bo if (IS_ERR(left)) 1122fb770ae4SLiu Bo left = NULL; 1123e66f709bSChris Mason 1124e66f709bSChris Mason /* first, try to make some room in the middle buffer */ 11255f39d397SChris Mason if (left) { 1126e66f709bSChris Mason u32 left_nr; 1127925baeddSChris Mason 1128bf77467aSJosef Bacik __btrfs_tree_lock(left, BTRFS_NESTING_LEFT); 1129b4ce94deSChris Mason 11305f39d397SChris Mason left_nr = btrfs_header_nritems(left); 11310b246afaSJeff Mahoney if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(fs_info) - 1) { 113233ade1f8SChris Mason wret = 1; 113333ade1f8SChris Mason } else { 11345f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, parent, 11359631e4ccSJosef Bacik pslot - 1, &left, 1136bf59a5a2SJosef Bacik BTRFS_NESTING_LEFT_COW); 113754aa1f4dSChris Mason if (ret) 113854aa1f4dSChris Mason wret = 1; 113954aa1f4dSChris Mason else { 1140d30a668fSDavid Sterba wret = push_node_left(trans, left, mid, 0); 114154aa1f4dSChris Mason } 114233ade1f8SChris Mason } 1143e66f709bSChris Mason if (wret < 0) 1144e66f709bSChris Mason ret = wret; 1145e66f709bSChris Mason if (wret == 0) { 11465f39d397SChris Mason struct btrfs_disk_key disk_key; 1147e66f709bSChris Mason orig_slot += left_nr; 11485f39d397SChris Mason btrfs_node_key(mid, &disk_key, 0); 1149f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_insert_key(parent, pslot, 1150f3a84ccdSFilipe Manana BTRFS_MOD_LOG_KEY_REPLACE, GFP_NOFS); 11510e82bcfeSDavid Sterba BUG_ON(ret < 0); 11525f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot); 11535f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 11545f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 11555f39d397SChris Mason path->nodes[level] = left; 1156e66f709bSChris Mason path->slots[level + 1] -= 1; 1157e66f709bSChris Mason path->slots[level] = orig_slot; 1158925baeddSChris Mason btrfs_tree_unlock(mid); 11595f39d397SChris Mason free_extent_buffer(mid); 1160e66f709bSChris Mason } else { 1161e66f709bSChris Mason orig_slot -= 11625f39d397SChris Mason btrfs_header_nritems(left); 1163e66f709bSChris Mason path->slots[level] = orig_slot; 1164925baeddSChris Mason btrfs_tree_unlock(left); 11655f39d397SChris Mason free_extent_buffer(left); 1166e66f709bSChris Mason } 1167e66f709bSChris Mason return 0; 1168e66f709bSChris Mason } 1169925baeddSChris Mason btrfs_tree_unlock(left); 11705f39d397SChris Mason free_extent_buffer(left); 1171e66f709bSChris Mason } 11724b231ae4SDavid Sterba right = btrfs_read_node_slot(parent, pslot + 1); 1173fb770ae4SLiu Bo if (IS_ERR(right)) 1174fb770ae4SLiu Bo right = NULL; 1175e66f709bSChris Mason 1176e66f709bSChris Mason /* 1177e66f709bSChris Mason * then try to empty the right most buffer into the middle 1178e66f709bSChris Mason */ 11795f39d397SChris Mason if (right) { 118033ade1f8SChris Mason u32 right_nr; 1181b4ce94deSChris Mason 1182bf77467aSJosef Bacik __btrfs_tree_lock(right, BTRFS_NESTING_RIGHT); 1183b4ce94deSChris Mason 11845f39d397SChris Mason right_nr = btrfs_header_nritems(right); 11850b246afaSJeff Mahoney if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(fs_info) - 1) { 118633ade1f8SChris Mason wret = 1; 118733ade1f8SChris Mason } else { 11885f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, 11895f39d397SChris Mason parent, pslot + 1, 1190bf59a5a2SJosef Bacik &right, BTRFS_NESTING_RIGHT_COW); 119154aa1f4dSChris Mason if (ret) 119254aa1f4dSChris Mason wret = 1; 119354aa1f4dSChris Mason else { 119455d32ed8SDavid Sterba wret = balance_node_right(trans, right, mid); 119533ade1f8SChris Mason } 119654aa1f4dSChris Mason } 1197e66f709bSChris Mason if (wret < 0) 1198e66f709bSChris Mason ret = wret; 1199e66f709bSChris Mason if (wret == 0) { 12005f39d397SChris Mason struct btrfs_disk_key disk_key; 12015f39d397SChris Mason 12025f39d397SChris Mason btrfs_node_key(right, &disk_key, 0); 1203f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_insert_key(parent, pslot + 1, 1204f3a84ccdSFilipe Manana BTRFS_MOD_LOG_KEY_REPLACE, GFP_NOFS); 12050e82bcfeSDavid Sterba BUG_ON(ret < 0); 12065f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot + 1); 12075f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 12085f39d397SChris Mason 12095f39d397SChris Mason if (btrfs_header_nritems(mid) <= orig_slot) { 12105f39d397SChris Mason path->nodes[level] = right; 1211e66f709bSChris Mason path->slots[level + 1] += 1; 1212e66f709bSChris Mason path->slots[level] = orig_slot - 12135f39d397SChris Mason btrfs_header_nritems(mid); 1214925baeddSChris Mason btrfs_tree_unlock(mid); 12155f39d397SChris Mason free_extent_buffer(mid); 1216e66f709bSChris Mason } else { 1217925baeddSChris Mason btrfs_tree_unlock(right); 12185f39d397SChris Mason free_extent_buffer(right); 1219e66f709bSChris Mason } 1220e66f709bSChris Mason return 0; 1221e66f709bSChris Mason } 1222925baeddSChris Mason btrfs_tree_unlock(right); 12235f39d397SChris Mason free_extent_buffer(right); 1224e66f709bSChris Mason } 1225e66f709bSChris Mason return 1; 1226e66f709bSChris Mason } 1227e66f709bSChris Mason 122874123bd7SChris Mason /* 1229d352ac68SChris Mason * readahead one full node of leaves, finding things that are close 1230d352ac68SChris Mason * to the block in 'slot', and triggering ra on them. 12313c69faecSChris Mason */ 12322ff7e61eSJeff Mahoney static void reada_for_search(struct btrfs_fs_info *fs_info, 1233e02119d5SChris Mason struct btrfs_path *path, 123401f46658SChris Mason int level, int slot, u64 objectid) 12353c69faecSChris Mason { 12365f39d397SChris Mason struct extent_buffer *node; 123701f46658SChris Mason struct btrfs_disk_key disk_key; 12383c69faecSChris Mason u32 nritems; 12393c69faecSChris Mason u64 search; 1240a7175319SChris Mason u64 target; 12416b80053dSChris Mason u64 nread = 0; 1242ace75066SFilipe Manana u64 nread_max; 12436b80053dSChris Mason u32 nr; 12446b80053dSChris Mason u32 blocksize; 12456b80053dSChris Mason u32 nscan = 0; 1246db94535dSChris Mason 1247ace75066SFilipe Manana if (level != 1 && path->reada != READA_FORWARD_ALWAYS) 12483c69faecSChris Mason return; 12493c69faecSChris Mason 12506702ed49SChris Mason if (!path->nodes[level]) 12516702ed49SChris Mason return; 12526702ed49SChris Mason 12535f39d397SChris Mason node = path->nodes[level]; 1254925baeddSChris Mason 1255ace75066SFilipe Manana /* 1256ace75066SFilipe Manana * Since the time between visiting leaves is much shorter than the time 1257ace75066SFilipe Manana * between visiting nodes, limit read ahead of nodes to 1, to avoid too 1258ace75066SFilipe Manana * much IO at once (possibly random). 1259ace75066SFilipe Manana */ 1260ace75066SFilipe Manana if (path->reada == READA_FORWARD_ALWAYS) { 1261ace75066SFilipe Manana if (level > 1) 1262ace75066SFilipe Manana nread_max = node->fs_info->nodesize; 1263ace75066SFilipe Manana else 1264ace75066SFilipe Manana nread_max = SZ_128K; 1265ace75066SFilipe Manana } else { 1266ace75066SFilipe Manana nread_max = SZ_64K; 1267ace75066SFilipe Manana } 1268ace75066SFilipe Manana 12693c69faecSChris Mason search = btrfs_node_blockptr(node, slot); 12700b246afaSJeff Mahoney blocksize = fs_info->nodesize; 1271069a2e37SFilipe Manana if (path->reada != READA_FORWARD_ALWAYS) { 1272069a2e37SFilipe Manana struct extent_buffer *eb; 1273069a2e37SFilipe Manana 12740b246afaSJeff Mahoney eb = find_extent_buffer(fs_info, search); 12755f39d397SChris Mason if (eb) { 12765f39d397SChris Mason free_extent_buffer(eb); 12773c69faecSChris Mason return; 12783c69faecSChris Mason } 1279069a2e37SFilipe Manana } 12803c69faecSChris Mason 1281a7175319SChris Mason target = search; 12826b80053dSChris Mason 12835f39d397SChris Mason nritems = btrfs_header_nritems(node); 12846b80053dSChris Mason nr = slot; 128525b8b936SJosef Bacik 12863c69faecSChris Mason while (1) { 1287e4058b54SDavid Sterba if (path->reada == READA_BACK) { 12886b80053dSChris Mason if (nr == 0) 12893c69faecSChris Mason break; 12906b80053dSChris Mason nr--; 1291ace75066SFilipe Manana } else if (path->reada == READA_FORWARD || 1292ace75066SFilipe Manana path->reada == READA_FORWARD_ALWAYS) { 12936b80053dSChris Mason nr++; 12946b80053dSChris Mason if (nr >= nritems) 12956b80053dSChris Mason break; 12963c69faecSChris Mason } 1297e4058b54SDavid Sterba if (path->reada == READA_BACK && objectid) { 129801f46658SChris Mason btrfs_node_key(node, &disk_key, nr); 129901f46658SChris Mason if (btrfs_disk_key_objectid(&disk_key) != objectid) 130001f46658SChris Mason break; 130101f46658SChris Mason } 13026b80053dSChris Mason search = btrfs_node_blockptr(node, nr); 1303ace75066SFilipe Manana if (path->reada == READA_FORWARD_ALWAYS || 1304ace75066SFilipe Manana (search <= target && target - search <= 65536) || 1305a7175319SChris Mason (search > target && search - target <= 65536)) { 1306bfb484d9SJosef Bacik btrfs_readahead_node_child(node, nr); 13076b80053dSChris Mason nread += blocksize; 13083c69faecSChris Mason } 13096b80053dSChris Mason nscan++; 1310ace75066SFilipe Manana if (nread > nread_max || nscan > 32) 13116b80053dSChris Mason break; 13123c69faecSChris Mason } 13133c69faecSChris Mason } 1314925baeddSChris Mason 1315bfb484d9SJosef Bacik static noinline void reada_for_balance(struct btrfs_path *path, int level) 1316b4ce94deSChris Mason { 1317bfb484d9SJosef Bacik struct extent_buffer *parent; 1318b4ce94deSChris Mason int slot; 1319b4ce94deSChris Mason int nritems; 1320b4ce94deSChris Mason 13218c594ea8SChris Mason parent = path->nodes[level + 1]; 1322b4ce94deSChris Mason if (!parent) 13230b08851fSJosef Bacik return; 1324b4ce94deSChris Mason 1325b4ce94deSChris Mason nritems = btrfs_header_nritems(parent); 13268c594ea8SChris Mason slot = path->slots[level + 1]; 1327b4ce94deSChris Mason 1328bfb484d9SJosef Bacik if (slot > 0) 1329bfb484d9SJosef Bacik btrfs_readahead_node_child(parent, slot - 1); 1330bfb484d9SJosef Bacik if (slot + 1 < nritems) 1331bfb484d9SJosef Bacik btrfs_readahead_node_child(parent, slot + 1); 1332b4ce94deSChris Mason } 1333b4ce94deSChris Mason 1334b4ce94deSChris Mason 1335b4ce94deSChris Mason /* 1336d397712bSChris Mason * when we walk down the tree, it is usually safe to unlock the higher layers 1337d397712bSChris Mason * in the tree. The exceptions are when our path goes through slot 0, because 1338d397712bSChris Mason * operations on the tree might require changing key pointers higher up in the 1339d397712bSChris Mason * tree. 1340d352ac68SChris Mason * 1341d397712bSChris Mason * callers might also have set path->keep_locks, which tells this code to keep 1342d397712bSChris Mason * the lock if the path points to the last slot in the block. This is part of 1343d397712bSChris Mason * walking through the tree, and selecting the next slot in the higher block. 1344d352ac68SChris Mason * 1345d397712bSChris Mason * lowest_unlock sets the lowest level in the tree we're allowed to unlock. so 1346d397712bSChris Mason * if lowest_unlock is 1, level 0 won't be unlocked 1347d352ac68SChris Mason */ 1348e02119d5SChris Mason static noinline void unlock_up(struct btrfs_path *path, int level, 1349f7c79f30SChris Mason int lowest_unlock, int min_write_lock_level, 1350f7c79f30SChris Mason int *write_lock_level) 1351925baeddSChris Mason { 1352925baeddSChris Mason int i; 1353925baeddSChris Mason int skip_level = level; 1354c1227996SNikolay Borisov bool check_skip = true; 1355925baeddSChris Mason 1356925baeddSChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 1357925baeddSChris Mason if (!path->nodes[i]) 1358925baeddSChris Mason break; 1359925baeddSChris Mason if (!path->locks[i]) 1360925baeddSChris Mason break; 1361c1227996SNikolay Borisov 1362c1227996SNikolay Borisov if (check_skip) { 1363c1227996SNikolay Borisov if (path->slots[i] == 0) { 1364925baeddSChris Mason skip_level = i + 1; 1365925baeddSChris Mason continue; 1366925baeddSChris Mason } 1367c1227996SNikolay Borisov 1368c1227996SNikolay Borisov if (path->keep_locks) { 1369925baeddSChris Mason u32 nritems; 1370c1227996SNikolay Borisov 1371c1227996SNikolay Borisov nritems = btrfs_header_nritems(path->nodes[i]); 1372051e1b9fSChris Mason if (nritems < 1 || path->slots[i] >= nritems - 1) { 1373925baeddSChris Mason skip_level = i + 1; 1374925baeddSChris Mason continue; 1375925baeddSChris Mason } 1376925baeddSChris Mason } 1377c1227996SNikolay Borisov } 1378051e1b9fSChris Mason 1379d80bb3f9SLiu Bo if (i >= lowest_unlock && i > skip_level) { 1380c1227996SNikolay Borisov check_skip = false; 1381c1227996SNikolay Borisov btrfs_tree_unlock_rw(path->nodes[i], path->locks[i]); 1382925baeddSChris Mason path->locks[i] = 0; 1383f7c79f30SChris Mason if (write_lock_level && 1384f7c79f30SChris Mason i > min_write_lock_level && 1385f7c79f30SChris Mason i <= *write_lock_level) { 1386f7c79f30SChris Mason *write_lock_level = i - 1; 1387f7c79f30SChris Mason } 1388925baeddSChris Mason } 1389925baeddSChris Mason } 1390925baeddSChris Mason } 1391925baeddSChris Mason 13923c69faecSChris Mason /* 1393c8c42864SChris Mason * helper function for btrfs_search_slot. The goal is to find a block 1394c8c42864SChris Mason * in cache without setting the path to blocking. If we find the block 1395c8c42864SChris Mason * we return zero and the path is unchanged. 1396c8c42864SChris Mason * 1397c8c42864SChris Mason * If we can't find the block, we set the path blocking and do some 1398c8c42864SChris Mason * reada. -EAGAIN is returned and the search must be repeated. 1399c8c42864SChris Mason */ 1400c8c42864SChris Mason static int 1401d07b8528SLiu Bo read_block_for_search(struct btrfs_root *root, struct btrfs_path *p, 1402c8c42864SChris Mason struct extent_buffer **eb_ret, int level, int slot, 1403cda79c54SDavid Sterba const struct btrfs_key *key) 1404c8c42864SChris Mason { 14050b246afaSJeff Mahoney struct btrfs_fs_info *fs_info = root->fs_info; 1406c8c42864SChris Mason u64 blocknr; 1407c8c42864SChris Mason u64 gen; 1408c8c42864SChris Mason struct extent_buffer *tmp; 1409581c1760SQu Wenruo struct btrfs_key first_key; 141076a05b35SChris Mason int ret; 1411581c1760SQu Wenruo int parent_level; 1412c8c42864SChris Mason 1413213ff4b7SNikolay Borisov blocknr = btrfs_node_blockptr(*eb_ret, slot); 1414213ff4b7SNikolay Borisov gen = btrfs_node_ptr_generation(*eb_ret, slot); 1415213ff4b7SNikolay Borisov parent_level = btrfs_header_level(*eb_ret); 1416213ff4b7SNikolay Borisov btrfs_node_key_to_cpu(*eb_ret, &first_key, slot); 1417c8c42864SChris Mason 14180b246afaSJeff Mahoney tmp = find_extent_buffer(fs_info, blocknr); 1419cb44921aSChris Mason if (tmp) { 1420ace75066SFilipe Manana if (p->reada == READA_FORWARD_ALWAYS) 1421ace75066SFilipe Manana reada_for_search(fs_info, p, level, slot, key->objectid); 1422ace75066SFilipe Manana 1423b9fab919SChris Mason /* first we do an atomic uptodate check */ 1424b9fab919SChris Mason if (btrfs_buffer_uptodate(tmp, gen, 1) > 0) { 1425448de471SQu Wenruo /* 1426448de471SQu Wenruo * Do extra check for first_key, eb can be stale due to 1427448de471SQu Wenruo * being cached, read from scrub, or have multiple 1428448de471SQu Wenruo * parents (shared tree blocks). 1429448de471SQu Wenruo */ 1430e064d5e9SDavid Sterba if (btrfs_verify_level_key(tmp, 1431448de471SQu Wenruo parent_level - 1, &first_key, gen)) { 1432448de471SQu Wenruo free_extent_buffer(tmp); 1433448de471SQu Wenruo return -EUCLEAN; 1434448de471SQu Wenruo } 1435c8c42864SChris Mason *eb_ret = tmp; 1436c8c42864SChris Mason return 0; 1437c8c42864SChris Mason } 1438bdf7c00eSJosef Bacik 1439b9fab919SChris Mason /* now we're allowed to do a blocking uptodate check */ 1440581c1760SQu Wenruo ret = btrfs_read_buffer(tmp, gen, parent_level - 1, &first_key); 1441*9a4ffa1bSQu Wenruo if (ret) { 1442cb44921aSChris Mason free_extent_buffer(tmp); 1443b3b4aa74SDavid Sterba btrfs_release_path(p); 1444cb44921aSChris Mason return -EIO; 1445cb44921aSChris Mason } 1446*9a4ffa1bSQu Wenruo *eb_ret = tmp; 1447*9a4ffa1bSQu Wenruo return 0; 1448*9a4ffa1bSQu Wenruo } 1449c8c42864SChris Mason 1450c8c42864SChris Mason /* 1451c8c42864SChris Mason * reduce lock contention at high levels 1452c8c42864SChris Mason * of the btree by dropping locks before 145376a05b35SChris Mason * we read. Don't release the lock on the current 145476a05b35SChris Mason * level because we need to walk this node to figure 145576a05b35SChris Mason * out which blocks to read. 1456c8c42864SChris Mason */ 14578c594ea8SChris Mason btrfs_unlock_up_safe(p, level + 1); 14588c594ea8SChris Mason 1459e4058b54SDavid Sterba if (p->reada != READA_NONE) 14602ff7e61eSJeff Mahoney reada_for_search(fs_info, p, level, slot, key->objectid); 1461c8c42864SChris Mason 146276a05b35SChris Mason ret = -EAGAIN; 14631b7ec85eSJosef Bacik tmp = read_tree_block(fs_info, blocknr, root->root_key.objectid, 14641b7ec85eSJosef Bacik gen, parent_level - 1, &first_key); 14654eb150d6SQu Wenruo if (IS_ERR(tmp)) { 14664eb150d6SQu Wenruo btrfs_release_path(p); 14674eb150d6SQu Wenruo return PTR_ERR(tmp); 14684eb150d6SQu Wenruo } 146976a05b35SChris Mason /* 147076a05b35SChris Mason * If the read above didn't mark this buffer up to date, 147176a05b35SChris Mason * it will never end up being up to date. Set ret to EIO now 147276a05b35SChris Mason * and give up so that our caller doesn't loop forever 147376a05b35SChris Mason * on our EAGAINs. 147476a05b35SChris Mason */ 1475e6a1d6fdSLiu Bo if (!extent_buffer_uptodate(tmp)) 147676a05b35SChris Mason ret = -EIO; 1477c8c42864SChris Mason free_extent_buffer(tmp); 147802a3307aSLiu Bo 147902a3307aSLiu Bo btrfs_release_path(p); 148076a05b35SChris Mason return ret; 1481c8c42864SChris Mason } 1482c8c42864SChris Mason 1483c8c42864SChris Mason /* 1484c8c42864SChris Mason * helper function for btrfs_search_slot. This does all of the checks 1485c8c42864SChris Mason * for node-level blocks and does any balancing required based on 1486c8c42864SChris Mason * the ins_len. 1487c8c42864SChris Mason * 1488c8c42864SChris Mason * If no extra work was required, zero is returned. If we had to 1489c8c42864SChris Mason * drop the path, -EAGAIN is returned and btrfs_search_slot must 1490c8c42864SChris Mason * start over 1491c8c42864SChris Mason */ 1492c8c42864SChris Mason static int 1493c8c42864SChris Mason setup_nodes_for_search(struct btrfs_trans_handle *trans, 1494c8c42864SChris Mason struct btrfs_root *root, struct btrfs_path *p, 1495bd681513SChris Mason struct extent_buffer *b, int level, int ins_len, 1496bd681513SChris Mason int *write_lock_level) 1497c8c42864SChris Mason { 14980b246afaSJeff Mahoney struct btrfs_fs_info *fs_info = root->fs_info; 149995b982deSNikolay Borisov int ret = 0; 15000b246afaSJeff Mahoney 1501c8c42864SChris Mason if ((p->search_for_split || ins_len > 0) && btrfs_header_nritems(b) >= 15020b246afaSJeff Mahoney BTRFS_NODEPTRS_PER_BLOCK(fs_info) - 3) { 1503c8c42864SChris Mason 1504bd681513SChris Mason if (*write_lock_level < level + 1) { 1505bd681513SChris Mason *write_lock_level = level + 1; 1506bd681513SChris Mason btrfs_release_path(p); 150795b982deSNikolay Borisov return -EAGAIN; 1508bd681513SChris Mason } 1509bd681513SChris Mason 1510bfb484d9SJosef Bacik reada_for_balance(p, level); 151195b982deSNikolay Borisov ret = split_node(trans, root, p, level); 1512c8c42864SChris Mason 1513c8c42864SChris Mason b = p->nodes[level]; 1514c8c42864SChris Mason } else if (ins_len < 0 && btrfs_header_nritems(b) < 15150b246afaSJeff Mahoney BTRFS_NODEPTRS_PER_BLOCK(fs_info) / 2) { 1516c8c42864SChris Mason 1517bd681513SChris Mason if (*write_lock_level < level + 1) { 1518bd681513SChris Mason *write_lock_level = level + 1; 1519bd681513SChris Mason btrfs_release_path(p); 152095b982deSNikolay Borisov return -EAGAIN; 1521bd681513SChris Mason } 1522bd681513SChris Mason 1523bfb484d9SJosef Bacik reada_for_balance(p, level); 152495b982deSNikolay Borisov ret = balance_level(trans, root, p, level); 152595b982deSNikolay Borisov if (ret) 152695b982deSNikolay Borisov return ret; 1527c8c42864SChris Mason 1528c8c42864SChris Mason b = p->nodes[level]; 1529c8c42864SChris Mason if (!b) { 1530b3b4aa74SDavid Sterba btrfs_release_path(p); 153195b982deSNikolay Borisov return -EAGAIN; 1532c8c42864SChris Mason } 1533c8c42864SChris Mason BUG_ON(btrfs_header_nritems(b) == 1); 1534c8c42864SChris Mason } 1535c8c42864SChris Mason return ret; 1536c8c42864SChris Mason } 1537c8c42864SChris Mason 1538381cf658SDavid Sterba int btrfs_find_item(struct btrfs_root *fs_root, struct btrfs_path *path, 1539e33d5c3dSKelley Nielsen u64 iobjectid, u64 ioff, u8 key_type, 1540e33d5c3dSKelley Nielsen struct btrfs_key *found_key) 1541e33d5c3dSKelley Nielsen { 1542e33d5c3dSKelley Nielsen int ret; 1543e33d5c3dSKelley Nielsen struct btrfs_key key; 1544e33d5c3dSKelley Nielsen struct extent_buffer *eb; 1545381cf658SDavid Sterba 1546381cf658SDavid Sterba ASSERT(path); 15471d4c08e0SDavid Sterba ASSERT(found_key); 1548e33d5c3dSKelley Nielsen 1549e33d5c3dSKelley Nielsen key.type = key_type; 1550e33d5c3dSKelley Nielsen key.objectid = iobjectid; 1551e33d5c3dSKelley Nielsen key.offset = ioff; 1552e33d5c3dSKelley Nielsen 1553e33d5c3dSKelley Nielsen ret = btrfs_search_slot(NULL, fs_root, &key, path, 0, 0); 15541d4c08e0SDavid Sterba if (ret < 0) 1555e33d5c3dSKelley Nielsen return ret; 1556e33d5c3dSKelley Nielsen 1557e33d5c3dSKelley Nielsen eb = path->nodes[0]; 1558e33d5c3dSKelley Nielsen if (ret && path->slots[0] >= btrfs_header_nritems(eb)) { 1559e33d5c3dSKelley Nielsen ret = btrfs_next_leaf(fs_root, path); 1560e33d5c3dSKelley Nielsen if (ret) 1561e33d5c3dSKelley Nielsen return ret; 1562e33d5c3dSKelley Nielsen eb = path->nodes[0]; 1563e33d5c3dSKelley Nielsen } 1564e33d5c3dSKelley Nielsen 1565e33d5c3dSKelley Nielsen btrfs_item_key_to_cpu(eb, found_key, path->slots[0]); 1566e33d5c3dSKelley Nielsen if (found_key->type != key.type || 1567e33d5c3dSKelley Nielsen found_key->objectid != key.objectid) 1568e33d5c3dSKelley Nielsen return 1; 1569e33d5c3dSKelley Nielsen 1570e33d5c3dSKelley Nielsen return 0; 1571e33d5c3dSKelley Nielsen } 1572e33d5c3dSKelley Nielsen 15731fc28d8eSLiu Bo static struct extent_buffer *btrfs_search_slot_get_root(struct btrfs_root *root, 15741fc28d8eSLiu Bo struct btrfs_path *p, 15751fc28d8eSLiu Bo int write_lock_level) 15761fc28d8eSLiu Bo { 15771fc28d8eSLiu Bo struct extent_buffer *b; 1578120de408SJosef Bacik int root_lock = 0; 15791fc28d8eSLiu Bo int level = 0; 15801fc28d8eSLiu Bo 15811fc28d8eSLiu Bo if (p->search_commit_root) { 15821fc28d8eSLiu Bo b = root->commit_root; 158367439dadSDavid Sterba atomic_inc(&b->refs); 15841fc28d8eSLiu Bo level = btrfs_header_level(b); 1585f9ddfd05SLiu Bo /* 1586f9ddfd05SLiu Bo * Ensure that all callers have set skip_locking when 1587f9ddfd05SLiu Bo * p->search_commit_root = 1. 1588f9ddfd05SLiu Bo */ 1589f9ddfd05SLiu Bo ASSERT(p->skip_locking == 1); 15901fc28d8eSLiu Bo 15911fc28d8eSLiu Bo goto out; 15921fc28d8eSLiu Bo } 15931fc28d8eSLiu Bo 15941fc28d8eSLiu Bo if (p->skip_locking) { 15951fc28d8eSLiu Bo b = btrfs_root_node(root); 15961fc28d8eSLiu Bo level = btrfs_header_level(b); 15971fc28d8eSLiu Bo goto out; 15981fc28d8eSLiu Bo } 15991fc28d8eSLiu Bo 1600120de408SJosef Bacik /* We try very hard to do read locks on the root */ 1601120de408SJosef Bacik root_lock = BTRFS_READ_LOCK; 1602120de408SJosef Bacik 16031fc28d8eSLiu Bo /* 1604662c653bSLiu Bo * If the level is set to maximum, we can skip trying to get the read 1605662c653bSLiu Bo * lock. 1606662c653bSLiu Bo */ 1607662c653bSLiu Bo if (write_lock_level < BTRFS_MAX_LEVEL) { 1608662c653bSLiu Bo /* 1609662c653bSLiu Bo * We don't know the level of the root node until we actually 1610662c653bSLiu Bo * have it read locked 16111fc28d8eSLiu Bo */ 16121bb96598SJosef Bacik b = btrfs_read_lock_root_node(root); 16131fc28d8eSLiu Bo level = btrfs_header_level(b); 16141fc28d8eSLiu Bo if (level > write_lock_level) 16151fc28d8eSLiu Bo goto out; 16161fc28d8eSLiu Bo 1617662c653bSLiu Bo /* Whoops, must trade for write lock */ 16181fc28d8eSLiu Bo btrfs_tree_read_unlock(b); 16191fc28d8eSLiu Bo free_extent_buffer(b); 1620662c653bSLiu Bo } 1621662c653bSLiu Bo 16221fc28d8eSLiu Bo b = btrfs_lock_root_node(root); 16231fc28d8eSLiu Bo root_lock = BTRFS_WRITE_LOCK; 16241fc28d8eSLiu Bo 16251fc28d8eSLiu Bo /* The level might have changed, check again */ 16261fc28d8eSLiu Bo level = btrfs_header_level(b); 16271fc28d8eSLiu Bo 16281fc28d8eSLiu Bo out: 1629120de408SJosef Bacik /* 1630120de408SJosef Bacik * The root may have failed to write out at some point, and thus is no 1631120de408SJosef Bacik * longer valid, return an error in this case. 1632120de408SJosef Bacik */ 1633120de408SJosef Bacik if (!extent_buffer_uptodate(b)) { 1634120de408SJosef Bacik if (root_lock) 1635120de408SJosef Bacik btrfs_tree_unlock_rw(b, root_lock); 1636120de408SJosef Bacik free_extent_buffer(b); 1637120de408SJosef Bacik return ERR_PTR(-EIO); 1638120de408SJosef Bacik } 1639120de408SJosef Bacik 16401fc28d8eSLiu Bo p->nodes[level] = b; 16411fc28d8eSLiu Bo if (!p->skip_locking) 16421fc28d8eSLiu Bo p->locks[level] = root_lock; 16431fc28d8eSLiu Bo /* 16441fc28d8eSLiu Bo * Callers are responsible for dropping b's references. 16451fc28d8eSLiu Bo */ 16461fc28d8eSLiu Bo return b; 16471fc28d8eSLiu Bo } 16481fc28d8eSLiu Bo 1649d96b3424SFilipe Manana /* 1650d96b3424SFilipe Manana * Replace the extent buffer at the lowest level of the path with a cloned 1651d96b3424SFilipe Manana * version. The purpose is to be able to use it safely, after releasing the 1652d96b3424SFilipe Manana * commit root semaphore, even if relocation is happening in parallel, the 1653d96b3424SFilipe Manana * transaction used for relocation is committed and the extent buffer is 1654d96b3424SFilipe Manana * reallocated in the next transaction. 1655d96b3424SFilipe Manana * 1656d96b3424SFilipe Manana * This is used in a context where the caller does not prevent transaction 1657d96b3424SFilipe Manana * commits from happening, either by holding a transaction handle or holding 1658d96b3424SFilipe Manana * some lock, while it's doing searches through a commit root. 1659d96b3424SFilipe Manana * At the moment it's only used for send operations. 1660d96b3424SFilipe Manana */ 1661d96b3424SFilipe Manana static int finish_need_commit_sem_search(struct btrfs_path *path) 1662d96b3424SFilipe Manana { 1663d96b3424SFilipe Manana const int i = path->lowest_level; 1664d96b3424SFilipe Manana const int slot = path->slots[i]; 1665d96b3424SFilipe Manana struct extent_buffer *lowest = path->nodes[i]; 1666d96b3424SFilipe Manana struct extent_buffer *clone; 1667d96b3424SFilipe Manana 1668d96b3424SFilipe Manana ASSERT(path->need_commit_sem); 1669d96b3424SFilipe Manana 1670d96b3424SFilipe Manana if (!lowest) 1671d96b3424SFilipe Manana return 0; 1672d96b3424SFilipe Manana 1673d96b3424SFilipe Manana lockdep_assert_held_read(&lowest->fs_info->commit_root_sem); 1674d96b3424SFilipe Manana 1675d96b3424SFilipe Manana clone = btrfs_clone_extent_buffer(lowest); 1676d96b3424SFilipe Manana if (!clone) 1677d96b3424SFilipe Manana return -ENOMEM; 1678d96b3424SFilipe Manana 1679d96b3424SFilipe Manana btrfs_release_path(path); 1680d96b3424SFilipe Manana path->nodes[i] = clone; 1681d96b3424SFilipe Manana path->slots[i] = slot; 1682d96b3424SFilipe Manana 1683d96b3424SFilipe Manana return 0; 1684d96b3424SFilipe Manana } 16851fc28d8eSLiu Bo 1686e2e58d0fSFilipe Manana static inline int search_for_key_slot(struct extent_buffer *eb, 1687e2e58d0fSFilipe Manana int search_low_slot, 1688e2e58d0fSFilipe Manana const struct btrfs_key *key, 1689e2e58d0fSFilipe Manana int prev_cmp, 1690e2e58d0fSFilipe Manana int *slot) 1691e2e58d0fSFilipe Manana { 1692e2e58d0fSFilipe Manana /* 1693e2e58d0fSFilipe Manana * If a previous call to btrfs_bin_search() on a parent node returned an 1694e2e58d0fSFilipe Manana * exact match (prev_cmp == 0), we can safely assume the target key will 1695e2e58d0fSFilipe Manana * always be at slot 0 on lower levels, since each key pointer 1696e2e58d0fSFilipe Manana * (struct btrfs_key_ptr) refers to the lowest key accessible from the 1697e2e58d0fSFilipe Manana * subtree it points to. Thus we can skip searching lower levels. 1698e2e58d0fSFilipe Manana */ 1699e2e58d0fSFilipe Manana if (prev_cmp == 0) { 1700e2e58d0fSFilipe Manana *slot = 0; 1701e2e58d0fSFilipe Manana return 0; 1702e2e58d0fSFilipe Manana } 1703e2e58d0fSFilipe Manana 1704e2e58d0fSFilipe Manana return generic_bin_search(eb, search_low_slot, key, slot); 1705e2e58d0fSFilipe Manana } 1706e2e58d0fSFilipe Manana 1707109324cfSFilipe Manana static int search_leaf(struct btrfs_trans_handle *trans, 1708109324cfSFilipe Manana struct btrfs_root *root, 1709109324cfSFilipe Manana const struct btrfs_key *key, 1710109324cfSFilipe Manana struct btrfs_path *path, 1711109324cfSFilipe Manana int ins_len, 1712109324cfSFilipe Manana int prev_cmp) 1713109324cfSFilipe Manana { 1714109324cfSFilipe Manana struct extent_buffer *leaf = path->nodes[0]; 1715109324cfSFilipe Manana int leaf_free_space = -1; 1716109324cfSFilipe Manana int search_low_slot = 0; 1717109324cfSFilipe Manana int ret; 1718109324cfSFilipe Manana bool do_bin_search = true; 1719109324cfSFilipe Manana 1720109324cfSFilipe Manana /* 1721109324cfSFilipe Manana * If we are doing an insertion, the leaf has enough free space and the 1722109324cfSFilipe Manana * destination slot for the key is not slot 0, then we can unlock our 1723109324cfSFilipe Manana * write lock on the parent, and any other upper nodes, before doing the 1724109324cfSFilipe Manana * binary search on the leaf (with search_for_key_slot()), allowing other 1725109324cfSFilipe Manana * tasks to lock the parent and any other upper nodes. 1726109324cfSFilipe Manana */ 1727109324cfSFilipe Manana if (ins_len > 0) { 1728109324cfSFilipe Manana /* 1729109324cfSFilipe Manana * Cache the leaf free space, since we will need it later and it 1730109324cfSFilipe Manana * will not change until then. 1731109324cfSFilipe Manana */ 1732109324cfSFilipe Manana leaf_free_space = btrfs_leaf_free_space(leaf); 1733109324cfSFilipe Manana 1734109324cfSFilipe Manana /* 1735109324cfSFilipe Manana * !path->locks[1] means we have a single node tree, the leaf is 1736109324cfSFilipe Manana * the root of the tree. 1737109324cfSFilipe Manana */ 1738109324cfSFilipe Manana if (path->locks[1] && leaf_free_space >= ins_len) { 1739109324cfSFilipe Manana struct btrfs_disk_key first_key; 1740109324cfSFilipe Manana 1741109324cfSFilipe Manana ASSERT(btrfs_header_nritems(leaf) > 0); 1742109324cfSFilipe Manana btrfs_item_key(leaf, &first_key, 0); 1743109324cfSFilipe Manana 1744109324cfSFilipe Manana /* 1745109324cfSFilipe Manana * Doing the extra comparison with the first key is cheap, 1746109324cfSFilipe Manana * taking into account that the first key is very likely 1747109324cfSFilipe Manana * already in a cache line because it immediately follows 1748109324cfSFilipe Manana * the extent buffer's header and we have recently accessed 1749109324cfSFilipe Manana * the header's level field. 1750109324cfSFilipe Manana */ 1751109324cfSFilipe Manana ret = comp_keys(&first_key, key); 1752109324cfSFilipe Manana if (ret < 0) { 1753109324cfSFilipe Manana /* 1754109324cfSFilipe Manana * The first key is smaller than the key we want 1755109324cfSFilipe Manana * to insert, so we are safe to unlock all upper 1756109324cfSFilipe Manana * nodes and we have to do the binary search. 1757109324cfSFilipe Manana * 1758109324cfSFilipe Manana * We do use btrfs_unlock_up_safe() and not 1759109324cfSFilipe Manana * unlock_up() because the later does not unlock 1760109324cfSFilipe Manana * nodes with a slot of 0 - we can safely unlock 1761109324cfSFilipe Manana * any node even if its slot is 0 since in this 1762109324cfSFilipe Manana * case the key does not end up at slot 0 of the 1763109324cfSFilipe Manana * leaf and there's no need to split the leaf. 1764109324cfSFilipe Manana */ 1765109324cfSFilipe Manana btrfs_unlock_up_safe(path, 1); 1766109324cfSFilipe Manana search_low_slot = 1; 1767109324cfSFilipe Manana } else { 1768109324cfSFilipe Manana /* 1769109324cfSFilipe Manana * The first key is >= then the key we want to 1770109324cfSFilipe Manana * insert, so we can skip the binary search as 1771109324cfSFilipe Manana * the target key will be at slot 0. 1772109324cfSFilipe Manana * 1773109324cfSFilipe Manana * We can not unlock upper nodes when the key is 1774109324cfSFilipe Manana * less than the first key, because we will need 1775109324cfSFilipe Manana * to update the key at slot 0 of the parent node 1776109324cfSFilipe Manana * and possibly of other upper nodes too. 1777109324cfSFilipe Manana * If the key matches the first key, then we can 1778109324cfSFilipe Manana * unlock all the upper nodes, using 1779109324cfSFilipe Manana * btrfs_unlock_up_safe() instead of unlock_up() 1780109324cfSFilipe Manana * as stated above. 1781109324cfSFilipe Manana */ 1782109324cfSFilipe Manana if (ret == 0) 1783109324cfSFilipe Manana btrfs_unlock_up_safe(path, 1); 1784109324cfSFilipe Manana /* 1785109324cfSFilipe Manana * ret is already 0 or 1, matching the result of 1786109324cfSFilipe Manana * a btrfs_bin_search() call, so there is no need 1787109324cfSFilipe Manana * to adjust it. 1788109324cfSFilipe Manana */ 1789109324cfSFilipe Manana do_bin_search = false; 1790109324cfSFilipe Manana path->slots[0] = 0; 1791109324cfSFilipe Manana } 1792109324cfSFilipe Manana } 1793109324cfSFilipe Manana } 1794109324cfSFilipe Manana 1795109324cfSFilipe Manana if (do_bin_search) { 1796109324cfSFilipe Manana ret = search_for_key_slot(leaf, search_low_slot, key, 1797109324cfSFilipe Manana prev_cmp, &path->slots[0]); 1798109324cfSFilipe Manana if (ret < 0) 1799109324cfSFilipe Manana return ret; 1800109324cfSFilipe Manana } 1801109324cfSFilipe Manana 1802109324cfSFilipe Manana if (ins_len > 0) { 1803109324cfSFilipe Manana /* 1804109324cfSFilipe Manana * Item key already exists. In this case, if we are allowed to 1805109324cfSFilipe Manana * insert the item (for example, in dir_item case, item key 1806109324cfSFilipe Manana * collision is allowed), it will be merged with the original 1807109324cfSFilipe Manana * item. Only the item size grows, no new btrfs item will be 1808109324cfSFilipe Manana * added. If search_for_extension is not set, ins_len already 1809109324cfSFilipe Manana * accounts the size btrfs_item, deduct it here so leaf space 1810109324cfSFilipe Manana * check will be correct. 1811109324cfSFilipe Manana */ 1812109324cfSFilipe Manana if (ret == 0 && !path->search_for_extension) { 1813109324cfSFilipe Manana ASSERT(ins_len >= sizeof(struct btrfs_item)); 1814109324cfSFilipe Manana ins_len -= sizeof(struct btrfs_item); 1815109324cfSFilipe Manana } 1816109324cfSFilipe Manana 1817109324cfSFilipe Manana ASSERT(leaf_free_space >= 0); 1818109324cfSFilipe Manana 1819109324cfSFilipe Manana if (leaf_free_space < ins_len) { 1820109324cfSFilipe Manana int err; 1821109324cfSFilipe Manana 1822109324cfSFilipe Manana err = split_leaf(trans, root, key, path, ins_len, 1823109324cfSFilipe Manana (ret == 0)); 1824bb8e9a60SFilipe Manana ASSERT(err <= 0); 1825bb8e9a60SFilipe Manana if (WARN_ON(err > 0)) 1826bb8e9a60SFilipe Manana err = -EUCLEAN; 1827109324cfSFilipe Manana if (err) 1828109324cfSFilipe Manana ret = err; 1829109324cfSFilipe Manana } 1830109324cfSFilipe Manana } 1831109324cfSFilipe Manana 1832109324cfSFilipe Manana return ret; 1833109324cfSFilipe Manana } 1834109324cfSFilipe Manana 1835c8c42864SChris Mason /* 18364271eceaSNikolay Borisov * btrfs_search_slot - look for a key in a tree and perform necessary 18374271eceaSNikolay Borisov * modifications to preserve tree invariants. 183874123bd7SChris Mason * 18394271eceaSNikolay Borisov * @trans: Handle of transaction, used when modifying the tree 18404271eceaSNikolay Borisov * @p: Holds all btree nodes along the search path 18414271eceaSNikolay Borisov * @root: The root node of the tree 18424271eceaSNikolay Borisov * @key: The key we are looking for 18439a664971Sethanwu * @ins_len: Indicates purpose of search: 18449a664971Sethanwu * >0 for inserts it's size of item inserted (*) 18459a664971Sethanwu * <0 for deletions 18469a664971Sethanwu * 0 for plain searches, not modifying the tree 18479a664971Sethanwu * 18489a664971Sethanwu * (*) If size of item inserted doesn't include 18499a664971Sethanwu * sizeof(struct btrfs_item), then p->search_for_extension must 18509a664971Sethanwu * be set. 18514271eceaSNikolay Borisov * @cow: boolean should CoW operations be performed. Must always be 1 18524271eceaSNikolay Borisov * when modifying the tree. 185397571fd0SChris Mason * 18544271eceaSNikolay Borisov * If @ins_len > 0, nodes and leaves will be split as we walk down the tree. 18554271eceaSNikolay Borisov * If @ins_len < 0, nodes will be merged as we walk down the tree (if possible) 18564271eceaSNikolay Borisov * 18574271eceaSNikolay Borisov * If @key is found, 0 is returned and you can find the item in the leaf level 18584271eceaSNikolay Borisov * of the path (level 0) 18594271eceaSNikolay Borisov * 18604271eceaSNikolay Borisov * If @key isn't found, 1 is returned and the leaf level of the path (level 0) 18614271eceaSNikolay Borisov * points to the slot where it should be inserted 18624271eceaSNikolay Borisov * 18634271eceaSNikolay Borisov * If an error is encountered while searching the tree a negative error number 18644271eceaSNikolay Borisov * is returned 186574123bd7SChris Mason */ 1866310712b2SOmar Sandoval int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root *root, 1867310712b2SOmar Sandoval const struct btrfs_key *key, struct btrfs_path *p, 1868310712b2SOmar Sandoval int ins_len, int cow) 1869be0e5c09SChris Mason { 1870d96b3424SFilipe Manana struct btrfs_fs_info *fs_info = root->fs_info; 18715f39d397SChris Mason struct extent_buffer *b; 1872be0e5c09SChris Mason int slot; 1873be0e5c09SChris Mason int ret; 187433c66f43SYan Zheng int err; 1875be0e5c09SChris Mason int level; 1876925baeddSChris Mason int lowest_unlock = 1; 1877bd681513SChris Mason /* everything at write_lock_level or lower must be write locked */ 1878bd681513SChris Mason int write_lock_level = 0; 18799f3a7427SChris Mason u8 lowest_level = 0; 1880f7c79f30SChris Mason int min_write_lock_level; 1881d7396f07SFilipe David Borba Manana int prev_cmp; 18829f3a7427SChris Mason 18836702ed49SChris Mason lowest_level = p->lowest_level; 1884323ac95bSChris Mason WARN_ON(lowest_level && ins_len > 0); 188522b0ebdaSChris Mason WARN_ON(p->nodes[0] != NULL); 1886eb653de1SFilipe David Borba Manana BUG_ON(!cow && ins_len); 188725179201SJosef Bacik 1888bd681513SChris Mason if (ins_len < 0) { 1889925baeddSChris Mason lowest_unlock = 2; 189065b51a00SChris Mason 1891bd681513SChris Mason /* when we are removing items, we might have to go up to level 1892bd681513SChris Mason * two as we update tree pointers Make sure we keep write 1893bd681513SChris Mason * for those levels as well 1894bd681513SChris Mason */ 1895bd681513SChris Mason write_lock_level = 2; 1896bd681513SChris Mason } else if (ins_len > 0) { 1897bd681513SChris Mason /* 1898bd681513SChris Mason * for inserting items, make sure we have a write lock on 1899bd681513SChris Mason * level 1 so we can update keys 1900bd681513SChris Mason */ 1901bd681513SChris Mason write_lock_level = 1; 1902bd681513SChris Mason } 1903bd681513SChris Mason 1904bd681513SChris Mason if (!cow) 1905bd681513SChris Mason write_lock_level = -1; 1906bd681513SChris Mason 190709a2a8f9SJosef Bacik if (cow && (p->keep_locks || p->lowest_level)) 1908bd681513SChris Mason write_lock_level = BTRFS_MAX_LEVEL; 1909bd681513SChris Mason 1910f7c79f30SChris Mason min_write_lock_level = write_lock_level; 1911f7c79f30SChris Mason 1912d96b3424SFilipe Manana if (p->need_commit_sem) { 1913d96b3424SFilipe Manana ASSERT(p->search_commit_root); 1914d96b3424SFilipe Manana down_read(&fs_info->commit_root_sem); 1915d96b3424SFilipe Manana } 1916d96b3424SFilipe Manana 1917bb803951SChris Mason again: 1918d7396f07SFilipe David Borba Manana prev_cmp = -1; 19191fc28d8eSLiu Bo b = btrfs_search_slot_get_root(root, p, write_lock_level); 1920be6821f8SFilipe Manana if (IS_ERR(b)) { 1921be6821f8SFilipe Manana ret = PTR_ERR(b); 1922be6821f8SFilipe Manana goto done; 1923be6821f8SFilipe Manana } 1924925baeddSChris Mason 1925eb60ceacSChris Mason while (b) { 1926f624d976SQu Wenruo int dec = 0; 1927f624d976SQu Wenruo 19285f39d397SChris Mason level = btrfs_header_level(b); 192965b51a00SChris Mason 193002217ed2SChris Mason if (cow) { 19319ea2c7c9SNikolay Borisov bool last_level = (level == (BTRFS_MAX_LEVEL - 1)); 19329ea2c7c9SNikolay Borisov 1933c8c42864SChris Mason /* 1934c8c42864SChris Mason * if we don't really need to cow this block 1935c8c42864SChris Mason * then we don't want to set the path blocking, 1936c8c42864SChris Mason * so we test it here 1937c8c42864SChris Mason */ 19385963ffcaSJosef Bacik if (!should_cow_block(trans, root, b)) 193965b51a00SChris Mason goto cow_done; 19405d4f98a2SYan Zheng 1941bd681513SChris Mason /* 1942bd681513SChris Mason * must have write locks on this node and the 1943bd681513SChris Mason * parent 1944bd681513SChris Mason */ 19455124e00eSJosef Bacik if (level > write_lock_level || 19465124e00eSJosef Bacik (level + 1 > write_lock_level && 19475124e00eSJosef Bacik level + 1 < BTRFS_MAX_LEVEL && 19485124e00eSJosef Bacik p->nodes[level + 1])) { 1949bd681513SChris Mason write_lock_level = level + 1; 1950bd681513SChris Mason btrfs_release_path(p); 1951bd681513SChris Mason goto again; 1952bd681513SChris Mason } 1953bd681513SChris Mason 19549ea2c7c9SNikolay Borisov if (last_level) 19559ea2c7c9SNikolay Borisov err = btrfs_cow_block(trans, root, b, NULL, 0, 19569631e4ccSJosef Bacik &b, 19579631e4ccSJosef Bacik BTRFS_NESTING_COW); 19589ea2c7c9SNikolay Borisov else 195933c66f43SYan Zheng err = btrfs_cow_block(trans, root, b, 1960e20d96d6SChris Mason p->nodes[level + 1], 19619631e4ccSJosef Bacik p->slots[level + 1], &b, 19629631e4ccSJosef Bacik BTRFS_NESTING_COW); 196333c66f43SYan Zheng if (err) { 196433c66f43SYan Zheng ret = err; 196565b51a00SChris Mason goto done; 196654aa1f4dSChris Mason } 196702217ed2SChris Mason } 196865b51a00SChris Mason cow_done: 1969eb60ceacSChris Mason p->nodes[level] = b; 1970b4ce94deSChris Mason 1971b4ce94deSChris Mason /* 1972b4ce94deSChris Mason * we have a lock on b and as long as we aren't changing 1973b4ce94deSChris Mason * the tree, there is no way to for the items in b to change. 1974b4ce94deSChris Mason * It is safe to drop the lock on our parent before we 1975b4ce94deSChris Mason * go through the expensive btree search on b. 1976b4ce94deSChris Mason * 1977eb653de1SFilipe David Borba Manana * If we're inserting or deleting (ins_len != 0), then we might 1978eb653de1SFilipe David Borba Manana * be changing slot zero, which may require changing the parent. 1979eb653de1SFilipe David Borba Manana * So, we can't drop the lock until after we know which slot 1980eb653de1SFilipe David Borba Manana * we're operating on. 1981b4ce94deSChris Mason */ 1982eb653de1SFilipe David Borba Manana if (!ins_len && !p->keep_locks) { 1983eb653de1SFilipe David Borba Manana int u = level + 1; 1984eb653de1SFilipe David Borba Manana 1985eb653de1SFilipe David Borba Manana if (u < BTRFS_MAX_LEVEL && p->locks[u]) { 1986eb653de1SFilipe David Borba Manana btrfs_tree_unlock_rw(p->nodes[u], p->locks[u]); 1987eb653de1SFilipe David Borba Manana p->locks[u] = 0; 1988eb653de1SFilipe David Borba Manana } 1989eb653de1SFilipe David Borba Manana } 1990b4ce94deSChris Mason 1991e2e58d0fSFilipe Manana if (level == 0) { 1992109324cfSFilipe Manana if (ins_len > 0) 1993e5e1c174SFilipe Manana ASSERT(write_lock_level >= 1); 1994bd681513SChris Mason 1995109324cfSFilipe Manana ret = search_leaf(trans, root, key, p, ins_len, prev_cmp); 1996459931ecSChris Mason if (!p->search_for_split) 1997f7c79f30SChris Mason unlock_up(p, level, lowest_unlock, 19984b6f8e96SLiu Bo min_write_lock_level, NULL); 199965b51a00SChris Mason goto done; 200065b51a00SChris Mason } 2001e2e58d0fSFilipe Manana 2002e2e58d0fSFilipe Manana ret = search_for_key_slot(b, 0, key, prev_cmp, &slot); 2003e2e58d0fSFilipe Manana if (ret < 0) 2004e2e58d0fSFilipe Manana goto done; 2005e2e58d0fSFilipe Manana prev_cmp = ret; 2006e2e58d0fSFilipe Manana 2007f624d976SQu Wenruo if (ret && slot > 0) { 2008f624d976SQu Wenruo dec = 1; 2009f624d976SQu Wenruo slot--; 2010f624d976SQu Wenruo } 2011f624d976SQu Wenruo p->slots[level] = slot; 2012f624d976SQu Wenruo err = setup_nodes_for_search(trans, root, p, b, level, ins_len, 2013f624d976SQu Wenruo &write_lock_level); 2014f624d976SQu Wenruo if (err == -EAGAIN) 2015f624d976SQu Wenruo goto again; 2016f624d976SQu Wenruo if (err) { 2017f624d976SQu Wenruo ret = err; 2018f624d976SQu Wenruo goto done; 2019f624d976SQu Wenruo } 2020f624d976SQu Wenruo b = p->nodes[level]; 2021f624d976SQu Wenruo slot = p->slots[level]; 2022f624d976SQu Wenruo 2023f624d976SQu Wenruo /* 2024f624d976SQu Wenruo * Slot 0 is special, if we change the key we have to update 2025f624d976SQu Wenruo * the parent pointer which means we must have a write lock on 2026f624d976SQu Wenruo * the parent 2027f624d976SQu Wenruo */ 2028f624d976SQu Wenruo if (slot == 0 && ins_len && write_lock_level < level + 1) { 2029f624d976SQu Wenruo write_lock_level = level + 1; 2030f624d976SQu Wenruo btrfs_release_path(p); 2031f624d976SQu Wenruo goto again; 2032f624d976SQu Wenruo } 2033f624d976SQu Wenruo 2034f624d976SQu Wenruo unlock_up(p, level, lowest_unlock, min_write_lock_level, 2035f624d976SQu Wenruo &write_lock_level); 2036f624d976SQu Wenruo 2037f624d976SQu Wenruo if (level == lowest_level) { 2038f624d976SQu Wenruo if (dec) 2039f624d976SQu Wenruo p->slots[level]++; 2040f624d976SQu Wenruo goto done; 2041f624d976SQu Wenruo } 2042f624d976SQu Wenruo 2043f624d976SQu Wenruo err = read_block_for_search(root, p, &b, level, slot, key); 2044f624d976SQu Wenruo if (err == -EAGAIN) 2045f624d976SQu Wenruo goto again; 2046f624d976SQu Wenruo if (err) { 2047f624d976SQu Wenruo ret = err; 2048f624d976SQu Wenruo goto done; 2049f624d976SQu Wenruo } 2050f624d976SQu Wenruo 2051f624d976SQu Wenruo if (!p->skip_locking) { 2052f624d976SQu Wenruo level = btrfs_header_level(b); 2053f624d976SQu Wenruo if (level <= write_lock_level) { 2054f624d976SQu Wenruo btrfs_tree_lock(b); 2055f624d976SQu Wenruo p->locks[level] = BTRFS_WRITE_LOCK; 2056f624d976SQu Wenruo } else { 2057fe596ca3SJosef Bacik btrfs_tree_read_lock(b); 2058f624d976SQu Wenruo p->locks[level] = BTRFS_READ_LOCK; 2059f624d976SQu Wenruo } 2060f624d976SQu Wenruo p->nodes[level] = b; 2061f624d976SQu Wenruo } 206265b51a00SChris Mason } 206365b51a00SChris Mason ret = 1; 206465b51a00SChris Mason done: 20655f5bc6b1SFilipe Manana if (ret < 0 && !p->skip_release_on_error) 2066b3b4aa74SDavid Sterba btrfs_release_path(p); 2067d96b3424SFilipe Manana 2068d96b3424SFilipe Manana if (p->need_commit_sem) { 2069d96b3424SFilipe Manana int ret2; 2070d96b3424SFilipe Manana 2071d96b3424SFilipe Manana ret2 = finish_need_commit_sem_search(p); 2072d96b3424SFilipe Manana up_read(&fs_info->commit_root_sem); 2073d96b3424SFilipe Manana if (ret2) 2074d96b3424SFilipe Manana ret = ret2; 2075d96b3424SFilipe Manana } 2076d96b3424SFilipe Manana 2077be0e5c09SChris Mason return ret; 2078be0e5c09SChris Mason } 2079f75e2b79SJosef Bacik ALLOW_ERROR_INJECTION(btrfs_search_slot, ERRNO); 2080be0e5c09SChris Mason 208174123bd7SChris Mason /* 20825d9e75c4SJan Schmidt * Like btrfs_search_slot, this looks for a key in the given tree. It uses the 20835d9e75c4SJan Schmidt * current state of the tree together with the operations recorded in the tree 20845d9e75c4SJan Schmidt * modification log to search for the key in a previous version of this tree, as 20855d9e75c4SJan Schmidt * denoted by the time_seq parameter. 20865d9e75c4SJan Schmidt * 20875d9e75c4SJan Schmidt * Naturally, there is no support for insert, delete or cow operations. 20885d9e75c4SJan Schmidt * 20895d9e75c4SJan Schmidt * The resulting path and return value will be set up as if we called 20905d9e75c4SJan Schmidt * btrfs_search_slot at that point in time with ins_len and cow both set to 0. 20915d9e75c4SJan Schmidt */ 2092310712b2SOmar Sandoval int btrfs_search_old_slot(struct btrfs_root *root, const struct btrfs_key *key, 20935d9e75c4SJan Schmidt struct btrfs_path *p, u64 time_seq) 20945d9e75c4SJan Schmidt { 20950b246afaSJeff Mahoney struct btrfs_fs_info *fs_info = root->fs_info; 20965d9e75c4SJan Schmidt struct extent_buffer *b; 20975d9e75c4SJan Schmidt int slot; 20985d9e75c4SJan Schmidt int ret; 20995d9e75c4SJan Schmidt int err; 21005d9e75c4SJan Schmidt int level; 21015d9e75c4SJan Schmidt int lowest_unlock = 1; 21025d9e75c4SJan Schmidt u8 lowest_level = 0; 21035d9e75c4SJan Schmidt 21045d9e75c4SJan Schmidt lowest_level = p->lowest_level; 21055d9e75c4SJan Schmidt WARN_ON(p->nodes[0] != NULL); 21065d9e75c4SJan Schmidt 21075d9e75c4SJan Schmidt if (p->search_commit_root) { 21085d9e75c4SJan Schmidt BUG_ON(time_seq); 21095d9e75c4SJan Schmidt return btrfs_search_slot(NULL, root, key, p, 0, 0); 21105d9e75c4SJan Schmidt } 21115d9e75c4SJan Schmidt 21125d9e75c4SJan Schmidt again: 2113f3a84ccdSFilipe Manana b = btrfs_get_old_root(root, time_seq); 2114315bed43SNikolay Borisov if (!b) { 2115315bed43SNikolay Borisov ret = -EIO; 2116315bed43SNikolay Borisov goto done; 2117315bed43SNikolay Borisov } 21185d9e75c4SJan Schmidt level = btrfs_header_level(b); 21195d9e75c4SJan Schmidt p->locks[level] = BTRFS_READ_LOCK; 21205d9e75c4SJan Schmidt 21215d9e75c4SJan Schmidt while (b) { 2122abe9339dSQu Wenruo int dec = 0; 2123abe9339dSQu Wenruo 21245d9e75c4SJan Schmidt level = btrfs_header_level(b); 21255d9e75c4SJan Schmidt p->nodes[level] = b; 21265d9e75c4SJan Schmidt 21275d9e75c4SJan Schmidt /* 21285d9e75c4SJan Schmidt * we have a lock on b and as long as we aren't changing 21295d9e75c4SJan Schmidt * the tree, there is no way to for the items in b to change. 21305d9e75c4SJan Schmidt * It is safe to drop the lock on our parent before we 21315d9e75c4SJan Schmidt * go through the expensive btree search on b. 21325d9e75c4SJan Schmidt */ 21335d9e75c4SJan Schmidt btrfs_unlock_up_safe(p, level + 1); 21345d9e75c4SJan Schmidt 2135995e9a16SNikolay Borisov ret = btrfs_bin_search(b, key, &slot); 2136cbca7d59SFilipe Manana if (ret < 0) 2137cbca7d59SFilipe Manana goto done; 21385d9e75c4SJan Schmidt 2139abe9339dSQu Wenruo if (level == 0) { 2140abe9339dSQu Wenruo p->slots[level] = slot; 2141abe9339dSQu Wenruo unlock_up(p, level, lowest_unlock, 0, NULL); 2142abe9339dSQu Wenruo goto done; 2143abe9339dSQu Wenruo } 2144abe9339dSQu Wenruo 21455d9e75c4SJan Schmidt if (ret && slot > 0) { 21465d9e75c4SJan Schmidt dec = 1; 2147abe9339dSQu Wenruo slot--; 21485d9e75c4SJan Schmidt } 21495d9e75c4SJan Schmidt p->slots[level] = slot; 21505d9e75c4SJan Schmidt unlock_up(p, level, lowest_unlock, 0, NULL); 21515d9e75c4SJan Schmidt 21525d9e75c4SJan Schmidt if (level == lowest_level) { 21535d9e75c4SJan Schmidt if (dec) 21545d9e75c4SJan Schmidt p->slots[level]++; 21555d9e75c4SJan Schmidt goto done; 21565d9e75c4SJan Schmidt } 21575d9e75c4SJan Schmidt 2158abe9339dSQu Wenruo err = read_block_for_search(root, p, &b, level, slot, key); 21595d9e75c4SJan Schmidt if (err == -EAGAIN) 21605d9e75c4SJan Schmidt goto again; 21615d9e75c4SJan Schmidt if (err) { 21625d9e75c4SJan Schmidt ret = err; 21635d9e75c4SJan Schmidt goto done; 21645d9e75c4SJan Schmidt } 21655d9e75c4SJan Schmidt 21665d9e75c4SJan Schmidt level = btrfs_header_level(b); 21675d9e75c4SJan Schmidt btrfs_tree_read_lock(b); 2168f3a84ccdSFilipe Manana b = btrfs_tree_mod_log_rewind(fs_info, p, b, time_seq); 2169db7f3436SJosef Bacik if (!b) { 2170db7f3436SJosef Bacik ret = -ENOMEM; 2171db7f3436SJosef Bacik goto done; 2172db7f3436SJosef Bacik } 21735d9e75c4SJan Schmidt p->locks[level] = BTRFS_READ_LOCK; 21745d9e75c4SJan Schmidt p->nodes[level] = b; 21755d9e75c4SJan Schmidt } 21765d9e75c4SJan Schmidt ret = 1; 21775d9e75c4SJan Schmidt done: 21785d9e75c4SJan Schmidt if (ret < 0) 21795d9e75c4SJan Schmidt btrfs_release_path(p); 21805d9e75c4SJan Schmidt 21815d9e75c4SJan Schmidt return ret; 21825d9e75c4SJan Schmidt } 21835d9e75c4SJan Schmidt 21845d9e75c4SJan Schmidt /* 21852f38b3e1SArne Jansen * helper to use instead of search slot if no exact match is needed but 21862f38b3e1SArne Jansen * instead the next or previous item should be returned. 21872f38b3e1SArne Jansen * When find_higher is true, the next higher item is returned, the next lower 21882f38b3e1SArne Jansen * otherwise. 21892f38b3e1SArne Jansen * When return_any and find_higher are both true, and no higher item is found, 21902f38b3e1SArne Jansen * return the next lower instead. 21912f38b3e1SArne Jansen * When return_any is true and find_higher is false, and no lower item is found, 21922f38b3e1SArne Jansen * return the next higher instead. 21932f38b3e1SArne Jansen * It returns 0 if any item is found, 1 if none is found (tree empty), and 21942f38b3e1SArne Jansen * < 0 on error 21952f38b3e1SArne Jansen */ 21962f38b3e1SArne Jansen int btrfs_search_slot_for_read(struct btrfs_root *root, 2197310712b2SOmar Sandoval const struct btrfs_key *key, 2198310712b2SOmar Sandoval struct btrfs_path *p, int find_higher, 2199310712b2SOmar Sandoval int return_any) 22002f38b3e1SArne Jansen { 22012f38b3e1SArne Jansen int ret; 22022f38b3e1SArne Jansen struct extent_buffer *leaf; 22032f38b3e1SArne Jansen 22042f38b3e1SArne Jansen again: 22052f38b3e1SArne Jansen ret = btrfs_search_slot(NULL, root, key, p, 0, 0); 22062f38b3e1SArne Jansen if (ret <= 0) 22072f38b3e1SArne Jansen return ret; 22082f38b3e1SArne Jansen /* 22092f38b3e1SArne Jansen * a return value of 1 means the path is at the position where the 22102f38b3e1SArne Jansen * item should be inserted. Normally this is the next bigger item, 22112f38b3e1SArne Jansen * but in case the previous item is the last in a leaf, path points 22122f38b3e1SArne Jansen * to the first free slot in the previous leaf, i.e. at an invalid 22132f38b3e1SArne Jansen * item. 22142f38b3e1SArne Jansen */ 22152f38b3e1SArne Jansen leaf = p->nodes[0]; 22162f38b3e1SArne Jansen 22172f38b3e1SArne Jansen if (find_higher) { 22182f38b3e1SArne Jansen if (p->slots[0] >= btrfs_header_nritems(leaf)) { 22192f38b3e1SArne Jansen ret = btrfs_next_leaf(root, p); 22202f38b3e1SArne Jansen if (ret <= 0) 22212f38b3e1SArne Jansen return ret; 22222f38b3e1SArne Jansen if (!return_any) 22232f38b3e1SArne Jansen return 1; 22242f38b3e1SArne Jansen /* 22252f38b3e1SArne Jansen * no higher item found, return the next 22262f38b3e1SArne Jansen * lower instead 22272f38b3e1SArne Jansen */ 22282f38b3e1SArne Jansen return_any = 0; 22292f38b3e1SArne Jansen find_higher = 0; 22302f38b3e1SArne Jansen btrfs_release_path(p); 22312f38b3e1SArne Jansen goto again; 22322f38b3e1SArne Jansen } 22332f38b3e1SArne Jansen } else { 22342f38b3e1SArne Jansen if (p->slots[0] == 0) { 22352f38b3e1SArne Jansen ret = btrfs_prev_leaf(root, p); 2236e6793769SArne Jansen if (ret < 0) 22372f38b3e1SArne Jansen return ret; 2238e6793769SArne Jansen if (!ret) { 223923c6bf6aSFilipe David Borba Manana leaf = p->nodes[0]; 224023c6bf6aSFilipe David Borba Manana if (p->slots[0] == btrfs_header_nritems(leaf)) 224123c6bf6aSFilipe David Borba Manana p->slots[0]--; 2242e6793769SArne Jansen return 0; 2243e6793769SArne Jansen } 22442f38b3e1SArne Jansen if (!return_any) 22452f38b3e1SArne Jansen return 1; 22462f38b3e1SArne Jansen /* 22472f38b3e1SArne Jansen * no lower item found, return the next 22482f38b3e1SArne Jansen * higher instead 22492f38b3e1SArne Jansen */ 22502f38b3e1SArne Jansen return_any = 0; 22512f38b3e1SArne Jansen find_higher = 1; 22522f38b3e1SArne Jansen btrfs_release_path(p); 22532f38b3e1SArne Jansen goto again; 2254e6793769SArne Jansen } else { 22552f38b3e1SArne Jansen --p->slots[0]; 22562f38b3e1SArne Jansen } 22572f38b3e1SArne Jansen } 22582f38b3e1SArne Jansen return 0; 22592f38b3e1SArne Jansen } 22602f38b3e1SArne Jansen 22612f38b3e1SArne Jansen /* 22620ff40a91SMarcos Paulo de Souza * Execute search and call btrfs_previous_item to traverse backwards if the item 22630ff40a91SMarcos Paulo de Souza * was not found. 22640ff40a91SMarcos Paulo de Souza * 22650ff40a91SMarcos Paulo de Souza * Return 0 if found, 1 if not found and < 0 if error. 22660ff40a91SMarcos Paulo de Souza */ 22670ff40a91SMarcos Paulo de Souza int btrfs_search_backwards(struct btrfs_root *root, struct btrfs_key *key, 22680ff40a91SMarcos Paulo de Souza struct btrfs_path *path) 22690ff40a91SMarcos Paulo de Souza { 22700ff40a91SMarcos Paulo de Souza int ret; 22710ff40a91SMarcos Paulo de Souza 22720ff40a91SMarcos Paulo de Souza ret = btrfs_search_slot(NULL, root, key, path, 0, 0); 22730ff40a91SMarcos Paulo de Souza if (ret > 0) 22740ff40a91SMarcos Paulo de Souza ret = btrfs_previous_item(root, path, key->objectid, key->type); 22750ff40a91SMarcos Paulo de Souza 22760ff40a91SMarcos Paulo de Souza if (ret == 0) 22770ff40a91SMarcos Paulo de Souza btrfs_item_key_to_cpu(path->nodes[0], key, path->slots[0]); 22780ff40a91SMarcos Paulo de Souza 22790ff40a91SMarcos Paulo de Souza return ret; 22800ff40a91SMarcos Paulo de Souza } 22810ff40a91SMarcos Paulo de Souza 22820ff40a91SMarcos Paulo de Souza /* 228374123bd7SChris Mason * adjust the pointers going up the tree, starting at level 228474123bd7SChris Mason * making sure the right key of each node is points to 'key'. 228574123bd7SChris Mason * This is used after shifting pointers to the left, so it stops 228674123bd7SChris Mason * fixing up pointers when a given leaf/node is not in slot 0 of the 228774123bd7SChris Mason * higher levels 2288aa5d6bedSChris Mason * 228974123bd7SChris Mason */ 2290b167fa91SNikolay Borisov static void fixup_low_keys(struct btrfs_path *path, 22915f39d397SChris Mason struct btrfs_disk_key *key, int level) 2292be0e5c09SChris Mason { 2293be0e5c09SChris Mason int i; 22945f39d397SChris Mason struct extent_buffer *t; 22950e82bcfeSDavid Sterba int ret; 22965f39d397SChris Mason 2297234b63a0SChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 2298be0e5c09SChris Mason int tslot = path->slots[i]; 22990e82bcfeSDavid Sterba 2300eb60ceacSChris Mason if (!path->nodes[i]) 2301be0e5c09SChris Mason break; 23025f39d397SChris Mason t = path->nodes[i]; 2303f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_insert_key(t, tslot, 2304f3a84ccdSFilipe Manana BTRFS_MOD_LOG_KEY_REPLACE, GFP_ATOMIC); 23050e82bcfeSDavid Sterba BUG_ON(ret < 0); 23065f39d397SChris Mason btrfs_set_node_key(t, key, tslot); 2307d6025579SChris Mason btrfs_mark_buffer_dirty(path->nodes[i]); 2308be0e5c09SChris Mason if (tslot != 0) 2309be0e5c09SChris Mason break; 2310be0e5c09SChris Mason } 2311be0e5c09SChris Mason } 2312be0e5c09SChris Mason 231374123bd7SChris Mason /* 231431840ae1SZheng Yan * update item key. 231531840ae1SZheng Yan * 231631840ae1SZheng Yan * This function isn't completely safe. It's the caller's responsibility 231731840ae1SZheng Yan * that the new key won't break the order 231831840ae1SZheng Yan */ 2319b7a0365eSDaniel Dressler void btrfs_set_item_key_safe(struct btrfs_fs_info *fs_info, 2320b7a0365eSDaniel Dressler struct btrfs_path *path, 2321310712b2SOmar Sandoval const struct btrfs_key *new_key) 232231840ae1SZheng Yan { 232331840ae1SZheng Yan struct btrfs_disk_key disk_key; 232431840ae1SZheng Yan struct extent_buffer *eb; 232531840ae1SZheng Yan int slot; 232631840ae1SZheng Yan 232731840ae1SZheng Yan eb = path->nodes[0]; 232831840ae1SZheng Yan slot = path->slots[0]; 232931840ae1SZheng Yan if (slot > 0) { 233031840ae1SZheng Yan btrfs_item_key(eb, &disk_key, slot - 1); 23317c15d410SQu Wenruo if (unlikely(comp_keys(&disk_key, new_key) >= 0)) { 23327c15d410SQu Wenruo btrfs_crit(fs_info, 23337c15d410SQu Wenruo "slot %u key (%llu %u %llu) new key (%llu %u %llu)", 23347c15d410SQu Wenruo slot, btrfs_disk_key_objectid(&disk_key), 23357c15d410SQu Wenruo btrfs_disk_key_type(&disk_key), 23367c15d410SQu Wenruo btrfs_disk_key_offset(&disk_key), 23377c15d410SQu Wenruo new_key->objectid, new_key->type, 23387c15d410SQu Wenruo new_key->offset); 23397c15d410SQu Wenruo btrfs_print_leaf(eb); 23407c15d410SQu Wenruo BUG(); 23417c15d410SQu Wenruo } 234231840ae1SZheng Yan } 234331840ae1SZheng Yan if (slot < btrfs_header_nritems(eb) - 1) { 234431840ae1SZheng Yan btrfs_item_key(eb, &disk_key, slot + 1); 23457c15d410SQu Wenruo if (unlikely(comp_keys(&disk_key, new_key) <= 0)) { 23467c15d410SQu Wenruo btrfs_crit(fs_info, 23477c15d410SQu Wenruo "slot %u key (%llu %u %llu) new key (%llu %u %llu)", 23487c15d410SQu Wenruo slot, btrfs_disk_key_objectid(&disk_key), 23497c15d410SQu Wenruo btrfs_disk_key_type(&disk_key), 23507c15d410SQu Wenruo btrfs_disk_key_offset(&disk_key), 23517c15d410SQu Wenruo new_key->objectid, new_key->type, 23527c15d410SQu Wenruo new_key->offset); 23537c15d410SQu Wenruo btrfs_print_leaf(eb); 23547c15d410SQu Wenruo BUG(); 23557c15d410SQu Wenruo } 235631840ae1SZheng Yan } 235731840ae1SZheng Yan 235831840ae1SZheng Yan btrfs_cpu_key_to_disk(&disk_key, new_key); 235931840ae1SZheng Yan btrfs_set_item_key(eb, &disk_key, slot); 236031840ae1SZheng Yan btrfs_mark_buffer_dirty(eb); 236131840ae1SZheng Yan if (slot == 0) 2362b167fa91SNikolay Borisov fixup_low_keys(path, &disk_key, 1); 236331840ae1SZheng Yan } 236431840ae1SZheng Yan 236531840ae1SZheng Yan /* 2366d16c702fSQu Wenruo * Check key order of two sibling extent buffers. 2367d16c702fSQu Wenruo * 2368d16c702fSQu Wenruo * Return true if something is wrong. 2369d16c702fSQu Wenruo * Return false if everything is fine. 2370d16c702fSQu Wenruo * 2371d16c702fSQu Wenruo * Tree-checker only works inside one tree block, thus the following 2372d16c702fSQu Wenruo * corruption can not be detected by tree-checker: 2373d16c702fSQu Wenruo * 2374d16c702fSQu Wenruo * Leaf @left | Leaf @right 2375d16c702fSQu Wenruo * -------------------------------------------------------------- 2376d16c702fSQu Wenruo * | 1 | 2 | 3 | 4 | 5 | f6 | | 7 | 8 | 2377d16c702fSQu Wenruo * 2378d16c702fSQu Wenruo * Key f6 in leaf @left itself is valid, but not valid when the next 2379d16c702fSQu Wenruo * key in leaf @right is 7. 2380d16c702fSQu Wenruo * This can only be checked at tree block merge time. 2381d16c702fSQu Wenruo * And since tree checker has ensured all key order in each tree block 2382d16c702fSQu Wenruo * is correct, we only need to bother the last key of @left and the first 2383d16c702fSQu Wenruo * key of @right. 2384d16c702fSQu Wenruo */ 2385d16c702fSQu Wenruo static bool check_sibling_keys(struct extent_buffer *left, 2386d16c702fSQu Wenruo struct extent_buffer *right) 2387d16c702fSQu Wenruo { 2388d16c702fSQu Wenruo struct btrfs_key left_last; 2389d16c702fSQu Wenruo struct btrfs_key right_first; 2390d16c702fSQu Wenruo int level = btrfs_header_level(left); 2391d16c702fSQu Wenruo int nr_left = btrfs_header_nritems(left); 2392d16c702fSQu Wenruo int nr_right = btrfs_header_nritems(right); 2393d16c702fSQu Wenruo 2394d16c702fSQu Wenruo /* No key to check in one of the tree blocks */ 2395d16c702fSQu Wenruo if (!nr_left || !nr_right) 2396d16c702fSQu Wenruo return false; 2397d16c702fSQu Wenruo 2398d16c702fSQu Wenruo if (level) { 2399d16c702fSQu Wenruo btrfs_node_key_to_cpu(left, &left_last, nr_left - 1); 2400d16c702fSQu Wenruo btrfs_node_key_to_cpu(right, &right_first, 0); 2401d16c702fSQu Wenruo } else { 2402d16c702fSQu Wenruo btrfs_item_key_to_cpu(left, &left_last, nr_left - 1); 2403d16c702fSQu Wenruo btrfs_item_key_to_cpu(right, &right_first, 0); 2404d16c702fSQu Wenruo } 2405d16c702fSQu Wenruo 2406d16c702fSQu Wenruo if (btrfs_comp_cpu_keys(&left_last, &right_first) >= 0) { 2407d16c702fSQu Wenruo btrfs_crit(left->fs_info, 2408d16c702fSQu Wenruo "bad key order, sibling blocks, left last (%llu %u %llu) right first (%llu %u %llu)", 2409d16c702fSQu Wenruo left_last.objectid, left_last.type, 2410d16c702fSQu Wenruo left_last.offset, right_first.objectid, 2411d16c702fSQu Wenruo right_first.type, right_first.offset); 2412d16c702fSQu Wenruo return true; 2413d16c702fSQu Wenruo } 2414d16c702fSQu Wenruo return false; 2415d16c702fSQu Wenruo } 2416d16c702fSQu Wenruo 2417d16c702fSQu Wenruo /* 241874123bd7SChris Mason * try to push data from one node into the next node left in the 241979f95c82SChris Mason * tree. 2420aa5d6bedSChris Mason * 2421aa5d6bedSChris Mason * returns 0 if some ptrs were pushed left, < 0 if there was some horrible 2422aa5d6bedSChris Mason * error, and > 0 if there was no room in the left hand block. 242374123bd7SChris Mason */ 242498ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans, 24252ff7e61eSJeff Mahoney struct extent_buffer *dst, 2426971a1f66SChris Mason struct extent_buffer *src, int empty) 2427be0e5c09SChris Mason { 2428d30a668fSDavid Sterba struct btrfs_fs_info *fs_info = trans->fs_info; 2429be0e5c09SChris Mason int push_items = 0; 2430bb803951SChris Mason int src_nritems; 2431bb803951SChris Mason int dst_nritems; 2432aa5d6bedSChris Mason int ret = 0; 2433be0e5c09SChris Mason 24345f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 24355f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 24360b246afaSJeff Mahoney push_items = BTRFS_NODEPTRS_PER_BLOCK(fs_info) - dst_nritems; 24377bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 24387bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 243954aa1f4dSChris Mason 2440bce4eae9SChris Mason if (!empty && src_nritems <= 8) 2441971a1f66SChris Mason return 1; 2442971a1f66SChris Mason 2443d397712bSChris Mason if (push_items <= 0) 2444be0e5c09SChris Mason return 1; 2445be0e5c09SChris Mason 2446bce4eae9SChris Mason if (empty) { 2447971a1f66SChris Mason push_items = min(src_nritems, push_items); 2448bce4eae9SChris Mason if (push_items < src_nritems) { 2449bce4eae9SChris Mason /* leave at least 8 pointers in the node if 2450bce4eae9SChris Mason * we aren't going to empty it 2451bce4eae9SChris Mason */ 2452bce4eae9SChris Mason if (src_nritems - push_items < 8) { 2453bce4eae9SChris Mason if (push_items <= 8) 2454bce4eae9SChris Mason return 1; 2455bce4eae9SChris Mason push_items -= 8; 2456bce4eae9SChris Mason } 2457bce4eae9SChris Mason } 2458bce4eae9SChris Mason } else 2459bce4eae9SChris Mason push_items = min(src_nritems - 8, push_items); 246079f95c82SChris Mason 2461d16c702fSQu Wenruo /* dst is the left eb, src is the middle eb */ 2462d16c702fSQu Wenruo if (check_sibling_keys(dst, src)) { 2463d16c702fSQu Wenruo ret = -EUCLEAN; 2464d16c702fSQu Wenruo btrfs_abort_transaction(trans, ret); 2465d16c702fSQu Wenruo return ret; 2466d16c702fSQu Wenruo } 2467f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_eb_copy(dst, src, dst_nritems, 0, push_items); 24685de865eeSFilipe David Borba Manana if (ret) { 246966642832SJeff Mahoney btrfs_abort_transaction(trans, ret); 24705de865eeSFilipe David Borba Manana return ret; 24715de865eeSFilipe David Borba Manana } 24725f39d397SChris Mason copy_extent_buffer(dst, src, 24735f39d397SChris Mason btrfs_node_key_ptr_offset(dst_nritems), 24745f39d397SChris Mason btrfs_node_key_ptr_offset(0), 2475123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 24765f39d397SChris Mason 2477bb803951SChris Mason if (push_items < src_nritems) { 247857911b8bSJan Schmidt /* 2479f3a84ccdSFilipe Manana * Don't call btrfs_tree_mod_log_insert_move() here, key removal 2480f3a84ccdSFilipe Manana * was already fully logged by btrfs_tree_mod_log_eb_copy() above. 248157911b8bSJan Schmidt */ 24825f39d397SChris Mason memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0), 24835f39d397SChris Mason btrfs_node_key_ptr_offset(push_items), 2484e2fa7227SChris Mason (src_nritems - push_items) * 2485123abc88SChris Mason sizeof(struct btrfs_key_ptr)); 2486bb803951SChris Mason } 24875f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 24885f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 24895f39d397SChris Mason btrfs_mark_buffer_dirty(src); 24905f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 249131840ae1SZheng Yan 2492bb803951SChris Mason return ret; 2493be0e5c09SChris Mason } 2494be0e5c09SChris Mason 249597571fd0SChris Mason /* 249679f95c82SChris Mason * try to push data from one node into the next node right in the 249779f95c82SChris Mason * tree. 249879f95c82SChris Mason * 249979f95c82SChris Mason * returns 0 if some ptrs were pushed, < 0 if there was some horrible 250079f95c82SChris Mason * error, and > 0 if there was no room in the right hand block. 250179f95c82SChris Mason * 250279f95c82SChris Mason * this will only push up to 1/2 the contents of the left node over 250379f95c82SChris Mason */ 25045f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 25055f39d397SChris Mason struct extent_buffer *dst, 25065f39d397SChris Mason struct extent_buffer *src) 250779f95c82SChris Mason { 250855d32ed8SDavid Sterba struct btrfs_fs_info *fs_info = trans->fs_info; 250979f95c82SChris Mason int push_items = 0; 251079f95c82SChris Mason int max_push; 251179f95c82SChris Mason int src_nritems; 251279f95c82SChris Mason int dst_nritems; 251379f95c82SChris Mason int ret = 0; 251479f95c82SChris Mason 25157bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 25167bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 25177bb86316SChris Mason 25185f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 25195f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 25200b246afaSJeff Mahoney push_items = BTRFS_NODEPTRS_PER_BLOCK(fs_info) - dst_nritems; 2521d397712bSChris Mason if (push_items <= 0) 252279f95c82SChris Mason return 1; 2523bce4eae9SChris Mason 2524d397712bSChris Mason if (src_nritems < 4) 2525bce4eae9SChris Mason return 1; 252679f95c82SChris Mason 252779f95c82SChris Mason max_push = src_nritems / 2 + 1; 252879f95c82SChris Mason /* don't try to empty the node */ 2529d397712bSChris Mason if (max_push >= src_nritems) 253079f95c82SChris Mason return 1; 2531252c38f0SYan 253279f95c82SChris Mason if (max_push < push_items) 253379f95c82SChris Mason push_items = max_push; 253479f95c82SChris Mason 2535d16c702fSQu Wenruo /* dst is the right eb, src is the middle eb */ 2536d16c702fSQu Wenruo if (check_sibling_keys(src, dst)) { 2537d16c702fSQu Wenruo ret = -EUCLEAN; 2538d16c702fSQu Wenruo btrfs_abort_transaction(trans, ret); 2539d16c702fSQu Wenruo return ret; 2540d16c702fSQu Wenruo } 2541f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_insert_move(dst, push_items, 0, dst_nritems); 2542bf1d3425SDavid Sterba BUG_ON(ret < 0); 25435f39d397SChris Mason memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items), 25445f39d397SChris Mason btrfs_node_key_ptr_offset(0), 25455f39d397SChris Mason (dst_nritems) * 25465f39d397SChris Mason sizeof(struct btrfs_key_ptr)); 2547d6025579SChris Mason 2548f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_eb_copy(dst, src, 0, src_nritems - push_items, 2549ed874f0dSDavid Sterba push_items); 25505de865eeSFilipe David Borba Manana if (ret) { 255166642832SJeff Mahoney btrfs_abort_transaction(trans, ret); 25525de865eeSFilipe David Borba Manana return ret; 25535de865eeSFilipe David Borba Manana } 25545f39d397SChris Mason copy_extent_buffer(dst, src, 25555f39d397SChris Mason btrfs_node_key_ptr_offset(0), 25565f39d397SChris Mason btrfs_node_key_ptr_offset(src_nritems - push_items), 2557123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 255879f95c82SChris Mason 25595f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 25605f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 256179f95c82SChris Mason 25625f39d397SChris Mason btrfs_mark_buffer_dirty(src); 25635f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 256431840ae1SZheng Yan 256579f95c82SChris Mason return ret; 256679f95c82SChris Mason } 256779f95c82SChris Mason 256879f95c82SChris Mason /* 256997571fd0SChris Mason * helper function to insert a new root level in the tree. 257097571fd0SChris Mason * A new node is allocated, and a single item is inserted to 257197571fd0SChris Mason * point to the existing root 2572aa5d6bedSChris Mason * 2573aa5d6bedSChris Mason * returns zero on success or < 0 on failure. 257497571fd0SChris Mason */ 2575d397712bSChris Mason static noinline int insert_new_root(struct btrfs_trans_handle *trans, 25765f39d397SChris Mason struct btrfs_root *root, 2577fdd99c72SLiu Bo struct btrfs_path *path, int level) 257874123bd7SChris Mason { 25790b246afaSJeff Mahoney struct btrfs_fs_info *fs_info = root->fs_info; 25807bb86316SChris Mason u64 lower_gen; 25815f39d397SChris Mason struct extent_buffer *lower; 25825f39d397SChris Mason struct extent_buffer *c; 2583925baeddSChris Mason struct extent_buffer *old; 25845f39d397SChris Mason struct btrfs_disk_key lower_key; 2585d9d19a01SDavid Sterba int ret; 25865c680ed6SChris Mason 25875c680ed6SChris Mason BUG_ON(path->nodes[level]); 25885c680ed6SChris Mason BUG_ON(path->nodes[level-1] != root->node); 25895c680ed6SChris Mason 25907bb86316SChris Mason lower = path->nodes[level-1]; 25917bb86316SChris Mason if (level == 1) 25927bb86316SChris Mason btrfs_item_key(lower, &lower_key, 0); 25937bb86316SChris Mason else 25947bb86316SChris Mason btrfs_node_key(lower, &lower_key, 0); 25957bb86316SChris Mason 259679bd3712SFilipe Manana c = btrfs_alloc_tree_block(trans, root, 0, root->root_key.objectid, 259779bd3712SFilipe Manana &lower_key, level, root->node->start, 0, 2598cf6f34aaSJosef Bacik BTRFS_NESTING_NEW_ROOT); 25995f39d397SChris Mason if (IS_ERR(c)) 26005f39d397SChris Mason return PTR_ERR(c); 2601925baeddSChris Mason 26020b246afaSJeff Mahoney root_add_used(root, fs_info->nodesize); 2603f0486c68SYan, Zheng 26045f39d397SChris Mason btrfs_set_header_nritems(c, 1); 26055f39d397SChris Mason btrfs_set_node_key(c, &lower_key, 0); 2606db94535dSChris Mason btrfs_set_node_blockptr(c, 0, lower->start); 26077bb86316SChris Mason lower_gen = btrfs_header_generation(lower); 260831840ae1SZheng Yan WARN_ON(lower_gen != trans->transid); 26097bb86316SChris Mason 26107bb86316SChris Mason btrfs_set_node_ptr_generation(c, 0, lower_gen); 26115f39d397SChris Mason 26125f39d397SChris Mason btrfs_mark_buffer_dirty(c); 2613d5719762SChris Mason 2614925baeddSChris Mason old = root->node; 2615406808abSFilipe Manana ret = btrfs_tree_mod_log_insert_root(root->node, c, false); 2616d9d19a01SDavid Sterba BUG_ON(ret < 0); 2617240f62c8SChris Mason rcu_assign_pointer(root->node, c); 2618925baeddSChris Mason 2619925baeddSChris Mason /* the super has an extra ref to root->node */ 2620925baeddSChris Mason free_extent_buffer(old); 2621925baeddSChris Mason 26220b86a832SChris Mason add_root_to_dirty_list(root); 262367439dadSDavid Sterba atomic_inc(&c->refs); 26245f39d397SChris Mason path->nodes[level] = c; 2625ac5887c8SJosef Bacik path->locks[level] = BTRFS_WRITE_LOCK; 262674123bd7SChris Mason path->slots[level] = 0; 262774123bd7SChris Mason return 0; 262874123bd7SChris Mason } 26295c680ed6SChris Mason 26305c680ed6SChris Mason /* 26315c680ed6SChris Mason * worker function to insert a single pointer in a node. 26325c680ed6SChris Mason * the node should have enough room for the pointer already 263397571fd0SChris Mason * 26345c680ed6SChris Mason * slot and level indicate where you want the key to go, and 26355c680ed6SChris Mason * blocknr is the block the key points to. 26365c680ed6SChris Mason */ 2637143bede5SJeff Mahoney static void insert_ptr(struct btrfs_trans_handle *trans, 26386ad3cf6dSDavid Sterba struct btrfs_path *path, 2639143bede5SJeff Mahoney struct btrfs_disk_key *key, u64 bytenr, 2640c3e06965SJan Schmidt int slot, int level) 26415c680ed6SChris Mason { 26425f39d397SChris Mason struct extent_buffer *lower; 26435c680ed6SChris Mason int nritems; 2644f3ea38daSJan Schmidt int ret; 26455c680ed6SChris Mason 26465c680ed6SChris Mason BUG_ON(!path->nodes[level]); 264749d0c642SFilipe Manana btrfs_assert_tree_write_locked(path->nodes[level]); 26485f39d397SChris Mason lower = path->nodes[level]; 26495f39d397SChris Mason nritems = btrfs_header_nritems(lower); 2650c293498bSStoyan Gaydarov BUG_ON(slot > nritems); 26516ad3cf6dSDavid Sterba BUG_ON(nritems == BTRFS_NODEPTRS_PER_BLOCK(trans->fs_info)); 265274123bd7SChris Mason if (slot != nritems) { 2653bf1d3425SDavid Sterba if (level) { 2654f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_insert_move(lower, slot + 1, 2655f3a84ccdSFilipe Manana slot, nritems - slot); 2656bf1d3425SDavid Sterba BUG_ON(ret < 0); 2657bf1d3425SDavid Sterba } 26585f39d397SChris Mason memmove_extent_buffer(lower, 26595f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 26605f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 2661123abc88SChris Mason (nritems - slot) * sizeof(struct btrfs_key_ptr)); 266274123bd7SChris Mason } 2663c3e06965SJan Schmidt if (level) { 2664f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_insert_key(lower, slot, 2665f3a84ccdSFilipe Manana BTRFS_MOD_LOG_KEY_ADD, GFP_NOFS); 2666f3ea38daSJan Schmidt BUG_ON(ret < 0); 2667f3ea38daSJan Schmidt } 26685f39d397SChris Mason btrfs_set_node_key(lower, key, slot); 2669db94535dSChris Mason btrfs_set_node_blockptr(lower, slot, bytenr); 267074493f7aSChris Mason WARN_ON(trans->transid == 0); 267174493f7aSChris Mason btrfs_set_node_ptr_generation(lower, slot, trans->transid); 26725f39d397SChris Mason btrfs_set_header_nritems(lower, nritems + 1); 26735f39d397SChris Mason btrfs_mark_buffer_dirty(lower); 267474123bd7SChris Mason } 267574123bd7SChris Mason 267697571fd0SChris Mason /* 267797571fd0SChris Mason * split the node at the specified level in path in two. 267897571fd0SChris Mason * The path is corrected to point to the appropriate node after the split 267997571fd0SChris Mason * 268097571fd0SChris Mason * Before splitting this tries to make some room in the node by pushing 268197571fd0SChris Mason * left and right, if either one works, it returns right away. 2682aa5d6bedSChris Mason * 2683aa5d6bedSChris Mason * returns 0 on success and < 0 on failure 268497571fd0SChris Mason */ 2685e02119d5SChris Mason static noinline int split_node(struct btrfs_trans_handle *trans, 2686e02119d5SChris Mason struct btrfs_root *root, 2687e02119d5SChris Mason struct btrfs_path *path, int level) 2688be0e5c09SChris Mason { 26890b246afaSJeff Mahoney struct btrfs_fs_info *fs_info = root->fs_info; 26905f39d397SChris Mason struct extent_buffer *c; 26915f39d397SChris Mason struct extent_buffer *split; 26925f39d397SChris Mason struct btrfs_disk_key disk_key; 2693be0e5c09SChris Mason int mid; 26945c680ed6SChris Mason int ret; 26957518a238SChris Mason u32 c_nritems; 2696be0e5c09SChris Mason 26975f39d397SChris Mason c = path->nodes[level]; 26987bb86316SChris Mason WARN_ON(btrfs_header_generation(c) != trans->transid); 26995f39d397SChris Mason if (c == root->node) { 2700d9abbf1cSJan Schmidt /* 270190f8d62eSJan Schmidt * trying to split the root, lets make a new one 270290f8d62eSJan Schmidt * 2703fdd99c72SLiu Bo * tree mod log: We don't log_removal old root in 270490f8d62eSJan Schmidt * insert_new_root, because that root buffer will be kept as a 270590f8d62eSJan Schmidt * normal node. We are going to log removal of half of the 2706f3a84ccdSFilipe Manana * elements below with btrfs_tree_mod_log_eb_copy(). We're 2707f3a84ccdSFilipe Manana * holding a tree lock on the buffer, which is why we cannot 2708f3a84ccdSFilipe Manana * race with other tree_mod_log users. 2709d9abbf1cSJan Schmidt */ 2710fdd99c72SLiu Bo ret = insert_new_root(trans, root, path, level + 1); 27115c680ed6SChris Mason if (ret) 27125c680ed6SChris Mason return ret; 2713b3612421SChris Mason } else { 2714e66f709bSChris Mason ret = push_nodes_for_insert(trans, root, path, level); 27155f39d397SChris Mason c = path->nodes[level]; 27165f39d397SChris Mason if (!ret && btrfs_header_nritems(c) < 27170b246afaSJeff Mahoney BTRFS_NODEPTRS_PER_BLOCK(fs_info) - 3) 2718e66f709bSChris Mason return 0; 271954aa1f4dSChris Mason if (ret < 0) 272054aa1f4dSChris Mason return ret; 27215c680ed6SChris Mason } 2722e66f709bSChris Mason 27235f39d397SChris Mason c_nritems = btrfs_header_nritems(c); 27245d4f98a2SYan Zheng mid = (c_nritems + 1) / 2; 27255d4f98a2SYan Zheng btrfs_node_key(c, &disk_key, mid); 27267bb86316SChris Mason 272779bd3712SFilipe Manana split = btrfs_alloc_tree_block(trans, root, 0, root->root_key.objectid, 272879bd3712SFilipe Manana &disk_key, level, c->start, 0, 272979bd3712SFilipe Manana BTRFS_NESTING_SPLIT); 27305f39d397SChris Mason if (IS_ERR(split)) 27315f39d397SChris Mason return PTR_ERR(split); 273254aa1f4dSChris Mason 27330b246afaSJeff Mahoney root_add_used(root, fs_info->nodesize); 2734bc877d28SNikolay Borisov ASSERT(btrfs_header_level(c) == level); 27355f39d397SChris Mason 2736f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_eb_copy(split, c, 0, mid, c_nritems - mid); 27375de865eeSFilipe David Borba Manana if (ret) { 273866642832SJeff Mahoney btrfs_abort_transaction(trans, ret); 27395de865eeSFilipe David Borba Manana return ret; 27405de865eeSFilipe David Borba Manana } 27415f39d397SChris Mason copy_extent_buffer(split, c, 27425f39d397SChris Mason btrfs_node_key_ptr_offset(0), 27435f39d397SChris Mason btrfs_node_key_ptr_offset(mid), 2744123abc88SChris Mason (c_nritems - mid) * sizeof(struct btrfs_key_ptr)); 27455f39d397SChris Mason btrfs_set_header_nritems(split, c_nritems - mid); 27465f39d397SChris Mason btrfs_set_header_nritems(c, mid); 2747aa5d6bedSChris Mason 27485f39d397SChris Mason btrfs_mark_buffer_dirty(c); 27495f39d397SChris Mason btrfs_mark_buffer_dirty(split); 27505f39d397SChris Mason 27516ad3cf6dSDavid Sterba insert_ptr(trans, path, &disk_key, split->start, 2752c3e06965SJan Schmidt path->slots[level + 1] + 1, level + 1); 2753aa5d6bedSChris Mason 27545de08d7dSChris Mason if (path->slots[level] >= mid) { 27555c680ed6SChris Mason path->slots[level] -= mid; 2756925baeddSChris Mason btrfs_tree_unlock(c); 27575f39d397SChris Mason free_extent_buffer(c); 27585f39d397SChris Mason path->nodes[level] = split; 27595c680ed6SChris Mason path->slots[level + 1] += 1; 2760eb60ceacSChris Mason } else { 2761925baeddSChris Mason btrfs_tree_unlock(split); 27625f39d397SChris Mason free_extent_buffer(split); 2763be0e5c09SChris Mason } 2764d5286a92SNikolay Borisov return 0; 2765be0e5c09SChris Mason } 2766be0e5c09SChris Mason 276774123bd7SChris Mason /* 276874123bd7SChris Mason * how many bytes are required to store the items in a leaf. start 276974123bd7SChris Mason * and nr indicate which items in the leaf to check. This totals up the 277074123bd7SChris Mason * space used both by the item structs and the item data 277174123bd7SChris Mason */ 27725f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr) 2773be0e5c09SChris Mason { 2774be0e5c09SChris Mason int data_len; 27755f39d397SChris Mason int nritems = btrfs_header_nritems(l); 2776d4dbff95SChris Mason int end = min(nritems, start + nr) - 1; 2777be0e5c09SChris Mason 2778be0e5c09SChris Mason if (!nr) 2779be0e5c09SChris Mason return 0; 27803212fa14SJosef Bacik data_len = btrfs_item_offset(l, start) + btrfs_item_size(l, start); 27813212fa14SJosef Bacik data_len = data_len - btrfs_item_offset(l, end); 27820783fcfcSChris Mason data_len += sizeof(struct btrfs_item) * nr; 2783d4dbff95SChris Mason WARN_ON(data_len < 0); 2784be0e5c09SChris Mason return data_len; 2785be0e5c09SChris Mason } 2786be0e5c09SChris Mason 278774123bd7SChris Mason /* 2788d4dbff95SChris Mason * The space between the end of the leaf items and 2789d4dbff95SChris Mason * the start of the leaf data. IOW, how much room 2790d4dbff95SChris Mason * the leaf has left for both items and data 2791d4dbff95SChris Mason */ 2792e902baacSDavid Sterba noinline int btrfs_leaf_free_space(struct extent_buffer *leaf) 2793d4dbff95SChris Mason { 2794e902baacSDavid Sterba struct btrfs_fs_info *fs_info = leaf->fs_info; 27955f39d397SChris Mason int nritems = btrfs_header_nritems(leaf); 27965f39d397SChris Mason int ret; 27970b246afaSJeff Mahoney 27980b246afaSJeff Mahoney ret = BTRFS_LEAF_DATA_SIZE(fs_info) - leaf_space_used(leaf, 0, nritems); 27995f39d397SChris Mason if (ret < 0) { 28000b246afaSJeff Mahoney btrfs_crit(fs_info, 2801efe120a0SFrank Holton "leaf free space ret %d, leaf data size %lu, used %d nritems %d", 2802da17066cSJeff Mahoney ret, 28030b246afaSJeff Mahoney (unsigned long) BTRFS_LEAF_DATA_SIZE(fs_info), 28045f39d397SChris Mason leaf_space_used(leaf, 0, nritems), nritems); 28055f39d397SChris Mason } 28065f39d397SChris Mason return ret; 2807d4dbff95SChris Mason } 2808d4dbff95SChris Mason 280999d8f83cSChris Mason /* 281099d8f83cSChris Mason * min slot controls the lowest index we're willing to push to the 281199d8f83cSChris Mason * right. We'll push up to and including min_slot, but no lower 281299d8f83cSChris Mason */ 2813f72f0010SDavid Sterba static noinline int __push_leaf_right(struct btrfs_path *path, 281444871b1bSChris Mason int data_size, int empty, 281544871b1bSChris Mason struct extent_buffer *right, 281699d8f83cSChris Mason int free_space, u32 left_nritems, 281799d8f83cSChris Mason u32 min_slot) 281800ec4c51SChris Mason { 2819f72f0010SDavid Sterba struct btrfs_fs_info *fs_info = right->fs_info; 28205f39d397SChris Mason struct extent_buffer *left = path->nodes[0]; 282144871b1bSChris Mason struct extent_buffer *upper = path->nodes[1]; 2822cfed81a0SChris Mason struct btrfs_map_token token; 28235f39d397SChris Mason struct btrfs_disk_key disk_key; 282400ec4c51SChris Mason int slot; 282534a38218SChris Mason u32 i; 282600ec4c51SChris Mason int push_space = 0; 282700ec4c51SChris Mason int push_items = 0; 282834a38218SChris Mason u32 nr; 28297518a238SChris Mason u32 right_nritems; 28305f39d397SChris Mason u32 data_end; 2831db94535dSChris Mason u32 this_item_size; 283200ec4c51SChris Mason 283334a38218SChris Mason if (empty) 283434a38218SChris Mason nr = 0; 283534a38218SChris Mason else 283699d8f83cSChris Mason nr = max_t(u32, 1, min_slot); 283734a38218SChris Mason 283831840ae1SZheng Yan if (path->slots[0] >= left_nritems) 283987b29b20SYan Zheng push_space += data_size; 284031840ae1SZheng Yan 284144871b1bSChris Mason slot = path->slots[1]; 284234a38218SChris Mason i = left_nritems - 1; 284334a38218SChris Mason while (i >= nr) { 284431840ae1SZheng Yan if (!empty && push_items > 0) { 284531840ae1SZheng Yan if (path->slots[0] > i) 284631840ae1SZheng Yan break; 284731840ae1SZheng Yan if (path->slots[0] == i) { 2848e902baacSDavid Sterba int space = btrfs_leaf_free_space(left); 2849e902baacSDavid Sterba 285031840ae1SZheng Yan if (space + push_space * 2 > free_space) 285131840ae1SZheng Yan break; 285231840ae1SZheng Yan } 285331840ae1SZheng Yan } 285431840ae1SZheng Yan 285500ec4c51SChris Mason if (path->slots[0] == i) 285687b29b20SYan Zheng push_space += data_size; 2857db94535dSChris Mason 28583212fa14SJosef Bacik this_item_size = btrfs_item_size(left, i); 285974794207SJosef Bacik if (this_item_size + sizeof(struct btrfs_item) + 286074794207SJosef Bacik push_space > free_space) 286100ec4c51SChris Mason break; 286231840ae1SZheng Yan 286300ec4c51SChris Mason push_items++; 286474794207SJosef Bacik push_space += this_item_size + sizeof(struct btrfs_item); 286534a38218SChris Mason if (i == 0) 286634a38218SChris Mason break; 286734a38218SChris Mason i--; 2868db94535dSChris Mason } 28695f39d397SChris Mason 2870925baeddSChris Mason if (push_items == 0) 2871925baeddSChris Mason goto out_unlock; 28725f39d397SChris Mason 28736c1500f2SJulia Lawall WARN_ON(!empty && push_items == left_nritems); 28745f39d397SChris Mason 287500ec4c51SChris Mason /* push left to right */ 28765f39d397SChris Mason right_nritems = btrfs_header_nritems(right); 287734a38218SChris Mason 2878dc2e724eSJosef Bacik push_space = btrfs_item_data_end(left, left_nritems - push_items); 28798f881e8cSDavid Sterba push_space -= leaf_data_end(left); 28805f39d397SChris Mason 288100ec4c51SChris Mason /* make room in the right data area */ 28828f881e8cSDavid Sterba data_end = leaf_data_end(right); 28835f39d397SChris Mason memmove_extent_buffer(right, 28843d9ec8c4SNikolay Borisov BTRFS_LEAF_DATA_OFFSET + data_end - push_space, 28853d9ec8c4SNikolay Borisov BTRFS_LEAF_DATA_OFFSET + data_end, 28860b246afaSJeff Mahoney BTRFS_LEAF_DATA_SIZE(fs_info) - data_end); 28875f39d397SChris Mason 288800ec4c51SChris Mason /* copy from the left data area */ 28893d9ec8c4SNikolay Borisov copy_extent_buffer(right, left, BTRFS_LEAF_DATA_OFFSET + 28900b246afaSJeff Mahoney BTRFS_LEAF_DATA_SIZE(fs_info) - push_space, 28918f881e8cSDavid Sterba BTRFS_LEAF_DATA_OFFSET + leaf_data_end(left), 2892d6025579SChris Mason push_space); 28935f39d397SChris Mason 28945f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(push_items), 28955f39d397SChris Mason btrfs_item_nr_offset(0), 28960783fcfcSChris Mason right_nritems * sizeof(struct btrfs_item)); 28975f39d397SChris Mason 289800ec4c51SChris Mason /* copy the items from left to right */ 28995f39d397SChris Mason copy_extent_buffer(right, left, btrfs_item_nr_offset(0), 29005f39d397SChris Mason btrfs_item_nr_offset(left_nritems - push_items), 29010783fcfcSChris Mason push_items * sizeof(struct btrfs_item)); 290200ec4c51SChris Mason 290300ec4c51SChris Mason /* update the item pointers */ 2904c82f823cSDavid Sterba btrfs_init_map_token(&token, right); 29057518a238SChris Mason right_nritems += push_items; 29065f39d397SChris Mason btrfs_set_header_nritems(right, right_nritems); 29070b246afaSJeff Mahoney push_space = BTRFS_LEAF_DATA_SIZE(fs_info); 29087518a238SChris Mason for (i = 0; i < right_nritems; i++) { 29093212fa14SJosef Bacik push_space -= btrfs_token_item_size(&token, i); 29103212fa14SJosef Bacik btrfs_set_token_item_offset(&token, i, push_space); 2911db94535dSChris Mason } 2912db94535dSChris Mason 29137518a238SChris Mason left_nritems -= push_items; 29145f39d397SChris Mason btrfs_set_header_nritems(left, left_nritems); 291500ec4c51SChris Mason 291634a38218SChris Mason if (left_nritems) 29175f39d397SChris Mason btrfs_mark_buffer_dirty(left); 2918f0486c68SYan, Zheng else 29196a884d7dSDavid Sterba btrfs_clean_tree_block(left); 2920f0486c68SYan, Zheng 29215f39d397SChris Mason btrfs_mark_buffer_dirty(right); 2922a429e513SChris Mason 29235f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 29245f39d397SChris Mason btrfs_set_node_key(upper, &disk_key, slot + 1); 2925d6025579SChris Mason btrfs_mark_buffer_dirty(upper); 292602217ed2SChris Mason 292700ec4c51SChris Mason /* then fixup the leaf pointer in the path */ 29287518a238SChris Mason if (path->slots[0] >= left_nritems) { 29297518a238SChris Mason path->slots[0] -= left_nritems; 2930925baeddSChris Mason if (btrfs_header_nritems(path->nodes[0]) == 0) 29316a884d7dSDavid Sterba btrfs_clean_tree_block(path->nodes[0]); 2932925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 29335f39d397SChris Mason free_extent_buffer(path->nodes[0]); 29345f39d397SChris Mason path->nodes[0] = right; 293500ec4c51SChris Mason path->slots[1] += 1; 293600ec4c51SChris Mason } else { 2937925baeddSChris Mason btrfs_tree_unlock(right); 29385f39d397SChris Mason free_extent_buffer(right); 293900ec4c51SChris Mason } 294000ec4c51SChris Mason return 0; 2941925baeddSChris Mason 2942925baeddSChris Mason out_unlock: 2943925baeddSChris Mason btrfs_tree_unlock(right); 2944925baeddSChris Mason free_extent_buffer(right); 2945925baeddSChris Mason return 1; 294600ec4c51SChris Mason } 2947925baeddSChris Mason 294800ec4c51SChris Mason /* 294944871b1bSChris Mason * push some data in the path leaf to the right, trying to free up at 295074123bd7SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 295144871b1bSChris Mason * 295244871b1bSChris Mason * returns 1 if the push failed because the other node didn't have enough 295344871b1bSChris Mason * room, 0 if everything worked out and < 0 if there were major errors. 295499d8f83cSChris Mason * 295599d8f83cSChris Mason * this will push starting from min_slot to the end of the leaf. It won't 295699d8f83cSChris Mason * push any slot lower than min_slot 295774123bd7SChris Mason */ 295844871b1bSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root 295999d8f83cSChris Mason *root, struct btrfs_path *path, 296099d8f83cSChris Mason int min_data_size, int data_size, 296199d8f83cSChris Mason int empty, u32 min_slot) 2962be0e5c09SChris Mason { 296344871b1bSChris Mason struct extent_buffer *left = path->nodes[0]; 296444871b1bSChris Mason struct extent_buffer *right; 296544871b1bSChris Mason struct extent_buffer *upper; 296644871b1bSChris Mason int slot; 296744871b1bSChris Mason int free_space; 296844871b1bSChris Mason u32 left_nritems; 296944871b1bSChris Mason int ret; 297044871b1bSChris Mason 297144871b1bSChris Mason if (!path->nodes[1]) 297244871b1bSChris Mason return 1; 297344871b1bSChris Mason 297444871b1bSChris Mason slot = path->slots[1]; 297544871b1bSChris Mason upper = path->nodes[1]; 297644871b1bSChris Mason if (slot >= btrfs_header_nritems(upper) - 1) 297744871b1bSChris Mason return 1; 297844871b1bSChris Mason 297949d0c642SFilipe Manana btrfs_assert_tree_write_locked(path->nodes[1]); 298044871b1bSChris Mason 29814b231ae4SDavid Sterba right = btrfs_read_node_slot(upper, slot + 1); 2982fb770ae4SLiu Bo /* 2983fb770ae4SLiu Bo * slot + 1 is not valid or we fail to read the right node, 2984fb770ae4SLiu Bo * no big deal, just return. 2985fb770ae4SLiu Bo */ 2986fb770ae4SLiu Bo if (IS_ERR(right)) 298791ca338dSTsutomu Itoh return 1; 298891ca338dSTsutomu Itoh 2989bf77467aSJosef Bacik __btrfs_tree_lock(right, BTRFS_NESTING_RIGHT); 299044871b1bSChris Mason 2991e902baacSDavid Sterba free_space = btrfs_leaf_free_space(right); 299244871b1bSChris Mason if (free_space < data_size) 299344871b1bSChris Mason goto out_unlock; 299444871b1bSChris Mason 299544871b1bSChris Mason ret = btrfs_cow_block(trans, root, right, upper, 2996bf59a5a2SJosef Bacik slot + 1, &right, BTRFS_NESTING_RIGHT_COW); 299744871b1bSChris Mason if (ret) 299844871b1bSChris Mason goto out_unlock; 299944871b1bSChris Mason 300044871b1bSChris Mason left_nritems = btrfs_header_nritems(left); 300144871b1bSChris Mason if (left_nritems == 0) 300244871b1bSChris Mason goto out_unlock; 300344871b1bSChris Mason 3004d16c702fSQu Wenruo if (check_sibling_keys(left, right)) { 3005d16c702fSQu Wenruo ret = -EUCLEAN; 3006d16c702fSQu Wenruo btrfs_tree_unlock(right); 3007d16c702fSQu Wenruo free_extent_buffer(right); 3008d16c702fSQu Wenruo return ret; 3009d16c702fSQu Wenruo } 30102ef1fed2SFilipe David Borba Manana if (path->slots[0] == left_nritems && !empty) { 30112ef1fed2SFilipe David Borba Manana /* Key greater than all keys in the leaf, right neighbor has 30122ef1fed2SFilipe David Borba Manana * enough room for it and we're not emptying our leaf to delete 30132ef1fed2SFilipe David Borba Manana * it, therefore use right neighbor to insert the new item and 301452042d8eSAndrea Gelmini * no need to touch/dirty our left leaf. */ 30152ef1fed2SFilipe David Borba Manana btrfs_tree_unlock(left); 30162ef1fed2SFilipe David Borba Manana free_extent_buffer(left); 30172ef1fed2SFilipe David Borba Manana path->nodes[0] = right; 30182ef1fed2SFilipe David Borba Manana path->slots[0] = 0; 30192ef1fed2SFilipe David Borba Manana path->slots[1]++; 30202ef1fed2SFilipe David Borba Manana return 0; 30212ef1fed2SFilipe David Borba Manana } 30222ef1fed2SFilipe David Borba Manana 3023f72f0010SDavid Sterba return __push_leaf_right(path, min_data_size, empty, 302499d8f83cSChris Mason right, free_space, left_nritems, min_slot); 302544871b1bSChris Mason out_unlock: 302644871b1bSChris Mason btrfs_tree_unlock(right); 302744871b1bSChris Mason free_extent_buffer(right); 302844871b1bSChris Mason return 1; 302944871b1bSChris Mason } 303044871b1bSChris Mason 303144871b1bSChris Mason /* 303244871b1bSChris Mason * push some data in the path leaf to the left, trying to free up at 303344871b1bSChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 303499d8f83cSChris Mason * 303599d8f83cSChris Mason * max_slot can put a limit on how far into the leaf we'll push items. The 303699d8f83cSChris Mason * item at 'max_slot' won't be touched. Use (u32)-1 to make us do all the 303799d8f83cSChris Mason * items 303844871b1bSChris Mason */ 30398087c193SDavid Sterba static noinline int __push_leaf_left(struct btrfs_path *path, int data_size, 304044871b1bSChris Mason int empty, struct extent_buffer *left, 304199d8f83cSChris Mason int free_space, u32 right_nritems, 304299d8f83cSChris Mason u32 max_slot) 304344871b1bSChris Mason { 30448087c193SDavid Sterba struct btrfs_fs_info *fs_info = left->fs_info; 30455f39d397SChris Mason struct btrfs_disk_key disk_key; 30465f39d397SChris Mason struct extent_buffer *right = path->nodes[0]; 3047be0e5c09SChris Mason int i; 3048be0e5c09SChris Mason int push_space = 0; 3049be0e5c09SChris Mason int push_items = 0; 30507518a238SChris Mason u32 old_left_nritems; 305134a38218SChris Mason u32 nr; 3052aa5d6bedSChris Mason int ret = 0; 3053db94535dSChris Mason u32 this_item_size; 3054db94535dSChris Mason u32 old_left_item_size; 3055cfed81a0SChris Mason struct btrfs_map_token token; 3056cfed81a0SChris Mason 305734a38218SChris Mason if (empty) 305899d8f83cSChris Mason nr = min(right_nritems, max_slot); 305934a38218SChris Mason else 306099d8f83cSChris Mason nr = min(right_nritems - 1, max_slot); 306134a38218SChris Mason 306234a38218SChris Mason for (i = 0; i < nr; i++) { 306331840ae1SZheng Yan if (!empty && push_items > 0) { 306431840ae1SZheng Yan if (path->slots[0] < i) 306531840ae1SZheng Yan break; 306631840ae1SZheng Yan if (path->slots[0] == i) { 3067e902baacSDavid Sterba int space = btrfs_leaf_free_space(right); 3068e902baacSDavid Sterba 306931840ae1SZheng Yan if (space + push_space * 2 > free_space) 307031840ae1SZheng Yan break; 307131840ae1SZheng Yan } 307231840ae1SZheng Yan } 307331840ae1SZheng Yan 3074be0e5c09SChris Mason if (path->slots[0] == i) 307587b29b20SYan Zheng push_space += data_size; 3076db94535dSChris Mason 30773212fa14SJosef Bacik this_item_size = btrfs_item_size(right, i); 307874794207SJosef Bacik if (this_item_size + sizeof(struct btrfs_item) + push_space > 307974794207SJosef Bacik free_space) 3080be0e5c09SChris Mason break; 3081db94535dSChris Mason 3082be0e5c09SChris Mason push_items++; 308374794207SJosef Bacik push_space += this_item_size + sizeof(struct btrfs_item); 3084be0e5c09SChris Mason } 3085db94535dSChris Mason 3086be0e5c09SChris Mason if (push_items == 0) { 3087925baeddSChris Mason ret = 1; 3088925baeddSChris Mason goto out; 3089be0e5c09SChris Mason } 3090fae7f21cSDulshani Gunawardhana WARN_ON(!empty && push_items == btrfs_header_nritems(right)); 30915f39d397SChris Mason 3092be0e5c09SChris Mason /* push data from right to left */ 30935f39d397SChris Mason copy_extent_buffer(left, right, 30945f39d397SChris Mason btrfs_item_nr_offset(btrfs_header_nritems(left)), 30955f39d397SChris Mason btrfs_item_nr_offset(0), 30965f39d397SChris Mason push_items * sizeof(struct btrfs_item)); 30975f39d397SChris Mason 30980b246afaSJeff Mahoney push_space = BTRFS_LEAF_DATA_SIZE(fs_info) - 30993212fa14SJosef Bacik btrfs_item_offset(right, push_items - 1); 31005f39d397SChris Mason 31013d9ec8c4SNikolay Borisov copy_extent_buffer(left, right, BTRFS_LEAF_DATA_OFFSET + 31028f881e8cSDavid Sterba leaf_data_end(left) - push_space, 31033d9ec8c4SNikolay Borisov BTRFS_LEAF_DATA_OFFSET + 31043212fa14SJosef Bacik btrfs_item_offset(right, push_items - 1), 3105be0e5c09SChris Mason push_space); 31065f39d397SChris Mason old_left_nritems = btrfs_header_nritems(left); 310787b29b20SYan Zheng BUG_ON(old_left_nritems <= 0); 3108eb60ceacSChris Mason 3109c82f823cSDavid Sterba btrfs_init_map_token(&token, left); 31103212fa14SJosef Bacik old_left_item_size = btrfs_item_offset(left, old_left_nritems - 1); 3111be0e5c09SChris Mason for (i = old_left_nritems; i < old_left_nritems + push_items; i++) { 31125f39d397SChris Mason u32 ioff; 3113db94535dSChris Mason 31143212fa14SJosef Bacik ioff = btrfs_token_item_offset(&token, i); 31153212fa14SJosef Bacik btrfs_set_token_item_offset(&token, i, 3116cc4c13d5SDavid Sterba ioff - (BTRFS_LEAF_DATA_SIZE(fs_info) - old_left_item_size)); 3117be0e5c09SChris Mason } 31185f39d397SChris Mason btrfs_set_header_nritems(left, old_left_nritems + push_items); 3119be0e5c09SChris Mason 3120be0e5c09SChris Mason /* fixup right node */ 312131b1a2bdSJulia Lawall if (push_items > right_nritems) 312231b1a2bdSJulia Lawall WARN(1, KERN_CRIT "push items %d nr %u\n", push_items, 3123d397712bSChris Mason right_nritems); 312434a38218SChris Mason 312534a38218SChris Mason if (push_items < right_nritems) { 31263212fa14SJosef Bacik push_space = btrfs_item_offset(right, push_items - 1) - 31278f881e8cSDavid Sterba leaf_data_end(right); 31283d9ec8c4SNikolay Borisov memmove_extent_buffer(right, BTRFS_LEAF_DATA_OFFSET + 31290b246afaSJeff Mahoney BTRFS_LEAF_DATA_SIZE(fs_info) - push_space, 31303d9ec8c4SNikolay Borisov BTRFS_LEAF_DATA_OFFSET + 31318f881e8cSDavid Sterba leaf_data_end(right), push_space); 31325f39d397SChris Mason 31335f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(0), 31345f39d397SChris Mason btrfs_item_nr_offset(push_items), 31355f39d397SChris Mason (btrfs_header_nritems(right) - push_items) * 31360783fcfcSChris Mason sizeof(struct btrfs_item)); 313734a38218SChris Mason } 3138c82f823cSDavid Sterba 3139c82f823cSDavid Sterba btrfs_init_map_token(&token, right); 3140eef1c494SYan right_nritems -= push_items; 3141eef1c494SYan btrfs_set_header_nritems(right, right_nritems); 31420b246afaSJeff Mahoney push_space = BTRFS_LEAF_DATA_SIZE(fs_info); 31435f39d397SChris Mason for (i = 0; i < right_nritems; i++) { 31443212fa14SJosef Bacik push_space = push_space - btrfs_token_item_size(&token, i); 31453212fa14SJosef Bacik btrfs_set_token_item_offset(&token, i, push_space); 3146db94535dSChris Mason } 3147eb60ceacSChris Mason 31485f39d397SChris Mason btrfs_mark_buffer_dirty(left); 314934a38218SChris Mason if (right_nritems) 31505f39d397SChris Mason btrfs_mark_buffer_dirty(right); 3151f0486c68SYan, Zheng else 31526a884d7dSDavid Sterba btrfs_clean_tree_block(right); 3153098f59c2SChris Mason 31545f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 3155b167fa91SNikolay Borisov fixup_low_keys(path, &disk_key, 1); 3156be0e5c09SChris Mason 3157be0e5c09SChris Mason /* then fixup the leaf pointer in the path */ 3158be0e5c09SChris Mason if (path->slots[0] < push_items) { 3159be0e5c09SChris Mason path->slots[0] += old_left_nritems; 3160925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 31615f39d397SChris Mason free_extent_buffer(path->nodes[0]); 31625f39d397SChris Mason path->nodes[0] = left; 3163be0e5c09SChris Mason path->slots[1] -= 1; 3164be0e5c09SChris Mason } else { 3165925baeddSChris Mason btrfs_tree_unlock(left); 31665f39d397SChris Mason free_extent_buffer(left); 3167be0e5c09SChris Mason path->slots[0] -= push_items; 3168be0e5c09SChris Mason } 3169eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 3170aa5d6bedSChris Mason return ret; 3171925baeddSChris Mason out: 3172925baeddSChris Mason btrfs_tree_unlock(left); 3173925baeddSChris Mason free_extent_buffer(left); 3174925baeddSChris Mason return ret; 3175be0e5c09SChris Mason } 3176be0e5c09SChris Mason 317774123bd7SChris Mason /* 317844871b1bSChris Mason * push some data in the path leaf to the left, trying to free up at 317944871b1bSChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 318099d8f83cSChris Mason * 318199d8f83cSChris Mason * max_slot can put a limit on how far into the leaf we'll push items. The 318299d8f83cSChris Mason * item at 'max_slot' won't be touched. Use (u32)-1 to make us push all the 318399d8f83cSChris Mason * items 318444871b1bSChris Mason */ 318544871b1bSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root 318699d8f83cSChris Mason *root, struct btrfs_path *path, int min_data_size, 318799d8f83cSChris Mason int data_size, int empty, u32 max_slot) 318844871b1bSChris Mason { 318944871b1bSChris Mason struct extent_buffer *right = path->nodes[0]; 319044871b1bSChris Mason struct extent_buffer *left; 319144871b1bSChris Mason int slot; 319244871b1bSChris Mason int free_space; 319344871b1bSChris Mason u32 right_nritems; 319444871b1bSChris Mason int ret = 0; 319544871b1bSChris Mason 319644871b1bSChris Mason slot = path->slots[1]; 319744871b1bSChris Mason if (slot == 0) 319844871b1bSChris Mason return 1; 319944871b1bSChris Mason if (!path->nodes[1]) 320044871b1bSChris Mason return 1; 320144871b1bSChris Mason 320244871b1bSChris Mason right_nritems = btrfs_header_nritems(right); 320344871b1bSChris Mason if (right_nritems == 0) 320444871b1bSChris Mason return 1; 320544871b1bSChris Mason 320649d0c642SFilipe Manana btrfs_assert_tree_write_locked(path->nodes[1]); 320744871b1bSChris Mason 32084b231ae4SDavid Sterba left = btrfs_read_node_slot(path->nodes[1], slot - 1); 3209fb770ae4SLiu Bo /* 3210fb770ae4SLiu Bo * slot - 1 is not valid or we fail to read the left node, 3211fb770ae4SLiu Bo * no big deal, just return. 3212fb770ae4SLiu Bo */ 3213fb770ae4SLiu Bo if (IS_ERR(left)) 321491ca338dSTsutomu Itoh return 1; 321591ca338dSTsutomu Itoh 3216bf77467aSJosef Bacik __btrfs_tree_lock(left, BTRFS_NESTING_LEFT); 321744871b1bSChris Mason 3218e902baacSDavid Sterba free_space = btrfs_leaf_free_space(left); 321944871b1bSChris Mason if (free_space < data_size) { 322044871b1bSChris Mason ret = 1; 322144871b1bSChris Mason goto out; 322244871b1bSChris Mason } 322344871b1bSChris Mason 322444871b1bSChris Mason ret = btrfs_cow_block(trans, root, left, 32259631e4ccSJosef Bacik path->nodes[1], slot - 1, &left, 3226bf59a5a2SJosef Bacik BTRFS_NESTING_LEFT_COW); 322744871b1bSChris Mason if (ret) { 322844871b1bSChris Mason /* we hit -ENOSPC, but it isn't fatal here */ 322979787eaaSJeff Mahoney if (ret == -ENOSPC) 323044871b1bSChris Mason ret = 1; 323144871b1bSChris Mason goto out; 323244871b1bSChris Mason } 323344871b1bSChris Mason 3234d16c702fSQu Wenruo if (check_sibling_keys(left, right)) { 3235d16c702fSQu Wenruo ret = -EUCLEAN; 3236d16c702fSQu Wenruo goto out; 3237d16c702fSQu Wenruo } 32388087c193SDavid Sterba return __push_leaf_left(path, min_data_size, 323999d8f83cSChris Mason empty, left, free_space, right_nritems, 324099d8f83cSChris Mason max_slot); 324144871b1bSChris Mason out: 324244871b1bSChris Mason btrfs_tree_unlock(left); 324344871b1bSChris Mason free_extent_buffer(left); 324444871b1bSChris Mason return ret; 324544871b1bSChris Mason } 324644871b1bSChris Mason 324744871b1bSChris Mason /* 324874123bd7SChris Mason * split the path's leaf in two, making sure there is at least data_size 324974123bd7SChris Mason * available for the resulting leaf level of the path. 325074123bd7SChris Mason */ 3251143bede5SJeff Mahoney static noinline void copy_for_split(struct btrfs_trans_handle *trans, 325244871b1bSChris Mason struct btrfs_path *path, 325344871b1bSChris Mason struct extent_buffer *l, 325444871b1bSChris Mason struct extent_buffer *right, 325544871b1bSChris Mason int slot, int mid, int nritems) 3256be0e5c09SChris Mason { 325794f94ad9SDavid Sterba struct btrfs_fs_info *fs_info = trans->fs_info; 3258be0e5c09SChris Mason int data_copy_size; 3259be0e5c09SChris Mason int rt_data_off; 3260be0e5c09SChris Mason int i; 3261d4dbff95SChris Mason struct btrfs_disk_key disk_key; 3262cfed81a0SChris Mason struct btrfs_map_token token; 3263cfed81a0SChris Mason 32645f39d397SChris Mason nritems = nritems - mid; 32655f39d397SChris Mason btrfs_set_header_nritems(right, nritems); 3266dc2e724eSJosef Bacik data_copy_size = btrfs_item_data_end(l, mid) - leaf_data_end(l); 32675f39d397SChris Mason 32685f39d397SChris Mason copy_extent_buffer(right, l, btrfs_item_nr_offset(0), 32695f39d397SChris Mason btrfs_item_nr_offset(mid), 32705f39d397SChris Mason nritems * sizeof(struct btrfs_item)); 32715f39d397SChris Mason 32725f39d397SChris Mason copy_extent_buffer(right, l, 32733d9ec8c4SNikolay Borisov BTRFS_LEAF_DATA_OFFSET + BTRFS_LEAF_DATA_SIZE(fs_info) - 32743d9ec8c4SNikolay Borisov data_copy_size, BTRFS_LEAF_DATA_OFFSET + 32758f881e8cSDavid Sterba leaf_data_end(l), data_copy_size); 327674123bd7SChris Mason 3277dc2e724eSJosef Bacik rt_data_off = BTRFS_LEAF_DATA_SIZE(fs_info) - btrfs_item_data_end(l, mid); 32785f39d397SChris Mason 3279c82f823cSDavid Sterba btrfs_init_map_token(&token, right); 32805f39d397SChris Mason for (i = 0; i < nritems; i++) { 3281db94535dSChris Mason u32 ioff; 3282db94535dSChris Mason 32833212fa14SJosef Bacik ioff = btrfs_token_item_offset(&token, i); 32843212fa14SJosef Bacik btrfs_set_token_item_offset(&token, i, ioff + rt_data_off); 32850783fcfcSChris Mason } 328674123bd7SChris Mason 32875f39d397SChris Mason btrfs_set_header_nritems(l, mid); 32885f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 32896ad3cf6dSDavid Sterba insert_ptr(trans, path, &disk_key, right->start, path->slots[1] + 1, 1); 32905f39d397SChris Mason 32915f39d397SChris Mason btrfs_mark_buffer_dirty(right); 32925f39d397SChris Mason btrfs_mark_buffer_dirty(l); 3293eb60ceacSChris Mason BUG_ON(path->slots[0] != slot); 32945f39d397SChris Mason 3295be0e5c09SChris Mason if (mid <= slot) { 3296925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 32975f39d397SChris Mason free_extent_buffer(path->nodes[0]); 32985f39d397SChris Mason path->nodes[0] = right; 3299be0e5c09SChris Mason path->slots[0] -= mid; 3300be0e5c09SChris Mason path->slots[1] += 1; 3301925baeddSChris Mason } else { 3302925baeddSChris Mason btrfs_tree_unlock(right); 33035f39d397SChris Mason free_extent_buffer(right); 3304925baeddSChris Mason } 33055f39d397SChris Mason 3306eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 330744871b1bSChris Mason } 330844871b1bSChris Mason 330944871b1bSChris Mason /* 331099d8f83cSChris Mason * double splits happen when we need to insert a big item in the middle 331199d8f83cSChris Mason * of a leaf. A double split can leave us with 3 mostly empty leaves: 331299d8f83cSChris Mason * leaf: [ slots 0 - N] [ our target ] [ N + 1 - total in leaf ] 331399d8f83cSChris Mason * A B C 331499d8f83cSChris Mason * 331599d8f83cSChris Mason * We avoid this by trying to push the items on either side of our target 331699d8f83cSChris Mason * into the adjacent leaves. If all goes well we can avoid the double split 331799d8f83cSChris Mason * completely. 331899d8f83cSChris Mason */ 331999d8f83cSChris Mason static noinline int push_for_double_split(struct btrfs_trans_handle *trans, 332099d8f83cSChris Mason struct btrfs_root *root, 332199d8f83cSChris Mason struct btrfs_path *path, 332299d8f83cSChris Mason int data_size) 332399d8f83cSChris Mason { 332499d8f83cSChris Mason int ret; 332599d8f83cSChris Mason int progress = 0; 332699d8f83cSChris Mason int slot; 332799d8f83cSChris Mason u32 nritems; 33285a4267caSFilipe David Borba Manana int space_needed = data_size; 332999d8f83cSChris Mason 333099d8f83cSChris Mason slot = path->slots[0]; 33315a4267caSFilipe David Borba Manana if (slot < btrfs_header_nritems(path->nodes[0])) 3332e902baacSDavid Sterba space_needed -= btrfs_leaf_free_space(path->nodes[0]); 333399d8f83cSChris Mason 333499d8f83cSChris Mason /* 333599d8f83cSChris Mason * try to push all the items after our slot into the 333699d8f83cSChris Mason * right leaf 333799d8f83cSChris Mason */ 33385a4267caSFilipe David Borba Manana ret = push_leaf_right(trans, root, path, 1, space_needed, 0, slot); 333999d8f83cSChris Mason if (ret < 0) 334099d8f83cSChris Mason return ret; 334199d8f83cSChris Mason 334299d8f83cSChris Mason if (ret == 0) 334399d8f83cSChris Mason progress++; 334499d8f83cSChris Mason 334599d8f83cSChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 334699d8f83cSChris Mason /* 334799d8f83cSChris Mason * our goal is to get our slot at the start or end of a leaf. If 334899d8f83cSChris Mason * we've done so we're done 334999d8f83cSChris Mason */ 335099d8f83cSChris Mason if (path->slots[0] == 0 || path->slots[0] == nritems) 335199d8f83cSChris Mason return 0; 335299d8f83cSChris Mason 3353e902baacSDavid Sterba if (btrfs_leaf_free_space(path->nodes[0]) >= data_size) 335499d8f83cSChris Mason return 0; 335599d8f83cSChris Mason 335699d8f83cSChris Mason /* try to push all the items before our slot into the next leaf */ 335799d8f83cSChris Mason slot = path->slots[0]; 3358263d3995SFilipe Manana space_needed = data_size; 3359263d3995SFilipe Manana if (slot > 0) 3360e902baacSDavid Sterba space_needed -= btrfs_leaf_free_space(path->nodes[0]); 33615a4267caSFilipe David Borba Manana ret = push_leaf_left(trans, root, path, 1, space_needed, 0, slot); 336299d8f83cSChris Mason if (ret < 0) 336399d8f83cSChris Mason return ret; 336499d8f83cSChris Mason 336599d8f83cSChris Mason if (ret == 0) 336699d8f83cSChris Mason progress++; 336799d8f83cSChris Mason 336899d8f83cSChris Mason if (progress) 336999d8f83cSChris Mason return 0; 337099d8f83cSChris Mason return 1; 337199d8f83cSChris Mason } 337299d8f83cSChris Mason 337399d8f83cSChris Mason /* 337444871b1bSChris Mason * split the path's leaf in two, making sure there is at least data_size 337544871b1bSChris Mason * available for the resulting leaf level of the path. 337644871b1bSChris Mason * 337744871b1bSChris Mason * returns 0 if all went well and < 0 on failure. 337844871b1bSChris Mason */ 337944871b1bSChris Mason static noinline int split_leaf(struct btrfs_trans_handle *trans, 338044871b1bSChris Mason struct btrfs_root *root, 3381310712b2SOmar Sandoval const struct btrfs_key *ins_key, 338244871b1bSChris Mason struct btrfs_path *path, int data_size, 338344871b1bSChris Mason int extend) 338444871b1bSChris Mason { 33855d4f98a2SYan Zheng struct btrfs_disk_key disk_key; 338644871b1bSChris Mason struct extent_buffer *l; 338744871b1bSChris Mason u32 nritems; 338844871b1bSChris Mason int mid; 338944871b1bSChris Mason int slot; 339044871b1bSChris Mason struct extent_buffer *right; 3391b7a0365eSDaniel Dressler struct btrfs_fs_info *fs_info = root->fs_info; 339244871b1bSChris Mason int ret = 0; 339344871b1bSChris Mason int wret; 33945d4f98a2SYan Zheng int split; 339544871b1bSChris Mason int num_doubles = 0; 339699d8f83cSChris Mason int tried_avoid_double = 0; 339744871b1bSChris Mason 3398a5719521SYan, Zheng l = path->nodes[0]; 3399a5719521SYan, Zheng slot = path->slots[0]; 34003212fa14SJosef Bacik if (extend && data_size + btrfs_item_size(l, slot) + 34010b246afaSJeff Mahoney sizeof(struct btrfs_item) > BTRFS_LEAF_DATA_SIZE(fs_info)) 3402a5719521SYan, Zheng return -EOVERFLOW; 3403a5719521SYan, Zheng 340444871b1bSChris Mason /* first try to make some room by pushing left and right */ 340533157e05SLiu Bo if (data_size && path->nodes[1]) { 34065a4267caSFilipe David Borba Manana int space_needed = data_size; 34075a4267caSFilipe David Borba Manana 34085a4267caSFilipe David Borba Manana if (slot < btrfs_header_nritems(l)) 3409e902baacSDavid Sterba space_needed -= btrfs_leaf_free_space(l); 34105a4267caSFilipe David Borba Manana 34115a4267caSFilipe David Borba Manana wret = push_leaf_right(trans, root, path, space_needed, 34125a4267caSFilipe David Borba Manana space_needed, 0, 0); 341344871b1bSChris Mason if (wret < 0) 341444871b1bSChris Mason return wret; 341544871b1bSChris Mason if (wret) { 3416263d3995SFilipe Manana space_needed = data_size; 3417263d3995SFilipe Manana if (slot > 0) 3418e902baacSDavid Sterba space_needed -= btrfs_leaf_free_space(l); 34195a4267caSFilipe David Borba Manana wret = push_leaf_left(trans, root, path, space_needed, 34205a4267caSFilipe David Borba Manana space_needed, 0, (u32)-1); 342144871b1bSChris Mason if (wret < 0) 342244871b1bSChris Mason return wret; 342344871b1bSChris Mason } 342444871b1bSChris Mason l = path->nodes[0]; 342544871b1bSChris Mason 342644871b1bSChris Mason /* did the pushes work? */ 3427e902baacSDavid Sterba if (btrfs_leaf_free_space(l) >= data_size) 342844871b1bSChris Mason return 0; 342944871b1bSChris Mason } 343044871b1bSChris Mason 343144871b1bSChris Mason if (!path->nodes[1]) { 3432fdd99c72SLiu Bo ret = insert_new_root(trans, root, path, 1); 343344871b1bSChris Mason if (ret) 343444871b1bSChris Mason return ret; 343544871b1bSChris Mason } 343644871b1bSChris Mason again: 34375d4f98a2SYan Zheng split = 1; 343844871b1bSChris Mason l = path->nodes[0]; 343944871b1bSChris Mason slot = path->slots[0]; 344044871b1bSChris Mason nritems = btrfs_header_nritems(l); 344144871b1bSChris Mason mid = (nritems + 1) / 2; 344244871b1bSChris Mason 34435d4f98a2SYan Zheng if (mid <= slot) { 34445d4f98a2SYan Zheng if (nritems == 1 || 34455d4f98a2SYan Zheng leaf_space_used(l, mid, nritems - mid) + data_size > 34460b246afaSJeff Mahoney BTRFS_LEAF_DATA_SIZE(fs_info)) { 34475d4f98a2SYan Zheng if (slot >= nritems) { 34485d4f98a2SYan Zheng split = 0; 34495d4f98a2SYan Zheng } else { 34505d4f98a2SYan Zheng mid = slot; 34515d4f98a2SYan Zheng if (mid != nritems && 34525d4f98a2SYan Zheng leaf_space_used(l, mid, nritems - mid) + 34530b246afaSJeff Mahoney data_size > BTRFS_LEAF_DATA_SIZE(fs_info)) { 345499d8f83cSChris Mason if (data_size && !tried_avoid_double) 345599d8f83cSChris Mason goto push_for_double; 34565d4f98a2SYan Zheng split = 2; 34575d4f98a2SYan Zheng } 34585d4f98a2SYan Zheng } 34595d4f98a2SYan Zheng } 34605d4f98a2SYan Zheng } else { 34615d4f98a2SYan Zheng if (leaf_space_used(l, 0, mid) + data_size > 34620b246afaSJeff Mahoney BTRFS_LEAF_DATA_SIZE(fs_info)) { 34635d4f98a2SYan Zheng if (!extend && data_size && slot == 0) { 34645d4f98a2SYan Zheng split = 0; 34655d4f98a2SYan Zheng } else if ((extend || !data_size) && slot == 0) { 34665d4f98a2SYan Zheng mid = 1; 34675d4f98a2SYan Zheng } else { 34685d4f98a2SYan Zheng mid = slot; 34695d4f98a2SYan Zheng if (mid != nritems && 34705d4f98a2SYan Zheng leaf_space_used(l, mid, nritems - mid) + 34710b246afaSJeff Mahoney data_size > BTRFS_LEAF_DATA_SIZE(fs_info)) { 347299d8f83cSChris Mason if (data_size && !tried_avoid_double) 347399d8f83cSChris Mason goto push_for_double; 34745d4f98a2SYan Zheng split = 2; 34755d4f98a2SYan Zheng } 34765d4f98a2SYan Zheng } 34775d4f98a2SYan Zheng } 34785d4f98a2SYan Zheng } 34795d4f98a2SYan Zheng 34805d4f98a2SYan Zheng if (split == 0) 34815d4f98a2SYan Zheng btrfs_cpu_key_to_disk(&disk_key, ins_key); 34825d4f98a2SYan Zheng else 34835d4f98a2SYan Zheng btrfs_item_key(l, &disk_key, mid); 34845d4f98a2SYan Zheng 3485ca9d473aSJosef Bacik /* 3486ca9d473aSJosef Bacik * We have to about BTRFS_NESTING_NEW_ROOT here if we've done a double 3487ca9d473aSJosef Bacik * split, because we're only allowed to have MAX_LOCKDEP_SUBCLASSES 3488ca9d473aSJosef Bacik * subclasses, which is 8 at the time of this patch, and we've maxed it 3489ca9d473aSJosef Bacik * out. In the future we could add a 3490ca9d473aSJosef Bacik * BTRFS_NESTING_SPLIT_THE_SPLITTENING if we need to, but for now just 3491ca9d473aSJosef Bacik * use BTRFS_NESTING_NEW_ROOT. 3492ca9d473aSJosef Bacik */ 349379bd3712SFilipe Manana right = btrfs_alloc_tree_block(trans, root, 0, root->root_key.objectid, 349479bd3712SFilipe Manana &disk_key, 0, l->start, 0, 349579bd3712SFilipe Manana num_doubles ? BTRFS_NESTING_NEW_ROOT : 3496ca9d473aSJosef Bacik BTRFS_NESTING_SPLIT); 3497f0486c68SYan, Zheng if (IS_ERR(right)) 349844871b1bSChris Mason return PTR_ERR(right); 3499f0486c68SYan, Zheng 35000b246afaSJeff Mahoney root_add_used(root, fs_info->nodesize); 350144871b1bSChris Mason 35025d4f98a2SYan Zheng if (split == 0) { 350344871b1bSChris Mason if (mid <= slot) { 350444871b1bSChris Mason btrfs_set_header_nritems(right, 0); 35056ad3cf6dSDavid Sterba insert_ptr(trans, path, &disk_key, 35062ff7e61eSJeff Mahoney right->start, path->slots[1] + 1, 1); 350744871b1bSChris Mason btrfs_tree_unlock(path->nodes[0]); 350844871b1bSChris Mason free_extent_buffer(path->nodes[0]); 350944871b1bSChris Mason path->nodes[0] = right; 351044871b1bSChris Mason path->slots[0] = 0; 351144871b1bSChris Mason path->slots[1] += 1; 351244871b1bSChris Mason } else { 351344871b1bSChris Mason btrfs_set_header_nritems(right, 0); 35146ad3cf6dSDavid Sterba insert_ptr(trans, path, &disk_key, 35152ff7e61eSJeff Mahoney right->start, path->slots[1], 1); 351644871b1bSChris Mason btrfs_tree_unlock(path->nodes[0]); 351744871b1bSChris Mason free_extent_buffer(path->nodes[0]); 351844871b1bSChris Mason path->nodes[0] = right; 351944871b1bSChris Mason path->slots[0] = 0; 3520143bede5SJeff Mahoney if (path->slots[1] == 0) 3521b167fa91SNikolay Borisov fixup_low_keys(path, &disk_key, 1); 35225d4f98a2SYan Zheng } 3523196e0249SLiu Bo /* 3524196e0249SLiu Bo * We create a new leaf 'right' for the required ins_len and 3525196e0249SLiu Bo * we'll do btrfs_mark_buffer_dirty() on this leaf after copying 3526196e0249SLiu Bo * the content of ins_len to 'right'. 3527196e0249SLiu Bo */ 352844871b1bSChris Mason return ret; 352944871b1bSChris Mason } 353044871b1bSChris Mason 353194f94ad9SDavid Sterba copy_for_split(trans, path, l, right, slot, mid, nritems); 353244871b1bSChris Mason 35335d4f98a2SYan Zheng if (split == 2) { 3534cc0c5538SChris Mason BUG_ON(num_doubles != 0); 3535cc0c5538SChris Mason num_doubles++; 3536cc0c5538SChris Mason goto again; 35373326d1b0SChris Mason } 353844871b1bSChris Mason 3539143bede5SJeff Mahoney return 0; 354099d8f83cSChris Mason 354199d8f83cSChris Mason push_for_double: 354299d8f83cSChris Mason push_for_double_split(trans, root, path, data_size); 354399d8f83cSChris Mason tried_avoid_double = 1; 3544e902baacSDavid Sterba if (btrfs_leaf_free_space(path->nodes[0]) >= data_size) 354599d8f83cSChris Mason return 0; 354699d8f83cSChris Mason goto again; 3547be0e5c09SChris Mason } 3548be0e5c09SChris Mason 3549ad48fd75SYan, Zheng static noinline int setup_leaf_for_split(struct btrfs_trans_handle *trans, 3550ad48fd75SYan, Zheng struct btrfs_root *root, 3551ad48fd75SYan, Zheng struct btrfs_path *path, int ins_len) 3552ad48fd75SYan, Zheng { 3553ad48fd75SYan, Zheng struct btrfs_key key; 3554ad48fd75SYan, Zheng struct extent_buffer *leaf; 3555ad48fd75SYan, Zheng struct btrfs_file_extent_item *fi; 3556ad48fd75SYan, Zheng u64 extent_len = 0; 3557ad48fd75SYan, Zheng u32 item_size; 3558ad48fd75SYan, Zheng int ret; 3559ad48fd75SYan, Zheng 3560ad48fd75SYan, Zheng leaf = path->nodes[0]; 3561ad48fd75SYan, Zheng btrfs_item_key_to_cpu(leaf, &key, path->slots[0]); 3562ad48fd75SYan, Zheng 3563ad48fd75SYan, Zheng BUG_ON(key.type != BTRFS_EXTENT_DATA_KEY && 3564ad48fd75SYan, Zheng key.type != BTRFS_EXTENT_CSUM_KEY); 3565ad48fd75SYan, Zheng 3566e902baacSDavid Sterba if (btrfs_leaf_free_space(leaf) >= ins_len) 3567ad48fd75SYan, Zheng return 0; 3568ad48fd75SYan, Zheng 35693212fa14SJosef Bacik item_size = btrfs_item_size(leaf, path->slots[0]); 3570ad48fd75SYan, Zheng if (key.type == BTRFS_EXTENT_DATA_KEY) { 3571ad48fd75SYan, Zheng fi = btrfs_item_ptr(leaf, path->slots[0], 3572ad48fd75SYan, Zheng struct btrfs_file_extent_item); 3573ad48fd75SYan, Zheng extent_len = btrfs_file_extent_num_bytes(leaf, fi); 3574ad48fd75SYan, Zheng } 3575b3b4aa74SDavid Sterba btrfs_release_path(path); 3576ad48fd75SYan, Zheng 3577ad48fd75SYan, Zheng path->keep_locks = 1; 3578ad48fd75SYan, Zheng path->search_for_split = 1; 3579ad48fd75SYan, Zheng ret = btrfs_search_slot(trans, root, &key, path, 0, 1); 3580ad48fd75SYan, Zheng path->search_for_split = 0; 3581a8df6fe6SFilipe Manana if (ret > 0) 3582a8df6fe6SFilipe Manana ret = -EAGAIN; 3583ad48fd75SYan, Zheng if (ret < 0) 3584ad48fd75SYan, Zheng goto err; 3585ad48fd75SYan, Zheng 3586ad48fd75SYan, Zheng ret = -EAGAIN; 3587ad48fd75SYan, Zheng leaf = path->nodes[0]; 3588a8df6fe6SFilipe Manana /* if our item isn't there, return now */ 35893212fa14SJosef Bacik if (item_size != btrfs_item_size(leaf, path->slots[0])) 3590ad48fd75SYan, Zheng goto err; 3591ad48fd75SYan, Zheng 3592109f6aefSChris Mason /* the leaf has changed, it now has room. return now */ 3593e902baacSDavid Sterba if (btrfs_leaf_free_space(path->nodes[0]) >= ins_len) 3594109f6aefSChris Mason goto err; 3595109f6aefSChris Mason 3596ad48fd75SYan, Zheng if (key.type == BTRFS_EXTENT_DATA_KEY) { 3597ad48fd75SYan, Zheng fi = btrfs_item_ptr(leaf, path->slots[0], 3598ad48fd75SYan, Zheng struct btrfs_file_extent_item); 3599ad48fd75SYan, Zheng if (extent_len != btrfs_file_extent_num_bytes(leaf, fi)) 3600ad48fd75SYan, Zheng goto err; 3601ad48fd75SYan, Zheng } 3602ad48fd75SYan, Zheng 3603ad48fd75SYan, Zheng ret = split_leaf(trans, root, &key, path, ins_len, 1); 3604f0486c68SYan, Zheng if (ret) 3605f0486c68SYan, Zheng goto err; 3606ad48fd75SYan, Zheng 3607ad48fd75SYan, Zheng path->keep_locks = 0; 3608ad48fd75SYan, Zheng btrfs_unlock_up_safe(path, 1); 3609ad48fd75SYan, Zheng return 0; 3610ad48fd75SYan, Zheng err: 3611ad48fd75SYan, Zheng path->keep_locks = 0; 3612ad48fd75SYan, Zheng return ret; 3613ad48fd75SYan, Zheng } 3614ad48fd75SYan, Zheng 361525263cd7SDavid Sterba static noinline int split_item(struct btrfs_path *path, 3616310712b2SOmar Sandoval const struct btrfs_key *new_key, 3617459931ecSChris Mason unsigned long split_offset) 3618459931ecSChris Mason { 3619459931ecSChris Mason struct extent_buffer *leaf; 3620c91666b1SJosef Bacik int orig_slot, slot; 3621ad48fd75SYan, Zheng char *buf; 3622459931ecSChris Mason u32 nritems; 3623ad48fd75SYan, Zheng u32 item_size; 3624459931ecSChris Mason u32 orig_offset; 3625459931ecSChris Mason struct btrfs_disk_key disk_key; 3626459931ecSChris Mason 3627459931ecSChris Mason leaf = path->nodes[0]; 3628e902baacSDavid Sterba BUG_ON(btrfs_leaf_free_space(leaf) < sizeof(struct btrfs_item)); 3629b9473439SChris Mason 3630c91666b1SJosef Bacik orig_slot = path->slots[0]; 36313212fa14SJosef Bacik orig_offset = btrfs_item_offset(leaf, path->slots[0]); 36323212fa14SJosef Bacik item_size = btrfs_item_size(leaf, path->slots[0]); 3633459931ecSChris Mason 3634459931ecSChris Mason buf = kmalloc(item_size, GFP_NOFS); 3635ad48fd75SYan, Zheng if (!buf) 3636ad48fd75SYan, Zheng return -ENOMEM; 3637ad48fd75SYan, Zheng 3638459931ecSChris Mason read_extent_buffer(leaf, buf, btrfs_item_ptr_offset(leaf, 3639459931ecSChris Mason path->slots[0]), item_size); 3640ad48fd75SYan, Zheng 3641459931ecSChris Mason slot = path->slots[0] + 1; 3642459931ecSChris Mason nritems = btrfs_header_nritems(leaf); 3643459931ecSChris Mason if (slot != nritems) { 3644459931ecSChris Mason /* shift the items */ 3645459931ecSChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1), 3646459931ecSChris Mason btrfs_item_nr_offset(slot), 3647459931ecSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 3648459931ecSChris Mason } 3649459931ecSChris Mason 3650459931ecSChris Mason btrfs_cpu_key_to_disk(&disk_key, new_key); 3651459931ecSChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 3652459931ecSChris Mason 36533212fa14SJosef Bacik btrfs_set_item_offset(leaf, slot, orig_offset); 36543212fa14SJosef Bacik btrfs_set_item_size(leaf, slot, item_size - split_offset); 3655459931ecSChris Mason 36563212fa14SJosef Bacik btrfs_set_item_offset(leaf, orig_slot, 3657459931ecSChris Mason orig_offset + item_size - split_offset); 36583212fa14SJosef Bacik btrfs_set_item_size(leaf, orig_slot, split_offset); 3659459931ecSChris Mason 3660459931ecSChris Mason btrfs_set_header_nritems(leaf, nritems + 1); 3661459931ecSChris Mason 3662459931ecSChris Mason /* write the data for the start of the original item */ 3663459931ecSChris Mason write_extent_buffer(leaf, buf, 3664459931ecSChris Mason btrfs_item_ptr_offset(leaf, path->slots[0]), 3665459931ecSChris Mason split_offset); 3666459931ecSChris Mason 3667459931ecSChris Mason /* write the data for the new item */ 3668459931ecSChris Mason write_extent_buffer(leaf, buf + split_offset, 3669459931ecSChris Mason btrfs_item_ptr_offset(leaf, slot), 3670459931ecSChris Mason item_size - split_offset); 3671459931ecSChris Mason btrfs_mark_buffer_dirty(leaf); 3672459931ecSChris Mason 3673e902baacSDavid Sterba BUG_ON(btrfs_leaf_free_space(leaf) < 0); 3674459931ecSChris Mason kfree(buf); 3675ad48fd75SYan, Zheng return 0; 3676ad48fd75SYan, Zheng } 3677ad48fd75SYan, Zheng 3678ad48fd75SYan, Zheng /* 3679ad48fd75SYan, Zheng * This function splits a single item into two items, 3680ad48fd75SYan, Zheng * giving 'new_key' to the new item and splitting the 3681ad48fd75SYan, Zheng * old one at split_offset (from the start of the item). 3682ad48fd75SYan, Zheng * 3683ad48fd75SYan, Zheng * The path may be released by this operation. After 3684ad48fd75SYan, Zheng * the split, the path is pointing to the old item. The 3685ad48fd75SYan, Zheng * new item is going to be in the same node as the old one. 3686ad48fd75SYan, Zheng * 3687ad48fd75SYan, Zheng * Note, the item being split must be smaller enough to live alone on 3688ad48fd75SYan, Zheng * a tree block with room for one extra struct btrfs_item 3689ad48fd75SYan, Zheng * 3690ad48fd75SYan, Zheng * This allows us to split the item in place, keeping a lock on the 3691ad48fd75SYan, Zheng * leaf the entire time. 3692ad48fd75SYan, Zheng */ 3693ad48fd75SYan, Zheng int btrfs_split_item(struct btrfs_trans_handle *trans, 3694ad48fd75SYan, Zheng struct btrfs_root *root, 3695ad48fd75SYan, Zheng struct btrfs_path *path, 3696310712b2SOmar Sandoval const struct btrfs_key *new_key, 3697ad48fd75SYan, Zheng unsigned long split_offset) 3698ad48fd75SYan, Zheng { 3699ad48fd75SYan, Zheng int ret; 3700ad48fd75SYan, Zheng ret = setup_leaf_for_split(trans, root, path, 3701ad48fd75SYan, Zheng sizeof(struct btrfs_item)); 3702ad48fd75SYan, Zheng if (ret) 3703459931ecSChris Mason return ret; 3704ad48fd75SYan, Zheng 370525263cd7SDavid Sterba ret = split_item(path, new_key, split_offset); 3706ad48fd75SYan, Zheng return ret; 3707ad48fd75SYan, Zheng } 3708ad48fd75SYan, Zheng 3709ad48fd75SYan, Zheng /* 3710d352ac68SChris Mason * make the item pointed to by the path smaller. new_size indicates 3711d352ac68SChris Mason * how small to make it, and from_end tells us if we just chop bytes 3712d352ac68SChris Mason * off the end of the item or if we shift the item to chop bytes off 3713d352ac68SChris Mason * the front. 3714d352ac68SChris Mason */ 371578ac4f9eSDavid Sterba void btrfs_truncate_item(struct btrfs_path *path, u32 new_size, int from_end) 3716b18c6685SChris Mason { 3717b18c6685SChris Mason int slot; 37185f39d397SChris Mason struct extent_buffer *leaf; 3719b18c6685SChris Mason u32 nritems; 3720b18c6685SChris Mason unsigned int data_end; 3721b18c6685SChris Mason unsigned int old_data_start; 3722b18c6685SChris Mason unsigned int old_size; 3723b18c6685SChris Mason unsigned int size_diff; 3724b18c6685SChris Mason int i; 3725cfed81a0SChris Mason struct btrfs_map_token token; 3726cfed81a0SChris Mason 37275f39d397SChris Mason leaf = path->nodes[0]; 3728179e29e4SChris Mason slot = path->slots[0]; 3729179e29e4SChris Mason 37303212fa14SJosef Bacik old_size = btrfs_item_size(leaf, slot); 3731179e29e4SChris Mason if (old_size == new_size) 3732143bede5SJeff Mahoney return; 3733b18c6685SChris Mason 37345f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 37358f881e8cSDavid Sterba data_end = leaf_data_end(leaf); 3736b18c6685SChris Mason 37373212fa14SJosef Bacik old_data_start = btrfs_item_offset(leaf, slot); 3738179e29e4SChris Mason 3739b18c6685SChris Mason size_diff = old_size - new_size; 3740b18c6685SChris Mason 3741b18c6685SChris Mason BUG_ON(slot < 0); 3742b18c6685SChris Mason BUG_ON(slot >= nritems); 3743b18c6685SChris Mason 3744b18c6685SChris Mason /* 3745b18c6685SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 3746b18c6685SChris Mason */ 3747b18c6685SChris Mason /* first correct the data pointers */ 3748c82f823cSDavid Sterba btrfs_init_map_token(&token, leaf); 3749b18c6685SChris Mason for (i = slot; i < nritems; i++) { 37505f39d397SChris Mason u32 ioff; 3751db94535dSChris Mason 37523212fa14SJosef Bacik ioff = btrfs_token_item_offset(&token, i); 37533212fa14SJosef Bacik btrfs_set_token_item_offset(&token, i, ioff + size_diff); 3754b18c6685SChris Mason } 3755db94535dSChris Mason 3756b18c6685SChris Mason /* shift the data */ 3757179e29e4SChris Mason if (from_end) { 37583d9ec8c4SNikolay Borisov memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET + 37593d9ec8c4SNikolay Borisov data_end + size_diff, BTRFS_LEAF_DATA_OFFSET + 3760b18c6685SChris Mason data_end, old_data_start + new_size - data_end); 3761179e29e4SChris Mason } else { 3762179e29e4SChris Mason struct btrfs_disk_key disk_key; 3763179e29e4SChris Mason u64 offset; 3764179e29e4SChris Mason 3765179e29e4SChris Mason btrfs_item_key(leaf, &disk_key, slot); 3766179e29e4SChris Mason 3767179e29e4SChris Mason if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) { 3768179e29e4SChris Mason unsigned long ptr; 3769179e29e4SChris Mason struct btrfs_file_extent_item *fi; 3770179e29e4SChris Mason 3771179e29e4SChris Mason fi = btrfs_item_ptr(leaf, slot, 3772179e29e4SChris Mason struct btrfs_file_extent_item); 3773179e29e4SChris Mason fi = (struct btrfs_file_extent_item *)( 3774179e29e4SChris Mason (unsigned long)fi - size_diff); 3775179e29e4SChris Mason 3776179e29e4SChris Mason if (btrfs_file_extent_type(leaf, fi) == 3777179e29e4SChris Mason BTRFS_FILE_EXTENT_INLINE) { 3778179e29e4SChris Mason ptr = btrfs_item_ptr_offset(leaf, slot); 3779179e29e4SChris Mason memmove_extent_buffer(leaf, ptr, 3780179e29e4SChris Mason (unsigned long)fi, 37817ec20afbSDavid Sterba BTRFS_FILE_EXTENT_INLINE_DATA_START); 3782179e29e4SChris Mason } 3783179e29e4SChris Mason } 3784179e29e4SChris Mason 37853d9ec8c4SNikolay Borisov memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET + 37863d9ec8c4SNikolay Borisov data_end + size_diff, BTRFS_LEAF_DATA_OFFSET + 3787179e29e4SChris Mason data_end, old_data_start - data_end); 3788179e29e4SChris Mason 3789179e29e4SChris Mason offset = btrfs_disk_key_offset(&disk_key); 3790179e29e4SChris Mason btrfs_set_disk_key_offset(&disk_key, offset + size_diff); 3791179e29e4SChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 3792179e29e4SChris Mason if (slot == 0) 3793b167fa91SNikolay Borisov fixup_low_keys(path, &disk_key, 1); 3794179e29e4SChris Mason } 37955f39d397SChris Mason 37963212fa14SJosef Bacik btrfs_set_item_size(leaf, slot, new_size); 37975f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 3798b18c6685SChris Mason 3799e902baacSDavid Sterba if (btrfs_leaf_free_space(leaf) < 0) { 3800a4f78750SDavid Sterba btrfs_print_leaf(leaf); 3801b18c6685SChris Mason BUG(); 38025f39d397SChris Mason } 3803b18c6685SChris Mason } 3804b18c6685SChris Mason 3805d352ac68SChris Mason /* 38068f69dbd2SStefan Behrens * make the item pointed to by the path bigger, data_size is the added size. 3807d352ac68SChris Mason */ 3808c71dd880SDavid Sterba void btrfs_extend_item(struct btrfs_path *path, u32 data_size) 38096567e837SChris Mason { 38106567e837SChris Mason int slot; 38115f39d397SChris Mason struct extent_buffer *leaf; 38126567e837SChris Mason u32 nritems; 38136567e837SChris Mason unsigned int data_end; 38146567e837SChris Mason unsigned int old_data; 38156567e837SChris Mason unsigned int old_size; 38166567e837SChris Mason int i; 3817cfed81a0SChris Mason struct btrfs_map_token token; 3818cfed81a0SChris Mason 38195f39d397SChris Mason leaf = path->nodes[0]; 38206567e837SChris Mason 38215f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 38228f881e8cSDavid Sterba data_end = leaf_data_end(leaf); 38236567e837SChris Mason 3824e902baacSDavid Sterba if (btrfs_leaf_free_space(leaf) < data_size) { 3825a4f78750SDavid Sterba btrfs_print_leaf(leaf); 38266567e837SChris Mason BUG(); 38275f39d397SChris Mason } 38286567e837SChris Mason slot = path->slots[0]; 3829dc2e724eSJosef Bacik old_data = btrfs_item_data_end(leaf, slot); 38306567e837SChris Mason 38316567e837SChris Mason BUG_ON(slot < 0); 38323326d1b0SChris Mason if (slot >= nritems) { 3833a4f78750SDavid Sterba btrfs_print_leaf(leaf); 3834c71dd880SDavid Sterba btrfs_crit(leaf->fs_info, "slot %d too large, nritems %d", 3835d397712bSChris Mason slot, nritems); 3836290342f6SArnd Bergmann BUG(); 38373326d1b0SChris Mason } 38386567e837SChris Mason 38396567e837SChris Mason /* 38406567e837SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 38416567e837SChris Mason */ 38426567e837SChris Mason /* first correct the data pointers */ 3843c82f823cSDavid Sterba btrfs_init_map_token(&token, leaf); 38446567e837SChris Mason for (i = slot; i < nritems; i++) { 38455f39d397SChris Mason u32 ioff; 3846db94535dSChris Mason 38473212fa14SJosef Bacik ioff = btrfs_token_item_offset(&token, i); 38483212fa14SJosef Bacik btrfs_set_token_item_offset(&token, i, ioff - data_size); 38496567e837SChris Mason } 38505f39d397SChris Mason 38516567e837SChris Mason /* shift the data */ 38523d9ec8c4SNikolay Borisov memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET + 38533d9ec8c4SNikolay Borisov data_end - data_size, BTRFS_LEAF_DATA_OFFSET + 38546567e837SChris Mason data_end, old_data - data_end); 38555f39d397SChris Mason 38566567e837SChris Mason data_end = old_data; 38573212fa14SJosef Bacik old_size = btrfs_item_size(leaf, slot); 38583212fa14SJosef Bacik btrfs_set_item_size(leaf, slot, old_size + data_size); 38595f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 38606567e837SChris Mason 3861e902baacSDavid Sterba if (btrfs_leaf_free_space(leaf) < 0) { 3862a4f78750SDavid Sterba btrfs_print_leaf(leaf); 38636567e837SChris Mason BUG(); 38645f39d397SChris Mason } 38656567e837SChris Mason } 38666567e837SChris Mason 3867da9ffb24SNikolay Borisov /** 3868da9ffb24SNikolay Borisov * setup_items_for_insert - Helper called before inserting one or more items 3869da9ffb24SNikolay Borisov * to a leaf. Main purpose is to save stack depth by doing the bulk of the work 3870da9ffb24SNikolay Borisov * in a function that doesn't call btrfs_search_slot 3871da9ffb24SNikolay Borisov * 3872da9ffb24SNikolay Borisov * @root: root we are inserting items to 3873da9ffb24SNikolay Borisov * @path: points to the leaf/slot where we are going to insert new items 3874b7ef5f3aSFilipe Manana * @batch: information about the batch of items to insert 387574123bd7SChris Mason */ 3876f0641656SFilipe Manana static void setup_items_for_insert(struct btrfs_root *root, struct btrfs_path *path, 3877b7ef5f3aSFilipe Manana const struct btrfs_item_batch *batch) 3878be0e5c09SChris Mason { 38790b246afaSJeff Mahoney struct btrfs_fs_info *fs_info = root->fs_info; 38809c58309dSChris Mason int i; 38817518a238SChris Mason u32 nritems; 3882be0e5c09SChris Mason unsigned int data_end; 3883e2fa7227SChris Mason struct btrfs_disk_key disk_key; 388444871b1bSChris Mason struct extent_buffer *leaf; 388544871b1bSChris Mason int slot; 3886cfed81a0SChris Mason struct btrfs_map_token token; 3887fc0d82e1SNikolay Borisov u32 total_size; 3888fc0d82e1SNikolay Borisov 3889b7ef5f3aSFilipe Manana /* 3890b7ef5f3aSFilipe Manana * Before anything else, update keys in the parent and other ancestors 3891b7ef5f3aSFilipe Manana * if needed, then release the write locks on them, so that other tasks 3892b7ef5f3aSFilipe Manana * can use them while we modify the leaf. 3893b7ef5f3aSFilipe Manana */ 389424cdc847SFilipe Manana if (path->slots[0] == 0) { 3895b7ef5f3aSFilipe Manana btrfs_cpu_key_to_disk(&disk_key, &batch->keys[0]); 3896b167fa91SNikolay Borisov fixup_low_keys(path, &disk_key, 1); 389724cdc847SFilipe Manana } 389824cdc847SFilipe Manana btrfs_unlock_up_safe(path, 1); 389924cdc847SFilipe Manana 39005f39d397SChris Mason leaf = path->nodes[0]; 390144871b1bSChris Mason slot = path->slots[0]; 390274123bd7SChris Mason 39035f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 39048f881e8cSDavid Sterba data_end = leaf_data_end(leaf); 3905b7ef5f3aSFilipe Manana total_size = batch->total_data_size + (batch->nr * sizeof(struct btrfs_item)); 3906eb60ceacSChris Mason 3907e902baacSDavid Sterba if (btrfs_leaf_free_space(leaf) < total_size) { 3908a4f78750SDavid Sterba btrfs_print_leaf(leaf); 39090b246afaSJeff Mahoney btrfs_crit(fs_info, "not enough freespace need %u have %d", 3910e902baacSDavid Sterba total_size, btrfs_leaf_free_space(leaf)); 3911be0e5c09SChris Mason BUG(); 3912d4dbff95SChris Mason } 39135f39d397SChris Mason 3914c82f823cSDavid Sterba btrfs_init_map_token(&token, leaf); 3915be0e5c09SChris Mason if (slot != nritems) { 3916dc2e724eSJosef Bacik unsigned int old_data = btrfs_item_data_end(leaf, slot); 3917be0e5c09SChris Mason 39185f39d397SChris Mason if (old_data < data_end) { 3919a4f78750SDavid Sterba btrfs_print_leaf(leaf); 39207269ddd2SNikolay Borisov btrfs_crit(fs_info, 39217269ddd2SNikolay Borisov "item at slot %d with data offset %u beyond data end of leaf %u", 39225f39d397SChris Mason slot, old_data, data_end); 3923290342f6SArnd Bergmann BUG(); 39245f39d397SChris Mason } 3925be0e5c09SChris Mason /* 3926be0e5c09SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 3927be0e5c09SChris Mason */ 3928be0e5c09SChris Mason /* first correct the data pointers */ 39290783fcfcSChris Mason for (i = slot; i < nritems; i++) { 39305f39d397SChris Mason u32 ioff; 3931db94535dSChris Mason 39323212fa14SJosef Bacik ioff = btrfs_token_item_offset(&token, i); 39333212fa14SJosef Bacik btrfs_set_token_item_offset(&token, i, 3934b7ef5f3aSFilipe Manana ioff - batch->total_data_size); 39350783fcfcSChris Mason } 3936be0e5c09SChris Mason /* shift the items */ 3937b7ef5f3aSFilipe Manana memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + batch->nr), 39385f39d397SChris Mason btrfs_item_nr_offset(slot), 39390783fcfcSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 3940be0e5c09SChris Mason 3941be0e5c09SChris Mason /* shift the data */ 39423d9ec8c4SNikolay Borisov memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET + 3943b7ef5f3aSFilipe Manana data_end - batch->total_data_size, 3944b7ef5f3aSFilipe Manana BTRFS_LEAF_DATA_OFFSET + data_end, 3945b7ef5f3aSFilipe Manana old_data - data_end); 3946be0e5c09SChris Mason data_end = old_data; 3947be0e5c09SChris Mason } 39485f39d397SChris Mason 394962e2749eSChris Mason /* setup the item for the new data */ 3950b7ef5f3aSFilipe Manana for (i = 0; i < batch->nr; i++) { 3951b7ef5f3aSFilipe Manana btrfs_cpu_key_to_disk(&disk_key, &batch->keys[i]); 39529c58309dSChris Mason btrfs_set_item_key(leaf, &disk_key, slot + i); 3953b7ef5f3aSFilipe Manana data_end -= batch->data_sizes[i]; 39543212fa14SJosef Bacik btrfs_set_token_item_offset(&token, slot + i, data_end); 39553212fa14SJosef Bacik btrfs_set_token_item_size(&token, slot + i, batch->data_sizes[i]); 39569c58309dSChris Mason } 395744871b1bSChris Mason 3958b7ef5f3aSFilipe Manana btrfs_set_header_nritems(leaf, nritems + batch->nr); 3959b9473439SChris Mason btrfs_mark_buffer_dirty(leaf); 3960aa5d6bedSChris Mason 3961e902baacSDavid Sterba if (btrfs_leaf_free_space(leaf) < 0) { 3962a4f78750SDavid Sterba btrfs_print_leaf(leaf); 3963be0e5c09SChris Mason BUG(); 39645f39d397SChris Mason } 396544871b1bSChris Mason } 396644871b1bSChris Mason 396744871b1bSChris Mason /* 3968f0641656SFilipe Manana * Insert a new item into a leaf. 3969f0641656SFilipe Manana * 3970f0641656SFilipe Manana * @root: The root of the btree. 3971f0641656SFilipe Manana * @path: A path pointing to the target leaf and slot. 3972f0641656SFilipe Manana * @key: The key of the new item. 3973f0641656SFilipe Manana * @data_size: The size of the data associated with the new key. 3974f0641656SFilipe Manana */ 3975f0641656SFilipe Manana void btrfs_setup_item_for_insert(struct btrfs_root *root, 3976f0641656SFilipe Manana struct btrfs_path *path, 3977f0641656SFilipe Manana const struct btrfs_key *key, 3978f0641656SFilipe Manana u32 data_size) 3979f0641656SFilipe Manana { 3980f0641656SFilipe Manana struct btrfs_item_batch batch; 3981f0641656SFilipe Manana 3982f0641656SFilipe Manana batch.keys = key; 3983f0641656SFilipe Manana batch.data_sizes = &data_size; 3984f0641656SFilipe Manana batch.total_data_size = data_size; 3985f0641656SFilipe Manana batch.nr = 1; 3986f0641656SFilipe Manana 3987f0641656SFilipe Manana setup_items_for_insert(root, path, &batch); 3988f0641656SFilipe Manana } 3989f0641656SFilipe Manana 3990f0641656SFilipe Manana /* 399144871b1bSChris Mason * Given a key and some data, insert items into the tree. 399244871b1bSChris Mason * This does all the path init required, making room in the tree if needed. 399344871b1bSChris Mason */ 399444871b1bSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans, 399544871b1bSChris Mason struct btrfs_root *root, 399644871b1bSChris Mason struct btrfs_path *path, 3997b7ef5f3aSFilipe Manana const struct btrfs_item_batch *batch) 399844871b1bSChris Mason { 399944871b1bSChris Mason int ret = 0; 400044871b1bSChris Mason int slot; 4001b7ef5f3aSFilipe Manana u32 total_size; 400244871b1bSChris Mason 4003b7ef5f3aSFilipe Manana total_size = batch->total_data_size + (batch->nr * sizeof(struct btrfs_item)); 4004b7ef5f3aSFilipe Manana ret = btrfs_search_slot(trans, root, &batch->keys[0], path, total_size, 1); 400544871b1bSChris Mason if (ret == 0) 400644871b1bSChris Mason return -EEXIST; 400744871b1bSChris Mason if (ret < 0) 4008143bede5SJeff Mahoney return ret; 400944871b1bSChris Mason 401044871b1bSChris Mason slot = path->slots[0]; 401144871b1bSChris Mason BUG_ON(slot < 0); 401244871b1bSChris Mason 4013b7ef5f3aSFilipe Manana setup_items_for_insert(root, path, batch); 4014143bede5SJeff Mahoney return 0; 401562e2749eSChris Mason } 401662e2749eSChris Mason 401762e2749eSChris Mason /* 401862e2749eSChris Mason * Given a key and some data, insert an item into the tree. 401962e2749eSChris Mason * This does all the path init required, making room in the tree if needed. 402062e2749eSChris Mason */ 4021310712b2SOmar Sandoval int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root *root, 4022310712b2SOmar Sandoval const struct btrfs_key *cpu_key, void *data, 4023310712b2SOmar Sandoval u32 data_size) 402462e2749eSChris Mason { 402562e2749eSChris Mason int ret = 0; 40262c90e5d6SChris Mason struct btrfs_path *path; 40275f39d397SChris Mason struct extent_buffer *leaf; 40285f39d397SChris Mason unsigned long ptr; 402962e2749eSChris Mason 40302c90e5d6SChris Mason path = btrfs_alloc_path(); 4031db5b493aSTsutomu Itoh if (!path) 4032db5b493aSTsutomu Itoh return -ENOMEM; 40332c90e5d6SChris Mason ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size); 403462e2749eSChris Mason if (!ret) { 40355f39d397SChris Mason leaf = path->nodes[0]; 40365f39d397SChris Mason ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); 40375f39d397SChris Mason write_extent_buffer(leaf, data, ptr, data_size); 40385f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 403962e2749eSChris Mason } 40402c90e5d6SChris Mason btrfs_free_path(path); 4041aa5d6bedSChris Mason return ret; 4042be0e5c09SChris Mason } 4043be0e5c09SChris Mason 404474123bd7SChris Mason /* 4045f0641656SFilipe Manana * This function duplicates an item, giving 'new_key' to the new item. 4046f0641656SFilipe Manana * It guarantees both items live in the same tree leaf and the new item is 4047f0641656SFilipe Manana * contiguous with the original item. 4048f0641656SFilipe Manana * 4049f0641656SFilipe Manana * This allows us to split a file extent in place, keeping a lock on the leaf 4050f0641656SFilipe Manana * the entire time. 4051f0641656SFilipe Manana */ 4052f0641656SFilipe Manana int btrfs_duplicate_item(struct btrfs_trans_handle *trans, 4053f0641656SFilipe Manana struct btrfs_root *root, 4054f0641656SFilipe Manana struct btrfs_path *path, 4055f0641656SFilipe Manana const struct btrfs_key *new_key) 4056f0641656SFilipe Manana { 4057f0641656SFilipe Manana struct extent_buffer *leaf; 4058f0641656SFilipe Manana int ret; 4059f0641656SFilipe Manana u32 item_size; 4060f0641656SFilipe Manana 4061f0641656SFilipe Manana leaf = path->nodes[0]; 40623212fa14SJosef Bacik item_size = btrfs_item_size(leaf, path->slots[0]); 4063f0641656SFilipe Manana ret = setup_leaf_for_split(trans, root, path, 4064f0641656SFilipe Manana item_size + sizeof(struct btrfs_item)); 4065f0641656SFilipe Manana if (ret) 4066f0641656SFilipe Manana return ret; 4067f0641656SFilipe Manana 4068f0641656SFilipe Manana path->slots[0]++; 4069f0641656SFilipe Manana btrfs_setup_item_for_insert(root, path, new_key, item_size); 4070f0641656SFilipe Manana leaf = path->nodes[0]; 4071f0641656SFilipe Manana memcpy_extent_buffer(leaf, 4072f0641656SFilipe Manana btrfs_item_ptr_offset(leaf, path->slots[0]), 4073f0641656SFilipe Manana btrfs_item_ptr_offset(leaf, path->slots[0] - 1), 4074f0641656SFilipe Manana item_size); 4075f0641656SFilipe Manana return 0; 4076f0641656SFilipe Manana } 4077f0641656SFilipe Manana 4078f0641656SFilipe Manana /* 40795de08d7dSChris Mason * delete the pointer from a given node. 408074123bd7SChris Mason * 4081d352ac68SChris Mason * the tree should have been previously balanced so the deletion does not 4082d352ac68SChris Mason * empty a node. 408374123bd7SChris Mason */ 4084afe5fea7STsutomu Itoh static void del_ptr(struct btrfs_root *root, struct btrfs_path *path, 4085afe5fea7STsutomu Itoh int level, int slot) 4086be0e5c09SChris Mason { 40875f39d397SChris Mason struct extent_buffer *parent = path->nodes[level]; 40887518a238SChris Mason u32 nritems; 4089f3ea38daSJan Schmidt int ret; 4090be0e5c09SChris Mason 40915f39d397SChris Mason nritems = btrfs_header_nritems(parent); 4092be0e5c09SChris Mason if (slot != nritems - 1) { 4093bf1d3425SDavid Sterba if (level) { 4094f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_insert_move(parent, slot, 4095f3a84ccdSFilipe Manana slot + 1, nritems - slot - 1); 4096bf1d3425SDavid Sterba BUG_ON(ret < 0); 4097bf1d3425SDavid Sterba } 40985f39d397SChris Mason memmove_extent_buffer(parent, 40995f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 41005f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 4101d6025579SChris Mason sizeof(struct btrfs_key_ptr) * 4102d6025579SChris Mason (nritems - slot - 1)); 410357ba86c0SChris Mason } else if (level) { 4104f3a84ccdSFilipe Manana ret = btrfs_tree_mod_log_insert_key(parent, slot, 4105f3a84ccdSFilipe Manana BTRFS_MOD_LOG_KEY_REMOVE, GFP_NOFS); 410657ba86c0SChris Mason BUG_ON(ret < 0); 4107be0e5c09SChris Mason } 4108f3ea38daSJan Schmidt 41097518a238SChris Mason nritems--; 41105f39d397SChris Mason btrfs_set_header_nritems(parent, nritems); 41117518a238SChris Mason if (nritems == 0 && parent == root->node) { 41125f39d397SChris Mason BUG_ON(btrfs_header_level(root->node) != 1); 4113eb60ceacSChris Mason /* just turn the root into a leaf and break */ 41145f39d397SChris Mason btrfs_set_header_level(root->node, 0); 4115bb803951SChris Mason } else if (slot == 0) { 41165f39d397SChris Mason struct btrfs_disk_key disk_key; 41175f39d397SChris Mason 41185f39d397SChris Mason btrfs_node_key(parent, &disk_key, 0); 4119b167fa91SNikolay Borisov fixup_low_keys(path, &disk_key, level + 1); 4120be0e5c09SChris Mason } 4121d6025579SChris Mason btrfs_mark_buffer_dirty(parent); 4122be0e5c09SChris Mason } 4123be0e5c09SChris Mason 412474123bd7SChris Mason /* 4125323ac95bSChris Mason * a helper function to delete the leaf pointed to by path->slots[1] and 41265d4f98a2SYan Zheng * path->nodes[1]. 4127323ac95bSChris Mason * 4128323ac95bSChris Mason * This deletes the pointer in path->nodes[1] and frees the leaf 4129323ac95bSChris Mason * block extent. zero is returned if it all worked out, < 0 otherwise. 4130323ac95bSChris Mason * 4131323ac95bSChris Mason * The path must have already been setup for deleting the leaf, including 4132323ac95bSChris Mason * all the proper balancing. path->nodes[1] must be locked. 4133323ac95bSChris Mason */ 4134143bede5SJeff Mahoney static noinline void btrfs_del_leaf(struct btrfs_trans_handle *trans, 4135323ac95bSChris Mason struct btrfs_root *root, 41365d4f98a2SYan Zheng struct btrfs_path *path, 41375d4f98a2SYan Zheng struct extent_buffer *leaf) 4138323ac95bSChris Mason { 41395d4f98a2SYan Zheng WARN_ON(btrfs_header_generation(leaf) != trans->transid); 4140afe5fea7STsutomu Itoh del_ptr(root, path, 1, path->slots[1]); 4141323ac95bSChris Mason 41424d081c41SChris Mason /* 41434d081c41SChris Mason * btrfs_free_extent is expensive, we want to make sure we 41444d081c41SChris Mason * aren't holding any locks when we call it 41454d081c41SChris Mason */ 41464d081c41SChris Mason btrfs_unlock_up_safe(path, 0); 41474d081c41SChris Mason 4148f0486c68SYan, Zheng root_sub_used(root, leaf->len); 4149f0486c68SYan, Zheng 415067439dadSDavid Sterba atomic_inc(&leaf->refs); 41517a163608SFilipe Manana btrfs_free_tree_block(trans, btrfs_root_id(root), leaf, 0, 1); 41523083ee2eSJosef Bacik free_extent_buffer_stale(leaf); 4153323ac95bSChris Mason } 4154323ac95bSChris Mason /* 415574123bd7SChris Mason * delete the item at the leaf level in path. If that empties 415674123bd7SChris Mason * the leaf, remove it from the tree 415774123bd7SChris Mason */ 415885e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root, 415985e21bacSChris Mason struct btrfs_path *path, int slot, int nr) 4160be0e5c09SChris Mason { 41610b246afaSJeff Mahoney struct btrfs_fs_info *fs_info = root->fs_info; 41625f39d397SChris Mason struct extent_buffer *leaf; 4163aa5d6bedSChris Mason int ret = 0; 4164aa5d6bedSChris Mason int wret; 41657518a238SChris Mason u32 nritems; 4166be0e5c09SChris Mason 41675f39d397SChris Mason leaf = path->nodes[0]; 41685f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 4169be0e5c09SChris Mason 417085e21bacSChris Mason if (slot + nr != nritems) { 41710cae23b6SFilipe Manana const u32 last_off = btrfs_item_offset(leaf, slot + nr - 1); 41720cae23b6SFilipe Manana const int data_end = leaf_data_end(leaf); 4173c82f823cSDavid Sterba struct btrfs_map_token token; 41740cae23b6SFilipe Manana u32 dsize = 0; 41750cae23b6SFilipe Manana int i; 41760cae23b6SFilipe Manana 41770cae23b6SFilipe Manana for (i = 0; i < nr; i++) 41780cae23b6SFilipe Manana dsize += btrfs_item_size(leaf, slot + i); 41795f39d397SChris Mason 41803d9ec8c4SNikolay Borisov memmove_extent_buffer(leaf, BTRFS_LEAF_DATA_OFFSET + 4181d6025579SChris Mason data_end + dsize, 41823d9ec8c4SNikolay Borisov BTRFS_LEAF_DATA_OFFSET + data_end, 418385e21bacSChris Mason last_off - data_end); 41845f39d397SChris Mason 4185c82f823cSDavid Sterba btrfs_init_map_token(&token, leaf); 418685e21bacSChris Mason for (i = slot + nr; i < nritems; i++) { 41875f39d397SChris Mason u32 ioff; 4188db94535dSChris Mason 41893212fa14SJosef Bacik ioff = btrfs_token_item_offset(&token, i); 41903212fa14SJosef Bacik btrfs_set_token_item_offset(&token, i, ioff + dsize); 41910783fcfcSChris Mason } 4192db94535dSChris Mason 41935f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot), 419485e21bacSChris Mason btrfs_item_nr_offset(slot + nr), 41950783fcfcSChris Mason sizeof(struct btrfs_item) * 419685e21bacSChris Mason (nritems - slot - nr)); 4197be0e5c09SChris Mason } 419885e21bacSChris Mason btrfs_set_header_nritems(leaf, nritems - nr); 419985e21bacSChris Mason nritems -= nr; 42005f39d397SChris Mason 420174123bd7SChris Mason /* delete the leaf if we've emptied it */ 42027518a238SChris Mason if (nritems == 0) { 42035f39d397SChris Mason if (leaf == root->node) { 42045f39d397SChris Mason btrfs_set_header_level(leaf, 0); 42059a8dd150SChris Mason } else { 42066a884d7dSDavid Sterba btrfs_clean_tree_block(leaf); 4207143bede5SJeff Mahoney btrfs_del_leaf(trans, root, path, leaf); 42089a8dd150SChris Mason } 4209be0e5c09SChris Mason } else { 42107518a238SChris Mason int used = leaf_space_used(leaf, 0, nritems); 4211aa5d6bedSChris Mason if (slot == 0) { 42125f39d397SChris Mason struct btrfs_disk_key disk_key; 42135f39d397SChris Mason 42145f39d397SChris Mason btrfs_item_key(leaf, &disk_key, 0); 4215b167fa91SNikolay Borisov fixup_low_keys(path, &disk_key, 1); 4216aa5d6bedSChris Mason } 4217aa5d6bedSChris Mason 42187c4063d1SFilipe Manana /* 42197c4063d1SFilipe Manana * Try to delete the leaf if it is mostly empty. We do this by 42207c4063d1SFilipe Manana * trying to move all its items into its left and right neighbours. 42217c4063d1SFilipe Manana * If we can't move all the items, then we don't delete it - it's 42227c4063d1SFilipe Manana * not ideal, but future insertions might fill the leaf with more 42237c4063d1SFilipe Manana * items, or items from other leaves might be moved later into our 42247c4063d1SFilipe Manana * leaf due to deletions on those leaves. 42257c4063d1SFilipe Manana */ 42260b246afaSJeff Mahoney if (used < BTRFS_LEAF_DATA_SIZE(fs_info) / 3) { 42277c4063d1SFilipe Manana u32 min_push_space; 42287c4063d1SFilipe Manana 4229be0e5c09SChris Mason /* push_leaf_left fixes the path. 4230be0e5c09SChris Mason * make sure the path still points to our leaf 4231be0e5c09SChris Mason * for possible call to del_ptr below 4232be0e5c09SChris Mason */ 42334920c9acSChris Mason slot = path->slots[1]; 423467439dadSDavid Sterba atomic_inc(&leaf->refs); 42357c4063d1SFilipe Manana /* 42367c4063d1SFilipe Manana * We want to be able to at least push one item to the 42377c4063d1SFilipe Manana * left neighbour leaf, and that's the first item. 42387c4063d1SFilipe Manana */ 42397c4063d1SFilipe Manana min_push_space = sizeof(struct btrfs_item) + 42407c4063d1SFilipe Manana btrfs_item_size(leaf, 0); 42417c4063d1SFilipe Manana wret = push_leaf_left(trans, root, path, 0, 42427c4063d1SFilipe Manana min_push_space, 1, (u32)-1); 424354aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 4244aa5d6bedSChris Mason ret = wret; 42455f39d397SChris Mason 42465f39d397SChris Mason if (path->nodes[0] == leaf && 42475f39d397SChris Mason btrfs_header_nritems(leaf)) { 42487c4063d1SFilipe Manana /* 42497c4063d1SFilipe Manana * If we were not able to push all items from our 42507c4063d1SFilipe Manana * leaf to its left neighbour, then attempt to 42517c4063d1SFilipe Manana * either push all the remaining items to the 42527c4063d1SFilipe Manana * right neighbour or none. There's no advantage 42537c4063d1SFilipe Manana * in pushing only some items, instead of all, as 42547c4063d1SFilipe Manana * it's pointless to end up with a leaf having 42557c4063d1SFilipe Manana * too few items while the neighbours can be full 42567c4063d1SFilipe Manana * or nearly full. 42577c4063d1SFilipe Manana */ 42587c4063d1SFilipe Manana nritems = btrfs_header_nritems(leaf); 42597c4063d1SFilipe Manana min_push_space = leaf_space_used(leaf, 0, nritems); 42607c4063d1SFilipe Manana wret = push_leaf_right(trans, root, path, 0, 42617c4063d1SFilipe Manana min_push_space, 1, 0); 426254aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 4263aa5d6bedSChris Mason ret = wret; 4264aa5d6bedSChris Mason } 42655f39d397SChris Mason 42665f39d397SChris Mason if (btrfs_header_nritems(leaf) == 0) { 4267323ac95bSChris Mason path->slots[1] = slot; 4268143bede5SJeff Mahoney btrfs_del_leaf(trans, root, path, leaf); 42695f39d397SChris Mason free_extent_buffer(leaf); 4270143bede5SJeff Mahoney ret = 0; 42715de08d7dSChris Mason } else { 4272925baeddSChris Mason /* if we're still in the path, make sure 4273925baeddSChris Mason * we're dirty. Otherwise, one of the 4274925baeddSChris Mason * push_leaf functions must have already 4275925baeddSChris Mason * dirtied this buffer 4276925baeddSChris Mason */ 4277925baeddSChris Mason if (path->nodes[0] == leaf) 42785f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 42795f39d397SChris Mason free_extent_buffer(leaf); 4280be0e5c09SChris Mason } 4281d5719762SChris Mason } else { 42825f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 4283be0e5c09SChris Mason } 42849a8dd150SChris Mason } 4285aa5d6bedSChris Mason return ret; 42869a8dd150SChris Mason } 42879a8dd150SChris Mason 428897571fd0SChris Mason /* 4289925baeddSChris Mason * search the tree again to find a leaf with lesser keys 42907bb86316SChris Mason * returns 0 if it found something or 1 if there are no lesser leaves. 42917bb86316SChris Mason * returns < 0 on io errors. 4292d352ac68SChris Mason * 4293d352ac68SChris Mason * This may release the path, and so you may lose any locks held at the 4294d352ac68SChris Mason * time you call it. 42957bb86316SChris Mason */ 429616e7549fSJosef Bacik int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path) 42977bb86316SChris Mason { 4298925baeddSChris Mason struct btrfs_key key; 4299925baeddSChris Mason struct btrfs_disk_key found_key; 4300925baeddSChris Mason int ret; 43017bb86316SChris Mason 4302925baeddSChris Mason btrfs_item_key_to_cpu(path->nodes[0], &key, 0); 4303925baeddSChris Mason 4304e8b0d724SFilipe David Borba Manana if (key.offset > 0) { 4305925baeddSChris Mason key.offset--; 4306e8b0d724SFilipe David Borba Manana } else if (key.type > 0) { 4307925baeddSChris Mason key.type--; 4308e8b0d724SFilipe David Borba Manana key.offset = (u64)-1; 4309e8b0d724SFilipe David Borba Manana } else if (key.objectid > 0) { 4310925baeddSChris Mason key.objectid--; 4311e8b0d724SFilipe David Borba Manana key.type = (u8)-1; 4312e8b0d724SFilipe David Borba Manana key.offset = (u64)-1; 4313e8b0d724SFilipe David Borba Manana } else { 43147bb86316SChris Mason return 1; 4315e8b0d724SFilipe David Borba Manana } 43167bb86316SChris Mason 4317b3b4aa74SDavid Sterba btrfs_release_path(path); 4318925baeddSChris Mason ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); 4319925baeddSChris Mason if (ret < 0) 4320925baeddSChris Mason return ret; 4321925baeddSChris Mason btrfs_item_key(path->nodes[0], &found_key, 0); 4322925baeddSChris Mason ret = comp_keys(&found_key, &key); 4323337c6f68SFilipe Manana /* 4324337c6f68SFilipe Manana * We might have had an item with the previous key in the tree right 4325337c6f68SFilipe Manana * before we released our path. And after we released our path, that 4326337c6f68SFilipe Manana * item might have been pushed to the first slot (0) of the leaf we 4327337c6f68SFilipe Manana * were holding due to a tree balance. Alternatively, an item with the 4328337c6f68SFilipe Manana * previous key can exist as the only element of a leaf (big fat item). 4329337c6f68SFilipe Manana * Therefore account for these 2 cases, so that our callers (like 4330337c6f68SFilipe Manana * btrfs_previous_item) don't miss an existing item with a key matching 4331337c6f68SFilipe Manana * the previous key we computed above. 4332337c6f68SFilipe Manana */ 4333337c6f68SFilipe Manana if (ret <= 0) 43347bb86316SChris Mason return 0; 4335925baeddSChris Mason return 1; 43367bb86316SChris Mason } 43377bb86316SChris Mason 43383f157a2fSChris Mason /* 43393f157a2fSChris Mason * A helper function to walk down the tree starting at min_key, and looking 4340de78b51aSEric Sandeen * for nodes or leaves that are have a minimum transaction id. 4341de78b51aSEric Sandeen * This is used by the btree defrag code, and tree logging 43423f157a2fSChris Mason * 43433f157a2fSChris Mason * This does not cow, but it does stuff the starting key it finds back 43443f157a2fSChris Mason * into min_key, so you can call btrfs_search_slot with cow=1 on the 43453f157a2fSChris Mason * key and get a writable path. 43463f157a2fSChris Mason * 43473f157a2fSChris Mason * This honors path->lowest_level to prevent descent past a given level 43483f157a2fSChris Mason * of the tree. 43493f157a2fSChris Mason * 4350d352ac68SChris Mason * min_trans indicates the oldest transaction that you are interested 4351d352ac68SChris Mason * in walking through. Any nodes or leaves older than min_trans are 4352d352ac68SChris Mason * skipped over (without reading them). 4353d352ac68SChris Mason * 43543f157a2fSChris Mason * returns zero if something useful was found, < 0 on error and 1 if there 43553f157a2fSChris Mason * was nothing in the tree that matched the search criteria. 43563f157a2fSChris Mason */ 43573f157a2fSChris Mason int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key, 4358de78b51aSEric Sandeen struct btrfs_path *path, 43593f157a2fSChris Mason u64 min_trans) 43603f157a2fSChris Mason { 43613f157a2fSChris Mason struct extent_buffer *cur; 43623f157a2fSChris Mason struct btrfs_key found_key; 43633f157a2fSChris Mason int slot; 43649652480bSYan int sret; 43653f157a2fSChris Mason u32 nritems; 43663f157a2fSChris Mason int level; 43673f157a2fSChris Mason int ret = 1; 4368f98de9b9SFilipe Manana int keep_locks = path->keep_locks; 43693f157a2fSChris Mason 4370f98de9b9SFilipe Manana path->keep_locks = 1; 43713f157a2fSChris Mason again: 4372bd681513SChris Mason cur = btrfs_read_lock_root_node(root); 43733f157a2fSChris Mason level = btrfs_header_level(cur); 4374e02119d5SChris Mason WARN_ON(path->nodes[level]); 43753f157a2fSChris Mason path->nodes[level] = cur; 4376bd681513SChris Mason path->locks[level] = BTRFS_READ_LOCK; 43773f157a2fSChris Mason 43783f157a2fSChris Mason if (btrfs_header_generation(cur) < min_trans) { 43793f157a2fSChris Mason ret = 1; 43803f157a2fSChris Mason goto out; 43813f157a2fSChris Mason } 43823f157a2fSChris Mason while (1) { 43833f157a2fSChris Mason nritems = btrfs_header_nritems(cur); 43843f157a2fSChris Mason level = btrfs_header_level(cur); 4385e3b83361SQu Wenruo sret = btrfs_bin_search(cur, min_key, &slot); 4386cbca7d59SFilipe Manana if (sret < 0) { 4387cbca7d59SFilipe Manana ret = sret; 4388cbca7d59SFilipe Manana goto out; 4389cbca7d59SFilipe Manana } 43903f157a2fSChris Mason 4391323ac95bSChris Mason /* at the lowest level, we're done, setup the path and exit */ 4392323ac95bSChris Mason if (level == path->lowest_level) { 4393e02119d5SChris Mason if (slot >= nritems) 4394e02119d5SChris Mason goto find_next_key; 43953f157a2fSChris Mason ret = 0; 43963f157a2fSChris Mason path->slots[level] = slot; 43973f157a2fSChris Mason btrfs_item_key_to_cpu(cur, &found_key, slot); 43983f157a2fSChris Mason goto out; 43993f157a2fSChris Mason } 44009652480bSYan if (sret && slot > 0) 44019652480bSYan slot--; 44023f157a2fSChris Mason /* 4403de78b51aSEric Sandeen * check this node pointer against the min_trans parameters. 4404260db43cSRandy Dunlap * If it is too old, skip to the next one. 44053f157a2fSChris Mason */ 44063f157a2fSChris Mason while (slot < nritems) { 44073f157a2fSChris Mason u64 gen; 4408e02119d5SChris Mason 44093f157a2fSChris Mason gen = btrfs_node_ptr_generation(cur, slot); 44103f157a2fSChris Mason if (gen < min_trans) { 44113f157a2fSChris Mason slot++; 44123f157a2fSChris Mason continue; 44133f157a2fSChris Mason } 44143f157a2fSChris Mason break; 44153f157a2fSChris Mason } 4416e02119d5SChris Mason find_next_key: 44173f157a2fSChris Mason /* 44183f157a2fSChris Mason * we didn't find a candidate key in this node, walk forward 44193f157a2fSChris Mason * and find another one 44203f157a2fSChris Mason */ 44213f157a2fSChris Mason if (slot >= nritems) { 4422e02119d5SChris Mason path->slots[level] = slot; 4423e02119d5SChris Mason sret = btrfs_find_next_key(root, path, min_key, level, 4424de78b51aSEric Sandeen min_trans); 4425e02119d5SChris Mason if (sret == 0) { 4426b3b4aa74SDavid Sterba btrfs_release_path(path); 44273f157a2fSChris Mason goto again; 44283f157a2fSChris Mason } else { 44293f157a2fSChris Mason goto out; 44303f157a2fSChris Mason } 44313f157a2fSChris Mason } 44323f157a2fSChris Mason /* save our key for returning back */ 44333f157a2fSChris Mason btrfs_node_key_to_cpu(cur, &found_key, slot); 44343f157a2fSChris Mason path->slots[level] = slot; 44353f157a2fSChris Mason if (level == path->lowest_level) { 44363f157a2fSChris Mason ret = 0; 44373f157a2fSChris Mason goto out; 44383f157a2fSChris Mason } 44394b231ae4SDavid Sterba cur = btrfs_read_node_slot(cur, slot); 4440fb770ae4SLiu Bo if (IS_ERR(cur)) { 4441fb770ae4SLiu Bo ret = PTR_ERR(cur); 4442fb770ae4SLiu Bo goto out; 4443fb770ae4SLiu Bo } 44443f157a2fSChris Mason 4445bd681513SChris Mason btrfs_tree_read_lock(cur); 4446b4ce94deSChris Mason 4447bd681513SChris Mason path->locks[level - 1] = BTRFS_READ_LOCK; 44483f157a2fSChris Mason path->nodes[level - 1] = cur; 4449f7c79f30SChris Mason unlock_up(path, level, 1, 0, NULL); 44503f157a2fSChris Mason } 44513f157a2fSChris Mason out: 4452f98de9b9SFilipe Manana path->keep_locks = keep_locks; 4453f98de9b9SFilipe Manana if (ret == 0) { 4454f98de9b9SFilipe Manana btrfs_unlock_up_safe(path, path->lowest_level + 1); 4455f98de9b9SFilipe Manana memcpy(min_key, &found_key, sizeof(found_key)); 4456f98de9b9SFilipe Manana } 44573f157a2fSChris Mason return ret; 44583f157a2fSChris Mason } 44593f157a2fSChris Mason 44603f157a2fSChris Mason /* 44613f157a2fSChris Mason * this is similar to btrfs_next_leaf, but does not try to preserve 44623f157a2fSChris Mason * and fixup the path. It looks for and returns the next key in the 4463de78b51aSEric Sandeen * tree based on the current path and the min_trans parameters. 44643f157a2fSChris Mason * 44653f157a2fSChris Mason * 0 is returned if another key is found, < 0 if there are any errors 44663f157a2fSChris Mason * and 1 is returned if there are no higher keys in the tree 44673f157a2fSChris Mason * 44683f157a2fSChris Mason * path->keep_locks should be set to 1 on the search made before 44693f157a2fSChris Mason * calling this function. 44703f157a2fSChris Mason */ 4471e7a84565SChris Mason int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path, 4472de78b51aSEric Sandeen struct btrfs_key *key, int level, u64 min_trans) 4473e7a84565SChris Mason { 4474e7a84565SChris Mason int slot; 4475e7a84565SChris Mason struct extent_buffer *c; 4476e7a84565SChris Mason 44776a9fb468SJosef Bacik WARN_ON(!path->keep_locks && !path->skip_locking); 4478e7a84565SChris Mason while (level < BTRFS_MAX_LEVEL) { 4479e7a84565SChris Mason if (!path->nodes[level]) 4480e7a84565SChris Mason return 1; 4481e7a84565SChris Mason 4482e7a84565SChris Mason slot = path->slots[level] + 1; 4483e7a84565SChris Mason c = path->nodes[level]; 44843f157a2fSChris Mason next: 4485e7a84565SChris Mason if (slot >= btrfs_header_nritems(c)) { 448633c66f43SYan Zheng int ret; 448733c66f43SYan Zheng int orig_lowest; 448833c66f43SYan Zheng struct btrfs_key cur_key; 448933c66f43SYan Zheng if (level + 1 >= BTRFS_MAX_LEVEL || 449033c66f43SYan Zheng !path->nodes[level + 1]) 4491e7a84565SChris Mason return 1; 449233c66f43SYan Zheng 44936a9fb468SJosef Bacik if (path->locks[level + 1] || path->skip_locking) { 449433c66f43SYan Zheng level++; 4495e7a84565SChris Mason continue; 4496e7a84565SChris Mason } 449733c66f43SYan Zheng 449833c66f43SYan Zheng slot = btrfs_header_nritems(c) - 1; 449933c66f43SYan Zheng if (level == 0) 450033c66f43SYan Zheng btrfs_item_key_to_cpu(c, &cur_key, slot); 450133c66f43SYan Zheng else 450233c66f43SYan Zheng btrfs_node_key_to_cpu(c, &cur_key, slot); 450333c66f43SYan Zheng 450433c66f43SYan Zheng orig_lowest = path->lowest_level; 4505b3b4aa74SDavid Sterba btrfs_release_path(path); 450633c66f43SYan Zheng path->lowest_level = level; 450733c66f43SYan Zheng ret = btrfs_search_slot(NULL, root, &cur_key, path, 450833c66f43SYan Zheng 0, 0); 450933c66f43SYan Zheng path->lowest_level = orig_lowest; 451033c66f43SYan Zheng if (ret < 0) 451133c66f43SYan Zheng return ret; 451233c66f43SYan Zheng 451333c66f43SYan Zheng c = path->nodes[level]; 451433c66f43SYan Zheng slot = path->slots[level]; 451533c66f43SYan Zheng if (ret == 0) 451633c66f43SYan Zheng slot++; 451733c66f43SYan Zheng goto next; 451833c66f43SYan Zheng } 451933c66f43SYan Zheng 4520e7a84565SChris Mason if (level == 0) 4521e7a84565SChris Mason btrfs_item_key_to_cpu(c, key, slot); 45223f157a2fSChris Mason else { 45233f157a2fSChris Mason u64 gen = btrfs_node_ptr_generation(c, slot); 45243f157a2fSChris Mason 45253f157a2fSChris Mason if (gen < min_trans) { 45263f157a2fSChris Mason slot++; 45273f157a2fSChris Mason goto next; 45283f157a2fSChris Mason } 4529e7a84565SChris Mason btrfs_node_key_to_cpu(c, key, slot); 45303f157a2fSChris Mason } 4531e7a84565SChris Mason return 0; 4532e7a84565SChris Mason } 4533e7a84565SChris Mason return 1; 4534e7a84565SChris Mason } 4535e7a84565SChris Mason 45363d7806ecSJan Schmidt int btrfs_next_old_leaf(struct btrfs_root *root, struct btrfs_path *path, 45373d7806ecSJan Schmidt u64 time_seq) 45383d7806ecSJan Schmidt { 4539d97e63b6SChris Mason int slot; 45408e73f275SChris Mason int level; 45415f39d397SChris Mason struct extent_buffer *c; 45428e73f275SChris Mason struct extent_buffer *next; 4543d96b3424SFilipe Manana struct btrfs_fs_info *fs_info = root->fs_info; 4544925baeddSChris Mason struct btrfs_key key; 4545d96b3424SFilipe Manana bool need_commit_sem = false; 4546925baeddSChris Mason u32 nritems; 4547925baeddSChris Mason int ret; 45480e46318dSJosef Bacik int i; 4549925baeddSChris Mason 4550925baeddSChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 4551d397712bSChris Mason if (nritems == 0) 4552925baeddSChris Mason return 1; 4553925baeddSChris Mason 45548e73f275SChris Mason btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1); 45558e73f275SChris Mason again: 45568e73f275SChris Mason level = 1; 45578e73f275SChris Mason next = NULL; 4558b3b4aa74SDavid Sterba btrfs_release_path(path); 45598e73f275SChris Mason 4560a2135011SChris Mason path->keep_locks = 1; 45618e73f275SChris Mason 4562d96b3424SFilipe Manana if (time_seq) { 45633d7806ecSJan Schmidt ret = btrfs_search_old_slot(root, &key, path, time_seq); 4564d96b3424SFilipe Manana } else { 4565d96b3424SFilipe Manana if (path->need_commit_sem) { 4566d96b3424SFilipe Manana path->need_commit_sem = 0; 4567d96b3424SFilipe Manana need_commit_sem = true; 4568d96b3424SFilipe Manana down_read(&fs_info->commit_root_sem); 4569d96b3424SFilipe Manana } 4570925baeddSChris Mason ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); 4571d96b3424SFilipe Manana } 4572925baeddSChris Mason path->keep_locks = 0; 4573925baeddSChris Mason 4574925baeddSChris Mason if (ret < 0) 4575d96b3424SFilipe Manana goto done; 4576925baeddSChris Mason 4577a2135011SChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 4578168fd7d2SChris Mason /* 4579168fd7d2SChris Mason * by releasing the path above we dropped all our locks. A balance 4580168fd7d2SChris Mason * could have added more items next to the key that used to be 4581168fd7d2SChris Mason * at the very end of the block. So, check again here and 4582168fd7d2SChris Mason * advance the path if there are now more items available. 4583168fd7d2SChris Mason */ 4584a2135011SChris Mason if (nritems > 0 && path->slots[0] < nritems - 1) { 4585e457afecSYan Zheng if (ret == 0) 4586168fd7d2SChris Mason path->slots[0]++; 45878e73f275SChris Mason ret = 0; 4588925baeddSChris Mason goto done; 4589925baeddSChris Mason } 45900b43e04fSLiu Bo /* 45910b43e04fSLiu Bo * So the above check misses one case: 45920b43e04fSLiu Bo * - after releasing the path above, someone has removed the item that 45930b43e04fSLiu Bo * used to be at the very end of the block, and balance between leafs 45940b43e04fSLiu Bo * gets another one with bigger key.offset to replace it. 45950b43e04fSLiu Bo * 45960b43e04fSLiu Bo * This one should be returned as well, or we can get leaf corruption 45970b43e04fSLiu Bo * later(esp. in __btrfs_drop_extents()). 45980b43e04fSLiu Bo * 45990b43e04fSLiu Bo * And a bit more explanation about this check, 46000b43e04fSLiu Bo * with ret > 0, the key isn't found, the path points to the slot 46010b43e04fSLiu Bo * where it should be inserted, so the path->slots[0] item must be the 46020b43e04fSLiu Bo * bigger one. 46030b43e04fSLiu Bo */ 46040b43e04fSLiu Bo if (nritems > 0 && ret > 0 && path->slots[0] == nritems - 1) { 46050b43e04fSLiu Bo ret = 0; 46060b43e04fSLiu Bo goto done; 46070b43e04fSLiu Bo } 4608d97e63b6SChris Mason 4609234b63a0SChris Mason while (level < BTRFS_MAX_LEVEL) { 46108e73f275SChris Mason if (!path->nodes[level]) { 46118e73f275SChris Mason ret = 1; 46128e73f275SChris Mason goto done; 46138e73f275SChris Mason } 46145f39d397SChris Mason 4615d97e63b6SChris Mason slot = path->slots[level] + 1; 4616d97e63b6SChris Mason c = path->nodes[level]; 46175f39d397SChris Mason if (slot >= btrfs_header_nritems(c)) { 4618d97e63b6SChris Mason level++; 46198e73f275SChris Mason if (level == BTRFS_MAX_LEVEL) { 46208e73f275SChris Mason ret = 1; 46218e73f275SChris Mason goto done; 46228e73f275SChris Mason } 4623d97e63b6SChris Mason continue; 4624d97e63b6SChris Mason } 46255f39d397SChris Mason 46260e46318dSJosef Bacik 46270e46318dSJosef Bacik /* 46280e46318dSJosef Bacik * Our current level is where we're going to start from, and to 46290e46318dSJosef Bacik * make sure lockdep doesn't complain we need to drop our locks 46300e46318dSJosef Bacik * and nodes from 0 to our current level. 46310e46318dSJosef Bacik */ 46320e46318dSJosef Bacik for (i = 0; i < level; i++) { 46330e46318dSJosef Bacik if (path->locks[level]) { 46340e46318dSJosef Bacik btrfs_tree_read_unlock(path->nodes[i]); 46350e46318dSJosef Bacik path->locks[i] = 0; 46360e46318dSJosef Bacik } 46370e46318dSJosef Bacik free_extent_buffer(path->nodes[i]); 46380e46318dSJosef Bacik path->nodes[i] = NULL; 4639925baeddSChris Mason } 46405f39d397SChris Mason 46418e73f275SChris Mason next = c; 4642d07b8528SLiu Bo ret = read_block_for_search(root, path, &next, level, 4643cda79c54SDavid Sterba slot, &key); 46448e73f275SChris Mason if (ret == -EAGAIN) 46458e73f275SChris Mason goto again; 46465f39d397SChris Mason 464776a05b35SChris Mason if (ret < 0) { 4648b3b4aa74SDavid Sterba btrfs_release_path(path); 464976a05b35SChris Mason goto done; 465076a05b35SChris Mason } 465176a05b35SChris Mason 46525cd57b2cSChris Mason if (!path->skip_locking) { 4653bd681513SChris Mason ret = btrfs_try_tree_read_lock(next); 4654d42244a0SJan Schmidt if (!ret && time_seq) { 4655d42244a0SJan Schmidt /* 4656d42244a0SJan Schmidt * If we don't get the lock, we may be racing 4657d42244a0SJan Schmidt * with push_leaf_left, holding that lock while 4658d42244a0SJan Schmidt * itself waiting for the leaf we've currently 4659d42244a0SJan Schmidt * locked. To solve this situation, we give up 4660d42244a0SJan Schmidt * on our lock and cycle. 4661d42244a0SJan Schmidt */ 4662cf538830SJan Schmidt free_extent_buffer(next); 4663d42244a0SJan Schmidt btrfs_release_path(path); 4664d42244a0SJan Schmidt cond_resched(); 4665d42244a0SJan Schmidt goto again; 4666d42244a0SJan Schmidt } 46670e46318dSJosef Bacik if (!ret) 46680e46318dSJosef Bacik btrfs_tree_read_lock(next); 4669bd681513SChris Mason } 4670d97e63b6SChris Mason break; 4671d97e63b6SChris Mason } 4672d97e63b6SChris Mason path->slots[level] = slot; 4673d97e63b6SChris Mason while (1) { 4674d97e63b6SChris Mason level--; 4675d97e63b6SChris Mason path->nodes[level] = next; 4676d97e63b6SChris Mason path->slots[level] = 0; 4677a74a4b97SChris Mason if (!path->skip_locking) 4678ffeb03cfSJosef Bacik path->locks[level] = BTRFS_READ_LOCK; 4679d97e63b6SChris Mason if (!level) 4680d97e63b6SChris Mason break; 4681b4ce94deSChris Mason 4682d07b8528SLiu Bo ret = read_block_for_search(root, path, &next, level, 4683cda79c54SDavid Sterba 0, &key); 46848e73f275SChris Mason if (ret == -EAGAIN) 46858e73f275SChris Mason goto again; 46868e73f275SChris Mason 468776a05b35SChris Mason if (ret < 0) { 4688b3b4aa74SDavid Sterba btrfs_release_path(path); 468976a05b35SChris Mason goto done; 469076a05b35SChris Mason } 469176a05b35SChris Mason 4692ffeb03cfSJosef Bacik if (!path->skip_locking) 46930e46318dSJosef Bacik btrfs_tree_read_lock(next); 4694d97e63b6SChris Mason } 46958e73f275SChris Mason ret = 0; 4696925baeddSChris Mason done: 4697f7c79f30SChris Mason unlock_up(path, 0, 1, 0, NULL); 4698d96b3424SFilipe Manana if (need_commit_sem) { 4699d96b3424SFilipe Manana int ret2; 4700d96b3424SFilipe Manana 4701d96b3424SFilipe Manana path->need_commit_sem = 1; 4702d96b3424SFilipe Manana ret2 = finish_need_commit_sem_search(path); 4703d96b3424SFilipe Manana up_read(&fs_info->commit_root_sem); 4704d96b3424SFilipe Manana if (ret2) 4705d96b3424SFilipe Manana ret = ret2; 4706d96b3424SFilipe Manana } 47078e73f275SChris Mason 47088e73f275SChris Mason return ret; 4709d97e63b6SChris Mason } 47100b86a832SChris Mason 47113f157a2fSChris Mason /* 47123f157a2fSChris Mason * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps 47133f157a2fSChris Mason * searching until it gets past min_objectid or finds an item of 'type' 47143f157a2fSChris Mason * 47153f157a2fSChris Mason * returns 0 if something is found, 1 if nothing was found and < 0 on error 47163f157a2fSChris Mason */ 47170b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root, 47180b86a832SChris Mason struct btrfs_path *path, u64 min_objectid, 47190b86a832SChris Mason int type) 47200b86a832SChris Mason { 47210b86a832SChris Mason struct btrfs_key found_key; 47220b86a832SChris Mason struct extent_buffer *leaf; 4723e02119d5SChris Mason u32 nritems; 47240b86a832SChris Mason int ret; 47250b86a832SChris Mason 47260b86a832SChris Mason while (1) { 47270b86a832SChris Mason if (path->slots[0] == 0) { 47280b86a832SChris Mason ret = btrfs_prev_leaf(root, path); 47290b86a832SChris Mason if (ret != 0) 47300b86a832SChris Mason return ret; 47310b86a832SChris Mason } else { 47320b86a832SChris Mason path->slots[0]--; 47330b86a832SChris Mason } 47340b86a832SChris Mason leaf = path->nodes[0]; 4735e02119d5SChris Mason nritems = btrfs_header_nritems(leaf); 4736e02119d5SChris Mason if (nritems == 0) 4737e02119d5SChris Mason return 1; 4738e02119d5SChris Mason if (path->slots[0] == nritems) 4739e02119d5SChris Mason path->slots[0]--; 4740e02119d5SChris Mason 47410b86a832SChris Mason btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); 4742e02119d5SChris Mason if (found_key.objectid < min_objectid) 4743e02119d5SChris Mason break; 47440a4eefbbSYan Zheng if (found_key.type == type) 47450a4eefbbSYan Zheng return 0; 4746e02119d5SChris Mason if (found_key.objectid == min_objectid && 4747e02119d5SChris Mason found_key.type < type) 4748e02119d5SChris Mason break; 47490b86a832SChris Mason } 47500b86a832SChris Mason return 1; 47510b86a832SChris Mason } 4752ade2e0b3SWang Shilong 4753ade2e0b3SWang Shilong /* 4754ade2e0b3SWang Shilong * search in extent tree to find a previous Metadata/Data extent item with 4755ade2e0b3SWang Shilong * min objecitd. 4756ade2e0b3SWang Shilong * 4757ade2e0b3SWang Shilong * returns 0 if something is found, 1 if nothing was found and < 0 on error 4758ade2e0b3SWang Shilong */ 4759ade2e0b3SWang Shilong int btrfs_previous_extent_item(struct btrfs_root *root, 4760ade2e0b3SWang Shilong struct btrfs_path *path, u64 min_objectid) 4761ade2e0b3SWang Shilong { 4762ade2e0b3SWang Shilong struct btrfs_key found_key; 4763ade2e0b3SWang Shilong struct extent_buffer *leaf; 4764ade2e0b3SWang Shilong u32 nritems; 4765ade2e0b3SWang Shilong int ret; 4766ade2e0b3SWang Shilong 4767ade2e0b3SWang Shilong while (1) { 4768ade2e0b3SWang Shilong if (path->slots[0] == 0) { 4769ade2e0b3SWang Shilong ret = btrfs_prev_leaf(root, path); 4770ade2e0b3SWang Shilong if (ret != 0) 4771ade2e0b3SWang Shilong return ret; 4772ade2e0b3SWang Shilong } else { 4773ade2e0b3SWang Shilong path->slots[0]--; 4774ade2e0b3SWang Shilong } 4775ade2e0b3SWang Shilong leaf = path->nodes[0]; 4776ade2e0b3SWang Shilong nritems = btrfs_header_nritems(leaf); 4777ade2e0b3SWang Shilong if (nritems == 0) 4778ade2e0b3SWang Shilong return 1; 4779ade2e0b3SWang Shilong if (path->slots[0] == nritems) 4780ade2e0b3SWang Shilong path->slots[0]--; 4781ade2e0b3SWang Shilong 4782ade2e0b3SWang Shilong btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); 4783ade2e0b3SWang Shilong if (found_key.objectid < min_objectid) 4784ade2e0b3SWang Shilong break; 4785ade2e0b3SWang Shilong if (found_key.type == BTRFS_EXTENT_ITEM_KEY || 4786ade2e0b3SWang Shilong found_key.type == BTRFS_METADATA_ITEM_KEY) 4787ade2e0b3SWang Shilong return 0; 4788ade2e0b3SWang Shilong if (found_key.objectid == min_objectid && 4789ade2e0b3SWang Shilong found_key.type < BTRFS_EXTENT_ITEM_KEY) 4790ade2e0b3SWang Shilong break; 4791ade2e0b3SWang Shilong } 4792ade2e0b3SWang Shilong return 1; 4793ade2e0b3SWang Shilong } 4794