16cbd5570SChris Mason /* 2d352ac68SChris Mason * Copyright (C) 2007,2008 Oracle. All rights reserved. 36cbd5570SChris Mason * 46cbd5570SChris Mason * This program is free software; you can redistribute it and/or 56cbd5570SChris Mason * modify it under the terms of the GNU General Public 66cbd5570SChris Mason * License v2 as published by the Free Software Foundation. 76cbd5570SChris Mason * 86cbd5570SChris Mason * This program is distributed in the hope that it will be useful, 96cbd5570SChris Mason * but WITHOUT ANY WARRANTY; without even the implied warranty of 106cbd5570SChris Mason * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 116cbd5570SChris Mason * General Public License for more details. 126cbd5570SChris Mason * 136cbd5570SChris Mason * You should have received a copy of the GNU General Public 146cbd5570SChris Mason * License along with this program; if not, write to the 156cbd5570SChris Mason * Free Software Foundation, Inc., 59 Temple Place - Suite 330, 166cbd5570SChris Mason * Boston, MA 021110-1307, USA. 176cbd5570SChris Mason */ 186cbd5570SChris Mason 19a6b6e75eSChris Mason #include <linux/sched.h> 20eb60ceacSChris Mason #include "ctree.h" 21eb60ceacSChris Mason #include "disk-io.h" 227f5c1516SChris Mason #include "transaction.h" 235f39d397SChris Mason #include "print-tree.h" 24925baeddSChris Mason #include "locking.h" 259a8dd150SChris Mason 26e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 27e089f05cSChris Mason *root, struct btrfs_path *path, int level); 28e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 29d4dbff95SChris Mason *root, struct btrfs_key *ins_key, 30cc0c5538SChris Mason struct btrfs_path *path, int data_size, int extend); 315f39d397SChris Mason static int push_node_left(struct btrfs_trans_handle *trans, 325f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *dst, 33971a1f66SChris Mason struct extent_buffer *src, int empty); 345f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 355f39d397SChris Mason struct btrfs_root *root, 365f39d397SChris Mason struct extent_buffer *dst_buf, 375f39d397SChris Mason struct extent_buffer *src_buf); 38e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 39e089f05cSChris Mason struct btrfs_path *path, int level, int slot); 40d97e63b6SChris Mason 41df24a2b9SChris Mason inline void btrfs_init_path(struct btrfs_path *p) 42df24a2b9SChris Mason { 43df24a2b9SChris Mason memset(p, 0, sizeof(*p)); 44df24a2b9SChris Mason } 45df24a2b9SChris Mason 462c90e5d6SChris Mason struct btrfs_path *btrfs_alloc_path(void) 472c90e5d6SChris Mason { 48df24a2b9SChris Mason struct btrfs_path *path; 49df24a2b9SChris Mason path = kmem_cache_alloc(btrfs_path_cachep, GFP_NOFS); 502cc58cf2SChris Mason if (path) { 51df24a2b9SChris Mason btrfs_init_path(path); 522cc58cf2SChris Mason path->reada = 1; 532cc58cf2SChris Mason } 54df24a2b9SChris Mason return path; 552c90e5d6SChris Mason } 562c90e5d6SChris Mason 57d352ac68SChris Mason /* this also releases the path */ 582c90e5d6SChris Mason void btrfs_free_path(struct btrfs_path *p) 592c90e5d6SChris Mason { 60df24a2b9SChris Mason btrfs_release_path(NULL, p); 612c90e5d6SChris Mason kmem_cache_free(btrfs_path_cachep, p); 622c90e5d6SChris Mason } 632c90e5d6SChris Mason 64d352ac68SChris Mason /* 65d352ac68SChris Mason * path release drops references on the extent buffers in the path 66d352ac68SChris Mason * and it drops any locks held by this path 67d352ac68SChris Mason * 68d352ac68SChris Mason * It is safe to call this on paths that no locks or extent buffers held. 69d352ac68SChris Mason */ 70e02119d5SChris Mason void noinline btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p) 71eb60ceacSChris Mason { 72eb60ceacSChris Mason int i; 73a2135011SChris Mason 74234b63a0SChris Mason for (i = 0; i < BTRFS_MAX_LEVEL; i++) { 753f157a2fSChris Mason p->slots[i] = 0; 76eb60ceacSChris Mason if (!p->nodes[i]) 77925baeddSChris Mason continue; 78925baeddSChris Mason if (p->locks[i]) { 79925baeddSChris Mason btrfs_tree_unlock(p->nodes[i]); 80925baeddSChris Mason p->locks[i] = 0; 81925baeddSChris Mason } 825f39d397SChris Mason free_extent_buffer(p->nodes[i]); 833f157a2fSChris Mason p->nodes[i] = NULL; 84eb60ceacSChris Mason } 85eb60ceacSChris Mason } 86eb60ceacSChris Mason 87d352ac68SChris Mason /* 88d352ac68SChris Mason * safely gets a reference on the root node of a tree. A lock 89d352ac68SChris Mason * is not taken, so a concurrent writer may put a different node 90d352ac68SChris Mason * at the root of the tree. See btrfs_lock_root_node for the 91d352ac68SChris Mason * looping required. 92d352ac68SChris Mason * 93d352ac68SChris Mason * The extent buffer returned by this has a reference taken, so 94d352ac68SChris Mason * it won't disappear. It may stop being the root of the tree 95d352ac68SChris Mason * at any time because there are no locks held. 96d352ac68SChris Mason */ 97925baeddSChris Mason struct extent_buffer *btrfs_root_node(struct btrfs_root *root) 98925baeddSChris Mason { 99925baeddSChris Mason struct extent_buffer *eb; 100925baeddSChris Mason spin_lock(&root->node_lock); 101925baeddSChris Mason eb = root->node; 102925baeddSChris Mason extent_buffer_get(eb); 103925baeddSChris Mason spin_unlock(&root->node_lock); 104925baeddSChris Mason return eb; 105925baeddSChris Mason } 106925baeddSChris Mason 107d352ac68SChris Mason /* loop around taking references on and locking the root node of the 108d352ac68SChris Mason * tree until you end up with a lock on the root. A locked buffer 109d352ac68SChris Mason * is returned, with a reference held. 110d352ac68SChris Mason */ 111925baeddSChris Mason struct extent_buffer *btrfs_lock_root_node(struct btrfs_root *root) 112925baeddSChris Mason { 113925baeddSChris Mason struct extent_buffer *eb; 114925baeddSChris Mason 115925baeddSChris Mason while(1) { 116925baeddSChris Mason eb = btrfs_root_node(root); 117925baeddSChris Mason btrfs_tree_lock(eb); 118925baeddSChris Mason 119925baeddSChris Mason spin_lock(&root->node_lock); 120925baeddSChris Mason if (eb == root->node) { 121925baeddSChris Mason spin_unlock(&root->node_lock); 122925baeddSChris Mason break; 123925baeddSChris Mason } 124925baeddSChris Mason spin_unlock(&root->node_lock); 125925baeddSChris Mason 126925baeddSChris Mason btrfs_tree_unlock(eb); 127925baeddSChris Mason free_extent_buffer(eb); 128925baeddSChris Mason } 129925baeddSChris Mason return eb; 130925baeddSChris Mason } 131925baeddSChris Mason 132d352ac68SChris Mason /* cowonly root (everything not a reference counted cow subvolume), just get 133d352ac68SChris Mason * put onto a simple dirty list. transaction.c walks this to make sure they 134d352ac68SChris Mason * get properly updated on disk. 135d352ac68SChris Mason */ 1360b86a832SChris Mason static void add_root_to_dirty_list(struct btrfs_root *root) 1370b86a832SChris Mason { 1380b86a832SChris Mason if (root->track_dirty && list_empty(&root->dirty_list)) { 1390b86a832SChris Mason list_add(&root->dirty_list, 1400b86a832SChris Mason &root->fs_info->dirty_cowonly_roots); 1410b86a832SChris Mason } 1420b86a832SChris Mason } 1430b86a832SChris Mason 144d352ac68SChris Mason /* 145d352ac68SChris Mason * used by snapshot creation to make a copy of a root for a tree with 146d352ac68SChris Mason * a given objectid. The buffer with the new root node is returned in 147d352ac68SChris Mason * cow_ret, and this func returns zero on success or a negative error code. 148d352ac68SChris Mason */ 149be20aa9dSChris Mason int btrfs_copy_root(struct btrfs_trans_handle *trans, 150be20aa9dSChris Mason struct btrfs_root *root, 151be20aa9dSChris Mason struct extent_buffer *buf, 152be20aa9dSChris Mason struct extent_buffer **cow_ret, u64 new_root_objectid) 153be20aa9dSChris Mason { 154be20aa9dSChris Mason struct extent_buffer *cow; 155be20aa9dSChris Mason u32 nritems; 156be20aa9dSChris Mason int ret = 0; 157be20aa9dSChris Mason int level; 1584aec2b52SChris Mason struct btrfs_root *new_root; 159be20aa9dSChris Mason 1604aec2b52SChris Mason new_root = kmalloc(sizeof(*new_root), GFP_NOFS); 1614aec2b52SChris Mason if (!new_root) 1624aec2b52SChris Mason return -ENOMEM; 1634aec2b52SChris Mason 1644aec2b52SChris Mason memcpy(new_root, root, sizeof(*new_root)); 1654aec2b52SChris Mason new_root->root_key.objectid = new_root_objectid; 166be20aa9dSChris Mason 167be20aa9dSChris Mason WARN_ON(root->ref_cows && trans->transid != 168be20aa9dSChris Mason root->fs_info->running_transaction->transid); 169be20aa9dSChris Mason WARN_ON(root->ref_cows && trans->transid != root->last_trans); 170be20aa9dSChris Mason 171be20aa9dSChris Mason level = btrfs_header_level(buf); 172be20aa9dSChris Mason nritems = btrfs_header_nritems(buf); 17331840ae1SZheng Yan 17431840ae1SZheng Yan cow = btrfs_alloc_free_block(trans, new_root, buf->len, 0, 17531840ae1SZheng Yan new_root_objectid, trans->transid, 176be20aa9dSChris Mason level, buf->start, 0); 1774aec2b52SChris Mason if (IS_ERR(cow)) { 1784aec2b52SChris Mason kfree(new_root); 179be20aa9dSChris Mason return PTR_ERR(cow); 1804aec2b52SChris Mason } 181be20aa9dSChris Mason 182be20aa9dSChris Mason copy_extent_buffer(cow, buf, 0, 0, cow->len); 183be20aa9dSChris Mason btrfs_set_header_bytenr(cow, cow->start); 184be20aa9dSChris Mason btrfs_set_header_generation(cow, trans->transid); 185be20aa9dSChris Mason btrfs_set_header_owner(cow, new_root_objectid); 18663b10fc4SChris Mason btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN); 187be20aa9dSChris Mason 1882b82032cSYan Zheng write_extent_buffer(cow, root->fs_info->fsid, 1892b82032cSYan Zheng (unsigned long)btrfs_header_fsid(cow), 1902b82032cSYan Zheng BTRFS_FSID_SIZE); 1912b82032cSYan Zheng 192be20aa9dSChris Mason WARN_ON(btrfs_header_generation(buf) > trans->transid); 19331840ae1SZheng Yan ret = btrfs_inc_ref(trans, new_root, buf, cow, NULL); 1944aec2b52SChris Mason kfree(new_root); 1954aec2b52SChris Mason 196be20aa9dSChris Mason if (ret) 197be20aa9dSChris Mason return ret; 198be20aa9dSChris Mason 199be20aa9dSChris Mason btrfs_mark_buffer_dirty(cow); 200be20aa9dSChris Mason *cow_ret = cow; 201be20aa9dSChris Mason return 0; 202be20aa9dSChris Mason } 203be20aa9dSChris Mason 204d352ac68SChris Mason /* 205d352ac68SChris Mason * does the dirty work in cow of a single block. The parent block 206d352ac68SChris Mason * (if supplied) is updated to point to the new cow copy. The new 207d352ac68SChris Mason * buffer is marked dirty and returned locked. If you modify the block 208d352ac68SChris Mason * it needs to be marked dirty again. 209d352ac68SChris Mason * 210d352ac68SChris Mason * search_start -- an allocation hint for the new block 211d352ac68SChris Mason * 212d352ac68SChris Mason * empty_size -- a hint that you plan on doing more cow. This is the size in bytes 213d352ac68SChris Mason * the allocator should try to find free next to the block it returns. This is 214d352ac68SChris Mason * just a hint and may be ignored by the allocator. 215d352ac68SChris Mason * 216d352ac68SChris Mason * prealloc_dest -- if you have already reserved a destination for the cow, 217d352ac68SChris Mason * this uses that block instead of allocating a new one. btrfs_alloc_reserved_extent 218d352ac68SChris Mason * is used to finish the allocation. 219d352ac68SChris Mason */ 220b2950863SChristoph Hellwig static int noinline __btrfs_cow_block(struct btrfs_trans_handle *trans, 2215f39d397SChris Mason struct btrfs_root *root, 2225f39d397SChris Mason struct extent_buffer *buf, 2235f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 2245f39d397SChris Mason struct extent_buffer **cow_ret, 22565b51a00SChris Mason u64 search_start, u64 empty_size, 22665b51a00SChris Mason u64 prealloc_dest) 2276702ed49SChris Mason { 22831840ae1SZheng Yan u64 parent_start; 2295f39d397SChris Mason struct extent_buffer *cow; 2307bb86316SChris Mason u32 nritems; 2316702ed49SChris Mason int ret = 0; 2327bb86316SChris Mason int level; 233925baeddSChris Mason int unlock_orig = 0; 2346702ed49SChris Mason 235925baeddSChris Mason if (*cow_ret == buf) 236925baeddSChris Mason unlock_orig = 1; 237925baeddSChris Mason 238925baeddSChris Mason WARN_ON(!btrfs_tree_locked(buf)); 239925baeddSChris Mason 24031840ae1SZheng Yan if (parent) 24131840ae1SZheng Yan parent_start = parent->start; 24231840ae1SZheng Yan else 24331840ae1SZheng Yan parent_start = 0; 24431840ae1SZheng Yan 2457bb86316SChris Mason WARN_ON(root->ref_cows && trans->transid != 2467bb86316SChris Mason root->fs_info->running_transaction->transid); 2476702ed49SChris Mason WARN_ON(root->ref_cows && trans->transid != root->last_trans); 2485f39d397SChris Mason 2497bb86316SChris Mason level = btrfs_header_level(buf); 2507bb86316SChris Mason nritems = btrfs_header_nritems(buf); 25131840ae1SZheng Yan 25265b51a00SChris Mason if (prealloc_dest) { 25365b51a00SChris Mason struct btrfs_key ins; 25465b51a00SChris Mason 25565b51a00SChris Mason ins.objectid = prealloc_dest; 25665b51a00SChris Mason ins.offset = buf->len; 25765b51a00SChris Mason ins.type = BTRFS_EXTENT_ITEM_KEY; 25865b51a00SChris Mason 25931840ae1SZheng Yan ret = btrfs_alloc_reserved_extent(trans, root, parent_start, 26065b51a00SChris Mason root->root_key.objectid, 2613bb1a1bcSYan Zheng trans->transid, level, &ins); 26265b51a00SChris Mason BUG_ON(ret); 26365b51a00SChris Mason cow = btrfs_init_new_buffer(trans, root, prealloc_dest, 26465b51a00SChris Mason buf->len); 26565b51a00SChris Mason } else { 266925baeddSChris Mason cow = btrfs_alloc_free_block(trans, root, buf->len, 26731840ae1SZheng Yan parent_start, 2687bb86316SChris Mason root->root_key.objectid, 26931840ae1SZheng Yan trans->transid, level, 27031840ae1SZheng Yan search_start, empty_size); 27165b51a00SChris Mason } 2726702ed49SChris Mason if (IS_ERR(cow)) 2736702ed49SChris Mason return PTR_ERR(cow); 2746702ed49SChris Mason 2755f39d397SChris Mason copy_extent_buffer(cow, buf, 0, 0, cow->len); 276db94535dSChris Mason btrfs_set_header_bytenr(cow, cow->start); 2775f39d397SChris Mason btrfs_set_header_generation(cow, trans->transid); 2785f39d397SChris Mason btrfs_set_header_owner(cow, root->root_key.objectid); 27963b10fc4SChris Mason btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN); 2806702ed49SChris Mason 2812b82032cSYan Zheng write_extent_buffer(cow, root->fs_info->fsid, 2822b82032cSYan Zheng (unsigned long)btrfs_header_fsid(cow), 2832b82032cSYan Zheng BTRFS_FSID_SIZE); 2842b82032cSYan Zheng 2855f39d397SChris Mason WARN_ON(btrfs_header_generation(buf) > trans->transid); 2865f39d397SChris Mason if (btrfs_header_generation(buf) != trans->transid) { 28731840ae1SZheng Yan u32 nr_extents; 28831840ae1SZheng Yan ret = btrfs_inc_ref(trans, root, buf, cow, &nr_extents); 2896702ed49SChris Mason if (ret) 2906702ed49SChris Mason return ret; 29131840ae1SZheng Yan 29231840ae1SZheng Yan ret = btrfs_cache_ref(trans, root, buf, nr_extents); 29331840ae1SZheng Yan WARN_ON(ret); 2941a40e23bSZheng Yan } else if (btrfs_header_owner(buf) == BTRFS_TREE_RELOC_OBJECTID) { 2951a40e23bSZheng Yan /* 2961a40e23bSZheng Yan * There are only two places that can drop reference to 2971a40e23bSZheng Yan * tree blocks owned by living reloc trees, one is here, 298f82d02d9SYan Zheng * the other place is btrfs_drop_subtree. In both places, 2991a40e23bSZheng Yan * we check reference count while tree block is locked. 3001a40e23bSZheng Yan * Furthermore, if reference count is one, it won't get 3011a40e23bSZheng Yan * increased by someone else. 3021a40e23bSZheng Yan */ 3031a40e23bSZheng Yan u32 refs; 3041a40e23bSZheng Yan ret = btrfs_lookup_extent_ref(trans, root, buf->start, 3051a40e23bSZheng Yan buf->len, &refs); 3061a40e23bSZheng Yan BUG_ON(ret); 3071a40e23bSZheng Yan if (refs == 1) { 3081a40e23bSZheng Yan ret = btrfs_update_ref(trans, root, buf, cow, 3091a40e23bSZheng Yan 0, nritems); 3101a40e23bSZheng Yan clean_tree_block(trans, root, buf); 3111a40e23bSZheng Yan } else { 3121a40e23bSZheng Yan ret = btrfs_inc_ref(trans, root, buf, cow, NULL); 3131a40e23bSZheng Yan } 3141a40e23bSZheng Yan BUG_ON(ret); 3156702ed49SChris Mason } else { 31631840ae1SZheng Yan ret = btrfs_update_ref(trans, root, buf, cow, 0, nritems); 31731840ae1SZheng Yan if (ret) 31831840ae1SZheng Yan return ret; 3196702ed49SChris Mason clean_tree_block(trans, root, buf); 3206702ed49SChris Mason } 3216702ed49SChris Mason 3221a40e23bSZheng Yan if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) { 3231a40e23bSZheng Yan ret = btrfs_reloc_tree_cache_ref(trans, root, cow, buf->start); 3241a40e23bSZheng Yan WARN_ON(ret); 3251a40e23bSZheng Yan } 3261a40e23bSZheng Yan 3276702ed49SChris Mason if (buf == root->node) { 328925baeddSChris Mason WARN_ON(parent && parent != buf); 329925baeddSChris Mason 330925baeddSChris Mason spin_lock(&root->node_lock); 3316702ed49SChris Mason root->node = cow; 3325f39d397SChris Mason extent_buffer_get(cow); 333925baeddSChris Mason spin_unlock(&root->node_lock); 334925baeddSChris Mason 3356702ed49SChris Mason if (buf != root->commit_root) { 336db94535dSChris Mason btrfs_free_extent(trans, root, buf->start, 33731840ae1SZheng Yan buf->len, buf->start, 33831840ae1SZheng Yan root->root_key.objectid, 33931840ae1SZheng Yan btrfs_header_generation(buf), 3403bb1a1bcSYan Zheng level, 1); 3416702ed49SChris Mason } 3425f39d397SChris Mason free_extent_buffer(buf); 3430b86a832SChris Mason add_root_to_dirty_list(root); 3446702ed49SChris Mason } else { 3455f39d397SChris Mason btrfs_set_node_blockptr(parent, parent_slot, 346db94535dSChris Mason cow->start); 34774493f7aSChris Mason WARN_ON(trans->transid == 0); 34874493f7aSChris Mason btrfs_set_node_ptr_generation(parent, parent_slot, 34974493f7aSChris Mason trans->transid); 3506702ed49SChris Mason btrfs_mark_buffer_dirty(parent); 3515f39d397SChris Mason WARN_ON(btrfs_header_generation(parent) != trans->transid); 3527bb86316SChris Mason btrfs_free_extent(trans, root, buf->start, buf->len, 35331840ae1SZheng Yan parent_start, btrfs_header_owner(parent), 3543bb1a1bcSYan Zheng btrfs_header_generation(parent), level, 1); 3556702ed49SChris Mason } 356925baeddSChris Mason if (unlock_orig) 357925baeddSChris Mason btrfs_tree_unlock(buf); 3585f39d397SChris Mason free_extent_buffer(buf); 3596702ed49SChris Mason btrfs_mark_buffer_dirty(cow); 3606702ed49SChris Mason *cow_ret = cow; 3616702ed49SChris Mason return 0; 3626702ed49SChris Mason } 3636702ed49SChris Mason 364d352ac68SChris Mason /* 365d352ac68SChris Mason * cows a single block, see __btrfs_cow_block for the real work. 366d352ac68SChris Mason * This version of it has extra checks so that a block isn't cow'd more than 367d352ac68SChris Mason * once per transaction, as long as it hasn't been written yet 368d352ac68SChris Mason */ 369e02119d5SChris Mason int noinline btrfs_cow_block(struct btrfs_trans_handle *trans, 3705f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *buf, 3715f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 37265b51a00SChris Mason struct extent_buffer **cow_ret, u64 prealloc_dest) 37302217ed2SChris Mason { 3746702ed49SChris Mason u64 search_start; 375f510cfecSChris Mason int ret; 376dc17ff8fSChris Mason 377ccd467d6SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 378ccd467d6SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 379ccd467d6SChris Mason root->fs_info->running_transaction->transid); 380ccd467d6SChris Mason WARN_ON(1); 381ccd467d6SChris Mason } 382ccd467d6SChris Mason if (trans->transid != root->fs_info->generation) { 383ccd467d6SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 384ccd467d6SChris Mason root->fs_info->generation); 385ccd467d6SChris Mason WARN_ON(1); 386ccd467d6SChris Mason } 387dc17ff8fSChris Mason 38863b10fc4SChris Mason spin_lock(&root->fs_info->hash_lock); 3895b21f2edSZheng Yan if (btrfs_header_generation(buf) == trans->transid && 3905b21f2edSZheng Yan btrfs_header_owner(buf) == root->root_key.objectid && 39163b10fc4SChris Mason !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN)) { 39202217ed2SChris Mason *cow_ret = buf; 39363b10fc4SChris Mason spin_unlock(&root->fs_info->hash_lock); 39465b51a00SChris Mason WARN_ON(prealloc_dest); 39502217ed2SChris Mason return 0; 39602217ed2SChris Mason } 39763b10fc4SChris Mason spin_unlock(&root->fs_info->hash_lock); 3980b86a832SChris Mason search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1); 399f510cfecSChris Mason ret = __btrfs_cow_block(trans, root, buf, parent, 40065b51a00SChris Mason parent_slot, cow_ret, search_start, 0, 40165b51a00SChris Mason prealloc_dest); 402f510cfecSChris Mason return ret; 4032c90e5d6SChris Mason } 4046702ed49SChris Mason 405d352ac68SChris Mason /* 406d352ac68SChris Mason * helper function for defrag to decide if two blocks pointed to by a 407d352ac68SChris Mason * node are actually close by 408d352ac68SChris Mason */ 4096b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize) 4106702ed49SChris Mason { 4116b80053dSChris Mason if (blocknr < other && other - (blocknr + blocksize) < 32768) 4126702ed49SChris Mason return 1; 4136b80053dSChris Mason if (blocknr > other && blocknr - (other + blocksize) < 32768) 4146702ed49SChris Mason return 1; 41502217ed2SChris Mason return 0; 41602217ed2SChris Mason } 41702217ed2SChris Mason 418081e9573SChris Mason /* 419081e9573SChris Mason * compare two keys in a memcmp fashion 420081e9573SChris Mason */ 421081e9573SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2) 422081e9573SChris Mason { 423081e9573SChris Mason struct btrfs_key k1; 424081e9573SChris Mason 425081e9573SChris Mason btrfs_disk_key_to_cpu(&k1, disk); 426081e9573SChris Mason 427081e9573SChris Mason if (k1.objectid > k2->objectid) 428081e9573SChris Mason return 1; 429081e9573SChris Mason if (k1.objectid < k2->objectid) 430081e9573SChris Mason return -1; 431081e9573SChris Mason if (k1.type > k2->type) 432081e9573SChris Mason return 1; 433081e9573SChris Mason if (k1.type < k2->type) 434081e9573SChris Mason return -1; 435081e9573SChris Mason if (k1.offset > k2->offset) 436081e9573SChris Mason return 1; 437081e9573SChris Mason if (k1.offset < k2->offset) 438081e9573SChris Mason return -1; 439081e9573SChris Mason return 0; 440081e9573SChris Mason } 441081e9573SChris Mason 442f3465ca4SJosef Bacik /* 443f3465ca4SJosef Bacik * same as comp_keys only with two btrfs_key's 444f3465ca4SJosef Bacik */ 445f3465ca4SJosef Bacik static int comp_cpu_keys(struct btrfs_key *k1, struct btrfs_key *k2) 446f3465ca4SJosef Bacik { 447f3465ca4SJosef Bacik if (k1->objectid > k2->objectid) 448f3465ca4SJosef Bacik return 1; 449f3465ca4SJosef Bacik if (k1->objectid < k2->objectid) 450f3465ca4SJosef Bacik return -1; 451f3465ca4SJosef Bacik if (k1->type > k2->type) 452f3465ca4SJosef Bacik return 1; 453f3465ca4SJosef Bacik if (k1->type < k2->type) 454f3465ca4SJosef Bacik return -1; 455f3465ca4SJosef Bacik if (k1->offset > k2->offset) 456f3465ca4SJosef Bacik return 1; 457f3465ca4SJosef Bacik if (k1->offset < k2->offset) 458f3465ca4SJosef Bacik return -1; 459f3465ca4SJosef Bacik return 0; 460f3465ca4SJosef Bacik } 461081e9573SChris Mason 462d352ac68SChris Mason /* 463d352ac68SChris Mason * this is used by the defrag code to go through all the 464d352ac68SChris Mason * leaves pointed to by a node and reallocate them so that 465d352ac68SChris Mason * disk order is close to key order 466d352ac68SChris Mason */ 4676702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans, 4685f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *parent, 469a6b6e75eSChris Mason int start_slot, int cache_only, u64 *last_ret, 470a6b6e75eSChris Mason struct btrfs_key *progress) 4716702ed49SChris Mason { 4726b80053dSChris Mason struct extent_buffer *cur; 4736702ed49SChris Mason u64 blocknr; 474ca7a79adSChris Mason u64 gen; 475e9d0b13bSChris Mason u64 search_start = *last_ret; 476e9d0b13bSChris Mason u64 last_block = 0; 4776702ed49SChris Mason u64 other; 4786702ed49SChris Mason u32 parent_nritems; 4796702ed49SChris Mason int end_slot; 4806702ed49SChris Mason int i; 4816702ed49SChris Mason int err = 0; 482f2183bdeSChris Mason int parent_level; 4836b80053dSChris Mason int uptodate; 4846b80053dSChris Mason u32 blocksize; 485081e9573SChris Mason int progress_passed = 0; 486081e9573SChris Mason struct btrfs_disk_key disk_key; 4876702ed49SChris Mason 4885708b959SChris Mason parent_level = btrfs_header_level(parent); 4895708b959SChris Mason if (cache_only && parent_level != 1) 4905708b959SChris Mason return 0; 4915708b959SChris Mason 4926702ed49SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 4936702ed49SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 4946702ed49SChris Mason root->fs_info->running_transaction->transid); 4956702ed49SChris Mason WARN_ON(1); 4966702ed49SChris Mason } 4976702ed49SChris Mason if (trans->transid != root->fs_info->generation) { 4986702ed49SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 4996702ed49SChris Mason root->fs_info->generation); 5006702ed49SChris Mason WARN_ON(1); 5016702ed49SChris Mason } 50286479a04SChris Mason 5036b80053dSChris Mason parent_nritems = btrfs_header_nritems(parent); 5046b80053dSChris Mason blocksize = btrfs_level_size(root, parent_level - 1); 5056702ed49SChris Mason end_slot = parent_nritems; 5066702ed49SChris Mason 5076702ed49SChris Mason if (parent_nritems == 1) 5086702ed49SChris Mason return 0; 5096702ed49SChris Mason 5106702ed49SChris Mason for (i = start_slot; i < end_slot; i++) { 5116702ed49SChris Mason int close = 1; 512a6b6e75eSChris Mason 5135708b959SChris Mason if (!parent->map_token) { 5145708b959SChris Mason map_extent_buffer(parent, 5155708b959SChris Mason btrfs_node_key_ptr_offset(i), 5165708b959SChris Mason sizeof(struct btrfs_key_ptr), 5175708b959SChris Mason &parent->map_token, &parent->kaddr, 5185708b959SChris Mason &parent->map_start, &parent->map_len, 5195708b959SChris Mason KM_USER1); 5205708b959SChris Mason } 521081e9573SChris Mason btrfs_node_key(parent, &disk_key, i); 522081e9573SChris Mason if (!progress_passed && comp_keys(&disk_key, progress) < 0) 523081e9573SChris Mason continue; 524081e9573SChris Mason 525081e9573SChris Mason progress_passed = 1; 5266b80053dSChris Mason blocknr = btrfs_node_blockptr(parent, i); 527ca7a79adSChris Mason gen = btrfs_node_ptr_generation(parent, i); 528e9d0b13bSChris Mason if (last_block == 0) 529e9d0b13bSChris Mason last_block = blocknr; 5305708b959SChris Mason 5316702ed49SChris Mason if (i > 0) { 5326b80053dSChris Mason other = btrfs_node_blockptr(parent, i - 1); 5336b80053dSChris Mason close = close_blocks(blocknr, other, blocksize); 5346702ed49SChris Mason } 5350ef3e66bSChris Mason if (!close && i < end_slot - 2) { 5366b80053dSChris Mason other = btrfs_node_blockptr(parent, i + 1); 5376b80053dSChris Mason close = close_blocks(blocknr, other, blocksize); 5386702ed49SChris Mason } 539e9d0b13bSChris Mason if (close) { 540e9d0b13bSChris Mason last_block = blocknr; 5416702ed49SChris Mason continue; 542e9d0b13bSChris Mason } 5435708b959SChris Mason if (parent->map_token) { 5445708b959SChris Mason unmap_extent_buffer(parent, parent->map_token, 5455708b959SChris Mason KM_USER1); 5465708b959SChris Mason parent->map_token = NULL; 5475708b959SChris Mason } 5486702ed49SChris Mason 5496b80053dSChris Mason cur = btrfs_find_tree_block(root, blocknr, blocksize); 5506b80053dSChris Mason if (cur) 5511259ab75SChris Mason uptodate = btrfs_buffer_uptodate(cur, gen); 5526b80053dSChris Mason else 5536b80053dSChris Mason uptodate = 0; 5545708b959SChris Mason if (!cur || !uptodate) { 5556702ed49SChris Mason if (cache_only) { 5566b80053dSChris Mason free_extent_buffer(cur); 5576702ed49SChris Mason continue; 5586702ed49SChris Mason } 5596b80053dSChris Mason if (!cur) { 5606b80053dSChris Mason cur = read_tree_block(root, blocknr, 561ca7a79adSChris Mason blocksize, gen); 5626b80053dSChris Mason } else if (!uptodate) { 563ca7a79adSChris Mason btrfs_read_buffer(cur, gen); 5646702ed49SChris Mason } 565f2183bdeSChris Mason } 566e9d0b13bSChris Mason if (search_start == 0) 5676b80053dSChris Mason search_start = last_block; 568e9d0b13bSChris Mason 569e7a84565SChris Mason btrfs_tree_lock(cur); 5706b80053dSChris Mason err = __btrfs_cow_block(trans, root, cur, parent, i, 571e7a84565SChris Mason &cur, search_start, 5726b80053dSChris Mason min(16 * blocksize, 57365b51a00SChris Mason (end_slot - i) * blocksize), 0); 574252c38f0SYan if (err) { 575e7a84565SChris Mason btrfs_tree_unlock(cur); 5766b80053dSChris Mason free_extent_buffer(cur); 5776702ed49SChris Mason break; 578252c38f0SYan } 579e7a84565SChris Mason search_start = cur->start; 580e7a84565SChris Mason last_block = cur->start; 581f2183bdeSChris Mason *last_ret = search_start; 582e7a84565SChris Mason btrfs_tree_unlock(cur); 583e7a84565SChris Mason free_extent_buffer(cur); 5846702ed49SChris Mason } 5855708b959SChris Mason if (parent->map_token) { 5865708b959SChris Mason unmap_extent_buffer(parent, parent->map_token, 5875708b959SChris Mason KM_USER1); 5885708b959SChris Mason parent->map_token = NULL; 5895708b959SChris Mason } 5906702ed49SChris Mason return err; 5916702ed49SChris Mason } 5926702ed49SChris Mason 59374123bd7SChris Mason /* 59474123bd7SChris Mason * The leaf data grows from end-to-front in the node. 59574123bd7SChris Mason * this returns the address of the start of the last item, 59674123bd7SChris Mason * which is the stop of the leaf data stack 59774123bd7SChris Mason */ 598123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root, 5995f39d397SChris Mason struct extent_buffer *leaf) 600be0e5c09SChris Mason { 6015f39d397SChris Mason u32 nr = btrfs_header_nritems(leaf); 602be0e5c09SChris Mason if (nr == 0) 603123abc88SChris Mason return BTRFS_LEAF_DATA_SIZE(root); 6045f39d397SChris Mason return btrfs_item_offset_nr(leaf, nr - 1); 605be0e5c09SChris Mason } 606be0e5c09SChris Mason 607d352ac68SChris Mason /* 608d352ac68SChris Mason * extra debugging checks to make sure all the items in a key are 609d352ac68SChris Mason * well formed and in the proper order 610d352ac68SChris Mason */ 611123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path, 612123abc88SChris Mason int level) 613aa5d6bedSChris Mason { 6145f39d397SChris Mason struct extent_buffer *parent = NULL; 6155f39d397SChris Mason struct extent_buffer *node = path->nodes[level]; 6165f39d397SChris Mason struct btrfs_disk_key parent_key; 6175f39d397SChris Mason struct btrfs_disk_key node_key; 618aa5d6bedSChris Mason int parent_slot; 6198d7be552SChris Mason int slot; 6208d7be552SChris Mason struct btrfs_key cpukey; 6215f39d397SChris Mason u32 nritems = btrfs_header_nritems(node); 622aa5d6bedSChris Mason 623aa5d6bedSChris Mason if (path->nodes[level + 1]) 6245f39d397SChris Mason parent = path->nodes[level + 1]; 625a1f39630SAneesh 6268d7be552SChris Mason slot = path->slots[level]; 6277518a238SChris Mason BUG_ON(nritems == 0); 6287518a238SChris Mason if (parent) { 629a1f39630SAneesh parent_slot = path->slots[level + 1]; 6305f39d397SChris Mason btrfs_node_key(parent, &parent_key, parent_slot); 6315f39d397SChris Mason btrfs_node_key(node, &node_key, 0); 6325f39d397SChris Mason BUG_ON(memcmp(&parent_key, &node_key, 633e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 6341d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 635db94535dSChris Mason btrfs_header_bytenr(node)); 636aa5d6bedSChris Mason } 637123abc88SChris Mason BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root)); 6388d7be552SChris Mason if (slot != 0) { 6395f39d397SChris Mason btrfs_node_key_to_cpu(node, &cpukey, slot - 1); 6405f39d397SChris Mason btrfs_node_key(node, &node_key, slot); 6415f39d397SChris Mason BUG_ON(comp_keys(&node_key, &cpukey) <= 0); 6428d7be552SChris Mason } 6438d7be552SChris Mason if (slot < nritems - 1) { 6445f39d397SChris Mason btrfs_node_key_to_cpu(node, &cpukey, slot + 1); 6455f39d397SChris Mason btrfs_node_key(node, &node_key, slot); 6465f39d397SChris Mason BUG_ON(comp_keys(&node_key, &cpukey) >= 0); 647aa5d6bedSChris Mason } 648aa5d6bedSChris Mason return 0; 649aa5d6bedSChris Mason } 650aa5d6bedSChris Mason 651d352ac68SChris Mason /* 652d352ac68SChris Mason * extra checking to make sure all the items in a leaf are 653d352ac68SChris Mason * well formed and in the proper order 654d352ac68SChris Mason */ 655123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path, 656123abc88SChris Mason int level) 657aa5d6bedSChris Mason { 6585f39d397SChris Mason struct extent_buffer *leaf = path->nodes[level]; 6595f39d397SChris Mason struct extent_buffer *parent = NULL; 660aa5d6bedSChris Mason int parent_slot; 6618d7be552SChris Mason struct btrfs_key cpukey; 6625f39d397SChris Mason struct btrfs_disk_key parent_key; 6635f39d397SChris Mason struct btrfs_disk_key leaf_key; 6645f39d397SChris Mason int slot = path->slots[0]; 6658d7be552SChris Mason 6665f39d397SChris Mason u32 nritems = btrfs_header_nritems(leaf); 667aa5d6bedSChris Mason 668aa5d6bedSChris Mason if (path->nodes[level + 1]) 6695f39d397SChris Mason parent = path->nodes[level + 1]; 6707518a238SChris Mason 6717518a238SChris Mason if (nritems == 0) 6727518a238SChris Mason return 0; 6737518a238SChris Mason 6747518a238SChris Mason if (parent) { 675a1f39630SAneesh parent_slot = path->slots[level + 1]; 6765f39d397SChris Mason btrfs_node_key(parent, &parent_key, parent_slot); 6775f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, 0); 6786702ed49SChris Mason 6795f39d397SChris Mason BUG_ON(memcmp(&parent_key, &leaf_key, 680e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 6811d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 682db94535dSChris Mason btrfs_header_bytenr(leaf)); 683aa5d6bedSChris Mason } 6845f39d397SChris Mason #if 0 6855f39d397SChris Mason for (i = 0; nritems > 1 && i < nritems - 2; i++) { 6865f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, i + 1); 6875f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, i); 6885f39d397SChris Mason if (comp_keys(&leaf_key, &cpukey) >= 0) { 6895f39d397SChris Mason btrfs_print_leaf(root, leaf); 6905f39d397SChris Mason printk("slot %d offset bad key\n", i); 6915f39d397SChris Mason BUG_ON(1); 6925f39d397SChris Mason } 6935f39d397SChris Mason if (btrfs_item_offset_nr(leaf, i) != 6945f39d397SChris Mason btrfs_item_end_nr(leaf, i + 1)) { 6955f39d397SChris Mason btrfs_print_leaf(root, leaf); 6965f39d397SChris Mason printk("slot %d offset bad\n", i); 6975f39d397SChris Mason BUG_ON(1); 6985f39d397SChris Mason } 6995f39d397SChris Mason if (i == 0) { 7005f39d397SChris Mason if (btrfs_item_offset_nr(leaf, i) + 7015f39d397SChris Mason btrfs_item_size_nr(leaf, i) != 7025f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 7035f39d397SChris Mason btrfs_print_leaf(root, leaf); 7045f39d397SChris Mason printk("slot %d first offset bad\n", i); 7055f39d397SChris Mason BUG_ON(1); 7065f39d397SChris Mason } 7075f39d397SChris Mason } 7085f39d397SChris Mason } 7095f39d397SChris Mason if (nritems > 0) { 7105f39d397SChris Mason if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) { 7115f39d397SChris Mason btrfs_print_leaf(root, leaf); 7125f39d397SChris Mason printk("slot %d bad size \n", nritems - 1); 7135f39d397SChris Mason BUG_ON(1); 7145f39d397SChris Mason } 7155f39d397SChris Mason } 7165f39d397SChris Mason #endif 7175f39d397SChris Mason if (slot != 0 && slot < nritems - 1) { 7185f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, slot); 7195f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1); 7205f39d397SChris Mason if (comp_keys(&leaf_key, &cpukey) <= 0) { 7215f39d397SChris Mason btrfs_print_leaf(root, leaf); 7225f39d397SChris Mason printk("slot %d offset bad key\n", slot); 7235f39d397SChris Mason BUG_ON(1); 7245f39d397SChris Mason } 7255f39d397SChris Mason if (btrfs_item_offset_nr(leaf, slot - 1) != 7265f39d397SChris Mason btrfs_item_end_nr(leaf, slot)) { 7275f39d397SChris Mason btrfs_print_leaf(root, leaf); 7285f39d397SChris Mason printk("slot %d offset bad\n", slot); 7295f39d397SChris Mason BUG_ON(1); 7305f39d397SChris Mason } 731aa5d6bedSChris Mason } 7328d7be552SChris Mason if (slot < nritems - 1) { 7335f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, slot); 7345f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1); 7355f39d397SChris Mason BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0); 7365f39d397SChris Mason if (btrfs_item_offset_nr(leaf, slot) != 7375f39d397SChris Mason btrfs_item_end_nr(leaf, slot + 1)) { 7385f39d397SChris Mason btrfs_print_leaf(root, leaf); 7395f39d397SChris Mason printk("slot %d offset bad\n", slot); 7405f39d397SChris Mason BUG_ON(1); 741aa5d6bedSChris Mason } 7425f39d397SChris Mason } 7435f39d397SChris Mason BUG_ON(btrfs_item_offset_nr(leaf, 0) + 7445f39d397SChris Mason btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root)); 745aa5d6bedSChris Mason return 0; 746aa5d6bedSChris Mason } 747aa5d6bedSChris Mason 74898ed5174SChris Mason static int noinline check_block(struct btrfs_root *root, 74998ed5174SChris Mason struct btrfs_path *path, int level) 750aa5d6bedSChris Mason { 751f188591eSChris Mason u64 found_start; 75285d824c4SChris Mason return 0; 753f188591eSChris Mason if (btrfs_header_level(path->nodes[level]) != level) 754f188591eSChris Mason printk("warning: bad level %Lu wanted %d found %d\n", 755f188591eSChris Mason path->nodes[level]->start, level, 756f188591eSChris Mason btrfs_header_level(path->nodes[level])); 757f188591eSChris Mason found_start = btrfs_header_bytenr(path->nodes[level]); 758f188591eSChris Mason if (found_start != path->nodes[level]->start) { 759f188591eSChris Mason printk("warning: bad bytentr %Lu found %Lu\n", 760f188591eSChris Mason path->nodes[level]->start, found_start); 761f188591eSChris Mason } 762db94535dSChris Mason #if 0 7635f39d397SChris Mason struct extent_buffer *buf = path->nodes[level]; 7645f39d397SChris Mason 765479965d6SChris Mason if (memcmp_extent_buffer(buf, root->fs_info->fsid, 766479965d6SChris Mason (unsigned long)btrfs_header_fsid(buf), 767479965d6SChris Mason BTRFS_FSID_SIZE)) { 7685f39d397SChris Mason printk("warning bad block %Lu\n", buf->start); 769db94535dSChris Mason return 1; 7705f39d397SChris Mason } 771db94535dSChris Mason #endif 772aa5d6bedSChris Mason if (level == 0) 773123abc88SChris Mason return check_leaf(root, path, level); 774123abc88SChris Mason return check_node(root, path, level); 775aa5d6bedSChris Mason } 776aa5d6bedSChris Mason 77774123bd7SChris Mason /* 7785f39d397SChris Mason * search for key in the extent_buffer. The items start at offset p, 7795f39d397SChris Mason * and they are item_size apart. There are 'max' items in p. 7805f39d397SChris Mason * 78174123bd7SChris Mason * the slot in the array is returned via slot, and it points to 78274123bd7SChris Mason * the place where you would insert key if it is not found in 78374123bd7SChris Mason * the array. 78474123bd7SChris Mason * 78574123bd7SChris Mason * slot may point to max if the key is bigger than all of the keys 78674123bd7SChris Mason */ 787e02119d5SChris Mason static noinline int generic_bin_search(struct extent_buffer *eb, 788e02119d5SChris Mason unsigned long p, 7895f39d397SChris Mason int item_size, struct btrfs_key *key, 790be0e5c09SChris Mason int max, int *slot) 791be0e5c09SChris Mason { 792be0e5c09SChris Mason int low = 0; 793be0e5c09SChris Mason int high = max; 794be0e5c09SChris Mason int mid; 795be0e5c09SChris Mason int ret; 796479965d6SChris Mason struct btrfs_disk_key *tmp = NULL; 7975f39d397SChris Mason struct btrfs_disk_key unaligned; 7985f39d397SChris Mason unsigned long offset; 7995f39d397SChris Mason char *map_token = NULL; 8005f39d397SChris Mason char *kaddr = NULL; 8015f39d397SChris Mason unsigned long map_start = 0; 8025f39d397SChris Mason unsigned long map_len = 0; 803479965d6SChris Mason int err; 804be0e5c09SChris Mason 805be0e5c09SChris Mason while(low < high) { 806be0e5c09SChris Mason mid = (low + high) / 2; 8075f39d397SChris Mason offset = p + mid * item_size; 8085f39d397SChris Mason 8095f39d397SChris Mason if (!map_token || offset < map_start || 8105f39d397SChris Mason (offset + sizeof(struct btrfs_disk_key)) > 8115f39d397SChris Mason map_start + map_len) { 812479965d6SChris Mason if (map_token) { 8135f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 814479965d6SChris Mason map_token = NULL; 815479965d6SChris Mason } 816934d375bSChris Mason 817934d375bSChris Mason err = map_private_extent_buffer(eb, offset, 818479965d6SChris Mason sizeof(struct btrfs_disk_key), 819479965d6SChris Mason &map_token, &kaddr, 8205f39d397SChris Mason &map_start, &map_len, KM_USER0); 8215f39d397SChris Mason 822479965d6SChris Mason if (!err) { 823479965d6SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 824479965d6SChris Mason map_start); 825479965d6SChris Mason } else { 8265f39d397SChris Mason read_extent_buffer(eb, &unaligned, 8275f39d397SChris Mason offset, sizeof(unaligned)); 8285f39d397SChris Mason tmp = &unaligned; 829479965d6SChris Mason } 830479965d6SChris Mason 8315f39d397SChris Mason } else { 8325f39d397SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 8335f39d397SChris Mason map_start); 8345f39d397SChris Mason } 835be0e5c09SChris Mason ret = comp_keys(tmp, key); 836be0e5c09SChris Mason 837be0e5c09SChris Mason if (ret < 0) 838be0e5c09SChris Mason low = mid + 1; 839be0e5c09SChris Mason else if (ret > 0) 840be0e5c09SChris Mason high = mid; 841be0e5c09SChris Mason else { 842be0e5c09SChris Mason *slot = mid; 843479965d6SChris Mason if (map_token) 8445f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 845be0e5c09SChris Mason return 0; 846be0e5c09SChris Mason } 847be0e5c09SChris Mason } 848be0e5c09SChris Mason *slot = low; 8495f39d397SChris Mason if (map_token) 8505f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 851be0e5c09SChris Mason return 1; 852be0e5c09SChris Mason } 853be0e5c09SChris Mason 85497571fd0SChris Mason /* 85597571fd0SChris Mason * simple bin_search frontend that does the right thing for 85697571fd0SChris Mason * leaves vs nodes 85797571fd0SChris Mason */ 8585f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key, 8595f39d397SChris Mason int level, int *slot) 860be0e5c09SChris Mason { 8615f39d397SChris Mason if (level == 0) { 8625f39d397SChris Mason return generic_bin_search(eb, 8635f39d397SChris Mason offsetof(struct btrfs_leaf, items), 8640783fcfcSChris Mason sizeof(struct btrfs_item), 8655f39d397SChris Mason key, btrfs_header_nritems(eb), 8667518a238SChris Mason slot); 867be0e5c09SChris Mason } else { 8685f39d397SChris Mason return generic_bin_search(eb, 8695f39d397SChris Mason offsetof(struct btrfs_node, ptrs), 870123abc88SChris Mason sizeof(struct btrfs_key_ptr), 8715f39d397SChris Mason key, btrfs_header_nritems(eb), 8727518a238SChris Mason slot); 873be0e5c09SChris Mason } 874be0e5c09SChris Mason return -1; 875be0e5c09SChris Mason } 876be0e5c09SChris Mason 877d352ac68SChris Mason /* given a node and slot number, this reads the blocks it points to. The 878d352ac68SChris Mason * extent buffer is returned with a reference taken (but unlocked). 879d352ac68SChris Mason * NULL is returned on error. 880d352ac68SChris Mason */ 881e02119d5SChris Mason static noinline struct extent_buffer *read_node_slot(struct btrfs_root *root, 8825f39d397SChris Mason struct extent_buffer *parent, int slot) 883bb803951SChris Mason { 884ca7a79adSChris Mason int level = btrfs_header_level(parent); 885bb803951SChris Mason if (slot < 0) 886bb803951SChris Mason return NULL; 8875f39d397SChris Mason if (slot >= btrfs_header_nritems(parent)) 888bb803951SChris Mason return NULL; 889ca7a79adSChris Mason 890ca7a79adSChris Mason BUG_ON(level == 0); 891ca7a79adSChris Mason 892db94535dSChris Mason return read_tree_block(root, btrfs_node_blockptr(parent, slot), 893ca7a79adSChris Mason btrfs_level_size(root, level - 1), 894ca7a79adSChris Mason btrfs_node_ptr_generation(parent, slot)); 895bb803951SChris Mason } 896bb803951SChris Mason 897d352ac68SChris Mason /* 898d352ac68SChris Mason * node level balancing, used to make sure nodes are in proper order for 899d352ac68SChris Mason * item deletion. We balance from the top down, so we have to make sure 900d352ac68SChris Mason * that a deletion won't leave an node completely empty later on. 901d352ac68SChris Mason */ 902e02119d5SChris Mason static noinline int balance_level(struct btrfs_trans_handle *trans, 90398ed5174SChris Mason struct btrfs_root *root, 90498ed5174SChris Mason struct btrfs_path *path, int level) 905bb803951SChris Mason { 9065f39d397SChris Mason struct extent_buffer *right = NULL; 9075f39d397SChris Mason struct extent_buffer *mid; 9085f39d397SChris Mason struct extent_buffer *left = NULL; 9095f39d397SChris Mason struct extent_buffer *parent = NULL; 910bb803951SChris Mason int ret = 0; 911bb803951SChris Mason int wret; 912bb803951SChris Mason int pslot; 913bb803951SChris Mason int orig_slot = path->slots[level]; 91454aa1f4dSChris Mason int err_on_enospc = 0; 91579f95c82SChris Mason u64 orig_ptr; 916bb803951SChris Mason 917bb803951SChris Mason if (level == 0) 918bb803951SChris Mason return 0; 919bb803951SChris Mason 9205f39d397SChris Mason mid = path->nodes[level]; 921925baeddSChris Mason WARN_ON(!path->locks[level]); 9227bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 9237bb86316SChris Mason 9241d4f8a0cSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 92579f95c82SChris Mason 926234b63a0SChris Mason if (level < BTRFS_MAX_LEVEL - 1) 9275f39d397SChris Mason parent = path->nodes[level + 1]; 928bb803951SChris Mason pslot = path->slots[level + 1]; 929bb803951SChris Mason 93040689478SChris Mason /* 93140689478SChris Mason * deal with the case where there is only one pointer in the root 93240689478SChris Mason * by promoting the node below to a root 93340689478SChris Mason */ 9345f39d397SChris Mason if (!parent) { 9355f39d397SChris Mason struct extent_buffer *child; 936bb803951SChris Mason 9375f39d397SChris Mason if (btrfs_header_nritems(mid) != 1) 938bb803951SChris Mason return 0; 939bb803951SChris Mason 940bb803951SChris Mason /* promote the child to a root */ 9415f39d397SChris Mason child = read_node_slot(root, mid, 0); 942925baeddSChris Mason btrfs_tree_lock(child); 943bb803951SChris Mason BUG_ON(!child); 94465b51a00SChris Mason ret = btrfs_cow_block(trans, root, child, mid, 0, &child, 0); 9452f375ab9SYan BUG_ON(ret); 9462f375ab9SYan 947925baeddSChris Mason spin_lock(&root->node_lock); 948bb803951SChris Mason root->node = child; 949925baeddSChris Mason spin_unlock(&root->node_lock); 950925baeddSChris Mason 95131840ae1SZheng Yan ret = btrfs_update_extent_ref(trans, root, child->start, 95231840ae1SZheng Yan mid->start, child->start, 95331840ae1SZheng Yan root->root_key.objectid, 9543bb1a1bcSYan Zheng trans->transid, level - 1); 95531840ae1SZheng Yan BUG_ON(ret); 95631840ae1SZheng Yan 9570b86a832SChris Mason add_root_to_dirty_list(root); 958925baeddSChris Mason btrfs_tree_unlock(child); 959925baeddSChris Mason path->locks[level] = 0; 960bb803951SChris Mason path->nodes[level] = NULL; 9615f39d397SChris Mason clean_tree_block(trans, root, mid); 962925baeddSChris Mason btrfs_tree_unlock(mid); 963bb803951SChris Mason /* once for the path */ 9645f39d397SChris Mason free_extent_buffer(mid); 9657bb86316SChris Mason ret = btrfs_free_extent(trans, root, mid->start, mid->len, 96631840ae1SZheng Yan mid->start, root->root_key.objectid, 9673bb1a1bcSYan Zheng btrfs_header_generation(mid), 9683bb1a1bcSYan Zheng level, 1); 969bb803951SChris Mason /* once for the root ptr */ 9705f39d397SChris Mason free_extent_buffer(mid); 971db94535dSChris Mason return ret; 972bb803951SChris Mason } 9735f39d397SChris Mason if (btrfs_header_nritems(mid) > 974123abc88SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) / 4) 975bb803951SChris Mason return 0; 976bb803951SChris Mason 9775f39d397SChris Mason if (btrfs_header_nritems(mid) < 2) 97854aa1f4dSChris Mason err_on_enospc = 1; 97954aa1f4dSChris Mason 9805f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 9815f39d397SChris Mason if (left) { 982925baeddSChris Mason btrfs_tree_lock(left); 9835f39d397SChris Mason wret = btrfs_cow_block(trans, root, left, 98465b51a00SChris Mason parent, pslot - 1, &left, 0); 98554aa1f4dSChris Mason if (wret) { 98654aa1f4dSChris Mason ret = wret; 98754aa1f4dSChris Mason goto enospc; 98854aa1f4dSChris Mason } 9892cc58cf2SChris Mason } 9905f39d397SChris Mason right = read_node_slot(root, parent, pslot + 1); 9915f39d397SChris Mason if (right) { 992925baeddSChris Mason btrfs_tree_lock(right); 9935f39d397SChris Mason wret = btrfs_cow_block(trans, root, right, 99465b51a00SChris Mason parent, pslot + 1, &right, 0); 9952cc58cf2SChris Mason if (wret) { 9962cc58cf2SChris Mason ret = wret; 9972cc58cf2SChris Mason goto enospc; 9982cc58cf2SChris Mason } 9992cc58cf2SChris Mason } 10002cc58cf2SChris Mason 10012cc58cf2SChris Mason /* first, try to make some room in the middle buffer */ 10025f39d397SChris Mason if (left) { 10035f39d397SChris Mason orig_slot += btrfs_header_nritems(left); 1004bce4eae9SChris Mason wret = push_node_left(trans, root, left, mid, 1); 100579f95c82SChris Mason if (wret < 0) 100679f95c82SChris Mason ret = wret; 10075f39d397SChris Mason if (btrfs_header_nritems(mid) < 2) 100854aa1f4dSChris Mason err_on_enospc = 1; 1009bb803951SChris Mason } 101079f95c82SChris Mason 101179f95c82SChris Mason /* 101279f95c82SChris Mason * then try to empty the right most buffer into the middle 101379f95c82SChris Mason */ 10145f39d397SChris Mason if (right) { 1015971a1f66SChris Mason wret = push_node_left(trans, root, mid, right, 1); 101654aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 101779f95c82SChris Mason ret = wret; 10185f39d397SChris Mason if (btrfs_header_nritems(right) == 0) { 1019db94535dSChris Mason u64 bytenr = right->start; 10207bb86316SChris Mason u64 generation = btrfs_header_generation(parent); 1021db94535dSChris Mason u32 blocksize = right->len; 1022db94535dSChris Mason 10235f39d397SChris Mason clean_tree_block(trans, root, right); 1024925baeddSChris Mason btrfs_tree_unlock(right); 10255f39d397SChris Mason free_extent_buffer(right); 1026bb803951SChris Mason right = NULL; 1027e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot + 1028e089f05cSChris Mason 1); 1029bb803951SChris Mason if (wret) 1030bb803951SChris Mason ret = wret; 1031db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 103231840ae1SZheng Yan blocksize, parent->start, 10337bb86316SChris Mason btrfs_header_owner(parent), 10343bb1a1bcSYan Zheng generation, level, 1); 1035bb803951SChris Mason if (wret) 1036bb803951SChris Mason ret = wret; 1037bb803951SChris Mason } else { 10385f39d397SChris Mason struct btrfs_disk_key right_key; 10395f39d397SChris Mason btrfs_node_key(right, &right_key, 0); 10405f39d397SChris Mason btrfs_set_node_key(parent, &right_key, pslot + 1); 10415f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 1042bb803951SChris Mason } 1043bb803951SChris Mason } 10445f39d397SChris Mason if (btrfs_header_nritems(mid) == 1) { 104579f95c82SChris Mason /* 104679f95c82SChris Mason * we're not allowed to leave a node with one item in the 104779f95c82SChris Mason * tree during a delete. A deletion from lower in the tree 104879f95c82SChris Mason * could try to delete the only pointer in this node. 104979f95c82SChris Mason * So, pull some keys from the left. 105079f95c82SChris Mason * There has to be a left pointer at this point because 105179f95c82SChris Mason * otherwise we would have pulled some pointers from the 105279f95c82SChris Mason * right 105379f95c82SChris Mason */ 10545f39d397SChris Mason BUG_ON(!left); 10555f39d397SChris Mason wret = balance_node_right(trans, root, mid, left); 105654aa1f4dSChris Mason if (wret < 0) { 105779f95c82SChris Mason ret = wret; 105854aa1f4dSChris Mason goto enospc; 105954aa1f4dSChris Mason } 1060bce4eae9SChris Mason if (wret == 1) { 1061bce4eae9SChris Mason wret = push_node_left(trans, root, left, mid, 1); 1062bce4eae9SChris Mason if (wret < 0) 1063bce4eae9SChris Mason ret = wret; 1064bce4eae9SChris Mason } 106579f95c82SChris Mason BUG_ON(wret == 1); 106679f95c82SChris Mason } 10675f39d397SChris Mason if (btrfs_header_nritems(mid) == 0) { 106879f95c82SChris Mason /* we've managed to empty the middle node, drop it */ 10697bb86316SChris Mason u64 root_gen = btrfs_header_generation(parent); 1070db94535dSChris Mason u64 bytenr = mid->start; 1071db94535dSChris Mason u32 blocksize = mid->len; 1072925baeddSChris Mason 10735f39d397SChris Mason clean_tree_block(trans, root, mid); 1074925baeddSChris Mason btrfs_tree_unlock(mid); 10755f39d397SChris Mason free_extent_buffer(mid); 1076bb803951SChris Mason mid = NULL; 1077e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot); 1078bb803951SChris Mason if (wret) 1079bb803951SChris Mason ret = wret; 10807bb86316SChris Mason wret = btrfs_free_extent(trans, root, bytenr, blocksize, 108131840ae1SZheng Yan parent->start, 10827bb86316SChris Mason btrfs_header_owner(parent), 10833bb1a1bcSYan Zheng root_gen, level, 1); 1084bb803951SChris Mason if (wret) 1085bb803951SChris Mason ret = wret; 108679f95c82SChris Mason } else { 108779f95c82SChris Mason /* update the parent key to reflect our changes */ 10885f39d397SChris Mason struct btrfs_disk_key mid_key; 10895f39d397SChris Mason btrfs_node_key(mid, &mid_key, 0); 10905f39d397SChris Mason btrfs_set_node_key(parent, &mid_key, pslot); 10915f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 109279f95c82SChris Mason } 1093bb803951SChris Mason 109479f95c82SChris Mason /* update the path */ 10955f39d397SChris Mason if (left) { 10965f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 10975f39d397SChris Mason extent_buffer_get(left); 1098925baeddSChris Mason /* left was locked after cow */ 10995f39d397SChris Mason path->nodes[level] = left; 1100bb803951SChris Mason path->slots[level + 1] -= 1; 1101bb803951SChris Mason path->slots[level] = orig_slot; 1102925baeddSChris Mason if (mid) { 1103925baeddSChris Mason btrfs_tree_unlock(mid); 11045f39d397SChris Mason free_extent_buffer(mid); 1105925baeddSChris Mason } 1106bb803951SChris Mason } else { 11075f39d397SChris Mason orig_slot -= btrfs_header_nritems(left); 1108bb803951SChris Mason path->slots[level] = orig_slot; 1109bb803951SChris Mason } 1110bb803951SChris Mason } 111179f95c82SChris Mason /* double check we haven't messed things up */ 1112123abc88SChris Mason check_block(root, path, level); 1113e20d96d6SChris Mason if (orig_ptr != 11145f39d397SChris Mason btrfs_node_blockptr(path->nodes[level], path->slots[level])) 111579f95c82SChris Mason BUG(); 111654aa1f4dSChris Mason enospc: 1117925baeddSChris Mason if (right) { 1118925baeddSChris Mason btrfs_tree_unlock(right); 11195f39d397SChris Mason free_extent_buffer(right); 1120925baeddSChris Mason } 1121925baeddSChris Mason if (left) { 1122925baeddSChris Mason if (path->nodes[level] != left) 1123925baeddSChris Mason btrfs_tree_unlock(left); 11245f39d397SChris Mason free_extent_buffer(left); 1125925baeddSChris Mason } 1126bb803951SChris Mason return ret; 1127bb803951SChris Mason } 1128bb803951SChris Mason 1129d352ac68SChris Mason /* Node balancing for insertion. Here we only split or push nodes around 1130d352ac68SChris Mason * when they are completely full. This is also done top down, so we 1131d352ac68SChris Mason * have to be pessimistic. 1132d352ac68SChris Mason */ 113398ed5174SChris Mason static int noinline push_nodes_for_insert(struct btrfs_trans_handle *trans, 1134e66f709bSChris Mason struct btrfs_root *root, 1135e66f709bSChris Mason struct btrfs_path *path, int level) 1136e66f709bSChris Mason { 11375f39d397SChris Mason struct extent_buffer *right = NULL; 11385f39d397SChris Mason struct extent_buffer *mid; 11395f39d397SChris Mason struct extent_buffer *left = NULL; 11405f39d397SChris Mason struct extent_buffer *parent = NULL; 1141e66f709bSChris Mason int ret = 0; 1142e66f709bSChris Mason int wret; 1143e66f709bSChris Mason int pslot; 1144e66f709bSChris Mason int orig_slot = path->slots[level]; 1145e66f709bSChris Mason u64 orig_ptr; 1146e66f709bSChris Mason 1147e66f709bSChris Mason if (level == 0) 1148e66f709bSChris Mason return 1; 1149e66f709bSChris Mason 11505f39d397SChris Mason mid = path->nodes[level]; 11517bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 1152e66f709bSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 1153e66f709bSChris Mason 1154e66f709bSChris Mason if (level < BTRFS_MAX_LEVEL - 1) 11555f39d397SChris Mason parent = path->nodes[level + 1]; 1156e66f709bSChris Mason pslot = path->slots[level + 1]; 1157e66f709bSChris Mason 11585f39d397SChris Mason if (!parent) 1159e66f709bSChris Mason return 1; 1160e66f709bSChris Mason 11615f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 1162e66f709bSChris Mason 1163e66f709bSChris Mason /* first, try to make some room in the middle buffer */ 11645f39d397SChris Mason if (left) { 1165e66f709bSChris Mason u32 left_nr; 1166925baeddSChris Mason 1167925baeddSChris Mason btrfs_tree_lock(left); 11685f39d397SChris Mason left_nr = btrfs_header_nritems(left); 116933ade1f8SChris Mason if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 117033ade1f8SChris Mason wret = 1; 117133ade1f8SChris Mason } else { 11725f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, parent, 117365b51a00SChris Mason pslot - 1, &left, 0); 117454aa1f4dSChris Mason if (ret) 117554aa1f4dSChris Mason wret = 1; 117654aa1f4dSChris Mason else { 117754aa1f4dSChris Mason wret = push_node_left(trans, root, 1178971a1f66SChris Mason left, mid, 0); 117954aa1f4dSChris Mason } 118033ade1f8SChris Mason } 1181e66f709bSChris Mason if (wret < 0) 1182e66f709bSChris Mason ret = wret; 1183e66f709bSChris Mason if (wret == 0) { 11845f39d397SChris Mason struct btrfs_disk_key disk_key; 1185e66f709bSChris Mason orig_slot += left_nr; 11865f39d397SChris Mason btrfs_node_key(mid, &disk_key, 0); 11875f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot); 11885f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 11895f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 11905f39d397SChris Mason path->nodes[level] = left; 1191e66f709bSChris Mason path->slots[level + 1] -= 1; 1192e66f709bSChris Mason path->slots[level] = orig_slot; 1193925baeddSChris Mason btrfs_tree_unlock(mid); 11945f39d397SChris Mason free_extent_buffer(mid); 1195e66f709bSChris Mason } else { 1196e66f709bSChris Mason orig_slot -= 11975f39d397SChris Mason btrfs_header_nritems(left); 1198e66f709bSChris Mason path->slots[level] = orig_slot; 1199925baeddSChris Mason btrfs_tree_unlock(left); 12005f39d397SChris Mason free_extent_buffer(left); 1201e66f709bSChris Mason } 1202e66f709bSChris Mason return 0; 1203e66f709bSChris Mason } 1204925baeddSChris Mason btrfs_tree_unlock(left); 12055f39d397SChris Mason free_extent_buffer(left); 1206e66f709bSChris Mason } 12075f39d397SChris Mason right = read_node_slot(root, parent, pslot + 1); 1208e66f709bSChris Mason 1209e66f709bSChris Mason /* 1210e66f709bSChris Mason * then try to empty the right most buffer into the middle 1211e66f709bSChris Mason */ 12125f39d397SChris Mason if (right) { 121333ade1f8SChris Mason u32 right_nr; 1214925baeddSChris Mason btrfs_tree_lock(right); 12155f39d397SChris Mason right_nr = btrfs_header_nritems(right); 121633ade1f8SChris Mason if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 121733ade1f8SChris Mason wret = 1; 121833ade1f8SChris Mason } else { 12195f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, 12205f39d397SChris Mason parent, pslot + 1, 122165b51a00SChris Mason &right, 0); 122254aa1f4dSChris Mason if (ret) 122354aa1f4dSChris Mason wret = 1; 122454aa1f4dSChris Mason else { 122533ade1f8SChris Mason wret = balance_node_right(trans, root, 12265f39d397SChris Mason right, mid); 122733ade1f8SChris Mason } 122854aa1f4dSChris Mason } 1229e66f709bSChris Mason if (wret < 0) 1230e66f709bSChris Mason ret = wret; 1231e66f709bSChris Mason if (wret == 0) { 12325f39d397SChris Mason struct btrfs_disk_key disk_key; 12335f39d397SChris Mason 12345f39d397SChris Mason btrfs_node_key(right, &disk_key, 0); 12355f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot + 1); 12365f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 12375f39d397SChris Mason 12385f39d397SChris Mason if (btrfs_header_nritems(mid) <= orig_slot) { 12395f39d397SChris Mason path->nodes[level] = right; 1240e66f709bSChris Mason path->slots[level + 1] += 1; 1241e66f709bSChris Mason path->slots[level] = orig_slot - 12425f39d397SChris Mason btrfs_header_nritems(mid); 1243925baeddSChris Mason btrfs_tree_unlock(mid); 12445f39d397SChris Mason free_extent_buffer(mid); 1245e66f709bSChris Mason } else { 1246925baeddSChris Mason btrfs_tree_unlock(right); 12475f39d397SChris Mason free_extent_buffer(right); 1248e66f709bSChris Mason } 1249e66f709bSChris Mason return 0; 1250e66f709bSChris Mason } 1251925baeddSChris Mason btrfs_tree_unlock(right); 12525f39d397SChris Mason free_extent_buffer(right); 1253e66f709bSChris Mason } 1254e66f709bSChris Mason return 1; 1255e66f709bSChris Mason } 1256e66f709bSChris Mason 125774123bd7SChris Mason /* 1258d352ac68SChris Mason * readahead one full node of leaves, finding things that are close 1259d352ac68SChris Mason * to the block in 'slot', and triggering ra on them. 12603c69faecSChris Mason */ 1261e02119d5SChris Mason static noinline void reada_for_search(struct btrfs_root *root, 1262e02119d5SChris Mason struct btrfs_path *path, 126301f46658SChris Mason int level, int slot, u64 objectid) 12643c69faecSChris Mason { 12655f39d397SChris Mason struct extent_buffer *node; 126601f46658SChris Mason struct btrfs_disk_key disk_key; 12673c69faecSChris Mason u32 nritems; 12683c69faecSChris Mason u64 search; 12696b80053dSChris Mason u64 lowest_read; 12706b80053dSChris Mason u64 highest_read; 12716b80053dSChris Mason u64 nread = 0; 12723c69faecSChris Mason int direction = path->reada; 12735f39d397SChris Mason struct extent_buffer *eb; 12746b80053dSChris Mason u32 nr; 12756b80053dSChris Mason u32 blocksize; 12766b80053dSChris Mason u32 nscan = 0; 1277db94535dSChris Mason 1278a6b6e75eSChris Mason if (level != 1) 12793c69faecSChris Mason return; 12803c69faecSChris Mason 12816702ed49SChris Mason if (!path->nodes[level]) 12826702ed49SChris Mason return; 12836702ed49SChris Mason 12845f39d397SChris Mason node = path->nodes[level]; 1285925baeddSChris Mason 12863c69faecSChris Mason search = btrfs_node_blockptr(node, slot); 12876b80053dSChris Mason blocksize = btrfs_level_size(root, level - 1); 12886b80053dSChris Mason eb = btrfs_find_tree_block(root, search, blocksize); 12895f39d397SChris Mason if (eb) { 12905f39d397SChris Mason free_extent_buffer(eb); 12913c69faecSChris Mason return; 12923c69faecSChris Mason } 12933c69faecSChris Mason 12946b80053dSChris Mason highest_read = search; 12956b80053dSChris Mason lowest_read = search; 12966b80053dSChris Mason 12975f39d397SChris Mason nritems = btrfs_header_nritems(node); 12986b80053dSChris Mason nr = slot; 12993c69faecSChris Mason while(1) { 13006b80053dSChris Mason if (direction < 0) { 13016b80053dSChris Mason if (nr == 0) 13023c69faecSChris Mason break; 13036b80053dSChris Mason nr--; 13046b80053dSChris Mason } else if (direction > 0) { 13056b80053dSChris Mason nr++; 13066b80053dSChris Mason if (nr >= nritems) 13076b80053dSChris Mason break; 13083c69faecSChris Mason } 130901f46658SChris Mason if (path->reada < 0 && objectid) { 131001f46658SChris Mason btrfs_node_key(node, &disk_key, nr); 131101f46658SChris Mason if (btrfs_disk_key_objectid(&disk_key) != objectid) 131201f46658SChris Mason break; 131301f46658SChris Mason } 13146b80053dSChris Mason search = btrfs_node_blockptr(node, nr); 13156b80053dSChris Mason if ((search >= lowest_read && search <= highest_read) || 13166f3577bdSChris Mason (search < lowest_read && lowest_read - search <= 16384) || 13176f3577bdSChris Mason (search > highest_read && search - highest_read <= 16384)) { 1318ca7a79adSChris Mason readahead_tree_block(root, search, blocksize, 1319ca7a79adSChris Mason btrfs_node_ptr_generation(node, nr)); 13206b80053dSChris Mason nread += blocksize; 13213c69faecSChris Mason } 13226b80053dSChris Mason nscan++; 13236f3577bdSChris Mason if (path->reada < 2 && (nread > (64 * 1024) || nscan > 32)) 13246b80053dSChris Mason break; 13256f3577bdSChris Mason if(nread > (256 * 1024) || nscan > 128) 13266b80053dSChris Mason break; 13276b80053dSChris Mason 13286b80053dSChris Mason if (search < lowest_read) 13296b80053dSChris Mason lowest_read = search; 13306b80053dSChris Mason if (search > highest_read) 13316b80053dSChris Mason highest_read = search; 13323c69faecSChris Mason } 13333c69faecSChris Mason } 1334925baeddSChris Mason 1335d352ac68SChris Mason /* 1336d352ac68SChris Mason * when we walk down the tree, it is usually safe to unlock the higher layers in 1337d352ac68SChris Mason * the tree. The exceptions are when our path goes through slot 0, because operations 1338d352ac68SChris Mason * on the tree might require changing key pointers higher up in the tree. 1339d352ac68SChris Mason * 1340d352ac68SChris Mason * callers might also have set path->keep_locks, which tells this code to 1341d352ac68SChris Mason * keep the lock if the path points to the last slot in the block. This is 1342d352ac68SChris Mason * part of walking through the tree, and selecting the next slot in the higher 1343d352ac68SChris Mason * block. 1344d352ac68SChris Mason * 1345d352ac68SChris Mason * lowest_unlock sets the lowest level in the tree we're allowed to unlock. 1346d352ac68SChris Mason * so 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, 1349e02119d5SChris Mason int lowest_unlock) 1350925baeddSChris Mason { 1351925baeddSChris Mason int i; 1352925baeddSChris Mason int skip_level = level; 1353051e1b9fSChris Mason int no_skips = 0; 1354925baeddSChris Mason struct extent_buffer *t; 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; 1361051e1b9fSChris Mason if (!no_skips && path->slots[i] == 0) { 1362925baeddSChris Mason skip_level = i + 1; 1363925baeddSChris Mason continue; 1364925baeddSChris Mason } 1365051e1b9fSChris Mason if (!no_skips && path->keep_locks) { 1366925baeddSChris Mason u32 nritems; 1367925baeddSChris Mason t = path->nodes[i]; 1368925baeddSChris Mason nritems = btrfs_header_nritems(t); 1369051e1b9fSChris Mason if (nritems < 1 || path->slots[i] >= nritems - 1) { 1370925baeddSChris Mason skip_level = i + 1; 1371925baeddSChris Mason continue; 1372925baeddSChris Mason } 1373925baeddSChris Mason } 1374051e1b9fSChris Mason if (skip_level < i && i >= lowest_unlock) 1375051e1b9fSChris Mason no_skips = 1; 1376051e1b9fSChris Mason 1377925baeddSChris Mason t = path->nodes[i]; 1378925baeddSChris Mason if (i >= lowest_unlock && i > skip_level && path->locks[i]) { 1379925baeddSChris Mason btrfs_tree_unlock(t); 1380925baeddSChris Mason path->locks[i] = 0; 1381925baeddSChris Mason } 1382925baeddSChris Mason } 1383925baeddSChris Mason } 1384925baeddSChris Mason 13853c69faecSChris Mason /* 138674123bd7SChris Mason * look for key in the tree. path is filled in with nodes along the way 138774123bd7SChris Mason * if key is found, we return zero and you can find the item in the leaf 138874123bd7SChris Mason * level of the path (level 0) 138974123bd7SChris Mason * 139074123bd7SChris Mason * If the key isn't found, the path points to the slot where it should 1391aa5d6bedSChris Mason * be inserted, and 1 is returned. If there are other errors during the 1392aa5d6bedSChris Mason * search a negative error number is returned. 139397571fd0SChris Mason * 139497571fd0SChris Mason * if ins_len > 0, nodes and leaves will be split as we walk down the 139597571fd0SChris Mason * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if 139697571fd0SChris Mason * possible) 139774123bd7SChris Mason */ 1398e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root 1399e089f05cSChris Mason *root, struct btrfs_key *key, struct btrfs_path *p, int 1400e089f05cSChris Mason ins_len, int cow) 1401be0e5c09SChris Mason { 14025f39d397SChris Mason struct extent_buffer *b; 1403051e1b9fSChris Mason struct extent_buffer *tmp; 1404be0e5c09SChris Mason int slot; 1405be0e5c09SChris Mason int ret; 1406be0e5c09SChris Mason int level; 14073c69faecSChris Mason int should_reada = p->reada; 1408925baeddSChris Mason int lowest_unlock = 1; 1409594a24ebSChris Mason int blocksize; 14109f3a7427SChris Mason u8 lowest_level = 0; 1411594a24ebSChris Mason u64 blocknr; 1412594a24ebSChris Mason u64 gen; 141365b51a00SChris Mason struct btrfs_key prealloc_block; 14149f3a7427SChris Mason 14156702ed49SChris Mason lowest_level = p->lowest_level; 1416323ac95bSChris Mason WARN_ON(lowest_level && ins_len > 0); 141722b0ebdaSChris Mason WARN_ON(p->nodes[0] != NULL); 141825179201SJosef Bacik 1419925baeddSChris Mason if (ins_len < 0) 1420925baeddSChris Mason lowest_unlock = 2; 142165b51a00SChris Mason 142265b51a00SChris Mason prealloc_block.objectid = 0; 142365b51a00SChris Mason 1424bb803951SChris Mason again: 14255cd57b2cSChris Mason if (p->skip_locking) 14265cd57b2cSChris Mason b = btrfs_root_node(root); 14275cd57b2cSChris Mason else 1428925baeddSChris Mason b = btrfs_lock_root_node(root); 1429925baeddSChris Mason 1430eb60ceacSChris Mason while (b) { 14315f39d397SChris Mason level = btrfs_header_level(b); 143265b51a00SChris Mason 143365b51a00SChris Mason /* 143465b51a00SChris Mason * setup the path here so we can release it under lock 143565b51a00SChris Mason * contention with the cow code 143665b51a00SChris Mason */ 143765b51a00SChris Mason p->nodes[level] = b; 143865b51a00SChris Mason if (!p->skip_locking) 143965b51a00SChris Mason p->locks[level] = 1; 144065b51a00SChris Mason 144102217ed2SChris Mason if (cow) { 144202217ed2SChris Mason int wret; 144365b51a00SChris Mason 144465b51a00SChris Mason /* is a cow on this block not required */ 144565b51a00SChris Mason spin_lock(&root->fs_info->hash_lock); 144665b51a00SChris Mason if (btrfs_header_generation(b) == trans->transid && 14475b21f2edSZheng Yan btrfs_header_owner(b) == root->root_key.objectid && 144865b51a00SChris Mason !btrfs_header_flag(b, BTRFS_HEADER_FLAG_WRITTEN)) { 144965b51a00SChris Mason spin_unlock(&root->fs_info->hash_lock); 145065b51a00SChris Mason goto cow_done; 145165b51a00SChris Mason } 145265b51a00SChris Mason spin_unlock(&root->fs_info->hash_lock); 145365b51a00SChris Mason 145465b51a00SChris Mason /* ok, we have to cow, is our old prealloc the right 145565b51a00SChris Mason * size? 145665b51a00SChris Mason */ 145765b51a00SChris Mason if (prealloc_block.objectid && 145865b51a00SChris Mason prealloc_block.offset != b->len) { 145965b51a00SChris Mason btrfs_free_reserved_extent(root, 146065b51a00SChris Mason prealloc_block.objectid, 146165b51a00SChris Mason prealloc_block.offset); 146265b51a00SChris Mason prealloc_block.objectid = 0; 146365b51a00SChris Mason } 146465b51a00SChris Mason 146565b51a00SChris Mason /* 146665b51a00SChris Mason * for higher level blocks, try not to allocate blocks 146765b51a00SChris Mason * with the block and the parent locks held. 146865b51a00SChris Mason */ 146965b51a00SChris Mason if (level > 1 && !prealloc_block.objectid && 147065b51a00SChris Mason btrfs_path_lock_waiting(p, level)) { 147165b51a00SChris Mason u32 size = b->len; 147265b51a00SChris Mason u64 hint = b->start; 147365b51a00SChris Mason 147465b51a00SChris Mason btrfs_release_path(root, p); 147565b51a00SChris Mason ret = btrfs_reserve_extent(trans, root, 147665b51a00SChris Mason size, size, 0, 147765b51a00SChris Mason hint, (u64)-1, 147865b51a00SChris Mason &prealloc_block, 0); 147965b51a00SChris Mason BUG_ON(ret); 148065b51a00SChris Mason goto again; 148165b51a00SChris Mason } 148265b51a00SChris Mason 1483e20d96d6SChris Mason wret = btrfs_cow_block(trans, root, b, 1484e20d96d6SChris Mason p->nodes[level + 1], 1485e20d96d6SChris Mason p->slots[level + 1], 148665b51a00SChris Mason &b, prealloc_block.objectid); 148765b51a00SChris Mason prealloc_block.objectid = 0; 148854aa1f4dSChris Mason if (wret) { 14895f39d397SChris Mason free_extent_buffer(b); 149065b51a00SChris Mason ret = wret; 149165b51a00SChris Mason goto done; 149254aa1f4dSChris Mason } 149302217ed2SChris Mason } 149465b51a00SChris Mason cow_done: 149502217ed2SChris Mason BUG_ON(!cow && ins_len); 14965f39d397SChris Mason if (level != btrfs_header_level(b)) 14972c90e5d6SChris Mason WARN_ON(1); 14985f39d397SChris Mason level = btrfs_header_level(b); 149965b51a00SChris Mason 1500eb60ceacSChris Mason p->nodes[level] = b; 15015cd57b2cSChris Mason if (!p->skip_locking) 1502925baeddSChris Mason p->locks[level] = 1; 150365b51a00SChris Mason 1504123abc88SChris Mason ret = check_block(root, p, level); 150565b51a00SChris Mason if (ret) { 150665b51a00SChris Mason ret = -1; 150765b51a00SChris Mason goto done; 150865b51a00SChris Mason } 1509925baeddSChris Mason 15105f39d397SChris Mason ret = bin_search(b, key, level, &slot); 15115f39d397SChris Mason if (level != 0) { 1512be0e5c09SChris Mason if (ret && slot > 0) 1513be0e5c09SChris Mason slot -= 1; 1514be0e5c09SChris Mason p->slots[level] = slot; 1515459931ecSChris Mason if ((p->search_for_split || ins_len > 0) && 1516459931ecSChris Mason btrfs_header_nritems(b) >= 15171514794eSChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) { 1518e089f05cSChris Mason int sret = split_node(trans, root, p, level); 15195c680ed6SChris Mason BUG_ON(sret > 0); 152065b51a00SChris Mason if (sret) { 152165b51a00SChris Mason ret = sret; 152265b51a00SChris Mason goto done; 152365b51a00SChris Mason } 15245c680ed6SChris Mason b = p->nodes[level]; 15255c680ed6SChris Mason slot = p->slots[level]; 1526bb803951SChris Mason } else if (ins_len < 0) { 1527e089f05cSChris Mason int sret = balance_level(trans, root, p, 1528e089f05cSChris Mason level); 152965b51a00SChris Mason if (sret) { 153065b51a00SChris Mason ret = sret; 153165b51a00SChris Mason goto done; 153265b51a00SChris Mason } 1533bb803951SChris Mason b = p->nodes[level]; 1534f510cfecSChris Mason if (!b) { 1535f510cfecSChris Mason btrfs_release_path(NULL, p); 1536bb803951SChris Mason goto again; 1537f510cfecSChris Mason } 1538bb803951SChris Mason slot = p->slots[level]; 15395f39d397SChris Mason BUG_ON(btrfs_header_nritems(b) == 1); 15405c680ed6SChris Mason } 1541f9efa9c7SChris Mason unlock_up(p, level, lowest_unlock); 1542f9efa9c7SChris Mason 15439f3a7427SChris Mason /* this is only true while dropping a snapshot */ 1544925baeddSChris Mason if (level == lowest_level) { 15455b21f2edSZheng Yan ret = 0; 15465b21f2edSZheng Yan goto done; 1547925baeddSChris Mason } 1548ca7a79adSChris Mason 1549594a24ebSChris Mason blocknr = btrfs_node_blockptr(b, slot); 1550594a24ebSChris Mason gen = btrfs_node_ptr_generation(b, slot); 1551594a24ebSChris Mason blocksize = btrfs_level_size(root, level - 1); 1552594a24ebSChris Mason 1553594a24ebSChris Mason tmp = btrfs_find_tree_block(root, blocknr, blocksize); 1554594a24ebSChris Mason if (tmp && btrfs_buffer_uptodate(tmp, gen)) { 1555051e1b9fSChris Mason b = tmp; 1556051e1b9fSChris Mason } else { 1557051e1b9fSChris Mason /* 1558051e1b9fSChris Mason * reduce lock contention at high levels 1559051e1b9fSChris Mason * of the btree by dropping locks before 1560051e1b9fSChris Mason * we read. 1561051e1b9fSChris Mason */ 1562051e1b9fSChris Mason if (level > 1) { 1563051e1b9fSChris Mason btrfs_release_path(NULL, p); 1564051e1b9fSChris Mason if (tmp) 1565051e1b9fSChris Mason free_extent_buffer(tmp); 1566f9efa9c7SChris Mason if (should_reada) 1567f9efa9c7SChris Mason reada_for_search(root, p, 1568f9efa9c7SChris Mason level, slot, 1569f9efa9c7SChris Mason key->objectid); 1570f9efa9c7SChris Mason 1571594a24ebSChris Mason tmp = read_tree_block(root, blocknr, 1572594a24ebSChris Mason blocksize, gen); 1573594a24ebSChris Mason if (tmp) 1574594a24ebSChris Mason free_extent_buffer(tmp); 1575051e1b9fSChris Mason goto again; 1576051e1b9fSChris Mason } else { 1577a74a4b97SChris Mason if (tmp) 1578a74a4b97SChris Mason free_extent_buffer(tmp); 1579f9efa9c7SChris Mason if (should_reada) 1580f9efa9c7SChris Mason reada_for_search(root, p, 1581f9efa9c7SChris Mason level, slot, 1582f9efa9c7SChris Mason key->objectid); 1583ca7a79adSChris Mason b = read_node_slot(root, b, slot); 1584051e1b9fSChris Mason } 1585051e1b9fSChris Mason } 15865cd57b2cSChris Mason if (!p->skip_locking) 1587925baeddSChris Mason btrfs_tree_lock(b); 1588be0e5c09SChris Mason } else { 1589be0e5c09SChris Mason p->slots[level] = slot; 15905f39d397SChris Mason if (ins_len > 0 && btrfs_leaf_free_space(root, b) < 15910783fcfcSChris Mason sizeof(struct btrfs_item) + ins_len) { 1592d4dbff95SChris Mason int sret = split_leaf(trans, root, key, 1593cc0c5538SChris Mason p, ins_len, ret == 0); 15945c680ed6SChris Mason BUG_ON(sret > 0); 159565b51a00SChris Mason if (sret) { 159665b51a00SChris Mason ret = sret; 159765b51a00SChris Mason goto done; 159865b51a00SChris Mason } 15995c680ed6SChris Mason } 1600459931ecSChris Mason if (!p->search_for_split) 1601925baeddSChris Mason unlock_up(p, level, lowest_unlock); 160265b51a00SChris Mason goto done; 160365b51a00SChris Mason } 160465b51a00SChris Mason } 160565b51a00SChris Mason ret = 1; 160665b51a00SChris Mason done: 160765b51a00SChris Mason if (prealloc_block.objectid) { 160865b51a00SChris Mason btrfs_free_reserved_extent(root, 160965b51a00SChris Mason prealloc_block.objectid, 161065b51a00SChris Mason prealloc_block.offset); 161165b51a00SChris Mason } 161265b51a00SChris Mason 1613be0e5c09SChris Mason return ret; 1614be0e5c09SChris Mason } 1615be0e5c09SChris Mason 16161a40e23bSZheng Yan int btrfs_merge_path(struct btrfs_trans_handle *trans, 16171a40e23bSZheng Yan struct btrfs_root *root, 16181a40e23bSZheng Yan struct btrfs_key *node_keys, 16191a40e23bSZheng Yan u64 *nodes, int lowest_level) 16201a40e23bSZheng Yan { 16211a40e23bSZheng Yan struct extent_buffer *eb; 16221a40e23bSZheng Yan struct extent_buffer *parent; 16231a40e23bSZheng Yan struct btrfs_key key; 16241a40e23bSZheng Yan u64 bytenr; 16251a40e23bSZheng Yan u64 generation; 16261a40e23bSZheng Yan u32 blocksize; 16271a40e23bSZheng Yan int level; 16281a40e23bSZheng Yan int slot; 16291a40e23bSZheng Yan int key_match; 16301a40e23bSZheng Yan int ret; 16311a40e23bSZheng Yan 16321a40e23bSZheng Yan eb = btrfs_lock_root_node(root); 16331a40e23bSZheng Yan ret = btrfs_cow_block(trans, root, eb, NULL, 0, &eb, 0); 16341a40e23bSZheng Yan BUG_ON(ret); 16351a40e23bSZheng Yan 16361a40e23bSZheng Yan parent = eb; 16371a40e23bSZheng Yan while (1) { 16381a40e23bSZheng Yan level = btrfs_header_level(parent); 16391a40e23bSZheng Yan if (level == 0 || level <= lowest_level) 16401a40e23bSZheng Yan break; 16411a40e23bSZheng Yan 16421a40e23bSZheng Yan ret = bin_search(parent, &node_keys[lowest_level], level, 16431a40e23bSZheng Yan &slot); 16441a40e23bSZheng Yan if (ret && slot > 0) 16451a40e23bSZheng Yan slot--; 16461a40e23bSZheng Yan 16471a40e23bSZheng Yan bytenr = btrfs_node_blockptr(parent, slot); 16481a40e23bSZheng Yan if (nodes[level - 1] == bytenr) 16491a40e23bSZheng Yan break; 16501a40e23bSZheng Yan 16511a40e23bSZheng Yan blocksize = btrfs_level_size(root, level - 1); 16521a40e23bSZheng Yan generation = btrfs_node_ptr_generation(parent, slot); 16531a40e23bSZheng Yan btrfs_node_key_to_cpu(eb, &key, slot); 16541a40e23bSZheng Yan key_match = !memcmp(&key, &node_keys[level - 1], sizeof(key)); 16551a40e23bSZheng Yan 1656f82d02d9SYan Zheng if (generation == trans->transid) { 16571a40e23bSZheng Yan eb = read_tree_block(root, bytenr, blocksize, 16581a40e23bSZheng Yan generation); 16591a40e23bSZheng Yan btrfs_tree_lock(eb); 1660f82d02d9SYan Zheng } 1661f82d02d9SYan Zheng 1662f82d02d9SYan Zheng /* 1663f82d02d9SYan Zheng * if node keys match and node pointer hasn't been modified 1664f82d02d9SYan Zheng * in the running transaction, we can merge the path. for 1665f82d02d9SYan Zheng * blocks owened by reloc trees, the node pointer check is 1666f82d02d9SYan Zheng * skipped, this is because these blocks are fully controlled 1667f82d02d9SYan Zheng * by the space balance code, no one else can modify them. 1668f82d02d9SYan Zheng */ 1669f82d02d9SYan Zheng if (!nodes[level - 1] || !key_match || 1670f82d02d9SYan Zheng (generation == trans->transid && 1671f82d02d9SYan Zheng btrfs_header_owner(eb) != BTRFS_TREE_RELOC_OBJECTID)) { 1672f82d02d9SYan Zheng if (level == 1 || level == lowest_level + 1) { 1673f82d02d9SYan Zheng if (generation == trans->transid) { 1674f82d02d9SYan Zheng btrfs_tree_unlock(eb); 1675f82d02d9SYan Zheng free_extent_buffer(eb); 1676f82d02d9SYan Zheng } 1677f82d02d9SYan Zheng break; 1678f82d02d9SYan Zheng } 1679f82d02d9SYan Zheng 1680f82d02d9SYan Zheng if (generation != trans->transid) { 1681f82d02d9SYan Zheng eb = read_tree_block(root, bytenr, blocksize, 1682f82d02d9SYan Zheng generation); 1683f82d02d9SYan Zheng btrfs_tree_lock(eb); 1684f82d02d9SYan Zheng } 16851a40e23bSZheng Yan 16861a40e23bSZheng Yan ret = btrfs_cow_block(trans, root, eb, parent, slot, 16871a40e23bSZheng Yan &eb, 0); 16881a40e23bSZheng Yan BUG_ON(ret); 16891a40e23bSZheng Yan 1690f82d02d9SYan Zheng if (root->root_key.objectid == 1691f82d02d9SYan Zheng BTRFS_TREE_RELOC_OBJECTID) { 1692f82d02d9SYan Zheng if (!nodes[level - 1]) { 1693f82d02d9SYan Zheng nodes[level - 1] = eb->start; 1694f82d02d9SYan Zheng memcpy(&node_keys[level - 1], &key, 1695f82d02d9SYan Zheng sizeof(node_keys[0])); 1696f82d02d9SYan Zheng } else { 1697f82d02d9SYan Zheng WARN_ON(1); 1698f82d02d9SYan Zheng } 1699f82d02d9SYan Zheng } 1700f82d02d9SYan Zheng 17011a40e23bSZheng Yan btrfs_tree_unlock(parent); 17021a40e23bSZheng Yan free_extent_buffer(parent); 17031a40e23bSZheng Yan parent = eb; 17041a40e23bSZheng Yan continue; 17051a40e23bSZheng Yan } 17061a40e23bSZheng Yan 17071a40e23bSZheng Yan btrfs_set_node_blockptr(parent, slot, nodes[level - 1]); 17081a40e23bSZheng Yan btrfs_set_node_ptr_generation(parent, slot, trans->transid); 17091a40e23bSZheng Yan btrfs_mark_buffer_dirty(parent); 17101a40e23bSZheng Yan 17111a40e23bSZheng Yan ret = btrfs_inc_extent_ref(trans, root, 17121a40e23bSZheng Yan nodes[level - 1], 17131a40e23bSZheng Yan blocksize, parent->start, 17141a40e23bSZheng Yan btrfs_header_owner(parent), 17151a40e23bSZheng Yan btrfs_header_generation(parent), 17163bb1a1bcSYan Zheng level - 1); 17171a40e23bSZheng Yan BUG_ON(ret); 1718f82d02d9SYan Zheng 1719f82d02d9SYan Zheng /* 1720f82d02d9SYan Zheng * If the block was created in the running transaction, 1721f82d02d9SYan Zheng * it's possible this is the last reference to it, so we 1722f82d02d9SYan Zheng * should drop the subtree. 1723f82d02d9SYan Zheng */ 1724f82d02d9SYan Zheng if (generation == trans->transid) { 1725f82d02d9SYan Zheng ret = btrfs_drop_subtree(trans, root, eb, parent); 1726f82d02d9SYan Zheng BUG_ON(ret); 1727f82d02d9SYan Zheng btrfs_tree_unlock(eb); 1728f82d02d9SYan Zheng free_extent_buffer(eb); 1729f82d02d9SYan Zheng } else { 17301a40e23bSZheng Yan ret = btrfs_free_extent(trans, root, bytenr, 17311a40e23bSZheng Yan blocksize, parent->start, 17321a40e23bSZheng Yan btrfs_header_owner(parent), 17331a40e23bSZheng Yan btrfs_header_generation(parent), 17343bb1a1bcSYan Zheng level - 1, 1); 17351a40e23bSZheng Yan BUG_ON(ret); 17361a40e23bSZheng Yan } 17371a40e23bSZheng Yan break; 17381a40e23bSZheng Yan } 17391a40e23bSZheng Yan btrfs_tree_unlock(parent); 17401a40e23bSZheng Yan free_extent_buffer(parent); 17411a40e23bSZheng Yan return 0; 17421a40e23bSZheng Yan } 17431a40e23bSZheng Yan 174474123bd7SChris Mason /* 174574123bd7SChris Mason * adjust the pointers going up the tree, starting at level 174674123bd7SChris Mason * making sure the right key of each node is points to 'key'. 174774123bd7SChris Mason * This is used after shifting pointers to the left, so it stops 174874123bd7SChris Mason * fixing up pointers when a given leaf/node is not in slot 0 of the 174974123bd7SChris Mason * higher levels 1750aa5d6bedSChris Mason * 1751aa5d6bedSChris Mason * If this fails to write a tree block, it returns -1, but continues 1752aa5d6bedSChris Mason * fixing up the blocks in ram so the tree is consistent. 175374123bd7SChris Mason */ 17545f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans, 17555f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 17565f39d397SChris Mason struct btrfs_disk_key *key, int level) 1757be0e5c09SChris Mason { 1758be0e5c09SChris Mason int i; 1759aa5d6bedSChris Mason int ret = 0; 17605f39d397SChris Mason struct extent_buffer *t; 17615f39d397SChris Mason 1762234b63a0SChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 1763be0e5c09SChris Mason int tslot = path->slots[i]; 1764eb60ceacSChris Mason if (!path->nodes[i]) 1765be0e5c09SChris Mason break; 17665f39d397SChris Mason t = path->nodes[i]; 17675f39d397SChris Mason btrfs_set_node_key(t, key, tslot); 1768d6025579SChris Mason btrfs_mark_buffer_dirty(path->nodes[i]); 1769be0e5c09SChris Mason if (tslot != 0) 1770be0e5c09SChris Mason break; 1771be0e5c09SChris Mason } 1772aa5d6bedSChris Mason return ret; 1773be0e5c09SChris Mason } 1774be0e5c09SChris Mason 177574123bd7SChris Mason /* 177631840ae1SZheng Yan * update item key. 177731840ae1SZheng Yan * 177831840ae1SZheng Yan * This function isn't completely safe. It's the caller's responsibility 177931840ae1SZheng Yan * that the new key won't break the order 178031840ae1SZheng Yan */ 178131840ae1SZheng Yan int btrfs_set_item_key_safe(struct btrfs_trans_handle *trans, 178231840ae1SZheng Yan struct btrfs_root *root, struct btrfs_path *path, 178331840ae1SZheng Yan struct btrfs_key *new_key) 178431840ae1SZheng Yan { 178531840ae1SZheng Yan struct btrfs_disk_key disk_key; 178631840ae1SZheng Yan struct extent_buffer *eb; 178731840ae1SZheng Yan int slot; 178831840ae1SZheng Yan 178931840ae1SZheng Yan eb = path->nodes[0]; 179031840ae1SZheng Yan slot = path->slots[0]; 179131840ae1SZheng Yan if (slot > 0) { 179231840ae1SZheng Yan btrfs_item_key(eb, &disk_key, slot - 1); 179331840ae1SZheng Yan if (comp_keys(&disk_key, new_key) >= 0) 179431840ae1SZheng Yan return -1; 179531840ae1SZheng Yan } 179631840ae1SZheng Yan if (slot < btrfs_header_nritems(eb) - 1) { 179731840ae1SZheng Yan btrfs_item_key(eb, &disk_key, slot + 1); 179831840ae1SZheng Yan if (comp_keys(&disk_key, new_key) <= 0) 179931840ae1SZheng Yan return -1; 180031840ae1SZheng Yan } 180131840ae1SZheng Yan 180231840ae1SZheng Yan btrfs_cpu_key_to_disk(&disk_key, new_key); 180331840ae1SZheng Yan btrfs_set_item_key(eb, &disk_key, slot); 180431840ae1SZheng Yan btrfs_mark_buffer_dirty(eb); 180531840ae1SZheng Yan if (slot == 0) 180631840ae1SZheng Yan fixup_low_keys(trans, root, path, &disk_key, 1); 180731840ae1SZheng Yan return 0; 180831840ae1SZheng Yan } 180931840ae1SZheng Yan 181031840ae1SZheng Yan /* 181174123bd7SChris Mason * try to push data from one node into the next node left in the 181279f95c82SChris Mason * tree. 1813aa5d6bedSChris Mason * 1814aa5d6bedSChris Mason * returns 0 if some ptrs were pushed left, < 0 if there was some horrible 1815aa5d6bedSChris Mason * error, and > 0 if there was no room in the left hand block. 181674123bd7SChris Mason */ 181798ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans, 181898ed5174SChris Mason struct btrfs_root *root, struct extent_buffer *dst, 1819971a1f66SChris Mason struct extent_buffer *src, int empty) 1820be0e5c09SChris Mason { 1821be0e5c09SChris Mason int push_items = 0; 1822bb803951SChris Mason int src_nritems; 1823bb803951SChris Mason int dst_nritems; 1824aa5d6bedSChris Mason int ret = 0; 1825be0e5c09SChris Mason 18265f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 18275f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1828123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 18297bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 18307bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 183154aa1f4dSChris Mason 1832bce4eae9SChris Mason if (!empty && src_nritems <= 8) 1833971a1f66SChris Mason return 1; 1834971a1f66SChris Mason 1835eb60ceacSChris Mason if (push_items <= 0) { 1836be0e5c09SChris Mason return 1; 1837eb60ceacSChris Mason } 1838be0e5c09SChris Mason 1839bce4eae9SChris Mason if (empty) { 1840971a1f66SChris Mason push_items = min(src_nritems, push_items); 1841bce4eae9SChris Mason if (push_items < src_nritems) { 1842bce4eae9SChris Mason /* leave at least 8 pointers in the node if 1843bce4eae9SChris Mason * we aren't going to empty it 1844bce4eae9SChris Mason */ 1845bce4eae9SChris Mason if (src_nritems - push_items < 8) { 1846bce4eae9SChris Mason if (push_items <= 8) 1847bce4eae9SChris Mason return 1; 1848bce4eae9SChris Mason push_items -= 8; 1849bce4eae9SChris Mason } 1850bce4eae9SChris Mason } 1851bce4eae9SChris Mason } else 1852bce4eae9SChris Mason push_items = min(src_nritems - 8, push_items); 185379f95c82SChris Mason 18545f39d397SChris Mason copy_extent_buffer(dst, src, 18555f39d397SChris Mason btrfs_node_key_ptr_offset(dst_nritems), 18565f39d397SChris Mason btrfs_node_key_ptr_offset(0), 1857123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 18585f39d397SChris Mason 1859bb803951SChris Mason if (push_items < src_nritems) { 18605f39d397SChris Mason memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0), 18615f39d397SChris Mason btrfs_node_key_ptr_offset(push_items), 1862e2fa7227SChris Mason (src_nritems - push_items) * 1863123abc88SChris Mason sizeof(struct btrfs_key_ptr)); 1864bb803951SChris Mason } 18655f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 18665f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 18675f39d397SChris Mason btrfs_mark_buffer_dirty(src); 18685f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 186931840ae1SZheng Yan 187031840ae1SZheng Yan ret = btrfs_update_ref(trans, root, src, dst, dst_nritems, push_items); 187131840ae1SZheng Yan BUG_ON(ret); 187231840ae1SZheng Yan 1873bb803951SChris Mason return ret; 1874be0e5c09SChris Mason } 1875be0e5c09SChris Mason 187697571fd0SChris Mason /* 187779f95c82SChris Mason * try to push data from one node into the next node right in the 187879f95c82SChris Mason * tree. 187979f95c82SChris Mason * 188079f95c82SChris Mason * returns 0 if some ptrs were pushed, < 0 if there was some horrible 188179f95c82SChris Mason * error, and > 0 if there was no room in the right hand block. 188279f95c82SChris Mason * 188379f95c82SChris Mason * this will only push up to 1/2 the contents of the left node over 188479f95c82SChris Mason */ 18855f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 18865f39d397SChris Mason struct btrfs_root *root, 18875f39d397SChris Mason struct extent_buffer *dst, 18885f39d397SChris Mason struct extent_buffer *src) 188979f95c82SChris Mason { 189079f95c82SChris Mason int push_items = 0; 189179f95c82SChris Mason int max_push; 189279f95c82SChris Mason int src_nritems; 189379f95c82SChris Mason int dst_nritems; 189479f95c82SChris Mason int ret = 0; 189579f95c82SChris Mason 18967bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 18977bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 18987bb86316SChris Mason 18995f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 19005f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1901123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 1902bce4eae9SChris Mason if (push_items <= 0) { 190379f95c82SChris Mason return 1; 1904bce4eae9SChris Mason } 1905bce4eae9SChris Mason 1906bce4eae9SChris Mason if (src_nritems < 4) { 1907bce4eae9SChris Mason return 1; 1908bce4eae9SChris Mason } 190979f95c82SChris Mason 191079f95c82SChris Mason max_push = src_nritems / 2 + 1; 191179f95c82SChris Mason /* don't try to empty the node */ 1912bce4eae9SChris Mason if (max_push >= src_nritems) { 191379f95c82SChris Mason return 1; 1914bce4eae9SChris Mason } 1915252c38f0SYan 191679f95c82SChris Mason if (max_push < push_items) 191779f95c82SChris Mason push_items = max_push; 191879f95c82SChris Mason 19195f39d397SChris Mason memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items), 19205f39d397SChris Mason btrfs_node_key_ptr_offset(0), 19215f39d397SChris Mason (dst_nritems) * 19225f39d397SChris Mason sizeof(struct btrfs_key_ptr)); 1923d6025579SChris Mason 19245f39d397SChris Mason copy_extent_buffer(dst, src, 19255f39d397SChris Mason btrfs_node_key_ptr_offset(0), 19265f39d397SChris Mason btrfs_node_key_ptr_offset(src_nritems - push_items), 1927123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 192879f95c82SChris Mason 19295f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 19305f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 193179f95c82SChris Mason 19325f39d397SChris Mason btrfs_mark_buffer_dirty(src); 19335f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 193431840ae1SZheng Yan 193531840ae1SZheng Yan ret = btrfs_update_ref(trans, root, src, dst, 0, push_items); 193631840ae1SZheng Yan BUG_ON(ret); 193731840ae1SZheng Yan 193879f95c82SChris Mason return ret; 193979f95c82SChris Mason } 194079f95c82SChris Mason 194179f95c82SChris Mason /* 194297571fd0SChris Mason * helper function to insert a new root level in the tree. 194397571fd0SChris Mason * A new node is allocated, and a single item is inserted to 194497571fd0SChris Mason * point to the existing root 1945aa5d6bedSChris Mason * 1946aa5d6bedSChris Mason * returns zero on success or < 0 on failure. 194797571fd0SChris Mason */ 194898ed5174SChris Mason static int noinline insert_new_root(struct btrfs_trans_handle *trans, 19495f39d397SChris Mason struct btrfs_root *root, 19505f39d397SChris Mason struct btrfs_path *path, int level) 195174123bd7SChris Mason { 19527bb86316SChris Mason u64 lower_gen; 19535f39d397SChris Mason struct extent_buffer *lower; 19545f39d397SChris Mason struct extent_buffer *c; 1955925baeddSChris Mason struct extent_buffer *old; 19565f39d397SChris Mason struct btrfs_disk_key lower_key; 195731840ae1SZheng Yan int ret; 19585c680ed6SChris Mason 19595c680ed6SChris Mason BUG_ON(path->nodes[level]); 19605c680ed6SChris Mason BUG_ON(path->nodes[level-1] != root->node); 19615c680ed6SChris Mason 19627bb86316SChris Mason lower = path->nodes[level-1]; 19637bb86316SChris Mason if (level == 1) 19647bb86316SChris Mason btrfs_item_key(lower, &lower_key, 0); 19657bb86316SChris Mason else 19667bb86316SChris Mason btrfs_node_key(lower, &lower_key, 0); 19677bb86316SChris Mason 196831840ae1SZheng Yan c = btrfs_alloc_free_block(trans, root, root->nodesize, 0, 196931840ae1SZheng Yan root->root_key.objectid, trans->transid, 1970ad3d81baSChristoph Hellwig level, root->node->start, 0); 19715f39d397SChris Mason if (IS_ERR(c)) 19725f39d397SChris Mason return PTR_ERR(c); 1973925baeddSChris Mason 19745f39d397SChris Mason memset_extent_buffer(c, 0, 0, root->nodesize); 19755f39d397SChris Mason btrfs_set_header_nritems(c, 1); 19765f39d397SChris Mason btrfs_set_header_level(c, level); 1977db94535dSChris Mason btrfs_set_header_bytenr(c, c->start); 19785f39d397SChris Mason btrfs_set_header_generation(c, trans->transid); 19795f39d397SChris Mason btrfs_set_header_owner(c, root->root_key.objectid); 1980d5719762SChris Mason 19815f39d397SChris Mason write_extent_buffer(c, root->fs_info->fsid, 19825f39d397SChris Mason (unsigned long)btrfs_header_fsid(c), 19835f39d397SChris Mason BTRFS_FSID_SIZE); 1984e17cade2SChris Mason 1985e17cade2SChris Mason write_extent_buffer(c, root->fs_info->chunk_tree_uuid, 1986e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(c), 1987e17cade2SChris Mason BTRFS_UUID_SIZE); 1988e17cade2SChris Mason 19895f39d397SChris Mason btrfs_set_node_key(c, &lower_key, 0); 1990db94535dSChris Mason btrfs_set_node_blockptr(c, 0, lower->start); 19917bb86316SChris Mason lower_gen = btrfs_header_generation(lower); 199231840ae1SZheng Yan WARN_ON(lower_gen != trans->transid); 19937bb86316SChris Mason 19947bb86316SChris Mason btrfs_set_node_ptr_generation(c, 0, lower_gen); 19955f39d397SChris Mason 19965f39d397SChris Mason btrfs_mark_buffer_dirty(c); 1997d5719762SChris Mason 1998925baeddSChris Mason spin_lock(&root->node_lock); 1999925baeddSChris Mason old = root->node; 20005f39d397SChris Mason root->node = c; 2001925baeddSChris Mason spin_unlock(&root->node_lock); 2002925baeddSChris Mason 200331840ae1SZheng Yan ret = btrfs_update_extent_ref(trans, root, lower->start, 200431840ae1SZheng Yan lower->start, c->start, 200531840ae1SZheng Yan root->root_key.objectid, 20063bb1a1bcSYan Zheng trans->transid, level - 1); 200731840ae1SZheng Yan BUG_ON(ret); 200831840ae1SZheng Yan 2009925baeddSChris Mason /* the super has an extra ref to root->node */ 2010925baeddSChris Mason free_extent_buffer(old); 2011925baeddSChris Mason 20120b86a832SChris Mason add_root_to_dirty_list(root); 20135f39d397SChris Mason extent_buffer_get(c); 20145f39d397SChris Mason path->nodes[level] = c; 2015925baeddSChris Mason path->locks[level] = 1; 201674123bd7SChris Mason path->slots[level] = 0; 201774123bd7SChris Mason return 0; 201874123bd7SChris Mason } 20195c680ed6SChris Mason 20205c680ed6SChris Mason /* 20215c680ed6SChris Mason * worker function to insert a single pointer in a node. 20225c680ed6SChris Mason * the node should have enough room for the pointer already 202397571fd0SChris Mason * 20245c680ed6SChris Mason * slot and level indicate where you want the key to go, and 20255c680ed6SChris Mason * blocknr is the block the key points to. 2026aa5d6bedSChris Mason * 2027aa5d6bedSChris Mason * returns zero on success and < 0 on any error 20285c680ed6SChris Mason */ 2029e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root 2030e089f05cSChris Mason *root, struct btrfs_path *path, struct btrfs_disk_key 2031db94535dSChris Mason *key, u64 bytenr, int slot, int level) 20325c680ed6SChris Mason { 20335f39d397SChris Mason struct extent_buffer *lower; 20345c680ed6SChris Mason int nritems; 20355c680ed6SChris Mason 20365c680ed6SChris Mason BUG_ON(!path->nodes[level]); 20375f39d397SChris Mason lower = path->nodes[level]; 20385f39d397SChris Mason nritems = btrfs_header_nritems(lower); 203974123bd7SChris Mason if (slot > nritems) 204074123bd7SChris Mason BUG(); 2041123abc88SChris Mason if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root)) 204274123bd7SChris Mason BUG(); 204374123bd7SChris Mason if (slot != nritems) { 20445f39d397SChris Mason memmove_extent_buffer(lower, 20455f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 20465f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 2047123abc88SChris Mason (nritems - slot) * sizeof(struct btrfs_key_ptr)); 204874123bd7SChris Mason } 20495f39d397SChris Mason btrfs_set_node_key(lower, key, slot); 2050db94535dSChris Mason btrfs_set_node_blockptr(lower, slot, bytenr); 205174493f7aSChris Mason WARN_ON(trans->transid == 0); 205274493f7aSChris Mason btrfs_set_node_ptr_generation(lower, slot, trans->transid); 20535f39d397SChris Mason btrfs_set_header_nritems(lower, nritems + 1); 20545f39d397SChris Mason btrfs_mark_buffer_dirty(lower); 205574123bd7SChris Mason return 0; 205674123bd7SChris Mason } 205774123bd7SChris Mason 205897571fd0SChris Mason /* 205997571fd0SChris Mason * split the node at the specified level in path in two. 206097571fd0SChris Mason * The path is corrected to point to the appropriate node after the split 206197571fd0SChris Mason * 206297571fd0SChris Mason * Before splitting this tries to make some room in the node by pushing 206397571fd0SChris Mason * left and right, if either one works, it returns right away. 2064aa5d6bedSChris Mason * 2065aa5d6bedSChris Mason * returns 0 on success and < 0 on failure 206697571fd0SChris Mason */ 2067e02119d5SChris Mason static noinline int split_node(struct btrfs_trans_handle *trans, 2068e02119d5SChris Mason struct btrfs_root *root, 2069e02119d5SChris Mason struct btrfs_path *path, int level) 2070be0e5c09SChris Mason { 20715f39d397SChris Mason struct extent_buffer *c; 20725f39d397SChris Mason struct extent_buffer *split; 20735f39d397SChris Mason struct btrfs_disk_key disk_key; 2074be0e5c09SChris Mason int mid; 20755c680ed6SChris Mason int ret; 2076aa5d6bedSChris Mason int wret; 20777518a238SChris Mason u32 c_nritems; 2078be0e5c09SChris Mason 20795f39d397SChris Mason c = path->nodes[level]; 20807bb86316SChris Mason WARN_ON(btrfs_header_generation(c) != trans->transid); 20815f39d397SChris Mason if (c == root->node) { 20825c680ed6SChris Mason /* trying to split the root, lets make a new one */ 2083e089f05cSChris Mason ret = insert_new_root(trans, root, path, level + 1); 20845c680ed6SChris Mason if (ret) 20855c680ed6SChris Mason return ret; 2086e66f709bSChris Mason } else { 2087e66f709bSChris Mason ret = push_nodes_for_insert(trans, root, path, level); 20885f39d397SChris Mason c = path->nodes[level]; 20895f39d397SChris Mason if (!ret && btrfs_header_nritems(c) < 2090c448acf0SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) 2091e66f709bSChris Mason return 0; 209254aa1f4dSChris Mason if (ret < 0) 209354aa1f4dSChris Mason return ret; 20945c680ed6SChris Mason } 2095e66f709bSChris Mason 20965f39d397SChris Mason c_nritems = btrfs_header_nritems(c); 20977bb86316SChris Mason 2098925baeddSChris Mason split = btrfs_alloc_free_block(trans, root, root->nodesize, 209931840ae1SZheng Yan path->nodes[level + 1]->start, 21007bb86316SChris Mason root->root_key.objectid, 210131840ae1SZheng Yan trans->transid, level, c->start, 0); 21025f39d397SChris Mason if (IS_ERR(split)) 21035f39d397SChris Mason return PTR_ERR(split); 210454aa1f4dSChris Mason 21055f39d397SChris Mason btrfs_set_header_flags(split, btrfs_header_flags(c)); 21065f39d397SChris Mason btrfs_set_header_level(split, btrfs_header_level(c)); 2107db94535dSChris Mason btrfs_set_header_bytenr(split, split->start); 21085f39d397SChris Mason btrfs_set_header_generation(split, trans->transid); 21095f39d397SChris Mason btrfs_set_header_owner(split, root->root_key.objectid); 211063b10fc4SChris Mason btrfs_set_header_flags(split, 0); 21115f39d397SChris Mason write_extent_buffer(split, root->fs_info->fsid, 21125f39d397SChris Mason (unsigned long)btrfs_header_fsid(split), 21135f39d397SChris Mason BTRFS_FSID_SIZE); 2114e17cade2SChris Mason write_extent_buffer(split, root->fs_info->chunk_tree_uuid, 2115e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(split), 2116e17cade2SChris Mason BTRFS_UUID_SIZE); 21175f39d397SChris Mason 21187518a238SChris Mason mid = (c_nritems + 1) / 2; 21195f39d397SChris Mason 21205f39d397SChris Mason copy_extent_buffer(split, c, 21215f39d397SChris Mason btrfs_node_key_ptr_offset(0), 21225f39d397SChris Mason btrfs_node_key_ptr_offset(mid), 2123123abc88SChris Mason (c_nritems - mid) * sizeof(struct btrfs_key_ptr)); 21245f39d397SChris Mason btrfs_set_header_nritems(split, c_nritems - mid); 21255f39d397SChris Mason btrfs_set_header_nritems(c, mid); 2126aa5d6bedSChris Mason ret = 0; 2127aa5d6bedSChris Mason 21285f39d397SChris Mason btrfs_mark_buffer_dirty(c); 21295f39d397SChris Mason btrfs_mark_buffer_dirty(split); 21305f39d397SChris Mason 21315f39d397SChris Mason btrfs_node_key(split, &disk_key, 0); 2132db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, split->start, 21335f39d397SChris Mason path->slots[level + 1] + 1, 2134123abc88SChris Mason level + 1); 2135aa5d6bedSChris Mason if (wret) 2136aa5d6bedSChris Mason ret = wret; 2137aa5d6bedSChris Mason 213831840ae1SZheng Yan ret = btrfs_update_ref(trans, root, c, split, 0, c_nritems - mid); 213931840ae1SZheng Yan BUG_ON(ret); 214031840ae1SZheng Yan 21415de08d7dSChris Mason if (path->slots[level] >= mid) { 21425c680ed6SChris Mason path->slots[level] -= mid; 2143925baeddSChris Mason btrfs_tree_unlock(c); 21445f39d397SChris Mason free_extent_buffer(c); 21455f39d397SChris Mason path->nodes[level] = split; 21465c680ed6SChris Mason path->slots[level + 1] += 1; 2147eb60ceacSChris Mason } else { 2148925baeddSChris Mason btrfs_tree_unlock(split); 21495f39d397SChris Mason free_extent_buffer(split); 2150be0e5c09SChris Mason } 2151aa5d6bedSChris Mason return ret; 2152be0e5c09SChris Mason } 2153be0e5c09SChris Mason 215474123bd7SChris Mason /* 215574123bd7SChris Mason * how many bytes are required to store the items in a leaf. start 215674123bd7SChris Mason * and nr indicate which items in the leaf to check. This totals up the 215774123bd7SChris Mason * space used both by the item structs and the item data 215874123bd7SChris Mason */ 21595f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr) 2160be0e5c09SChris Mason { 2161be0e5c09SChris Mason int data_len; 21625f39d397SChris Mason int nritems = btrfs_header_nritems(l); 2163d4dbff95SChris Mason int end = min(nritems, start + nr) - 1; 2164be0e5c09SChris Mason 2165be0e5c09SChris Mason if (!nr) 2166be0e5c09SChris Mason return 0; 21675f39d397SChris Mason data_len = btrfs_item_end_nr(l, start); 21685f39d397SChris Mason data_len = data_len - btrfs_item_offset_nr(l, end); 21690783fcfcSChris Mason data_len += sizeof(struct btrfs_item) * nr; 2170d4dbff95SChris Mason WARN_ON(data_len < 0); 2171be0e5c09SChris Mason return data_len; 2172be0e5c09SChris Mason } 2173be0e5c09SChris Mason 217474123bd7SChris Mason /* 2175d4dbff95SChris Mason * The space between the end of the leaf items and 2176d4dbff95SChris Mason * the start of the leaf data. IOW, how much room 2177d4dbff95SChris Mason * the leaf has left for both items and data 2178d4dbff95SChris Mason */ 2179e02119d5SChris Mason int noinline btrfs_leaf_free_space(struct btrfs_root *root, 2180e02119d5SChris Mason struct extent_buffer *leaf) 2181d4dbff95SChris Mason { 21825f39d397SChris Mason int nritems = btrfs_header_nritems(leaf); 21835f39d397SChris Mason int ret; 21845f39d397SChris Mason ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems); 21855f39d397SChris Mason if (ret < 0) { 21865f39d397SChris Mason printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n", 2187ae2f5411SJens Axboe ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root), 21885f39d397SChris Mason leaf_space_used(leaf, 0, nritems), nritems); 21895f39d397SChris Mason } 21905f39d397SChris Mason return ret; 2191d4dbff95SChris Mason } 2192d4dbff95SChris Mason 2193d4dbff95SChris Mason /* 219400ec4c51SChris Mason * push some data in the path leaf to the right, trying to free up at 219500ec4c51SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 2196aa5d6bedSChris Mason * 2197aa5d6bedSChris Mason * returns 1 if the push failed because the other node didn't have enough 2198aa5d6bedSChris Mason * room, 0 if everything worked out and < 0 if there were major errors. 219900ec4c51SChris Mason */ 2200e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root 220134a38218SChris Mason *root, struct btrfs_path *path, int data_size, 220234a38218SChris Mason int empty) 220300ec4c51SChris Mason { 22045f39d397SChris Mason struct extent_buffer *left = path->nodes[0]; 22055f39d397SChris Mason struct extent_buffer *right; 22065f39d397SChris Mason struct extent_buffer *upper; 22075f39d397SChris Mason struct btrfs_disk_key disk_key; 220800ec4c51SChris Mason int slot; 220934a38218SChris Mason u32 i; 221000ec4c51SChris Mason int free_space; 221100ec4c51SChris Mason int push_space = 0; 221200ec4c51SChris Mason int push_items = 0; 22130783fcfcSChris Mason struct btrfs_item *item; 22147518a238SChris Mason u32 left_nritems; 221534a38218SChris Mason u32 nr; 22167518a238SChris Mason u32 right_nritems; 22175f39d397SChris Mason u32 data_end; 2218db94535dSChris Mason u32 this_item_size; 221954aa1f4dSChris Mason int ret; 222000ec4c51SChris Mason 222100ec4c51SChris Mason slot = path->slots[1]; 222200ec4c51SChris Mason if (!path->nodes[1]) { 222300ec4c51SChris Mason return 1; 222400ec4c51SChris Mason } 222500ec4c51SChris Mason upper = path->nodes[1]; 22265f39d397SChris Mason if (slot >= btrfs_header_nritems(upper) - 1) 222700ec4c51SChris Mason return 1; 22285f39d397SChris Mason 2229a2135011SChris Mason WARN_ON(!btrfs_tree_locked(path->nodes[1])); 2230a2135011SChris Mason 2231ca7a79adSChris Mason right = read_node_slot(root, upper, slot + 1); 2232925baeddSChris Mason btrfs_tree_lock(right); 2233123abc88SChris Mason free_space = btrfs_leaf_free_space(root, right); 2234925baeddSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) 2235925baeddSChris Mason goto out_unlock; 223602217ed2SChris Mason 22375f39d397SChris Mason /* cow and double check */ 22385f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, upper, 223965b51a00SChris Mason slot + 1, &right, 0); 2240925baeddSChris Mason if (ret) 2241925baeddSChris Mason goto out_unlock; 2242925baeddSChris Mason 22435f39d397SChris Mason free_space = btrfs_leaf_free_space(root, right); 2244925baeddSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) 2245925baeddSChris Mason goto out_unlock; 22465f39d397SChris Mason 22475f39d397SChris Mason left_nritems = btrfs_header_nritems(left); 2248925baeddSChris Mason if (left_nritems == 0) 2249925baeddSChris Mason goto out_unlock; 22505f39d397SChris Mason 225134a38218SChris Mason if (empty) 225234a38218SChris Mason nr = 0; 225334a38218SChris Mason else 225434a38218SChris Mason nr = 1; 225534a38218SChris Mason 225631840ae1SZheng Yan if (path->slots[0] >= left_nritems) 225731840ae1SZheng Yan push_space += data_size + sizeof(*item); 225831840ae1SZheng Yan 225934a38218SChris Mason i = left_nritems - 1; 226034a38218SChris Mason while (i >= nr) { 22615f39d397SChris Mason item = btrfs_item_nr(left, i); 2262db94535dSChris Mason 226331840ae1SZheng Yan if (!empty && push_items > 0) { 226431840ae1SZheng Yan if (path->slots[0] > i) 226531840ae1SZheng Yan break; 226631840ae1SZheng Yan if (path->slots[0] == i) { 226731840ae1SZheng Yan int space = btrfs_leaf_free_space(root, left); 226831840ae1SZheng Yan if (space + push_space * 2 > free_space) 226931840ae1SZheng Yan break; 227031840ae1SZheng Yan } 227131840ae1SZheng Yan } 227231840ae1SZheng Yan 227300ec4c51SChris Mason if (path->slots[0] == i) 227400ec4c51SChris Mason push_space += data_size + sizeof(*item); 2275db94535dSChris Mason 2276db94535dSChris Mason if (!left->map_token) { 2277db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 2278db94535dSChris Mason sizeof(struct btrfs_item), 2279db94535dSChris Mason &left->map_token, &left->kaddr, 2280db94535dSChris Mason &left->map_start, &left->map_len, 2281db94535dSChris Mason KM_USER1); 2282db94535dSChris Mason } 2283db94535dSChris Mason 2284db94535dSChris Mason this_item_size = btrfs_item_size(left, item); 2285db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 228600ec4c51SChris Mason break; 228731840ae1SZheng Yan 228800ec4c51SChris Mason push_items++; 2289db94535dSChris Mason push_space += this_item_size + sizeof(*item); 229034a38218SChris Mason if (i == 0) 229134a38218SChris Mason break; 229234a38218SChris Mason i--; 2293db94535dSChris Mason } 2294db94535dSChris Mason if (left->map_token) { 2295db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 2296db94535dSChris Mason left->map_token = NULL; 229700ec4c51SChris Mason } 22985f39d397SChris Mason 2299925baeddSChris Mason if (push_items == 0) 2300925baeddSChris Mason goto out_unlock; 23015f39d397SChris Mason 230234a38218SChris Mason if (!empty && push_items == left_nritems) 2303a429e513SChris Mason WARN_ON(1); 23045f39d397SChris Mason 230500ec4c51SChris Mason /* push left to right */ 23065f39d397SChris Mason right_nritems = btrfs_header_nritems(right); 230734a38218SChris Mason 23085f39d397SChris Mason push_space = btrfs_item_end_nr(left, left_nritems - push_items); 2309123abc88SChris Mason push_space -= leaf_data_end(root, left); 23105f39d397SChris Mason 231100ec4c51SChris Mason /* make room in the right data area */ 23125f39d397SChris Mason data_end = leaf_data_end(root, right); 23135f39d397SChris Mason memmove_extent_buffer(right, 23145f39d397SChris Mason btrfs_leaf_data(right) + data_end - push_space, 23155f39d397SChris Mason btrfs_leaf_data(right) + data_end, 23165f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root) - data_end); 23175f39d397SChris Mason 231800ec4c51SChris Mason /* copy from the left data area */ 23195f39d397SChris Mason copy_extent_buffer(right, left, btrfs_leaf_data(right) + 2320d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 2321d6025579SChris Mason btrfs_leaf_data(left) + leaf_data_end(root, left), 2322d6025579SChris Mason push_space); 23235f39d397SChris Mason 23245f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(push_items), 23255f39d397SChris Mason btrfs_item_nr_offset(0), 23260783fcfcSChris Mason right_nritems * sizeof(struct btrfs_item)); 23275f39d397SChris Mason 232800ec4c51SChris Mason /* copy the items from left to right */ 23295f39d397SChris Mason copy_extent_buffer(right, left, btrfs_item_nr_offset(0), 23305f39d397SChris Mason btrfs_item_nr_offset(left_nritems - push_items), 23310783fcfcSChris Mason push_items * sizeof(struct btrfs_item)); 233200ec4c51SChris Mason 233300ec4c51SChris Mason /* update the item pointers */ 23347518a238SChris Mason right_nritems += push_items; 23355f39d397SChris Mason btrfs_set_header_nritems(right, right_nritems); 2336123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 23377518a238SChris Mason for (i = 0; i < right_nritems; i++) { 23385f39d397SChris Mason item = btrfs_item_nr(right, i); 2339db94535dSChris Mason if (!right->map_token) { 2340db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2341db94535dSChris Mason sizeof(struct btrfs_item), 2342db94535dSChris Mason &right->map_token, &right->kaddr, 2343db94535dSChris Mason &right->map_start, &right->map_len, 2344db94535dSChris Mason KM_USER1); 2345db94535dSChris Mason } 2346db94535dSChris Mason push_space -= btrfs_item_size(right, item); 2347db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 2348db94535dSChris Mason } 2349db94535dSChris Mason 2350db94535dSChris Mason if (right->map_token) { 2351db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2352db94535dSChris Mason right->map_token = NULL; 235300ec4c51SChris Mason } 23547518a238SChris Mason left_nritems -= push_items; 23555f39d397SChris Mason btrfs_set_header_nritems(left, left_nritems); 235600ec4c51SChris Mason 235734a38218SChris Mason if (left_nritems) 23585f39d397SChris Mason btrfs_mark_buffer_dirty(left); 23595f39d397SChris Mason btrfs_mark_buffer_dirty(right); 2360a429e513SChris Mason 236131840ae1SZheng Yan ret = btrfs_update_ref(trans, root, left, right, 0, push_items); 236231840ae1SZheng Yan BUG_ON(ret); 236331840ae1SZheng Yan 23645f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 23655f39d397SChris Mason btrfs_set_node_key(upper, &disk_key, slot + 1); 2366d6025579SChris Mason btrfs_mark_buffer_dirty(upper); 236702217ed2SChris Mason 236800ec4c51SChris Mason /* then fixup the leaf pointer in the path */ 23697518a238SChris Mason if (path->slots[0] >= left_nritems) { 23707518a238SChris Mason path->slots[0] -= left_nritems; 2371925baeddSChris Mason if (btrfs_header_nritems(path->nodes[0]) == 0) 2372925baeddSChris Mason clean_tree_block(trans, root, path->nodes[0]); 2373925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 23745f39d397SChris Mason free_extent_buffer(path->nodes[0]); 23755f39d397SChris Mason path->nodes[0] = right; 237600ec4c51SChris Mason path->slots[1] += 1; 237700ec4c51SChris Mason } else { 2378925baeddSChris Mason btrfs_tree_unlock(right); 23795f39d397SChris Mason free_extent_buffer(right); 238000ec4c51SChris Mason } 238100ec4c51SChris Mason return 0; 2382925baeddSChris Mason 2383925baeddSChris Mason out_unlock: 2384925baeddSChris Mason btrfs_tree_unlock(right); 2385925baeddSChris Mason free_extent_buffer(right); 2386925baeddSChris Mason return 1; 238700ec4c51SChris Mason } 2388925baeddSChris Mason 238900ec4c51SChris Mason /* 239074123bd7SChris Mason * push some data in the path leaf to the left, trying to free up at 239174123bd7SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 239274123bd7SChris Mason */ 2393e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root 239434a38218SChris Mason *root, struct btrfs_path *path, int data_size, 239534a38218SChris Mason int empty) 2396be0e5c09SChris Mason { 23975f39d397SChris Mason struct btrfs_disk_key disk_key; 23985f39d397SChris Mason struct extent_buffer *right = path->nodes[0]; 23995f39d397SChris Mason struct extent_buffer *left; 2400be0e5c09SChris Mason int slot; 2401be0e5c09SChris Mason int i; 2402be0e5c09SChris Mason int free_space; 2403be0e5c09SChris Mason int push_space = 0; 2404be0e5c09SChris Mason int push_items = 0; 24050783fcfcSChris Mason struct btrfs_item *item; 24067518a238SChris Mason u32 old_left_nritems; 24075f39d397SChris Mason u32 right_nritems; 240834a38218SChris Mason u32 nr; 2409aa5d6bedSChris Mason int ret = 0; 2410aa5d6bedSChris Mason int wret; 2411db94535dSChris Mason u32 this_item_size; 2412db94535dSChris Mason u32 old_left_item_size; 2413be0e5c09SChris Mason 2414be0e5c09SChris Mason slot = path->slots[1]; 24155f39d397SChris Mason if (slot == 0) 2416be0e5c09SChris Mason return 1; 24175f39d397SChris Mason if (!path->nodes[1]) 2418be0e5c09SChris Mason return 1; 24195f39d397SChris Mason 24203685f791SChris Mason right_nritems = btrfs_header_nritems(right); 24213685f791SChris Mason if (right_nritems == 0) { 24223685f791SChris Mason return 1; 24233685f791SChris Mason } 24243685f791SChris Mason 2425a2135011SChris Mason WARN_ON(!btrfs_tree_locked(path->nodes[1])); 2426a2135011SChris Mason 2427ca7a79adSChris Mason left = read_node_slot(root, path->nodes[1], slot - 1); 2428925baeddSChris Mason btrfs_tree_lock(left); 2429123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 24300783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 2431925baeddSChris Mason ret = 1; 2432925baeddSChris Mason goto out; 2433be0e5c09SChris Mason } 243402217ed2SChris Mason 243502217ed2SChris Mason /* cow and double check */ 24365f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, 243765b51a00SChris Mason path->nodes[1], slot - 1, &left, 0); 243854aa1f4dSChris Mason if (ret) { 243954aa1f4dSChris Mason /* we hit -ENOSPC, but it isn't fatal here */ 2440925baeddSChris Mason ret = 1; 2441925baeddSChris Mason goto out; 244254aa1f4dSChris Mason } 24433685f791SChris Mason 2444123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 24450783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 2446925baeddSChris Mason ret = 1; 2447925baeddSChris Mason goto out; 244802217ed2SChris Mason } 244902217ed2SChris Mason 245034a38218SChris Mason if (empty) 245134a38218SChris Mason nr = right_nritems; 245234a38218SChris Mason else 245334a38218SChris Mason nr = right_nritems - 1; 245434a38218SChris Mason 245534a38218SChris Mason for (i = 0; i < nr; i++) { 24565f39d397SChris Mason item = btrfs_item_nr(right, i); 2457db94535dSChris Mason if (!right->map_token) { 2458db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2459db94535dSChris Mason sizeof(struct btrfs_item), 2460db94535dSChris Mason &right->map_token, &right->kaddr, 2461db94535dSChris Mason &right->map_start, &right->map_len, 2462db94535dSChris Mason KM_USER1); 2463db94535dSChris Mason } 2464db94535dSChris Mason 246531840ae1SZheng Yan if (!empty && push_items > 0) { 246631840ae1SZheng Yan if (path->slots[0] < i) 246731840ae1SZheng Yan break; 246831840ae1SZheng Yan if (path->slots[0] == i) { 246931840ae1SZheng Yan int space = btrfs_leaf_free_space(root, right); 247031840ae1SZheng Yan if (space + push_space * 2 > free_space) 247131840ae1SZheng Yan break; 247231840ae1SZheng Yan } 247331840ae1SZheng Yan } 247431840ae1SZheng Yan 2475be0e5c09SChris Mason if (path->slots[0] == i) 2476be0e5c09SChris Mason push_space += data_size + sizeof(*item); 2477db94535dSChris Mason 2478db94535dSChris Mason this_item_size = btrfs_item_size(right, item); 2479db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 2480be0e5c09SChris Mason break; 2481db94535dSChris Mason 2482be0e5c09SChris Mason push_items++; 2483db94535dSChris Mason push_space += this_item_size + sizeof(*item); 2484be0e5c09SChris Mason } 2485db94535dSChris Mason 2486db94535dSChris Mason if (right->map_token) { 2487db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2488db94535dSChris Mason right->map_token = NULL; 2489db94535dSChris Mason } 2490db94535dSChris Mason 2491be0e5c09SChris Mason if (push_items == 0) { 2492925baeddSChris Mason ret = 1; 2493925baeddSChris Mason goto out; 2494be0e5c09SChris Mason } 249534a38218SChris Mason if (!empty && push_items == btrfs_header_nritems(right)) 2496a429e513SChris Mason WARN_ON(1); 24975f39d397SChris Mason 2498be0e5c09SChris Mason /* push data from right to left */ 24995f39d397SChris Mason copy_extent_buffer(left, right, 25005f39d397SChris Mason btrfs_item_nr_offset(btrfs_header_nritems(left)), 25015f39d397SChris Mason btrfs_item_nr_offset(0), 25025f39d397SChris Mason push_items * sizeof(struct btrfs_item)); 25035f39d397SChris Mason 2504123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root) - 25055f39d397SChris Mason btrfs_item_offset_nr(right, push_items -1); 25065f39d397SChris Mason 25075f39d397SChris Mason copy_extent_buffer(left, right, btrfs_leaf_data(left) + 2508d6025579SChris Mason leaf_data_end(root, left) - push_space, 2509123abc88SChris Mason btrfs_leaf_data(right) + 25105f39d397SChris Mason btrfs_item_offset_nr(right, push_items - 1), 2511be0e5c09SChris Mason push_space); 25125f39d397SChris Mason old_left_nritems = btrfs_header_nritems(left); 2513eb60ceacSChris Mason BUG_ON(old_left_nritems < 0); 2514eb60ceacSChris Mason 2515db94535dSChris Mason old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1); 2516be0e5c09SChris Mason for (i = old_left_nritems; i < old_left_nritems + push_items; i++) { 25175f39d397SChris Mason u32 ioff; 2518db94535dSChris Mason 25195f39d397SChris Mason item = btrfs_item_nr(left, i); 2520db94535dSChris Mason if (!left->map_token) { 2521db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 2522db94535dSChris Mason sizeof(struct btrfs_item), 2523db94535dSChris Mason &left->map_token, &left->kaddr, 2524db94535dSChris Mason &left->map_start, &left->map_len, 2525db94535dSChris Mason KM_USER1); 2526db94535dSChris Mason } 2527db94535dSChris Mason 25285f39d397SChris Mason ioff = btrfs_item_offset(left, item); 25295f39d397SChris Mason btrfs_set_item_offset(left, item, 2530db94535dSChris Mason ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size)); 2531be0e5c09SChris Mason } 25325f39d397SChris Mason btrfs_set_header_nritems(left, old_left_nritems + push_items); 2533db94535dSChris Mason if (left->map_token) { 2534db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 2535db94535dSChris Mason left->map_token = NULL; 2536db94535dSChris Mason } 2537be0e5c09SChris Mason 2538be0e5c09SChris Mason /* fixup right node */ 253934a38218SChris Mason if (push_items > right_nritems) { 254034a38218SChris Mason printk("push items %d nr %u\n", push_items, right_nritems); 254134a38218SChris Mason WARN_ON(1); 254234a38218SChris Mason } 254334a38218SChris Mason 254434a38218SChris Mason if (push_items < right_nritems) { 25455f39d397SChris Mason push_space = btrfs_item_offset_nr(right, push_items - 1) - 2546123abc88SChris Mason leaf_data_end(root, right); 25475f39d397SChris Mason memmove_extent_buffer(right, btrfs_leaf_data(right) + 2548d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 2549d6025579SChris Mason btrfs_leaf_data(right) + 2550123abc88SChris Mason leaf_data_end(root, right), push_space); 25515f39d397SChris Mason 25525f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(0), 25535f39d397SChris Mason btrfs_item_nr_offset(push_items), 25545f39d397SChris Mason (btrfs_header_nritems(right) - push_items) * 25550783fcfcSChris Mason sizeof(struct btrfs_item)); 255634a38218SChris Mason } 2557eef1c494SYan right_nritems -= push_items; 2558eef1c494SYan btrfs_set_header_nritems(right, right_nritems); 2559123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 25605f39d397SChris Mason for (i = 0; i < right_nritems; i++) { 25615f39d397SChris Mason item = btrfs_item_nr(right, i); 2562db94535dSChris Mason 2563db94535dSChris Mason if (!right->map_token) { 2564db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2565db94535dSChris Mason sizeof(struct btrfs_item), 2566db94535dSChris Mason &right->map_token, &right->kaddr, 2567db94535dSChris Mason &right->map_start, &right->map_len, 2568db94535dSChris Mason KM_USER1); 2569db94535dSChris Mason } 2570db94535dSChris Mason 2571db94535dSChris Mason push_space = push_space - btrfs_item_size(right, item); 2572db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 2573db94535dSChris Mason } 2574db94535dSChris Mason if (right->map_token) { 2575db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2576db94535dSChris Mason right->map_token = NULL; 2577be0e5c09SChris Mason } 2578eb60ceacSChris Mason 25795f39d397SChris Mason btrfs_mark_buffer_dirty(left); 258034a38218SChris Mason if (right_nritems) 25815f39d397SChris Mason btrfs_mark_buffer_dirty(right); 2582098f59c2SChris Mason 258331840ae1SZheng Yan ret = btrfs_update_ref(trans, root, right, left, 258431840ae1SZheng Yan old_left_nritems, push_items); 258531840ae1SZheng Yan BUG_ON(ret); 258631840ae1SZheng Yan 25875f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 25885f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, 1); 2589aa5d6bedSChris Mason if (wret) 2590aa5d6bedSChris Mason ret = wret; 2591be0e5c09SChris Mason 2592be0e5c09SChris Mason /* then fixup the leaf pointer in the path */ 2593be0e5c09SChris Mason if (path->slots[0] < push_items) { 2594be0e5c09SChris Mason path->slots[0] += old_left_nritems; 2595925baeddSChris Mason if (btrfs_header_nritems(path->nodes[0]) == 0) 2596925baeddSChris Mason clean_tree_block(trans, root, path->nodes[0]); 2597925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 25985f39d397SChris Mason free_extent_buffer(path->nodes[0]); 25995f39d397SChris Mason path->nodes[0] = left; 2600be0e5c09SChris Mason path->slots[1] -= 1; 2601be0e5c09SChris Mason } else { 2602925baeddSChris Mason btrfs_tree_unlock(left); 26035f39d397SChris Mason free_extent_buffer(left); 2604be0e5c09SChris Mason path->slots[0] -= push_items; 2605be0e5c09SChris Mason } 2606eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 2607aa5d6bedSChris Mason return ret; 2608925baeddSChris Mason out: 2609925baeddSChris Mason btrfs_tree_unlock(left); 2610925baeddSChris Mason free_extent_buffer(left); 2611925baeddSChris Mason return ret; 2612be0e5c09SChris Mason } 2613be0e5c09SChris Mason 261474123bd7SChris Mason /* 261574123bd7SChris Mason * split the path's leaf in two, making sure there is at least data_size 261674123bd7SChris Mason * available for the resulting leaf level of the path. 2617aa5d6bedSChris Mason * 2618aa5d6bedSChris Mason * returns 0 if all went well and < 0 on failure. 261974123bd7SChris Mason */ 2620e02119d5SChris Mason static noinline int split_leaf(struct btrfs_trans_handle *trans, 2621e02119d5SChris Mason struct btrfs_root *root, 2622e02119d5SChris Mason struct btrfs_key *ins_key, 2623e02119d5SChris Mason struct btrfs_path *path, int data_size, 2624e02119d5SChris Mason int extend) 2625be0e5c09SChris Mason { 26265f39d397SChris Mason struct extent_buffer *l; 26277518a238SChris Mason u32 nritems; 2628eb60ceacSChris Mason int mid; 2629eb60ceacSChris Mason int slot; 26305f39d397SChris Mason struct extent_buffer *right; 26310783fcfcSChris Mason int space_needed = data_size + sizeof(struct btrfs_item); 2632be0e5c09SChris Mason int data_copy_size; 2633be0e5c09SChris Mason int rt_data_off; 2634be0e5c09SChris Mason int i; 2635d4dbff95SChris Mason int ret = 0; 2636aa5d6bedSChris Mason int wret; 2637cc0c5538SChris Mason int double_split; 2638cc0c5538SChris Mason int num_doubles = 0; 2639d4dbff95SChris Mason struct btrfs_disk_key disk_key; 2640be0e5c09SChris Mason 2641459931ecSChris Mason if (extend && data_size) 2642cc0c5538SChris Mason space_needed = data_size; 2643cc0c5538SChris Mason 264440689478SChris Mason /* first try to make some room by pushing left and right */ 2645459931ecSChris Mason if (data_size && ins_key->type != BTRFS_DIR_ITEM_KEY) { 264634a38218SChris Mason wret = push_leaf_right(trans, root, path, data_size, 0); 26473326d1b0SChris Mason if (wret < 0) { 2648eaee50e8SChris Mason return wret; 26493326d1b0SChris Mason } 2650eaee50e8SChris Mason if (wret) { 265134a38218SChris Mason wret = push_leaf_left(trans, root, path, data_size, 0); 2652eaee50e8SChris Mason if (wret < 0) 2653eaee50e8SChris Mason return wret; 2654eaee50e8SChris Mason } 26555f39d397SChris Mason l = path->nodes[0]; 2656aa5d6bedSChris Mason 2657aa5d6bedSChris Mason /* did the pushes work? */ 2658cc0c5538SChris Mason if (btrfs_leaf_free_space(root, l) >= space_needed) 2659be0e5c09SChris Mason return 0; 26603326d1b0SChris Mason } 2661aa5d6bedSChris Mason 26625c680ed6SChris Mason if (!path->nodes[1]) { 2663e089f05cSChris Mason ret = insert_new_root(trans, root, path, 1); 26645c680ed6SChris Mason if (ret) 26655c680ed6SChris Mason return ret; 26665c680ed6SChris Mason } 2667cc0c5538SChris Mason again: 2668cc0c5538SChris Mason double_split = 0; 2669cc0c5538SChris Mason l = path->nodes[0]; 2670eb60ceacSChris Mason slot = path->slots[0]; 26715f39d397SChris Mason nritems = btrfs_header_nritems(l); 2672eb60ceacSChris Mason mid = (nritems + 1)/ 2; 267354aa1f4dSChris Mason 2674925baeddSChris Mason right = btrfs_alloc_free_block(trans, root, root->leafsize, 267531840ae1SZheng Yan path->nodes[1]->start, 26767bb86316SChris Mason root->root_key.objectid, 267731840ae1SZheng Yan trans->transid, 0, l->start, 0); 2678cea9e445SChris Mason if (IS_ERR(right)) { 2679cea9e445SChris Mason BUG_ON(1); 26805f39d397SChris Mason return PTR_ERR(right); 2681cea9e445SChris Mason } 268254aa1f4dSChris Mason 26835f39d397SChris Mason memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header)); 2684db94535dSChris Mason btrfs_set_header_bytenr(right, right->start); 26855f39d397SChris Mason btrfs_set_header_generation(right, trans->transid); 26865f39d397SChris Mason btrfs_set_header_owner(right, root->root_key.objectid); 26875f39d397SChris Mason btrfs_set_header_level(right, 0); 26885f39d397SChris Mason write_extent_buffer(right, root->fs_info->fsid, 26895f39d397SChris Mason (unsigned long)btrfs_header_fsid(right), 26905f39d397SChris Mason BTRFS_FSID_SIZE); 2691e17cade2SChris Mason 2692e17cade2SChris Mason write_extent_buffer(right, root->fs_info->chunk_tree_uuid, 2693e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(right), 2694e17cade2SChris Mason BTRFS_UUID_SIZE); 2695d4dbff95SChris Mason if (mid <= slot) { 2696d4dbff95SChris Mason if (nritems == 1 || 2697d4dbff95SChris Mason leaf_space_used(l, mid, nritems - mid) + space_needed > 2698d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 2699d4dbff95SChris Mason if (slot >= nritems) { 2700d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 27015f39d397SChris Mason btrfs_set_header_nritems(right, 0); 2702d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 2703db94535dSChris Mason &disk_key, right->start, 2704d4dbff95SChris Mason path->slots[1] + 1, 1); 2705d4dbff95SChris Mason if (wret) 2706d4dbff95SChris Mason ret = wret; 2707925baeddSChris Mason 2708925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 27095f39d397SChris Mason free_extent_buffer(path->nodes[0]); 27105f39d397SChris Mason path->nodes[0] = right; 2711d4dbff95SChris Mason path->slots[0] = 0; 2712d4dbff95SChris Mason path->slots[1] += 1; 27130ef8b242SChris Mason btrfs_mark_buffer_dirty(right); 2714d4dbff95SChris Mason return ret; 2715d4dbff95SChris Mason } 2716d4dbff95SChris Mason mid = slot; 27173326d1b0SChris Mason if (mid != nritems && 27183326d1b0SChris Mason leaf_space_used(l, mid, nritems - mid) + 27193326d1b0SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 2720d4dbff95SChris Mason double_split = 1; 2721d4dbff95SChris Mason } 27223326d1b0SChris Mason } 2723d4dbff95SChris Mason } else { 2724d4dbff95SChris Mason if (leaf_space_used(l, 0, mid + 1) + space_needed > 2725d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 2726459931ecSChris Mason if (!extend && data_size && slot == 0) { 2727d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 27285f39d397SChris Mason btrfs_set_header_nritems(right, 0); 2729d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 2730d4dbff95SChris Mason &disk_key, 2731db94535dSChris Mason right->start, 2732098f59c2SChris Mason path->slots[1], 1); 2733d4dbff95SChris Mason if (wret) 2734d4dbff95SChris Mason ret = wret; 2735925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 27365f39d397SChris Mason free_extent_buffer(path->nodes[0]); 27375f39d397SChris Mason path->nodes[0] = right; 2738d4dbff95SChris Mason path->slots[0] = 0; 2739a429e513SChris Mason if (path->slots[1] == 0) { 2740a429e513SChris Mason wret = fixup_low_keys(trans, root, 2741a429e513SChris Mason path, &disk_key, 1); 2742a429e513SChris Mason if (wret) 2743a429e513SChris Mason ret = wret; 2744a429e513SChris Mason } 27450ef8b242SChris Mason btrfs_mark_buffer_dirty(right); 2746d4dbff95SChris Mason return ret; 2747459931ecSChris Mason } else if ((extend || !data_size) && slot == 0) { 2748cc0c5538SChris Mason mid = 1; 2749cc0c5538SChris Mason } else { 2750d4dbff95SChris Mason mid = slot; 27515ee78ac7SChris Mason if (mid != nritems && 27525ee78ac7SChris Mason leaf_space_used(l, mid, nritems - mid) + 27535ee78ac7SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 2754d4dbff95SChris Mason double_split = 1; 2755d4dbff95SChris Mason } 2756d4dbff95SChris Mason } 27575ee78ac7SChris Mason } 2758cc0c5538SChris Mason } 27595f39d397SChris Mason nritems = nritems - mid; 27605f39d397SChris Mason btrfs_set_header_nritems(right, nritems); 27615f39d397SChris Mason data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l); 27625f39d397SChris Mason 27635f39d397SChris Mason copy_extent_buffer(right, l, btrfs_item_nr_offset(0), 27645f39d397SChris Mason btrfs_item_nr_offset(mid), 27655f39d397SChris Mason nritems * sizeof(struct btrfs_item)); 27665f39d397SChris Mason 27675f39d397SChris Mason copy_extent_buffer(right, l, 2768d6025579SChris Mason btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - 2769123abc88SChris Mason data_copy_size, btrfs_leaf_data(l) + 2770123abc88SChris Mason leaf_data_end(root, l), data_copy_size); 277174123bd7SChris Mason 27725f39d397SChris Mason rt_data_off = BTRFS_LEAF_DATA_SIZE(root) - 27735f39d397SChris Mason btrfs_item_end_nr(l, mid); 27745f39d397SChris Mason 27755f39d397SChris Mason for (i = 0; i < nritems; i++) { 27765f39d397SChris Mason struct btrfs_item *item = btrfs_item_nr(right, i); 2777db94535dSChris Mason u32 ioff; 2778db94535dSChris Mason 2779db94535dSChris Mason if (!right->map_token) { 2780db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2781db94535dSChris Mason sizeof(struct btrfs_item), 2782db94535dSChris Mason &right->map_token, &right->kaddr, 2783db94535dSChris Mason &right->map_start, &right->map_len, 2784db94535dSChris Mason KM_USER1); 2785db94535dSChris Mason } 2786db94535dSChris Mason 2787db94535dSChris Mason ioff = btrfs_item_offset(right, item); 27885f39d397SChris Mason btrfs_set_item_offset(right, item, ioff + rt_data_off); 27890783fcfcSChris Mason } 279074123bd7SChris Mason 2791db94535dSChris Mason if (right->map_token) { 2792db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2793db94535dSChris Mason right->map_token = NULL; 2794db94535dSChris Mason } 2795db94535dSChris Mason 27965f39d397SChris Mason btrfs_set_header_nritems(l, mid); 2797aa5d6bedSChris Mason ret = 0; 27985f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 2799db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, right->start, 2800db94535dSChris Mason path->slots[1] + 1, 1); 2801aa5d6bedSChris Mason if (wret) 2802aa5d6bedSChris Mason ret = wret; 28035f39d397SChris Mason 28045f39d397SChris Mason btrfs_mark_buffer_dirty(right); 28055f39d397SChris Mason btrfs_mark_buffer_dirty(l); 2806eb60ceacSChris Mason BUG_ON(path->slots[0] != slot); 28075f39d397SChris Mason 280831840ae1SZheng Yan ret = btrfs_update_ref(trans, root, l, right, 0, nritems); 280931840ae1SZheng Yan BUG_ON(ret); 281031840ae1SZheng Yan 2811be0e5c09SChris Mason if (mid <= slot) { 2812925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 28135f39d397SChris Mason free_extent_buffer(path->nodes[0]); 28145f39d397SChris Mason path->nodes[0] = right; 2815be0e5c09SChris Mason path->slots[0] -= mid; 2816be0e5c09SChris Mason path->slots[1] += 1; 2817925baeddSChris Mason } else { 2818925baeddSChris Mason btrfs_tree_unlock(right); 28195f39d397SChris Mason free_extent_buffer(right); 2820925baeddSChris Mason } 28215f39d397SChris Mason 2822eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 2823d4dbff95SChris Mason 2824cc0c5538SChris Mason if (double_split) { 2825cc0c5538SChris Mason BUG_ON(num_doubles != 0); 2826cc0c5538SChris Mason num_doubles++; 2827cc0c5538SChris Mason goto again; 28283326d1b0SChris Mason } 2829be0e5c09SChris Mason return ret; 2830be0e5c09SChris Mason } 2831be0e5c09SChris Mason 2832d352ac68SChris Mason /* 2833459931ecSChris Mason * This function splits a single item into two items, 2834459931ecSChris Mason * giving 'new_key' to the new item and splitting the 2835459931ecSChris Mason * old one at split_offset (from the start of the item). 2836459931ecSChris Mason * 2837459931ecSChris Mason * The path may be released by this operation. After 2838459931ecSChris Mason * the split, the path is pointing to the old item. The 2839459931ecSChris Mason * new item is going to be in the same node as the old one. 2840459931ecSChris Mason * 2841459931ecSChris Mason * Note, the item being split must be smaller enough to live alone on 2842459931ecSChris Mason * a tree block with room for one extra struct btrfs_item 2843459931ecSChris Mason * 2844459931ecSChris Mason * This allows us to split the item in place, keeping a lock on the 2845459931ecSChris Mason * leaf the entire time. 2846459931ecSChris Mason */ 2847459931ecSChris Mason int btrfs_split_item(struct btrfs_trans_handle *trans, 2848459931ecSChris Mason struct btrfs_root *root, 2849459931ecSChris Mason struct btrfs_path *path, 2850459931ecSChris Mason struct btrfs_key *new_key, 2851459931ecSChris Mason unsigned long split_offset) 2852459931ecSChris Mason { 2853459931ecSChris Mason u32 item_size; 2854459931ecSChris Mason struct extent_buffer *leaf; 2855459931ecSChris Mason struct btrfs_key orig_key; 2856459931ecSChris Mason struct btrfs_item *item; 2857459931ecSChris Mason struct btrfs_item *new_item; 2858459931ecSChris Mason int ret = 0; 2859459931ecSChris Mason int slot; 2860459931ecSChris Mason u32 nritems; 2861459931ecSChris Mason u32 orig_offset; 2862459931ecSChris Mason struct btrfs_disk_key disk_key; 2863459931ecSChris Mason char *buf; 2864459931ecSChris Mason 2865459931ecSChris Mason leaf = path->nodes[0]; 2866459931ecSChris Mason btrfs_item_key_to_cpu(leaf, &orig_key, path->slots[0]); 2867459931ecSChris Mason if (btrfs_leaf_free_space(root, leaf) >= sizeof(struct btrfs_item)) 2868459931ecSChris Mason goto split; 2869459931ecSChris Mason 2870459931ecSChris Mason item_size = btrfs_item_size_nr(leaf, path->slots[0]); 2871459931ecSChris Mason btrfs_release_path(root, path); 2872459931ecSChris Mason 2873459931ecSChris Mason path->search_for_split = 1; 2874459931ecSChris Mason path->keep_locks = 1; 2875459931ecSChris Mason 2876459931ecSChris Mason ret = btrfs_search_slot(trans, root, &orig_key, path, 0, 1); 2877459931ecSChris Mason path->search_for_split = 0; 2878459931ecSChris Mason 2879459931ecSChris Mason /* if our item isn't there or got smaller, return now */ 2880459931ecSChris Mason if (ret != 0 || item_size != btrfs_item_size_nr(path->nodes[0], 2881459931ecSChris Mason path->slots[0])) { 2882459931ecSChris Mason path->keep_locks = 0; 2883459931ecSChris Mason return -EAGAIN; 2884459931ecSChris Mason } 2885459931ecSChris Mason 2886459931ecSChris Mason ret = split_leaf(trans, root, &orig_key, path, 0, 0); 2887459931ecSChris Mason path->keep_locks = 0; 2888459931ecSChris Mason BUG_ON(ret); 2889459931ecSChris Mason 2890459931ecSChris Mason leaf = path->nodes[0]; 2891*42dc7babSChris Mason BUG_ON(btrfs_leaf_free_space(root, leaf) < sizeof(struct btrfs_item)); 2892459931ecSChris Mason 2893459931ecSChris Mason split: 2894459931ecSChris Mason item = btrfs_item_nr(leaf, path->slots[0]); 2895459931ecSChris Mason orig_offset = btrfs_item_offset(leaf, item); 2896459931ecSChris Mason item_size = btrfs_item_size(leaf, item); 2897459931ecSChris Mason 2898459931ecSChris Mason 2899459931ecSChris Mason buf = kmalloc(item_size, GFP_NOFS); 2900459931ecSChris Mason read_extent_buffer(leaf, buf, btrfs_item_ptr_offset(leaf, 2901459931ecSChris Mason path->slots[0]), item_size); 2902459931ecSChris Mason slot = path->slots[0] + 1; 2903459931ecSChris Mason leaf = path->nodes[0]; 2904459931ecSChris Mason 2905459931ecSChris Mason nritems = btrfs_header_nritems(leaf); 2906459931ecSChris Mason 2907459931ecSChris Mason if (slot != nritems) { 2908459931ecSChris Mason /* shift the items */ 2909459931ecSChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1), 2910459931ecSChris Mason btrfs_item_nr_offset(slot), 2911459931ecSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 2912459931ecSChris Mason 2913459931ecSChris Mason } 2914459931ecSChris Mason 2915459931ecSChris Mason btrfs_cpu_key_to_disk(&disk_key, new_key); 2916459931ecSChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 2917459931ecSChris Mason 2918459931ecSChris Mason new_item = btrfs_item_nr(leaf, slot); 2919459931ecSChris Mason 2920459931ecSChris Mason btrfs_set_item_offset(leaf, new_item, orig_offset); 2921459931ecSChris Mason btrfs_set_item_size(leaf, new_item, item_size - split_offset); 2922459931ecSChris Mason 2923459931ecSChris Mason btrfs_set_item_offset(leaf, item, 2924459931ecSChris Mason orig_offset + item_size - split_offset); 2925459931ecSChris Mason btrfs_set_item_size(leaf, item, split_offset); 2926459931ecSChris Mason 2927459931ecSChris Mason btrfs_set_header_nritems(leaf, nritems + 1); 2928459931ecSChris Mason 2929459931ecSChris Mason /* write the data for the start of the original item */ 2930459931ecSChris Mason write_extent_buffer(leaf, buf, 2931459931ecSChris Mason btrfs_item_ptr_offset(leaf, path->slots[0]), 2932459931ecSChris Mason split_offset); 2933459931ecSChris Mason 2934459931ecSChris Mason /* write the data for the new item */ 2935459931ecSChris Mason write_extent_buffer(leaf, buf + split_offset, 2936459931ecSChris Mason btrfs_item_ptr_offset(leaf, slot), 2937459931ecSChris Mason item_size - split_offset); 2938459931ecSChris Mason btrfs_mark_buffer_dirty(leaf); 2939459931ecSChris Mason 2940459931ecSChris Mason ret = 0; 2941459931ecSChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 2942459931ecSChris Mason btrfs_print_leaf(root, leaf); 2943459931ecSChris Mason BUG(); 2944459931ecSChris Mason } 2945459931ecSChris Mason kfree(buf); 2946459931ecSChris Mason return ret; 2947459931ecSChris Mason } 2948459931ecSChris Mason 2949459931ecSChris Mason /* 2950d352ac68SChris Mason * make the item pointed to by the path smaller. new_size indicates 2951d352ac68SChris Mason * how small to make it, and from_end tells us if we just chop bytes 2952d352ac68SChris Mason * off the end of the item or if we shift the item to chop bytes off 2953d352ac68SChris Mason * the front. 2954d352ac68SChris Mason */ 2955b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans, 2956b18c6685SChris Mason struct btrfs_root *root, 2957b18c6685SChris Mason struct btrfs_path *path, 2958179e29e4SChris Mason u32 new_size, int from_end) 2959b18c6685SChris Mason { 2960b18c6685SChris Mason int ret = 0; 2961b18c6685SChris Mason int slot; 2962b18c6685SChris Mason int slot_orig; 29635f39d397SChris Mason struct extent_buffer *leaf; 29645f39d397SChris Mason struct btrfs_item *item; 2965b18c6685SChris Mason u32 nritems; 2966b18c6685SChris Mason unsigned int data_end; 2967b18c6685SChris Mason unsigned int old_data_start; 2968b18c6685SChris Mason unsigned int old_size; 2969b18c6685SChris Mason unsigned int size_diff; 2970b18c6685SChris Mason int i; 2971b18c6685SChris Mason 2972b18c6685SChris Mason slot_orig = path->slots[0]; 29735f39d397SChris Mason leaf = path->nodes[0]; 2974179e29e4SChris Mason slot = path->slots[0]; 2975179e29e4SChris Mason 2976179e29e4SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 2977179e29e4SChris Mason if (old_size == new_size) 2978179e29e4SChris Mason return 0; 2979b18c6685SChris Mason 29805f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2981b18c6685SChris Mason data_end = leaf_data_end(root, leaf); 2982b18c6685SChris Mason 29835f39d397SChris Mason old_data_start = btrfs_item_offset_nr(leaf, slot); 2984179e29e4SChris Mason 2985b18c6685SChris Mason size_diff = old_size - new_size; 2986b18c6685SChris Mason 2987b18c6685SChris Mason BUG_ON(slot < 0); 2988b18c6685SChris Mason BUG_ON(slot >= nritems); 2989b18c6685SChris Mason 2990b18c6685SChris Mason /* 2991b18c6685SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2992b18c6685SChris Mason */ 2993b18c6685SChris Mason /* first correct the data pointers */ 2994b18c6685SChris Mason for (i = slot; i < nritems; i++) { 29955f39d397SChris Mason u32 ioff; 29965f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2997db94535dSChris Mason 2998db94535dSChris Mason if (!leaf->map_token) { 2999db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 3000db94535dSChris Mason sizeof(struct btrfs_item), 3001db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 3002db94535dSChris Mason &leaf->map_start, &leaf->map_len, 3003db94535dSChris Mason KM_USER1); 3004db94535dSChris Mason } 3005db94535dSChris Mason 30065f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 30075f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + size_diff); 3008b18c6685SChris Mason } 3009db94535dSChris Mason 3010db94535dSChris Mason if (leaf->map_token) { 3011db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 3012db94535dSChris Mason leaf->map_token = NULL; 3013db94535dSChris Mason } 3014db94535dSChris Mason 3015b18c6685SChris Mason /* shift the data */ 3016179e29e4SChris Mason if (from_end) { 30175f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 3018b18c6685SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 3019b18c6685SChris Mason data_end, old_data_start + new_size - data_end); 3020179e29e4SChris Mason } else { 3021179e29e4SChris Mason struct btrfs_disk_key disk_key; 3022179e29e4SChris Mason u64 offset; 3023179e29e4SChris Mason 3024179e29e4SChris Mason btrfs_item_key(leaf, &disk_key, slot); 3025179e29e4SChris Mason 3026179e29e4SChris Mason if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) { 3027179e29e4SChris Mason unsigned long ptr; 3028179e29e4SChris Mason struct btrfs_file_extent_item *fi; 3029179e29e4SChris Mason 3030179e29e4SChris Mason fi = btrfs_item_ptr(leaf, slot, 3031179e29e4SChris Mason struct btrfs_file_extent_item); 3032179e29e4SChris Mason fi = (struct btrfs_file_extent_item *)( 3033179e29e4SChris Mason (unsigned long)fi - size_diff); 3034179e29e4SChris Mason 3035179e29e4SChris Mason if (btrfs_file_extent_type(leaf, fi) == 3036179e29e4SChris Mason BTRFS_FILE_EXTENT_INLINE) { 3037179e29e4SChris Mason ptr = btrfs_item_ptr_offset(leaf, slot); 3038179e29e4SChris Mason memmove_extent_buffer(leaf, ptr, 3039179e29e4SChris Mason (unsigned long)fi, 3040179e29e4SChris Mason offsetof(struct btrfs_file_extent_item, 3041179e29e4SChris Mason disk_bytenr)); 3042179e29e4SChris Mason } 3043179e29e4SChris Mason } 3044179e29e4SChris Mason 3045179e29e4SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 3046179e29e4SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 3047179e29e4SChris Mason data_end, old_data_start - data_end); 3048179e29e4SChris Mason 3049179e29e4SChris Mason offset = btrfs_disk_key_offset(&disk_key); 3050179e29e4SChris Mason btrfs_set_disk_key_offset(&disk_key, offset + size_diff); 3051179e29e4SChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 3052179e29e4SChris Mason if (slot == 0) 3053179e29e4SChris Mason fixup_low_keys(trans, root, path, &disk_key, 1); 3054179e29e4SChris Mason } 30555f39d397SChris Mason 30565f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 30575f39d397SChris Mason btrfs_set_item_size(leaf, item, new_size); 30585f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 3059b18c6685SChris Mason 3060b18c6685SChris Mason ret = 0; 30615f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 30625f39d397SChris Mason btrfs_print_leaf(root, leaf); 3063b18c6685SChris Mason BUG(); 30645f39d397SChris Mason } 3065b18c6685SChris Mason return ret; 3066b18c6685SChris Mason } 3067b18c6685SChris Mason 3068d352ac68SChris Mason /* 3069d352ac68SChris Mason * make the item pointed to by the path bigger, data_size is the new size. 3070d352ac68SChris Mason */ 30715f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans, 30725f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 30735f39d397SChris Mason u32 data_size) 30746567e837SChris Mason { 30756567e837SChris Mason int ret = 0; 30766567e837SChris Mason int slot; 30776567e837SChris Mason int slot_orig; 30785f39d397SChris Mason struct extent_buffer *leaf; 30795f39d397SChris Mason struct btrfs_item *item; 30806567e837SChris Mason u32 nritems; 30816567e837SChris Mason unsigned int data_end; 30826567e837SChris Mason unsigned int old_data; 30836567e837SChris Mason unsigned int old_size; 30846567e837SChris Mason int i; 30856567e837SChris Mason 30866567e837SChris Mason slot_orig = path->slots[0]; 30875f39d397SChris Mason leaf = path->nodes[0]; 30886567e837SChris Mason 30895f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 30906567e837SChris Mason data_end = leaf_data_end(root, leaf); 30916567e837SChris Mason 30925f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < data_size) { 30935f39d397SChris Mason btrfs_print_leaf(root, leaf); 30946567e837SChris Mason BUG(); 30955f39d397SChris Mason } 30966567e837SChris Mason slot = path->slots[0]; 30975f39d397SChris Mason old_data = btrfs_item_end_nr(leaf, slot); 30986567e837SChris Mason 30996567e837SChris Mason BUG_ON(slot < 0); 31003326d1b0SChris Mason if (slot >= nritems) { 31013326d1b0SChris Mason btrfs_print_leaf(root, leaf); 31023326d1b0SChris Mason printk("slot %d too large, nritems %d\n", slot, nritems); 31033326d1b0SChris Mason BUG_ON(1); 31043326d1b0SChris Mason } 31056567e837SChris Mason 31066567e837SChris Mason /* 31076567e837SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 31086567e837SChris Mason */ 31096567e837SChris Mason /* first correct the data pointers */ 31106567e837SChris Mason for (i = slot; i < nritems; i++) { 31115f39d397SChris Mason u32 ioff; 31125f39d397SChris Mason item = btrfs_item_nr(leaf, i); 3113db94535dSChris Mason 3114db94535dSChris Mason if (!leaf->map_token) { 3115db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 3116db94535dSChris Mason sizeof(struct btrfs_item), 3117db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 3118db94535dSChris Mason &leaf->map_start, &leaf->map_len, 3119db94535dSChris Mason KM_USER1); 3120db94535dSChris Mason } 31215f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 31225f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff - data_size); 31236567e837SChris Mason } 31245f39d397SChris Mason 3125db94535dSChris Mason if (leaf->map_token) { 3126db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 3127db94535dSChris Mason leaf->map_token = NULL; 3128db94535dSChris Mason } 3129db94535dSChris Mason 31306567e837SChris Mason /* shift the data */ 31315f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 31326567e837SChris Mason data_end - data_size, btrfs_leaf_data(leaf) + 31336567e837SChris Mason data_end, old_data - data_end); 31345f39d397SChris Mason 31356567e837SChris Mason data_end = old_data; 31365f39d397SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 31375f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 31385f39d397SChris Mason btrfs_set_item_size(leaf, item, old_size + data_size); 31395f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 31406567e837SChris Mason 31416567e837SChris Mason ret = 0; 31425f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 31435f39d397SChris Mason btrfs_print_leaf(root, leaf); 31446567e837SChris Mason BUG(); 31455f39d397SChris Mason } 31466567e837SChris Mason return ret; 31476567e837SChris Mason } 31486567e837SChris Mason 314974123bd7SChris Mason /* 3150d352ac68SChris Mason * Given a key and some data, insert items into the tree. 315174123bd7SChris Mason * This does all the path init required, making room in the tree if needed. 3152f3465ca4SJosef Bacik * Returns the number of keys that were inserted. 3153f3465ca4SJosef Bacik */ 3154f3465ca4SJosef Bacik int btrfs_insert_some_items(struct btrfs_trans_handle *trans, 3155f3465ca4SJosef Bacik struct btrfs_root *root, 3156f3465ca4SJosef Bacik struct btrfs_path *path, 3157f3465ca4SJosef Bacik struct btrfs_key *cpu_key, u32 *data_size, 3158f3465ca4SJosef Bacik int nr) 3159f3465ca4SJosef Bacik { 3160f3465ca4SJosef Bacik struct extent_buffer *leaf; 3161f3465ca4SJosef Bacik struct btrfs_item *item; 3162f3465ca4SJosef Bacik int ret = 0; 3163f3465ca4SJosef Bacik int slot; 3164f3465ca4SJosef Bacik int i; 3165f3465ca4SJosef Bacik u32 nritems; 3166f3465ca4SJosef Bacik u32 total_data = 0; 3167f3465ca4SJosef Bacik u32 total_size = 0; 3168f3465ca4SJosef Bacik unsigned int data_end; 3169f3465ca4SJosef Bacik struct btrfs_disk_key disk_key; 3170f3465ca4SJosef Bacik struct btrfs_key found_key; 3171f3465ca4SJosef Bacik 3172f3465ca4SJosef Bacik found_key.objectid = 0; 3173f3465ca4SJosef Bacik nr = min_t(int, nr, BTRFS_NODEPTRS_PER_BLOCK(root)); 3174f3465ca4SJosef Bacik 3175f3465ca4SJosef Bacik for (i = 0; i < nr; i++) 3176f3465ca4SJosef Bacik total_data += data_size[i]; 3177f3465ca4SJosef Bacik 3178f3465ca4SJosef Bacik total_data = min_t(u32, total_data, BTRFS_LEAF_DATA_SIZE(root)); 3179f3465ca4SJosef Bacik total_size = total_data + (nr * sizeof(struct btrfs_item)); 3180f3465ca4SJosef Bacik ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1); 3181f3465ca4SJosef Bacik if (ret == 0) 3182f3465ca4SJosef Bacik return -EEXIST; 3183f3465ca4SJosef Bacik if (ret < 0) 3184f3465ca4SJosef Bacik goto out; 3185f3465ca4SJosef Bacik 3186f3465ca4SJosef Bacik leaf = path->nodes[0]; 3187f3465ca4SJosef Bacik 3188f3465ca4SJosef Bacik nritems = btrfs_header_nritems(leaf); 3189f3465ca4SJosef Bacik data_end = leaf_data_end(root, leaf); 3190f3465ca4SJosef Bacik 3191f3465ca4SJosef Bacik if (btrfs_leaf_free_space(root, leaf) < total_size) { 3192f3465ca4SJosef Bacik for (i = nr; i >= 0; i--) { 3193f3465ca4SJosef Bacik total_data -= data_size[i]; 3194f3465ca4SJosef Bacik total_size -= data_size[i] + sizeof(struct btrfs_item); 3195f3465ca4SJosef Bacik if (total_size < btrfs_leaf_free_space(root, leaf)) 3196f3465ca4SJosef Bacik break; 3197f3465ca4SJosef Bacik } 3198f3465ca4SJosef Bacik nr = i; 3199f3465ca4SJosef Bacik } 3200f3465ca4SJosef Bacik 3201f3465ca4SJosef Bacik slot = path->slots[0]; 3202f3465ca4SJosef Bacik BUG_ON(slot < 0); 3203f3465ca4SJosef Bacik 3204f3465ca4SJosef Bacik if (slot != nritems) { 3205f3465ca4SJosef Bacik unsigned int old_data = btrfs_item_end_nr(leaf, slot); 3206f3465ca4SJosef Bacik 3207f3465ca4SJosef Bacik item = btrfs_item_nr(leaf, slot); 3208f3465ca4SJosef Bacik btrfs_item_key_to_cpu(leaf, &found_key, slot); 3209f3465ca4SJosef Bacik 3210f3465ca4SJosef Bacik /* figure out how many keys we can insert in here */ 3211f3465ca4SJosef Bacik total_data = data_size[0]; 3212f3465ca4SJosef Bacik for (i = 1; i < nr; i++) { 3213f3465ca4SJosef Bacik if (comp_cpu_keys(&found_key, cpu_key + i) <= 0) 3214f3465ca4SJosef Bacik break; 3215f3465ca4SJosef Bacik total_data += data_size[i]; 3216f3465ca4SJosef Bacik } 3217f3465ca4SJosef Bacik nr = i; 3218f3465ca4SJosef Bacik 3219f3465ca4SJosef Bacik if (old_data < data_end) { 3220f3465ca4SJosef Bacik btrfs_print_leaf(root, leaf); 3221f3465ca4SJosef Bacik printk("slot %d old_data %d data_end %d\n", 3222f3465ca4SJosef Bacik slot, old_data, data_end); 3223f3465ca4SJosef Bacik BUG_ON(1); 3224f3465ca4SJosef Bacik } 3225f3465ca4SJosef Bacik /* 3226f3465ca4SJosef Bacik * item0..itemN ... dataN.offset..dataN.size .. data0.size 3227f3465ca4SJosef Bacik */ 3228f3465ca4SJosef Bacik /* first correct the data pointers */ 3229f3465ca4SJosef Bacik WARN_ON(leaf->map_token); 3230f3465ca4SJosef Bacik for (i = slot; i < nritems; i++) { 3231f3465ca4SJosef Bacik u32 ioff; 3232f3465ca4SJosef Bacik 3233f3465ca4SJosef Bacik item = btrfs_item_nr(leaf, i); 3234f3465ca4SJosef Bacik if (!leaf->map_token) { 3235f3465ca4SJosef Bacik map_extent_buffer(leaf, (unsigned long)item, 3236f3465ca4SJosef Bacik sizeof(struct btrfs_item), 3237f3465ca4SJosef Bacik &leaf->map_token, &leaf->kaddr, 3238f3465ca4SJosef Bacik &leaf->map_start, &leaf->map_len, 3239f3465ca4SJosef Bacik KM_USER1); 3240f3465ca4SJosef Bacik } 3241f3465ca4SJosef Bacik 3242f3465ca4SJosef Bacik ioff = btrfs_item_offset(leaf, item); 3243f3465ca4SJosef Bacik btrfs_set_item_offset(leaf, item, ioff - total_data); 3244f3465ca4SJosef Bacik } 3245f3465ca4SJosef Bacik if (leaf->map_token) { 3246f3465ca4SJosef Bacik unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 3247f3465ca4SJosef Bacik leaf->map_token = NULL; 3248f3465ca4SJosef Bacik } 3249f3465ca4SJosef Bacik 3250f3465ca4SJosef Bacik /* shift the items */ 3251f3465ca4SJosef Bacik memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr), 3252f3465ca4SJosef Bacik btrfs_item_nr_offset(slot), 3253f3465ca4SJosef Bacik (nritems - slot) * sizeof(struct btrfs_item)); 3254f3465ca4SJosef Bacik 3255f3465ca4SJosef Bacik /* shift the data */ 3256f3465ca4SJosef Bacik memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 3257f3465ca4SJosef Bacik data_end - total_data, btrfs_leaf_data(leaf) + 3258f3465ca4SJosef Bacik data_end, old_data - data_end); 3259f3465ca4SJosef Bacik data_end = old_data; 3260f3465ca4SJosef Bacik } else { 3261f3465ca4SJosef Bacik /* 3262f3465ca4SJosef Bacik * this sucks but it has to be done, if we are inserting at 3263f3465ca4SJosef Bacik * the end of the leaf only insert 1 of the items, since we 3264f3465ca4SJosef Bacik * have no way of knowing whats on the next leaf and we'd have 3265f3465ca4SJosef Bacik * to drop our current locks to figure it out 3266f3465ca4SJosef Bacik */ 3267f3465ca4SJosef Bacik nr = 1; 3268f3465ca4SJosef Bacik } 3269f3465ca4SJosef Bacik 3270f3465ca4SJosef Bacik /* setup the item for the new data */ 3271f3465ca4SJosef Bacik for (i = 0; i < nr; i++) { 3272f3465ca4SJosef Bacik btrfs_cpu_key_to_disk(&disk_key, cpu_key + i); 3273f3465ca4SJosef Bacik btrfs_set_item_key(leaf, &disk_key, slot + i); 3274f3465ca4SJosef Bacik item = btrfs_item_nr(leaf, slot + i); 3275f3465ca4SJosef Bacik btrfs_set_item_offset(leaf, item, data_end - data_size[i]); 3276f3465ca4SJosef Bacik data_end -= data_size[i]; 3277f3465ca4SJosef Bacik btrfs_set_item_size(leaf, item, data_size[i]); 3278f3465ca4SJosef Bacik } 3279f3465ca4SJosef Bacik btrfs_set_header_nritems(leaf, nritems + nr); 3280f3465ca4SJosef Bacik btrfs_mark_buffer_dirty(leaf); 3281f3465ca4SJosef Bacik 3282f3465ca4SJosef Bacik ret = 0; 3283f3465ca4SJosef Bacik if (slot == 0) { 3284f3465ca4SJosef Bacik btrfs_cpu_key_to_disk(&disk_key, cpu_key); 3285f3465ca4SJosef Bacik ret = fixup_low_keys(trans, root, path, &disk_key, 1); 3286f3465ca4SJosef Bacik } 3287f3465ca4SJosef Bacik 3288f3465ca4SJosef Bacik if (btrfs_leaf_free_space(root, leaf) < 0) { 3289f3465ca4SJosef Bacik btrfs_print_leaf(root, leaf); 3290f3465ca4SJosef Bacik BUG(); 3291f3465ca4SJosef Bacik } 3292f3465ca4SJosef Bacik out: 3293f3465ca4SJosef Bacik if (!ret) 3294f3465ca4SJosef Bacik ret = nr; 3295f3465ca4SJosef Bacik return ret; 3296f3465ca4SJosef Bacik } 3297f3465ca4SJosef Bacik 3298f3465ca4SJosef Bacik /* 3299f3465ca4SJosef Bacik * Given a key and some data, insert items into the tree. 3300f3465ca4SJosef Bacik * This does all the path init required, making room in the tree if needed. 330174123bd7SChris Mason */ 33029c58309dSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans, 33035f39d397SChris Mason struct btrfs_root *root, 33045f39d397SChris Mason struct btrfs_path *path, 33059c58309dSChris Mason struct btrfs_key *cpu_key, u32 *data_size, 33069c58309dSChris Mason int nr) 3307be0e5c09SChris Mason { 33085f39d397SChris Mason struct extent_buffer *leaf; 33095f39d397SChris Mason struct btrfs_item *item; 3310aa5d6bedSChris Mason int ret = 0; 3311be0e5c09SChris Mason int slot; 3312eb60ceacSChris Mason int slot_orig; 33139c58309dSChris Mason int i; 33147518a238SChris Mason u32 nritems; 33159c58309dSChris Mason u32 total_size = 0; 33169c58309dSChris Mason u32 total_data = 0; 3317be0e5c09SChris Mason unsigned int data_end; 3318e2fa7227SChris Mason struct btrfs_disk_key disk_key; 3319e2fa7227SChris Mason 33209c58309dSChris Mason for (i = 0; i < nr; i++) { 33219c58309dSChris Mason total_data += data_size[i]; 33229c58309dSChris Mason } 3323be0e5c09SChris Mason 33247b128766SJosef Bacik total_size = total_data + (nr * sizeof(struct btrfs_item)); 33259c58309dSChris Mason ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1); 33260f9dd46cSJosef Bacik if (ret == 0) 3327f0930a37SChris Mason return -EEXIST; 3328ed2ff2cbSChris Mason if (ret < 0) 3329ed2ff2cbSChris Mason goto out; 3330be0e5c09SChris Mason 333162e2749eSChris Mason slot_orig = path->slots[0]; 33325f39d397SChris Mason leaf = path->nodes[0]; 333374123bd7SChris Mason 33345f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 3335123abc88SChris Mason data_end = leaf_data_end(root, leaf); 3336eb60ceacSChris Mason 3337f25956ccSChris Mason if (btrfs_leaf_free_space(root, leaf) < total_size) { 33383326d1b0SChris Mason btrfs_print_leaf(root, leaf); 33393326d1b0SChris Mason printk("not enough freespace need %u have %d\n", 33409c58309dSChris Mason total_size, btrfs_leaf_free_space(root, leaf)); 3341be0e5c09SChris Mason BUG(); 3342d4dbff95SChris Mason } 33435f39d397SChris Mason 334462e2749eSChris Mason slot = path->slots[0]; 3345eb60ceacSChris Mason BUG_ON(slot < 0); 33465f39d397SChris Mason 3347be0e5c09SChris Mason if (slot != nritems) { 33485f39d397SChris Mason unsigned int old_data = btrfs_item_end_nr(leaf, slot); 3349be0e5c09SChris Mason 33505f39d397SChris Mason if (old_data < data_end) { 33515f39d397SChris Mason btrfs_print_leaf(root, leaf); 33525f39d397SChris Mason printk("slot %d old_data %d data_end %d\n", 33535f39d397SChris Mason slot, old_data, data_end); 33545f39d397SChris Mason BUG_ON(1); 33555f39d397SChris Mason } 3356be0e5c09SChris Mason /* 3357be0e5c09SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 3358be0e5c09SChris Mason */ 3359be0e5c09SChris Mason /* first correct the data pointers */ 3360db94535dSChris Mason WARN_ON(leaf->map_token); 33610783fcfcSChris Mason for (i = slot; i < nritems; i++) { 33625f39d397SChris Mason u32 ioff; 3363db94535dSChris Mason 33645f39d397SChris Mason item = btrfs_item_nr(leaf, i); 3365db94535dSChris Mason if (!leaf->map_token) { 3366db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 3367db94535dSChris Mason sizeof(struct btrfs_item), 3368db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 3369db94535dSChris Mason &leaf->map_start, &leaf->map_len, 3370db94535dSChris Mason KM_USER1); 3371db94535dSChris Mason } 3372db94535dSChris Mason 33735f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 33749c58309dSChris Mason btrfs_set_item_offset(leaf, item, ioff - total_data); 33750783fcfcSChris Mason } 3376db94535dSChris Mason if (leaf->map_token) { 3377db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 3378db94535dSChris Mason leaf->map_token = NULL; 3379db94535dSChris Mason } 3380be0e5c09SChris Mason 3381be0e5c09SChris Mason /* shift the items */ 33829c58309dSChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr), 33835f39d397SChris Mason btrfs_item_nr_offset(slot), 33840783fcfcSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 3385be0e5c09SChris Mason 3386be0e5c09SChris Mason /* shift the data */ 33875f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 33889c58309dSChris Mason data_end - total_data, btrfs_leaf_data(leaf) + 3389be0e5c09SChris Mason data_end, old_data - data_end); 3390be0e5c09SChris Mason data_end = old_data; 3391be0e5c09SChris Mason } 33925f39d397SChris Mason 339362e2749eSChris Mason /* setup the item for the new data */ 33949c58309dSChris Mason for (i = 0; i < nr; i++) { 33959c58309dSChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key + i); 33969c58309dSChris Mason btrfs_set_item_key(leaf, &disk_key, slot + i); 33979c58309dSChris Mason item = btrfs_item_nr(leaf, slot + i); 33989c58309dSChris Mason btrfs_set_item_offset(leaf, item, data_end - data_size[i]); 33999c58309dSChris Mason data_end -= data_size[i]; 34009c58309dSChris Mason btrfs_set_item_size(leaf, item, data_size[i]); 34019c58309dSChris Mason } 34029c58309dSChris Mason btrfs_set_header_nritems(leaf, nritems + nr); 34035f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 3404aa5d6bedSChris Mason 3405aa5d6bedSChris Mason ret = 0; 34065a01a2e3SChris Mason if (slot == 0) { 34075a01a2e3SChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key); 3408e089f05cSChris Mason ret = fixup_low_keys(trans, root, path, &disk_key, 1); 34095a01a2e3SChris Mason } 3410aa5d6bedSChris Mason 34115f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 34125f39d397SChris Mason btrfs_print_leaf(root, leaf); 3413be0e5c09SChris Mason BUG(); 34145f39d397SChris Mason } 3415ed2ff2cbSChris Mason out: 341662e2749eSChris Mason return ret; 341762e2749eSChris Mason } 341862e2749eSChris Mason 341962e2749eSChris Mason /* 342062e2749eSChris Mason * Given a key and some data, insert an item into the tree. 342162e2749eSChris Mason * This does all the path init required, making room in the tree if needed. 342262e2749eSChris Mason */ 3423e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root 3424e089f05cSChris Mason *root, struct btrfs_key *cpu_key, void *data, u32 3425e089f05cSChris Mason data_size) 342662e2749eSChris Mason { 342762e2749eSChris Mason int ret = 0; 34282c90e5d6SChris Mason struct btrfs_path *path; 34295f39d397SChris Mason struct extent_buffer *leaf; 34305f39d397SChris Mason unsigned long ptr; 343162e2749eSChris Mason 34322c90e5d6SChris Mason path = btrfs_alloc_path(); 34332c90e5d6SChris Mason BUG_ON(!path); 34342c90e5d6SChris Mason ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size); 343562e2749eSChris Mason if (!ret) { 34365f39d397SChris Mason leaf = path->nodes[0]; 34375f39d397SChris Mason ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); 34385f39d397SChris Mason write_extent_buffer(leaf, data, ptr, data_size); 34395f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 344062e2749eSChris Mason } 34412c90e5d6SChris Mason btrfs_free_path(path); 3442aa5d6bedSChris Mason return ret; 3443be0e5c09SChris Mason } 3444be0e5c09SChris Mason 344574123bd7SChris Mason /* 34465de08d7dSChris Mason * delete the pointer from a given node. 344774123bd7SChris Mason * 3448d352ac68SChris Mason * the tree should have been previously balanced so the deletion does not 3449d352ac68SChris Mason * empty a node. 345074123bd7SChris Mason */ 3451e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 3452e089f05cSChris Mason struct btrfs_path *path, int level, int slot) 3453be0e5c09SChris Mason { 34545f39d397SChris Mason struct extent_buffer *parent = path->nodes[level]; 34557518a238SChris Mason u32 nritems; 3456aa5d6bedSChris Mason int ret = 0; 3457bb803951SChris Mason int wret; 3458be0e5c09SChris Mason 34595f39d397SChris Mason nritems = btrfs_header_nritems(parent); 3460be0e5c09SChris Mason if (slot != nritems -1) { 34615f39d397SChris Mason memmove_extent_buffer(parent, 34625f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 34635f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 3464d6025579SChris Mason sizeof(struct btrfs_key_ptr) * 3465d6025579SChris Mason (nritems - slot - 1)); 3466be0e5c09SChris Mason } 34677518a238SChris Mason nritems--; 34685f39d397SChris Mason btrfs_set_header_nritems(parent, nritems); 34697518a238SChris Mason if (nritems == 0 && parent == root->node) { 34705f39d397SChris Mason BUG_ON(btrfs_header_level(root->node) != 1); 3471eb60ceacSChris Mason /* just turn the root into a leaf and break */ 34725f39d397SChris Mason btrfs_set_header_level(root->node, 0); 3473bb803951SChris Mason } else if (slot == 0) { 34745f39d397SChris Mason struct btrfs_disk_key disk_key; 34755f39d397SChris Mason 34765f39d397SChris Mason btrfs_node_key(parent, &disk_key, 0); 34775f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, level + 1); 34780f70abe2SChris Mason if (wret) 34790f70abe2SChris Mason ret = wret; 3480be0e5c09SChris Mason } 3481d6025579SChris Mason btrfs_mark_buffer_dirty(parent); 3482aa5d6bedSChris Mason return ret; 3483be0e5c09SChris Mason } 3484be0e5c09SChris Mason 348574123bd7SChris Mason /* 3486323ac95bSChris Mason * a helper function to delete the leaf pointed to by path->slots[1] and 3487323ac95bSChris Mason * path->nodes[1]. bytenr is the node block pointer, but since the callers 3488323ac95bSChris Mason * already know it, it is faster to have them pass it down than to 3489323ac95bSChris Mason * read it out of the node again. 3490323ac95bSChris Mason * 3491323ac95bSChris Mason * This deletes the pointer in path->nodes[1] and frees the leaf 3492323ac95bSChris Mason * block extent. zero is returned if it all worked out, < 0 otherwise. 3493323ac95bSChris Mason * 3494323ac95bSChris Mason * The path must have already been setup for deleting the leaf, including 3495323ac95bSChris Mason * all the proper balancing. path->nodes[1] must be locked. 3496323ac95bSChris Mason */ 3497323ac95bSChris Mason noinline int btrfs_del_leaf(struct btrfs_trans_handle *trans, 3498323ac95bSChris Mason struct btrfs_root *root, 3499323ac95bSChris Mason struct btrfs_path *path, u64 bytenr) 3500323ac95bSChris Mason { 3501323ac95bSChris Mason int ret; 3502323ac95bSChris Mason u64 root_gen = btrfs_header_generation(path->nodes[1]); 3503323ac95bSChris Mason 3504323ac95bSChris Mason ret = del_ptr(trans, root, path, 1, path->slots[1]); 3505323ac95bSChris Mason if (ret) 3506323ac95bSChris Mason return ret; 3507323ac95bSChris Mason 3508323ac95bSChris Mason ret = btrfs_free_extent(trans, root, bytenr, 3509323ac95bSChris Mason btrfs_level_size(root, 0), 3510323ac95bSChris Mason path->nodes[1]->start, 3511323ac95bSChris Mason btrfs_header_owner(path->nodes[1]), 35123bb1a1bcSYan Zheng root_gen, 0, 1); 3513323ac95bSChris Mason return ret; 3514323ac95bSChris Mason } 3515323ac95bSChris Mason /* 351674123bd7SChris Mason * delete the item at the leaf level in path. If that empties 351774123bd7SChris Mason * the leaf, remove it from the tree 351874123bd7SChris Mason */ 351985e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root, 352085e21bacSChris Mason struct btrfs_path *path, int slot, int nr) 3521be0e5c09SChris Mason { 35225f39d397SChris Mason struct extent_buffer *leaf; 35235f39d397SChris Mason struct btrfs_item *item; 352485e21bacSChris Mason int last_off; 352585e21bacSChris Mason int dsize = 0; 3526aa5d6bedSChris Mason int ret = 0; 3527aa5d6bedSChris Mason int wret; 352885e21bacSChris Mason int i; 35297518a238SChris Mason u32 nritems; 3530be0e5c09SChris Mason 35315f39d397SChris Mason leaf = path->nodes[0]; 353285e21bacSChris Mason last_off = btrfs_item_offset_nr(leaf, slot + nr - 1); 353385e21bacSChris Mason 353485e21bacSChris Mason for (i = 0; i < nr; i++) 353585e21bacSChris Mason dsize += btrfs_item_size_nr(leaf, slot + i); 353685e21bacSChris Mason 35375f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 3538be0e5c09SChris Mason 353985e21bacSChris Mason if (slot + nr != nritems) { 3540123abc88SChris Mason int data_end = leaf_data_end(root, leaf); 35415f39d397SChris Mason 35425f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 3543d6025579SChris Mason data_end + dsize, 3544123abc88SChris Mason btrfs_leaf_data(leaf) + data_end, 354585e21bacSChris Mason last_off - data_end); 35465f39d397SChris Mason 354785e21bacSChris Mason for (i = slot + nr; i < nritems; i++) { 35485f39d397SChris Mason u32 ioff; 3549db94535dSChris Mason 35505f39d397SChris Mason item = btrfs_item_nr(leaf, i); 3551db94535dSChris Mason if (!leaf->map_token) { 3552db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 3553db94535dSChris Mason sizeof(struct btrfs_item), 3554db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 3555db94535dSChris Mason &leaf->map_start, &leaf->map_len, 3556db94535dSChris Mason KM_USER1); 3557db94535dSChris Mason } 35585f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 35595f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + dsize); 35600783fcfcSChris Mason } 3561db94535dSChris Mason 3562db94535dSChris Mason if (leaf->map_token) { 3563db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 3564db94535dSChris Mason leaf->map_token = NULL; 3565db94535dSChris Mason } 3566db94535dSChris Mason 35675f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot), 356885e21bacSChris Mason btrfs_item_nr_offset(slot + nr), 35690783fcfcSChris Mason sizeof(struct btrfs_item) * 357085e21bacSChris Mason (nritems - slot - nr)); 3571be0e5c09SChris Mason } 357285e21bacSChris Mason btrfs_set_header_nritems(leaf, nritems - nr); 357385e21bacSChris Mason nritems -= nr; 35745f39d397SChris Mason 357574123bd7SChris Mason /* delete the leaf if we've emptied it */ 35767518a238SChris Mason if (nritems == 0) { 35775f39d397SChris Mason if (leaf == root->node) { 35785f39d397SChris Mason btrfs_set_header_level(leaf, 0); 35799a8dd150SChris Mason } else { 3580323ac95bSChris Mason ret = btrfs_del_leaf(trans, root, path, leaf->start); 3581323ac95bSChris Mason BUG_ON(ret); 35829a8dd150SChris Mason } 3583be0e5c09SChris Mason } else { 35847518a238SChris Mason int used = leaf_space_used(leaf, 0, nritems); 3585aa5d6bedSChris Mason if (slot == 0) { 35865f39d397SChris Mason struct btrfs_disk_key disk_key; 35875f39d397SChris Mason 35885f39d397SChris Mason btrfs_item_key(leaf, &disk_key, 0); 3589e089f05cSChris Mason wret = fixup_low_keys(trans, root, path, 35905f39d397SChris Mason &disk_key, 1); 3591aa5d6bedSChris Mason if (wret) 3592aa5d6bedSChris Mason ret = wret; 3593aa5d6bedSChris Mason } 3594aa5d6bedSChris Mason 359574123bd7SChris Mason /* delete the leaf if it is mostly empty */ 359685e21bacSChris Mason if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) { 3597be0e5c09SChris Mason /* push_leaf_left fixes the path. 3598be0e5c09SChris Mason * make sure the path still points to our leaf 3599be0e5c09SChris Mason * for possible call to del_ptr below 3600be0e5c09SChris Mason */ 36014920c9acSChris Mason slot = path->slots[1]; 36025f39d397SChris Mason extent_buffer_get(leaf); 36035f39d397SChris Mason 360485e21bacSChris Mason wret = push_leaf_left(trans, root, path, 1, 1); 360554aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 3606aa5d6bedSChris Mason ret = wret; 36075f39d397SChris Mason 36085f39d397SChris Mason if (path->nodes[0] == leaf && 36095f39d397SChris Mason btrfs_header_nritems(leaf)) { 361085e21bacSChris Mason wret = push_leaf_right(trans, root, path, 1, 1); 361154aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 3612aa5d6bedSChris Mason ret = wret; 3613aa5d6bedSChris Mason } 36145f39d397SChris Mason 36155f39d397SChris Mason if (btrfs_header_nritems(leaf) == 0) { 3616323ac95bSChris Mason path->slots[1] = slot; 3617323ac95bSChris Mason ret = btrfs_del_leaf(trans, root, path, leaf->start); 3618323ac95bSChris Mason BUG_ON(ret); 36195f39d397SChris Mason free_extent_buffer(leaf); 36205de08d7dSChris Mason } else { 3621925baeddSChris Mason /* if we're still in the path, make sure 3622925baeddSChris Mason * we're dirty. Otherwise, one of the 3623925baeddSChris Mason * push_leaf functions must have already 3624925baeddSChris Mason * dirtied this buffer 3625925baeddSChris Mason */ 3626925baeddSChris Mason if (path->nodes[0] == leaf) 36275f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 36285f39d397SChris Mason free_extent_buffer(leaf); 3629be0e5c09SChris Mason } 3630d5719762SChris Mason } else { 36315f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 3632be0e5c09SChris Mason } 36339a8dd150SChris Mason } 3634aa5d6bedSChris Mason return ret; 36359a8dd150SChris Mason } 36369a8dd150SChris Mason 363797571fd0SChris Mason /* 3638925baeddSChris Mason * search the tree again to find a leaf with lesser keys 36397bb86316SChris Mason * returns 0 if it found something or 1 if there are no lesser leaves. 36407bb86316SChris Mason * returns < 0 on io errors. 3641d352ac68SChris Mason * 3642d352ac68SChris Mason * This may release the path, and so you may lose any locks held at the 3643d352ac68SChris Mason * time you call it. 36447bb86316SChris Mason */ 36457bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path) 36467bb86316SChris Mason { 3647925baeddSChris Mason struct btrfs_key key; 3648925baeddSChris Mason struct btrfs_disk_key found_key; 3649925baeddSChris Mason int ret; 36507bb86316SChris Mason 3651925baeddSChris Mason btrfs_item_key_to_cpu(path->nodes[0], &key, 0); 3652925baeddSChris Mason 3653925baeddSChris Mason if (key.offset > 0) 3654925baeddSChris Mason key.offset--; 3655925baeddSChris Mason else if (key.type > 0) 3656925baeddSChris Mason key.type--; 3657925baeddSChris Mason else if (key.objectid > 0) 3658925baeddSChris Mason key.objectid--; 3659925baeddSChris Mason else 36607bb86316SChris Mason return 1; 36617bb86316SChris Mason 3662925baeddSChris Mason btrfs_release_path(root, path); 3663925baeddSChris Mason ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); 3664925baeddSChris Mason if (ret < 0) 3665925baeddSChris Mason return ret; 3666925baeddSChris Mason btrfs_item_key(path->nodes[0], &found_key, 0); 3667925baeddSChris Mason ret = comp_keys(&found_key, &key); 3668925baeddSChris Mason if (ret < 0) 36697bb86316SChris Mason return 0; 3670925baeddSChris Mason return 1; 36717bb86316SChris Mason } 36727bb86316SChris Mason 36733f157a2fSChris Mason /* 36743f157a2fSChris Mason * A helper function to walk down the tree starting at min_key, and looking 36753f157a2fSChris Mason * for nodes or leaves that are either in cache or have a minimum 3676d352ac68SChris Mason * transaction id. This is used by the btree defrag code, and tree logging 36773f157a2fSChris Mason * 36783f157a2fSChris Mason * This does not cow, but it does stuff the starting key it finds back 36793f157a2fSChris Mason * into min_key, so you can call btrfs_search_slot with cow=1 on the 36803f157a2fSChris Mason * key and get a writable path. 36813f157a2fSChris Mason * 36823f157a2fSChris Mason * This does lock as it descends, and path->keep_locks should be set 36833f157a2fSChris Mason * to 1 by the caller. 36843f157a2fSChris Mason * 36853f157a2fSChris Mason * This honors path->lowest_level to prevent descent past a given level 36863f157a2fSChris Mason * of the tree. 36873f157a2fSChris Mason * 3688d352ac68SChris Mason * min_trans indicates the oldest transaction that you are interested 3689d352ac68SChris Mason * in walking through. Any nodes or leaves older than min_trans are 3690d352ac68SChris Mason * skipped over (without reading them). 3691d352ac68SChris Mason * 36923f157a2fSChris Mason * returns zero if something useful was found, < 0 on error and 1 if there 36933f157a2fSChris Mason * was nothing in the tree that matched the search criteria. 36943f157a2fSChris Mason */ 36953f157a2fSChris Mason int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key, 3696e02119d5SChris Mason struct btrfs_key *max_key, 36973f157a2fSChris Mason struct btrfs_path *path, int cache_only, 36983f157a2fSChris Mason u64 min_trans) 36993f157a2fSChris Mason { 37003f157a2fSChris Mason struct extent_buffer *cur; 37013f157a2fSChris Mason struct btrfs_key found_key; 37023f157a2fSChris Mason int slot; 37039652480bSYan int sret; 37043f157a2fSChris Mason u32 nritems; 37053f157a2fSChris Mason int level; 37063f157a2fSChris Mason int ret = 1; 37073f157a2fSChris Mason 3708934d375bSChris Mason WARN_ON(!path->keep_locks); 37093f157a2fSChris Mason again: 37103f157a2fSChris Mason cur = btrfs_lock_root_node(root); 37113f157a2fSChris Mason level = btrfs_header_level(cur); 3712e02119d5SChris Mason WARN_ON(path->nodes[level]); 37133f157a2fSChris Mason path->nodes[level] = cur; 37143f157a2fSChris Mason path->locks[level] = 1; 37153f157a2fSChris Mason 37163f157a2fSChris Mason if (btrfs_header_generation(cur) < min_trans) { 37173f157a2fSChris Mason ret = 1; 37183f157a2fSChris Mason goto out; 37193f157a2fSChris Mason } 37203f157a2fSChris Mason while(1) { 37213f157a2fSChris Mason nritems = btrfs_header_nritems(cur); 37223f157a2fSChris Mason level = btrfs_header_level(cur); 37239652480bSYan sret = bin_search(cur, min_key, level, &slot); 37243f157a2fSChris Mason 3725323ac95bSChris Mason /* at the lowest level, we're done, setup the path and exit */ 3726323ac95bSChris Mason if (level == path->lowest_level) { 3727e02119d5SChris Mason if (slot >= nritems) 3728e02119d5SChris Mason goto find_next_key; 37293f157a2fSChris Mason ret = 0; 37303f157a2fSChris Mason path->slots[level] = slot; 37313f157a2fSChris Mason btrfs_item_key_to_cpu(cur, &found_key, slot); 37323f157a2fSChris Mason goto out; 37333f157a2fSChris Mason } 37349652480bSYan if (sret && slot > 0) 37359652480bSYan slot--; 37363f157a2fSChris Mason /* 37373f157a2fSChris Mason * check this node pointer against the cache_only and 37383f157a2fSChris Mason * min_trans parameters. If it isn't in cache or is too 37393f157a2fSChris Mason * old, skip to the next one. 37403f157a2fSChris Mason */ 37413f157a2fSChris Mason while(slot < nritems) { 37423f157a2fSChris Mason u64 blockptr; 37433f157a2fSChris Mason u64 gen; 37443f157a2fSChris Mason struct extent_buffer *tmp; 3745e02119d5SChris Mason struct btrfs_disk_key disk_key; 3746e02119d5SChris Mason 37473f157a2fSChris Mason blockptr = btrfs_node_blockptr(cur, slot); 37483f157a2fSChris Mason gen = btrfs_node_ptr_generation(cur, slot); 37493f157a2fSChris Mason if (gen < min_trans) { 37503f157a2fSChris Mason slot++; 37513f157a2fSChris Mason continue; 37523f157a2fSChris Mason } 37533f157a2fSChris Mason if (!cache_only) 37543f157a2fSChris Mason break; 37553f157a2fSChris Mason 3756e02119d5SChris Mason if (max_key) { 3757e02119d5SChris Mason btrfs_node_key(cur, &disk_key, slot); 3758e02119d5SChris Mason if (comp_keys(&disk_key, max_key) >= 0) { 3759e02119d5SChris Mason ret = 1; 3760e02119d5SChris Mason goto out; 3761e02119d5SChris Mason } 3762e02119d5SChris Mason } 3763e02119d5SChris Mason 37643f157a2fSChris Mason tmp = btrfs_find_tree_block(root, blockptr, 37653f157a2fSChris Mason btrfs_level_size(root, level - 1)); 37663f157a2fSChris Mason 37673f157a2fSChris Mason if (tmp && btrfs_buffer_uptodate(tmp, gen)) { 37683f157a2fSChris Mason free_extent_buffer(tmp); 37693f157a2fSChris Mason break; 37703f157a2fSChris Mason } 37713f157a2fSChris Mason if (tmp) 37723f157a2fSChris Mason free_extent_buffer(tmp); 37733f157a2fSChris Mason slot++; 37743f157a2fSChris Mason } 3775e02119d5SChris Mason find_next_key: 37763f157a2fSChris Mason /* 37773f157a2fSChris Mason * we didn't find a candidate key in this node, walk forward 37783f157a2fSChris Mason * and find another one 37793f157a2fSChris Mason */ 37803f157a2fSChris Mason if (slot >= nritems) { 3781e02119d5SChris Mason path->slots[level] = slot; 3782e02119d5SChris Mason sret = btrfs_find_next_key(root, path, min_key, level, 37833f157a2fSChris Mason cache_only, min_trans); 3784e02119d5SChris Mason if (sret == 0) { 37853f157a2fSChris Mason btrfs_release_path(root, path); 37863f157a2fSChris Mason goto again; 37873f157a2fSChris Mason } else { 37883f157a2fSChris Mason goto out; 37893f157a2fSChris Mason } 37903f157a2fSChris Mason } 37913f157a2fSChris Mason /* save our key for returning back */ 37923f157a2fSChris Mason btrfs_node_key_to_cpu(cur, &found_key, slot); 37933f157a2fSChris Mason path->slots[level] = slot; 37943f157a2fSChris Mason if (level == path->lowest_level) { 37953f157a2fSChris Mason ret = 0; 37963f157a2fSChris Mason unlock_up(path, level, 1); 37973f157a2fSChris Mason goto out; 37983f157a2fSChris Mason } 37993f157a2fSChris Mason cur = read_node_slot(root, cur, slot); 38003f157a2fSChris Mason 38013f157a2fSChris Mason btrfs_tree_lock(cur); 38023f157a2fSChris Mason path->locks[level - 1] = 1; 38033f157a2fSChris Mason path->nodes[level - 1] = cur; 38043f157a2fSChris Mason unlock_up(path, level, 1); 38053f157a2fSChris Mason } 38063f157a2fSChris Mason out: 38073f157a2fSChris Mason if (ret == 0) 38083f157a2fSChris Mason memcpy(min_key, &found_key, sizeof(found_key)); 38093f157a2fSChris Mason return ret; 38103f157a2fSChris Mason } 38113f157a2fSChris Mason 38123f157a2fSChris Mason /* 38133f157a2fSChris Mason * this is similar to btrfs_next_leaf, but does not try to preserve 38143f157a2fSChris Mason * and fixup the path. It looks for and returns the next key in the 38153f157a2fSChris Mason * tree based on the current path and the cache_only and min_trans 38163f157a2fSChris Mason * parameters. 38173f157a2fSChris Mason * 38183f157a2fSChris Mason * 0 is returned if another key is found, < 0 if there are any errors 38193f157a2fSChris Mason * and 1 is returned if there are no higher keys in the tree 38203f157a2fSChris Mason * 38213f157a2fSChris Mason * path->keep_locks should be set to 1 on the search made before 38223f157a2fSChris Mason * calling this function. 38233f157a2fSChris Mason */ 3824e7a84565SChris Mason int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path, 38253f157a2fSChris Mason struct btrfs_key *key, int lowest_level, 38263f157a2fSChris Mason int cache_only, u64 min_trans) 3827e7a84565SChris Mason { 3828e7a84565SChris Mason int level = lowest_level; 3829e7a84565SChris Mason int slot; 3830e7a84565SChris Mason struct extent_buffer *c; 3831e7a84565SChris Mason 3832934d375bSChris Mason WARN_ON(!path->keep_locks); 3833e7a84565SChris Mason while(level < BTRFS_MAX_LEVEL) { 3834e7a84565SChris Mason if (!path->nodes[level]) 3835e7a84565SChris Mason return 1; 3836e7a84565SChris Mason 3837e7a84565SChris Mason slot = path->slots[level] + 1; 3838e7a84565SChris Mason c = path->nodes[level]; 38393f157a2fSChris Mason next: 3840e7a84565SChris Mason if (slot >= btrfs_header_nritems(c)) { 3841e7a84565SChris Mason level++; 3842e7a84565SChris Mason if (level == BTRFS_MAX_LEVEL) { 3843e7a84565SChris Mason return 1; 3844e7a84565SChris Mason } 3845e7a84565SChris Mason continue; 3846e7a84565SChris Mason } 3847e7a84565SChris Mason if (level == 0) 3848e7a84565SChris Mason btrfs_item_key_to_cpu(c, key, slot); 38493f157a2fSChris Mason else { 38503f157a2fSChris Mason u64 blockptr = btrfs_node_blockptr(c, slot); 38513f157a2fSChris Mason u64 gen = btrfs_node_ptr_generation(c, slot); 38523f157a2fSChris Mason 38533f157a2fSChris Mason if (cache_only) { 38543f157a2fSChris Mason struct extent_buffer *cur; 38553f157a2fSChris Mason cur = btrfs_find_tree_block(root, blockptr, 38563f157a2fSChris Mason btrfs_level_size(root, level - 1)); 38573f157a2fSChris Mason if (!cur || !btrfs_buffer_uptodate(cur, gen)) { 38583f157a2fSChris Mason slot++; 38593f157a2fSChris Mason if (cur) 38603f157a2fSChris Mason free_extent_buffer(cur); 38613f157a2fSChris Mason goto next; 38623f157a2fSChris Mason } 38633f157a2fSChris Mason free_extent_buffer(cur); 38643f157a2fSChris Mason } 38653f157a2fSChris Mason if (gen < min_trans) { 38663f157a2fSChris Mason slot++; 38673f157a2fSChris Mason goto next; 38683f157a2fSChris Mason } 3869e7a84565SChris Mason btrfs_node_key_to_cpu(c, key, slot); 38703f157a2fSChris Mason } 3871e7a84565SChris Mason return 0; 3872e7a84565SChris Mason } 3873e7a84565SChris Mason return 1; 3874e7a84565SChris Mason } 3875e7a84565SChris Mason 38767bb86316SChris Mason /* 3877925baeddSChris Mason * search the tree again to find a leaf with greater keys 38780f70abe2SChris Mason * returns 0 if it found something or 1 if there are no greater leaves. 38790f70abe2SChris Mason * returns < 0 on io errors. 388097571fd0SChris Mason */ 3881234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path) 3882d97e63b6SChris Mason { 3883d97e63b6SChris Mason int slot; 3884d97e63b6SChris Mason int level = 1; 38855f39d397SChris Mason struct extent_buffer *c; 38865f39d397SChris Mason struct extent_buffer *next = NULL; 3887925baeddSChris Mason struct btrfs_key key; 3888925baeddSChris Mason u32 nritems; 3889925baeddSChris Mason int ret; 3890925baeddSChris Mason 3891925baeddSChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 3892925baeddSChris Mason if (nritems == 0) { 3893925baeddSChris Mason return 1; 3894925baeddSChris Mason } 3895925baeddSChris Mason 3896925baeddSChris Mason btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1); 3897925baeddSChris Mason 3898925baeddSChris Mason btrfs_release_path(root, path); 3899a2135011SChris Mason path->keep_locks = 1; 3900925baeddSChris Mason ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); 3901925baeddSChris Mason path->keep_locks = 0; 3902925baeddSChris Mason 3903925baeddSChris Mason if (ret < 0) 3904925baeddSChris Mason return ret; 3905925baeddSChris Mason 3906a2135011SChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 3907168fd7d2SChris Mason /* 3908168fd7d2SChris Mason * by releasing the path above we dropped all our locks. A balance 3909168fd7d2SChris Mason * could have added more items next to the key that used to be 3910168fd7d2SChris Mason * at the very end of the block. So, check again here and 3911168fd7d2SChris Mason * advance the path if there are now more items available. 3912168fd7d2SChris Mason */ 3913a2135011SChris Mason if (nritems > 0 && path->slots[0] < nritems - 1) { 3914168fd7d2SChris Mason path->slots[0]++; 3915925baeddSChris Mason goto done; 3916925baeddSChris Mason } 3917d97e63b6SChris Mason 3918234b63a0SChris Mason while(level < BTRFS_MAX_LEVEL) { 3919d97e63b6SChris Mason if (!path->nodes[level]) 39200f70abe2SChris Mason return 1; 39215f39d397SChris Mason 3922d97e63b6SChris Mason slot = path->slots[level] + 1; 3923d97e63b6SChris Mason c = path->nodes[level]; 39245f39d397SChris Mason if (slot >= btrfs_header_nritems(c)) { 3925d97e63b6SChris Mason level++; 3926925baeddSChris Mason if (level == BTRFS_MAX_LEVEL) { 39277bb86316SChris Mason return 1; 3928925baeddSChris Mason } 3929d97e63b6SChris Mason continue; 3930d97e63b6SChris Mason } 39315f39d397SChris Mason 3932925baeddSChris Mason if (next) { 3933925baeddSChris Mason btrfs_tree_unlock(next); 39345f39d397SChris Mason free_extent_buffer(next); 3935925baeddSChris Mason } 39365f39d397SChris Mason 39370bd40a71SChris Mason if (level == 1 && (path->locks[1] || path->skip_locking) && 39380bd40a71SChris Mason path->reada) 393901f46658SChris Mason reada_for_search(root, path, level, slot, 0); 39405f39d397SChris Mason 3941ca7a79adSChris Mason next = read_node_slot(root, c, slot); 39425cd57b2cSChris Mason if (!path->skip_locking) { 3943051e1b9fSChris Mason WARN_ON(!btrfs_tree_locked(c)); 3944925baeddSChris Mason btrfs_tree_lock(next); 39455cd57b2cSChris Mason } 3946d97e63b6SChris Mason break; 3947d97e63b6SChris Mason } 3948d97e63b6SChris Mason path->slots[level] = slot; 3949d97e63b6SChris Mason while(1) { 3950d97e63b6SChris Mason level--; 3951d97e63b6SChris Mason c = path->nodes[level]; 3952925baeddSChris Mason if (path->locks[level]) 3953925baeddSChris Mason btrfs_tree_unlock(c); 39545f39d397SChris Mason free_extent_buffer(c); 3955d97e63b6SChris Mason path->nodes[level] = next; 3956d97e63b6SChris Mason path->slots[level] = 0; 3957a74a4b97SChris Mason if (!path->skip_locking) 3958925baeddSChris Mason path->locks[level] = 1; 3959d97e63b6SChris Mason if (!level) 3960d97e63b6SChris Mason break; 3961925baeddSChris Mason if (level == 1 && path->locks[1] && path->reada) 3962925baeddSChris Mason reada_for_search(root, path, level, slot, 0); 3963ca7a79adSChris Mason next = read_node_slot(root, next, 0); 39645cd57b2cSChris Mason if (!path->skip_locking) { 3965a2135011SChris Mason WARN_ON(!btrfs_tree_locked(path->nodes[level])); 3966925baeddSChris Mason btrfs_tree_lock(next); 3967d97e63b6SChris Mason } 39685cd57b2cSChris Mason } 3969925baeddSChris Mason done: 3970925baeddSChris Mason unlock_up(path, 0, 1); 3971d97e63b6SChris Mason return 0; 3972d97e63b6SChris Mason } 39730b86a832SChris Mason 39743f157a2fSChris Mason /* 39753f157a2fSChris Mason * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps 39763f157a2fSChris Mason * searching until it gets past min_objectid or finds an item of 'type' 39773f157a2fSChris Mason * 39783f157a2fSChris Mason * returns 0 if something is found, 1 if nothing was found and < 0 on error 39793f157a2fSChris Mason */ 39800b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root, 39810b86a832SChris Mason struct btrfs_path *path, u64 min_objectid, 39820b86a832SChris Mason int type) 39830b86a832SChris Mason { 39840b86a832SChris Mason struct btrfs_key found_key; 39850b86a832SChris Mason struct extent_buffer *leaf; 3986e02119d5SChris Mason u32 nritems; 39870b86a832SChris Mason int ret; 39880b86a832SChris Mason 39890b86a832SChris Mason while(1) { 39900b86a832SChris Mason if (path->slots[0] == 0) { 39910b86a832SChris Mason ret = btrfs_prev_leaf(root, path); 39920b86a832SChris Mason if (ret != 0) 39930b86a832SChris Mason return ret; 39940b86a832SChris Mason } else { 39950b86a832SChris Mason path->slots[0]--; 39960b86a832SChris Mason } 39970b86a832SChris Mason leaf = path->nodes[0]; 3998e02119d5SChris Mason nritems = btrfs_header_nritems(leaf); 3999e02119d5SChris Mason if (nritems == 0) 4000e02119d5SChris Mason return 1; 4001e02119d5SChris Mason if (path->slots[0] == nritems) 4002e02119d5SChris Mason path->slots[0]--; 4003e02119d5SChris Mason 40040b86a832SChris Mason btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); 40050b86a832SChris Mason if (found_key.type == type) 40060b86a832SChris Mason return 0; 4007e02119d5SChris Mason if (found_key.objectid < min_objectid) 4008e02119d5SChris Mason break; 4009e02119d5SChris Mason if (found_key.objectid == min_objectid && 4010e02119d5SChris Mason found_key.type < type) 4011e02119d5SChris Mason break; 40120b86a832SChris Mason } 40130b86a832SChris Mason return 1; 40140b86a832SChris Mason } 4015