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++) { 683f157a2fSChris 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]); 763f157a2fSChris 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 if (ins_len < 0) 1255925baeddSChris Mason lowest_unlock = 2; 1256bb803951SChris Mason again: 12575cd57b2cSChris Mason if (p->skip_locking) 12585cd57b2cSChris Mason b = btrfs_root_node(root); 12595cd57b2cSChris Mason else 1260925baeddSChris Mason b = btrfs_lock_root_node(root); 1261925baeddSChris Mason 1262eb60ceacSChris Mason while (b) { 12635f39d397SChris Mason level = btrfs_header_level(b); 126402217ed2SChris Mason if (cow) { 126502217ed2SChris Mason int wret; 1266e20d96d6SChris Mason wret = btrfs_cow_block(trans, root, b, 1267e20d96d6SChris Mason p->nodes[level + 1], 1268e20d96d6SChris Mason p->slots[level + 1], 1269252c38f0SYan &b); 127054aa1f4dSChris Mason if (wret) { 12715f39d397SChris Mason free_extent_buffer(b); 127254aa1f4dSChris Mason return wret; 127354aa1f4dSChris Mason } 127402217ed2SChris Mason } 127502217ed2SChris Mason BUG_ON(!cow && ins_len); 12765f39d397SChris Mason if (level != btrfs_header_level(b)) 12772c90e5d6SChris Mason WARN_ON(1); 12785f39d397SChris Mason level = btrfs_header_level(b); 1279eb60ceacSChris Mason p->nodes[level] = b; 12805cd57b2cSChris Mason if (!p->skip_locking) 1281925baeddSChris Mason p->locks[level] = 1; 1282123abc88SChris Mason ret = check_block(root, p, level); 1283aa5d6bedSChris Mason if (ret) 1284aa5d6bedSChris Mason return -1; 1285925baeddSChris Mason 12865f39d397SChris Mason ret = bin_search(b, key, level, &slot); 12875f39d397SChris Mason if (level != 0) { 1288be0e5c09SChris Mason if (ret && slot > 0) 1289be0e5c09SChris Mason slot -= 1; 1290be0e5c09SChris Mason p->slots[level] = slot; 12915f39d397SChris Mason if (ins_len > 0 && btrfs_header_nritems(b) >= 12921514794eSChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) { 1293e089f05cSChris Mason int sret = split_node(trans, root, p, level); 12945c680ed6SChris Mason BUG_ON(sret > 0); 12955c680ed6SChris Mason if (sret) 12965c680ed6SChris Mason return sret; 12975c680ed6SChris Mason b = p->nodes[level]; 12985c680ed6SChris Mason slot = p->slots[level]; 1299bb803951SChris Mason } else if (ins_len < 0) { 1300e089f05cSChris Mason int sret = balance_level(trans, root, p, 1301e089f05cSChris Mason level); 1302bb803951SChris Mason if (sret) 1303bb803951SChris Mason return sret; 1304bb803951SChris Mason b = p->nodes[level]; 1305f510cfecSChris Mason if (!b) { 1306f510cfecSChris Mason btrfs_release_path(NULL, p); 1307bb803951SChris Mason goto again; 1308f510cfecSChris Mason } 1309bb803951SChris Mason slot = p->slots[level]; 13105f39d397SChris Mason BUG_ON(btrfs_header_nritems(b) == 1); 13115c680ed6SChris Mason } 1312f9efa9c7SChris Mason unlock_up(p, level, lowest_unlock); 1313f9efa9c7SChris Mason 13149f3a7427SChris Mason /* this is only true while dropping a snapshot */ 1315925baeddSChris Mason if (level == lowest_level) { 13169f3a7427SChris Mason break; 1317925baeddSChris Mason } 1318ca7a79adSChris Mason 1319594a24ebSChris Mason blocknr = btrfs_node_blockptr(b, slot); 1320594a24ebSChris Mason gen = btrfs_node_ptr_generation(b, slot); 1321594a24ebSChris Mason blocksize = btrfs_level_size(root, level - 1); 1322594a24ebSChris Mason 1323594a24ebSChris Mason tmp = btrfs_find_tree_block(root, blocknr, blocksize); 1324594a24ebSChris Mason if (tmp && btrfs_buffer_uptodate(tmp, gen)) { 1325051e1b9fSChris Mason b = tmp; 1326051e1b9fSChris Mason } else { 1327051e1b9fSChris Mason /* 1328051e1b9fSChris Mason * reduce lock contention at high levels 1329051e1b9fSChris Mason * of the btree by dropping locks before 1330051e1b9fSChris Mason * we read. 1331051e1b9fSChris Mason */ 1332051e1b9fSChris Mason if (level > 1) { 1333051e1b9fSChris Mason btrfs_release_path(NULL, p); 1334051e1b9fSChris Mason if (tmp) 1335051e1b9fSChris Mason free_extent_buffer(tmp); 1336f9efa9c7SChris Mason if (should_reada) 1337f9efa9c7SChris Mason reada_for_search(root, p, 1338f9efa9c7SChris Mason level, slot, 1339f9efa9c7SChris Mason key->objectid); 1340f9efa9c7SChris Mason 1341594a24ebSChris Mason tmp = read_tree_block(root, blocknr, 1342594a24ebSChris Mason blocksize, gen); 1343594a24ebSChris Mason if (tmp) 1344594a24ebSChris Mason free_extent_buffer(tmp); 1345051e1b9fSChris Mason goto again; 1346051e1b9fSChris Mason } else { 1347a74a4b97SChris Mason if (tmp) 1348a74a4b97SChris Mason free_extent_buffer(tmp); 1349f9efa9c7SChris Mason if (should_reada) 1350f9efa9c7SChris Mason reada_for_search(root, p, 1351f9efa9c7SChris Mason level, slot, 1352f9efa9c7SChris Mason key->objectid); 1353ca7a79adSChris Mason b = read_node_slot(root, b, slot); 1354051e1b9fSChris Mason } 1355051e1b9fSChris Mason } 13565cd57b2cSChris Mason if (!p->skip_locking) 1357925baeddSChris Mason btrfs_tree_lock(b); 1358be0e5c09SChris Mason } else { 1359be0e5c09SChris Mason p->slots[level] = slot; 13605f39d397SChris Mason if (ins_len > 0 && btrfs_leaf_free_space(root, b) < 13610783fcfcSChris Mason sizeof(struct btrfs_item) + ins_len) { 1362d4dbff95SChris Mason int sret = split_leaf(trans, root, key, 1363cc0c5538SChris Mason p, ins_len, ret == 0); 13645c680ed6SChris Mason BUG_ON(sret > 0); 13655c680ed6SChris Mason if (sret) 13665c680ed6SChris Mason return sret; 13675c680ed6SChris Mason } 1368925baeddSChris Mason unlock_up(p, level, lowest_unlock); 1369be0e5c09SChris Mason return ret; 1370be0e5c09SChris Mason } 1371be0e5c09SChris Mason } 1372aa5d6bedSChris Mason return 1; 1373be0e5c09SChris Mason } 1374be0e5c09SChris Mason 137574123bd7SChris Mason /* 137674123bd7SChris Mason * adjust the pointers going up the tree, starting at level 137774123bd7SChris Mason * making sure the right key of each node is points to 'key'. 137874123bd7SChris Mason * This is used after shifting pointers to the left, so it stops 137974123bd7SChris Mason * fixing up pointers when a given leaf/node is not in slot 0 of the 138074123bd7SChris Mason * higher levels 1381aa5d6bedSChris Mason * 1382aa5d6bedSChris Mason * If this fails to write a tree block, it returns -1, but continues 1383aa5d6bedSChris Mason * fixing up the blocks in ram so the tree is consistent. 138474123bd7SChris Mason */ 13855f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans, 13865f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 13875f39d397SChris Mason struct btrfs_disk_key *key, int level) 1388be0e5c09SChris Mason { 1389be0e5c09SChris Mason int i; 1390aa5d6bedSChris Mason int ret = 0; 13915f39d397SChris Mason struct extent_buffer *t; 13925f39d397SChris Mason 1393234b63a0SChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 1394be0e5c09SChris Mason int tslot = path->slots[i]; 1395eb60ceacSChris Mason if (!path->nodes[i]) 1396be0e5c09SChris Mason break; 13975f39d397SChris Mason t = path->nodes[i]; 13985f39d397SChris Mason btrfs_set_node_key(t, key, tslot); 1399d6025579SChris Mason btrfs_mark_buffer_dirty(path->nodes[i]); 1400be0e5c09SChris Mason if (tslot != 0) 1401be0e5c09SChris Mason break; 1402be0e5c09SChris Mason } 1403aa5d6bedSChris Mason return ret; 1404be0e5c09SChris Mason } 1405be0e5c09SChris Mason 140674123bd7SChris Mason /* 140774123bd7SChris Mason * try to push data from one node into the next node left in the 140879f95c82SChris Mason * tree. 1409aa5d6bedSChris Mason * 1410aa5d6bedSChris Mason * returns 0 if some ptrs were pushed left, < 0 if there was some horrible 1411aa5d6bedSChris Mason * error, and > 0 if there was no room in the left hand block. 141274123bd7SChris Mason */ 141398ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans, 141498ed5174SChris Mason struct btrfs_root *root, struct extent_buffer *dst, 1415971a1f66SChris Mason struct extent_buffer *src, int empty) 1416be0e5c09SChris Mason { 1417be0e5c09SChris Mason int push_items = 0; 1418bb803951SChris Mason int src_nritems; 1419bb803951SChris Mason int dst_nritems; 1420aa5d6bedSChris Mason int ret = 0; 1421be0e5c09SChris Mason 14225f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 14235f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1424123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 14257bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 14267bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 142754aa1f4dSChris Mason 1428bce4eae9SChris Mason if (!empty && src_nritems <= 8) 1429971a1f66SChris Mason return 1; 1430971a1f66SChris Mason 1431eb60ceacSChris Mason if (push_items <= 0) { 1432be0e5c09SChris Mason return 1; 1433eb60ceacSChris Mason } 1434be0e5c09SChris Mason 1435bce4eae9SChris Mason if (empty) { 1436971a1f66SChris Mason push_items = min(src_nritems, push_items); 1437bce4eae9SChris Mason if (push_items < src_nritems) { 1438bce4eae9SChris Mason /* leave at least 8 pointers in the node if 1439bce4eae9SChris Mason * we aren't going to empty it 1440bce4eae9SChris Mason */ 1441bce4eae9SChris Mason if (src_nritems - push_items < 8) { 1442bce4eae9SChris Mason if (push_items <= 8) 1443bce4eae9SChris Mason return 1; 1444bce4eae9SChris Mason push_items -= 8; 1445bce4eae9SChris Mason } 1446bce4eae9SChris Mason } 1447bce4eae9SChris Mason } else 1448bce4eae9SChris Mason push_items = min(src_nritems - 8, push_items); 144979f95c82SChris Mason 14505f39d397SChris Mason copy_extent_buffer(dst, src, 14515f39d397SChris Mason btrfs_node_key_ptr_offset(dst_nritems), 14525f39d397SChris Mason btrfs_node_key_ptr_offset(0), 1453123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 14545f39d397SChris Mason 1455bb803951SChris Mason if (push_items < src_nritems) { 14565f39d397SChris Mason memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0), 14575f39d397SChris Mason btrfs_node_key_ptr_offset(push_items), 1458e2fa7227SChris Mason (src_nritems - push_items) * 1459123abc88SChris Mason sizeof(struct btrfs_key_ptr)); 1460bb803951SChris Mason } 14615f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 14625f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 14635f39d397SChris Mason btrfs_mark_buffer_dirty(src); 14645f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 1465bb803951SChris Mason return ret; 1466be0e5c09SChris Mason } 1467be0e5c09SChris Mason 146897571fd0SChris Mason /* 146979f95c82SChris Mason * try to push data from one node into the next node right in the 147079f95c82SChris Mason * tree. 147179f95c82SChris Mason * 147279f95c82SChris Mason * returns 0 if some ptrs were pushed, < 0 if there was some horrible 147379f95c82SChris Mason * error, and > 0 if there was no room in the right hand block. 147479f95c82SChris Mason * 147579f95c82SChris Mason * this will only push up to 1/2 the contents of the left node over 147679f95c82SChris Mason */ 14775f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 14785f39d397SChris Mason struct btrfs_root *root, 14795f39d397SChris Mason struct extent_buffer *dst, 14805f39d397SChris Mason struct extent_buffer *src) 148179f95c82SChris Mason { 148279f95c82SChris Mason int push_items = 0; 148379f95c82SChris Mason int max_push; 148479f95c82SChris Mason int src_nritems; 148579f95c82SChris Mason int dst_nritems; 148679f95c82SChris Mason int ret = 0; 148779f95c82SChris Mason 14887bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 14897bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 14907bb86316SChris Mason 14915f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 14925f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1493123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 1494bce4eae9SChris Mason if (push_items <= 0) { 149579f95c82SChris Mason return 1; 1496bce4eae9SChris Mason } 1497bce4eae9SChris Mason 1498bce4eae9SChris Mason if (src_nritems < 4) { 1499bce4eae9SChris Mason return 1; 1500bce4eae9SChris Mason } 150179f95c82SChris Mason 150279f95c82SChris Mason max_push = src_nritems / 2 + 1; 150379f95c82SChris Mason /* don't try to empty the node */ 1504bce4eae9SChris Mason if (max_push >= src_nritems) { 150579f95c82SChris Mason return 1; 1506bce4eae9SChris Mason } 1507252c38f0SYan 150879f95c82SChris Mason if (max_push < push_items) 150979f95c82SChris Mason push_items = max_push; 151079f95c82SChris Mason 15115f39d397SChris Mason memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items), 15125f39d397SChris Mason btrfs_node_key_ptr_offset(0), 15135f39d397SChris Mason (dst_nritems) * 15145f39d397SChris Mason sizeof(struct btrfs_key_ptr)); 1515d6025579SChris Mason 15165f39d397SChris Mason copy_extent_buffer(dst, src, 15175f39d397SChris Mason btrfs_node_key_ptr_offset(0), 15185f39d397SChris Mason btrfs_node_key_ptr_offset(src_nritems - push_items), 1519123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 152079f95c82SChris Mason 15215f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 15225f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 152379f95c82SChris Mason 15245f39d397SChris Mason btrfs_mark_buffer_dirty(src); 15255f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 152679f95c82SChris Mason return ret; 152779f95c82SChris Mason } 152879f95c82SChris Mason 152979f95c82SChris Mason /* 153097571fd0SChris Mason * helper function to insert a new root level in the tree. 153197571fd0SChris Mason * A new node is allocated, and a single item is inserted to 153297571fd0SChris Mason * point to the existing root 1533aa5d6bedSChris Mason * 1534aa5d6bedSChris Mason * returns zero on success or < 0 on failure. 153597571fd0SChris Mason */ 153698ed5174SChris Mason static int noinline insert_new_root(struct btrfs_trans_handle *trans, 15375f39d397SChris Mason struct btrfs_root *root, 15385f39d397SChris Mason struct btrfs_path *path, int level) 153974123bd7SChris Mason { 15407bb86316SChris Mason u64 root_gen; 15417bb86316SChris Mason u64 lower_gen; 15425f39d397SChris Mason struct extent_buffer *lower; 15435f39d397SChris Mason struct extent_buffer *c; 1544925baeddSChris Mason struct extent_buffer *old; 15455f39d397SChris Mason struct btrfs_disk_key lower_key; 15465c680ed6SChris Mason 15475c680ed6SChris Mason BUG_ON(path->nodes[level]); 15485c680ed6SChris Mason BUG_ON(path->nodes[level-1] != root->node); 15495c680ed6SChris Mason 15507bb86316SChris Mason if (root->ref_cows) 15517bb86316SChris Mason root_gen = trans->transid; 15527bb86316SChris Mason else 15537bb86316SChris Mason root_gen = 0; 15547bb86316SChris Mason 15557bb86316SChris Mason lower = path->nodes[level-1]; 15567bb86316SChris Mason if (level == 1) 15577bb86316SChris Mason btrfs_item_key(lower, &lower_key, 0); 15587bb86316SChris Mason else 15597bb86316SChris Mason btrfs_node_key(lower, &lower_key, 0); 15607bb86316SChris Mason 1561925baeddSChris Mason c = btrfs_alloc_free_block(trans, root, root->nodesize, 15627bb86316SChris Mason root->root_key.objectid, 15637bb86316SChris Mason root_gen, lower_key.objectid, level, 1564db94535dSChris Mason root->node->start, 0); 15655f39d397SChris Mason if (IS_ERR(c)) 15665f39d397SChris Mason return PTR_ERR(c); 1567925baeddSChris Mason 15685f39d397SChris Mason memset_extent_buffer(c, 0, 0, root->nodesize); 15695f39d397SChris Mason btrfs_set_header_nritems(c, 1); 15705f39d397SChris Mason btrfs_set_header_level(c, level); 1571db94535dSChris Mason btrfs_set_header_bytenr(c, c->start); 15725f39d397SChris Mason btrfs_set_header_generation(c, trans->transid); 15735f39d397SChris Mason btrfs_set_header_owner(c, root->root_key.objectid); 1574d5719762SChris Mason 15755f39d397SChris Mason write_extent_buffer(c, root->fs_info->fsid, 15765f39d397SChris Mason (unsigned long)btrfs_header_fsid(c), 15775f39d397SChris Mason BTRFS_FSID_SIZE); 1578e17cade2SChris Mason 1579e17cade2SChris Mason write_extent_buffer(c, root->fs_info->chunk_tree_uuid, 1580e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(c), 1581e17cade2SChris Mason BTRFS_UUID_SIZE); 1582e17cade2SChris Mason 15835f39d397SChris Mason btrfs_set_node_key(c, &lower_key, 0); 1584db94535dSChris Mason btrfs_set_node_blockptr(c, 0, lower->start); 15857bb86316SChris Mason lower_gen = btrfs_header_generation(lower); 15867bb86316SChris Mason WARN_ON(lower_gen == 0); 15877bb86316SChris Mason 15887bb86316SChris Mason btrfs_set_node_ptr_generation(c, 0, lower_gen); 15895f39d397SChris Mason 15905f39d397SChris Mason btrfs_mark_buffer_dirty(c); 1591d5719762SChris Mason 1592925baeddSChris Mason spin_lock(&root->node_lock); 1593925baeddSChris Mason old = root->node; 15945f39d397SChris Mason root->node = c; 1595925baeddSChris Mason spin_unlock(&root->node_lock); 1596925baeddSChris Mason 1597925baeddSChris Mason /* the super has an extra ref to root->node */ 1598925baeddSChris Mason free_extent_buffer(old); 1599925baeddSChris Mason 16000b86a832SChris Mason add_root_to_dirty_list(root); 16015f39d397SChris Mason extent_buffer_get(c); 16025f39d397SChris Mason path->nodes[level] = c; 1603925baeddSChris Mason path->locks[level] = 1; 160474123bd7SChris Mason path->slots[level] = 0; 16057bb86316SChris Mason 16067bb86316SChris Mason if (root->ref_cows && lower_gen != trans->transid) { 16077bb86316SChris Mason struct btrfs_path *back_path = btrfs_alloc_path(); 16087bb86316SChris Mason int ret; 1609925baeddSChris Mason mutex_lock(&root->fs_info->alloc_mutex); 16107bb86316SChris Mason ret = btrfs_insert_extent_backref(trans, 16117bb86316SChris Mason root->fs_info->extent_root, 16127bb86316SChris Mason path, lower->start, 16137bb86316SChris Mason root->root_key.objectid, 16147bb86316SChris Mason trans->transid, 0, 0); 16157bb86316SChris Mason BUG_ON(ret); 1616925baeddSChris Mason mutex_unlock(&root->fs_info->alloc_mutex); 16177bb86316SChris Mason btrfs_free_path(back_path); 16187bb86316SChris Mason } 161974123bd7SChris Mason return 0; 162074123bd7SChris Mason } 16215c680ed6SChris Mason 16225c680ed6SChris Mason /* 16235c680ed6SChris Mason * worker function to insert a single pointer in a node. 16245c680ed6SChris Mason * the node should have enough room for the pointer already 162597571fd0SChris Mason * 16265c680ed6SChris Mason * slot and level indicate where you want the key to go, and 16275c680ed6SChris Mason * blocknr is the block the key points to. 1628aa5d6bedSChris Mason * 1629aa5d6bedSChris Mason * returns zero on success and < 0 on any error 16305c680ed6SChris Mason */ 1631e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root 1632e089f05cSChris Mason *root, struct btrfs_path *path, struct btrfs_disk_key 1633db94535dSChris Mason *key, u64 bytenr, int slot, int level) 16345c680ed6SChris Mason { 16355f39d397SChris Mason struct extent_buffer *lower; 16365c680ed6SChris Mason int nritems; 16375c680ed6SChris Mason 16385c680ed6SChris Mason BUG_ON(!path->nodes[level]); 16395f39d397SChris Mason lower = path->nodes[level]; 16405f39d397SChris Mason nritems = btrfs_header_nritems(lower); 164174123bd7SChris Mason if (slot > nritems) 164274123bd7SChris Mason BUG(); 1643123abc88SChris Mason if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root)) 164474123bd7SChris Mason BUG(); 164574123bd7SChris Mason if (slot != nritems) { 16465f39d397SChris Mason memmove_extent_buffer(lower, 16475f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 16485f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 1649123abc88SChris Mason (nritems - slot) * sizeof(struct btrfs_key_ptr)); 165074123bd7SChris Mason } 16515f39d397SChris Mason btrfs_set_node_key(lower, key, slot); 1652db94535dSChris Mason btrfs_set_node_blockptr(lower, slot, bytenr); 165374493f7aSChris Mason WARN_ON(trans->transid == 0); 165474493f7aSChris Mason btrfs_set_node_ptr_generation(lower, slot, trans->transid); 16555f39d397SChris Mason btrfs_set_header_nritems(lower, nritems + 1); 16565f39d397SChris Mason btrfs_mark_buffer_dirty(lower); 165774123bd7SChris Mason return 0; 165874123bd7SChris Mason } 165974123bd7SChris Mason 166097571fd0SChris Mason /* 166197571fd0SChris Mason * split the node at the specified level in path in two. 166297571fd0SChris Mason * The path is corrected to point to the appropriate node after the split 166397571fd0SChris Mason * 166497571fd0SChris Mason * Before splitting this tries to make some room in the node by pushing 166597571fd0SChris Mason * left and right, if either one works, it returns right away. 1666aa5d6bedSChris Mason * 1667aa5d6bedSChris Mason * returns 0 on success and < 0 on failure 166897571fd0SChris Mason */ 1669e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 1670e089f05cSChris Mason *root, struct btrfs_path *path, int level) 1671be0e5c09SChris Mason { 16727bb86316SChris Mason u64 root_gen; 16735f39d397SChris Mason struct extent_buffer *c; 16745f39d397SChris Mason struct extent_buffer *split; 16755f39d397SChris Mason struct btrfs_disk_key disk_key; 1676be0e5c09SChris Mason int mid; 16775c680ed6SChris Mason int ret; 1678aa5d6bedSChris Mason int wret; 16797518a238SChris Mason u32 c_nritems; 1680be0e5c09SChris Mason 16815f39d397SChris Mason c = path->nodes[level]; 16827bb86316SChris Mason WARN_ON(btrfs_header_generation(c) != trans->transid); 16835f39d397SChris Mason if (c == root->node) { 16845c680ed6SChris Mason /* trying to split the root, lets make a new one */ 1685e089f05cSChris Mason ret = insert_new_root(trans, root, path, level + 1); 16865c680ed6SChris Mason if (ret) 16875c680ed6SChris Mason return ret; 1688e66f709bSChris Mason } else { 1689e66f709bSChris Mason ret = push_nodes_for_insert(trans, root, path, level); 16905f39d397SChris Mason c = path->nodes[level]; 16915f39d397SChris Mason if (!ret && btrfs_header_nritems(c) < 1692c448acf0SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) 1693e66f709bSChris Mason return 0; 169454aa1f4dSChris Mason if (ret < 0) 169554aa1f4dSChris Mason return ret; 16965c680ed6SChris Mason } 1697e66f709bSChris Mason 16985f39d397SChris Mason c_nritems = btrfs_header_nritems(c); 16997bb86316SChris Mason if (root->ref_cows) 17007bb86316SChris Mason root_gen = trans->transid; 17017bb86316SChris Mason else 17027bb86316SChris Mason root_gen = 0; 17037bb86316SChris Mason 17047bb86316SChris Mason btrfs_node_key(c, &disk_key, 0); 1705925baeddSChris Mason split = btrfs_alloc_free_block(trans, root, root->nodesize, 17067bb86316SChris Mason root->root_key.objectid, 17077bb86316SChris Mason root_gen, 17087bb86316SChris Mason btrfs_disk_key_objectid(&disk_key), 17097bb86316SChris Mason level, c->start, 0); 17105f39d397SChris Mason if (IS_ERR(split)) 17115f39d397SChris Mason return PTR_ERR(split); 171254aa1f4dSChris Mason 17135f39d397SChris Mason btrfs_set_header_flags(split, btrfs_header_flags(c)); 17145f39d397SChris Mason btrfs_set_header_level(split, btrfs_header_level(c)); 1715db94535dSChris Mason btrfs_set_header_bytenr(split, split->start); 17165f39d397SChris Mason btrfs_set_header_generation(split, trans->transid); 17175f39d397SChris Mason btrfs_set_header_owner(split, root->root_key.objectid); 171863b10fc4SChris Mason btrfs_set_header_flags(split, 0); 17195f39d397SChris Mason write_extent_buffer(split, root->fs_info->fsid, 17205f39d397SChris Mason (unsigned long)btrfs_header_fsid(split), 17215f39d397SChris Mason BTRFS_FSID_SIZE); 1722e17cade2SChris Mason write_extent_buffer(split, root->fs_info->chunk_tree_uuid, 1723e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(split), 1724e17cade2SChris Mason BTRFS_UUID_SIZE); 17255f39d397SChris Mason 17267518a238SChris Mason mid = (c_nritems + 1) / 2; 17275f39d397SChris Mason 17285f39d397SChris Mason copy_extent_buffer(split, c, 17295f39d397SChris Mason btrfs_node_key_ptr_offset(0), 17305f39d397SChris Mason btrfs_node_key_ptr_offset(mid), 1731123abc88SChris Mason (c_nritems - mid) * sizeof(struct btrfs_key_ptr)); 17325f39d397SChris Mason btrfs_set_header_nritems(split, c_nritems - mid); 17335f39d397SChris Mason btrfs_set_header_nritems(c, mid); 1734aa5d6bedSChris Mason ret = 0; 1735aa5d6bedSChris Mason 17365f39d397SChris Mason btrfs_mark_buffer_dirty(c); 17375f39d397SChris Mason btrfs_mark_buffer_dirty(split); 17385f39d397SChris Mason 17395f39d397SChris Mason btrfs_node_key(split, &disk_key, 0); 1740db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, split->start, 17415f39d397SChris Mason path->slots[level + 1] + 1, 1742123abc88SChris Mason level + 1); 1743aa5d6bedSChris Mason if (wret) 1744aa5d6bedSChris Mason ret = wret; 1745aa5d6bedSChris Mason 17465de08d7dSChris Mason if (path->slots[level] >= mid) { 17475c680ed6SChris Mason path->slots[level] -= mid; 1748925baeddSChris Mason btrfs_tree_unlock(c); 17495f39d397SChris Mason free_extent_buffer(c); 17505f39d397SChris Mason path->nodes[level] = split; 17515c680ed6SChris Mason path->slots[level + 1] += 1; 1752eb60ceacSChris Mason } else { 1753925baeddSChris Mason btrfs_tree_unlock(split); 17545f39d397SChris Mason free_extent_buffer(split); 1755be0e5c09SChris Mason } 1756aa5d6bedSChris Mason return ret; 1757be0e5c09SChris Mason } 1758be0e5c09SChris Mason 175974123bd7SChris Mason /* 176074123bd7SChris Mason * how many bytes are required to store the items in a leaf. start 176174123bd7SChris Mason * and nr indicate which items in the leaf to check. This totals up the 176274123bd7SChris Mason * space used both by the item structs and the item data 176374123bd7SChris Mason */ 17645f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr) 1765be0e5c09SChris Mason { 1766be0e5c09SChris Mason int data_len; 17675f39d397SChris Mason int nritems = btrfs_header_nritems(l); 1768d4dbff95SChris Mason int end = min(nritems, start + nr) - 1; 1769be0e5c09SChris Mason 1770be0e5c09SChris Mason if (!nr) 1771be0e5c09SChris Mason return 0; 17725f39d397SChris Mason data_len = btrfs_item_end_nr(l, start); 17735f39d397SChris Mason data_len = data_len - btrfs_item_offset_nr(l, end); 17740783fcfcSChris Mason data_len += sizeof(struct btrfs_item) * nr; 1775d4dbff95SChris Mason WARN_ON(data_len < 0); 1776be0e5c09SChris Mason return data_len; 1777be0e5c09SChris Mason } 1778be0e5c09SChris Mason 177974123bd7SChris Mason /* 1780d4dbff95SChris Mason * The space between the end of the leaf items and 1781d4dbff95SChris Mason * the start of the leaf data. IOW, how much room 1782d4dbff95SChris Mason * the leaf has left for both items and data 1783d4dbff95SChris Mason */ 17845f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf) 1785d4dbff95SChris Mason { 17865f39d397SChris Mason int nritems = btrfs_header_nritems(leaf); 17875f39d397SChris Mason int ret; 17885f39d397SChris Mason ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems); 17895f39d397SChris Mason if (ret < 0) { 17905f39d397SChris Mason printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n", 1791ae2f5411SJens Axboe ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root), 17925f39d397SChris Mason leaf_space_used(leaf, 0, nritems), nritems); 17935f39d397SChris Mason } 17945f39d397SChris Mason return ret; 1795d4dbff95SChris Mason } 1796d4dbff95SChris Mason 1797d4dbff95SChris Mason /* 179800ec4c51SChris Mason * push some data in the path leaf to the right, trying to free up at 179900ec4c51SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 1800aa5d6bedSChris Mason * 1801aa5d6bedSChris Mason * returns 1 if the push failed because the other node didn't have enough 1802aa5d6bedSChris Mason * room, 0 if everything worked out and < 0 if there were major errors. 180300ec4c51SChris Mason */ 1804e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root 180534a38218SChris Mason *root, struct btrfs_path *path, int data_size, 180634a38218SChris Mason int empty) 180700ec4c51SChris Mason { 18085f39d397SChris Mason struct extent_buffer *left = path->nodes[0]; 18095f39d397SChris Mason struct extent_buffer *right; 18105f39d397SChris Mason struct extent_buffer *upper; 18115f39d397SChris Mason struct btrfs_disk_key disk_key; 181200ec4c51SChris Mason int slot; 181334a38218SChris Mason u32 i; 181400ec4c51SChris Mason int free_space; 181500ec4c51SChris Mason int push_space = 0; 181600ec4c51SChris Mason int push_items = 0; 18170783fcfcSChris Mason struct btrfs_item *item; 18187518a238SChris Mason u32 left_nritems; 181934a38218SChris Mason u32 nr; 18207518a238SChris Mason u32 right_nritems; 18215f39d397SChris Mason u32 data_end; 1822db94535dSChris Mason u32 this_item_size; 182354aa1f4dSChris Mason int ret; 182400ec4c51SChris Mason 182500ec4c51SChris Mason slot = path->slots[1]; 182600ec4c51SChris Mason if (!path->nodes[1]) { 182700ec4c51SChris Mason return 1; 182800ec4c51SChris Mason } 182900ec4c51SChris Mason upper = path->nodes[1]; 18305f39d397SChris Mason if (slot >= btrfs_header_nritems(upper) - 1) 183100ec4c51SChris Mason return 1; 18325f39d397SChris Mason 1833a2135011SChris Mason WARN_ON(!btrfs_tree_locked(path->nodes[1])); 1834a2135011SChris Mason 1835ca7a79adSChris Mason right = read_node_slot(root, upper, slot + 1); 1836925baeddSChris Mason btrfs_tree_lock(right); 1837123abc88SChris Mason free_space = btrfs_leaf_free_space(root, right); 1838925baeddSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) 1839925baeddSChris Mason goto out_unlock; 184002217ed2SChris Mason 18415f39d397SChris Mason /* cow and double check */ 18425f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, upper, 18435f39d397SChris Mason slot + 1, &right); 1844925baeddSChris Mason if (ret) 1845925baeddSChris Mason goto out_unlock; 1846925baeddSChris Mason 18475f39d397SChris Mason free_space = btrfs_leaf_free_space(root, right); 1848925baeddSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) 1849925baeddSChris Mason goto out_unlock; 18505f39d397SChris Mason 18515f39d397SChris Mason left_nritems = btrfs_header_nritems(left); 1852925baeddSChris Mason if (left_nritems == 0) 1853925baeddSChris Mason goto out_unlock; 18545f39d397SChris Mason 185534a38218SChris Mason if (empty) 185634a38218SChris Mason nr = 0; 185734a38218SChris Mason else 185834a38218SChris Mason nr = 1; 185934a38218SChris Mason 186034a38218SChris Mason i = left_nritems - 1; 186134a38218SChris Mason while (i >= nr) { 18625f39d397SChris Mason item = btrfs_item_nr(left, i); 1863db94535dSChris Mason 186400ec4c51SChris Mason if (path->slots[0] == i) 186500ec4c51SChris Mason push_space += data_size + sizeof(*item); 1866db94535dSChris Mason 1867db94535dSChris Mason if (!left->map_token) { 1868db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 1869db94535dSChris Mason sizeof(struct btrfs_item), 1870db94535dSChris Mason &left->map_token, &left->kaddr, 1871db94535dSChris Mason &left->map_start, &left->map_len, 1872db94535dSChris Mason KM_USER1); 1873db94535dSChris Mason } 1874db94535dSChris Mason 1875db94535dSChris Mason this_item_size = btrfs_item_size(left, item); 1876db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 187700ec4c51SChris Mason break; 187800ec4c51SChris Mason push_items++; 1879db94535dSChris Mason push_space += this_item_size + sizeof(*item); 188034a38218SChris Mason if (i == 0) 188134a38218SChris Mason break; 188234a38218SChris Mason i--; 1883db94535dSChris Mason } 1884db94535dSChris Mason if (left->map_token) { 1885db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 1886db94535dSChris Mason left->map_token = NULL; 188700ec4c51SChris Mason } 18885f39d397SChris Mason 1889925baeddSChris Mason if (push_items == 0) 1890925baeddSChris Mason goto out_unlock; 18915f39d397SChris Mason 189234a38218SChris Mason if (!empty && push_items == left_nritems) 1893a429e513SChris Mason WARN_ON(1); 18945f39d397SChris Mason 189500ec4c51SChris Mason /* push left to right */ 18965f39d397SChris Mason right_nritems = btrfs_header_nritems(right); 189734a38218SChris Mason 18985f39d397SChris Mason push_space = btrfs_item_end_nr(left, left_nritems - push_items); 1899123abc88SChris Mason push_space -= leaf_data_end(root, left); 19005f39d397SChris Mason 190100ec4c51SChris Mason /* make room in the right data area */ 19025f39d397SChris Mason data_end = leaf_data_end(root, right); 19035f39d397SChris Mason memmove_extent_buffer(right, 19045f39d397SChris Mason btrfs_leaf_data(right) + data_end - push_space, 19055f39d397SChris Mason btrfs_leaf_data(right) + data_end, 19065f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root) - data_end); 19075f39d397SChris Mason 190800ec4c51SChris Mason /* copy from the left data area */ 19095f39d397SChris Mason copy_extent_buffer(right, left, btrfs_leaf_data(right) + 1910d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 1911d6025579SChris Mason btrfs_leaf_data(left) + leaf_data_end(root, left), 1912d6025579SChris Mason push_space); 19135f39d397SChris Mason 19145f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(push_items), 19155f39d397SChris Mason btrfs_item_nr_offset(0), 19160783fcfcSChris Mason right_nritems * sizeof(struct btrfs_item)); 19175f39d397SChris Mason 191800ec4c51SChris Mason /* copy the items from left to right */ 19195f39d397SChris Mason copy_extent_buffer(right, left, btrfs_item_nr_offset(0), 19205f39d397SChris Mason btrfs_item_nr_offset(left_nritems - push_items), 19210783fcfcSChris Mason push_items * sizeof(struct btrfs_item)); 192200ec4c51SChris Mason 192300ec4c51SChris Mason /* update the item pointers */ 19247518a238SChris Mason right_nritems += push_items; 19255f39d397SChris Mason btrfs_set_header_nritems(right, right_nritems); 1926123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 19277518a238SChris Mason for (i = 0; i < right_nritems; i++) { 19285f39d397SChris Mason item = btrfs_item_nr(right, i); 1929db94535dSChris Mason if (!right->map_token) { 1930db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1931db94535dSChris Mason sizeof(struct btrfs_item), 1932db94535dSChris Mason &right->map_token, &right->kaddr, 1933db94535dSChris Mason &right->map_start, &right->map_len, 1934db94535dSChris Mason KM_USER1); 1935db94535dSChris Mason } 1936db94535dSChris Mason push_space -= btrfs_item_size(right, item); 1937db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 1938db94535dSChris Mason } 1939db94535dSChris Mason 1940db94535dSChris Mason if (right->map_token) { 1941db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1942db94535dSChris Mason right->map_token = NULL; 194300ec4c51SChris Mason } 19447518a238SChris Mason left_nritems -= push_items; 19455f39d397SChris Mason btrfs_set_header_nritems(left, left_nritems); 194600ec4c51SChris Mason 194734a38218SChris Mason if (left_nritems) 19485f39d397SChris Mason btrfs_mark_buffer_dirty(left); 19495f39d397SChris Mason btrfs_mark_buffer_dirty(right); 1950a429e513SChris Mason 19515f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 19525f39d397SChris Mason btrfs_set_node_key(upper, &disk_key, slot + 1); 1953d6025579SChris Mason btrfs_mark_buffer_dirty(upper); 195402217ed2SChris Mason 195500ec4c51SChris Mason /* then fixup the leaf pointer in the path */ 19567518a238SChris Mason if (path->slots[0] >= left_nritems) { 19577518a238SChris Mason path->slots[0] -= left_nritems; 1958925baeddSChris Mason if (btrfs_header_nritems(path->nodes[0]) == 0) 1959925baeddSChris Mason clean_tree_block(trans, root, path->nodes[0]); 1960925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 19615f39d397SChris Mason free_extent_buffer(path->nodes[0]); 19625f39d397SChris Mason path->nodes[0] = right; 196300ec4c51SChris Mason path->slots[1] += 1; 196400ec4c51SChris Mason } else { 1965925baeddSChris Mason btrfs_tree_unlock(right); 19665f39d397SChris Mason free_extent_buffer(right); 196700ec4c51SChris Mason } 196800ec4c51SChris Mason return 0; 1969925baeddSChris Mason 1970925baeddSChris Mason out_unlock: 1971925baeddSChris Mason btrfs_tree_unlock(right); 1972925baeddSChris Mason free_extent_buffer(right); 1973925baeddSChris Mason return 1; 197400ec4c51SChris Mason } 1975925baeddSChris Mason 197600ec4c51SChris Mason /* 197774123bd7SChris Mason * push some data in the path leaf to the left, trying to free up at 197874123bd7SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 197974123bd7SChris Mason */ 1980e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root 198134a38218SChris Mason *root, struct btrfs_path *path, int data_size, 198234a38218SChris Mason int empty) 1983be0e5c09SChris Mason { 19845f39d397SChris Mason struct btrfs_disk_key disk_key; 19855f39d397SChris Mason struct extent_buffer *right = path->nodes[0]; 19865f39d397SChris Mason struct extent_buffer *left; 1987be0e5c09SChris Mason int slot; 1988be0e5c09SChris Mason int i; 1989be0e5c09SChris Mason int free_space; 1990be0e5c09SChris Mason int push_space = 0; 1991be0e5c09SChris Mason int push_items = 0; 19920783fcfcSChris Mason struct btrfs_item *item; 19937518a238SChris Mason u32 old_left_nritems; 19945f39d397SChris Mason u32 right_nritems; 199534a38218SChris Mason u32 nr; 1996aa5d6bedSChris Mason int ret = 0; 1997aa5d6bedSChris Mason int wret; 1998db94535dSChris Mason u32 this_item_size; 1999db94535dSChris Mason u32 old_left_item_size; 2000be0e5c09SChris Mason 2001be0e5c09SChris Mason slot = path->slots[1]; 20025f39d397SChris Mason if (slot == 0) 2003be0e5c09SChris Mason return 1; 20045f39d397SChris Mason if (!path->nodes[1]) 2005be0e5c09SChris Mason return 1; 20065f39d397SChris Mason 20073685f791SChris Mason right_nritems = btrfs_header_nritems(right); 20083685f791SChris Mason if (right_nritems == 0) { 20093685f791SChris Mason return 1; 20103685f791SChris Mason } 20113685f791SChris Mason 2012a2135011SChris Mason WARN_ON(!btrfs_tree_locked(path->nodes[1])); 2013a2135011SChris Mason 2014ca7a79adSChris Mason left = read_node_slot(root, path->nodes[1], slot - 1); 2015925baeddSChris Mason btrfs_tree_lock(left); 2016123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 20170783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 2018925baeddSChris Mason ret = 1; 2019925baeddSChris Mason goto out; 2020be0e5c09SChris Mason } 202102217ed2SChris Mason 202202217ed2SChris Mason /* cow and double check */ 20235f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, 20245f39d397SChris Mason path->nodes[1], slot - 1, &left); 202554aa1f4dSChris Mason if (ret) { 202654aa1f4dSChris Mason /* we hit -ENOSPC, but it isn't fatal here */ 2027925baeddSChris Mason ret = 1; 2028925baeddSChris Mason goto out; 202954aa1f4dSChris Mason } 20303685f791SChris Mason 2031123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 20320783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 2033925baeddSChris Mason ret = 1; 2034925baeddSChris Mason goto out; 203502217ed2SChris Mason } 203602217ed2SChris Mason 203734a38218SChris Mason if (empty) 203834a38218SChris Mason nr = right_nritems; 203934a38218SChris Mason else 204034a38218SChris Mason nr = right_nritems - 1; 204134a38218SChris Mason 204234a38218SChris Mason for (i = 0; i < nr; i++) { 20435f39d397SChris Mason item = btrfs_item_nr(right, i); 2044db94535dSChris Mason if (!right->map_token) { 2045db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2046db94535dSChris Mason sizeof(struct btrfs_item), 2047db94535dSChris Mason &right->map_token, &right->kaddr, 2048db94535dSChris Mason &right->map_start, &right->map_len, 2049db94535dSChris Mason KM_USER1); 2050db94535dSChris Mason } 2051db94535dSChris Mason 2052be0e5c09SChris Mason if (path->slots[0] == i) 2053be0e5c09SChris Mason push_space += data_size + sizeof(*item); 2054db94535dSChris Mason 2055db94535dSChris Mason this_item_size = btrfs_item_size(right, item); 2056db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 2057be0e5c09SChris Mason break; 2058db94535dSChris Mason 2059be0e5c09SChris Mason push_items++; 2060db94535dSChris Mason push_space += this_item_size + sizeof(*item); 2061be0e5c09SChris Mason } 2062db94535dSChris Mason 2063db94535dSChris Mason if (right->map_token) { 2064db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2065db94535dSChris Mason right->map_token = NULL; 2066db94535dSChris Mason } 2067db94535dSChris Mason 2068be0e5c09SChris Mason if (push_items == 0) { 2069925baeddSChris Mason ret = 1; 2070925baeddSChris Mason goto out; 2071be0e5c09SChris Mason } 207234a38218SChris Mason if (!empty && push_items == btrfs_header_nritems(right)) 2073a429e513SChris Mason WARN_ON(1); 20745f39d397SChris Mason 2075be0e5c09SChris Mason /* push data from right to left */ 20765f39d397SChris Mason copy_extent_buffer(left, right, 20775f39d397SChris Mason btrfs_item_nr_offset(btrfs_header_nritems(left)), 20785f39d397SChris Mason btrfs_item_nr_offset(0), 20795f39d397SChris Mason push_items * sizeof(struct btrfs_item)); 20805f39d397SChris Mason 2081123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root) - 20825f39d397SChris Mason btrfs_item_offset_nr(right, push_items -1); 20835f39d397SChris Mason 20845f39d397SChris Mason copy_extent_buffer(left, right, btrfs_leaf_data(left) + 2085d6025579SChris Mason leaf_data_end(root, left) - push_space, 2086123abc88SChris Mason btrfs_leaf_data(right) + 20875f39d397SChris Mason btrfs_item_offset_nr(right, push_items - 1), 2088be0e5c09SChris Mason push_space); 20895f39d397SChris Mason old_left_nritems = btrfs_header_nritems(left); 2090eb60ceacSChris Mason BUG_ON(old_left_nritems < 0); 2091eb60ceacSChris Mason 2092db94535dSChris Mason old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1); 2093be0e5c09SChris Mason for (i = old_left_nritems; i < old_left_nritems + push_items; i++) { 20945f39d397SChris Mason u32 ioff; 2095db94535dSChris Mason 20965f39d397SChris Mason item = btrfs_item_nr(left, i); 2097db94535dSChris Mason if (!left->map_token) { 2098db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 2099db94535dSChris Mason sizeof(struct btrfs_item), 2100db94535dSChris Mason &left->map_token, &left->kaddr, 2101db94535dSChris Mason &left->map_start, &left->map_len, 2102db94535dSChris Mason KM_USER1); 2103db94535dSChris Mason } 2104db94535dSChris Mason 21055f39d397SChris Mason ioff = btrfs_item_offset(left, item); 21065f39d397SChris Mason btrfs_set_item_offset(left, item, 2107db94535dSChris Mason ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size)); 2108be0e5c09SChris Mason } 21095f39d397SChris Mason btrfs_set_header_nritems(left, old_left_nritems + push_items); 2110db94535dSChris Mason if (left->map_token) { 2111db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 2112db94535dSChris Mason left->map_token = NULL; 2113db94535dSChris Mason } 2114be0e5c09SChris Mason 2115be0e5c09SChris Mason /* fixup right node */ 211634a38218SChris Mason if (push_items > right_nritems) { 211734a38218SChris Mason printk("push items %d nr %u\n", push_items, right_nritems); 211834a38218SChris Mason WARN_ON(1); 211934a38218SChris Mason } 212034a38218SChris Mason 212134a38218SChris Mason if (push_items < right_nritems) { 21225f39d397SChris Mason push_space = btrfs_item_offset_nr(right, push_items - 1) - 2123123abc88SChris Mason leaf_data_end(root, right); 21245f39d397SChris Mason memmove_extent_buffer(right, btrfs_leaf_data(right) + 2125d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 2126d6025579SChris Mason btrfs_leaf_data(right) + 2127123abc88SChris Mason leaf_data_end(root, right), push_space); 21285f39d397SChris Mason 21295f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(0), 21305f39d397SChris Mason btrfs_item_nr_offset(push_items), 21315f39d397SChris Mason (btrfs_header_nritems(right) - push_items) * 21320783fcfcSChris Mason sizeof(struct btrfs_item)); 213334a38218SChris Mason } 2134eef1c494SYan right_nritems -= push_items; 2135eef1c494SYan btrfs_set_header_nritems(right, right_nritems); 2136123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 21375f39d397SChris Mason for (i = 0; i < right_nritems; i++) { 21385f39d397SChris Mason item = btrfs_item_nr(right, i); 2139db94535dSChris Mason 2140db94535dSChris Mason if (!right->map_token) { 2141db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2142db94535dSChris Mason sizeof(struct btrfs_item), 2143db94535dSChris Mason &right->map_token, &right->kaddr, 2144db94535dSChris Mason &right->map_start, &right->map_len, 2145db94535dSChris Mason KM_USER1); 2146db94535dSChris Mason } 2147db94535dSChris Mason 2148db94535dSChris Mason push_space = push_space - btrfs_item_size(right, item); 2149db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 2150db94535dSChris Mason } 2151db94535dSChris Mason if (right->map_token) { 2152db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2153db94535dSChris Mason right->map_token = NULL; 2154be0e5c09SChris Mason } 2155eb60ceacSChris Mason 21565f39d397SChris Mason btrfs_mark_buffer_dirty(left); 215734a38218SChris Mason if (right_nritems) 21585f39d397SChris Mason btrfs_mark_buffer_dirty(right); 2159098f59c2SChris Mason 21605f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 21615f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, 1); 2162aa5d6bedSChris Mason if (wret) 2163aa5d6bedSChris Mason ret = wret; 2164be0e5c09SChris Mason 2165be0e5c09SChris Mason /* then fixup the leaf pointer in the path */ 2166be0e5c09SChris Mason if (path->slots[0] < push_items) { 2167be0e5c09SChris Mason path->slots[0] += old_left_nritems; 2168925baeddSChris Mason if (btrfs_header_nritems(path->nodes[0]) == 0) 2169925baeddSChris Mason clean_tree_block(trans, root, path->nodes[0]); 2170925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 21715f39d397SChris Mason free_extent_buffer(path->nodes[0]); 21725f39d397SChris Mason path->nodes[0] = left; 2173be0e5c09SChris Mason path->slots[1] -= 1; 2174be0e5c09SChris Mason } else { 2175925baeddSChris Mason btrfs_tree_unlock(left); 21765f39d397SChris Mason free_extent_buffer(left); 2177be0e5c09SChris Mason path->slots[0] -= push_items; 2178be0e5c09SChris Mason } 2179eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 2180aa5d6bedSChris Mason return ret; 2181925baeddSChris Mason out: 2182925baeddSChris Mason btrfs_tree_unlock(left); 2183925baeddSChris Mason free_extent_buffer(left); 2184925baeddSChris Mason return ret; 2185be0e5c09SChris Mason } 2186be0e5c09SChris Mason 218774123bd7SChris Mason /* 218874123bd7SChris Mason * split the path's leaf in two, making sure there is at least data_size 218974123bd7SChris Mason * available for the resulting leaf level of the path. 2190aa5d6bedSChris Mason * 2191aa5d6bedSChris Mason * returns 0 if all went well and < 0 on failure. 219274123bd7SChris Mason */ 2193e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 2194d4dbff95SChris Mason *root, struct btrfs_key *ins_key, 2195cc0c5538SChris Mason struct btrfs_path *path, int data_size, int extend) 2196be0e5c09SChris Mason { 21977bb86316SChris Mason u64 root_gen; 21985f39d397SChris Mason struct extent_buffer *l; 21997518a238SChris Mason u32 nritems; 2200eb60ceacSChris Mason int mid; 2201eb60ceacSChris Mason int slot; 22025f39d397SChris Mason struct extent_buffer *right; 22030783fcfcSChris Mason int space_needed = data_size + sizeof(struct btrfs_item); 2204be0e5c09SChris Mason int data_copy_size; 2205be0e5c09SChris Mason int rt_data_off; 2206be0e5c09SChris Mason int i; 2207d4dbff95SChris Mason int ret = 0; 2208aa5d6bedSChris Mason int wret; 2209cc0c5538SChris Mason int double_split; 2210cc0c5538SChris Mason int num_doubles = 0; 2211d4dbff95SChris Mason struct btrfs_disk_key disk_key; 2212be0e5c09SChris Mason 2213cc0c5538SChris Mason if (extend) 2214cc0c5538SChris Mason space_needed = data_size; 2215cc0c5538SChris Mason 22167bb86316SChris Mason if (root->ref_cows) 22177bb86316SChris Mason root_gen = trans->transid; 22187bb86316SChris Mason else 22197bb86316SChris Mason root_gen = 0; 22207bb86316SChris Mason 222140689478SChris Mason /* first try to make some room by pushing left and right */ 22223685f791SChris Mason if (ins_key->type != BTRFS_DIR_ITEM_KEY) { 222334a38218SChris Mason wret = push_leaf_right(trans, root, path, data_size, 0); 22243326d1b0SChris Mason if (wret < 0) { 2225eaee50e8SChris Mason return wret; 22263326d1b0SChris Mason } 2227eaee50e8SChris Mason if (wret) { 222834a38218SChris Mason wret = push_leaf_left(trans, root, path, data_size, 0); 2229eaee50e8SChris Mason if (wret < 0) 2230eaee50e8SChris Mason return wret; 2231eaee50e8SChris Mason } 22325f39d397SChris Mason l = path->nodes[0]; 2233aa5d6bedSChris Mason 2234aa5d6bedSChris Mason /* did the pushes work? */ 2235cc0c5538SChris Mason if (btrfs_leaf_free_space(root, l) >= space_needed) 2236be0e5c09SChris Mason return 0; 22373326d1b0SChris Mason } 2238aa5d6bedSChris Mason 22395c680ed6SChris Mason if (!path->nodes[1]) { 2240e089f05cSChris Mason ret = insert_new_root(trans, root, path, 1); 22415c680ed6SChris Mason if (ret) 22425c680ed6SChris Mason return ret; 22435c680ed6SChris Mason } 2244cc0c5538SChris Mason again: 2245cc0c5538SChris Mason double_split = 0; 2246cc0c5538SChris Mason l = path->nodes[0]; 2247eb60ceacSChris Mason slot = path->slots[0]; 22485f39d397SChris Mason nritems = btrfs_header_nritems(l); 2249eb60ceacSChris Mason mid = (nritems + 1)/ 2; 225054aa1f4dSChris Mason 22517bb86316SChris Mason btrfs_item_key(l, &disk_key, 0); 22527bb86316SChris Mason 2253925baeddSChris Mason right = btrfs_alloc_free_block(trans, root, root->leafsize, 22547bb86316SChris Mason root->root_key.objectid, 22557bb86316SChris Mason root_gen, disk_key.objectid, 0, 2256db94535dSChris Mason l->start, 0); 2257cea9e445SChris Mason if (IS_ERR(right)) { 2258cea9e445SChris Mason BUG_ON(1); 22595f39d397SChris Mason return PTR_ERR(right); 2260cea9e445SChris Mason } 226154aa1f4dSChris Mason 22625f39d397SChris Mason memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header)); 2263db94535dSChris Mason btrfs_set_header_bytenr(right, right->start); 22645f39d397SChris Mason btrfs_set_header_generation(right, trans->transid); 22655f39d397SChris Mason btrfs_set_header_owner(right, root->root_key.objectid); 22665f39d397SChris Mason btrfs_set_header_level(right, 0); 22675f39d397SChris Mason write_extent_buffer(right, root->fs_info->fsid, 22685f39d397SChris Mason (unsigned long)btrfs_header_fsid(right), 22695f39d397SChris Mason BTRFS_FSID_SIZE); 2270e17cade2SChris Mason 2271e17cade2SChris Mason write_extent_buffer(right, root->fs_info->chunk_tree_uuid, 2272e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(right), 2273e17cade2SChris Mason BTRFS_UUID_SIZE); 2274d4dbff95SChris Mason if (mid <= slot) { 2275d4dbff95SChris Mason if (nritems == 1 || 2276d4dbff95SChris Mason leaf_space_used(l, mid, nritems - mid) + space_needed > 2277d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 2278d4dbff95SChris Mason if (slot >= nritems) { 2279d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 22805f39d397SChris Mason btrfs_set_header_nritems(right, 0); 2281d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 2282db94535dSChris Mason &disk_key, right->start, 2283d4dbff95SChris Mason path->slots[1] + 1, 1); 2284d4dbff95SChris Mason if (wret) 2285d4dbff95SChris Mason ret = wret; 2286925baeddSChris Mason 2287925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 22885f39d397SChris Mason free_extent_buffer(path->nodes[0]); 22895f39d397SChris Mason path->nodes[0] = right; 2290d4dbff95SChris Mason path->slots[0] = 0; 2291d4dbff95SChris Mason path->slots[1] += 1; 22920ef8b242SChris Mason btrfs_mark_buffer_dirty(right); 2293d4dbff95SChris Mason return ret; 2294d4dbff95SChris Mason } 2295d4dbff95SChris Mason mid = slot; 22963326d1b0SChris Mason if (mid != nritems && 22973326d1b0SChris Mason leaf_space_used(l, mid, nritems - mid) + 22983326d1b0SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 2299d4dbff95SChris Mason double_split = 1; 2300d4dbff95SChris Mason } 23013326d1b0SChris Mason } 2302d4dbff95SChris Mason } else { 2303d4dbff95SChris Mason if (leaf_space_used(l, 0, mid + 1) + space_needed > 2304d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 2305cc0c5538SChris Mason if (!extend && slot == 0) { 2306d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 23075f39d397SChris Mason btrfs_set_header_nritems(right, 0); 2308d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 2309d4dbff95SChris Mason &disk_key, 2310db94535dSChris Mason right->start, 2311098f59c2SChris Mason path->slots[1], 1); 2312d4dbff95SChris Mason if (wret) 2313d4dbff95SChris Mason ret = wret; 2314925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 23155f39d397SChris Mason free_extent_buffer(path->nodes[0]); 23165f39d397SChris Mason path->nodes[0] = right; 2317d4dbff95SChris Mason path->slots[0] = 0; 2318a429e513SChris Mason if (path->slots[1] == 0) { 2319a429e513SChris Mason wret = fixup_low_keys(trans, root, 2320a429e513SChris Mason path, &disk_key, 1); 2321a429e513SChris Mason if (wret) 2322a429e513SChris Mason ret = wret; 2323a429e513SChris Mason } 23240ef8b242SChris Mason btrfs_mark_buffer_dirty(right); 2325d4dbff95SChris Mason return ret; 2326cc0c5538SChris Mason } else if (extend && slot == 0) { 2327cc0c5538SChris Mason mid = 1; 2328cc0c5538SChris Mason } else { 2329d4dbff95SChris Mason mid = slot; 23305ee78ac7SChris Mason if (mid != nritems && 23315ee78ac7SChris Mason leaf_space_used(l, mid, nritems - mid) + 23325ee78ac7SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 2333d4dbff95SChris Mason double_split = 1; 2334d4dbff95SChris Mason } 2335d4dbff95SChris Mason } 23365ee78ac7SChris Mason } 2337cc0c5538SChris Mason } 23385f39d397SChris Mason nritems = nritems - mid; 23395f39d397SChris Mason btrfs_set_header_nritems(right, nritems); 23405f39d397SChris Mason data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l); 23415f39d397SChris Mason 23425f39d397SChris Mason copy_extent_buffer(right, l, btrfs_item_nr_offset(0), 23435f39d397SChris Mason btrfs_item_nr_offset(mid), 23445f39d397SChris Mason nritems * sizeof(struct btrfs_item)); 23455f39d397SChris Mason 23465f39d397SChris Mason copy_extent_buffer(right, l, 2347d6025579SChris Mason btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - 2348123abc88SChris Mason data_copy_size, btrfs_leaf_data(l) + 2349123abc88SChris Mason leaf_data_end(root, l), data_copy_size); 235074123bd7SChris Mason 23515f39d397SChris Mason rt_data_off = BTRFS_LEAF_DATA_SIZE(root) - 23525f39d397SChris Mason btrfs_item_end_nr(l, mid); 23535f39d397SChris Mason 23545f39d397SChris Mason for (i = 0; i < nritems; i++) { 23555f39d397SChris Mason struct btrfs_item *item = btrfs_item_nr(right, i); 2356db94535dSChris Mason u32 ioff; 2357db94535dSChris Mason 2358db94535dSChris Mason if (!right->map_token) { 2359db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2360db94535dSChris Mason sizeof(struct btrfs_item), 2361db94535dSChris Mason &right->map_token, &right->kaddr, 2362db94535dSChris Mason &right->map_start, &right->map_len, 2363db94535dSChris Mason KM_USER1); 2364db94535dSChris Mason } 2365db94535dSChris Mason 2366db94535dSChris Mason ioff = btrfs_item_offset(right, item); 23675f39d397SChris Mason btrfs_set_item_offset(right, item, ioff + rt_data_off); 23680783fcfcSChris Mason } 236974123bd7SChris Mason 2370db94535dSChris Mason if (right->map_token) { 2371db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2372db94535dSChris Mason right->map_token = NULL; 2373db94535dSChris Mason } 2374db94535dSChris Mason 23755f39d397SChris Mason btrfs_set_header_nritems(l, mid); 2376aa5d6bedSChris Mason ret = 0; 23775f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 2378db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, right->start, 2379db94535dSChris Mason path->slots[1] + 1, 1); 2380aa5d6bedSChris Mason if (wret) 2381aa5d6bedSChris Mason ret = wret; 23825f39d397SChris Mason 23835f39d397SChris Mason btrfs_mark_buffer_dirty(right); 23845f39d397SChris Mason btrfs_mark_buffer_dirty(l); 2385eb60ceacSChris Mason BUG_ON(path->slots[0] != slot); 23865f39d397SChris Mason 2387be0e5c09SChris Mason if (mid <= slot) { 2388925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 23895f39d397SChris Mason free_extent_buffer(path->nodes[0]); 23905f39d397SChris Mason path->nodes[0] = right; 2391be0e5c09SChris Mason path->slots[0] -= mid; 2392be0e5c09SChris Mason path->slots[1] += 1; 2393925baeddSChris Mason } else { 2394925baeddSChris Mason btrfs_tree_unlock(right); 23955f39d397SChris Mason free_extent_buffer(right); 2396925baeddSChris Mason } 23975f39d397SChris Mason 2398eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 2399d4dbff95SChris Mason 2400cc0c5538SChris Mason if (double_split) { 2401cc0c5538SChris Mason BUG_ON(num_doubles != 0); 2402cc0c5538SChris Mason num_doubles++; 2403cc0c5538SChris Mason goto again; 24043326d1b0SChris Mason } 2405be0e5c09SChris Mason return ret; 2406be0e5c09SChris Mason } 2407be0e5c09SChris Mason 2408b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans, 2409b18c6685SChris Mason struct btrfs_root *root, 2410b18c6685SChris Mason struct btrfs_path *path, 2411179e29e4SChris Mason u32 new_size, int from_end) 2412b18c6685SChris Mason { 2413b18c6685SChris Mason int ret = 0; 2414b18c6685SChris Mason int slot; 2415b18c6685SChris Mason int slot_orig; 24165f39d397SChris Mason struct extent_buffer *leaf; 24175f39d397SChris Mason struct btrfs_item *item; 2418b18c6685SChris Mason u32 nritems; 2419b18c6685SChris Mason unsigned int data_end; 2420b18c6685SChris Mason unsigned int old_data_start; 2421b18c6685SChris Mason unsigned int old_size; 2422b18c6685SChris Mason unsigned int size_diff; 2423b18c6685SChris Mason int i; 2424b18c6685SChris Mason 2425b18c6685SChris Mason slot_orig = path->slots[0]; 24265f39d397SChris Mason leaf = path->nodes[0]; 2427179e29e4SChris Mason slot = path->slots[0]; 2428179e29e4SChris Mason 2429179e29e4SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 2430179e29e4SChris Mason if (old_size == new_size) 2431179e29e4SChris Mason return 0; 2432b18c6685SChris Mason 24335f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2434b18c6685SChris Mason data_end = leaf_data_end(root, leaf); 2435b18c6685SChris Mason 24365f39d397SChris Mason old_data_start = btrfs_item_offset_nr(leaf, slot); 2437179e29e4SChris Mason 2438b18c6685SChris Mason size_diff = old_size - new_size; 2439b18c6685SChris Mason 2440b18c6685SChris Mason BUG_ON(slot < 0); 2441b18c6685SChris Mason BUG_ON(slot >= nritems); 2442b18c6685SChris Mason 2443b18c6685SChris Mason /* 2444b18c6685SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2445b18c6685SChris Mason */ 2446b18c6685SChris Mason /* first correct the data pointers */ 2447b18c6685SChris Mason for (i = slot; i < nritems; i++) { 24485f39d397SChris Mason u32 ioff; 24495f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2450db94535dSChris Mason 2451db94535dSChris Mason if (!leaf->map_token) { 2452db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2453db94535dSChris Mason sizeof(struct btrfs_item), 2454db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2455db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2456db94535dSChris Mason KM_USER1); 2457db94535dSChris Mason } 2458db94535dSChris Mason 24595f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 24605f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + size_diff); 2461b18c6685SChris Mason } 2462db94535dSChris Mason 2463db94535dSChris Mason if (leaf->map_token) { 2464db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2465db94535dSChris Mason leaf->map_token = NULL; 2466db94535dSChris Mason } 2467db94535dSChris Mason 2468b18c6685SChris Mason /* shift the data */ 2469179e29e4SChris Mason if (from_end) { 24705f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2471b18c6685SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 2472b18c6685SChris Mason data_end, old_data_start + new_size - data_end); 2473179e29e4SChris Mason } else { 2474179e29e4SChris Mason struct btrfs_disk_key disk_key; 2475179e29e4SChris Mason u64 offset; 2476179e29e4SChris Mason 2477179e29e4SChris Mason btrfs_item_key(leaf, &disk_key, slot); 2478179e29e4SChris Mason 2479179e29e4SChris Mason if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) { 2480179e29e4SChris Mason unsigned long ptr; 2481179e29e4SChris Mason struct btrfs_file_extent_item *fi; 2482179e29e4SChris Mason 2483179e29e4SChris Mason fi = btrfs_item_ptr(leaf, slot, 2484179e29e4SChris Mason struct btrfs_file_extent_item); 2485179e29e4SChris Mason fi = (struct btrfs_file_extent_item *)( 2486179e29e4SChris Mason (unsigned long)fi - size_diff); 2487179e29e4SChris Mason 2488179e29e4SChris Mason if (btrfs_file_extent_type(leaf, fi) == 2489179e29e4SChris Mason BTRFS_FILE_EXTENT_INLINE) { 2490179e29e4SChris Mason ptr = btrfs_item_ptr_offset(leaf, slot); 2491179e29e4SChris Mason memmove_extent_buffer(leaf, ptr, 2492179e29e4SChris Mason (unsigned long)fi, 2493179e29e4SChris Mason offsetof(struct btrfs_file_extent_item, 2494179e29e4SChris Mason disk_bytenr)); 2495179e29e4SChris Mason } 2496179e29e4SChris Mason } 2497179e29e4SChris Mason 2498179e29e4SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2499179e29e4SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 2500179e29e4SChris Mason data_end, old_data_start - data_end); 2501179e29e4SChris Mason 2502179e29e4SChris Mason offset = btrfs_disk_key_offset(&disk_key); 2503179e29e4SChris Mason btrfs_set_disk_key_offset(&disk_key, offset + size_diff); 2504179e29e4SChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 2505179e29e4SChris Mason if (slot == 0) 2506179e29e4SChris Mason fixup_low_keys(trans, root, path, &disk_key, 1); 2507179e29e4SChris Mason } 25085f39d397SChris Mason 25095f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 25105f39d397SChris Mason btrfs_set_item_size(leaf, item, new_size); 25115f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2512b18c6685SChris Mason 2513b18c6685SChris Mason ret = 0; 25145f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 25155f39d397SChris Mason btrfs_print_leaf(root, leaf); 2516b18c6685SChris Mason BUG(); 25175f39d397SChris Mason } 2518b18c6685SChris Mason return ret; 2519b18c6685SChris Mason } 2520b18c6685SChris Mason 25215f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans, 25225f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 25235f39d397SChris Mason u32 data_size) 25246567e837SChris Mason { 25256567e837SChris Mason int ret = 0; 25266567e837SChris Mason int slot; 25276567e837SChris Mason int slot_orig; 25285f39d397SChris Mason struct extent_buffer *leaf; 25295f39d397SChris Mason struct btrfs_item *item; 25306567e837SChris Mason u32 nritems; 25316567e837SChris Mason unsigned int data_end; 25326567e837SChris Mason unsigned int old_data; 25336567e837SChris Mason unsigned int old_size; 25346567e837SChris Mason int i; 25356567e837SChris Mason 25366567e837SChris Mason slot_orig = path->slots[0]; 25375f39d397SChris Mason leaf = path->nodes[0]; 25386567e837SChris Mason 25395f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 25406567e837SChris Mason data_end = leaf_data_end(root, leaf); 25416567e837SChris Mason 25425f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < data_size) { 25435f39d397SChris Mason btrfs_print_leaf(root, leaf); 25446567e837SChris Mason BUG(); 25455f39d397SChris Mason } 25466567e837SChris Mason slot = path->slots[0]; 25475f39d397SChris Mason old_data = btrfs_item_end_nr(leaf, slot); 25486567e837SChris Mason 25496567e837SChris Mason BUG_ON(slot < 0); 25503326d1b0SChris Mason if (slot >= nritems) { 25513326d1b0SChris Mason btrfs_print_leaf(root, leaf); 25523326d1b0SChris Mason printk("slot %d too large, nritems %d\n", slot, nritems); 25533326d1b0SChris Mason BUG_ON(1); 25543326d1b0SChris Mason } 25556567e837SChris Mason 25566567e837SChris Mason /* 25576567e837SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 25586567e837SChris Mason */ 25596567e837SChris Mason /* first correct the data pointers */ 25606567e837SChris Mason for (i = slot; i < nritems; i++) { 25615f39d397SChris Mason u32 ioff; 25625f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2563db94535dSChris Mason 2564db94535dSChris Mason if (!leaf->map_token) { 2565db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2566db94535dSChris Mason sizeof(struct btrfs_item), 2567db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2568db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2569db94535dSChris Mason KM_USER1); 2570db94535dSChris Mason } 25715f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 25725f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff - data_size); 25736567e837SChris Mason } 25745f39d397SChris Mason 2575db94535dSChris Mason if (leaf->map_token) { 2576db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2577db94535dSChris Mason leaf->map_token = NULL; 2578db94535dSChris Mason } 2579db94535dSChris Mason 25806567e837SChris Mason /* shift the data */ 25815f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 25826567e837SChris Mason data_end - data_size, btrfs_leaf_data(leaf) + 25836567e837SChris Mason data_end, old_data - data_end); 25845f39d397SChris Mason 25856567e837SChris Mason data_end = old_data; 25865f39d397SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 25875f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 25885f39d397SChris Mason btrfs_set_item_size(leaf, item, old_size + data_size); 25895f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 25906567e837SChris Mason 25916567e837SChris Mason ret = 0; 25925f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 25935f39d397SChris Mason btrfs_print_leaf(root, leaf); 25946567e837SChris Mason BUG(); 25955f39d397SChris Mason } 25966567e837SChris Mason return ret; 25976567e837SChris Mason } 25986567e837SChris Mason 259974123bd7SChris Mason /* 260074123bd7SChris Mason * Given a key and some data, insert an item into the tree. 260174123bd7SChris Mason * This does all the path init required, making room in the tree if needed. 260274123bd7SChris Mason */ 26039c58309dSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans, 26045f39d397SChris Mason struct btrfs_root *root, 26055f39d397SChris Mason struct btrfs_path *path, 26069c58309dSChris Mason struct btrfs_key *cpu_key, u32 *data_size, 26079c58309dSChris Mason int nr) 2608be0e5c09SChris Mason { 26095f39d397SChris Mason struct extent_buffer *leaf; 26105f39d397SChris Mason struct btrfs_item *item; 2611aa5d6bedSChris Mason int ret = 0; 2612be0e5c09SChris Mason int slot; 2613eb60ceacSChris Mason int slot_orig; 26149c58309dSChris Mason int i; 26157518a238SChris Mason u32 nritems; 26169c58309dSChris Mason u32 total_size = 0; 26179c58309dSChris Mason u32 total_data = 0; 2618be0e5c09SChris Mason unsigned int data_end; 2619e2fa7227SChris Mason struct btrfs_disk_key disk_key; 2620e2fa7227SChris Mason 26219c58309dSChris Mason for (i = 0; i < nr; i++) { 26229c58309dSChris Mason total_data += data_size[i]; 26239c58309dSChris Mason } 2624be0e5c09SChris Mason 26259c58309dSChris Mason total_size = total_data + (nr - 1) * sizeof(struct btrfs_item); 26269c58309dSChris Mason ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1); 2627eb60ceacSChris Mason if (ret == 0) { 2628f0930a37SChris Mason return -EEXIST; 2629aa5d6bedSChris Mason } 2630ed2ff2cbSChris Mason if (ret < 0) 2631ed2ff2cbSChris Mason goto out; 2632be0e5c09SChris Mason 263362e2749eSChris Mason slot_orig = path->slots[0]; 26345f39d397SChris Mason leaf = path->nodes[0]; 263574123bd7SChris Mason 26365f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2637123abc88SChris Mason data_end = leaf_data_end(root, leaf); 2638eb60ceacSChris Mason 2639123abc88SChris Mason if (btrfs_leaf_free_space(root, leaf) < 26409c58309dSChris Mason sizeof(struct btrfs_item) + total_size) { 26413326d1b0SChris Mason btrfs_print_leaf(root, leaf); 26423326d1b0SChris Mason printk("not enough freespace need %u have %d\n", 26439c58309dSChris Mason total_size, btrfs_leaf_free_space(root, leaf)); 2644be0e5c09SChris Mason BUG(); 2645d4dbff95SChris Mason } 26465f39d397SChris Mason 264762e2749eSChris Mason slot = path->slots[0]; 2648eb60ceacSChris Mason BUG_ON(slot < 0); 26495f39d397SChris Mason 2650be0e5c09SChris Mason if (slot != nritems) { 2651be0e5c09SChris Mason int i; 26525f39d397SChris Mason unsigned int old_data = btrfs_item_end_nr(leaf, slot); 2653be0e5c09SChris Mason 26545f39d397SChris Mason if (old_data < data_end) { 26555f39d397SChris Mason btrfs_print_leaf(root, leaf); 26565f39d397SChris Mason printk("slot %d old_data %d data_end %d\n", 26575f39d397SChris Mason slot, old_data, data_end); 26585f39d397SChris Mason BUG_ON(1); 26595f39d397SChris Mason } 2660be0e5c09SChris Mason /* 2661be0e5c09SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2662be0e5c09SChris Mason */ 2663be0e5c09SChris Mason /* first correct the data pointers */ 2664db94535dSChris Mason WARN_ON(leaf->map_token); 26650783fcfcSChris Mason for (i = slot; i < nritems; i++) { 26665f39d397SChris Mason u32 ioff; 2667db94535dSChris Mason 26685f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2669db94535dSChris Mason if (!leaf->map_token) { 2670db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2671db94535dSChris Mason sizeof(struct btrfs_item), 2672db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2673db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2674db94535dSChris Mason KM_USER1); 2675db94535dSChris Mason } 2676db94535dSChris Mason 26775f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 26789c58309dSChris Mason btrfs_set_item_offset(leaf, item, ioff - total_data); 26790783fcfcSChris Mason } 2680db94535dSChris Mason if (leaf->map_token) { 2681db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2682db94535dSChris Mason leaf->map_token = NULL; 2683db94535dSChris Mason } 2684be0e5c09SChris Mason 2685be0e5c09SChris Mason /* shift the items */ 26869c58309dSChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr), 26875f39d397SChris Mason btrfs_item_nr_offset(slot), 26880783fcfcSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 2689be0e5c09SChris Mason 2690be0e5c09SChris Mason /* shift the data */ 26915f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 26929c58309dSChris Mason data_end - total_data, btrfs_leaf_data(leaf) + 2693be0e5c09SChris Mason data_end, old_data - data_end); 2694be0e5c09SChris Mason data_end = old_data; 2695be0e5c09SChris Mason } 26965f39d397SChris Mason 269762e2749eSChris Mason /* setup the item for the new data */ 26989c58309dSChris Mason for (i = 0; i < nr; i++) { 26999c58309dSChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key + i); 27009c58309dSChris Mason btrfs_set_item_key(leaf, &disk_key, slot + i); 27019c58309dSChris Mason item = btrfs_item_nr(leaf, slot + i); 27029c58309dSChris Mason btrfs_set_item_offset(leaf, item, data_end - data_size[i]); 27039c58309dSChris Mason data_end -= data_size[i]; 27049c58309dSChris Mason btrfs_set_item_size(leaf, item, data_size[i]); 27059c58309dSChris Mason } 27069c58309dSChris Mason btrfs_set_header_nritems(leaf, nritems + nr); 27075f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2708aa5d6bedSChris Mason 2709aa5d6bedSChris Mason ret = 0; 27105a01a2e3SChris Mason if (slot == 0) { 27115a01a2e3SChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key); 2712e089f05cSChris Mason ret = fixup_low_keys(trans, root, path, &disk_key, 1); 27135a01a2e3SChris Mason } 2714aa5d6bedSChris Mason 27155f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 27165f39d397SChris Mason btrfs_print_leaf(root, leaf); 2717be0e5c09SChris Mason BUG(); 27185f39d397SChris Mason } 2719ed2ff2cbSChris Mason out: 272062e2749eSChris Mason return ret; 272162e2749eSChris Mason } 272262e2749eSChris Mason 272362e2749eSChris Mason /* 272462e2749eSChris Mason * Given a key and some data, insert an item into the tree. 272562e2749eSChris Mason * This does all the path init required, making room in the tree if needed. 272662e2749eSChris Mason */ 2727e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root 2728e089f05cSChris Mason *root, struct btrfs_key *cpu_key, void *data, u32 2729e089f05cSChris Mason data_size) 273062e2749eSChris Mason { 273162e2749eSChris Mason int ret = 0; 27322c90e5d6SChris Mason struct btrfs_path *path; 27335f39d397SChris Mason struct extent_buffer *leaf; 27345f39d397SChris Mason unsigned long ptr; 273562e2749eSChris Mason 27362c90e5d6SChris Mason path = btrfs_alloc_path(); 27372c90e5d6SChris Mason BUG_ON(!path); 27382c90e5d6SChris Mason ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size); 273962e2749eSChris Mason if (!ret) { 27405f39d397SChris Mason leaf = path->nodes[0]; 27415f39d397SChris Mason ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); 27425f39d397SChris Mason write_extent_buffer(leaf, data, ptr, data_size); 27435f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 274462e2749eSChris Mason } 27452c90e5d6SChris Mason btrfs_free_path(path); 2746aa5d6bedSChris Mason return ret; 2747be0e5c09SChris Mason } 2748be0e5c09SChris Mason 274974123bd7SChris Mason /* 27505de08d7dSChris Mason * delete the pointer from a given node. 275174123bd7SChris Mason * 275274123bd7SChris Mason * If the delete empties a node, the node is removed from the tree, 275374123bd7SChris Mason * continuing all the way the root if required. The root is converted into 275474123bd7SChris Mason * a leaf if all the nodes are emptied. 275574123bd7SChris Mason */ 2756e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 2757e089f05cSChris Mason struct btrfs_path *path, int level, int slot) 2758be0e5c09SChris Mason { 27595f39d397SChris Mason struct extent_buffer *parent = path->nodes[level]; 27607518a238SChris Mason u32 nritems; 2761aa5d6bedSChris Mason int ret = 0; 2762bb803951SChris Mason int wret; 2763be0e5c09SChris Mason 27645f39d397SChris Mason nritems = btrfs_header_nritems(parent); 2765be0e5c09SChris Mason if (slot != nritems -1) { 27665f39d397SChris Mason memmove_extent_buffer(parent, 27675f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 27685f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 2769d6025579SChris Mason sizeof(struct btrfs_key_ptr) * 2770d6025579SChris Mason (nritems - slot - 1)); 2771be0e5c09SChris Mason } 27727518a238SChris Mason nritems--; 27735f39d397SChris Mason btrfs_set_header_nritems(parent, nritems); 27747518a238SChris Mason if (nritems == 0 && parent == root->node) { 27755f39d397SChris Mason BUG_ON(btrfs_header_level(root->node) != 1); 2776eb60ceacSChris Mason /* just turn the root into a leaf and break */ 27775f39d397SChris Mason btrfs_set_header_level(root->node, 0); 2778bb803951SChris Mason } else if (slot == 0) { 27795f39d397SChris Mason struct btrfs_disk_key disk_key; 27805f39d397SChris Mason 27815f39d397SChris Mason btrfs_node_key(parent, &disk_key, 0); 27825f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, level + 1); 27830f70abe2SChris Mason if (wret) 27840f70abe2SChris Mason ret = wret; 2785be0e5c09SChris Mason } 2786d6025579SChris Mason btrfs_mark_buffer_dirty(parent); 2787aa5d6bedSChris Mason return ret; 2788be0e5c09SChris Mason } 2789be0e5c09SChris Mason 279074123bd7SChris Mason /* 279174123bd7SChris Mason * delete the item at the leaf level in path. If that empties 279274123bd7SChris Mason * the leaf, remove it from the tree 279374123bd7SChris Mason */ 279485e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root, 279585e21bacSChris Mason struct btrfs_path *path, int slot, int nr) 2796be0e5c09SChris Mason { 27975f39d397SChris Mason struct extent_buffer *leaf; 27985f39d397SChris Mason struct btrfs_item *item; 279985e21bacSChris Mason int last_off; 280085e21bacSChris Mason int dsize = 0; 2801aa5d6bedSChris Mason int ret = 0; 2802aa5d6bedSChris Mason int wret; 280385e21bacSChris Mason int i; 28047518a238SChris Mason u32 nritems; 2805be0e5c09SChris Mason 28065f39d397SChris Mason leaf = path->nodes[0]; 280785e21bacSChris Mason last_off = btrfs_item_offset_nr(leaf, slot + nr - 1); 280885e21bacSChris Mason 280985e21bacSChris Mason for (i = 0; i < nr; i++) 281085e21bacSChris Mason dsize += btrfs_item_size_nr(leaf, slot + i); 281185e21bacSChris Mason 28125f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2813be0e5c09SChris Mason 281485e21bacSChris Mason if (slot + nr != nritems) { 2815be0e5c09SChris Mason int i; 2816123abc88SChris Mason int data_end = leaf_data_end(root, leaf); 28175f39d397SChris Mason 28185f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2819d6025579SChris Mason data_end + dsize, 2820123abc88SChris Mason btrfs_leaf_data(leaf) + data_end, 282185e21bacSChris Mason last_off - data_end); 28225f39d397SChris Mason 282385e21bacSChris Mason for (i = slot + nr; i < nritems; i++) { 28245f39d397SChris Mason u32 ioff; 2825db94535dSChris Mason 28265f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2827db94535dSChris Mason if (!leaf->map_token) { 2828db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2829db94535dSChris Mason sizeof(struct btrfs_item), 2830db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2831db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2832db94535dSChris Mason KM_USER1); 2833db94535dSChris Mason } 28345f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 28355f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + dsize); 28360783fcfcSChris Mason } 2837db94535dSChris Mason 2838db94535dSChris Mason if (leaf->map_token) { 2839db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2840db94535dSChris Mason leaf->map_token = NULL; 2841db94535dSChris Mason } 2842db94535dSChris Mason 28435f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot), 284485e21bacSChris Mason btrfs_item_nr_offset(slot + nr), 28450783fcfcSChris Mason sizeof(struct btrfs_item) * 284685e21bacSChris Mason (nritems - slot - nr)); 2847be0e5c09SChris Mason } 284885e21bacSChris Mason btrfs_set_header_nritems(leaf, nritems - nr); 284985e21bacSChris Mason nritems -= nr; 28505f39d397SChris Mason 285174123bd7SChris Mason /* delete the leaf if we've emptied it */ 28527518a238SChris Mason if (nritems == 0) { 28535f39d397SChris Mason if (leaf == root->node) { 28545f39d397SChris Mason btrfs_set_header_level(leaf, 0); 28559a8dd150SChris Mason } else { 28567bb86316SChris Mason u64 root_gen = btrfs_header_generation(path->nodes[1]); 2857e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, path->slots[1]); 2858aa5d6bedSChris Mason if (wret) 2859aa5d6bedSChris Mason ret = wret; 2860e089f05cSChris Mason wret = btrfs_free_extent(trans, root, 28617bb86316SChris Mason leaf->start, leaf->len, 28627bb86316SChris Mason btrfs_header_owner(path->nodes[1]), 28637bb86316SChris Mason root_gen, 0, 0, 1); 28640f70abe2SChris Mason if (wret) 28650f70abe2SChris Mason ret = wret; 28669a8dd150SChris Mason } 2867be0e5c09SChris Mason } else { 28687518a238SChris Mason int used = leaf_space_used(leaf, 0, nritems); 2869aa5d6bedSChris Mason if (slot == 0) { 28705f39d397SChris Mason struct btrfs_disk_key disk_key; 28715f39d397SChris Mason 28725f39d397SChris Mason btrfs_item_key(leaf, &disk_key, 0); 2873e089f05cSChris Mason wret = fixup_low_keys(trans, root, path, 28745f39d397SChris Mason &disk_key, 1); 2875aa5d6bedSChris Mason if (wret) 2876aa5d6bedSChris Mason ret = wret; 2877aa5d6bedSChris Mason } 2878aa5d6bedSChris Mason 287974123bd7SChris Mason /* delete the leaf if it is mostly empty */ 288085e21bacSChris Mason if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) { 2881be0e5c09SChris Mason /* push_leaf_left fixes the path. 2882be0e5c09SChris Mason * make sure the path still points to our leaf 2883be0e5c09SChris Mason * for possible call to del_ptr below 2884be0e5c09SChris Mason */ 28854920c9acSChris Mason slot = path->slots[1]; 28865f39d397SChris Mason extent_buffer_get(leaf); 28875f39d397SChris Mason 288885e21bacSChris Mason wret = push_leaf_left(trans, root, path, 1, 1); 288954aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2890aa5d6bedSChris Mason ret = wret; 28915f39d397SChris Mason 28925f39d397SChris Mason if (path->nodes[0] == leaf && 28935f39d397SChris Mason btrfs_header_nritems(leaf)) { 289485e21bacSChris Mason wret = push_leaf_right(trans, root, path, 1, 1); 289554aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2896aa5d6bedSChris Mason ret = wret; 2897aa5d6bedSChris Mason } 28985f39d397SChris Mason 28995f39d397SChris Mason if (btrfs_header_nritems(leaf) == 0) { 29007bb86316SChris Mason u64 root_gen; 2901db94535dSChris Mason u64 bytenr = leaf->start; 2902db94535dSChris Mason u32 blocksize = leaf->len; 29035f39d397SChris Mason 29047bb86316SChris Mason root_gen = btrfs_header_generation( 29057bb86316SChris Mason path->nodes[1]); 29067bb86316SChris Mason 2907e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, slot); 2908aa5d6bedSChris Mason if (wret) 2909aa5d6bedSChris Mason ret = wret; 29105f39d397SChris Mason 29115f39d397SChris Mason free_extent_buffer(leaf); 2912db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 29137bb86316SChris Mason blocksize, 29147bb86316SChris Mason btrfs_header_owner(path->nodes[1]), 29157bb86316SChris Mason root_gen, 0, 0, 1); 29160f70abe2SChris Mason if (wret) 29170f70abe2SChris Mason ret = wret; 29185de08d7dSChris Mason } else { 2919925baeddSChris Mason /* if we're still in the path, make sure 2920925baeddSChris Mason * we're dirty. Otherwise, one of the 2921925baeddSChris Mason * push_leaf functions must have already 2922925baeddSChris Mason * dirtied this buffer 2923925baeddSChris Mason */ 2924925baeddSChris Mason if (path->nodes[0] == leaf) 29255f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 29265f39d397SChris Mason free_extent_buffer(leaf); 2927be0e5c09SChris Mason } 2928d5719762SChris Mason } else { 29295f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2930be0e5c09SChris Mason } 29319a8dd150SChris Mason } 2932aa5d6bedSChris Mason return ret; 29339a8dd150SChris Mason } 29349a8dd150SChris Mason 293597571fd0SChris Mason /* 2936925baeddSChris Mason * search the tree again to find a leaf with lesser keys 29377bb86316SChris Mason * returns 0 if it found something or 1 if there are no lesser leaves. 29387bb86316SChris Mason * returns < 0 on io errors. 29397bb86316SChris Mason */ 29407bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path) 29417bb86316SChris Mason { 2942925baeddSChris Mason struct btrfs_key key; 2943925baeddSChris Mason struct btrfs_disk_key found_key; 2944925baeddSChris Mason int ret; 29457bb86316SChris Mason 2946925baeddSChris Mason btrfs_item_key_to_cpu(path->nodes[0], &key, 0); 2947925baeddSChris Mason 2948925baeddSChris Mason if (key.offset > 0) 2949925baeddSChris Mason key.offset--; 2950925baeddSChris Mason else if (key.type > 0) 2951925baeddSChris Mason key.type--; 2952925baeddSChris Mason else if (key.objectid > 0) 2953925baeddSChris Mason key.objectid--; 2954925baeddSChris Mason else 29557bb86316SChris Mason return 1; 29567bb86316SChris Mason 2957925baeddSChris Mason btrfs_release_path(root, path); 2958925baeddSChris Mason ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); 2959925baeddSChris Mason if (ret < 0) 2960925baeddSChris Mason return ret; 2961925baeddSChris Mason btrfs_item_key(path->nodes[0], &found_key, 0); 2962925baeddSChris Mason ret = comp_keys(&found_key, &key); 2963925baeddSChris Mason if (ret < 0) 29647bb86316SChris Mason return 0; 2965925baeddSChris Mason return 1; 29667bb86316SChris Mason } 29677bb86316SChris Mason 29683f157a2fSChris Mason /* 29693f157a2fSChris Mason * A helper function to walk down the tree starting at min_key, and looking 29703f157a2fSChris Mason * for nodes or leaves that are either in cache or have a minimum 29713f157a2fSChris Mason * transaction id. This is used by the btree defrag code, but could 29723f157a2fSChris Mason * also be used to search for blocks that have changed since a given 29733f157a2fSChris Mason * transaction id. 29743f157a2fSChris Mason * 29753f157a2fSChris Mason * This does not cow, but it does stuff the starting key it finds back 29763f157a2fSChris Mason * into min_key, so you can call btrfs_search_slot with cow=1 on the 29773f157a2fSChris Mason * key and get a writable path. 29783f157a2fSChris Mason * 29793f157a2fSChris Mason * This does lock as it descends, and path->keep_locks should be set 29803f157a2fSChris Mason * to 1 by the caller. 29813f157a2fSChris Mason * 29823f157a2fSChris Mason * This honors path->lowest_level to prevent descent past a given level 29833f157a2fSChris Mason * of the tree. 29843f157a2fSChris Mason * 29853f157a2fSChris Mason * returns zero if something useful was found, < 0 on error and 1 if there 29863f157a2fSChris Mason * was nothing in the tree that matched the search criteria. 29873f157a2fSChris Mason */ 29883f157a2fSChris Mason int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key, 29893f157a2fSChris Mason struct btrfs_path *path, int cache_only, 29903f157a2fSChris Mason u64 min_trans) 29913f157a2fSChris Mason { 29923f157a2fSChris Mason struct extent_buffer *cur; 29933f157a2fSChris Mason struct btrfs_key found_key; 29943f157a2fSChris Mason int slot; 29953f157a2fSChris Mason u32 nritems; 29963f157a2fSChris Mason int level; 29973f157a2fSChris Mason int ret = 1; 29983f157a2fSChris Mason 29993f157a2fSChris Mason again: 30003f157a2fSChris Mason cur = btrfs_lock_root_node(root); 30013f157a2fSChris Mason level = btrfs_header_level(cur); 30023f157a2fSChris Mason path->nodes[level] = cur; 30033f157a2fSChris Mason path->locks[level] = 1; 30043f157a2fSChris Mason 30053f157a2fSChris Mason if (btrfs_header_generation(cur) < min_trans) { 30063f157a2fSChris Mason ret = 1; 30073f157a2fSChris Mason goto out; 30083f157a2fSChris Mason } 30093f157a2fSChris Mason while(1) { 30103f157a2fSChris Mason nritems = btrfs_header_nritems(cur); 30113f157a2fSChris Mason level = btrfs_header_level(cur); 30123f157a2fSChris Mason bin_search(cur, min_key, level, &slot); 30133f157a2fSChris Mason 30143f157a2fSChris Mason /* at level = 0, we're done, setup the path and exit */ 30153f157a2fSChris Mason if (level == 0) { 30163f157a2fSChris Mason ret = 0; 30173f157a2fSChris Mason path->slots[level] = slot; 30183f157a2fSChris Mason btrfs_item_key_to_cpu(cur, &found_key, slot); 30193f157a2fSChris Mason goto out; 30203f157a2fSChris Mason } 30213f157a2fSChris Mason /* 30223f157a2fSChris Mason * check this node pointer against the cache_only and 30233f157a2fSChris Mason * min_trans parameters. If it isn't in cache or is too 30243f157a2fSChris Mason * old, skip to the next one. 30253f157a2fSChris Mason */ 30263f157a2fSChris Mason while(slot < nritems) { 30273f157a2fSChris Mason u64 blockptr; 30283f157a2fSChris Mason u64 gen; 30293f157a2fSChris Mason struct extent_buffer *tmp; 30303f157a2fSChris Mason blockptr = btrfs_node_blockptr(cur, slot); 30313f157a2fSChris Mason gen = btrfs_node_ptr_generation(cur, slot); 30323f157a2fSChris Mason if (gen < min_trans) { 30333f157a2fSChris Mason slot++; 30343f157a2fSChris Mason continue; 30353f157a2fSChris Mason } 30363f157a2fSChris Mason if (!cache_only) 30373f157a2fSChris Mason break; 30383f157a2fSChris Mason 30393f157a2fSChris Mason tmp = btrfs_find_tree_block(root, blockptr, 30403f157a2fSChris Mason btrfs_level_size(root, level - 1)); 30413f157a2fSChris Mason 30423f157a2fSChris Mason if (tmp && btrfs_buffer_uptodate(tmp, gen)) { 30433f157a2fSChris Mason free_extent_buffer(tmp); 30443f157a2fSChris Mason break; 30453f157a2fSChris Mason } 30463f157a2fSChris Mason if (tmp) 30473f157a2fSChris Mason free_extent_buffer(tmp); 30483f157a2fSChris Mason slot++; 30493f157a2fSChris Mason } 30503f157a2fSChris Mason /* 30513f157a2fSChris Mason * we didn't find a candidate key in this node, walk forward 30523f157a2fSChris Mason * and find another one 30533f157a2fSChris Mason */ 30543f157a2fSChris Mason if (slot >= nritems) { 30553f157a2fSChris Mason ret = btrfs_find_next_key(root, path, min_key, level, 30563f157a2fSChris Mason cache_only, min_trans); 30573f157a2fSChris Mason if (ret == 0) { 30583f157a2fSChris Mason btrfs_release_path(root, path); 30593f157a2fSChris Mason goto again; 30603f157a2fSChris Mason } else { 30613f157a2fSChris Mason goto out; 30623f157a2fSChris Mason } 30633f157a2fSChris Mason } 30643f157a2fSChris Mason /* save our key for returning back */ 30653f157a2fSChris Mason btrfs_node_key_to_cpu(cur, &found_key, slot); 30663f157a2fSChris Mason path->slots[level] = slot; 30673f157a2fSChris Mason if (level == path->lowest_level) { 30683f157a2fSChris Mason ret = 0; 30693f157a2fSChris Mason unlock_up(path, level, 1); 30703f157a2fSChris Mason goto out; 30713f157a2fSChris Mason } 30723f157a2fSChris Mason cur = read_node_slot(root, cur, slot); 30733f157a2fSChris Mason 30743f157a2fSChris Mason btrfs_tree_lock(cur); 30753f157a2fSChris Mason path->locks[level - 1] = 1; 30763f157a2fSChris Mason path->nodes[level - 1] = cur; 30773f157a2fSChris Mason unlock_up(path, level, 1); 30783f157a2fSChris Mason } 30793f157a2fSChris Mason out: 30803f157a2fSChris Mason if (ret == 0) 30813f157a2fSChris Mason memcpy(min_key, &found_key, sizeof(found_key)); 30823f157a2fSChris Mason return ret; 30833f157a2fSChris Mason } 30843f157a2fSChris Mason 30853f157a2fSChris Mason /* 30863f157a2fSChris Mason * this is similar to btrfs_next_leaf, but does not try to preserve 30873f157a2fSChris Mason * and fixup the path. It looks for and returns the next key in the 30883f157a2fSChris Mason * tree based on the current path and the cache_only and min_trans 30893f157a2fSChris Mason * parameters. 30903f157a2fSChris Mason * 30913f157a2fSChris Mason * 0 is returned if another key is found, < 0 if there are any errors 30923f157a2fSChris Mason * and 1 is returned if there are no higher keys in the tree 30933f157a2fSChris Mason * 30943f157a2fSChris Mason * path->keep_locks should be set to 1 on the search made before 30953f157a2fSChris Mason * calling this function. 30963f157a2fSChris Mason */ 3097e7a84565SChris Mason int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path, 30983f157a2fSChris Mason struct btrfs_key *key, int lowest_level, 30993f157a2fSChris Mason int cache_only, u64 min_trans) 3100e7a84565SChris Mason { 3101e7a84565SChris Mason int level = lowest_level; 3102e7a84565SChris Mason int slot; 3103e7a84565SChris Mason struct extent_buffer *c; 3104e7a84565SChris Mason 3105e7a84565SChris Mason while(level < BTRFS_MAX_LEVEL) { 3106e7a84565SChris Mason if (!path->nodes[level]) 3107e7a84565SChris Mason return 1; 3108e7a84565SChris Mason 3109e7a84565SChris Mason slot = path->slots[level] + 1; 3110e7a84565SChris Mason c = path->nodes[level]; 31113f157a2fSChris Mason next: 3112e7a84565SChris Mason if (slot >= btrfs_header_nritems(c)) { 3113e7a84565SChris Mason level++; 3114e7a84565SChris Mason if (level == BTRFS_MAX_LEVEL) { 3115e7a84565SChris Mason return 1; 3116e7a84565SChris Mason } 3117e7a84565SChris Mason continue; 3118e7a84565SChris Mason } 3119e7a84565SChris Mason if (level == 0) 3120e7a84565SChris Mason btrfs_item_key_to_cpu(c, key, slot); 31213f157a2fSChris Mason else { 31223f157a2fSChris Mason u64 blockptr = btrfs_node_blockptr(c, slot); 31233f157a2fSChris Mason u64 gen = btrfs_node_ptr_generation(c, slot); 31243f157a2fSChris Mason 31253f157a2fSChris Mason if (cache_only) { 31263f157a2fSChris Mason struct extent_buffer *cur; 31273f157a2fSChris Mason cur = btrfs_find_tree_block(root, blockptr, 31283f157a2fSChris Mason btrfs_level_size(root, level - 1)); 31293f157a2fSChris Mason if (!cur || !btrfs_buffer_uptodate(cur, gen)) { 31303f157a2fSChris Mason slot++; 31313f157a2fSChris Mason if (cur) 31323f157a2fSChris Mason free_extent_buffer(cur); 31333f157a2fSChris Mason goto next; 31343f157a2fSChris Mason } 31353f157a2fSChris Mason free_extent_buffer(cur); 31363f157a2fSChris Mason } 31373f157a2fSChris Mason if (gen < min_trans) { 31383f157a2fSChris Mason slot++; 31393f157a2fSChris Mason goto next; 31403f157a2fSChris Mason } 3141e7a84565SChris Mason btrfs_node_key_to_cpu(c, key, slot); 31423f157a2fSChris Mason } 3143e7a84565SChris Mason return 0; 3144e7a84565SChris Mason } 3145e7a84565SChris Mason return 1; 3146e7a84565SChris Mason } 3147e7a84565SChris Mason 31487bb86316SChris Mason /* 3149925baeddSChris Mason * search the tree again to find a leaf with greater keys 31500f70abe2SChris Mason * returns 0 if it found something or 1 if there are no greater leaves. 31510f70abe2SChris Mason * returns < 0 on io errors. 315297571fd0SChris Mason */ 3153234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path) 3154d97e63b6SChris Mason { 3155d97e63b6SChris Mason int slot; 3156d97e63b6SChris Mason int level = 1; 31575f39d397SChris Mason struct extent_buffer *c; 31585f39d397SChris Mason struct extent_buffer *next = NULL; 3159925baeddSChris Mason struct btrfs_key key; 3160925baeddSChris Mason u32 nritems; 3161925baeddSChris Mason int ret; 3162925baeddSChris Mason 3163925baeddSChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 3164925baeddSChris Mason if (nritems == 0) { 3165925baeddSChris Mason return 1; 3166925baeddSChris Mason } 3167925baeddSChris Mason 3168925baeddSChris Mason btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1); 3169925baeddSChris Mason 3170925baeddSChris Mason btrfs_release_path(root, path); 3171a2135011SChris Mason path->keep_locks = 1; 3172925baeddSChris Mason ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); 3173925baeddSChris Mason path->keep_locks = 0; 3174925baeddSChris Mason 3175925baeddSChris Mason if (ret < 0) 3176925baeddSChris Mason return ret; 3177925baeddSChris Mason 3178a2135011SChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 3179168fd7d2SChris Mason /* 3180168fd7d2SChris Mason * by releasing the path above we dropped all our locks. A balance 3181168fd7d2SChris Mason * could have added more items next to the key that used to be 3182168fd7d2SChris Mason * at the very end of the block. So, check again here and 3183168fd7d2SChris Mason * advance the path if there are now more items available. 3184168fd7d2SChris Mason */ 3185a2135011SChris Mason if (nritems > 0 && path->slots[0] < nritems - 1) { 3186168fd7d2SChris Mason path->slots[0]++; 3187925baeddSChris Mason goto done; 3188925baeddSChris Mason } 3189d97e63b6SChris Mason 3190234b63a0SChris Mason while(level < BTRFS_MAX_LEVEL) { 3191d97e63b6SChris Mason if (!path->nodes[level]) 31920f70abe2SChris Mason return 1; 31935f39d397SChris Mason 3194d97e63b6SChris Mason slot = path->slots[level] + 1; 3195d97e63b6SChris Mason c = path->nodes[level]; 31965f39d397SChris Mason if (slot >= btrfs_header_nritems(c)) { 3197d97e63b6SChris Mason level++; 3198925baeddSChris Mason if (level == BTRFS_MAX_LEVEL) { 31997bb86316SChris Mason return 1; 3200925baeddSChris Mason } 3201d97e63b6SChris Mason continue; 3202d97e63b6SChris Mason } 32035f39d397SChris Mason 3204925baeddSChris Mason if (next) { 3205925baeddSChris Mason btrfs_tree_unlock(next); 32065f39d397SChris Mason free_extent_buffer(next); 3207925baeddSChris Mason } 32085f39d397SChris Mason 3209*0bd40a71SChris Mason if (level == 1 && (path->locks[1] || path->skip_locking) && 3210*0bd40a71SChris Mason path->reada) 321101f46658SChris Mason reada_for_search(root, path, level, slot, 0); 32125f39d397SChris Mason 3213ca7a79adSChris Mason next = read_node_slot(root, c, slot); 32145cd57b2cSChris Mason if (!path->skip_locking) { 3215051e1b9fSChris Mason WARN_ON(!btrfs_tree_locked(c)); 3216925baeddSChris Mason btrfs_tree_lock(next); 32175cd57b2cSChris Mason } 3218d97e63b6SChris Mason break; 3219d97e63b6SChris Mason } 3220d97e63b6SChris Mason path->slots[level] = slot; 3221d97e63b6SChris Mason while(1) { 3222d97e63b6SChris Mason level--; 3223d97e63b6SChris Mason c = path->nodes[level]; 3224925baeddSChris Mason if (path->locks[level]) 3225925baeddSChris Mason btrfs_tree_unlock(c); 32265f39d397SChris Mason free_extent_buffer(c); 3227d97e63b6SChris Mason path->nodes[level] = next; 3228d97e63b6SChris Mason path->slots[level] = 0; 3229a74a4b97SChris Mason if (!path->skip_locking) 3230925baeddSChris Mason path->locks[level] = 1; 3231d97e63b6SChris Mason if (!level) 3232d97e63b6SChris Mason break; 3233925baeddSChris Mason if (level == 1 && path->locks[1] && path->reada) 3234925baeddSChris Mason reada_for_search(root, path, level, slot, 0); 3235ca7a79adSChris Mason next = read_node_slot(root, next, 0); 32365cd57b2cSChris Mason if (!path->skip_locking) { 3237a2135011SChris Mason WARN_ON(!btrfs_tree_locked(path->nodes[level])); 3238925baeddSChris Mason btrfs_tree_lock(next); 3239d97e63b6SChris Mason } 32405cd57b2cSChris Mason } 3241925baeddSChris Mason done: 3242925baeddSChris Mason unlock_up(path, 0, 1); 3243d97e63b6SChris Mason return 0; 3244d97e63b6SChris Mason } 32450b86a832SChris Mason 32463f157a2fSChris Mason /* 32473f157a2fSChris Mason * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps 32483f157a2fSChris Mason * searching until it gets past min_objectid or finds an item of 'type' 32493f157a2fSChris Mason * 32503f157a2fSChris Mason * returns 0 if something is found, 1 if nothing was found and < 0 on error 32513f157a2fSChris Mason */ 32520b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root, 32530b86a832SChris Mason struct btrfs_path *path, u64 min_objectid, 32540b86a832SChris Mason int type) 32550b86a832SChris Mason { 32560b86a832SChris Mason struct btrfs_key found_key; 32570b86a832SChris Mason struct extent_buffer *leaf; 32580b86a832SChris Mason int ret; 32590b86a832SChris Mason 32600b86a832SChris Mason while(1) { 32610b86a832SChris Mason if (path->slots[0] == 0) { 32620b86a832SChris Mason ret = btrfs_prev_leaf(root, path); 32630b86a832SChris Mason if (ret != 0) 32640b86a832SChris Mason return ret; 32650b86a832SChris Mason } else { 32660b86a832SChris Mason path->slots[0]--; 32670b86a832SChris Mason } 32680b86a832SChris Mason leaf = path->nodes[0]; 32690b86a832SChris Mason btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); 32700b86a832SChris Mason if (found_key.type == type) 32710b86a832SChris Mason return 0; 32720b86a832SChris Mason } 32730b86a832SChris Mason return 1; 32740b86a832SChris Mason } 32750b86a832SChris Mason 3276