16cbd5570SChris Mason /* 26cbd5570SChris Mason * Copyright (C) 2007 Oracle. All rights reserved. 36cbd5570SChris Mason * 46cbd5570SChris Mason * This program is free software; you can redistribute it and/or 56cbd5570SChris Mason * modify it under the terms of the GNU General Public 66cbd5570SChris Mason * License v2 as published by the Free Software Foundation. 76cbd5570SChris Mason * 86cbd5570SChris Mason * This program is distributed in the hope that it will be useful, 96cbd5570SChris Mason * but WITHOUT ANY WARRANTY; without even the implied warranty of 106cbd5570SChris Mason * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 116cbd5570SChris Mason * General Public License for more details. 126cbd5570SChris Mason * 136cbd5570SChris Mason * You should have received a copy of the GNU General Public 146cbd5570SChris Mason * License along with this program; if not, write to the 156cbd5570SChris Mason * Free Software Foundation, Inc., 59 Temple Place - Suite 330, 166cbd5570SChris Mason * Boston, MA 021110-1307, USA. 176cbd5570SChris Mason */ 186cbd5570SChris Mason 19a6b6e75eSChris Mason #include <linux/sched.h> 20eb60ceacSChris Mason #include "ctree.h" 21eb60ceacSChris Mason #include "disk-io.h" 227f5c1516SChris Mason #include "transaction.h" 235f39d397SChris Mason #include "print-tree.h" 24925baeddSChris Mason #include "locking.h" 259a8dd150SChris Mason 26e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 27e089f05cSChris Mason *root, struct btrfs_path *path, int level); 28e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 29d4dbff95SChris Mason *root, struct btrfs_key *ins_key, 30cc0c5538SChris Mason struct btrfs_path *path, int data_size, int extend); 315f39d397SChris Mason static int push_node_left(struct btrfs_trans_handle *trans, 325f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *dst, 33971a1f66SChris Mason struct extent_buffer *src, int empty); 345f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 355f39d397SChris Mason struct btrfs_root *root, 365f39d397SChris Mason struct extent_buffer *dst_buf, 375f39d397SChris Mason struct extent_buffer *src_buf); 38e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 39e089f05cSChris Mason struct btrfs_path *path, int level, int slot); 40d97e63b6SChris Mason 41df24a2b9SChris Mason inline void btrfs_init_path(struct btrfs_path *p) 42df24a2b9SChris Mason { 43df24a2b9SChris Mason memset(p, 0, sizeof(*p)); 44df24a2b9SChris Mason } 45df24a2b9SChris Mason 462c90e5d6SChris Mason struct btrfs_path *btrfs_alloc_path(void) 472c90e5d6SChris Mason { 48df24a2b9SChris Mason struct btrfs_path *path; 49df24a2b9SChris Mason path = kmem_cache_alloc(btrfs_path_cachep, GFP_NOFS); 502cc58cf2SChris Mason if (path) { 51df24a2b9SChris Mason btrfs_init_path(path); 522cc58cf2SChris Mason path->reada = 1; 532cc58cf2SChris Mason } 54df24a2b9SChris Mason return path; 552c90e5d6SChris Mason } 562c90e5d6SChris Mason 572c90e5d6SChris Mason void btrfs_free_path(struct btrfs_path *p) 582c90e5d6SChris Mason { 59df24a2b9SChris Mason btrfs_release_path(NULL, p); 602c90e5d6SChris Mason kmem_cache_free(btrfs_path_cachep, p); 612c90e5d6SChris Mason } 622c90e5d6SChris Mason 63234b63a0SChris Mason void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p) 64eb60ceacSChris Mason { 65eb60ceacSChris Mason int i; 66a2135011SChris Mason 67234b63a0SChris Mason for (i = 0; i < BTRFS_MAX_LEVEL; i++) { 68*3f157a2fSChris Mason p->slots[i] = 0; 69eb60ceacSChris Mason if (!p->nodes[i]) 70925baeddSChris Mason continue; 71925baeddSChris Mason if (p->locks[i]) { 72925baeddSChris Mason btrfs_tree_unlock(p->nodes[i]); 73925baeddSChris Mason p->locks[i] = 0; 74925baeddSChris Mason } 755f39d397SChris Mason free_extent_buffer(p->nodes[i]); 76*3f157a2fSChris Mason p->nodes[i] = NULL; 77eb60ceacSChris Mason } 78eb60ceacSChris Mason } 79eb60ceacSChris Mason 80925baeddSChris Mason struct extent_buffer *btrfs_root_node(struct btrfs_root *root) 81925baeddSChris Mason { 82925baeddSChris Mason struct extent_buffer *eb; 83925baeddSChris Mason spin_lock(&root->node_lock); 84925baeddSChris Mason eb = root->node; 85925baeddSChris Mason extent_buffer_get(eb); 86925baeddSChris Mason spin_unlock(&root->node_lock); 87925baeddSChris Mason return eb; 88925baeddSChris Mason } 89925baeddSChris Mason 90925baeddSChris Mason struct extent_buffer *btrfs_lock_root_node(struct btrfs_root *root) 91925baeddSChris Mason { 92925baeddSChris Mason struct extent_buffer *eb; 93925baeddSChris Mason 94925baeddSChris Mason while(1) { 95925baeddSChris Mason eb = btrfs_root_node(root); 96925baeddSChris Mason btrfs_tree_lock(eb); 97925baeddSChris Mason 98925baeddSChris Mason spin_lock(&root->node_lock); 99925baeddSChris Mason if (eb == root->node) { 100925baeddSChris Mason spin_unlock(&root->node_lock); 101925baeddSChris Mason break; 102925baeddSChris Mason } 103925baeddSChris Mason spin_unlock(&root->node_lock); 104925baeddSChris Mason 105925baeddSChris Mason btrfs_tree_unlock(eb); 106925baeddSChris Mason free_extent_buffer(eb); 107925baeddSChris Mason } 108925baeddSChris Mason return eb; 109925baeddSChris Mason } 110925baeddSChris Mason 1110b86a832SChris Mason static void add_root_to_dirty_list(struct btrfs_root *root) 1120b86a832SChris Mason { 1130b86a832SChris Mason if (root->track_dirty && list_empty(&root->dirty_list)) { 1140b86a832SChris Mason list_add(&root->dirty_list, 1150b86a832SChris Mason &root->fs_info->dirty_cowonly_roots); 1160b86a832SChris Mason } 1170b86a832SChris Mason } 1180b86a832SChris Mason 119be20aa9dSChris Mason int btrfs_copy_root(struct btrfs_trans_handle *trans, 120be20aa9dSChris Mason struct btrfs_root *root, 121be20aa9dSChris Mason struct extent_buffer *buf, 122be20aa9dSChris Mason struct extent_buffer **cow_ret, u64 new_root_objectid) 123be20aa9dSChris Mason { 124be20aa9dSChris Mason struct extent_buffer *cow; 125be20aa9dSChris Mason u32 nritems; 126be20aa9dSChris Mason int ret = 0; 127be20aa9dSChris Mason int level; 128be20aa9dSChris Mason struct btrfs_key first_key; 1294aec2b52SChris Mason struct btrfs_root *new_root; 130be20aa9dSChris Mason 1314aec2b52SChris Mason new_root = kmalloc(sizeof(*new_root), GFP_NOFS); 1324aec2b52SChris Mason if (!new_root) 1334aec2b52SChris Mason return -ENOMEM; 1344aec2b52SChris Mason 1354aec2b52SChris Mason memcpy(new_root, root, sizeof(*new_root)); 1364aec2b52SChris Mason new_root->root_key.objectid = new_root_objectid; 137be20aa9dSChris Mason 138be20aa9dSChris Mason WARN_ON(root->ref_cows && trans->transid != 139be20aa9dSChris Mason root->fs_info->running_transaction->transid); 140be20aa9dSChris Mason WARN_ON(root->ref_cows && trans->transid != root->last_trans); 141be20aa9dSChris Mason 142be20aa9dSChris Mason level = btrfs_header_level(buf); 143be20aa9dSChris Mason nritems = btrfs_header_nritems(buf); 144be20aa9dSChris Mason if (nritems) { 145be20aa9dSChris Mason if (level == 0) 146be20aa9dSChris Mason btrfs_item_key_to_cpu(buf, &first_key, 0); 147be20aa9dSChris Mason else 148be20aa9dSChris Mason btrfs_node_key_to_cpu(buf, &first_key, 0); 149be20aa9dSChris Mason } else { 150be20aa9dSChris Mason first_key.objectid = 0; 151be20aa9dSChris Mason } 152925baeddSChris Mason cow = btrfs_alloc_free_block(trans, new_root, buf->len, 153be20aa9dSChris Mason new_root_objectid, 154be20aa9dSChris Mason trans->transid, first_key.objectid, 155be20aa9dSChris Mason level, buf->start, 0); 1564aec2b52SChris Mason if (IS_ERR(cow)) { 1574aec2b52SChris Mason kfree(new_root); 158be20aa9dSChris Mason return PTR_ERR(cow); 1594aec2b52SChris Mason } 160be20aa9dSChris Mason 161be20aa9dSChris Mason copy_extent_buffer(cow, buf, 0, 0, cow->len); 162be20aa9dSChris Mason btrfs_set_header_bytenr(cow, cow->start); 163be20aa9dSChris Mason btrfs_set_header_generation(cow, trans->transid); 164be20aa9dSChris Mason btrfs_set_header_owner(cow, new_root_objectid); 16563b10fc4SChris Mason btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN); 166be20aa9dSChris Mason 167be20aa9dSChris Mason WARN_ON(btrfs_header_generation(buf) > trans->transid); 1684aec2b52SChris Mason ret = btrfs_inc_ref(trans, new_root, buf); 1694aec2b52SChris Mason kfree(new_root); 1704aec2b52SChris Mason 171be20aa9dSChris Mason if (ret) 172be20aa9dSChris Mason return ret; 173be20aa9dSChris Mason 174be20aa9dSChris Mason btrfs_mark_buffer_dirty(cow); 175be20aa9dSChris Mason *cow_ret = cow; 176be20aa9dSChris Mason return 0; 177be20aa9dSChris Mason } 178be20aa9dSChris Mason 179be20aa9dSChris Mason int __btrfs_cow_block(struct btrfs_trans_handle *trans, 1805f39d397SChris Mason struct btrfs_root *root, 1815f39d397SChris Mason struct extent_buffer *buf, 1825f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 1835f39d397SChris Mason struct extent_buffer **cow_ret, 1845f39d397SChris Mason u64 search_start, u64 empty_size) 1856702ed49SChris Mason { 1867bb86316SChris Mason u64 root_gen; 1875f39d397SChris Mason struct extent_buffer *cow; 1887bb86316SChris Mason u32 nritems; 1896702ed49SChris Mason int ret = 0; 1906702ed49SChris Mason int different_trans = 0; 1917bb86316SChris Mason int level; 192925baeddSChris Mason int unlock_orig = 0; 1937bb86316SChris Mason struct btrfs_key first_key; 1946702ed49SChris Mason 195925baeddSChris Mason if (*cow_ret == buf) 196925baeddSChris Mason unlock_orig = 1; 197925baeddSChris Mason 198925baeddSChris Mason WARN_ON(!btrfs_tree_locked(buf)); 199925baeddSChris Mason 2007bb86316SChris Mason if (root->ref_cows) { 2017bb86316SChris Mason root_gen = trans->transid; 2027bb86316SChris Mason } else { 2037bb86316SChris Mason root_gen = 0; 2047bb86316SChris Mason } 2057bb86316SChris Mason WARN_ON(root->ref_cows && trans->transid != 2067bb86316SChris Mason root->fs_info->running_transaction->transid); 2076702ed49SChris Mason WARN_ON(root->ref_cows && trans->transid != root->last_trans); 2085f39d397SChris Mason 2097bb86316SChris Mason level = btrfs_header_level(buf); 2107bb86316SChris Mason nritems = btrfs_header_nritems(buf); 2117bb86316SChris Mason if (nritems) { 2127bb86316SChris Mason if (level == 0) 2137bb86316SChris Mason btrfs_item_key_to_cpu(buf, &first_key, 0); 2147bb86316SChris Mason else 2157bb86316SChris Mason btrfs_node_key_to_cpu(buf, &first_key, 0); 2167bb86316SChris Mason } else { 2177bb86316SChris Mason first_key.objectid = 0; 2187bb86316SChris Mason } 219925baeddSChris Mason cow = btrfs_alloc_free_block(trans, root, buf->len, 2207bb86316SChris Mason root->root_key.objectid, 2217bb86316SChris Mason root_gen, first_key.objectid, level, 222db94535dSChris Mason search_start, empty_size); 2236702ed49SChris Mason if (IS_ERR(cow)) 2246702ed49SChris Mason return PTR_ERR(cow); 2256702ed49SChris Mason 2265f39d397SChris Mason copy_extent_buffer(cow, buf, 0, 0, cow->len); 227db94535dSChris Mason btrfs_set_header_bytenr(cow, cow->start); 2285f39d397SChris Mason btrfs_set_header_generation(cow, trans->transid); 2295f39d397SChris Mason btrfs_set_header_owner(cow, root->root_key.objectid); 23063b10fc4SChris Mason btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN); 2316702ed49SChris Mason 2325f39d397SChris Mason WARN_ON(btrfs_header_generation(buf) > trans->transid); 2335f39d397SChris Mason if (btrfs_header_generation(buf) != trans->transid) { 2346702ed49SChris Mason different_trans = 1; 2356702ed49SChris Mason ret = btrfs_inc_ref(trans, root, buf); 2366702ed49SChris Mason if (ret) 2376702ed49SChris Mason return ret; 2386702ed49SChris Mason } else { 2396702ed49SChris Mason clean_tree_block(trans, root, buf); 2406702ed49SChris Mason } 2416702ed49SChris Mason 2426702ed49SChris Mason if (buf == root->node) { 243925baeddSChris Mason WARN_ON(parent && parent != buf); 2447bb86316SChris Mason root_gen = btrfs_header_generation(buf); 245925baeddSChris Mason 246925baeddSChris Mason spin_lock(&root->node_lock); 2476702ed49SChris Mason root->node = cow; 2485f39d397SChris Mason extent_buffer_get(cow); 249925baeddSChris Mason spin_unlock(&root->node_lock); 250925baeddSChris Mason 2516702ed49SChris Mason if (buf != root->commit_root) { 252db94535dSChris Mason btrfs_free_extent(trans, root, buf->start, 2537bb86316SChris Mason buf->len, root->root_key.objectid, 2547bb86316SChris Mason root_gen, 0, 0, 1); 2556702ed49SChris Mason } 2565f39d397SChris Mason free_extent_buffer(buf); 2570b86a832SChris Mason add_root_to_dirty_list(root); 2586702ed49SChris Mason } else { 2597bb86316SChris Mason root_gen = btrfs_header_generation(parent); 2605f39d397SChris Mason btrfs_set_node_blockptr(parent, parent_slot, 261db94535dSChris Mason cow->start); 26274493f7aSChris Mason WARN_ON(trans->transid == 0); 26374493f7aSChris Mason btrfs_set_node_ptr_generation(parent, parent_slot, 26474493f7aSChris Mason trans->transid); 2656702ed49SChris Mason btrfs_mark_buffer_dirty(parent); 2665f39d397SChris Mason WARN_ON(btrfs_header_generation(parent) != trans->transid); 2677bb86316SChris Mason btrfs_free_extent(trans, root, buf->start, buf->len, 2687bb86316SChris Mason btrfs_header_owner(parent), root_gen, 2697bb86316SChris Mason 0, 0, 1); 2706702ed49SChris Mason } 271925baeddSChris Mason if (unlock_orig) 272925baeddSChris Mason btrfs_tree_unlock(buf); 2735f39d397SChris Mason free_extent_buffer(buf); 2746702ed49SChris Mason btrfs_mark_buffer_dirty(cow); 2756702ed49SChris Mason *cow_ret = cow; 2766702ed49SChris Mason return 0; 2776702ed49SChris Mason } 2786702ed49SChris Mason 2795f39d397SChris Mason int btrfs_cow_block(struct btrfs_trans_handle *trans, 2805f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *buf, 2815f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 2825f39d397SChris Mason struct extent_buffer **cow_ret) 28302217ed2SChris Mason { 2846702ed49SChris Mason u64 search_start; 285dc17ff8fSChris Mason u64 header_trans; 286f510cfecSChris Mason int ret; 287dc17ff8fSChris Mason 288ccd467d6SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 289ccd467d6SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 290ccd467d6SChris Mason root->fs_info->running_transaction->transid); 291ccd467d6SChris Mason WARN_ON(1); 292ccd467d6SChris Mason } 293ccd467d6SChris Mason if (trans->transid != root->fs_info->generation) { 294ccd467d6SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 295ccd467d6SChris Mason root->fs_info->generation); 296ccd467d6SChris Mason WARN_ON(1); 297ccd467d6SChris Mason } 298dc17ff8fSChris Mason 299dc17ff8fSChris Mason header_trans = btrfs_header_generation(buf); 30063b10fc4SChris Mason spin_lock(&root->fs_info->hash_lock); 30163b10fc4SChris Mason if (header_trans == trans->transid && 30263b10fc4SChris Mason !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN)) { 30302217ed2SChris Mason *cow_ret = buf; 30463b10fc4SChris Mason spin_unlock(&root->fs_info->hash_lock); 30502217ed2SChris Mason return 0; 30602217ed2SChris Mason } 30763b10fc4SChris Mason spin_unlock(&root->fs_info->hash_lock); 3080b86a832SChris Mason search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1); 309f510cfecSChris Mason ret = __btrfs_cow_block(trans, root, buf, parent, 3106702ed49SChris Mason parent_slot, cow_ret, search_start, 0); 311f510cfecSChris Mason return ret; 3122c90e5d6SChris Mason } 3136702ed49SChris Mason 3146b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize) 3156702ed49SChris Mason { 3166b80053dSChris Mason if (blocknr < other && other - (blocknr + blocksize) < 32768) 3176702ed49SChris Mason return 1; 3186b80053dSChris Mason if (blocknr > other && blocknr - (other + blocksize) < 32768) 3196702ed49SChris Mason return 1; 32002217ed2SChris Mason return 0; 32102217ed2SChris Mason } 32202217ed2SChris Mason 323081e9573SChris Mason /* 324081e9573SChris Mason * compare two keys in a memcmp fashion 325081e9573SChris Mason */ 326081e9573SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2) 327081e9573SChris Mason { 328081e9573SChris Mason struct btrfs_key k1; 329081e9573SChris Mason 330081e9573SChris Mason btrfs_disk_key_to_cpu(&k1, disk); 331081e9573SChris Mason 332081e9573SChris Mason if (k1.objectid > k2->objectid) 333081e9573SChris Mason return 1; 334081e9573SChris Mason if (k1.objectid < k2->objectid) 335081e9573SChris Mason return -1; 336081e9573SChris Mason if (k1.type > k2->type) 337081e9573SChris Mason return 1; 338081e9573SChris Mason if (k1.type < k2->type) 339081e9573SChris Mason return -1; 340081e9573SChris Mason if (k1.offset > k2->offset) 341081e9573SChris Mason return 1; 342081e9573SChris Mason if (k1.offset < k2->offset) 343081e9573SChris Mason return -1; 344081e9573SChris Mason return 0; 345081e9573SChris Mason } 346081e9573SChris Mason 347081e9573SChris Mason 3486702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans, 3495f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *parent, 350a6b6e75eSChris Mason int start_slot, int cache_only, u64 *last_ret, 351a6b6e75eSChris Mason struct btrfs_key *progress) 3526702ed49SChris Mason { 3536b80053dSChris Mason struct extent_buffer *cur; 3546702ed49SChris Mason u64 blocknr; 355ca7a79adSChris Mason u64 gen; 356e9d0b13bSChris Mason u64 search_start = *last_ret; 357e9d0b13bSChris Mason u64 last_block = 0; 3586702ed49SChris Mason u64 other; 3596702ed49SChris Mason u32 parent_nritems; 3606702ed49SChris Mason int end_slot; 3616702ed49SChris Mason int i; 3626702ed49SChris Mason int err = 0; 363f2183bdeSChris Mason int parent_level; 3646b80053dSChris Mason int uptodate; 3656b80053dSChris Mason u32 blocksize; 366081e9573SChris Mason int progress_passed = 0; 367081e9573SChris Mason struct btrfs_disk_key disk_key; 3686702ed49SChris Mason 3695708b959SChris Mason parent_level = btrfs_header_level(parent); 3705708b959SChris Mason if (cache_only && parent_level != 1) 3715708b959SChris Mason return 0; 3725708b959SChris Mason 3736702ed49SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 3746702ed49SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 3756702ed49SChris Mason root->fs_info->running_transaction->transid); 3766702ed49SChris Mason WARN_ON(1); 3776702ed49SChris Mason } 3786702ed49SChris Mason if (trans->transid != root->fs_info->generation) { 3796702ed49SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 3806702ed49SChris Mason root->fs_info->generation); 3816702ed49SChris Mason WARN_ON(1); 3826702ed49SChris Mason } 38386479a04SChris Mason 3846b80053dSChris Mason parent_nritems = btrfs_header_nritems(parent); 3856b80053dSChris Mason blocksize = btrfs_level_size(root, parent_level - 1); 3866702ed49SChris Mason end_slot = parent_nritems; 3876702ed49SChris Mason 3886702ed49SChris Mason if (parent_nritems == 1) 3896702ed49SChris Mason return 0; 3906702ed49SChris Mason 3916702ed49SChris Mason for (i = start_slot; i < end_slot; i++) { 3926702ed49SChris Mason int close = 1; 393a6b6e75eSChris Mason 3945708b959SChris Mason if (!parent->map_token) { 3955708b959SChris Mason map_extent_buffer(parent, 3965708b959SChris Mason btrfs_node_key_ptr_offset(i), 3975708b959SChris Mason sizeof(struct btrfs_key_ptr), 3985708b959SChris Mason &parent->map_token, &parent->kaddr, 3995708b959SChris Mason &parent->map_start, &parent->map_len, 4005708b959SChris Mason KM_USER1); 4015708b959SChris Mason } 402081e9573SChris Mason btrfs_node_key(parent, &disk_key, i); 403081e9573SChris Mason if (!progress_passed && comp_keys(&disk_key, progress) < 0) 404081e9573SChris Mason continue; 405081e9573SChris Mason 406081e9573SChris Mason progress_passed = 1; 4076b80053dSChris Mason blocknr = btrfs_node_blockptr(parent, i); 408ca7a79adSChris Mason gen = btrfs_node_ptr_generation(parent, i); 409e9d0b13bSChris Mason if (last_block == 0) 410e9d0b13bSChris Mason last_block = blocknr; 4115708b959SChris Mason 4126702ed49SChris Mason if (i > 0) { 4136b80053dSChris Mason other = btrfs_node_blockptr(parent, i - 1); 4146b80053dSChris Mason close = close_blocks(blocknr, other, blocksize); 4156702ed49SChris Mason } 4160ef3e66bSChris Mason if (!close && i < end_slot - 2) { 4176b80053dSChris Mason other = btrfs_node_blockptr(parent, i + 1); 4186b80053dSChris Mason close = close_blocks(blocknr, other, blocksize); 4196702ed49SChris Mason } 420e9d0b13bSChris Mason if (close) { 421e9d0b13bSChris Mason last_block = blocknr; 4226702ed49SChris Mason continue; 423e9d0b13bSChris Mason } 4245708b959SChris Mason if (parent->map_token) { 4255708b959SChris Mason unmap_extent_buffer(parent, parent->map_token, 4265708b959SChris Mason KM_USER1); 4275708b959SChris Mason parent->map_token = NULL; 4285708b959SChris Mason } 4296702ed49SChris Mason 4306b80053dSChris Mason cur = btrfs_find_tree_block(root, blocknr, blocksize); 4316b80053dSChris Mason if (cur) 4321259ab75SChris Mason uptodate = btrfs_buffer_uptodate(cur, gen); 4336b80053dSChris Mason else 4346b80053dSChris Mason uptodate = 0; 4355708b959SChris Mason if (!cur || !uptodate) { 4366702ed49SChris Mason if (cache_only) { 4376b80053dSChris Mason free_extent_buffer(cur); 4386702ed49SChris Mason continue; 4396702ed49SChris Mason } 4406b80053dSChris Mason if (!cur) { 4416b80053dSChris Mason cur = read_tree_block(root, blocknr, 442ca7a79adSChris Mason blocksize, gen); 4436b80053dSChris Mason } else if (!uptodate) { 444ca7a79adSChris Mason btrfs_read_buffer(cur, gen); 4456702ed49SChris Mason } 446f2183bdeSChris Mason } 447e9d0b13bSChris Mason if (search_start == 0) 4486b80053dSChris Mason search_start = last_block; 449e9d0b13bSChris Mason 450e7a84565SChris Mason btrfs_tree_lock(cur); 4516b80053dSChris Mason err = __btrfs_cow_block(trans, root, cur, parent, i, 452e7a84565SChris Mason &cur, search_start, 4536b80053dSChris Mason min(16 * blocksize, 4546b80053dSChris Mason (end_slot - i) * blocksize)); 455252c38f0SYan if (err) { 456e7a84565SChris Mason btrfs_tree_unlock(cur); 4576b80053dSChris Mason free_extent_buffer(cur); 4586702ed49SChris Mason break; 459252c38f0SYan } 460e7a84565SChris Mason search_start = cur->start; 461e7a84565SChris Mason last_block = cur->start; 462f2183bdeSChris Mason *last_ret = search_start; 463e7a84565SChris Mason btrfs_tree_unlock(cur); 464e7a84565SChris Mason free_extent_buffer(cur); 4656702ed49SChris Mason } 4665708b959SChris Mason if (parent->map_token) { 4675708b959SChris Mason unmap_extent_buffer(parent, parent->map_token, 4685708b959SChris Mason KM_USER1); 4695708b959SChris Mason parent->map_token = NULL; 4705708b959SChris Mason } 4716702ed49SChris Mason return err; 4726702ed49SChris Mason } 4736702ed49SChris Mason 47474123bd7SChris Mason /* 47574123bd7SChris Mason * The leaf data grows from end-to-front in the node. 47674123bd7SChris Mason * this returns the address of the start of the last item, 47774123bd7SChris Mason * which is the stop of the leaf data stack 47874123bd7SChris Mason */ 479123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root, 4805f39d397SChris Mason struct extent_buffer *leaf) 481be0e5c09SChris Mason { 4825f39d397SChris Mason u32 nr = btrfs_header_nritems(leaf); 483be0e5c09SChris Mason if (nr == 0) 484123abc88SChris Mason return BTRFS_LEAF_DATA_SIZE(root); 4855f39d397SChris Mason return btrfs_item_offset_nr(leaf, nr - 1); 486be0e5c09SChris Mason } 487be0e5c09SChris Mason 488123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path, 489123abc88SChris Mason int level) 490aa5d6bedSChris Mason { 4915f39d397SChris Mason struct extent_buffer *parent = NULL; 4925f39d397SChris Mason struct extent_buffer *node = path->nodes[level]; 4935f39d397SChris Mason struct btrfs_disk_key parent_key; 4945f39d397SChris Mason struct btrfs_disk_key node_key; 495aa5d6bedSChris Mason int parent_slot; 4968d7be552SChris Mason int slot; 4978d7be552SChris Mason struct btrfs_key cpukey; 4985f39d397SChris Mason u32 nritems = btrfs_header_nritems(node); 499aa5d6bedSChris Mason 500aa5d6bedSChris Mason if (path->nodes[level + 1]) 5015f39d397SChris Mason parent = path->nodes[level + 1]; 502a1f39630SAneesh 5038d7be552SChris Mason slot = path->slots[level]; 5047518a238SChris Mason BUG_ON(nritems == 0); 5057518a238SChris Mason if (parent) { 506a1f39630SAneesh parent_slot = path->slots[level + 1]; 5075f39d397SChris Mason btrfs_node_key(parent, &parent_key, parent_slot); 5085f39d397SChris Mason btrfs_node_key(node, &node_key, 0); 5095f39d397SChris Mason BUG_ON(memcmp(&parent_key, &node_key, 510e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 5111d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 512db94535dSChris Mason btrfs_header_bytenr(node)); 513aa5d6bedSChris Mason } 514123abc88SChris Mason BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root)); 5158d7be552SChris Mason if (slot != 0) { 5165f39d397SChris Mason btrfs_node_key_to_cpu(node, &cpukey, slot - 1); 5175f39d397SChris Mason btrfs_node_key(node, &node_key, slot); 5185f39d397SChris Mason BUG_ON(comp_keys(&node_key, &cpukey) <= 0); 5198d7be552SChris Mason } 5208d7be552SChris Mason if (slot < nritems - 1) { 5215f39d397SChris Mason btrfs_node_key_to_cpu(node, &cpukey, slot + 1); 5225f39d397SChris Mason btrfs_node_key(node, &node_key, slot); 5235f39d397SChris Mason BUG_ON(comp_keys(&node_key, &cpukey) >= 0); 524aa5d6bedSChris Mason } 525aa5d6bedSChris Mason return 0; 526aa5d6bedSChris Mason } 527aa5d6bedSChris Mason 528123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path, 529123abc88SChris Mason int level) 530aa5d6bedSChris Mason { 5315f39d397SChris Mason struct extent_buffer *leaf = path->nodes[level]; 5325f39d397SChris Mason struct extent_buffer *parent = NULL; 533aa5d6bedSChris Mason int parent_slot; 5348d7be552SChris Mason struct btrfs_key cpukey; 5355f39d397SChris Mason struct btrfs_disk_key parent_key; 5365f39d397SChris Mason struct btrfs_disk_key leaf_key; 5375f39d397SChris Mason int slot = path->slots[0]; 5388d7be552SChris Mason 5395f39d397SChris Mason u32 nritems = btrfs_header_nritems(leaf); 540aa5d6bedSChris Mason 541aa5d6bedSChris Mason if (path->nodes[level + 1]) 5425f39d397SChris Mason parent = path->nodes[level + 1]; 5437518a238SChris Mason 5447518a238SChris Mason if (nritems == 0) 5457518a238SChris Mason return 0; 5467518a238SChris Mason 5477518a238SChris Mason if (parent) { 548a1f39630SAneesh parent_slot = path->slots[level + 1]; 5495f39d397SChris Mason btrfs_node_key(parent, &parent_key, parent_slot); 5505f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, 0); 5516702ed49SChris Mason 5525f39d397SChris Mason BUG_ON(memcmp(&parent_key, &leaf_key, 553e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 5541d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 555db94535dSChris Mason btrfs_header_bytenr(leaf)); 556aa5d6bedSChris Mason } 5575f39d397SChris Mason #if 0 5585f39d397SChris Mason for (i = 0; nritems > 1 && i < nritems - 2; i++) { 5595f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, i + 1); 5605f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, i); 5615f39d397SChris Mason if (comp_keys(&leaf_key, &cpukey) >= 0) { 5625f39d397SChris Mason btrfs_print_leaf(root, leaf); 5635f39d397SChris Mason printk("slot %d offset bad key\n", i); 5645f39d397SChris Mason BUG_ON(1); 5655f39d397SChris Mason } 5665f39d397SChris Mason if (btrfs_item_offset_nr(leaf, i) != 5675f39d397SChris Mason btrfs_item_end_nr(leaf, i + 1)) { 5685f39d397SChris Mason btrfs_print_leaf(root, leaf); 5695f39d397SChris Mason printk("slot %d offset bad\n", i); 5705f39d397SChris Mason BUG_ON(1); 5715f39d397SChris Mason } 5725f39d397SChris Mason if (i == 0) { 5735f39d397SChris Mason if (btrfs_item_offset_nr(leaf, i) + 5745f39d397SChris Mason btrfs_item_size_nr(leaf, i) != 5755f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 5765f39d397SChris Mason btrfs_print_leaf(root, leaf); 5775f39d397SChris Mason printk("slot %d first offset bad\n", i); 5785f39d397SChris Mason BUG_ON(1); 5795f39d397SChris Mason } 5805f39d397SChris Mason } 5815f39d397SChris Mason } 5825f39d397SChris Mason if (nritems > 0) { 5835f39d397SChris Mason if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) { 5845f39d397SChris Mason btrfs_print_leaf(root, leaf); 5855f39d397SChris Mason printk("slot %d bad size \n", nritems - 1); 5865f39d397SChris Mason BUG_ON(1); 5875f39d397SChris Mason } 5885f39d397SChris Mason } 5895f39d397SChris Mason #endif 5905f39d397SChris Mason if (slot != 0 && slot < nritems - 1) { 5915f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, slot); 5925f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1); 5935f39d397SChris Mason if (comp_keys(&leaf_key, &cpukey) <= 0) { 5945f39d397SChris Mason btrfs_print_leaf(root, leaf); 5955f39d397SChris Mason printk("slot %d offset bad key\n", slot); 5965f39d397SChris Mason BUG_ON(1); 5975f39d397SChris Mason } 5985f39d397SChris Mason if (btrfs_item_offset_nr(leaf, slot - 1) != 5995f39d397SChris Mason btrfs_item_end_nr(leaf, slot)) { 6005f39d397SChris Mason btrfs_print_leaf(root, leaf); 6015f39d397SChris Mason printk("slot %d offset bad\n", slot); 6025f39d397SChris Mason BUG_ON(1); 6035f39d397SChris Mason } 604aa5d6bedSChris Mason } 6058d7be552SChris Mason if (slot < nritems - 1) { 6065f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, slot); 6075f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1); 6085f39d397SChris Mason BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0); 6095f39d397SChris Mason if (btrfs_item_offset_nr(leaf, slot) != 6105f39d397SChris Mason btrfs_item_end_nr(leaf, slot + 1)) { 6115f39d397SChris Mason btrfs_print_leaf(root, leaf); 6125f39d397SChris Mason printk("slot %d offset bad\n", slot); 6135f39d397SChris Mason BUG_ON(1); 614aa5d6bedSChris Mason } 6155f39d397SChris Mason } 6165f39d397SChris Mason BUG_ON(btrfs_item_offset_nr(leaf, 0) + 6175f39d397SChris Mason btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root)); 618aa5d6bedSChris Mason return 0; 619aa5d6bedSChris Mason } 620aa5d6bedSChris Mason 62198ed5174SChris Mason static int noinline check_block(struct btrfs_root *root, 62298ed5174SChris Mason struct btrfs_path *path, int level) 623aa5d6bedSChris Mason { 624f188591eSChris Mason u64 found_start; 62585d824c4SChris Mason return 0; 626f188591eSChris Mason if (btrfs_header_level(path->nodes[level]) != level) 627f188591eSChris Mason printk("warning: bad level %Lu wanted %d found %d\n", 628f188591eSChris Mason path->nodes[level]->start, level, 629f188591eSChris Mason btrfs_header_level(path->nodes[level])); 630f188591eSChris Mason found_start = btrfs_header_bytenr(path->nodes[level]); 631f188591eSChris Mason if (found_start != path->nodes[level]->start) { 632f188591eSChris Mason printk("warning: bad bytentr %Lu found %Lu\n", 633f188591eSChris Mason path->nodes[level]->start, found_start); 634f188591eSChris Mason } 635db94535dSChris Mason #if 0 6365f39d397SChris Mason struct extent_buffer *buf = path->nodes[level]; 6375f39d397SChris Mason 638479965d6SChris Mason if (memcmp_extent_buffer(buf, root->fs_info->fsid, 639479965d6SChris Mason (unsigned long)btrfs_header_fsid(buf), 640479965d6SChris Mason BTRFS_FSID_SIZE)) { 6415f39d397SChris Mason printk("warning bad block %Lu\n", buf->start); 642db94535dSChris Mason return 1; 6435f39d397SChris Mason } 644db94535dSChris Mason #endif 645aa5d6bedSChris Mason if (level == 0) 646123abc88SChris Mason return check_leaf(root, path, level); 647123abc88SChris Mason return check_node(root, path, level); 648aa5d6bedSChris Mason } 649aa5d6bedSChris Mason 65074123bd7SChris Mason /* 6515f39d397SChris Mason * search for key in the extent_buffer. The items start at offset p, 6525f39d397SChris Mason * and they are item_size apart. There are 'max' items in p. 6535f39d397SChris Mason * 65474123bd7SChris Mason * the slot in the array is returned via slot, and it points to 65574123bd7SChris Mason * the place where you would insert key if it is not found in 65674123bd7SChris Mason * the array. 65774123bd7SChris Mason * 65874123bd7SChris Mason * slot may point to max if the key is bigger than all of the keys 65974123bd7SChris Mason */ 6605f39d397SChris Mason static int generic_bin_search(struct extent_buffer *eb, unsigned long p, 6615f39d397SChris Mason int item_size, struct btrfs_key *key, 662be0e5c09SChris Mason int max, int *slot) 663be0e5c09SChris Mason { 664be0e5c09SChris Mason int low = 0; 665be0e5c09SChris Mason int high = max; 666be0e5c09SChris Mason int mid; 667be0e5c09SChris Mason int ret; 668479965d6SChris Mason struct btrfs_disk_key *tmp = NULL; 6695f39d397SChris Mason struct btrfs_disk_key unaligned; 6705f39d397SChris Mason unsigned long offset; 6715f39d397SChris Mason char *map_token = NULL; 6725f39d397SChris Mason char *kaddr = NULL; 6735f39d397SChris Mason unsigned long map_start = 0; 6745f39d397SChris Mason unsigned long map_len = 0; 675479965d6SChris Mason int err; 676be0e5c09SChris Mason 677be0e5c09SChris Mason while(low < high) { 678be0e5c09SChris Mason mid = (low + high) / 2; 6795f39d397SChris Mason offset = p + mid * item_size; 6805f39d397SChris Mason 6815f39d397SChris Mason if (!map_token || offset < map_start || 6825f39d397SChris Mason (offset + sizeof(struct btrfs_disk_key)) > 6835f39d397SChris Mason map_start + map_len) { 684479965d6SChris Mason if (map_token) { 6855f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 686479965d6SChris Mason map_token = NULL; 687479965d6SChris Mason } 688479965d6SChris Mason err = map_extent_buffer(eb, offset, 689479965d6SChris Mason sizeof(struct btrfs_disk_key), 690479965d6SChris Mason &map_token, &kaddr, 6915f39d397SChris Mason &map_start, &map_len, KM_USER0); 6925f39d397SChris Mason 693479965d6SChris Mason if (!err) { 694479965d6SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 695479965d6SChris Mason map_start); 696479965d6SChris Mason } else { 6975f39d397SChris Mason read_extent_buffer(eb, &unaligned, 6985f39d397SChris Mason offset, sizeof(unaligned)); 6995f39d397SChris Mason tmp = &unaligned; 700479965d6SChris Mason } 701479965d6SChris Mason 7025f39d397SChris Mason } else { 7035f39d397SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 7045f39d397SChris Mason map_start); 7055f39d397SChris Mason } 706be0e5c09SChris Mason ret = comp_keys(tmp, key); 707be0e5c09SChris Mason 708be0e5c09SChris Mason if (ret < 0) 709be0e5c09SChris Mason low = mid + 1; 710be0e5c09SChris Mason else if (ret > 0) 711be0e5c09SChris Mason high = mid; 712be0e5c09SChris Mason else { 713be0e5c09SChris Mason *slot = mid; 714479965d6SChris Mason if (map_token) 7155f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 716be0e5c09SChris Mason return 0; 717be0e5c09SChris Mason } 718be0e5c09SChris Mason } 719be0e5c09SChris Mason *slot = low; 7205f39d397SChris Mason if (map_token) 7215f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 722be0e5c09SChris Mason return 1; 723be0e5c09SChris Mason } 724be0e5c09SChris Mason 72597571fd0SChris Mason /* 72697571fd0SChris Mason * simple bin_search frontend that does the right thing for 72797571fd0SChris Mason * leaves vs nodes 72897571fd0SChris Mason */ 7295f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key, 7305f39d397SChris Mason int level, int *slot) 731be0e5c09SChris Mason { 7325f39d397SChris Mason if (level == 0) { 7335f39d397SChris Mason return generic_bin_search(eb, 7345f39d397SChris Mason offsetof(struct btrfs_leaf, items), 7350783fcfcSChris Mason sizeof(struct btrfs_item), 7365f39d397SChris Mason key, btrfs_header_nritems(eb), 7377518a238SChris Mason slot); 738be0e5c09SChris Mason } else { 7395f39d397SChris Mason return generic_bin_search(eb, 7405f39d397SChris Mason offsetof(struct btrfs_node, ptrs), 741123abc88SChris Mason sizeof(struct btrfs_key_ptr), 7425f39d397SChris Mason key, btrfs_header_nritems(eb), 7437518a238SChris Mason slot); 744be0e5c09SChris Mason } 745be0e5c09SChris Mason return -1; 746be0e5c09SChris Mason } 747be0e5c09SChris Mason 7485f39d397SChris Mason static struct extent_buffer *read_node_slot(struct btrfs_root *root, 7495f39d397SChris Mason struct extent_buffer *parent, int slot) 750bb803951SChris Mason { 751ca7a79adSChris Mason int level = btrfs_header_level(parent); 752bb803951SChris Mason if (slot < 0) 753bb803951SChris Mason return NULL; 7545f39d397SChris Mason if (slot >= btrfs_header_nritems(parent)) 755bb803951SChris Mason return NULL; 756ca7a79adSChris Mason 757ca7a79adSChris Mason BUG_ON(level == 0); 758ca7a79adSChris Mason 759db94535dSChris Mason return read_tree_block(root, btrfs_node_blockptr(parent, slot), 760ca7a79adSChris Mason btrfs_level_size(root, level - 1), 761ca7a79adSChris Mason btrfs_node_ptr_generation(parent, slot)); 762bb803951SChris Mason } 763bb803951SChris Mason 76498ed5174SChris Mason static int balance_level(struct btrfs_trans_handle *trans, 76598ed5174SChris Mason struct btrfs_root *root, 76698ed5174SChris Mason struct btrfs_path *path, int level) 767bb803951SChris Mason { 7685f39d397SChris Mason struct extent_buffer *right = NULL; 7695f39d397SChris Mason struct extent_buffer *mid; 7705f39d397SChris Mason struct extent_buffer *left = NULL; 7715f39d397SChris Mason struct extent_buffer *parent = NULL; 772bb803951SChris Mason int ret = 0; 773bb803951SChris Mason int wret; 774bb803951SChris Mason int pslot; 775bb803951SChris Mason int orig_slot = path->slots[level]; 77654aa1f4dSChris Mason int err_on_enospc = 0; 77779f95c82SChris Mason u64 orig_ptr; 778bb803951SChris Mason 779bb803951SChris Mason if (level == 0) 780bb803951SChris Mason return 0; 781bb803951SChris Mason 7825f39d397SChris Mason mid = path->nodes[level]; 783925baeddSChris Mason WARN_ON(!path->locks[level]); 7847bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 7857bb86316SChris Mason 7861d4f8a0cSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 78779f95c82SChris Mason 788234b63a0SChris Mason if (level < BTRFS_MAX_LEVEL - 1) 7895f39d397SChris Mason parent = path->nodes[level + 1]; 790bb803951SChris Mason pslot = path->slots[level + 1]; 791bb803951SChris Mason 79240689478SChris Mason /* 79340689478SChris Mason * deal with the case where there is only one pointer in the root 79440689478SChris Mason * by promoting the node below to a root 79540689478SChris Mason */ 7965f39d397SChris Mason if (!parent) { 7975f39d397SChris Mason struct extent_buffer *child; 798bb803951SChris Mason 7995f39d397SChris Mason if (btrfs_header_nritems(mid) != 1) 800bb803951SChris Mason return 0; 801bb803951SChris Mason 802bb803951SChris Mason /* promote the child to a root */ 8035f39d397SChris Mason child = read_node_slot(root, mid, 0); 804925baeddSChris Mason btrfs_tree_lock(child); 805bb803951SChris Mason BUG_ON(!child); 8062f375ab9SYan ret = btrfs_cow_block(trans, root, child, mid, 0, &child); 8072f375ab9SYan BUG_ON(ret); 8082f375ab9SYan 809925baeddSChris Mason spin_lock(&root->node_lock); 810bb803951SChris Mason root->node = child; 811925baeddSChris Mason spin_unlock(&root->node_lock); 812925baeddSChris Mason 8130b86a832SChris Mason add_root_to_dirty_list(root); 814925baeddSChris Mason btrfs_tree_unlock(child); 815925baeddSChris Mason path->locks[level] = 0; 816bb803951SChris Mason path->nodes[level] = NULL; 8175f39d397SChris Mason clean_tree_block(trans, root, mid); 818925baeddSChris Mason btrfs_tree_unlock(mid); 819bb803951SChris Mason /* once for the path */ 8205f39d397SChris Mason free_extent_buffer(mid); 8217bb86316SChris Mason ret = btrfs_free_extent(trans, root, mid->start, mid->len, 8227bb86316SChris Mason root->root_key.objectid, 8237bb86316SChris Mason btrfs_header_generation(mid), 0, 0, 1); 824bb803951SChris Mason /* once for the root ptr */ 8255f39d397SChris Mason free_extent_buffer(mid); 826db94535dSChris Mason return ret; 827bb803951SChris Mason } 8285f39d397SChris Mason if (btrfs_header_nritems(mid) > 829123abc88SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) / 4) 830bb803951SChris Mason return 0; 831bb803951SChris Mason 8325f39d397SChris Mason if (btrfs_header_nritems(mid) < 2) 83354aa1f4dSChris Mason err_on_enospc = 1; 83454aa1f4dSChris Mason 8355f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 8365f39d397SChris Mason if (left) { 837925baeddSChris Mason btrfs_tree_lock(left); 8385f39d397SChris Mason wret = btrfs_cow_block(trans, root, left, 8395f39d397SChris Mason parent, pslot - 1, &left); 84054aa1f4dSChris Mason if (wret) { 84154aa1f4dSChris Mason ret = wret; 84254aa1f4dSChris Mason goto enospc; 84354aa1f4dSChris Mason } 8442cc58cf2SChris Mason } 8455f39d397SChris Mason right = read_node_slot(root, parent, pslot + 1); 8465f39d397SChris Mason if (right) { 847925baeddSChris Mason btrfs_tree_lock(right); 8485f39d397SChris Mason wret = btrfs_cow_block(trans, root, right, 8495f39d397SChris Mason parent, pslot + 1, &right); 8502cc58cf2SChris Mason if (wret) { 8512cc58cf2SChris Mason ret = wret; 8522cc58cf2SChris Mason goto enospc; 8532cc58cf2SChris Mason } 8542cc58cf2SChris Mason } 8552cc58cf2SChris Mason 8562cc58cf2SChris Mason /* first, try to make some room in the middle buffer */ 8575f39d397SChris Mason if (left) { 8585f39d397SChris Mason orig_slot += btrfs_header_nritems(left); 859bce4eae9SChris Mason wret = push_node_left(trans, root, left, mid, 1); 86079f95c82SChris Mason if (wret < 0) 86179f95c82SChris Mason ret = wret; 8625f39d397SChris Mason if (btrfs_header_nritems(mid) < 2) 86354aa1f4dSChris Mason err_on_enospc = 1; 864bb803951SChris Mason } 86579f95c82SChris Mason 86679f95c82SChris Mason /* 86779f95c82SChris Mason * then try to empty the right most buffer into the middle 86879f95c82SChris Mason */ 8695f39d397SChris Mason if (right) { 870971a1f66SChris Mason wret = push_node_left(trans, root, mid, right, 1); 87154aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 87279f95c82SChris Mason ret = wret; 8735f39d397SChris Mason if (btrfs_header_nritems(right) == 0) { 874db94535dSChris Mason u64 bytenr = right->start; 8757bb86316SChris Mason u64 generation = btrfs_header_generation(parent); 876db94535dSChris Mason u32 blocksize = right->len; 877db94535dSChris Mason 8785f39d397SChris Mason clean_tree_block(trans, root, right); 879925baeddSChris Mason btrfs_tree_unlock(right); 8805f39d397SChris Mason free_extent_buffer(right); 881bb803951SChris Mason right = NULL; 882e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot + 883e089f05cSChris Mason 1); 884bb803951SChris Mason if (wret) 885bb803951SChris Mason ret = wret; 886db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 8877bb86316SChris Mason blocksize, 8887bb86316SChris Mason btrfs_header_owner(parent), 8897bb86316SChris Mason generation, 0, 0, 1); 890bb803951SChris Mason if (wret) 891bb803951SChris Mason ret = wret; 892bb803951SChris Mason } else { 8935f39d397SChris Mason struct btrfs_disk_key right_key; 8945f39d397SChris Mason btrfs_node_key(right, &right_key, 0); 8955f39d397SChris Mason btrfs_set_node_key(parent, &right_key, pslot + 1); 8965f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 897bb803951SChris Mason } 898bb803951SChris Mason } 8995f39d397SChris Mason if (btrfs_header_nritems(mid) == 1) { 90079f95c82SChris Mason /* 90179f95c82SChris Mason * we're not allowed to leave a node with one item in the 90279f95c82SChris Mason * tree during a delete. A deletion from lower in the tree 90379f95c82SChris Mason * could try to delete the only pointer in this node. 90479f95c82SChris Mason * So, pull some keys from the left. 90579f95c82SChris Mason * There has to be a left pointer at this point because 90679f95c82SChris Mason * otherwise we would have pulled some pointers from the 90779f95c82SChris Mason * right 90879f95c82SChris Mason */ 9095f39d397SChris Mason BUG_ON(!left); 9105f39d397SChris Mason wret = balance_node_right(trans, root, mid, left); 91154aa1f4dSChris Mason if (wret < 0) { 91279f95c82SChris Mason ret = wret; 91354aa1f4dSChris Mason goto enospc; 91454aa1f4dSChris Mason } 915bce4eae9SChris Mason if (wret == 1) { 916bce4eae9SChris Mason wret = push_node_left(trans, root, left, mid, 1); 917bce4eae9SChris Mason if (wret < 0) 918bce4eae9SChris Mason ret = wret; 919bce4eae9SChris Mason } 92079f95c82SChris Mason BUG_ON(wret == 1); 92179f95c82SChris Mason } 9225f39d397SChris Mason if (btrfs_header_nritems(mid) == 0) { 92379f95c82SChris Mason /* we've managed to empty the middle node, drop it */ 9247bb86316SChris Mason u64 root_gen = btrfs_header_generation(parent); 925db94535dSChris Mason u64 bytenr = mid->start; 926db94535dSChris Mason u32 blocksize = mid->len; 927925baeddSChris Mason 9285f39d397SChris Mason clean_tree_block(trans, root, mid); 929925baeddSChris Mason btrfs_tree_unlock(mid); 9305f39d397SChris Mason free_extent_buffer(mid); 931bb803951SChris Mason mid = NULL; 932e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot); 933bb803951SChris Mason if (wret) 934bb803951SChris Mason ret = wret; 9357bb86316SChris Mason wret = btrfs_free_extent(trans, root, bytenr, blocksize, 9367bb86316SChris Mason btrfs_header_owner(parent), 9377bb86316SChris Mason root_gen, 0, 0, 1); 938bb803951SChris Mason if (wret) 939bb803951SChris Mason ret = wret; 94079f95c82SChris Mason } else { 94179f95c82SChris Mason /* update the parent key to reflect our changes */ 9425f39d397SChris Mason struct btrfs_disk_key mid_key; 9435f39d397SChris Mason btrfs_node_key(mid, &mid_key, 0); 9445f39d397SChris Mason btrfs_set_node_key(parent, &mid_key, pslot); 9455f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 94679f95c82SChris Mason } 947bb803951SChris Mason 94879f95c82SChris Mason /* update the path */ 9495f39d397SChris Mason if (left) { 9505f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 9515f39d397SChris Mason extent_buffer_get(left); 952925baeddSChris Mason /* left was locked after cow */ 9535f39d397SChris Mason path->nodes[level] = left; 954bb803951SChris Mason path->slots[level + 1] -= 1; 955bb803951SChris Mason path->slots[level] = orig_slot; 956925baeddSChris Mason if (mid) { 957925baeddSChris Mason btrfs_tree_unlock(mid); 9585f39d397SChris Mason free_extent_buffer(mid); 959925baeddSChris Mason } 960bb803951SChris Mason } else { 9615f39d397SChris Mason orig_slot -= btrfs_header_nritems(left); 962bb803951SChris Mason path->slots[level] = orig_slot; 963bb803951SChris Mason } 964bb803951SChris Mason } 96579f95c82SChris Mason /* double check we haven't messed things up */ 966123abc88SChris Mason check_block(root, path, level); 967e20d96d6SChris Mason if (orig_ptr != 9685f39d397SChris Mason btrfs_node_blockptr(path->nodes[level], path->slots[level])) 96979f95c82SChris Mason BUG(); 97054aa1f4dSChris Mason enospc: 971925baeddSChris Mason if (right) { 972925baeddSChris Mason btrfs_tree_unlock(right); 9735f39d397SChris Mason free_extent_buffer(right); 974925baeddSChris Mason } 975925baeddSChris Mason if (left) { 976925baeddSChris Mason if (path->nodes[level] != left) 977925baeddSChris Mason btrfs_tree_unlock(left); 9785f39d397SChris Mason free_extent_buffer(left); 979925baeddSChris Mason } 980bb803951SChris Mason return ret; 981bb803951SChris Mason } 982bb803951SChris Mason 983e66f709bSChris Mason /* returns zero if the push worked, non-zero otherwise */ 98498ed5174SChris Mason static int noinline push_nodes_for_insert(struct btrfs_trans_handle *trans, 985e66f709bSChris Mason struct btrfs_root *root, 986e66f709bSChris Mason struct btrfs_path *path, int level) 987e66f709bSChris Mason { 9885f39d397SChris Mason struct extent_buffer *right = NULL; 9895f39d397SChris Mason struct extent_buffer *mid; 9905f39d397SChris Mason struct extent_buffer *left = NULL; 9915f39d397SChris Mason struct extent_buffer *parent = NULL; 992e66f709bSChris Mason int ret = 0; 993e66f709bSChris Mason int wret; 994e66f709bSChris Mason int pslot; 995e66f709bSChris Mason int orig_slot = path->slots[level]; 996e66f709bSChris Mason u64 orig_ptr; 997e66f709bSChris Mason 998e66f709bSChris Mason if (level == 0) 999e66f709bSChris Mason return 1; 1000e66f709bSChris Mason 10015f39d397SChris Mason mid = path->nodes[level]; 10027bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 1003e66f709bSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 1004e66f709bSChris Mason 1005e66f709bSChris Mason if (level < BTRFS_MAX_LEVEL - 1) 10065f39d397SChris Mason parent = path->nodes[level + 1]; 1007e66f709bSChris Mason pslot = path->slots[level + 1]; 1008e66f709bSChris Mason 10095f39d397SChris Mason if (!parent) 1010e66f709bSChris Mason return 1; 1011e66f709bSChris Mason 10125f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 1013e66f709bSChris Mason 1014e66f709bSChris Mason /* first, try to make some room in the middle buffer */ 10155f39d397SChris Mason if (left) { 1016e66f709bSChris Mason u32 left_nr; 1017925baeddSChris Mason 1018925baeddSChris Mason btrfs_tree_lock(left); 10195f39d397SChris Mason left_nr = btrfs_header_nritems(left); 102033ade1f8SChris Mason if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 102133ade1f8SChris Mason wret = 1; 102233ade1f8SChris Mason } else { 10235f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, parent, 10245f39d397SChris Mason pslot - 1, &left); 102554aa1f4dSChris Mason if (ret) 102654aa1f4dSChris Mason wret = 1; 102754aa1f4dSChris Mason else { 102854aa1f4dSChris Mason wret = push_node_left(trans, root, 1029971a1f66SChris Mason left, mid, 0); 103054aa1f4dSChris Mason } 103133ade1f8SChris Mason } 1032e66f709bSChris Mason if (wret < 0) 1033e66f709bSChris Mason ret = wret; 1034e66f709bSChris Mason if (wret == 0) { 10355f39d397SChris Mason struct btrfs_disk_key disk_key; 1036e66f709bSChris Mason orig_slot += left_nr; 10375f39d397SChris Mason btrfs_node_key(mid, &disk_key, 0); 10385f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot); 10395f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 10405f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 10415f39d397SChris Mason path->nodes[level] = left; 1042e66f709bSChris Mason path->slots[level + 1] -= 1; 1043e66f709bSChris Mason path->slots[level] = orig_slot; 1044925baeddSChris Mason btrfs_tree_unlock(mid); 10455f39d397SChris Mason free_extent_buffer(mid); 1046e66f709bSChris Mason } else { 1047e66f709bSChris Mason orig_slot -= 10485f39d397SChris Mason btrfs_header_nritems(left); 1049e66f709bSChris Mason path->slots[level] = orig_slot; 1050925baeddSChris Mason btrfs_tree_unlock(left); 10515f39d397SChris Mason free_extent_buffer(left); 1052e66f709bSChris Mason } 1053e66f709bSChris Mason return 0; 1054e66f709bSChris Mason } 1055925baeddSChris Mason btrfs_tree_unlock(left); 10565f39d397SChris Mason free_extent_buffer(left); 1057e66f709bSChris Mason } 10585f39d397SChris Mason right = read_node_slot(root, parent, pslot + 1); 1059e66f709bSChris Mason 1060e66f709bSChris Mason /* 1061e66f709bSChris Mason * then try to empty the right most buffer into the middle 1062e66f709bSChris Mason */ 10635f39d397SChris Mason if (right) { 106433ade1f8SChris Mason u32 right_nr; 1065925baeddSChris Mason btrfs_tree_lock(right); 10665f39d397SChris Mason right_nr = btrfs_header_nritems(right); 106733ade1f8SChris Mason if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 106833ade1f8SChris Mason wret = 1; 106933ade1f8SChris Mason } else { 10705f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, 10715f39d397SChris Mason parent, pslot + 1, 10725f39d397SChris Mason &right); 107354aa1f4dSChris Mason if (ret) 107454aa1f4dSChris Mason wret = 1; 107554aa1f4dSChris Mason else { 107633ade1f8SChris Mason wret = balance_node_right(trans, root, 10775f39d397SChris Mason right, mid); 107833ade1f8SChris Mason } 107954aa1f4dSChris Mason } 1080e66f709bSChris Mason if (wret < 0) 1081e66f709bSChris Mason ret = wret; 1082e66f709bSChris Mason if (wret == 0) { 10835f39d397SChris Mason struct btrfs_disk_key disk_key; 10845f39d397SChris Mason 10855f39d397SChris Mason btrfs_node_key(right, &disk_key, 0); 10865f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot + 1); 10875f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 10885f39d397SChris Mason 10895f39d397SChris Mason if (btrfs_header_nritems(mid) <= orig_slot) { 10905f39d397SChris Mason path->nodes[level] = right; 1091e66f709bSChris Mason path->slots[level + 1] += 1; 1092e66f709bSChris Mason path->slots[level] = orig_slot - 10935f39d397SChris Mason btrfs_header_nritems(mid); 1094925baeddSChris Mason btrfs_tree_unlock(mid); 10955f39d397SChris Mason free_extent_buffer(mid); 1096e66f709bSChris Mason } else { 1097925baeddSChris Mason btrfs_tree_unlock(right); 10985f39d397SChris Mason free_extent_buffer(right); 1099e66f709bSChris Mason } 1100e66f709bSChris Mason return 0; 1101e66f709bSChris Mason } 1102925baeddSChris Mason btrfs_tree_unlock(right); 11035f39d397SChris Mason free_extent_buffer(right); 1104e66f709bSChris Mason } 1105e66f709bSChris Mason return 1; 1106e66f709bSChris Mason } 1107e66f709bSChris Mason 110874123bd7SChris Mason /* 11093c69faecSChris Mason * readahead one full node of leaves 11103c69faecSChris Mason */ 11113c69faecSChris Mason static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path, 111201f46658SChris Mason int level, int slot, u64 objectid) 11133c69faecSChris Mason { 11145f39d397SChris Mason struct extent_buffer *node; 111501f46658SChris Mason struct btrfs_disk_key disk_key; 11163c69faecSChris Mason u32 nritems; 11173c69faecSChris Mason u64 search; 11186b80053dSChris Mason u64 lowest_read; 11196b80053dSChris Mason u64 highest_read; 11206b80053dSChris Mason u64 nread = 0; 11213c69faecSChris Mason int direction = path->reada; 11225f39d397SChris Mason struct extent_buffer *eb; 11236b80053dSChris Mason u32 nr; 11246b80053dSChris Mason u32 blocksize; 11256b80053dSChris Mason u32 nscan = 0; 1126db94535dSChris Mason 1127a6b6e75eSChris Mason if (level != 1) 11283c69faecSChris Mason return; 11293c69faecSChris Mason 11306702ed49SChris Mason if (!path->nodes[level]) 11316702ed49SChris Mason return; 11326702ed49SChris Mason 11335f39d397SChris Mason node = path->nodes[level]; 1134925baeddSChris Mason 11353c69faecSChris Mason search = btrfs_node_blockptr(node, slot); 11366b80053dSChris Mason blocksize = btrfs_level_size(root, level - 1); 11376b80053dSChris Mason eb = btrfs_find_tree_block(root, search, blocksize); 11385f39d397SChris Mason if (eb) { 11395f39d397SChris Mason free_extent_buffer(eb); 11403c69faecSChris Mason return; 11413c69faecSChris Mason } 11423c69faecSChris Mason 11436b80053dSChris Mason highest_read = search; 11446b80053dSChris Mason lowest_read = search; 11456b80053dSChris Mason 11465f39d397SChris Mason nritems = btrfs_header_nritems(node); 11476b80053dSChris Mason nr = slot; 11483c69faecSChris Mason while(1) { 11496b80053dSChris Mason if (direction < 0) { 11506b80053dSChris Mason if (nr == 0) 11513c69faecSChris Mason break; 11526b80053dSChris Mason nr--; 11536b80053dSChris Mason } else if (direction > 0) { 11546b80053dSChris Mason nr++; 11556b80053dSChris Mason if (nr >= nritems) 11566b80053dSChris Mason break; 11573c69faecSChris Mason } 115801f46658SChris Mason if (path->reada < 0 && objectid) { 115901f46658SChris Mason btrfs_node_key(node, &disk_key, nr); 116001f46658SChris Mason if (btrfs_disk_key_objectid(&disk_key) != objectid) 116101f46658SChris Mason break; 116201f46658SChris Mason } 11636b80053dSChris Mason search = btrfs_node_blockptr(node, nr); 11646b80053dSChris Mason if ((search >= lowest_read && search <= highest_read) || 11656b80053dSChris Mason (search < lowest_read && lowest_read - search <= 32768) || 11666b80053dSChris Mason (search > highest_read && search - highest_read <= 32768)) { 1167ca7a79adSChris Mason readahead_tree_block(root, search, blocksize, 1168ca7a79adSChris Mason btrfs_node_ptr_generation(node, nr)); 11696b80053dSChris Mason nread += blocksize; 11703c69faecSChris Mason } 11716b80053dSChris Mason nscan++; 11726b80053dSChris Mason if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32)) 11736b80053dSChris Mason break; 11746b80053dSChris Mason if(nread > (1024 * 1024) || nscan > 128) 11756b80053dSChris Mason break; 11766b80053dSChris Mason 11776b80053dSChris Mason if (search < lowest_read) 11786b80053dSChris Mason lowest_read = search; 11796b80053dSChris Mason if (search > highest_read) 11806b80053dSChris Mason highest_read = search; 11813c69faecSChris Mason } 11823c69faecSChris Mason } 1183925baeddSChris Mason 1184925baeddSChris Mason static void unlock_up(struct btrfs_path *path, int level, int lowest_unlock) 1185925baeddSChris Mason { 1186925baeddSChris Mason int i; 1187925baeddSChris Mason int skip_level = level; 1188051e1b9fSChris Mason int no_skips = 0; 1189925baeddSChris Mason struct extent_buffer *t; 1190925baeddSChris Mason 1191925baeddSChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 1192925baeddSChris Mason if (!path->nodes[i]) 1193925baeddSChris Mason break; 1194925baeddSChris Mason if (!path->locks[i]) 1195925baeddSChris Mason break; 1196051e1b9fSChris Mason if (!no_skips && path->slots[i] == 0) { 1197925baeddSChris Mason skip_level = i + 1; 1198925baeddSChris Mason continue; 1199925baeddSChris Mason } 1200051e1b9fSChris Mason if (!no_skips && path->keep_locks) { 1201925baeddSChris Mason u32 nritems; 1202925baeddSChris Mason t = path->nodes[i]; 1203925baeddSChris Mason nritems = btrfs_header_nritems(t); 1204051e1b9fSChris Mason if (nritems < 1 || path->slots[i] >= nritems - 1) { 1205925baeddSChris Mason skip_level = i + 1; 1206925baeddSChris Mason continue; 1207925baeddSChris Mason } 1208925baeddSChris Mason } 1209051e1b9fSChris Mason if (skip_level < i && i >= lowest_unlock) 1210051e1b9fSChris Mason no_skips = 1; 1211051e1b9fSChris Mason 1212925baeddSChris Mason t = path->nodes[i]; 1213925baeddSChris Mason if (i >= lowest_unlock && i > skip_level && path->locks[i]) { 1214925baeddSChris Mason btrfs_tree_unlock(t); 1215925baeddSChris Mason path->locks[i] = 0; 1216925baeddSChris Mason } 1217925baeddSChris Mason } 1218925baeddSChris Mason } 1219925baeddSChris Mason 12203c69faecSChris Mason /* 122174123bd7SChris Mason * look for key in the tree. path is filled in with nodes along the way 122274123bd7SChris Mason * if key is found, we return zero and you can find the item in the leaf 122374123bd7SChris Mason * level of the path (level 0) 122474123bd7SChris Mason * 122574123bd7SChris Mason * If the key isn't found, the path points to the slot where it should 1226aa5d6bedSChris Mason * be inserted, and 1 is returned. If there are other errors during the 1227aa5d6bedSChris Mason * search a negative error number is returned. 122897571fd0SChris Mason * 122997571fd0SChris Mason * if ins_len > 0, nodes and leaves will be split as we walk down the 123097571fd0SChris Mason * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if 123197571fd0SChris Mason * possible) 123274123bd7SChris Mason */ 1233e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root 1234e089f05cSChris Mason *root, struct btrfs_key *key, struct btrfs_path *p, int 1235e089f05cSChris Mason ins_len, int cow) 1236be0e5c09SChris Mason { 12375f39d397SChris Mason struct extent_buffer *b; 1238051e1b9fSChris Mason struct extent_buffer *tmp; 1239be0e5c09SChris Mason int slot; 1240be0e5c09SChris Mason int ret; 1241be0e5c09SChris Mason int level; 12423c69faecSChris Mason int should_reada = p->reada; 1243925baeddSChris Mason int lowest_unlock = 1; 1244594a24ebSChris Mason int blocksize; 12459f3a7427SChris Mason u8 lowest_level = 0; 1246594a24ebSChris Mason u64 blocknr; 1247594a24ebSChris Mason u64 gen; 12489f3a7427SChris Mason 12496702ed49SChris Mason lowest_level = p->lowest_level; 12506702ed49SChris Mason WARN_ON(lowest_level && ins_len); 125122b0ebdaSChris Mason WARN_ON(p->nodes[0] != NULL); 1252333db94cSChris Mason WARN_ON(cow && root == root->fs_info->extent_root && 1253925baeddSChris Mason !mutex_is_locked(&root->fs_info->alloc_mutex)); 1254925baeddSChris Mason WARN_ON(root == root->fs_info->chunk_root && 1255925baeddSChris Mason !mutex_is_locked(&root->fs_info->chunk_mutex)); 1256925baeddSChris Mason WARN_ON(root == root->fs_info->dev_root && 1257925baeddSChris Mason !mutex_is_locked(&root->fs_info->chunk_mutex)); 1258925baeddSChris Mason if (ins_len < 0) 1259925baeddSChris Mason lowest_unlock = 2; 1260bb803951SChris Mason again: 12615cd57b2cSChris Mason if (p->skip_locking) 12625cd57b2cSChris Mason b = btrfs_root_node(root); 12635cd57b2cSChris Mason else 1264925baeddSChris Mason b = btrfs_lock_root_node(root); 1265925baeddSChris Mason 1266eb60ceacSChris Mason while (b) { 12675f39d397SChris Mason level = btrfs_header_level(b); 126802217ed2SChris Mason if (cow) { 126902217ed2SChris Mason int wret; 1270e20d96d6SChris Mason wret = btrfs_cow_block(trans, root, b, 1271e20d96d6SChris Mason p->nodes[level + 1], 1272e20d96d6SChris Mason p->slots[level + 1], 1273252c38f0SYan &b); 127454aa1f4dSChris Mason if (wret) { 12755f39d397SChris Mason free_extent_buffer(b); 127654aa1f4dSChris Mason return wret; 127754aa1f4dSChris Mason } 127802217ed2SChris Mason } 127902217ed2SChris Mason BUG_ON(!cow && ins_len); 12805f39d397SChris Mason if (level != btrfs_header_level(b)) 12812c90e5d6SChris Mason WARN_ON(1); 12825f39d397SChris Mason level = btrfs_header_level(b); 1283eb60ceacSChris Mason p->nodes[level] = b; 12845cd57b2cSChris Mason if (!p->skip_locking) 1285925baeddSChris Mason p->locks[level] = 1; 1286123abc88SChris Mason ret = check_block(root, p, level); 1287aa5d6bedSChris Mason if (ret) 1288aa5d6bedSChris Mason return -1; 1289925baeddSChris Mason 12905f39d397SChris Mason ret = bin_search(b, key, level, &slot); 12915f39d397SChris Mason if (level != 0) { 1292be0e5c09SChris Mason if (ret && slot > 0) 1293be0e5c09SChris Mason slot -= 1; 1294be0e5c09SChris Mason p->slots[level] = slot; 12955f39d397SChris Mason if (ins_len > 0 && btrfs_header_nritems(b) >= 12961514794eSChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) { 1297e089f05cSChris Mason int sret = split_node(trans, root, p, level); 12985c680ed6SChris Mason BUG_ON(sret > 0); 12995c680ed6SChris Mason if (sret) 13005c680ed6SChris Mason return sret; 13015c680ed6SChris Mason b = p->nodes[level]; 13025c680ed6SChris Mason slot = p->slots[level]; 1303bb803951SChris Mason } else if (ins_len < 0) { 1304e089f05cSChris Mason int sret = balance_level(trans, root, p, 1305e089f05cSChris Mason level); 1306bb803951SChris Mason if (sret) 1307bb803951SChris Mason return sret; 1308bb803951SChris Mason b = p->nodes[level]; 1309f510cfecSChris Mason if (!b) { 1310f510cfecSChris Mason btrfs_release_path(NULL, p); 1311bb803951SChris Mason goto again; 1312f510cfecSChris Mason } 1313bb803951SChris Mason slot = p->slots[level]; 13145f39d397SChris Mason BUG_ON(btrfs_header_nritems(b) == 1); 13155c680ed6SChris Mason } 13169f3a7427SChris Mason /* this is only true while dropping a snapshot */ 1317925baeddSChris Mason if (level == lowest_level) { 1318925baeddSChris Mason unlock_up(p, level, lowest_unlock); 13199f3a7427SChris Mason break; 1320925baeddSChris Mason } 1321ca7a79adSChris Mason 13226702ed49SChris Mason if (should_reada) 132301f46658SChris Mason reada_for_search(root, p, level, slot, 132401f46658SChris Mason key->objectid); 1325ca7a79adSChris Mason 1326594a24ebSChris Mason blocknr = btrfs_node_blockptr(b, slot); 1327594a24ebSChris Mason gen = btrfs_node_ptr_generation(b, slot); 1328594a24ebSChris Mason blocksize = btrfs_level_size(root, level - 1); 1329594a24ebSChris Mason 1330594a24ebSChris Mason tmp = btrfs_find_tree_block(root, blocknr, blocksize); 1331594a24ebSChris Mason if (tmp && btrfs_buffer_uptodate(tmp, gen)) { 1332051e1b9fSChris Mason b = tmp; 1333051e1b9fSChris Mason } else { 1334051e1b9fSChris Mason /* 1335051e1b9fSChris Mason * reduce lock contention at high levels 1336051e1b9fSChris Mason * of the btree by dropping locks before 1337051e1b9fSChris Mason * we read. 1338051e1b9fSChris Mason */ 1339051e1b9fSChris Mason if (level > 1) { 1340051e1b9fSChris Mason btrfs_release_path(NULL, p); 1341051e1b9fSChris Mason if (tmp) 1342051e1b9fSChris Mason free_extent_buffer(tmp); 1343594a24ebSChris Mason tmp = read_tree_block(root, blocknr, 1344594a24ebSChris Mason blocksize, gen); 1345594a24ebSChris Mason if (tmp) 1346594a24ebSChris Mason free_extent_buffer(tmp); 1347051e1b9fSChris Mason goto again; 1348051e1b9fSChris Mason } else { 1349a74a4b97SChris Mason if (tmp) 1350a74a4b97SChris Mason free_extent_buffer(tmp); 1351ca7a79adSChris Mason b = read_node_slot(root, b, slot); 1352051e1b9fSChris Mason } 1353051e1b9fSChris Mason } 13545cd57b2cSChris Mason if (!p->skip_locking) 1355925baeddSChris Mason btrfs_tree_lock(b); 1356051e1b9fSChris Mason unlock_up(p, level, lowest_unlock); 1357be0e5c09SChris Mason } else { 1358be0e5c09SChris Mason p->slots[level] = slot; 13595f39d397SChris Mason if (ins_len > 0 && btrfs_leaf_free_space(root, b) < 13600783fcfcSChris Mason sizeof(struct btrfs_item) + ins_len) { 1361d4dbff95SChris Mason int sret = split_leaf(trans, root, key, 1362cc0c5538SChris Mason p, ins_len, ret == 0); 13635c680ed6SChris Mason BUG_ON(sret > 0); 13645c680ed6SChris Mason if (sret) 13655c680ed6SChris Mason return sret; 13665c680ed6SChris Mason } 1367925baeddSChris Mason unlock_up(p, level, lowest_unlock); 1368be0e5c09SChris Mason return ret; 1369be0e5c09SChris Mason } 1370be0e5c09SChris Mason } 1371aa5d6bedSChris Mason return 1; 1372be0e5c09SChris Mason } 1373be0e5c09SChris Mason 137474123bd7SChris Mason /* 137574123bd7SChris Mason * adjust the pointers going up the tree, starting at level 137674123bd7SChris Mason * making sure the right key of each node is points to 'key'. 137774123bd7SChris Mason * This is used after shifting pointers to the left, so it stops 137874123bd7SChris Mason * fixing up pointers when a given leaf/node is not in slot 0 of the 137974123bd7SChris Mason * higher levels 1380aa5d6bedSChris Mason * 1381aa5d6bedSChris Mason * If this fails to write a tree block, it returns -1, but continues 1382aa5d6bedSChris Mason * fixing up the blocks in ram so the tree is consistent. 138374123bd7SChris Mason */ 13845f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans, 13855f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 13865f39d397SChris Mason struct btrfs_disk_key *key, int level) 1387be0e5c09SChris Mason { 1388be0e5c09SChris Mason int i; 1389aa5d6bedSChris Mason int ret = 0; 13905f39d397SChris Mason struct extent_buffer *t; 13915f39d397SChris Mason 1392234b63a0SChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 1393be0e5c09SChris Mason int tslot = path->slots[i]; 1394eb60ceacSChris Mason if (!path->nodes[i]) 1395be0e5c09SChris Mason break; 13965f39d397SChris Mason t = path->nodes[i]; 13975f39d397SChris Mason btrfs_set_node_key(t, key, tslot); 1398d6025579SChris Mason btrfs_mark_buffer_dirty(path->nodes[i]); 1399be0e5c09SChris Mason if (tslot != 0) 1400be0e5c09SChris Mason break; 1401be0e5c09SChris Mason } 1402aa5d6bedSChris Mason return ret; 1403be0e5c09SChris Mason } 1404be0e5c09SChris Mason 140574123bd7SChris Mason /* 140674123bd7SChris Mason * try to push data from one node into the next node left in the 140779f95c82SChris Mason * tree. 1408aa5d6bedSChris Mason * 1409aa5d6bedSChris Mason * returns 0 if some ptrs were pushed left, < 0 if there was some horrible 1410aa5d6bedSChris Mason * error, and > 0 if there was no room in the left hand block. 141174123bd7SChris Mason */ 141298ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans, 141398ed5174SChris Mason struct btrfs_root *root, struct extent_buffer *dst, 1414971a1f66SChris Mason struct extent_buffer *src, int empty) 1415be0e5c09SChris Mason { 1416be0e5c09SChris Mason int push_items = 0; 1417bb803951SChris Mason int src_nritems; 1418bb803951SChris Mason int dst_nritems; 1419aa5d6bedSChris Mason int ret = 0; 1420be0e5c09SChris Mason 14215f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 14225f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1423123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 14247bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 14257bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 142654aa1f4dSChris Mason 1427bce4eae9SChris Mason if (!empty && src_nritems <= 8) 1428971a1f66SChris Mason return 1; 1429971a1f66SChris Mason 1430eb60ceacSChris Mason if (push_items <= 0) { 1431be0e5c09SChris Mason return 1; 1432eb60ceacSChris Mason } 1433be0e5c09SChris Mason 1434bce4eae9SChris Mason if (empty) { 1435971a1f66SChris Mason push_items = min(src_nritems, push_items); 1436bce4eae9SChris Mason if (push_items < src_nritems) { 1437bce4eae9SChris Mason /* leave at least 8 pointers in the node if 1438bce4eae9SChris Mason * we aren't going to empty it 1439bce4eae9SChris Mason */ 1440bce4eae9SChris Mason if (src_nritems - push_items < 8) { 1441bce4eae9SChris Mason if (push_items <= 8) 1442bce4eae9SChris Mason return 1; 1443bce4eae9SChris Mason push_items -= 8; 1444bce4eae9SChris Mason } 1445bce4eae9SChris Mason } 1446bce4eae9SChris Mason } else 1447bce4eae9SChris Mason push_items = min(src_nritems - 8, push_items); 144879f95c82SChris Mason 14495f39d397SChris Mason copy_extent_buffer(dst, src, 14505f39d397SChris Mason btrfs_node_key_ptr_offset(dst_nritems), 14515f39d397SChris Mason btrfs_node_key_ptr_offset(0), 1452123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 14535f39d397SChris Mason 1454bb803951SChris Mason if (push_items < src_nritems) { 14555f39d397SChris Mason memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0), 14565f39d397SChris Mason btrfs_node_key_ptr_offset(push_items), 1457e2fa7227SChris Mason (src_nritems - push_items) * 1458123abc88SChris Mason sizeof(struct btrfs_key_ptr)); 1459bb803951SChris Mason } 14605f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 14615f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 14625f39d397SChris Mason btrfs_mark_buffer_dirty(src); 14635f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 1464bb803951SChris Mason return ret; 1465be0e5c09SChris Mason } 1466be0e5c09SChris Mason 146797571fd0SChris Mason /* 146879f95c82SChris Mason * try to push data from one node into the next node right in the 146979f95c82SChris Mason * tree. 147079f95c82SChris Mason * 147179f95c82SChris Mason * returns 0 if some ptrs were pushed, < 0 if there was some horrible 147279f95c82SChris Mason * error, and > 0 if there was no room in the right hand block. 147379f95c82SChris Mason * 147479f95c82SChris Mason * this will only push up to 1/2 the contents of the left node over 147579f95c82SChris Mason */ 14765f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 14775f39d397SChris Mason struct btrfs_root *root, 14785f39d397SChris Mason struct extent_buffer *dst, 14795f39d397SChris Mason struct extent_buffer *src) 148079f95c82SChris Mason { 148179f95c82SChris Mason int push_items = 0; 148279f95c82SChris Mason int max_push; 148379f95c82SChris Mason int src_nritems; 148479f95c82SChris Mason int dst_nritems; 148579f95c82SChris Mason int ret = 0; 148679f95c82SChris Mason 14877bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 14887bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 14897bb86316SChris Mason 14905f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 14915f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1492123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 1493bce4eae9SChris Mason if (push_items <= 0) { 149479f95c82SChris Mason return 1; 1495bce4eae9SChris Mason } 1496bce4eae9SChris Mason 1497bce4eae9SChris Mason if (src_nritems < 4) { 1498bce4eae9SChris Mason return 1; 1499bce4eae9SChris Mason } 150079f95c82SChris Mason 150179f95c82SChris Mason max_push = src_nritems / 2 + 1; 150279f95c82SChris Mason /* don't try to empty the node */ 1503bce4eae9SChris Mason if (max_push >= src_nritems) { 150479f95c82SChris Mason return 1; 1505bce4eae9SChris Mason } 1506252c38f0SYan 150779f95c82SChris Mason if (max_push < push_items) 150879f95c82SChris Mason push_items = max_push; 150979f95c82SChris Mason 15105f39d397SChris Mason memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items), 15115f39d397SChris Mason btrfs_node_key_ptr_offset(0), 15125f39d397SChris Mason (dst_nritems) * 15135f39d397SChris Mason sizeof(struct btrfs_key_ptr)); 1514d6025579SChris Mason 15155f39d397SChris Mason copy_extent_buffer(dst, src, 15165f39d397SChris Mason btrfs_node_key_ptr_offset(0), 15175f39d397SChris Mason btrfs_node_key_ptr_offset(src_nritems - push_items), 1518123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 151979f95c82SChris Mason 15205f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 15215f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 152279f95c82SChris Mason 15235f39d397SChris Mason btrfs_mark_buffer_dirty(src); 15245f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 152579f95c82SChris Mason return ret; 152679f95c82SChris Mason } 152779f95c82SChris Mason 152879f95c82SChris Mason /* 152997571fd0SChris Mason * helper function to insert a new root level in the tree. 153097571fd0SChris Mason * A new node is allocated, and a single item is inserted to 153197571fd0SChris Mason * point to the existing root 1532aa5d6bedSChris Mason * 1533aa5d6bedSChris Mason * returns zero on success or < 0 on failure. 153497571fd0SChris Mason */ 153598ed5174SChris Mason static int noinline insert_new_root(struct btrfs_trans_handle *trans, 15365f39d397SChris Mason struct btrfs_root *root, 15375f39d397SChris Mason struct btrfs_path *path, int level) 153874123bd7SChris Mason { 15397bb86316SChris Mason u64 root_gen; 15407bb86316SChris Mason u64 lower_gen; 15415f39d397SChris Mason struct extent_buffer *lower; 15425f39d397SChris Mason struct extent_buffer *c; 1543925baeddSChris Mason struct extent_buffer *old; 15445f39d397SChris Mason struct btrfs_disk_key lower_key; 15455c680ed6SChris Mason 15465c680ed6SChris Mason BUG_ON(path->nodes[level]); 15475c680ed6SChris Mason BUG_ON(path->nodes[level-1] != root->node); 15485c680ed6SChris Mason 15497bb86316SChris Mason if (root->ref_cows) 15507bb86316SChris Mason root_gen = trans->transid; 15517bb86316SChris Mason else 15527bb86316SChris Mason root_gen = 0; 15537bb86316SChris Mason 15547bb86316SChris Mason lower = path->nodes[level-1]; 15557bb86316SChris Mason if (level == 1) 15567bb86316SChris Mason btrfs_item_key(lower, &lower_key, 0); 15577bb86316SChris Mason else 15587bb86316SChris Mason btrfs_node_key(lower, &lower_key, 0); 15597bb86316SChris Mason 1560925baeddSChris Mason c = btrfs_alloc_free_block(trans, root, root->nodesize, 15617bb86316SChris Mason root->root_key.objectid, 15627bb86316SChris Mason root_gen, lower_key.objectid, level, 1563db94535dSChris Mason root->node->start, 0); 15645f39d397SChris Mason if (IS_ERR(c)) 15655f39d397SChris Mason return PTR_ERR(c); 1566925baeddSChris Mason 15675f39d397SChris Mason memset_extent_buffer(c, 0, 0, root->nodesize); 15685f39d397SChris Mason btrfs_set_header_nritems(c, 1); 15695f39d397SChris Mason btrfs_set_header_level(c, level); 1570db94535dSChris Mason btrfs_set_header_bytenr(c, c->start); 15715f39d397SChris Mason btrfs_set_header_generation(c, trans->transid); 15725f39d397SChris Mason btrfs_set_header_owner(c, root->root_key.objectid); 1573d5719762SChris Mason 15745f39d397SChris Mason write_extent_buffer(c, root->fs_info->fsid, 15755f39d397SChris Mason (unsigned long)btrfs_header_fsid(c), 15765f39d397SChris Mason BTRFS_FSID_SIZE); 1577e17cade2SChris Mason 1578e17cade2SChris Mason write_extent_buffer(c, root->fs_info->chunk_tree_uuid, 1579e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(c), 1580e17cade2SChris Mason BTRFS_UUID_SIZE); 1581e17cade2SChris Mason 15825f39d397SChris Mason btrfs_set_node_key(c, &lower_key, 0); 1583db94535dSChris Mason btrfs_set_node_blockptr(c, 0, lower->start); 15847bb86316SChris Mason lower_gen = btrfs_header_generation(lower); 15857bb86316SChris Mason WARN_ON(lower_gen == 0); 15867bb86316SChris Mason 15877bb86316SChris Mason btrfs_set_node_ptr_generation(c, 0, lower_gen); 15885f39d397SChris Mason 15895f39d397SChris Mason btrfs_mark_buffer_dirty(c); 1590d5719762SChris Mason 1591925baeddSChris Mason spin_lock(&root->node_lock); 1592925baeddSChris Mason old = root->node; 15935f39d397SChris Mason root->node = c; 1594925baeddSChris Mason spin_unlock(&root->node_lock); 1595925baeddSChris Mason 1596925baeddSChris Mason /* the super has an extra ref to root->node */ 1597925baeddSChris Mason free_extent_buffer(old); 1598925baeddSChris Mason 15990b86a832SChris Mason add_root_to_dirty_list(root); 16005f39d397SChris Mason extent_buffer_get(c); 16015f39d397SChris Mason path->nodes[level] = c; 1602925baeddSChris Mason path->locks[level] = 1; 160374123bd7SChris Mason path->slots[level] = 0; 16047bb86316SChris Mason 16057bb86316SChris Mason if (root->ref_cows && lower_gen != trans->transid) { 16067bb86316SChris Mason struct btrfs_path *back_path = btrfs_alloc_path(); 16077bb86316SChris Mason int ret; 1608925baeddSChris Mason mutex_lock(&root->fs_info->alloc_mutex); 16097bb86316SChris Mason ret = btrfs_insert_extent_backref(trans, 16107bb86316SChris Mason root->fs_info->extent_root, 16117bb86316SChris Mason path, lower->start, 16127bb86316SChris Mason root->root_key.objectid, 16137bb86316SChris Mason trans->transid, 0, 0); 16147bb86316SChris Mason BUG_ON(ret); 1615925baeddSChris Mason mutex_unlock(&root->fs_info->alloc_mutex); 16167bb86316SChris Mason btrfs_free_path(back_path); 16177bb86316SChris Mason } 161874123bd7SChris Mason return 0; 161974123bd7SChris Mason } 16205c680ed6SChris Mason 16215c680ed6SChris Mason /* 16225c680ed6SChris Mason * worker function to insert a single pointer in a node. 16235c680ed6SChris Mason * the node should have enough room for the pointer already 162497571fd0SChris Mason * 16255c680ed6SChris Mason * slot and level indicate where you want the key to go, and 16265c680ed6SChris Mason * blocknr is the block the key points to. 1627aa5d6bedSChris Mason * 1628aa5d6bedSChris Mason * returns zero on success and < 0 on any error 16295c680ed6SChris Mason */ 1630e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root 1631e089f05cSChris Mason *root, struct btrfs_path *path, struct btrfs_disk_key 1632db94535dSChris Mason *key, u64 bytenr, int slot, int level) 16335c680ed6SChris Mason { 16345f39d397SChris Mason struct extent_buffer *lower; 16355c680ed6SChris Mason int nritems; 16365c680ed6SChris Mason 16375c680ed6SChris Mason BUG_ON(!path->nodes[level]); 16385f39d397SChris Mason lower = path->nodes[level]; 16395f39d397SChris Mason nritems = btrfs_header_nritems(lower); 164074123bd7SChris Mason if (slot > nritems) 164174123bd7SChris Mason BUG(); 1642123abc88SChris Mason if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root)) 164374123bd7SChris Mason BUG(); 164474123bd7SChris Mason if (slot != nritems) { 16455f39d397SChris Mason memmove_extent_buffer(lower, 16465f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 16475f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 1648123abc88SChris Mason (nritems - slot) * sizeof(struct btrfs_key_ptr)); 164974123bd7SChris Mason } 16505f39d397SChris Mason btrfs_set_node_key(lower, key, slot); 1651db94535dSChris Mason btrfs_set_node_blockptr(lower, slot, bytenr); 165274493f7aSChris Mason WARN_ON(trans->transid == 0); 165374493f7aSChris Mason btrfs_set_node_ptr_generation(lower, slot, trans->transid); 16545f39d397SChris Mason btrfs_set_header_nritems(lower, nritems + 1); 16555f39d397SChris Mason btrfs_mark_buffer_dirty(lower); 165674123bd7SChris Mason return 0; 165774123bd7SChris Mason } 165874123bd7SChris Mason 165997571fd0SChris Mason /* 166097571fd0SChris Mason * split the node at the specified level in path in two. 166197571fd0SChris Mason * The path is corrected to point to the appropriate node after the split 166297571fd0SChris Mason * 166397571fd0SChris Mason * Before splitting this tries to make some room in the node by pushing 166497571fd0SChris Mason * left and right, if either one works, it returns right away. 1665aa5d6bedSChris Mason * 1666aa5d6bedSChris Mason * returns 0 on success and < 0 on failure 166797571fd0SChris Mason */ 1668e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 1669e089f05cSChris Mason *root, struct btrfs_path *path, int level) 1670be0e5c09SChris Mason { 16717bb86316SChris Mason u64 root_gen; 16725f39d397SChris Mason struct extent_buffer *c; 16735f39d397SChris Mason struct extent_buffer *split; 16745f39d397SChris Mason struct btrfs_disk_key disk_key; 1675be0e5c09SChris Mason int mid; 16765c680ed6SChris Mason int ret; 1677aa5d6bedSChris Mason int wret; 16787518a238SChris Mason u32 c_nritems; 1679be0e5c09SChris Mason 16805f39d397SChris Mason c = path->nodes[level]; 16817bb86316SChris Mason WARN_ON(btrfs_header_generation(c) != trans->transid); 16825f39d397SChris Mason if (c == root->node) { 16835c680ed6SChris Mason /* trying to split the root, lets make a new one */ 1684e089f05cSChris Mason ret = insert_new_root(trans, root, path, level + 1); 16855c680ed6SChris Mason if (ret) 16865c680ed6SChris Mason return ret; 1687e66f709bSChris Mason } else { 1688e66f709bSChris Mason ret = push_nodes_for_insert(trans, root, path, level); 16895f39d397SChris Mason c = path->nodes[level]; 16905f39d397SChris Mason if (!ret && btrfs_header_nritems(c) < 1691c448acf0SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) 1692e66f709bSChris Mason return 0; 169354aa1f4dSChris Mason if (ret < 0) 169454aa1f4dSChris Mason return ret; 16955c680ed6SChris Mason } 1696e66f709bSChris Mason 16975f39d397SChris Mason c_nritems = btrfs_header_nritems(c); 16987bb86316SChris Mason if (root->ref_cows) 16997bb86316SChris Mason root_gen = trans->transid; 17007bb86316SChris Mason else 17017bb86316SChris Mason root_gen = 0; 17027bb86316SChris Mason 17037bb86316SChris Mason btrfs_node_key(c, &disk_key, 0); 1704925baeddSChris Mason split = btrfs_alloc_free_block(trans, root, root->nodesize, 17057bb86316SChris Mason root->root_key.objectid, 17067bb86316SChris Mason root_gen, 17077bb86316SChris Mason btrfs_disk_key_objectid(&disk_key), 17087bb86316SChris Mason level, c->start, 0); 17095f39d397SChris Mason if (IS_ERR(split)) 17105f39d397SChris Mason return PTR_ERR(split); 171154aa1f4dSChris Mason 17125f39d397SChris Mason btrfs_set_header_flags(split, btrfs_header_flags(c)); 17135f39d397SChris Mason btrfs_set_header_level(split, btrfs_header_level(c)); 1714db94535dSChris Mason btrfs_set_header_bytenr(split, split->start); 17155f39d397SChris Mason btrfs_set_header_generation(split, trans->transid); 17165f39d397SChris Mason btrfs_set_header_owner(split, root->root_key.objectid); 171763b10fc4SChris Mason btrfs_set_header_flags(split, 0); 17185f39d397SChris Mason write_extent_buffer(split, root->fs_info->fsid, 17195f39d397SChris Mason (unsigned long)btrfs_header_fsid(split), 17205f39d397SChris Mason BTRFS_FSID_SIZE); 1721e17cade2SChris Mason write_extent_buffer(split, root->fs_info->chunk_tree_uuid, 1722e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(split), 1723e17cade2SChris Mason BTRFS_UUID_SIZE); 17245f39d397SChris Mason 17257518a238SChris Mason mid = (c_nritems + 1) / 2; 17265f39d397SChris Mason 17275f39d397SChris Mason copy_extent_buffer(split, c, 17285f39d397SChris Mason btrfs_node_key_ptr_offset(0), 17295f39d397SChris Mason btrfs_node_key_ptr_offset(mid), 1730123abc88SChris Mason (c_nritems - mid) * sizeof(struct btrfs_key_ptr)); 17315f39d397SChris Mason btrfs_set_header_nritems(split, c_nritems - mid); 17325f39d397SChris Mason btrfs_set_header_nritems(c, mid); 1733aa5d6bedSChris Mason ret = 0; 1734aa5d6bedSChris Mason 17355f39d397SChris Mason btrfs_mark_buffer_dirty(c); 17365f39d397SChris Mason btrfs_mark_buffer_dirty(split); 17375f39d397SChris Mason 17385f39d397SChris Mason btrfs_node_key(split, &disk_key, 0); 1739db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, split->start, 17405f39d397SChris Mason path->slots[level + 1] + 1, 1741123abc88SChris Mason level + 1); 1742aa5d6bedSChris Mason if (wret) 1743aa5d6bedSChris Mason ret = wret; 1744aa5d6bedSChris Mason 17455de08d7dSChris Mason if (path->slots[level] >= mid) { 17465c680ed6SChris Mason path->slots[level] -= mid; 1747925baeddSChris Mason btrfs_tree_unlock(c); 17485f39d397SChris Mason free_extent_buffer(c); 17495f39d397SChris Mason path->nodes[level] = split; 17505c680ed6SChris Mason path->slots[level + 1] += 1; 1751eb60ceacSChris Mason } else { 1752925baeddSChris Mason btrfs_tree_unlock(split); 17535f39d397SChris Mason free_extent_buffer(split); 1754be0e5c09SChris Mason } 1755aa5d6bedSChris Mason return ret; 1756be0e5c09SChris Mason } 1757be0e5c09SChris Mason 175874123bd7SChris Mason /* 175974123bd7SChris Mason * how many bytes are required to store the items in a leaf. start 176074123bd7SChris Mason * and nr indicate which items in the leaf to check. This totals up the 176174123bd7SChris Mason * space used both by the item structs and the item data 176274123bd7SChris Mason */ 17635f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr) 1764be0e5c09SChris Mason { 1765be0e5c09SChris Mason int data_len; 17665f39d397SChris Mason int nritems = btrfs_header_nritems(l); 1767d4dbff95SChris Mason int end = min(nritems, start + nr) - 1; 1768be0e5c09SChris Mason 1769be0e5c09SChris Mason if (!nr) 1770be0e5c09SChris Mason return 0; 17715f39d397SChris Mason data_len = btrfs_item_end_nr(l, start); 17725f39d397SChris Mason data_len = data_len - btrfs_item_offset_nr(l, end); 17730783fcfcSChris Mason data_len += sizeof(struct btrfs_item) * nr; 1774d4dbff95SChris Mason WARN_ON(data_len < 0); 1775be0e5c09SChris Mason return data_len; 1776be0e5c09SChris Mason } 1777be0e5c09SChris Mason 177874123bd7SChris Mason /* 1779d4dbff95SChris Mason * The space between the end of the leaf items and 1780d4dbff95SChris Mason * the start of the leaf data. IOW, how much room 1781d4dbff95SChris Mason * the leaf has left for both items and data 1782d4dbff95SChris Mason */ 17835f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf) 1784d4dbff95SChris Mason { 17855f39d397SChris Mason int nritems = btrfs_header_nritems(leaf); 17865f39d397SChris Mason int ret; 17875f39d397SChris Mason ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems); 17885f39d397SChris Mason if (ret < 0) { 17895f39d397SChris Mason printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n", 1790ae2f5411SJens Axboe ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root), 17915f39d397SChris Mason leaf_space_used(leaf, 0, nritems), nritems); 17925f39d397SChris Mason } 17935f39d397SChris Mason return ret; 1794d4dbff95SChris Mason } 1795d4dbff95SChris Mason 1796d4dbff95SChris Mason /* 179700ec4c51SChris Mason * push some data in the path leaf to the right, trying to free up at 179800ec4c51SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 1799aa5d6bedSChris Mason * 1800aa5d6bedSChris Mason * returns 1 if the push failed because the other node didn't have enough 1801aa5d6bedSChris Mason * room, 0 if everything worked out and < 0 if there were major errors. 180200ec4c51SChris Mason */ 1803e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root 180434a38218SChris Mason *root, struct btrfs_path *path, int data_size, 180534a38218SChris Mason int empty) 180600ec4c51SChris Mason { 18075f39d397SChris Mason struct extent_buffer *left = path->nodes[0]; 18085f39d397SChris Mason struct extent_buffer *right; 18095f39d397SChris Mason struct extent_buffer *upper; 18105f39d397SChris Mason struct btrfs_disk_key disk_key; 181100ec4c51SChris Mason int slot; 181234a38218SChris Mason u32 i; 181300ec4c51SChris Mason int free_space; 181400ec4c51SChris Mason int push_space = 0; 181500ec4c51SChris Mason int push_items = 0; 18160783fcfcSChris Mason struct btrfs_item *item; 18177518a238SChris Mason u32 left_nritems; 181834a38218SChris Mason u32 nr; 18197518a238SChris Mason u32 right_nritems; 18205f39d397SChris Mason u32 data_end; 1821db94535dSChris Mason u32 this_item_size; 182254aa1f4dSChris Mason int ret; 182300ec4c51SChris Mason 182400ec4c51SChris Mason slot = path->slots[1]; 182500ec4c51SChris Mason if (!path->nodes[1]) { 182600ec4c51SChris Mason return 1; 182700ec4c51SChris Mason } 182800ec4c51SChris Mason upper = path->nodes[1]; 18295f39d397SChris Mason if (slot >= btrfs_header_nritems(upper) - 1) 183000ec4c51SChris Mason return 1; 18315f39d397SChris Mason 1832a2135011SChris Mason WARN_ON(!btrfs_tree_locked(path->nodes[1])); 1833a2135011SChris Mason 1834ca7a79adSChris Mason right = read_node_slot(root, upper, slot + 1); 1835925baeddSChris Mason btrfs_tree_lock(right); 1836123abc88SChris Mason free_space = btrfs_leaf_free_space(root, right); 1837925baeddSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) 1838925baeddSChris Mason goto out_unlock; 183902217ed2SChris Mason 18405f39d397SChris Mason /* cow and double check */ 18415f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, upper, 18425f39d397SChris Mason slot + 1, &right); 1843925baeddSChris Mason if (ret) 1844925baeddSChris Mason goto out_unlock; 1845925baeddSChris Mason 18465f39d397SChris Mason free_space = btrfs_leaf_free_space(root, right); 1847925baeddSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) 1848925baeddSChris Mason goto out_unlock; 18495f39d397SChris Mason 18505f39d397SChris Mason left_nritems = btrfs_header_nritems(left); 1851925baeddSChris Mason if (left_nritems == 0) 1852925baeddSChris Mason goto out_unlock; 18535f39d397SChris Mason 185434a38218SChris Mason if (empty) 185534a38218SChris Mason nr = 0; 185634a38218SChris Mason else 185734a38218SChris Mason nr = 1; 185834a38218SChris Mason 185934a38218SChris Mason i = left_nritems - 1; 186034a38218SChris Mason while (i >= nr) { 18615f39d397SChris Mason item = btrfs_item_nr(left, i); 1862db94535dSChris Mason 186300ec4c51SChris Mason if (path->slots[0] == i) 186400ec4c51SChris Mason push_space += data_size + sizeof(*item); 1865db94535dSChris Mason 1866db94535dSChris Mason if (!left->map_token) { 1867db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 1868db94535dSChris Mason sizeof(struct btrfs_item), 1869db94535dSChris Mason &left->map_token, &left->kaddr, 1870db94535dSChris Mason &left->map_start, &left->map_len, 1871db94535dSChris Mason KM_USER1); 1872db94535dSChris Mason } 1873db94535dSChris Mason 1874db94535dSChris Mason this_item_size = btrfs_item_size(left, item); 1875db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 187600ec4c51SChris Mason break; 187700ec4c51SChris Mason push_items++; 1878db94535dSChris Mason push_space += this_item_size + sizeof(*item); 187934a38218SChris Mason if (i == 0) 188034a38218SChris Mason break; 188134a38218SChris Mason i--; 1882db94535dSChris Mason } 1883db94535dSChris Mason if (left->map_token) { 1884db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 1885db94535dSChris Mason left->map_token = NULL; 188600ec4c51SChris Mason } 18875f39d397SChris Mason 1888925baeddSChris Mason if (push_items == 0) 1889925baeddSChris Mason goto out_unlock; 18905f39d397SChris Mason 189134a38218SChris Mason if (!empty && push_items == left_nritems) 1892a429e513SChris Mason WARN_ON(1); 18935f39d397SChris Mason 189400ec4c51SChris Mason /* push left to right */ 18955f39d397SChris Mason right_nritems = btrfs_header_nritems(right); 189634a38218SChris Mason 18975f39d397SChris Mason push_space = btrfs_item_end_nr(left, left_nritems - push_items); 1898123abc88SChris Mason push_space -= leaf_data_end(root, left); 18995f39d397SChris Mason 190000ec4c51SChris Mason /* make room in the right data area */ 19015f39d397SChris Mason data_end = leaf_data_end(root, right); 19025f39d397SChris Mason memmove_extent_buffer(right, 19035f39d397SChris Mason btrfs_leaf_data(right) + data_end - push_space, 19045f39d397SChris Mason btrfs_leaf_data(right) + data_end, 19055f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root) - data_end); 19065f39d397SChris Mason 190700ec4c51SChris Mason /* copy from the left data area */ 19085f39d397SChris Mason copy_extent_buffer(right, left, btrfs_leaf_data(right) + 1909d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 1910d6025579SChris Mason btrfs_leaf_data(left) + leaf_data_end(root, left), 1911d6025579SChris Mason push_space); 19125f39d397SChris Mason 19135f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(push_items), 19145f39d397SChris Mason btrfs_item_nr_offset(0), 19150783fcfcSChris Mason right_nritems * sizeof(struct btrfs_item)); 19165f39d397SChris Mason 191700ec4c51SChris Mason /* copy the items from left to right */ 19185f39d397SChris Mason copy_extent_buffer(right, left, btrfs_item_nr_offset(0), 19195f39d397SChris Mason btrfs_item_nr_offset(left_nritems - push_items), 19200783fcfcSChris Mason push_items * sizeof(struct btrfs_item)); 192100ec4c51SChris Mason 192200ec4c51SChris Mason /* update the item pointers */ 19237518a238SChris Mason right_nritems += push_items; 19245f39d397SChris Mason btrfs_set_header_nritems(right, right_nritems); 1925123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 19267518a238SChris Mason for (i = 0; i < right_nritems; i++) { 19275f39d397SChris Mason item = btrfs_item_nr(right, i); 1928db94535dSChris Mason if (!right->map_token) { 1929db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1930db94535dSChris Mason sizeof(struct btrfs_item), 1931db94535dSChris Mason &right->map_token, &right->kaddr, 1932db94535dSChris Mason &right->map_start, &right->map_len, 1933db94535dSChris Mason KM_USER1); 1934db94535dSChris Mason } 1935db94535dSChris Mason push_space -= btrfs_item_size(right, item); 1936db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 1937db94535dSChris Mason } 1938db94535dSChris Mason 1939db94535dSChris Mason if (right->map_token) { 1940db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1941db94535dSChris Mason right->map_token = NULL; 194200ec4c51SChris Mason } 19437518a238SChris Mason left_nritems -= push_items; 19445f39d397SChris Mason btrfs_set_header_nritems(left, left_nritems); 194500ec4c51SChris Mason 194634a38218SChris Mason if (left_nritems) 19475f39d397SChris Mason btrfs_mark_buffer_dirty(left); 19485f39d397SChris Mason btrfs_mark_buffer_dirty(right); 1949a429e513SChris Mason 19505f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 19515f39d397SChris Mason btrfs_set_node_key(upper, &disk_key, slot + 1); 1952d6025579SChris Mason btrfs_mark_buffer_dirty(upper); 195302217ed2SChris Mason 195400ec4c51SChris Mason /* then fixup the leaf pointer in the path */ 19557518a238SChris Mason if (path->slots[0] >= left_nritems) { 19567518a238SChris Mason path->slots[0] -= left_nritems; 1957925baeddSChris Mason if (btrfs_header_nritems(path->nodes[0]) == 0) 1958925baeddSChris Mason clean_tree_block(trans, root, path->nodes[0]); 1959925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 19605f39d397SChris Mason free_extent_buffer(path->nodes[0]); 19615f39d397SChris Mason path->nodes[0] = right; 196200ec4c51SChris Mason path->slots[1] += 1; 196300ec4c51SChris Mason } else { 1964925baeddSChris Mason btrfs_tree_unlock(right); 19655f39d397SChris Mason free_extent_buffer(right); 196600ec4c51SChris Mason } 196700ec4c51SChris Mason return 0; 1968925baeddSChris Mason 1969925baeddSChris Mason out_unlock: 1970925baeddSChris Mason btrfs_tree_unlock(right); 1971925baeddSChris Mason free_extent_buffer(right); 1972925baeddSChris Mason return 1; 197300ec4c51SChris Mason } 1974925baeddSChris Mason 197500ec4c51SChris Mason /* 197674123bd7SChris Mason * push some data in the path leaf to the left, trying to free up at 197774123bd7SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 197874123bd7SChris Mason */ 1979e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root 198034a38218SChris Mason *root, struct btrfs_path *path, int data_size, 198134a38218SChris Mason int empty) 1982be0e5c09SChris Mason { 19835f39d397SChris Mason struct btrfs_disk_key disk_key; 19845f39d397SChris Mason struct extent_buffer *right = path->nodes[0]; 19855f39d397SChris Mason struct extent_buffer *left; 1986be0e5c09SChris Mason int slot; 1987be0e5c09SChris Mason int i; 1988be0e5c09SChris Mason int free_space; 1989be0e5c09SChris Mason int push_space = 0; 1990be0e5c09SChris Mason int push_items = 0; 19910783fcfcSChris Mason struct btrfs_item *item; 19927518a238SChris Mason u32 old_left_nritems; 19935f39d397SChris Mason u32 right_nritems; 199434a38218SChris Mason u32 nr; 1995aa5d6bedSChris Mason int ret = 0; 1996aa5d6bedSChris Mason int wret; 1997db94535dSChris Mason u32 this_item_size; 1998db94535dSChris Mason u32 old_left_item_size; 1999be0e5c09SChris Mason 2000be0e5c09SChris Mason slot = path->slots[1]; 20015f39d397SChris Mason if (slot == 0) 2002be0e5c09SChris Mason return 1; 20035f39d397SChris Mason if (!path->nodes[1]) 2004be0e5c09SChris Mason return 1; 20055f39d397SChris Mason 20063685f791SChris Mason right_nritems = btrfs_header_nritems(right); 20073685f791SChris Mason if (right_nritems == 0) { 20083685f791SChris Mason return 1; 20093685f791SChris Mason } 20103685f791SChris Mason 2011a2135011SChris Mason WARN_ON(!btrfs_tree_locked(path->nodes[1])); 2012a2135011SChris Mason 2013ca7a79adSChris Mason left = read_node_slot(root, path->nodes[1], slot - 1); 2014925baeddSChris Mason btrfs_tree_lock(left); 2015123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 20160783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 2017925baeddSChris Mason ret = 1; 2018925baeddSChris Mason goto out; 2019be0e5c09SChris Mason } 202002217ed2SChris Mason 202102217ed2SChris Mason /* cow and double check */ 20225f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, 20235f39d397SChris Mason path->nodes[1], slot - 1, &left); 202454aa1f4dSChris Mason if (ret) { 202554aa1f4dSChris Mason /* we hit -ENOSPC, but it isn't fatal here */ 2026925baeddSChris Mason ret = 1; 2027925baeddSChris Mason goto out; 202854aa1f4dSChris Mason } 20293685f791SChris Mason 2030123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 20310783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 2032925baeddSChris Mason ret = 1; 2033925baeddSChris Mason goto out; 203402217ed2SChris Mason } 203502217ed2SChris Mason 203634a38218SChris Mason if (empty) 203734a38218SChris Mason nr = right_nritems; 203834a38218SChris Mason else 203934a38218SChris Mason nr = right_nritems - 1; 204034a38218SChris Mason 204134a38218SChris Mason for (i = 0; i < nr; i++) { 20425f39d397SChris Mason item = btrfs_item_nr(right, i); 2043db94535dSChris Mason if (!right->map_token) { 2044db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2045db94535dSChris Mason sizeof(struct btrfs_item), 2046db94535dSChris Mason &right->map_token, &right->kaddr, 2047db94535dSChris Mason &right->map_start, &right->map_len, 2048db94535dSChris Mason KM_USER1); 2049db94535dSChris Mason } 2050db94535dSChris Mason 2051be0e5c09SChris Mason if (path->slots[0] == i) 2052be0e5c09SChris Mason push_space += data_size + sizeof(*item); 2053db94535dSChris Mason 2054db94535dSChris Mason this_item_size = btrfs_item_size(right, item); 2055db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 2056be0e5c09SChris Mason break; 2057db94535dSChris Mason 2058be0e5c09SChris Mason push_items++; 2059db94535dSChris Mason push_space += this_item_size + sizeof(*item); 2060be0e5c09SChris Mason } 2061db94535dSChris Mason 2062db94535dSChris Mason if (right->map_token) { 2063db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2064db94535dSChris Mason right->map_token = NULL; 2065db94535dSChris Mason } 2066db94535dSChris Mason 2067be0e5c09SChris Mason if (push_items == 0) { 2068925baeddSChris Mason ret = 1; 2069925baeddSChris Mason goto out; 2070be0e5c09SChris Mason } 207134a38218SChris Mason if (!empty && push_items == btrfs_header_nritems(right)) 2072a429e513SChris Mason WARN_ON(1); 20735f39d397SChris Mason 2074be0e5c09SChris Mason /* push data from right to left */ 20755f39d397SChris Mason copy_extent_buffer(left, right, 20765f39d397SChris Mason btrfs_item_nr_offset(btrfs_header_nritems(left)), 20775f39d397SChris Mason btrfs_item_nr_offset(0), 20785f39d397SChris Mason push_items * sizeof(struct btrfs_item)); 20795f39d397SChris Mason 2080123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root) - 20815f39d397SChris Mason btrfs_item_offset_nr(right, push_items -1); 20825f39d397SChris Mason 20835f39d397SChris Mason copy_extent_buffer(left, right, btrfs_leaf_data(left) + 2084d6025579SChris Mason leaf_data_end(root, left) - push_space, 2085123abc88SChris Mason btrfs_leaf_data(right) + 20865f39d397SChris Mason btrfs_item_offset_nr(right, push_items - 1), 2087be0e5c09SChris Mason push_space); 20885f39d397SChris Mason old_left_nritems = btrfs_header_nritems(left); 2089eb60ceacSChris Mason BUG_ON(old_left_nritems < 0); 2090eb60ceacSChris Mason 2091db94535dSChris Mason old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1); 2092be0e5c09SChris Mason for (i = old_left_nritems; i < old_left_nritems + push_items; i++) { 20935f39d397SChris Mason u32 ioff; 2094db94535dSChris Mason 20955f39d397SChris Mason item = btrfs_item_nr(left, i); 2096db94535dSChris Mason if (!left->map_token) { 2097db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 2098db94535dSChris Mason sizeof(struct btrfs_item), 2099db94535dSChris Mason &left->map_token, &left->kaddr, 2100db94535dSChris Mason &left->map_start, &left->map_len, 2101db94535dSChris Mason KM_USER1); 2102db94535dSChris Mason } 2103db94535dSChris Mason 21045f39d397SChris Mason ioff = btrfs_item_offset(left, item); 21055f39d397SChris Mason btrfs_set_item_offset(left, item, 2106db94535dSChris Mason ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size)); 2107be0e5c09SChris Mason } 21085f39d397SChris Mason btrfs_set_header_nritems(left, old_left_nritems + push_items); 2109db94535dSChris Mason if (left->map_token) { 2110db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 2111db94535dSChris Mason left->map_token = NULL; 2112db94535dSChris Mason } 2113be0e5c09SChris Mason 2114be0e5c09SChris Mason /* fixup right node */ 211534a38218SChris Mason if (push_items > right_nritems) { 211634a38218SChris Mason printk("push items %d nr %u\n", push_items, right_nritems); 211734a38218SChris Mason WARN_ON(1); 211834a38218SChris Mason } 211934a38218SChris Mason 212034a38218SChris Mason if (push_items < right_nritems) { 21215f39d397SChris Mason push_space = btrfs_item_offset_nr(right, push_items - 1) - 2122123abc88SChris Mason leaf_data_end(root, right); 21235f39d397SChris Mason memmove_extent_buffer(right, btrfs_leaf_data(right) + 2124d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 2125d6025579SChris Mason btrfs_leaf_data(right) + 2126123abc88SChris Mason leaf_data_end(root, right), push_space); 21275f39d397SChris Mason 21285f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(0), 21295f39d397SChris Mason btrfs_item_nr_offset(push_items), 21305f39d397SChris Mason (btrfs_header_nritems(right) - push_items) * 21310783fcfcSChris Mason sizeof(struct btrfs_item)); 213234a38218SChris Mason } 2133eef1c494SYan right_nritems -= push_items; 2134eef1c494SYan btrfs_set_header_nritems(right, right_nritems); 2135123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 21365f39d397SChris Mason for (i = 0; i < right_nritems; i++) { 21375f39d397SChris Mason item = btrfs_item_nr(right, i); 2138db94535dSChris Mason 2139db94535dSChris Mason if (!right->map_token) { 2140db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2141db94535dSChris Mason sizeof(struct btrfs_item), 2142db94535dSChris Mason &right->map_token, &right->kaddr, 2143db94535dSChris Mason &right->map_start, &right->map_len, 2144db94535dSChris Mason KM_USER1); 2145db94535dSChris Mason } 2146db94535dSChris Mason 2147db94535dSChris Mason push_space = push_space - btrfs_item_size(right, item); 2148db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 2149db94535dSChris Mason } 2150db94535dSChris Mason if (right->map_token) { 2151db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2152db94535dSChris Mason right->map_token = NULL; 2153be0e5c09SChris Mason } 2154eb60ceacSChris Mason 21555f39d397SChris Mason btrfs_mark_buffer_dirty(left); 215634a38218SChris Mason if (right_nritems) 21575f39d397SChris Mason btrfs_mark_buffer_dirty(right); 2158098f59c2SChris Mason 21595f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 21605f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, 1); 2161aa5d6bedSChris Mason if (wret) 2162aa5d6bedSChris Mason ret = wret; 2163be0e5c09SChris Mason 2164be0e5c09SChris Mason /* then fixup the leaf pointer in the path */ 2165be0e5c09SChris Mason if (path->slots[0] < push_items) { 2166be0e5c09SChris Mason path->slots[0] += old_left_nritems; 2167925baeddSChris Mason if (btrfs_header_nritems(path->nodes[0]) == 0) 2168925baeddSChris Mason clean_tree_block(trans, root, path->nodes[0]); 2169925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 21705f39d397SChris Mason free_extent_buffer(path->nodes[0]); 21715f39d397SChris Mason path->nodes[0] = left; 2172be0e5c09SChris Mason path->slots[1] -= 1; 2173be0e5c09SChris Mason } else { 2174925baeddSChris Mason btrfs_tree_unlock(left); 21755f39d397SChris Mason free_extent_buffer(left); 2176be0e5c09SChris Mason path->slots[0] -= push_items; 2177be0e5c09SChris Mason } 2178eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 2179aa5d6bedSChris Mason return ret; 2180925baeddSChris Mason out: 2181925baeddSChris Mason btrfs_tree_unlock(left); 2182925baeddSChris Mason free_extent_buffer(left); 2183925baeddSChris Mason return ret; 2184be0e5c09SChris Mason } 2185be0e5c09SChris Mason 218674123bd7SChris Mason /* 218774123bd7SChris Mason * split the path's leaf in two, making sure there is at least data_size 218874123bd7SChris Mason * available for the resulting leaf level of the path. 2189aa5d6bedSChris Mason * 2190aa5d6bedSChris Mason * returns 0 if all went well and < 0 on failure. 219174123bd7SChris Mason */ 2192e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 2193d4dbff95SChris Mason *root, struct btrfs_key *ins_key, 2194cc0c5538SChris Mason struct btrfs_path *path, int data_size, int extend) 2195be0e5c09SChris Mason { 21967bb86316SChris Mason u64 root_gen; 21975f39d397SChris Mason struct extent_buffer *l; 21987518a238SChris Mason u32 nritems; 2199eb60ceacSChris Mason int mid; 2200eb60ceacSChris Mason int slot; 22015f39d397SChris Mason struct extent_buffer *right; 22020783fcfcSChris Mason int space_needed = data_size + sizeof(struct btrfs_item); 2203be0e5c09SChris Mason int data_copy_size; 2204be0e5c09SChris Mason int rt_data_off; 2205be0e5c09SChris Mason int i; 2206d4dbff95SChris Mason int ret = 0; 2207aa5d6bedSChris Mason int wret; 2208cc0c5538SChris Mason int double_split; 2209cc0c5538SChris Mason int num_doubles = 0; 2210d4dbff95SChris Mason struct btrfs_disk_key disk_key; 2211be0e5c09SChris Mason 2212cc0c5538SChris Mason if (extend) 2213cc0c5538SChris Mason space_needed = data_size; 2214cc0c5538SChris Mason 22157bb86316SChris Mason if (root->ref_cows) 22167bb86316SChris Mason root_gen = trans->transid; 22177bb86316SChris Mason else 22187bb86316SChris Mason root_gen = 0; 22197bb86316SChris Mason 222040689478SChris Mason /* first try to make some room by pushing left and right */ 22213685f791SChris Mason if (ins_key->type != BTRFS_DIR_ITEM_KEY) { 222234a38218SChris Mason wret = push_leaf_right(trans, root, path, data_size, 0); 22233326d1b0SChris Mason if (wret < 0) { 2224eaee50e8SChris Mason return wret; 22253326d1b0SChris Mason } 2226eaee50e8SChris Mason if (wret) { 222734a38218SChris Mason wret = push_leaf_left(trans, root, path, data_size, 0); 2228eaee50e8SChris Mason if (wret < 0) 2229eaee50e8SChris Mason return wret; 2230eaee50e8SChris Mason } 22315f39d397SChris Mason l = path->nodes[0]; 2232aa5d6bedSChris Mason 2233aa5d6bedSChris Mason /* did the pushes work? */ 2234cc0c5538SChris Mason if (btrfs_leaf_free_space(root, l) >= space_needed) 2235be0e5c09SChris Mason return 0; 22363326d1b0SChris Mason } 2237aa5d6bedSChris Mason 22385c680ed6SChris Mason if (!path->nodes[1]) { 2239e089f05cSChris Mason ret = insert_new_root(trans, root, path, 1); 22405c680ed6SChris Mason if (ret) 22415c680ed6SChris Mason return ret; 22425c680ed6SChris Mason } 2243cc0c5538SChris Mason again: 2244cc0c5538SChris Mason double_split = 0; 2245cc0c5538SChris Mason l = path->nodes[0]; 2246eb60ceacSChris Mason slot = path->slots[0]; 22475f39d397SChris Mason nritems = btrfs_header_nritems(l); 2248eb60ceacSChris Mason mid = (nritems + 1)/ 2; 224954aa1f4dSChris Mason 22507bb86316SChris Mason btrfs_item_key(l, &disk_key, 0); 22517bb86316SChris Mason 2252925baeddSChris Mason right = btrfs_alloc_free_block(trans, root, root->leafsize, 22537bb86316SChris Mason root->root_key.objectid, 22547bb86316SChris Mason root_gen, disk_key.objectid, 0, 2255db94535dSChris Mason l->start, 0); 2256cea9e445SChris Mason if (IS_ERR(right)) { 2257cea9e445SChris Mason BUG_ON(1); 22585f39d397SChris Mason return PTR_ERR(right); 2259cea9e445SChris Mason } 226054aa1f4dSChris Mason 22615f39d397SChris Mason memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header)); 2262db94535dSChris Mason btrfs_set_header_bytenr(right, right->start); 22635f39d397SChris Mason btrfs_set_header_generation(right, trans->transid); 22645f39d397SChris Mason btrfs_set_header_owner(right, root->root_key.objectid); 22655f39d397SChris Mason btrfs_set_header_level(right, 0); 22665f39d397SChris Mason write_extent_buffer(right, root->fs_info->fsid, 22675f39d397SChris Mason (unsigned long)btrfs_header_fsid(right), 22685f39d397SChris Mason BTRFS_FSID_SIZE); 2269e17cade2SChris Mason 2270e17cade2SChris Mason write_extent_buffer(right, root->fs_info->chunk_tree_uuid, 2271e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(right), 2272e17cade2SChris Mason BTRFS_UUID_SIZE); 2273d4dbff95SChris Mason if (mid <= slot) { 2274d4dbff95SChris Mason if (nritems == 1 || 2275d4dbff95SChris Mason leaf_space_used(l, mid, nritems - mid) + space_needed > 2276d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 2277d4dbff95SChris Mason if (slot >= nritems) { 2278d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 22795f39d397SChris Mason btrfs_set_header_nritems(right, 0); 2280d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 2281db94535dSChris Mason &disk_key, right->start, 2282d4dbff95SChris Mason path->slots[1] + 1, 1); 2283d4dbff95SChris Mason if (wret) 2284d4dbff95SChris Mason ret = wret; 2285925baeddSChris Mason 2286925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 22875f39d397SChris Mason free_extent_buffer(path->nodes[0]); 22885f39d397SChris Mason path->nodes[0] = right; 2289d4dbff95SChris Mason path->slots[0] = 0; 2290d4dbff95SChris Mason path->slots[1] += 1; 22910ef8b242SChris Mason btrfs_mark_buffer_dirty(right); 2292d4dbff95SChris Mason return ret; 2293d4dbff95SChris Mason } 2294d4dbff95SChris Mason mid = slot; 22953326d1b0SChris Mason if (mid != nritems && 22963326d1b0SChris Mason leaf_space_used(l, mid, nritems - mid) + 22973326d1b0SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 2298d4dbff95SChris Mason double_split = 1; 2299d4dbff95SChris Mason } 23003326d1b0SChris Mason } 2301d4dbff95SChris Mason } else { 2302d4dbff95SChris Mason if (leaf_space_used(l, 0, mid + 1) + space_needed > 2303d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 2304cc0c5538SChris Mason if (!extend && slot == 0) { 2305d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 23065f39d397SChris Mason btrfs_set_header_nritems(right, 0); 2307d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 2308d4dbff95SChris Mason &disk_key, 2309db94535dSChris Mason right->start, 2310098f59c2SChris Mason path->slots[1], 1); 2311d4dbff95SChris Mason if (wret) 2312d4dbff95SChris Mason ret = wret; 2313925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 23145f39d397SChris Mason free_extent_buffer(path->nodes[0]); 23155f39d397SChris Mason path->nodes[0] = right; 2316d4dbff95SChris Mason path->slots[0] = 0; 2317a429e513SChris Mason if (path->slots[1] == 0) { 2318a429e513SChris Mason wret = fixup_low_keys(trans, root, 2319a429e513SChris Mason path, &disk_key, 1); 2320a429e513SChris Mason if (wret) 2321a429e513SChris Mason ret = wret; 2322a429e513SChris Mason } 23230ef8b242SChris Mason btrfs_mark_buffer_dirty(right); 2324d4dbff95SChris Mason return ret; 2325cc0c5538SChris Mason } else if (extend && slot == 0) { 2326cc0c5538SChris Mason mid = 1; 2327cc0c5538SChris Mason } else { 2328d4dbff95SChris Mason mid = slot; 23295ee78ac7SChris Mason if (mid != nritems && 23305ee78ac7SChris Mason leaf_space_used(l, mid, nritems - mid) + 23315ee78ac7SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 2332d4dbff95SChris Mason double_split = 1; 2333d4dbff95SChris Mason } 2334d4dbff95SChris Mason } 23355ee78ac7SChris Mason } 2336cc0c5538SChris Mason } 23375f39d397SChris Mason nritems = nritems - mid; 23385f39d397SChris Mason btrfs_set_header_nritems(right, nritems); 23395f39d397SChris Mason data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l); 23405f39d397SChris Mason 23415f39d397SChris Mason copy_extent_buffer(right, l, btrfs_item_nr_offset(0), 23425f39d397SChris Mason btrfs_item_nr_offset(mid), 23435f39d397SChris Mason nritems * sizeof(struct btrfs_item)); 23445f39d397SChris Mason 23455f39d397SChris Mason copy_extent_buffer(right, l, 2346d6025579SChris Mason btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - 2347123abc88SChris Mason data_copy_size, btrfs_leaf_data(l) + 2348123abc88SChris Mason leaf_data_end(root, l), data_copy_size); 234974123bd7SChris Mason 23505f39d397SChris Mason rt_data_off = BTRFS_LEAF_DATA_SIZE(root) - 23515f39d397SChris Mason btrfs_item_end_nr(l, mid); 23525f39d397SChris Mason 23535f39d397SChris Mason for (i = 0; i < nritems; i++) { 23545f39d397SChris Mason struct btrfs_item *item = btrfs_item_nr(right, i); 2355db94535dSChris Mason u32 ioff; 2356db94535dSChris Mason 2357db94535dSChris Mason if (!right->map_token) { 2358db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2359db94535dSChris Mason sizeof(struct btrfs_item), 2360db94535dSChris Mason &right->map_token, &right->kaddr, 2361db94535dSChris Mason &right->map_start, &right->map_len, 2362db94535dSChris Mason KM_USER1); 2363db94535dSChris Mason } 2364db94535dSChris Mason 2365db94535dSChris Mason ioff = btrfs_item_offset(right, item); 23665f39d397SChris Mason btrfs_set_item_offset(right, item, ioff + rt_data_off); 23670783fcfcSChris Mason } 236874123bd7SChris Mason 2369db94535dSChris Mason if (right->map_token) { 2370db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2371db94535dSChris Mason right->map_token = NULL; 2372db94535dSChris Mason } 2373db94535dSChris Mason 23745f39d397SChris Mason btrfs_set_header_nritems(l, mid); 2375aa5d6bedSChris Mason ret = 0; 23765f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 2377db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, right->start, 2378db94535dSChris Mason path->slots[1] + 1, 1); 2379aa5d6bedSChris Mason if (wret) 2380aa5d6bedSChris Mason ret = wret; 23815f39d397SChris Mason 23825f39d397SChris Mason btrfs_mark_buffer_dirty(right); 23835f39d397SChris Mason btrfs_mark_buffer_dirty(l); 2384eb60ceacSChris Mason BUG_ON(path->slots[0] != slot); 23855f39d397SChris Mason 2386be0e5c09SChris Mason if (mid <= slot) { 2387925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 23885f39d397SChris Mason free_extent_buffer(path->nodes[0]); 23895f39d397SChris Mason path->nodes[0] = right; 2390be0e5c09SChris Mason path->slots[0] -= mid; 2391be0e5c09SChris Mason path->slots[1] += 1; 2392925baeddSChris Mason } else { 2393925baeddSChris Mason btrfs_tree_unlock(right); 23945f39d397SChris Mason free_extent_buffer(right); 2395925baeddSChris Mason } 23965f39d397SChris Mason 2397eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 2398d4dbff95SChris Mason 2399cc0c5538SChris Mason if (double_split) { 2400cc0c5538SChris Mason BUG_ON(num_doubles != 0); 2401cc0c5538SChris Mason num_doubles++; 2402cc0c5538SChris Mason goto again; 24033326d1b0SChris Mason } 2404be0e5c09SChris Mason return ret; 2405be0e5c09SChris Mason } 2406be0e5c09SChris Mason 2407b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans, 2408b18c6685SChris Mason struct btrfs_root *root, 2409b18c6685SChris Mason struct btrfs_path *path, 2410179e29e4SChris Mason u32 new_size, int from_end) 2411b18c6685SChris Mason { 2412b18c6685SChris Mason int ret = 0; 2413b18c6685SChris Mason int slot; 2414b18c6685SChris Mason int slot_orig; 24155f39d397SChris Mason struct extent_buffer *leaf; 24165f39d397SChris Mason struct btrfs_item *item; 2417b18c6685SChris Mason u32 nritems; 2418b18c6685SChris Mason unsigned int data_end; 2419b18c6685SChris Mason unsigned int old_data_start; 2420b18c6685SChris Mason unsigned int old_size; 2421b18c6685SChris Mason unsigned int size_diff; 2422b18c6685SChris Mason int i; 2423b18c6685SChris Mason 2424b18c6685SChris Mason slot_orig = path->slots[0]; 24255f39d397SChris Mason leaf = path->nodes[0]; 2426179e29e4SChris Mason slot = path->slots[0]; 2427179e29e4SChris Mason 2428179e29e4SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 2429179e29e4SChris Mason if (old_size == new_size) 2430179e29e4SChris Mason return 0; 2431b18c6685SChris Mason 24325f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2433b18c6685SChris Mason data_end = leaf_data_end(root, leaf); 2434b18c6685SChris Mason 24355f39d397SChris Mason old_data_start = btrfs_item_offset_nr(leaf, slot); 2436179e29e4SChris Mason 2437b18c6685SChris Mason size_diff = old_size - new_size; 2438b18c6685SChris Mason 2439b18c6685SChris Mason BUG_ON(slot < 0); 2440b18c6685SChris Mason BUG_ON(slot >= nritems); 2441b18c6685SChris Mason 2442b18c6685SChris Mason /* 2443b18c6685SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2444b18c6685SChris Mason */ 2445b18c6685SChris Mason /* first correct the data pointers */ 2446b18c6685SChris Mason for (i = slot; i < nritems; i++) { 24475f39d397SChris Mason u32 ioff; 24485f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2449db94535dSChris Mason 2450db94535dSChris Mason if (!leaf->map_token) { 2451db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2452db94535dSChris Mason sizeof(struct btrfs_item), 2453db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2454db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2455db94535dSChris Mason KM_USER1); 2456db94535dSChris Mason } 2457db94535dSChris Mason 24585f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 24595f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + size_diff); 2460b18c6685SChris Mason } 2461db94535dSChris Mason 2462db94535dSChris Mason if (leaf->map_token) { 2463db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2464db94535dSChris Mason leaf->map_token = NULL; 2465db94535dSChris Mason } 2466db94535dSChris Mason 2467b18c6685SChris Mason /* shift the data */ 2468179e29e4SChris Mason if (from_end) { 24695f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2470b18c6685SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 2471b18c6685SChris Mason data_end, old_data_start + new_size - data_end); 2472179e29e4SChris Mason } else { 2473179e29e4SChris Mason struct btrfs_disk_key disk_key; 2474179e29e4SChris Mason u64 offset; 2475179e29e4SChris Mason 2476179e29e4SChris Mason btrfs_item_key(leaf, &disk_key, slot); 2477179e29e4SChris Mason 2478179e29e4SChris Mason if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) { 2479179e29e4SChris Mason unsigned long ptr; 2480179e29e4SChris Mason struct btrfs_file_extent_item *fi; 2481179e29e4SChris Mason 2482179e29e4SChris Mason fi = btrfs_item_ptr(leaf, slot, 2483179e29e4SChris Mason struct btrfs_file_extent_item); 2484179e29e4SChris Mason fi = (struct btrfs_file_extent_item *)( 2485179e29e4SChris Mason (unsigned long)fi - size_diff); 2486179e29e4SChris Mason 2487179e29e4SChris Mason if (btrfs_file_extent_type(leaf, fi) == 2488179e29e4SChris Mason BTRFS_FILE_EXTENT_INLINE) { 2489179e29e4SChris Mason ptr = btrfs_item_ptr_offset(leaf, slot); 2490179e29e4SChris Mason memmove_extent_buffer(leaf, ptr, 2491179e29e4SChris Mason (unsigned long)fi, 2492179e29e4SChris Mason offsetof(struct btrfs_file_extent_item, 2493179e29e4SChris Mason disk_bytenr)); 2494179e29e4SChris Mason } 2495179e29e4SChris Mason } 2496179e29e4SChris Mason 2497179e29e4SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2498179e29e4SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 2499179e29e4SChris Mason data_end, old_data_start - data_end); 2500179e29e4SChris Mason 2501179e29e4SChris Mason offset = btrfs_disk_key_offset(&disk_key); 2502179e29e4SChris Mason btrfs_set_disk_key_offset(&disk_key, offset + size_diff); 2503179e29e4SChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 2504179e29e4SChris Mason if (slot == 0) 2505179e29e4SChris Mason fixup_low_keys(trans, root, path, &disk_key, 1); 2506179e29e4SChris Mason } 25075f39d397SChris Mason 25085f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 25095f39d397SChris Mason btrfs_set_item_size(leaf, item, new_size); 25105f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2511b18c6685SChris Mason 2512b18c6685SChris Mason ret = 0; 25135f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 25145f39d397SChris Mason btrfs_print_leaf(root, leaf); 2515b18c6685SChris Mason BUG(); 25165f39d397SChris Mason } 2517b18c6685SChris Mason return ret; 2518b18c6685SChris Mason } 2519b18c6685SChris Mason 25205f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans, 25215f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 25225f39d397SChris Mason u32 data_size) 25236567e837SChris Mason { 25246567e837SChris Mason int ret = 0; 25256567e837SChris Mason int slot; 25266567e837SChris Mason int slot_orig; 25275f39d397SChris Mason struct extent_buffer *leaf; 25285f39d397SChris Mason struct btrfs_item *item; 25296567e837SChris Mason u32 nritems; 25306567e837SChris Mason unsigned int data_end; 25316567e837SChris Mason unsigned int old_data; 25326567e837SChris Mason unsigned int old_size; 25336567e837SChris Mason int i; 25346567e837SChris Mason 25356567e837SChris Mason slot_orig = path->slots[0]; 25365f39d397SChris Mason leaf = path->nodes[0]; 25376567e837SChris Mason 25385f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 25396567e837SChris Mason data_end = leaf_data_end(root, leaf); 25406567e837SChris Mason 25415f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < data_size) { 25425f39d397SChris Mason btrfs_print_leaf(root, leaf); 25436567e837SChris Mason BUG(); 25445f39d397SChris Mason } 25456567e837SChris Mason slot = path->slots[0]; 25465f39d397SChris Mason old_data = btrfs_item_end_nr(leaf, slot); 25476567e837SChris Mason 25486567e837SChris Mason BUG_ON(slot < 0); 25493326d1b0SChris Mason if (slot >= nritems) { 25503326d1b0SChris Mason btrfs_print_leaf(root, leaf); 25513326d1b0SChris Mason printk("slot %d too large, nritems %d\n", slot, nritems); 25523326d1b0SChris Mason BUG_ON(1); 25533326d1b0SChris Mason } 25546567e837SChris Mason 25556567e837SChris Mason /* 25566567e837SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 25576567e837SChris Mason */ 25586567e837SChris Mason /* first correct the data pointers */ 25596567e837SChris Mason for (i = slot; i < nritems; i++) { 25605f39d397SChris Mason u32 ioff; 25615f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2562db94535dSChris Mason 2563db94535dSChris Mason if (!leaf->map_token) { 2564db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2565db94535dSChris Mason sizeof(struct btrfs_item), 2566db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2567db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2568db94535dSChris Mason KM_USER1); 2569db94535dSChris Mason } 25705f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 25715f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff - data_size); 25726567e837SChris Mason } 25735f39d397SChris Mason 2574db94535dSChris Mason if (leaf->map_token) { 2575db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2576db94535dSChris Mason leaf->map_token = NULL; 2577db94535dSChris Mason } 2578db94535dSChris Mason 25796567e837SChris Mason /* shift the data */ 25805f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 25816567e837SChris Mason data_end - data_size, btrfs_leaf_data(leaf) + 25826567e837SChris Mason data_end, old_data - data_end); 25835f39d397SChris Mason 25846567e837SChris Mason data_end = old_data; 25855f39d397SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 25865f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 25875f39d397SChris Mason btrfs_set_item_size(leaf, item, old_size + data_size); 25885f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 25896567e837SChris Mason 25906567e837SChris Mason ret = 0; 25915f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 25925f39d397SChris Mason btrfs_print_leaf(root, leaf); 25936567e837SChris Mason BUG(); 25945f39d397SChris Mason } 25956567e837SChris Mason return ret; 25966567e837SChris Mason } 25976567e837SChris Mason 259874123bd7SChris Mason /* 259974123bd7SChris Mason * Given a key and some data, insert an item into the tree. 260074123bd7SChris Mason * This does all the path init required, making room in the tree if needed. 260174123bd7SChris Mason */ 26029c58309dSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans, 26035f39d397SChris Mason struct btrfs_root *root, 26045f39d397SChris Mason struct btrfs_path *path, 26059c58309dSChris Mason struct btrfs_key *cpu_key, u32 *data_size, 26069c58309dSChris Mason int nr) 2607be0e5c09SChris Mason { 26085f39d397SChris Mason struct extent_buffer *leaf; 26095f39d397SChris Mason struct btrfs_item *item; 2610aa5d6bedSChris Mason int ret = 0; 2611be0e5c09SChris Mason int slot; 2612eb60ceacSChris Mason int slot_orig; 26139c58309dSChris Mason int i; 26147518a238SChris Mason u32 nritems; 26159c58309dSChris Mason u32 total_size = 0; 26169c58309dSChris Mason u32 total_data = 0; 2617be0e5c09SChris Mason unsigned int data_end; 2618e2fa7227SChris Mason struct btrfs_disk_key disk_key; 2619e2fa7227SChris Mason 26209c58309dSChris Mason for (i = 0; i < nr; i++) { 26219c58309dSChris Mason total_data += data_size[i]; 26229c58309dSChris Mason } 2623be0e5c09SChris Mason 26249c58309dSChris Mason total_size = total_data + (nr - 1) * sizeof(struct btrfs_item); 26259c58309dSChris Mason ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1); 2626eb60ceacSChris Mason if (ret == 0) { 2627f0930a37SChris Mason return -EEXIST; 2628aa5d6bedSChris Mason } 2629ed2ff2cbSChris Mason if (ret < 0) 2630ed2ff2cbSChris Mason goto out; 2631be0e5c09SChris Mason 263262e2749eSChris Mason slot_orig = path->slots[0]; 26335f39d397SChris Mason leaf = path->nodes[0]; 263474123bd7SChris Mason 26355f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2636123abc88SChris Mason data_end = leaf_data_end(root, leaf); 2637eb60ceacSChris Mason 2638123abc88SChris Mason if (btrfs_leaf_free_space(root, leaf) < 26399c58309dSChris Mason sizeof(struct btrfs_item) + total_size) { 26403326d1b0SChris Mason btrfs_print_leaf(root, leaf); 26413326d1b0SChris Mason printk("not enough freespace need %u have %d\n", 26429c58309dSChris Mason total_size, btrfs_leaf_free_space(root, leaf)); 2643be0e5c09SChris Mason BUG(); 2644d4dbff95SChris Mason } 26455f39d397SChris Mason 264662e2749eSChris Mason slot = path->slots[0]; 2647eb60ceacSChris Mason BUG_ON(slot < 0); 26485f39d397SChris Mason 2649be0e5c09SChris Mason if (slot != nritems) { 2650be0e5c09SChris Mason int i; 26515f39d397SChris Mason unsigned int old_data = btrfs_item_end_nr(leaf, slot); 2652be0e5c09SChris Mason 26535f39d397SChris Mason if (old_data < data_end) { 26545f39d397SChris Mason btrfs_print_leaf(root, leaf); 26555f39d397SChris Mason printk("slot %d old_data %d data_end %d\n", 26565f39d397SChris Mason slot, old_data, data_end); 26575f39d397SChris Mason BUG_ON(1); 26585f39d397SChris Mason } 2659be0e5c09SChris Mason /* 2660be0e5c09SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2661be0e5c09SChris Mason */ 2662be0e5c09SChris Mason /* first correct the data pointers */ 2663db94535dSChris Mason WARN_ON(leaf->map_token); 26640783fcfcSChris Mason for (i = slot; i < nritems; i++) { 26655f39d397SChris Mason u32 ioff; 2666db94535dSChris Mason 26675f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2668db94535dSChris Mason if (!leaf->map_token) { 2669db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2670db94535dSChris Mason sizeof(struct btrfs_item), 2671db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2672db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2673db94535dSChris Mason KM_USER1); 2674db94535dSChris Mason } 2675db94535dSChris Mason 26765f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 26779c58309dSChris Mason btrfs_set_item_offset(leaf, item, ioff - total_data); 26780783fcfcSChris Mason } 2679db94535dSChris Mason if (leaf->map_token) { 2680db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2681db94535dSChris Mason leaf->map_token = NULL; 2682db94535dSChris Mason } 2683be0e5c09SChris Mason 2684be0e5c09SChris Mason /* shift the items */ 26859c58309dSChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr), 26865f39d397SChris Mason btrfs_item_nr_offset(slot), 26870783fcfcSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 2688be0e5c09SChris Mason 2689be0e5c09SChris Mason /* shift the data */ 26905f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 26919c58309dSChris Mason data_end - total_data, btrfs_leaf_data(leaf) + 2692be0e5c09SChris Mason data_end, old_data - data_end); 2693be0e5c09SChris Mason data_end = old_data; 2694be0e5c09SChris Mason } 26955f39d397SChris Mason 269662e2749eSChris Mason /* setup the item for the new data */ 26979c58309dSChris Mason for (i = 0; i < nr; i++) { 26989c58309dSChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key + i); 26999c58309dSChris Mason btrfs_set_item_key(leaf, &disk_key, slot + i); 27009c58309dSChris Mason item = btrfs_item_nr(leaf, slot + i); 27019c58309dSChris Mason btrfs_set_item_offset(leaf, item, data_end - data_size[i]); 27029c58309dSChris Mason data_end -= data_size[i]; 27039c58309dSChris Mason btrfs_set_item_size(leaf, item, data_size[i]); 27049c58309dSChris Mason } 27059c58309dSChris Mason btrfs_set_header_nritems(leaf, nritems + nr); 27065f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2707aa5d6bedSChris Mason 2708aa5d6bedSChris Mason ret = 0; 27095a01a2e3SChris Mason if (slot == 0) { 27105a01a2e3SChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key); 2711e089f05cSChris Mason ret = fixup_low_keys(trans, root, path, &disk_key, 1); 27125a01a2e3SChris Mason } 2713aa5d6bedSChris Mason 27145f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 27155f39d397SChris Mason btrfs_print_leaf(root, leaf); 2716be0e5c09SChris Mason BUG(); 27175f39d397SChris Mason } 2718ed2ff2cbSChris Mason out: 271962e2749eSChris Mason return ret; 272062e2749eSChris Mason } 272162e2749eSChris Mason 272262e2749eSChris Mason /* 272362e2749eSChris Mason * Given a key and some data, insert an item into the tree. 272462e2749eSChris Mason * This does all the path init required, making room in the tree if needed. 272562e2749eSChris Mason */ 2726e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root 2727e089f05cSChris Mason *root, struct btrfs_key *cpu_key, void *data, u32 2728e089f05cSChris Mason data_size) 272962e2749eSChris Mason { 273062e2749eSChris Mason int ret = 0; 27312c90e5d6SChris Mason struct btrfs_path *path; 27325f39d397SChris Mason struct extent_buffer *leaf; 27335f39d397SChris Mason unsigned long ptr; 273462e2749eSChris Mason 27352c90e5d6SChris Mason path = btrfs_alloc_path(); 27362c90e5d6SChris Mason BUG_ON(!path); 27372c90e5d6SChris Mason ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size); 273862e2749eSChris Mason if (!ret) { 27395f39d397SChris Mason leaf = path->nodes[0]; 27405f39d397SChris Mason ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); 27415f39d397SChris Mason write_extent_buffer(leaf, data, ptr, data_size); 27425f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 274362e2749eSChris Mason } 27442c90e5d6SChris Mason btrfs_free_path(path); 2745aa5d6bedSChris Mason return ret; 2746be0e5c09SChris Mason } 2747be0e5c09SChris Mason 274874123bd7SChris Mason /* 27495de08d7dSChris Mason * delete the pointer from a given node. 275074123bd7SChris Mason * 275174123bd7SChris Mason * If the delete empties a node, the node is removed from the tree, 275274123bd7SChris Mason * continuing all the way the root if required. The root is converted into 275374123bd7SChris Mason * a leaf if all the nodes are emptied. 275474123bd7SChris Mason */ 2755e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 2756e089f05cSChris Mason struct btrfs_path *path, int level, int slot) 2757be0e5c09SChris Mason { 27585f39d397SChris Mason struct extent_buffer *parent = path->nodes[level]; 27597518a238SChris Mason u32 nritems; 2760aa5d6bedSChris Mason int ret = 0; 2761bb803951SChris Mason int wret; 2762be0e5c09SChris Mason 27635f39d397SChris Mason nritems = btrfs_header_nritems(parent); 2764be0e5c09SChris Mason if (slot != nritems -1) { 27655f39d397SChris Mason memmove_extent_buffer(parent, 27665f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 27675f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 2768d6025579SChris Mason sizeof(struct btrfs_key_ptr) * 2769d6025579SChris Mason (nritems - slot - 1)); 2770be0e5c09SChris Mason } 27717518a238SChris Mason nritems--; 27725f39d397SChris Mason btrfs_set_header_nritems(parent, nritems); 27737518a238SChris Mason if (nritems == 0 && parent == root->node) { 27745f39d397SChris Mason BUG_ON(btrfs_header_level(root->node) != 1); 2775eb60ceacSChris Mason /* just turn the root into a leaf and break */ 27765f39d397SChris Mason btrfs_set_header_level(root->node, 0); 2777bb803951SChris Mason } else if (slot == 0) { 27785f39d397SChris Mason struct btrfs_disk_key disk_key; 27795f39d397SChris Mason 27805f39d397SChris Mason btrfs_node_key(parent, &disk_key, 0); 27815f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, level + 1); 27820f70abe2SChris Mason if (wret) 27830f70abe2SChris Mason ret = wret; 2784be0e5c09SChris Mason } 2785d6025579SChris Mason btrfs_mark_buffer_dirty(parent); 2786aa5d6bedSChris Mason return ret; 2787be0e5c09SChris Mason } 2788be0e5c09SChris Mason 278974123bd7SChris Mason /* 279074123bd7SChris Mason * delete the item at the leaf level in path. If that empties 279174123bd7SChris Mason * the leaf, remove it from the tree 279274123bd7SChris Mason */ 279385e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root, 279485e21bacSChris Mason struct btrfs_path *path, int slot, int nr) 2795be0e5c09SChris Mason { 27965f39d397SChris Mason struct extent_buffer *leaf; 27975f39d397SChris Mason struct btrfs_item *item; 279885e21bacSChris Mason int last_off; 279985e21bacSChris Mason int dsize = 0; 2800aa5d6bedSChris Mason int ret = 0; 2801aa5d6bedSChris Mason int wret; 280285e21bacSChris Mason int i; 28037518a238SChris Mason u32 nritems; 2804be0e5c09SChris Mason 28055f39d397SChris Mason leaf = path->nodes[0]; 280685e21bacSChris Mason last_off = btrfs_item_offset_nr(leaf, slot + nr - 1); 280785e21bacSChris Mason 280885e21bacSChris Mason for (i = 0; i < nr; i++) 280985e21bacSChris Mason dsize += btrfs_item_size_nr(leaf, slot + i); 281085e21bacSChris Mason 28115f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2812be0e5c09SChris Mason 281385e21bacSChris Mason if (slot + nr != nritems) { 2814be0e5c09SChris Mason int i; 2815123abc88SChris Mason int data_end = leaf_data_end(root, leaf); 28165f39d397SChris Mason 28175f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2818d6025579SChris Mason data_end + dsize, 2819123abc88SChris Mason btrfs_leaf_data(leaf) + data_end, 282085e21bacSChris Mason last_off - data_end); 28215f39d397SChris Mason 282285e21bacSChris Mason for (i = slot + nr; i < nritems; i++) { 28235f39d397SChris Mason u32 ioff; 2824db94535dSChris Mason 28255f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2826db94535dSChris Mason if (!leaf->map_token) { 2827db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2828db94535dSChris Mason sizeof(struct btrfs_item), 2829db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2830db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2831db94535dSChris Mason KM_USER1); 2832db94535dSChris Mason } 28335f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 28345f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + dsize); 28350783fcfcSChris Mason } 2836db94535dSChris Mason 2837db94535dSChris Mason if (leaf->map_token) { 2838db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2839db94535dSChris Mason leaf->map_token = NULL; 2840db94535dSChris Mason } 2841db94535dSChris Mason 28425f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot), 284385e21bacSChris Mason btrfs_item_nr_offset(slot + nr), 28440783fcfcSChris Mason sizeof(struct btrfs_item) * 284585e21bacSChris Mason (nritems - slot - nr)); 2846be0e5c09SChris Mason } 284785e21bacSChris Mason btrfs_set_header_nritems(leaf, nritems - nr); 284885e21bacSChris Mason nritems -= nr; 28495f39d397SChris Mason 285074123bd7SChris Mason /* delete the leaf if we've emptied it */ 28517518a238SChris Mason if (nritems == 0) { 28525f39d397SChris Mason if (leaf == root->node) { 28535f39d397SChris Mason btrfs_set_header_level(leaf, 0); 28549a8dd150SChris Mason } else { 28557bb86316SChris Mason u64 root_gen = btrfs_header_generation(path->nodes[1]); 2856e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, path->slots[1]); 2857aa5d6bedSChris Mason if (wret) 2858aa5d6bedSChris Mason ret = wret; 2859e089f05cSChris Mason wret = btrfs_free_extent(trans, root, 28607bb86316SChris Mason leaf->start, leaf->len, 28617bb86316SChris Mason btrfs_header_owner(path->nodes[1]), 28627bb86316SChris Mason root_gen, 0, 0, 1); 28630f70abe2SChris Mason if (wret) 28640f70abe2SChris Mason ret = wret; 28659a8dd150SChris Mason } 2866be0e5c09SChris Mason } else { 28677518a238SChris Mason int used = leaf_space_used(leaf, 0, nritems); 2868aa5d6bedSChris Mason if (slot == 0) { 28695f39d397SChris Mason struct btrfs_disk_key disk_key; 28705f39d397SChris Mason 28715f39d397SChris Mason btrfs_item_key(leaf, &disk_key, 0); 2872e089f05cSChris Mason wret = fixup_low_keys(trans, root, path, 28735f39d397SChris Mason &disk_key, 1); 2874aa5d6bedSChris Mason if (wret) 2875aa5d6bedSChris Mason ret = wret; 2876aa5d6bedSChris Mason } 2877aa5d6bedSChris Mason 287874123bd7SChris Mason /* delete the leaf if it is mostly empty */ 287985e21bacSChris Mason if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) { 2880be0e5c09SChris Mason /* push_leaf_left fixes the path. 2881be0e5c09SChris Mason * make sure the path still points to our leaf 2882be0e5c09SChris Mason * for possible call to del_ptr below 2883be0e5c09SChris Mason */ 28844920c9acSChris Mason slot = path->slots[1]; 28855f39d397SChris Mason extent_buffer_get(leaf); 28865f39d397SChris Mason 288785e21bacSChris Mason wret = push_leaf_left(trans, root, path, 1, 1); 288854aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2889aa5d6bedSChris Mason ret = wret; 28905f39d397SChris Mason 28915f39d397SChris Mason if (path->nodes[0] == leaf && 28925f39d397SChris Mason btrfs_header_nritems(leaf)) { 289385e21bacSChris Mason wret = push_leaf_right(trans, root, path, 1, 1); 289454aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2895aa5d6bedSChris Mason ret = wret; 2896aa5d6bedSChris Mason } 28975f39d397SChris Mason 28985f39d397SChris Mason if (btrfs_header_nritems(leaf) == 0) { 28997bb86316SChris Mason u64 root_gen; 2900db94535dSChris Mason u64 bytenr = leaf->start; 2901db94535dSChris Mason u32 blocksize = leaf->len; 29025f39d397SChris Mason 29037bb86316SChris Mason root_gen = btrfs_header_generation( 29047bb86316SChris Mason path->nodes[1]); 29057bb86316SChris Mason 2906e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, slot); 2907aa5d6bedSChris Mason if (wret) 2908aa5d6bedSChris Mason ret = wret; 29095f39d397SChris Mason 29105f39d397SChris Mason free_extent_buffer(leaf); 2911db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 29127bb86316SChris Mason blocksize, 29137bb86316SChris Mason btrfs_header_owner(path->nodes[1]), 29147bb86316SChris Mason root_gen, 0, 0, 1); 29150f70abe2SChris Mason if (wret) 29160f70abe2SChris Mason ret = wret; 29175de08d7dSChris Mason } else { 2918925baeddSChris Mason /* if we're still in the path, make sure 2919925baeddSChris Mason * we're dirty. Otherwise, one of the 2920925baeddSChris Mason * push_leaf functions must have already 2921925baeddSChris Mason * dirtied this buffer 2922925baeddSChris Mason */ 2923925baeddSChris Mason if (path->nodes[0] == leaf) 29245f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 29255f39d397SChris Mason free_extent_buffer(leaf); 2926be0e5c09SChris Mason } 2927d5719762SChris Mason } else { 29285f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2929be0e5c09SChris Mason } 29309a8dd150SChris Mason } 2931aa5d6bedSChris Mason return ret; 29329a8dd150SChris Mason } 29339a8dd150SChris Mason 293497571fd0SChris Mason /* 2935925baeddSChris Mason * search the tree again to find a leaf with lesser keys 29367bb86316SChris Mason * returns 0 if it found something or 1 if there are no lesser leaves. 29377bb86316SChris Mason * returns < 0 on io errors. 29387bb86316SChris Mason */ 29397bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path) 29407bb86316SChris Mason { 2941925baeddSChris Mason struct btrfs_key key; 2942925baeddSChris Mason struct btrfs_disk_key found_key; 2943925baeddSChris Mason int ret; 29447bb86316SChris Mason 2945925baeddSChris Mason btrfs_item_key_to_cpu(path->nodes[0], &key, 0); 2946925baeddSChris Mason 2947925baeddSChris Mason if (key.offset > 0) 2948925baeddSChris Mason key.offset--; 2949925baeddSChris Mason else if (key.type > 0) 2950925baeddSChris Mason key.type--; 2951925baeddSChris Mason else if (key.objectid > 0) 2952925baeddSChris Mason key.objectid--; 2953925baeddSChris Mason else 29547bb86316SChris Mason return 1; 29557bb86316SChris Mason 2956925baeddSChris Mason btrfs_release_path(root, path); 2957925baeddSChris Mason ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); 2958925baeddSChris Mason if (ret < 0) 2959925baeddSChris Mason return ret; 2960925baeddSChris Mason btrfs_item_key(path->nodes[0], &found_key, 0); 2961925baeddSChris Mason ret = comp_keys(&found_key, &key); 2962925baeddSChris Mason if (ret < 0) 29637bb86316SChris Mason return 0; 2964925baeddSChris Mason return 1; 29657bb86316SChris Mason } 29667bb86316SChris Mason 2967*3f157a2fSChris Mason /* 2968*3f157a2fSChris Mason * A helper function to walk down the tree starting at min_key, and looking 2969*3f157a2fSChris Mason * for nodes or leaves that are either in cache or have a minimum 2970*3f157a2fSChris Mason * transaction id. This is used by the btree defrag code, but could 2971*3f157a2fSChris Mason * also be used to search for blocks that have changed since a given 2972*3f157a2fSChris Mason * transaction id. 2973*3f157a2fSChris Mason * 2974*3f157a2fSChris Mason * This does not cow, but it does stuff the starting key it finds back 2975*3f157a2fSChris Mason * into min_key, so you can call btrfs_search_slot with cow=1 on the 2976*3f157a2fSChris Mason * key and get a writable path. 2977*3f157a2fSChris Mason * 2978*3f157a2fSChris Mason * This does lock as it descends, and path->keep_locks should be set 2979*3f157a2fSChris Mason * to 1 by the caller. 2980*3f157a2fSChris Mason * 2981*3f157a2fSChris Mason * This honors path->lowest_level to prevent descent past a given level 2982*3f157a2fSChris Mason * of the tree. 2983*3f157a2fSChris Mason * 2984*3f157a2fSChris Mason * returns zero if something useful was found, < 0 on error and 1 if there 2985*3f157a2fSChris Mason * was nothing in the tree that matched the search criteria. 2986*3f157a2fSChris Mason */ 2987*3f157a2fSChris Mason int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key, 2988*3f157a2fSChris Mason struct btrfs_path *path, int cache_only, 2989*3f157a2fSChris Mason u64 min_trans) 2990*3f157a2fSChris Mason { 2991*3f157a2fSChris Mason struct extent_buffer *cur; 2992*3f157a2fSChris Mason struct btrfs_key found_key; 2993*3f157a2fSChris Mason int slot; 2994*3f157a2fSChris Mason u32 nritems; 2995*3f157a2fSChris Mason int level; 2996*3f157a2fSChris Mason int ret = 1; 2997*3f157a2fSChris Mason 2998*3f157a2fSChris Mason again: 2999*3f157a2fSChris Mason cur = btrfs_lock_root_node(root); 3000*3f157a2fSChris Mason level = btrfs_header_level(cur); 3001*3f157a2fSChris Mason path->nodes[level] = cur; 3002*3f157a2fSChris Mason path->locks[level] = 1; 3003*3f157a2fSChris Mason 3004*3f157a2fSChris Mason if (btrfs_header_generation(cur) < min_trans) { 3005*3f157a2fSChris Mason ret = 1; 3006*3f157a2fSChris Mason goto out; 3007*3f157a2fSChris Mason } 3008*3f157a2fSChris Mason while(1) { 3009*3f157a2fSChris Mason nritems = btrfs_header_nritems(cur); 3010*3f157a2fSChris Mason level = btrfs_header_level(cur); 3011*3f157a2fSChris Mason bin_search(cur, min_key, level, &slot); 3012*3f157a2fSChris Mason 3013*3f157a2fSChris Mason /* at level = 0, we're done, setup the path and exit */ 3014*3f157a2fSChris Mason if (level == 0) { 3015*3f157a2fSChris Mason ret = 0; 3016*3f157a2fSChris Mason path->slots[level] = slot; 3017*3f157a2fSChris Mason btrfs_item_key_to_cpu(cur, &found_key, slot); 3018*3f157a2fSChris Mason goto out; 3019*3f157a2fSChris Mason } 3020*3f157a2fSChris Mason /* 3021*3f157a2fSChris Mason * check this node pointer against the cache_only and 3022*3f157a2fSChris Mason * min_trans parameters. If it isn't in cache or is too 3023*3f157a2fSChris Mason * old, skip to the next one. 3024*3f157a2fSChris Mason */ 3025*3f157a2fSChris Mason while(slot < nritems) { 3026*3f157a2fSChris Mason u64 blockptr; 3027*3f157a2fSChris Mason u64 gen; 3028*3f157a2fSChris Mason struct extent_buffer *tmp; 3029*3f157a2fSChris Mason blockptr = btrfs_node_blockptr(cur, slot); 3030*3f157a2fSChris Mason gen = btrfs_node_ptr_generation(cur, slot); 3031*3f157a2fSChris Mason if (gen < min_trans) { 3032*3f157a2fSChris Mason slot++; 3033*3f157a2fSChris Mason continue; 3034*3f157a2fSChris Mason } 3035*3f157a2fSChris Mason if (!cache_only) 3036*3f157a2fSChris Mason break; 3037*3f157a2fSChris Mason 3038*3f157a2fSChris Mason tmp = btrfs_find_tree_block(root, blockptr, 3039*3f157a2fSChris Mason btrfs_level_size(root, level - 1)); 3040*3f157a2fSChris Mason 3041*3f157a2fSChris Mason if (tmp && btrfs_buffer_uptodate(tmp, gen)) { 3042*3f157a2fSChris Mason free_extent_buffer(tmp); 3043*3f157a2fSChris Mason break; 3044*3f157a2fSChris Mason } 3045*3f157a2fSChris Mason if (tmp) 3046*3f157a2fSChris Mason free_extent_buffer(tmp); 3047*3f157a2fSChris Mason slot++; 3048*3f157a2fSChris Mason } 3049*3f157a2fSChris Mason /* 3050*3f157a2fSChris Mason * we didn't find a candidate key in this node, walk forward 3051*3f157a2fSChris Mason * and find another one 3052*3f157a2fSChris Mason */ 3053*3f157a2fSChris Mason if (slot >= nritems) { 3054*3f157a2fSChris Mason ret = btrfs_find_next_key(root, path, min_key, level, 3055*3f157a2fSChris Mason cache_only, min_trans); 3056*3f157a2fSChris Mason if (ret == 0) { 3057*3f157a2fSChris Mason btrfs_release_path(root, path); 3058*3f157a2fSChris Mason goto again; 3059*3f157a2fSChris Mason } else { 3060*3f157a2fSChris Mason goto out; 3061*3f157a2fSChris Mason } 3062*3f157a2fSChris Mason } 3063*3f157a2fSChris Mason /* save our key for returning back */ 3064*3f157a2fSChris Mason btrfs_node_key_to_cpu(cur, &found_key, slot); 3065*3f157a2fSChris Mason path->slots[level] = slot; 3066*3f157a2fSChris Mason if (level == path->lowest_level) { 3067*3f157a2fSChris Mason ret = 0; 3068*3f157a2fSChris Mason unlock_up(path, level, 1); 3069*3f157a2fSChris Mason goto out; 3070*3f157a2fSChris Mason } 3071*3f157a2fSChris Mason cur = read_node_slot(root, cur, slot); 3072*3f157a2fSChris Mason 3073*3f157a2fSChris Mason btrfs_tree_lock(cur); 3074*3f157a2fSChris Mason path->locks[level - 1] = 1; 3075*3f157a2fSChris Mason path->nodes[level - 1] = cur; 3076*3f157a2fSChris Mason unlock_up(path, level, 1); 3077*3f157a2fSChris Mason } 3078*3f157a2fSChris Mason out: 3079*3f157a2fSChris Mason if (ret == 0) 3080*3f157a2fSChris Mason memcpy(min_key, &found_key, sizeof(found_key)); 3081*3f157a2fSChris Mason return ret; 3082*3f157a2fSChris Mason } 3083*3f157a2fSChris Mason 3084*3f157a2fSChris Mason /* 3085*3f157a2fSChris Mason * this is similar to btrfs_next_leaf, but does not try to preserve 3086*3f157a2fSChris Mason * and fixup the path. It looks for and returns the next key in the 3087*3f157a2fSChris Mason * tree based on the current path and the cache_only and min_trans 3088*3f157a2fSChris Mason * parameters. 3089*3f157a2fSChris Mason * 3090*3f157a2fSChris Mason * 0 is returned if another key is found, < 0 if there are any errors 3091*3f157a2fSChris Mason * and 1 is returned if there are no higher keys in the tree 3092*3f157a2fSChris Mason * 3093*3f157a2fSChris Mason * path->keep_locks should be set to 1 on the search made before 3094*3f157a2fSChris Mason * calling this function. 3095*3f157a2fSChris Mason */ 3096e7a84565SChris Mason int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path, 3097*3f157a2fSChris Mason struct btrfs_key *key, int lowest_level, 3098*3f157a2fSChris Mason int cache_only, u64 min_trans) 3099e7a84565SChris Mason { 3100e7a84565SChris Mason int level = lowest_level; 3101e7a84565SChris Mason int slot; 3102e7a84565SChris Mason struct extent_buffer *c; 3103e7a84565SChris Mason 3104e7a84565SChris Mason while(level < BTRFS_MAX_LEVEL) { 3105e7a84565SChris Mason if (!path->nodes[level]) 3106e7a84565SChris Mason return 1; 3107e7a84565SChris Mason 3108e7a84565SChris Mason slot = path->slots[level] + 1; 3109e7a84565SChris Mason c = path->nodes[level]; 3110*3f157a2fSChris Mason next: 3111e7a84565SChris Mason if (slot >= btrfs_header_nritems(c)) { 3112e7a84565SChris Mason level++; 3113e7a84565SChris Mason if (level == BTRFS_MAX_LEVEL) { 3114e7a84565SChris Mason return 1; 3115e7a84565SChris Mason } 3116e7a84565SChris Mason continue; 3117e7a84565SChris Mason } 3118e7a84565SChris Mason if (level == 0) 3119e7a84565SChris Mason btrfs_item_key_to_cpu(c, key, slot); 3120*3f157a2fSChris Mason else { 3121*3f157a2fSChris Mason u64 blockptr = btrfs_node_blockptr(c, slot); 3122*3f157a2fSChris Mason u64 gen = btrfs_node_ptr_generation(c, slot); 3123*3f157a2fSChris Mason 3124*3f157a2fSChris Mason if (cache_only) { 3125*3f157a2fSChris Mason struct extent_buffer *cur; 3126*3f157a2fSChris Mason cur = btrfs_find_tree_block(root, blockptr, 3127*3f157a2fSChris Mason btrfs_level_size(root, level - 1)); 3128*3f157a2fSChris Mason if (!cur || !btrfs_buffer_uptodate(cur, gen)) { 3129*3f157a2fSChris Mason slot++; 3130*3f157a2fSChris Mason if (cur) 3131*3f157a2fSChris Mason free_extent_buffer(cur); 3132*3f157a2fSChris Mason goto next; 3133*3f157a2fSChris Mason } 3134*3f157a2fSChris Mason free_extent_buffer(cur); 3135*3f157a2fSChris Mason } 3136*3f157a2fSChris Mason if (gen < min_trans) { 3137*3f157a2fSChris Mason slot++; 3138*3f157a2fSChris Mason goto next; 3139*3f157a2fSChris Mason } 3140e7a84565SChris Mason btrfs_node_key_to_cpu(c, key, slot); 3141*3f157a2fSChris Mason } 3142e7a84565SChris Mason return 0; 3143e7a84565SChris Mason } 3144e7a84565SChris Mason return 1; 3145e7a84565SChris Mason } 3146e7a84565SChris Mason 31477bb86316SChris Mason /* 3148925baeddSChris Mason * search the tree again to find a leaf with greater keys 31490f70abe2SChris Mason * returns 0 if it found something or 1 if there are no greater leaves. 31500f70abe2SChris Mason * returns < 0 on io errors. 315197571fd0SChris Mason */ 3152234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path) 3153d97e63b6SChris Mason { 3154d97e63b6SChris Mason int slot; 3155d97e63b6SChris Mason int level = 1; 31565f39d397SChris Mason struct extent_buffer *c; 31575f39d397SChris Mason struct extent_buffer *next = NULL; 3158925baeddSChris Mason struct btrfs_key key; 3159925baeddSChris Mason u32 nritems; 3160925baeddSChris Mason int ret; 3161925baeddSChris Mason 3162925baeddSChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 3163925baeddSChris Mason if (nritems == 0) { 3164925baeddSChris Mason return 1; 3165925baeddSChris Mason } 3166925baeddSChris Mason 3167925baeddSChris Mason btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1); 3168925baeddSChris Mason 3169925baeddSChris Mason btrfs_release_path(root, path); 3170a2135011SChris Mason path->keep_locks = 1; 3171925baeddSChris Mason ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); 3172925baeddSChris Mason path->keep_locks = 0; 3173925baeddSChris Mason 3174925baeddSChris Mason if (ret < 0) 3175925baeddSChris Mason return ret; 3176925baeddSChris Mason 3177a2135011SChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 3178168fd7d2SChris Mason /* 3179168fd7d2SChris Mason * by releasing the path above we dropped all our locks. A balance 3180168fd7d2SChris Mason * could have added more items next to the key that used to be 3181168fd7d2SChris Mason * at the very end of the block. So, check again here and 3182168fd7d2SChris Mason * advance the path if there are now more items available. 3183168fd7d2SChris Mason */ 3184a2135011SChris Mason if (nritems > 0 && path->slots[0] < nritems - 1) { 3185168fd7d2SChris Mason path->slots[0]++; 3186925baeddSChris Mason goto done; 3187925baeddSChris Mason } 3188d97e63b6SChris Mason 3189234b63a0SChris Mason while(level < BTRFS_MAX_LEVEL) { 3190d97e63b6SChris Mason if (!path->nodes[level]) 31910f70abe2SChris Mason return 1; 31925f39d397SChris Mason 3193d97e63b6SChris Mason slot = path->slots[level] + 1; 3194d97e63b6SChris Mason c = path->nodes[level]; 31955f39d397SChris Mason if (slot >= btrfs_header_nritems(c)) { 3196d97e63b6SChris Mason level++; 3197925baeddSChris Mason if (level == BTRFS_MAX_LEVEL) { 31987bb86316SChris Mason return 1; 3199925baeddSChris Mason } 3200d97e63b6SChris Mason continue; 3201d97e63b6SChris Mason } 32025f39d397SChris Mason 3203925baeddSChris Mason if (next) { 3204925baeddSChris Mason btrfs_tree_unlock(next); 32055f39d397SChris Mason free_extent_buffer(next); 3206925baeddSChris Mason } 32075f39d397SChris Mason 3208925baeddSChris Mason if (level == 1 && path->locks[1] && path->reada) 320901f46658SChris Mason reada_for_search(root, path, level, slot, 0); 32105f39d397SChris Mason 3211ca7a79adSChris Mason next = read_node_slot(root, c, slot); 32125cd57b2cSChris Mason if (!path->skip_locking) { 3213051e1b9fSChris Mason WARN_ON(!btrfs_tree_locked(c)); 3214925baeddSChris Mason btrfs_tree_lock(next); 32155cd57b2cSChris Mason } 3216d97e63b6SChris Mason break; 3217d97e63b6SChris Mason } 3218d97e63b6SChris Mason path->slots[level] = slot; 3219d97e63b6SChris Mason while(1) { 3220d97e63b6SChris Mason level--; 3221d97e63b6SChris Mason c = path->nodes[level]; 3222925baeddSChris Mason if (path->locks[level]) 3223925baeddSChris Mason btrfs_tree_unlock(c); 32245f39d397SChris Mason free_extent_buffer(c); 3225d97e63b6SChris Mason path->nodes[level] = next; 3226d97e63b6SChris Mason path->slots[level] = 0; 3227a74a4b97SChris Mason if (!path->skip_locking) 3228925baeddSChris Mason path->locks[level] = 1; 3229d97e63b6SChris Mason if (!level) 3230d97e63b6SChris Mason break; 3231925baeddSChris Mason if (level == 1 && path->locks[1] && path->reada) 3232925baeddSChris Mason reada_for_search(root, path, level, slot, 0); 3233ca7a79adSChris Mason next = read_node_slot(root, next, 0); 32345cd57b2cSChris Mason if (!path->skip_locking) { 3235a2135011SChris Mason WARN_ON(!btrfs_tree_locked(path->nodes[level])); 3236925baeddSChris Mason btrfs_tree_lock(next); 3237d97e63b6SChris Mason } 32385cd57b2cSChris Mason } 3239925baeddSChris Mason done: 3240925baeddSChris Mason unlock_up(path, 0, 1); 3241d97e63b6SChris Mason return 0; 3242d97e63b6SChris Mason } 32430b86a832SChris Mason 3244*3f157a2fSChris Mason /* 3245*3f157a2fSChris Mason * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps 3246*3f157a2fSChris Mason * searching until it gets past min_objectid or finds an item of 'type' 3247*3f157a2fSChris Mason * 3248*3f157a2fSChris Mason * returns 0 if something is found, 1 if nothing was found and < 0 on error 3249*3f157a2fSChris Mason */ 32500b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root, 32510b86a832SChris Mason struct btrfs_path *path, u64 min_objectid, 32520b86a832SChris Mason int type) 32530b86a832SChris Mason { 32540b86a832SChris Mason struct btrfs_key found_key; 32550b86a832SChris Mason struct extent_buffer *leaf; 32560b86a832SChris Mason int ret; 32570b86a832SChris Mason 32580b86a832SChris Mason while(1) { 32590b86a832SChris Mason if (path->slots[0] == 0) { 32600b86a832SChris Mason ret = btrfs_prev_leaf(root, path); 32610b86a832SChris Mason if (ret != 0) 32620b86a832SChris Mason return ret; 32630b86a832SChris Mason } else { 32640b86a832SChris Mason path->slots[0]--; 32650b86a832SChris Mason } 32660b86a832SChris Mason leaf = path->nodes[0]; 32670b86a832SChris Mason btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); 32680b86a832SChris Mason if (found_key.type == type) 32690b86a832SChris Mason return 0; 32700b86a832SChris Mason } 32710b86a832SChris Mason return 1; 32720b86a832SChris Mason } 32730b86a832SChris Mason 3274