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; 66*a2135011SChris Mason int skip = p->skip_locking; 67*a2135011SChris Mason int keep = p->keep_locks; 68*a2135011SChris Mason 69234b63a0SChris Mason for (i = 0; i < BTRFS_MAX_LEVEL; i++) { 70eb60ceacSChris Mason if (!p->nodes[i]) 71925baeddSChris Mason continue; 72925baeddSChris Mason if (p->locks[i]) { 73925baeddSChris Mason btrfs_tree_unlock(p->nodes[i]); 74925baeddSChris Mason p->locks[i] = 0; 75925baeddSChris Mason } 765f39d397SChris Mason free_extent_buffer(p->nodes[i]); 77eb60ceacSChris Mason } 78aa5d6bedSChris Mason memset(p, 0, sizeof(*p)); 79*a2135011SChris Mason p->skip_locking = skip; 80*a2135011SChris Mason p->keep_locks = keep; 81eb60ceacSChris Mason } 82eb60ceacSChris Mason 83925baeddSChris Mason struct extent_buffer *btrfs_root_node(struct btrfs_root *root) 84925baeddSChris Mason { 85925baeddSChris Mason struct extent_buffer *eb; 86925baeddSChris Mason spin_lock(&root->node_lock); 87925baeddSChris Mason eb = root->node; 88925baeddSChris Mason extent_buffer_get(eb); 89925baeddSChris Mason spin_unlock(&root->node_lock); 90925baeddSChris Mason return eb; 91925baeddSChris Mason } 92925baeddSChris Mason 93925baeddSChris Mason struct extent_buffer *btrfs_lock_root_node(struct btrfs_root *root) 94925baeddSChris Mason { 95925baeddSChris Mason struct extent_buffer *eb; 96925baeddSChris Mason 97925baeddSChris Mason while(1) { 98925baeddSChris Mason eb = btrfs_root_node(root); 99925baeddSChris Mason btrfs_tree_lock(eb); 100925baeddSChris Mason 101925baeddSChris Mason spin_lock(&root->node_lock); 102925baeddSChris Mason if (eb == root->node) { 103925baeddSChris Mason spin_unlock(&root->node_lock); 104925baeddSChris Mason break; 105925baeddSChris Mason } 106925baeddSChris Mason spin_unlock(&root->node_lock); 107925baeddSChris Mason 108925baeddSChris Mason btrfs_tree_unlock(eb); 109925baeddSChris Mason free_extent_buffer(eb); 110925baeddSChris Mason } 111925baeddSChris Mason return eb; 112925baeddSChris Mason } 113925baeddSChris Mason 1140b86a832SChris Mason static void add_root_to_dirty_list(struct btrfs_root *root) 1150b86a832SChris Mason { 1160b86a832SChris Mason if (root->track_dirty && list_empty(&root->dirty_list)) { 1170b86a832SChris Mason list_add(&root->dirty_list, 1180b86a832SChris Mason &root->fs_info->dirty_cowonly_roots); 1190b86a832SChris Mason } 1200b86a832SChris Mason } 1210b86a832SChris Mason 122be20aa9dSChris Mason int btrfs_copy_root(struct btrfs_trans_handle *trans, 123be20aa9dSChris Mason struct btrfs_root *root, 124be20aa9dSChris Mason struct extent_buffer *buf, 125be20aa9dSChris Mason struct extent_buffer **cow_ret, u64 new_root_objectid) 126be20aa9dSChris Mason { 127be20aa9dSChris Mason struct extent_buffer *cow; 128be20aa9dSChris Mason u32 nritems; 129be20aa9dSChris Mason int ret = 0; 130be20aa9dSChris Mason int level; 131be20aa9dSChris Mason struct btrfs_key first_key; 1324aec2b52SChris Mason struct btrfs_root *new_root; 133be20aa9dSChris Mason 1344aec2b52SChris Mason new_root = kmalloc(sizeof(*new_root), GFP_NOFS); 1354aec2b52SChris Mason if (!new_root) 1364aec2b52SChris Mason return -ENOMEM; 1374aec2b52SChris Mason 1384aec2b52SChris Mason memcpy(new_root, root, sizeof(*new_root)); 1394aec2b52SChris Mason new_root->root_key.objectid = new_root_objectid; 140be20aa9dSChris Mason 141be20aa9dSChris Mason WARN_ON(root->ref_cows && trans->transid != 142be20aa9dSChris Mason root->fs_info->running_transaction->transid); 143be20aa9dSChris Mason WARN_ON(root->ref_cows && trans->transid != root->last_trans); 144be20aa9dSChris Mason 145be20aa9dSChris Mason level = btrfs_header_level(buf); 146be20aa9dSChris Mason nritems = btrfs_header_nritems(buf); 147be20aa9dSChris Mason if (nritems) { 148be20aa9dSChris Mason if (level == 0) 149be20aa9dSChris Mason btrfs_item_key_to_cpu(buf, &first_key, 0); 150be20aa9dSChris Mason else 151be20aa9dSChris Mason btrfs_node_key_to_cpu(buf, &first_key, 0); 152be20aa9dSChris Mason } else { 153be20aa9dSChris Mason first_key.objectid = 0; 154be20aa9dSChris Mason } 155925baeddSChris Mason cow = btrfs_alloc_free_block(trans, new_root, buf->len, 156be20aa9dSChris Mason new_root_objectid, 157be20aa9dSChris Mason trans->transid, first_key.objectid, 158be20aa9dSChris Mason level, buf->start, 0); 1594aec2b52SChris Mason if (IS_ERR(cow)) { 1604aec2b52SChris Mason kfree(new_root); 161be20aa9dSChris Mason return PTR_ERR(cow); 1624aec2b52SChris Mason } 163be20aa9dSChris Mason 164be20aa9dSChris Mason copy_extent_buffer(cow, buf, 0, 0, cow->len); 165be20aa9dSChris Mason btrfs_set_header_bytenr(cow, cow->start); 166be20aa9dSChris Mason btrfs_set_header_generation(cow, trans->transid); 167be20aa9dSChris Mason btrfs_set_header_owner(cow, new_root_objectid); 16863b10fc4SChris Mason btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN); 169be20aa9dSChris Mason 170be20aa9dSChris Mason WARN_ON(btrfs_header_generation(buf) > trans->transid); 1714aec2b52SChris Mason ret = btrfs_inc_ref(trans, new_root, buf); 1724aec2b52SChris Mason kfree(new_root); 1734aec2b52SChris Mason 174be20aa9dSChris Mason if (ret) 175be20aa9dSChris Mason return ret; 176be20aa9dSChris Mason 177be20aa9dSChris Mason btrfs_mark_buffer_dirty(cow); 178be20aa9dSChris Mason *cow_ret = cow; 179be20aa9dSChris Mason return 0; 180be20aa9dSChris Mason } 181be20aa9dSChris Mason 182be20aa9dSChris Mason int __btrfs_cow_block(struct btrfs_trans_handle *trans, 1835f39d397SChris Mason struct btrfs_root *root, 1845f39d397SChris Mason struct extent_buffer *buf, 1855f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 1865f39d397SChris Mason struct extent_buffer **cow_ret, 1875f39d397SChris Mason u64 search_start, u64 empty_size) 1886702ed49SChris Mason { 1897bb86316SChris Mason u64 root_gen; 1905f39d397SChris Mason struct extent_buffer *cow; 1917bb86316SChris Mason u32 nritems; 1926702ed49SChris Mason int ret = 0; 1936702ed49SChris Mason int different_trans = 0; 1947bb86316SChris Mason int level; 195925baeddSChris Mason int unlock_orig = 0; 1967bb86316SChris Mason struct btrfs_key first_key; 1976702ed49SChris Mason 198925baeddSChris Mason if (*cow_ret == buf) 199925baeddSChris Mason unlock_orig = 1; 200925baeddSChris Mason 201925baeddSChris Mason WARN_ON(!btrfs_tree_locked(buf)); 202925baeddSChris Mason 2037bb86316SChris Mason if (root->ref_cows) { 2047bb86316SChris Mason root_gen = trans->transid; 2057bb86316SChris Mason } else { 2067bb86316SChris Mason root_gen = 0; 2077bb86316SChris Mason } 2087bb86316SChris Mason WARN_ON(root->ref_cows && trans->transid != 2097bb86316SChris Mason root->fs_info->running_transaction->transid); 2106702ed49SChris Mason WARN_ON(root->ref_cows && trans->transid != root->last_trans); 2115f39d397SChris Mason 2127bb86316SChris Mason level = btrfs_header_level(buf); 2137bb86316SChris Mason nritems = btrfs_header_nritems(buf); 2147bb86316SChris Mason if (nritems) { 2157bb86316SChris Mason if (level == 0) 2167bb86316SChris Mason btrfs_item_key_to_cpu(buf, &first_key, 0); 2177bb86316SChris Mason else 2187bb86316SChris Mason btrfs_node_key_to_cpu(buf, &first_key, 0); 2197bb86316SChris Mason } else { 2207bb86316SChris Mason first_key.objectid = 0; 2217bb86316SChris Mason } 222925baeddSChris Mason cow = btrfs_alloc_free_block(trans, root, buf->len, 2237bb86316SChris Mason root->root_key.objectid, 2247bb86316SChris Mason root_gen, first_key.objectid, level, 225db94535dSChris Mason search_start, empty_size); 2266702ed49SChris Mason if (IS_ERR(cow)) 2276702ed49SChris Mason return PTR_ERR(cow); 2286702ed49SChris Mason 2295f39d397SChris Mason copy_extent_buffer(cow, buf, 0, 0, cow->len); 230db94535dSChris Mason btrfs_set_header_bytenr(cow, cow->start); 2315f39d397SChris Mason btrfs_set_header_generation(cow, trans->transid); 2325f39d397SChris Mason btrfs_set_header_owner(cow, root->root_key.objectid); 23363b10fc4SChris Mason btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN); 2346702ed49SChris Mason 2355f39d397SChris Mason WARN_ON(btrfs_header_generation(buf) > trans->transid); 2365f39d397SChris Mason if (btrfs_header_generation(buf) != trans->transid) { 2376702ed49SChris Mason different_trans = 1; 2386702ed49SChris Mason ret = btrfs_inc_ref(trans, root, buf); 2396702ed49SChris Mason if (ret) 2406702ed49SChris Mason return ret; 2416702ed49SChris Mason } else { 2426702ed49SChris Mason clean_tree_block(trans, root, buf); 2436702ed49SChris Mason } 2446702ed49SChris Mason 2456702ed49SChris Mason if (buf == root->node) { 246925baeddSChris Mason WARN_ON(parent && parent != buf); 2477bb86316SChris Mason root_gen = btrfs_header_generation(buf); 248925baeddSChris Mason 249925baeddSChris Mason spin_lock(&root->node_lock); 2506702ed49SChris Mason root->node = cow; 2515f39d397SChris Mason extent_buffer_get(cow); 252925baeddSChris Mason spin_unlock(&root->node_lock); 253925baeddSChris Mason 2546702ed49SChris Mason if (buf != root->commit_root) { 255db94535dSChris Mason btrfs_free_extent(trans, root, buf->start, 2567bb86316SChris Mason buf->len, root->root_key.objectid, 2577bb86316SChris Mason root_gen, 0, 0, 1); 2586702ed49SChris Mason } 2595f39d397SChris Mason free_extent_buffer(buf); 2600b86a832SChris Mason add_root_to_dirty_list(root); 2616702ed49SChris Mason } else { 2627bb86316SChris Mason root_gen = btrfs_header_generation(parent); 2635f39d397SChris Mason btrfs_set_node_blockptr(parent, parent_slot, 264db94535dSChris Mason cow->start); 26574493f7aSChris Mason WARN_ON(trans->transid == 0); 26674493f7aSChris Mason btrfs_set_node_ptr_generation(parent, parent_slot, 26774493f7aSChris Mason trans->transid); 2686702ed49SChris Mason btrfs_mark_buffer_dirty(parent); 2695f39d397SChris Mason WARN_ON(btrfs_header_generation(parent) != trans->transid); 2707bb86316SChris Mason btrfs_free_extent(trans, root, buf->start, buf->len, 2717bb86316SChris Mason btrfs_header_owner(parent), root_gen, 2727bb86316SChris Mason 0, 0, 1); 2736702ed49SChris Mason } 274925baeddSChris Mason if (unlock_orig) 275925baeddSChris Mason btrfs_tree_unlock(buf); 2765f39d397SChris Mason free_extent_buffer(buf); 2776702ed49SChris Mason btrfs_mark_buffer_dirty(cow); 2786702ed49SChris Mason *cow_ret = cow; 2796702ed49SChris Mason return 0; 2806702ed49SChris Mason } 2816702ed49SChris Mason 2825f39d397SChris Mason int btrfs_cow_block(struct btrfs_trans_handle *trans, 2835f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *buf, 2845f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 2855f39d397SChris Mason struct extent_buffer **cow_ret) 28602217ed2SChris Mason { 2876702ed49SChris Mason u64 search_start; 288dc17ff8fSChris Mason u64 header_trans; 289f510cfecSChris Mason int ret; 290dc17ff8fSChris Mason 291ccd467d6SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 292ccd467d6SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 293ccd467d6SChris Mason root->fs_info->running_transaction->transid); 294ccd467d6SChris Mason WARN_ON(1); 295ccd467d6SChris Mason } 296ccd467d6SChris Mason if (trans->transid != root->fs_info->generation) { 297ccd467d6SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 298ccd467d6SChris Mason root->fs_info->generation); 299ccd467d6SChris Mason WARN_ON(1); 300ccd467d6SChris Mason } 301dc17ff8fSChris Mason 302dc17ff8fSChris Mason header_trans = btrfs_header_generation(buf); 30363b10fc4SChris Mason spin_lock(&root->fs_info->hash_lock); 30463b10fc4SChris Mason if (header_trans == trans->transid && 30563b10fc4SChris Mason !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN)) { 30602217ed2SChris Mason *cow_ret = buf; 30763b10fc4SChris Mason spin_unlock(&root->fs_info->hash_lock); 30802217ed2SChris Mason return 0; 30902217ed2SChris Mason } 31063b10fc4SChris Mason spin_unlock(&root->fs_info->hash_lock); 3110b86a832SChris Mason search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1); 312f510cfecSChris Mason ret = __btrfs_cow_block(trans, root, buf, parent, 3136702ed49SChris Mason parent_slot, cow_ret, search_start, 0); 314f510cfecSChris Mason return ret; 3152c90e5d6SChris Mason } 3166702ed49SChris Mason 3176b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize) 3186702ed49SChris Mason { 3196b80053dSChris Mason if (blocknr < other && other - (blocknr + blocksize) < 32768) 3206702ed49SChris Mason return 1; 3216b80053dSChris Mason if (blocknr > other && blocknr - (other + blocksize) < 32768) 3226702ed49SChris Mason return 1; 32302217ed2SChris Mason return 0; 32402217ed2SChris Mason } 32502217ed2SChris Mason 326081e9573SChris Mason /* 327081e9573SChris Mason * compare two keys in a memcmp fashion 328081e9573SChris Mason */ 329081e9573SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2) 330081e9573SChris Mason { 331081e9573SChris Mason struct btrfs_key k1; 332081e9573SChris Mason 333081e9573SChris Mason btrfs_disk_key_to_cpu(&k1, disk); 334081e9573SChris Mason 335081e9573SChris Mason if (k1.objectid > k2->objectid) 336081e9573SChris Mason return 1; 337081e9573SChris Mason if (k1.objectid < k2->objectid) 338081e9573SChris Mason return -1; 339081e9573SChris Mason if (k1.type > k2->type) 340081e9573SChris Mason return 1; 341081e9573SChris Mason if (k1.type < k2->type) 342081e9573SChris Mason return -1; 343081e9573SChris Mason if (k1.offset > k2->offset) 344081e9573SChris Mason return 1; 345081e9573SChris Mason if (k1.offset < k2->offset) 346081e9573SChris Mason return -1; 347081e9573SChris Mason return 0; 348081e9573SChris Mason } 349081e9573SChris Mason 350081e9573SChris Mason 3516702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans, 3525f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *parent, 353a6b6e75eSChris Mason int start_slot, int cache_only, u64 *last_ret, 354a6b6e75eSChris Mason struct btrfs_key *progress) 3556702ed49SChris Mason { 3566b80053dSChris Mason struct extent_buffer *cur; 3576b80053dSChris Mason struct extent_buffer *tmp; 3586702ed49SChris Mason u64 blocknr; 359ca7a79adSChris Mason u64 gen; 360e9d0b13bSChris Mason u64 search_start = *last_ret; 361e9d0b13bSChris Mason u64 last_block = 0; 3626702ed49SChris Mason u64 other; 3636702ed49SChris Mason u32 parent_nritems; 3646702ed49SChris Mason int end_slot; 3656702ed49SChris Mason int i; 3666702ed49SChris Mason int err = 0; 367f2183bdeSChris Mason int parent_level; 3686b80053dSChris Mason int uptodate; 3696b80053dSChris Mason u32 blocksize; 370081e9573SChris Mason int progress_passed = 0; 371081e9573SChris Mason struct btrfs_disk_key disk_key; 3726702ed49SChris Mason 373925baeddSChris Mason /* FIXME this code needs locking */ 374925baeddSChris Mason return 0; 375925baeddSChris Mason 3765708b959SChris Mason parent_level = btrfs_header_level(parent); 3775708b959SChris Mason if (cache_only && parent_level != 1) 3785708b959SChris Mason return 0; 3795708b959SChris Mason 3806702ed49SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 3816702ed49SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 3826702ed49SChris Mason root->fs_info->running_transaction->transid); 3836702ed49SChris Mason WARN_ON(1); 3846702ed49SChris Mason } 3856702ed49SChris Mason if (trans->transid != root->fs_info->generation) { 3866702ed49SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 3876702ed49SChris Mason root->fs_info->generation); 3886702ed49SChris Mason WARN_ON(1); 3896702ed49SChris Mason } 39086479a04SChris Mason 3916b80053dSChris Mason parent_nritems = btrfs_header_nritems(parent); 3926b80053dSChris Mason blocksize = btrfs_level_size(root, parent_level - 1); 3936702ed49SChris Mason end_slot = parent_nritems; 3946702ed49SChris Mason 3956702ed49SChris Mason if (parent_nritems == 1) 3966702ed49SChris Mason return 0; 3976702ed49SChris Mason 3986702ed49SChris Mason for (i = start_slot; i < end_slot; i++) { 3996702ed49SChris Mason int close = 1; 400a6b6e75eSChris Mason 4015708b959SChris Mason if (!parent->map_token) { 4025708b959SChris Mason map_extent_buffer(parent, 4035708b959SChris Mason btrfs_node_key_ptr_offset(i), 4045708b959SChris Mason sizeof(struct btrfs_key_ptr), 4055708b959SChris Mason &parent->map_token, &parent->kaddr, 4065708b959SChris Mason &parent->map_start, &parent->map_len, 4075708b959SChris Mason KM_USER1); 4085708b959SChris Mason } 409081e9573SChris Mason btrfs_node_key(parent, &disk_key, i); 410081e9573SChris Mason if (!progress_passed && comp_keys(&disk_key, progress) < 0) 411081e9573SChris Mason continue; 412081e9573SChris Mason 413081e9573SChris Mason progress_passed = 1; 4146b80053dSChris Mason blocknr = btrfs_node_blockptr(parent, i); 415ca7a79adSChris Mason gen = btrfs_node_ptr_generation(parent, i); 416e9d0b13bSChris Mason if (last_block == 0) 417e9d0b13bSChris Mason last_block = blocknr; 4185708b959SChris Mason 4196702ed49SChris Mason if (i > 0) { 4206b80053dSChris Mason other = btrfs_node_blockptr(parent, i - 1); 4216b80053dSChris Mason close = close_blocks(blocknr, other, blocksize); 4226702ed49SChris Mason } 4230ef3e66bSChris Mason if (!close && i < end_slot - 2) { 4246b80053dSChris Mason other = btrfs_node_blockptr(parent, i + 1); 4256b80053dSChris Mason close = close_blocks(blocknr, other, blocksize); 4266702ed49SChris Mason } 427e9d0b13bSChris Mason if (close) { 428e9d0b13bSChris Mason last_block = blocknr; 4296702ed49SChris Mason continue; 430e9d0b13bSChris Mason } 4315708b959SChris Mason if (parent->map_token) { 4325708b959SChris Mason unmap_extent_buffer(parent, parent->map_token, 4335708b959SChris Mason KM_USER1); 4345708b959SChris Mason parent->map_token = NULL; 4355708b959SChris Mason } 4366702ed49SChris Mason 4376b80053dSChris Mason cur = btrfs_find_tree_block(root, blocknr, blocksize); 4386b80053dSChris Mason if (cur) 4391259ab75SChris Mason uptodate = btrfs_buffer_uptodate(cur, gen); 4406b80053dSChris Mason else 4416b80053dSChris Mason uptodate = 0; 4425708b959SChris Mason if (!cur || !uptodate) { 4436702ed49SChris Mason if (cache_only) { 4446b80053dSChris Mason free_extent_buffer(cur); 4456702ed49SChris Mason continue; 4466702ed49SChris Mason } 4476b80053dSChris Mason if (!cur) { 4486b80053dSChris Mason cur = read_tree_block(root, blocknr, 449ca7a79adSChris Mason blocksize, gen); 4506b80053dSChris Mason } else if (!uptodate) { 451ca7a79adSChris Mason btrfs_read_buffer(cur, gen); 4526702ed49SChris Mason } 453f2183bdeSChris Mason } 454e9d0b13bSChris Mason if (search_start == 0) 4556b80053dSChris Mason search_start = last_block; 456e9d0b13bSChris Mason 4576b80053dSChris Mason err = __btrfs_cow_block(trans, root, cur, parent, i, 4586b80053dSChris Mason &tmp, search_start, 4596b80053dSChris Mason min(16 * blocksize, 4606b80053dSChris Mason (end_slot - i) * blocksize)); 461252c38f0SYan if (err) { 4626b80053dSChris Mason free_extent_buffer(cur); 4636702ed49SChris Mason break; 464252c38f0SYan } 4656b80053dSChris Mason search_start = tmp->start; 4665708b959SChris Mason last_block = tmp->start; 467f2183bdeSChris Mason *last_ret = search_start; 468f2183bdeSChris Mason if (parent_level == 1) 4696b80053dSChris Mason btrfs_clear_buffer_defrag(tmp); 4706b80053dSChris Mason free_extent_buffer(tmp); 4716702ed49SChris Mason } 4725708b959SChris Mason if (parent->map_token) { 4735708b959SChris Mason unmap_extent_buffer(parent, parent->map_token, 4745708b959SChris Mason KM_USER1); 4755708b959SChris Mason parent->map_token = NULL; 4765708b959SChris Mason } 4776702ed49SChris Mason return err; 4786702ed49SChris Mason } 4796702ed49SChris Mason 48074123bd7SChris Mason /* 48174123bd7SChris Mason * The leaf data grows from end-to-front in the node. 48274123bd7SChris Mason * this returns the address of the start of the last item, 48374123bd7SChris Mason * which is the stop of the leaf data stack 48474123bd7SChris Mason */ 485123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root, 4865f39d397SChris Mason struct extent_buffer *leaf) 487be0e5c09SChris Mason { 4885f39d397SChris Mason u32 nr = btrfs_header_nritems(leaf); 489be0e5c09SChris Mason if (nr == 0) 490123abc88SChris Mason return BTRFS_LEAF_DATA_SIZE(root); 4915f39d397SChris Mason return btrfs_item_offset_nr(leaf, nr - 1); 492be0e5c09SChris Mason } 493be0e5c09SChris Mason 494123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path, 495123abc88SChris Mason int level) 496aa5d6bedSChris Mason { 4975f39d397SChris Mason struct extent_buffer *parent = NULL; 4985f39d397SChris Mason struct extent_buffer *node = path->nodes[level]; 4995f39d397SChris Mason struct btrfs_disk_key parent_key; 5005f39d397SChris Mason struct btrfs_disk_key node_key; 501aa5d6bedSChris Mason int parent_slot; 5028d7be552SChris Mason int slot; 5038d7be552SChris Mason struct btrfs_key cpukey; 5045f39d397SChris Mason u32 nritems = btrfs_header_nritems(node); 505aa5d6bedSChris Mason 506aa5d6bedSChris Mason if (path->nodes[level + 1]) 5075f39d397SChris Mason parent = path->nodes[level + 1]; 508a1f39630SAneesh 5098d7be552SChris Mason slot = path->slots[level]; 5107518a238SChris Mason BUG_ON(nritems == 0); 5117518a238SChris Mason if (parent) { 512a1f39630SAneesh parent_slot = path->slots[level + 1]; 5135f39d397SChris Mason btrfs_node_key(parent, &parent_key, parent_slot); 5145f39d397SChris Mason btrfs_node_key(node, &node_key, 0); 5155f39d397SChris Mason BUG_ON(memcmp(&parent_key, &node_key, 516e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 5171d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 518db94535dSChris Mason btrfs_header_bytenr(node)); 519aa5d6bedSChris Mason } 520123abc88SChris Mason BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root)); 5218d7be552SChris Mason if (slot != 0) { 5225f39d397SChris Mason btrfs_node_key_to_cpu(node, &cpukey, slot - 1); 5235f39d397SChris Mason btrfs_node_key(node, &node_key, slot); 5245f39d397SChris Mason BUG_ON(comp_keys(&node_key, &cpukey) <= 0); 5258d7be552SChris Mason } 5268d7be552SChris Mason if (slot < nritems - 1) { 5275f39d397SChris Mason btrfs_node_key_to_cpu(node, &cpukey, slot + 1); 5285f39d397SChris Mason btrfs_node_key(node, &node_key, slot); 5295f39d397SChris Mason BUG_ON(comp_keys(&node_key, &cpukey) >= 0); 530aa5d6bedSChris Mason } 531aa5d6bedSChris Mason return 0; 532aa5d6bedSChris Mason } 533aa5d6bedSChris Mason 534123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path, 535123abc88SChris Mason int level) 536aa5d6bedSChris Mason { 5375f39d397SChris Mason struct extent_buffer *leaf = path->nodes[level]; 5385f39d397SChris Mason struct extent_buffer *parent = NULL; 539aa5d6bedSChris Mason int parent_slot; 5408d7be552SChris Mason struct btrfs_key cpukey; 5415f39d397SChris Mason struct btrfs_disk_key parent_key; 5425f39d397SChris Mason struct btrfs_disk_key leaf_key; 5435f39d397SChris Mason int slot = path->slots[0]; 5448d7be552SChris Mason 5455f39d397SChris Mason u32 nritems = btrfs_header_nritems(leaf); 546aa5d6bedSChris Mason 547aa5d6bedSChris Mason if (path->nodes[level + 1]) 5485f39d397SChris Mason parent = path->nodes[level + 1]; 5497518a238SChris Mason 5507518a238SChris Mason if (nritems == 0) 5517518a238SChris Mason return 0; 5527518a238SChris Mason 5537518a238SChris Mason if (parent) { 554a1f39630SAneesh parent_slot = path->slots[level + 1]; 5555f39d397SChris Mason btrfs_node_key(parent, &parent_key, parent_slot); 5565f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, 0); 5576702ed49SChris Mason 5585f39d397SChris Mason BUG_ON(memcmp(&parent_key, &leaf_key, 559e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 5601d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 561db94535dSChris Mason btrfs_header_bytenr(leaf)); 562aa5d6bedSChris Mason } 5635f39d397SChris Mason #if 0 5645f39d397SChris Mason for (i = 0; nritems > 1 && i < nritems - 2; i++) { 5655f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, i + 1); 5665f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, i); 5675f39d397SChris Mason if (comp_keys(&leaf_key, &cpukey) >= 0) { 5685f39d397SChris Mason btrfs_print_leaf(root, leaf); 5695f39d397SChris Mason printk("slot %d offset bad key\n", i); 5705f39d397SChris Mason BUG_ON(1); 5715f39d397SChris Mason } 5725f39d397SChris Mason if (btrfs_item_offset_nr(leaf, i) != 5735f39d397SChris Mason btrfs_item_end_nr(leaf, i + 1)) { 5745f39d397SChris Mason btrfs_print_leaf(root, leaf); 5755f39d397SChris Mason printk("slot %d offset bad\n", i); 5765f39d397SChris Mason BUG_ON(1); 5775f39d397SChris Mason } 5785f39d397SChris Mason if (i == 0) { 5795f39d397SChris Mason if (btrfs_item_offset_nr(leaf, i) + 5805f39d397SChris Mason btrfs_item_size_nr(leaf, i) != 5815f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 5825f39d397SChris Mason btrfs_print_leaf(root, leaf); 5835f39d397SChris Mason printk("slot %d first offset bad\n", i); 5845f39d397SChris Mason BUG_ON(1); 5855f39d397SChris Mason } 5865f39d397SChris Mason } 5875f39d397SChris Mason } 5885f39d397SChris Mason if (nritems > 0) { 5895f39d397SChris Mason if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) { 5905f39d397SChris Mason btrfs_print_leaf(root, leaf); 5915f39d397SChris Mason printk("slot %d bad size \n", nritems - 1); 5925f39d397SChris Mason BUG_ON(1); 5935f39d397SChris Mason } 5945f39d397SChris Mason } 5955f39d397SChris Mason #endif 5965f39d397SChris Mason if (slot != 0 && slot < nritems - 1) { 5975f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, slot); 5985f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1); 5995f39d397SChris Mason if (comp_keys(&leaf_key, &cpukey) <= 0) { 6005f39d397SChris Mason btrfs_print_leaf(root, leaf); 6015f39d397SChris Mason printk("slot %d offset bad key\n", slot); 6025f39d397SChris Mason BUG_ON(1); 6035f39d397SChris Mason } 6045f39d397SChris Mason if (btrfs_item_offset_nr(leaf, slot - 1) != 6055f39d397SChris Mason btrfs_item_end_nr(leaf, slot)) { 6065f39d397SChris Mason btrfs_print_leaf(root, leaf); 6075f39d397SChris Mason printk("slot %d offset bad\n", slot); 6085f39d397SChris Mason BUG_ON(1); 6095f39d397SChris Mason } 610aa5d6bedSChris Mason } 6118d7be552SChris Mason if (slot < nritems - 1) { 6125f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, slot); 6135f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1); 6145f39d397SChris Mason BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0); 6155f39d397SChris Mason if (btrfs_item_offset_nr(leaf, slot) != 6165f39d397SChris Mason btrfs_item_end_nr(leaf, slot + 1)) { 6175f39d397SChris Mason btrfs_print_leaf(root, leaf); 6185f39d397SChris Mason printk("slot %d offset bad\n", slot); 6195f39d397SChris Mason BUG_ON(1); 620aa5d6bedSChris Mason } 6215f39d397SChris Mason } 6225f39d397SChris Mason BUG_ON(btrfs_item_offset_nr(leaf, 0) + 6235f39d397SChris Mason btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root)); 624aa5d6bedSChris Mason return 0; 625aa5d6bedSChris Mason } 626aa5d6bedSChris Mason 62798ed5174SChris Mason static int noinline check_block(struct btrfs_root *root, 62898ed5174SChris Mason struct btrfs_path *path, int level) 629aa5d6bedSChris Mason { 630f188591eSChris Mason u64 found_start; 63185d824c4SChris Mason return 0; 632f188591eSChris Mason if (btrfs_header_level(path->nodes[level]) != level) 633f188591eSChris Mason printk("warning: bad level %Lu wanted %d found %d\n", 634f188591eSChris Mason path->nodes[level]->start, level, 635f188591eSChris Mason btrfs_header_level(path->nodes[level])); 636f188591eSChris Mason found_start = btrfs_header_bytenr(path->nodes[level]); 637f188591eSChris Mason if (found_start != path->nodes[level]->start) { 638f188591eSChris Mason printk("warning: bad bytentr %Lu found %Lu\n", 639f188591eSChris Mason path->nodes[level]->start, found_start); 640f188591eSChris Mason } 641db94535dSChris Mason #if 0 6425f39d397SChris Mason struct extent_buffer *buf = path->nodes[level]; 6435f39d397SChris Mason 644479965d6SChris Mason if (memcmp_extent_buffer(buf, root->fs_info->fsid, 645479965d6SChris Mason (unsigned long)btrfs_header_fsid(buf), 646479965d6SChris Mason BTRFS_FSID_SIZE)) { 6475f39d397SChris Mason printk("warning bad block %Lu\n", buf->start); 648db94535dSChris Mason return 1; 6495f39d397SChris Mason } 650db94535dSChris Mason #endif 651aa5d6bedSChris Mason if (level == 0) 652123abc88SChris Mason return check_leaf(root, path, level); 653123abc88SChris Mason return check_node(root, path, level); 654aa5d6bedSChris Mason } 655aa5d6bedSChris Mason 65674123bd7SChris Mason /* 6575f39d397SChris Mason * search for key in the extent_buffer. The items start at offset p, 6585f39d397SChris Mason * and they are item_size apart. There are 'max' items in p. 6595f39d397SChris Mason * 66074123bd7SChris Mason * the slot in the array is returned via slot, and it points to 66174123bd7SChris Mason * the place where you would insert key if it is not found in 66274123bd7SChris Mason * the array. 66374123bd7SChris Mason * 66474123bd7SChris Mason * slot may point to max if the key is bigger than all of the keys 66574123bd7SChris Mason */ 6665f39d397SChris Mason static int generic_bin_search(struct extent_buffer *eb, unsigned long p, 6675f39d397SChris Mason int item_size, struct btrfs_key *key, 668be0e5c09SChris Mason int max, int *slot) 669be0e5c09SChris Mason { 670be0e5c09SChris Mason int low = 0; 671be0e5c09SChris Mason int high = max; 672be0e5c09SChris Mason int mid; 673be0e5c09SChris Mason int ret; 674479965d6SChris Mason struct btrfs_disk_key *tmp = NULL; 6755f39d397SChris Mason struct btrfs_disk_key unaligned; 6765f39d397SChris Mason unsigned long offset; 6775f39d397SChris Mason char *map_token = NULL; 6785f39d397SChris Mason char *kaddr = NULL; 6795f39d397SChris Mason unsigned long map_start = 0; 6805f39d397SChris Mason unsigned long map_len = 0; 681479965d6SChris Mason int err; 682be0e5c09SChris Mason 683be0e5c09SChris Mason while(low < high) { 684be0e5c09SChris Mason mid = (low + high) / 2; 6855f39d397SChris Mason offset = p + mid * item_size; 6865f39d397SChris Mason 6875f39d397SChris Mason if (!map_token || offset < map_start || 6885f39d397SChris Mason (offset + sizeof(struct btrfs_disk_key)) > 6895f39d397SChris Mason map_start + map_len) { 690479965d6SChris Mason if (map_token) { 6915f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 692479965d6SChris Mason map_token = NULL; 693479965d6SChris Mason } 694479965d6SChris Mason err = map_extent_buffer(eb, offset, 695479965d6SChris Mason sizeof(struct btrfs_disk_key), 696479965d6SChris Mason &map_token, &kaddr, 6975f39d397SChris Mason &map_start, &map_len, KM_USER0); 6985f39d397SChris Mason 699479965d6SChris Mason if (!err) { 700479965d6SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 701479965d6SChris Mason map_start); 702479965d6SChris Mason } else { 7035f39d397SChris Mason read_extent_buffer(eb, &unaligned, 7045f39d397SChris Mason offset, sizeof(unaligned)); 7055f39d397SChris Mason tmp = &unaligned; 706479965d6SChris Mason } 707479965d6SChris Mason 7085f39d397SChris Mason } else { 7095f39d397SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 7105f39d397SChris Mason map_start); 7115f39d397SChris Mason } 712be0e5c09SChris Mason ret = comp_keys(tmp, key); 713be0e5c09SChris Mason 714be0e5c09SChris Mason if (ret < 0) 715be0e5c09SChris Mason low = mid + 1; 716be0e5c09SChris Mason else if (ret > 0) 717be0e5c09SChris Mason high = mid; 718be0e5c09SChris Mason else { 719be0e5c09SChris Mason *slot = mid; 720479965d6SChris Mason if (map_token) 7215f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 722be0e5c09SChris Mason return 0; 723be0e5c09SChris Mason } 724be0e5c09SChris Mason } 725be0e5c09SChris Mason *slot = low; 7265f39d397SChris Mason if (map_token) 7275f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 728be0e5c09SChris Mason return 1; 729be0e5c09SChris Mason } 730be0e5c09SChris Mason 73197571fd0SChris Mason /* 73297571fd0SChris Mason * simple bin_search frontend that does the right thing for 73397571fd0SChris Mason * leaves vs nodes 73497571fd0SChris Mason */ 7355f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key, 7365f39d397SChris Mason int level, int *slot) 737be0e5c09SChris Mason { 7385f39d397SChris Mason if (level == 0) { 7395f39d397SChris Mason return generic_bin_search(eb, 7405f39d397SChris Mason offsetof(struct btrfs_leaf, items), 7410783fcfcSChris Mason sizeof(struct btrfs_item), 7425f39d397SChris Mason key, btrfs_header_nritems(eb), 7437518a238SChris Mason slot); 744be0e5c09SChris Mason } else { 7455f39d397SChris Mason return generic_bin_search(eb, 7465f39d397SChris Mason offsetof(struct btrfs_node, ptrs), 747123abc88SChris Mason sizeof(struct btrfs_key_ptr), 7485f39d397SChris Mason key, btrfs_header_nritems(eb), 7497518a238SChris Mason slot); 750be0e5c09SChris Mason } 751be0e5c09SChris Mason return -1; 752be0e5c09SChris Mason } 753be0e5c09SChris Mason 7545f39d397SChris Mason static struct extent_buffer *read_node_slot(struct btrfs_root *root, 7555f39d397SChris Mason struct extent_buffer *parent, int slot) 756bb803951SChris Mason { 757ca7a79adSChris Mason int level = btrfs_header_level(parent); 758bb803951SChris Mason if (slot < 0) 759bb803951SChris Mason return NULL; 7605f39d397SChris Mason if (slot >= btrfs_header_nritems(parent)) 761bb803951SChris Mason return NULL; 762ca7a79adSChris Mason 763ca7a79adSChris Mason BUG_ON(level == 0); 764ca7a79adSChris Mason 765db94535dSChris Mason return read_tree_block(root, btrfs_node_blockptr(parent, slot), 766ca7a79adSChris Mason btrfs_level_size(root, level - 1), 767ca7a79adSChris Mason btrfs_node_ptr_generation(parent, slot)); 768bb803951SChris Mason } 769bb803951SChris Mason 77098ed5174SChris Mason static int balance_level(struct btrfs_trans_handle *trans, 77198ed5174SChris Mason struct btrfs_root *root, 77298ed5174SChris Mason struct btrfs_path *path, int level) 773bb803951SChris Mason { 7745f39d397SChris Mason struct extent_buffer *right = NULL; 7755f39d397SChris Mason struct extent_buffer *mid; 7765f39d397SChris Mason struct extent_buffer *left = NULL; 7775f39d397SChris Mason struct extent_buffer *parent = NULL; 778bb803951SChris Mason int ret = 0; 779bb803951SChris Mason int wret; 780bb803951SChris Mason int pslot; 781bb803951SChris Mason int orig_slot = path->slots[level]; 78254aa1f4dSChris Mason int err_on_enospc = 0; 78379f95c82SChris Mason u64 orig_ptr; 784bb803951SChris Mason 785bb803951SChris Mason if (level == 0) 786bb803951SChris Mason return 0; 787bb803951SChris Mason 7885f39d397SChris Mason mid = path->nodes[level]; 789925baeddSChris Mason WARN_ON(!path->locks[level]); 7907bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 7917bb86316SChris Mason 7921d4f8a0cSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 79379f95c82SChris Mason 794234b63a0SChris Mason if (level < BTRFS_MAX_LEVEL - 1) 7955f39d397SChris Mason parent = path->nodes[level + 1]; 796bb803951SChris Mason pslot = path->slots[level + 1]; 797bb803951SChris Mason 79840689478SChris Mason /* 79940689478SChris Mason * deal with the case where there is only one pointer in the root 80040689478SChris Mason * by promoting the node below to a root 80140689478SChris Mason */ 8025f39d397SChris Mason if (!parent) { 8035f39d397SChris Mason struct extent_buffer *child; 804bb803951SChris Mason 8055f39d397SChris Mason if (btrfs_header_nritems(mid) != 1) 806bb803951SChris Mason return 0; 807bb803951SChris Mason 808bb803951SChris Mason /* promote the child to a root */ 8095f39d397SChris Mason child = read_node_slot(root, mid, 0); 810925baeddSChris Mason btrfs_tree_lock(child); 811bb803951SChris Mason BUG_ON(!child); 8122f375ab9SYan ret = btrfs_cow_block(trans, root, child, mid, 0, &child); 8132f375ab9SYan BUG_ON(ret); 8142f375ab9SYan 815925baeddSChris Mason spin_lock(&root->node_lock); 816bb803951SChris Mason root->node = child; 817925baeddSChris Mason spin_unlock(&root->node_lock); 818925baeddSChris Mason 8190b86a832SChris Mason add_root_to_dirty_list(root); 820925baeddSChris Mason btrfs_tree_unlock(child); 821925baeddSChris Mason path->locks[level] = 0; 822bb803951SChris Mason path->nodes[level] = NULL; 8235f39d397SChris Mason clean_tree_block(trans, root, mid); 824925baeddSChris Mason btrfs_tree_unlock(mid); 825bb803951SChris Mason /* once for the path */ 8265f39d397SChris Mason free_extent_buffer(mid); 8277bb86316SChris Mason ret = btrfs_free_extent(trans, root, mid->start, mid->len, 8287bb86316SChris Mason root->root_key.objectid, 8297bb86316SChris Mason btrfs_header_generation(mid), 0, 0, 1); 830bb803951SChris Mason /* once for the root ptr */ 8315f39d397SChris Mason free_extent_buffer(mid); 832db94535dSChris Mason return ret; 833bb803951SChris Mason } 8345f39d397SChris Mason if (btrfs_header_nritems(mid) > 835123abc88SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) / 4) 836bb803951SChris Mason return 0; 837bb803951SChris Mason 8385f39d397SChris Mason if (btrfs_header_nritems(mid) < 2) 83954aa1f4dSChris Mason err_on_enospc = 1; 84054aa1f4dSChris Mason 8415f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 8425f39d397SChris Mason if (left) { 843925baeddSChris Mason btrfs_tree_lock(left); 8445f39d397SChris Mason wret = btrfs_cow_block(trans, root, left, 8455f39d397SChris Mason parent, pslot - 1, &left); 84654aa1f4dSChris Mason if (wret) { 84754aa1f4dSChris Mason ret = wret; 84854aa1f4dSChris Mason goto enospc; 84954aa1f4dSChris Mason } 8502cc58cf2SChris Mason } 8515f39d397SChris Mason right = read_node_slot(root, parent, pslot + 1); 8525f39d397SChris Mason if (right) { 853925baeddSChris Mason btrfs_tree_lock(right); 8545f39d397SChris Mason wret = btrfs_cow_block(trans, root, right, 8555f39d397SChris Mason parent, pslot + 1, &right); 8562cc58cf2SChris Mason if (wret) { 8572cc58cf2SChris Mason ret = wret; 8582cc58cf2SChris Mason goto enospc; 8592cc58cf2SChris Mason } 8602cc58cf2SChris Mason } 8612cc58cf2SChris Mason 8622cc58cf2SChris Mason /* first, try to make some room in the middle buffer */ 8635f39d397SChris Mason if (left) { 8645f39d397SChris Mason orig_slot += btrfs_header_nritems(left); 865bce4eae9SChris Mason wret = push_node_left(trans, root, left, mid, 1); 86679f95c82SChris Mason if (wret < 0) 86779f95c82SChris Mason ret = wret; 8685f39d397SChris Mason if (btrfs_header_nritems(mid) < 2) 86954aa1f4dSChris Mason err_on_enospc = 1; 870bb803951SChris Mason } 87179f95c82SChris Mason 87279f95c82SChris Mason /* 87379f95c82SChris Mason * then try to empty the right most buffer into the middle 87479f95c82SChris Mason */ 8755f39d397SChris Mason if (right) { 876971a1f66SChris Mason wret = push_node_left(trans, root, mid, right, 1); 87754aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 87879f95c82SChris Mason ret = wret; 8795f39d397SChris Mason if (btrfs_header_nritems(right) == 0) { 880db94535dSChris Mason u64 bytenr = right->start; 8817bb86316SChris Mason u64 generation = btrfs_header_generation(parent); 882db94535dSChris Mason u32 blocksize = right->len; 883db94535dSChris Mason 8845f39d397SChris Mason clean_tree_block(trans, root, right); 885925baeddSChris Mason btrfs_tree_unlock(right); 8865f39d397SChris Mason free_extent_buffer(right); 887bb803951SChris Mason right = NULL; 888e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot + 889e089f05cSChris Mason 1); 890bb803951SChris Mason if (wret) 891bb803951SChris Mason ret = wret; 892db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 8937bb86316SChris Mason blocksize, 8947bb86316SChris Mason btrfs_header_owner(parent), 8957bb86316SChris Mason generation, 0, 0, 1); 896bb803951SChris Mason if (wret) 897bb803951SChris Mason ret = wret; 898bb803951SChris Mason } else { 8995f39d397SChris Mason struct btrfs_disk_key right_key; 9005f39d397SChris Mason btrfs_node_key(right, &right_key, 0); 9015f39d397SChris Mason btrfs_set_node_key(parent, &right_key, pslot + 1); 9025f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 903bb803951SChris Mason } 904bb803951SChris Mason } 9055f39d397SChris Mason if (btrfs_header_nritems(mid) == 1) { 90679f95c82SChris Mason /* 90779f95c82SChris Mason * we're not allowed to leave a node with one item in the 90879f95c82SChris Mason * tree during a delete. A deletion from lower in the tree 90979f95c82SChris Mason * could try to delete the only pointer in this node. 91079f95c82SChris Mason * So, pull some keys from the left. 91179f95c82SChris Mason * There has to be a left pointer at this point because 91279f95c82SChris Mason * otherwise we would have pulled some pointers from the 91379f95c82SChris Mason * right 91479f95c82SChris Mason */ 9155f39d397SChris Mason BUG_ON(!left); 9165f39d397SChris Mason wret = balance_node_right(trans, root, mid, left); 91754aa1f4dSChris Mason if (wret < 0) { 91879f95c82SChris Mason ret = wret; 91954aa1f4dSChris Mason goto enospc; 92054aa1f4dSChris Mason } 921bce4eae9SChris Mason if (wret == 1) { 922bce4eae9SChris Mason wret = push_node_left(trans, root, left, mid, 1); 923bce4eae9SChris Mason if (wret < 0) 924bce4eae9SChris Mason ret = wret; 925bce4eae9SChris Mason } 92679f95c82SChris Mason BUG_ON(wret == 1); 92779f95c82SChris Mason } 9285f39d397SChris Mason if (btrfs_header_nritems(mid) == 0) { 92979f95c82SChris Mason /* we've managed to empty the middle node, drop it */ 9307bb86316SChris Mason u64 root_gen = btrfs_header_generation(parent); 931db94535dSChris Mason u64 bytenr = mid->start; 932db94535dSChris Mason u32 blocksize = mid->len; 933925baeddSChris Mason 9345f39d397SChris Mason clean_tree_block(trans, root, mid); 935925baeddSChris Mason btrfs_tree_unlock(mid); 9365f39d397SChris Mason free_extent_buffer(mid); 937bb803951SChris Mason mid = NULL; 938e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot); 939bb803951SChris Mason if (wret) 940bb803951SChris Mason ret = wret; 9417bb86316SChris Mason wret = btrfs_free_extent(trans, root, bytenr, blocksize, 9427bb86316SChris Mason btrfs_header_owner(parent), 9437bb86316SChris Mason root_gen, 0, 0, 1); 944bb803951SChris Mason if (wret) 945bb803951SChris Mason ret = wret; 94679f95c82SChris Mason } else { 94779f95c82SChris Mason /* update the parent key to reflect our changes */ 9485f39d397SChris Mason struct btrfs_disk_key mid_key; 9495f39d397SChris Mason btrfs_node_key(mid, &mid_key, 0); 9505f39d397SChris Mason btrfs_set_node_key(parent, &mid_key, pslot); 9515f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 95279f95c82SChris Mason } 953bb803951SChris Mason 95479f95c82SChris Mason /* update the path */ 9555f39d397SChris Mason if (left) { 9565f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 9575f39d397SChris Mason extent_buffer_get(left); 958925baeddSChris Mason /* left was locked after cow */ 9595f39d397SChris Mason path->nodes[level] = left; 960bb803951SChris Mason path->slots[level + 1] -= 1; 961bb803951SChris Mason path->slots[level] = orig_slot; 962925baeddSChris Mason if (mid) { 963925baeddSChris Mason btrfs_tree_unlock(mid); 9645f39d397SChris Mason free_extent_buffer(mid); 965925baeddSChris Mason } 966bb803951SChris Mason } else { 9675f39d397SChris Mason orig_slot -= btrfs_header_nritems(left); 968bb803951SChris Mason path->slots[level] = orig_slot; 969bb803951SChris Mason } 970bb803951SChris Mason } 97179f95c82SChris Mason /* double check we haven't messed things up */ 972123abc88SChris Mason check_block(root, path, level); 973e20d96d6SChris Mason if (orig_ptr != 9745f39d397SChris Mason btrfs_node_blockptr(path->nodes[level], path->slots[level])) 97579f95c82SChris Mason BUG(); 97654aa1f4dSChris Mason enospc: 977925baeddSChris Mason if (right) { 978925baeddSChris Mason btrfs_tree_unlock(right); 9795f39d397SChris Mason free_extent_buffer(right); 980925baeddSChris Mason } 981925baeddSChris Mason if (left) { 982925baeddSChris Mason if (path->nodes[level] != left) 983925baeddSChris Mason btrfs_tree_unlock(left); 9845f39d397SChris Mason free_extent_buffer(left); 985925baeddSChris Mason } 986bb803951SChris Mason return ret; 987bb803951SChris Mason } 988bb803951SChris Mason 989e66f709bSChris Mason /* returns zero if the push worked, non-zero otherwise */ 99098ed5174SChris Mason static int noinline push_nodes_for_insert(struct btrfs_trans_handle *trans, 991e66f709bSChris Mason struct btrfs_root *root, 992e66f709bSChris Mason struct btrfs_path *path, int level) 993e66f709bSChris Mason { 9945f39d397SChris Mason struct extent_buffer *right = NULL; 9955f39d397SChris Mason struct extent_buffer *mid; 9965f39d397SChris Mason struct extent_buffer *left = NULL; 9975f39d397SChris Mason struct extent_buffer *parent = NULL; 998e66f709bSChris Mason int ret = 0; 999e66f709bSChris Mason int wret; 1000e66f709bSChris Mason int pslot; 1001e66f709bSChris Mason int orig_slot = path->slots[level]; 1002e66f709bSChris Mason u64 orig_ptr; 1003e66f709bSChris Mason 1004e66f709bSChris Mason if (level == 0) 1005e66f709bSChris Mason return 1; 1006e66f709bSChris Mason 10075f39d397SChris Mason mid = path->nodes[level]; 10087bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 1009e66f709bSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 1010e66f709bSChris Mason 1011e66f709bSChris Mason if (level < BTRFS_MAX_LEVEL - 1) 10125f39d397SChris Mason parent = path->nodes[level + 1]; 1013e66f709bSChris Mason pslot = path->slots[level + 1]; 1014e66f709bSChris Mason 10155f39d397SChris Mason if (!parent) 1016e66f709bSChris Mason return 1; 1017e66f709bSChris Mason 10185f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 1019e66f709bSChris Mason 1020e66f709bSChris Mason /* first, try to make some room in the middle buffer */ 10215f39d397SChris Mason if (left) { 1022e66f709bSChris Mason u32 left_nr; 1023925baeddSChris Mason 1024925baeddSChris Mason btrfs_tree_lock(left); 10255f39d397SChris Mason left_nr = btrfs_header_nritems(left); 102633ade1f8SChris Mason if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 102733ade1f8SChris Mason wret = 1; 102833ade1f8SChris Mason } else { 10295f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, parent, 10305f39d397SChris Mason pslot - 1, &left); 103154aa1f4dSChris Mason if (ret) 103254aa1f4dSChris Mason wret = 1; 103354aa1f4dSChris Mason else { 103454aa1f4dSChris Mason wret = push_node_left(trans, root, 1035971a1f66SChris Mason left, mid, 0); 103654aa1f4dSChris Mason } 103733ade1f8SChris Mason } 1038e66f709bSChris Mason if (wret < 0) 1039e66f709bSChris Mason ret = wret; 1040e66f709bSChris Mason if (wret == 0) { 10415f39d397SChris Mason struct btrfs_disk_key disk_key; 1042e66f709bSChris Mason orig_slot += left_nr; 10435f39d397SChris Mason btrfs_node_key(mid, &disk_key, 0); 10445f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot); 10455f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 10465f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 10475f39d397SChris Mason path->nodes[level] = left; 1048e66f709bSChris Mason path->slots[level + 1] -= 1; 1049e66f709bSChris Mason path->slots[level] = orig_slot; 1050925baeddSChris Mason btrfs_tree_unlock(mid); 10515f39d397SChris Mason free_extent_buffer(mid); 1052e66f709bSChris Mason } else { 1053e66f709bSChris Mason orig_slot -= 10545f39d397SChris Mason btrfs_header_nritems(left); 1055e66f709bSChris Mason path->slots[level] = orig_slot; 1056925baeddSChris Mason btrfs_tree_unlock(left); 10575f39d397SChris Mason free_extent_buffer(left); 1058e66f709bSChris Mason } 1059e66f709bSChris Mason return 0; 1060e66f709bSChris Mason } 1061925baeddSChris Mason btrfs_tree_unlock(left); 10625f39d397SChris Mason free_extent_buffer(left); 1063e66f709bSChris Mason } 10645f39d397SChris Mason right = read_node_slot(root, parent, pslot + 1); 1065e66f709bSChris Mason 1066e66f709bSChris Mason /* 1067e66f709bSChris Mason * then try to empty the right most buffer into the middle 1068e66f709bSChris Mason */ 10695f39d397SChris Mason if (right) { 107033ade1f8SChris Mason u32 right_nr; 1071925baeddSChris Mason btrfs_tree_lock(right); 10725f39d397SChris Mason right_nr = btrfs_header_nritems(right); 107333ade1f8SChris Mason if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 107433ade1f8SChris Mason wret = 1; 107533ade1f8SChris Mason } else { 10765f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, 10775f39d397SChris Mason parent, pslot + 1, 10785f39d397SChris Mason &right); 107954aa1f4dSChris Mason if (ret) 108054aa1f4dSChris Mason wret = 1; 108154aa1f4dSChris Mason else { 108233ade1f8SChris Mason wret = balance_node_right(trans, root, 10835f39d397SChris Mason right, mid); 108433ade1f8SChris Mason } 108554aa1f4dSChris Mason } 1086e66f709bSChris Mason if (wret < 0) 1087e66f709bSChris Mason ret = wret; 1088e66f709bSChris Mason if (wret == 0) { 10895f39d397SChris Mason struct btrfs_disk_key disk_key; 10905f39d397SChris Mason 10915f39d397SChris Mason btrfs_node_key(right, &disk_key, 0); 10925f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot + 1); 10935f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 10945f39d397SChris Mason 10955f39d397SChris Mason if (btrfs_header_nritems(mid) <= orig_slot) { 10965f39d397SChris Mason path->nodes[level] = right; 1097e66f709bSChris Mason path->slots[level + 1] += 1; 1098e66f709bSChris Mason path->slots[level] = orig_slot - 10995f39d397SChris Mason btrfs_header_nritems(mid); 1100925baeddSChris Mason btrfs_tree_unlock(mid); 11015f39d397SChris Mason free_extent_buffer(mid); 1102e66f709bSChris Mason } else { 1103925baeddSChris Mason btrfs_tree_unlock(right); 11045f39d397SChris Mason free_extent_buffer(right); 1105e66f709bSChris Mason } 1106e66f709bSChris Mason return 0; 1107e66f709bSChris Mason } 1108925baeddSChris Mason btrfs_tree_unlock(right); 11095f39d397SChris Mason free_extent_buffer(right); 1110e66f709bSChris Mason } 1111e66f709bSChris Mason return 1; 1112e66f709bSChris Mason } 1113e66f709bSChris Mason 111474123bd7SChris Mason /* 11153c69faecSChris Mason * readahead one full node of leaves 11163c69faecSChris Mason */ 11173c69faecSChris Mason static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path, 111801f46658SChris Mason int level, int slot, u64 objectid) 11193c69faecSChris Mason { 11205f39d397SChris Mason struct extent_buffer *node; 112101f46658SChris Mason struct btrfs_disk_key disk_key; 11223c69faecSChris Mason u32 nritems; 11233c69faecSChris Mason u64 search; 11246b80053dSChris Mason u64 lowest_read; 11256b80053dSChris Mason u64 highest_read; 11266b80053dSChris Mason u64 nread = 0; 11273c69faecSChris Mason int direction = path->reada; 11285f39d397SChris Mason struct extent_buffer *eb; 11296b80053dSChris Mason u32 nr; 11306b80053dSChris Mason u32 blocksize; 11316b80053dSChris Mason u32 nscan = 0; 1132db94535dSChris Mason 1133a6b6e75eSChris Mason if (level != 1) 11343c69faecSChris Mason return; 11353c69faecSChris Mason 11366702ed49SChris Mason if (!path->nodes[level]) 11376702ed49SChris Mason return; 11386702ed49SChris Mason 11395f39d397SChris Mason node = path->nodes[level]; 1140925baeddSChris Mason WARN_ON(!path->skip_locking && !btrfs_tree_locked(node)); 1141925baeddSChris Mason 11423c69faecSChris Mason search = btrfs_node_blockptr(node, slot); 11436b80053dSChris Mason blocksize = btrfs_level_size(root, level - 1); 11446b80053dSChris Mason eb = btrfs_find_tree_block(root, search, blocksize); 11455f39d397SChris Mason if (eb) { 11465f39d397SChris Mason free_extent_buffer(eb); 11473c69faecSChris Mason return; 11483c69faecSChris Mason } 11493c69faecSChris Mason 11506b80053dSChris Mason highest_read = search; 11516b80053dSChris Mason lowest_read = search; 11526b80053dSChris Mason 11535f39d397SChris Mason nritems = btrfs_header_nritems(node); 11546b80053dSChris Mason nr = slot; 11553c69faecSChris Mason while(1) { 11566b80053dSChris Mason if (direction < 0) { 11576b80053dSChris Mason if (nr == 0) 11583c69faecSChris Mason break; 11596b80053dSChris Mason nr--; 11606b80053dSChris Mason } else if (direction > 0) { 11616b80053dSChris Mason nr++; 11626b80053dSChris Mason if (nr >= nritems) 11636b80053dSChris Mason break; 11643c69faecSChris Mason } 116501f46658SChris Mason if (path->reada < 0 && objectid) { 116601f46658SChris Mason btrfs_node_key(node, &disk_key, nr); 116701f46658SChris Mason if (btrfs_disk_key_objectid(&disk_key) != objectid) 116801f46658SChris Mason break; 116901f46658SChris Mason } 11706b80053dSChris Mason search = btrfs_node_blockptr(node, nr); 11716b80053dSChris Mason if ((search >= lowest_read && search <= highest_read) || 11726b80053dSChris Mason (search < lowest_read && lowest_read - search <= 32768) || 11736b80053dSChris Mason (search > highest_read && search - highest_read <= 32768)) { 1174ca7a79adSChris Mason readahead_tree_block(root, search, blocksize, 1175ca7a79adSChris Mason btrfs_node_ptr_generation(node, nr)); 11766b80053dSChris Mason nread += blocksize; 11773c69faecSChris Mason } 11786b80053dSChris Mason nscan++; 11796b80053dSChris Mason if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32)) 11806b80053dSChris Mason break; 11816b80053dSChris Mason if(nread > (1024 * 1024) || nscan > 128) 11826b80053dSChris Mason break; 11836b80053dSChris Mason 11846b80053dSChris Mason if (search < lowest_read) 11856b80053dSChris Mason lowest_read = search; 11866b80053dSChris Mason if (search > highest_read) 11876b80053dSChris Mason highest_read = search; 11883c69faecSChris Mason } 11893c69faecSChris Mason } 1190925baeddSChris Mason 1191925baeddSChris Mason static void unlock_up(struct btrfs_path *path, int level, int lowest_unlock) 1192925baeddSChris Mason { 1193925baeddSChris Mason int i; 1194925baeddSChris Mason int skip_level = level; 1195925baeddSChris Mason struct extent_buffer *t; 1196925baeddSChris Mason 1197925baeddSChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 1198925baeddSChris Mason if (!path->nodes[i]) 1199925baeddSChris Mason break; 1200925baeddSChris Mason if (!path->locks[i]) 1201925baeddSChris Mason break; 1202925baeddSChris Mason if (path->slots[i] == 0) { 1203925baeddSChris Mason skip_level = i + 1; 1204925baeddSChris Mason continue; 1205925baeddSChris Mason } 1206925baeddSChris Mason if (path->keep_locks) { 1207925baeddSChris Mason u32 nritems; 1208925baeddSChris Mason t = path->nodes[i]; 1209925baeddSChris Mason nritems = btrfs_header_nritems(t); 1210*a2135011SChris Mason if (nritems < 2 || path->slots[i] >= nritems - 2) { 1211*a2135011SChris Mason if (path->keep_locks) { 1212*a2135011SChris Mason //printk("path %p skip level now %d\n", path, skip_level); 1213*a2135011SChris Mason } 1214925baeddSChris Mason skip_level = i + 1; 1215925baeddSChris Mason continue; 1216925baeddSChris Mason } 1217925baeddSChris Mason } 1218925baeddSChris Mason t = path->nodes[i]; 1219925baeddSChris Mason if (i >= lowest_unlock && i > skip_level && path->locks[i]) { 1220*a2135011SChris Mason if (path->keep_locks) { 1221*a2135011SChris Mason //printk("path %p unlocking level %d slot %d nritems %d skip_level %d\n", path, i, path->slots[i], btrfs_header_nritems(t), skip_level); 1222*a2135011SChris Mason } 1223925baeddSChris Mason btrfs_tree_unlock(t); 1224925baeddSChris Mason path->locks[i] = 0; 1225925baeddSChris Mason } 1226925baeddSChris Mason } 1227925baeddSChris Mason } 1228925baeddSChris Mason 12293c69faecSChris Mason /* 123074123bd7SChris Mason * look for key in the tree. path is filled in with nodes along the way 123174123bd7SChris Mason * if key is found, we return zero and you can find the item in the leaf 123274123bd7SChris Mason * level of the path (level 0) 123374123bd7SChris Mason * 123474123bd7SChris Mason * If the key isn't found, the path points to the slot where it should 1235aa5d6bedSChris Mason * be inserted, and 1 is returned. If there are other errors during the 1236aa5d6bedSChris Mason * search a negative error number is returned. 123797571fd0SChris Mason * 123897571fd0SChris Mason * if ins_len > 0, nodes and leaves will be split as we walk down the 123997571fd0SChris Mason * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if 124097571fd0SChris Mason * possible) 124174123bd7SChris Mason */ 1242e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root 1243e089f05cSChris Mason *root, struct btrfs_key *key, struct btrfs_path *p, int 1244e089f05cSChris Mason ins_len, int cow) 1245be0e5c09SChris Mason { 12465f39d397SChris Mason struct extent_buffer *b; 1247be0e5c09SChris Mason int slot; 1248be0e5c09SChris Mason int ret; 1249be0e5c09SChris Mason int level; 12503c69faecSChris Mason int should_reada = p->reada; 1251925baeddSChris Mason int lowest_unlock = 1; 12529f3a7427SChris Mason u8 lowest_level = 0; 12539f3a7427SChris Mason 12546702ed49SChris Mason lowest_level = p->lowest_level; 12556702ed49SChris Mason WARN_ON(lowest_level && ins_len); 125622b0ebdaSChris Mason WARN_ON(p->nodes[0] != NULL); 1257925baeddSChris Mason WARN_ON(root == root->fs_info->extent_root && 1258925baeddSChris Mason !mutex_is_locked(&root->fs_info->alloc_mutex)); 1259925baeddSChris Mason WARN_ON(root == root->fs_info->chunk_root && 1260925baeddSChris Mason !mutex_is_locked(&root->fs_info->chunk_mutex)); 1261925baeddSChris Mason WARN_ON(root == root->fs_info->dev_root && 1262925baeddSChris Mason !mutex_is_locked(&root->fs_info->chunk_mutex)); 1263925baeddSChris Mason if (ins_len < 0) 1264925baeddSChris Mason lowest_unlock = 2; 1265bb803951SChris Mason again: 1266925baeddSChris Mason if (!p->skip_locking) 1267925baeddSChris Mason b = btrfs_lock_root_node(root); 1268925baeddSChris Mason else 1269925baeddSChris Mason b = btrfs_root_node(root); 1270925baeddSChris Mason 1271eb60ceacSChris Mason while (b) { 12725f39d397SChris Mason level = btrfs_header_level(b); 127302217ed2SChris Mason if (cow) { 127402217ed2SChris Mason int wret; 1275e20d96d6SChris Mason wret = btrfs_cow_block(trans, root, b, 1276e20d96d6SChris Mason p->nodes[level + 1], 1277e20d96d6SChris Mason p->slots[level + 1], 1278252c38f0SYan &b); 127954aa1f4dSChris Mason if (wret) { 12805f39d397SChris Mason free_extent_buffer(b); 128154aa1f4dSChris Mason return wret; 128254aa1f4dSChris Mason } 128302217ed2SChris Mason } 128402217ed2SChris Mason BUG_ON(!cow && ins_len); 12855f39d397SChris Mason if (level != btrfs_header_level(b)) 12862c90e5d6SChris Mason WARN_ON(1); 12875f39d397SChris Mason level = btrfs_header_level(b); 1288eb60ceacSChris Mason p->nodes[level] = b; 1289925baeddSChris Mason if (!p->skip_locking) 1290925baeddSChris Mason p->locks[level] = 1; 1291123abc88SChris Mason ret = check_block(root, p, level); 1292aa5d6bedSChris Mason if (ret) 1293aa5d6bedSChris Mason return -1; 1294925baeddSChris Mason 12955f39d397SChris Mason ret = bin_search(b, key, level, &slot); 12965f39d397SChris Mason if (level != 0) { 1297be0e5c09SChris Mason if (ret && slot > 0) 1298be0e5c09SChris Mason slot -= 1; 1299be0e5c09SChris Mason p->slots[level] = slot; 13005f39d397SChris Mason if (ins_len > 0 && btrfs_header_nritems(b) >= 13011514794eSChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) { 1302e089f05cSChris Mason int sret = split_node(trans, root, p, level); 13035c680ed6SChris Mason BUG_ON(sret > 0); 13045c680ed6SChris Mason if (sret) 13055c680ed6SChris Mason return sret; 13065c680ed6SChris Mason b = p->nodes[level]; 13075c680ed6SChris Mason slot = p->slots[level]; 1308bb803951SChris Mason } else if (ins_len < 0) { 1309e089f05cSChris Mason int sret = balance_level(trans, root, p, 1310e089f05cSChris Mason level); 1311bb803951SChris Mason if (sret) 1312bb803951SChris Mason return sret; 1313bb803951SChris Mason b = p->nodes[level]; 1314f510cfecSChris Mason if (!b) { 1315f510cfecSChris Mason btrfs_release_path(NULL, p); 1316bb803951SChris Mason goto again; 1317f510cfecSChris Mason } 1318bb803951SChris Mason slot = p->slots[level]; 13195f39d397SChris Mason BUG_ON(btrfs_header_nritems(b) == 1); 13205c680ed6SChris Mason } 13219f3a7427SChris Mason /* this is only true while dropping a snapshot */ 1322925baeddSChris Mason if (level == lowest_level) { 1323925baeddSChris Mason unlock_up(p, level, lowest_unlock); 13249f3a7427SChris Mason break; 1325925baeddSChris Mason } 1326ca7a79adSChris Mason 13276702ed49SChris Mason if (should_reada) 132801f46658SChris Mason reada_for_search(root, p, level, slot, 132901f46658SChris Mason key->objectid); 1330ca7a79adSChris Mason 1331ca7a79adSChris Mason b = read_node_slot(root, b, slot); 1332925baeddSChris Mason if (!p->skip_locking) 1333925baeddSChris Mason btrfs_tree_lock(b); 1334*a2135011SChris Mason unlock_up(p, level + 1, lowest_unlock); 1335be0e5c09SChris Mason } else { 1336be0e5c09SChris Mason p->slots[level] = slot; 13375f39d397SChris Mason if (ins_len > 0 && btrfs_leaf_free_space(root, b) < 13380783fcfcSChris Mason sizeof(struct btrfs_item) + ins_len) { 1339d4dbff95SChris Mason int sret = split_leaf(trans, root, key, 1340cc0c5538SChris Mason p, ins_len, ret == 0); 13415c680ed6SChris Mason BUG_ON(sret > 0); 13425c680ed6SChris Mason if (sret) 13435c680ed6SChris Mason return sret; 13445c680ed6SChris Mason } 1345925baeddSChris Mason unlock_up(p, level, lowest_unlock); 1346be0e5c09SChris Mason return ret; 1347be0e5c09SChris Mason } 1348be0e5c09SChris Mason } 1349aa5d6bedSChris Mason return 1; 1350be0e5c09SChris Mason } 1351be0e5c09SChris Mason 135274123bd7SChris Mason /* 135374123bd7SChris Mason * adjust the pointers going up the tree, starting at level 135474123bd7SChris Mason * making sure the right key of each node is points to 'key'. 135574123bd7SChris Mason * This is used after shifting pointers to the left, so it stops 135674123bd7SChris Mason * fixing up pointers when a given leaf/node is not in slot 0 of the 135774123bd7SChris Mason * higher levels 1358aa5d6bedSChris Mason * 1359aa5d6bedSChris Mason * If this fails to write a tree block, it returns -1, but continues 1360aa5d6bedSChris Mason * fixing up the blocks in ram so the tree is consistent. 136174123bd7SChris Mason */ 13625f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans, 13635f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 13645f39d397SChris Mason struct btrfs_disk_key *key, int level) 1365be0e5c09SChris Mason { 1366be0e5c09SChris Mason int i; 1367aa5d6bedSChris Mason int ret = 0; 13685f39d397SChris Mason struct extent_buffer *t; 13695f39d397SChris Mason 1370234b63a0SChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 1371be0e5c09SChris Mason int tslot = path->slots[i]; 1372eb60ceacSChris Mason if (!path->nodes[i]) 1373be0e5c09SChris Mason break; 13745f39d397SChris Mason t = path->nodes[i]; 13755f39d397SChris Mason btrfs_set_node_key(t, key, tslot); 1376925baeddSChris Mason if (!btrfs_tree_locked(path->nodes[i])) { 1377925baeddSChris Mason int ii; 1378925baeddSChris Mason printk("fixup without lock on level %d\n", btrfs_header_level(path->nodes[i])); 1379925baeddSChris Mason for (ii = 0; ii < BTRFS_MAX_LEVEL; ii++) { 1380925baeddSChris Mason printk("level %d slot %d\n", ii, path->slots[ii]); 1381925baeddSChris Mason } 1382925baeddSChris Mason } 1383d6025579SChris Mason btrfs_mark_buffer_dirty(path->nodes[i]); 1384be0e5c09SChris Mason if (tslot != 0) 1385be0e5c09SChris Mason break; 1386be0e5c09SChris Mason } 1387aa5d6bedSChris Mason return ret; 1388be0e5c09SChris Mason } 1389be0e5c09SChris Mason 139074123bd7SChris Mason /* 139174123bd7SChris Mason * try to push data from one node into the next node left in the 139279f95c82SChris Mason * tree. 1393aa5d6bedSChris Mason * 1394aa5d6bedSChris Mason * returns 0 if some ptrs were pushed left, < 0 if there was some horrible 1395aa5d6bedSChris Mason * error, and > 0 if there was no room in the left hand block. 139674123bd7SChris Mason */ 139798ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans, 139898ed5174SChris Mason struct btrfs_root *root, struct extent_buffer *dst, 1399971a1f66SChris Mason struct extent_buffer *src, int empty) 1400be0e5c09SChris Mason { 1401be0e5c09SChris Mason int push_items = 0; 1402bb803951SChris Mason int src_nritems; 1403bb803951SChris Mason int dst_nritems; 1404aa5d6bedSChris Mason int ret = 0; 1405be0e5c09SChris Mason 14065f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 14075f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1408123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 14097bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 14107bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 141154aa1f4dSChris Mason 1412bce4eae9SChris Mason if (!empty && src_nritems <= 8) 1413971a1f66SChris Mason return 1; 1414971a1f66SChris Mason 1415eb60ceacSChris Mason if (push_items <= 0) { 1416be0e5c09SChris Mason return 1; 1417eb60ceacSChris Mason } 1418be0e5c09SChris Mason 1419bce4eae9SChris Mason if (empty) { 1420971a1f66SChris Mason push_items = min(src_nritems, push_items); 1421bce4eae9SChris Mason if (push_items < src_nritems) { 1422bce4eae9SChris Mason /* leave at least 8 pointers in the node if 1423bce4eae9SChris Mason * we aren't going to empty it 1424bce4eae9SChris Mason */ 1425bce4eae9SChris Mason if (src_nritems - push_items < 8) { 1426bce4eae9SChris Mason if (push_items <= 8) 1427bce4eae9SChris Mason return 1; 1428bce4eae9SChris Mason push_items -= 8; 1429bce4eae9SChris Mason } 1430bce4eae9SChris Mason } 1431bce4eae9SChris Mason } else 1432bce4eae9SChris Mason push_items = min(src_nritems - 8, push_items); 143379f95c82SChris Mason 14345f39d397SChris Mason copy_extent_buffer(dst, src, 14355f39d397SChris Mason btrfs_node_key_ptr_offset(dst_nritems), 14365f39d397SChris Mason btrfs_node_key_ptr_offset(0), 1437123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 14385f39d397SChris Mason 1439bb803951SChris Mason if (push_items < src_nritems) { 14405f39d397SChris Mason memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0), 14415f39d397SChris Mason btrfs_node_key_ptr_offset(push_items), 1442e2fa7227SChris Mason (src_nritems - push_items) * 1443123abc88SChris Mason sizeof(struct btrfs_key_ptr)); 1444bb803951SChris Mason } 14455f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 14465f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 14475f39d397SChris Mason btrfs_mark_buffer_dirty(src); 14485f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 1449bb803951SChris Mason return ret; 1450be0e5c09SChris Mason } 1451be0e5c09SChris Mason 145297571fd0SChris Mason /* 145379f95c82SChris Mason * try to push data from one node into the next node right in the 145479f95c82SChris Mason * tree. 145579f95c82SChris Mason * 145679f95c82SChris Mason * returns 0 if some ptrs were pushed, < 0 if there was some horrible 145779f95c82SChris Mason * error, and > 0 if there was no room in the right hand block. 145879f95c82SChris Mason * 145979f95c82SChris Mason * this will only push up to 1/2 the contents of the left node over 146079f95c82SChris Mason */ 14615f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 14625f39d397SChris Mason struct btrfs_root *root, 14635f39d397SChris Mason struct extent_buffer *dst, 14645f39d397SChris Mason struct extent_buffer *src) 146579f95c82SChris Mason { 146679f95c82SChris Mason int push_items = 0; 146779f95c82SChris Mason int max_push; 146879f95c82SChris Mason int src_nritems; 146979f95c82SChris Mason int dst_nritems; 147079f95c82SChris Mason int ret = 0; 147179f95c82SChris Mason 14727bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 14737bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 14747bb86316SChris Mason 14755f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 14765f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1477123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 1478bce4eae9SChris Mason if (push_items <= 0) { 147979f95c82SChris Mason return 1; 1480bce4eae9SChris Mason } 1481bce4eae9SChris Mason 1482bce4eae9SChris Mason if (src_nritems < 4) { 1483bce4eae9SChris Mason return 1; 1484bce4eae9SChris Mason } 148579f95c82SChris Mason 148679f95c82SChris Mason max_push = src_nritems / 2 + 1; 148779f95c82SChris Mason /* don't try to empty the node */ 1488bce4eae9SChris Mason if (max_push >= src_nritems) { 148979f95c82SChris Mason return 1; 1490bce4eae9SChris Mason } 1491252c38f0SYan 149279f95c82SChris Mason if (max_push < push_items) 149379f95c82SChris Mason push_items = max_push; 149479f95c82SChris Mason 14955f39d397SChris Mason memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items), 14965f39d397SChris Mason btrfs_node_key_ptr_offset(0), 14975f39d397SChris Mason (dst_nritems) * 14985f39d397SChris Mason sizeof(struct btrfs_key_ptr)); 1499d6025579SChris Mason 15005f39d397SChris Mason copy_extent_buffer(dst, src, 15015f39d397SChris Mason btrfs_node_key_ptr_offset(0), 15025f39d397SChris Mason btrfs_node_key_ptr_offset(src_nritems - push_items), 1503123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 150479f95c82SChris Mason 15055f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 15065f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 150779f95c82SChris Mason 15085f39d397SChris Mason btrfs_mark_buffer_dirty(src); 15095f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 151079f95c82SChris Mason return ret; 151179f95c82SChris Mason } 151279f95c82SChris Mason 151379f95c82SChris Mason /* 151497571fd0SChris Mason * helper function to insert a new root level in the tree. 151597571fd0SChris Mason * A new node is allocated, and a single item is inserted to 151697571fd0SChris Mason * point to the existing root 1517aa5d6bedSChris Mason * 1518aa5d6bedSChris Mason * returns zero on success or < 0 on failure. 151997571fd0SChris Mason */ 152098ed5174SChris Mason static int noinline insert_new_root(struct btrfs_trans_handle *trans, 15215f39d397SChris Mason struct btrfs_root *root, 15225f39d397SChris Mason struct btrfs_path *path, int level) 152374123bd7SChris Mason { 15247bb86316SChris Mason u64 root_gen; 15257bb86316SChris Mason u64 lower_gen; 15265f39d397SChris Mason struct extent_buffer *lower; 15275f39d397SChris Mason struct extent_buffer *c; 1528925baeddSChris Mason struct extent_buffer *old; 15295f39d397SChris Mason struct btrfs_disk_key lower_key; 15305c680ed6SChris Mason 15315c680ed6SChris Mason BUG_ON(path->nodes[level]); 15325c680ed6SChris Mason BUG_ON(path->nodes[level-1] != root->node); 15335c680ed6SChris Mason 15347bb86316SChris Mason if (root->ref_cows) 15357bb86316SChris Mason root_gen = trans->transid; 15367bb86316SChris Mason else 15377bb86316SChris Mason root_gen = 0; 15387bb86316SChris Mason 15397bb86316SChris Mason lower = path->nodes[level-1]; 15407bb86316SChris Mason if (level == 1) 15417bb86316SChris Mason btrfs_item_key(lower, &lower_key, 0); 15427bb86316SChris Mason else 15437bb86316SChris Mason btrfs_node_key(lower, &lower_key, 0); 15447bb86316SChris Mason 1545925baeddSChris Mason c = btrfs_alloc_free_block(trans, root, root->nodesize, 15467bb86316SChris Mason root->root_key.objectid, 15477bb86316SChris Mason root_gen, lower_key.objectid, level, 1548db94535dSChris Mason root->node->start, 0); 15495f39d397SChris Mason if (IS_ERR(c)) 15505f39d397SChris Mason return PTR_ERR(c); 1551925baeddSChris Mason 15525f39d397SChris Mason memset_extent_buffer(c, 0, 0, root->nodesize); 15535f39d397SChris Mason btrfs_set_header_nritems(c, 1); 15545f39d397SChris Mason btrfs_set_header_level(c, level); 1555db94535dSChris Mason btrfs_set_header_bytenr(c, c->start); 15565f39d397SChris Mason btrfs_set_header_generation(c, trans->transid); 15575f39d397SChris Mason btrfs_set_header_owner(c, root->root_key.objectid); 1558d5719762SChris Mason 15595f39d397SChris Mason write_extent_buffer(c, root->fs_info->fsid, 15605f39d397SChris Mason (unsigned long)btrfs_header_fsid(c), 15615f39d397SChris Mason BTRFS_FSID_SIZE); 1562e17cade2SChris Mason 1563e17cade2SChris Mason write_extent_buffer(c, root->fs_info->chunk_tree_uuid, 1564e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(c), 1565e17cade2SChris Mason BTRFS_UUID_SIZE); 1566e17cade2SChris Mason 15675f39d397SChris Mason btrfs_set_node_key(c, &lower_key, 0); 1568db94535dSChris Mason btrfs_set_node_blockptr(c, 0, lower->start); 15697bb86316SChris Mason lower_gen = btrfs_header_generation(lower); 15707bb86316SChris Mason WARN_ON(lower_gen == 0); 15717bb86316SChris Mason 15727bb86316SChris Mason btrfs_set_node_ptr_generation(c, 0, lower_gen); 15735f39d397SChris Mason 15745f39d397SChris Mason btrfs_mark_buffer_dirty(c); 1575d5719762SChris Mason 1576925baeddSChris Mason spin_lock(&root->node_lock); 1577925baeddSChris Mason old = root->node; 15785f39d397SChris Mason root->node = c; 1579925baeddSChris Mason spin_unlock(&root->node_lock); 1580925baeddSChris Mason 1581925baeddSChris Mason /* the super has an extra ref to root->node */ 1582925baeddSChris Mason free_extent_buffer(old); 1583925baeddSChris Mason 15840b86a832SChris Mason add_root_to_dirty_list(root); 15855f39d397SChris Mason extent_buffer_get(c); 15865f39d397SChris Mason path->nodes[level] = c; 1587925baeddSChris Mason path->locks[level] = 1; 158874123bd7SChris Mason path->slots[level] = 0; 15897bb86316SChris Mason 15907bb86316SChris Mason if (root->ref_cows && lower_gen != trans->transid) { 15917bb86316SChris Mason struct btrfs_path *back_path = btrfs_alloc_path(); 15927bb86316SChris Mason int ret; 1593925baeddSChris Mason mutex_lock(&root->fs_info->alloc_mutex); 15947bb86316SChris Mason ret = btrfs_insert_extent_backref(trans, 15957bb86316SChris Mason root->fs_info->extent_root, 15967bb86316SChris Mason path, lower->start, 15977bb86316SChris Mason root->root_key.objectid, 15987bb86316SChris Mason trans->transid, 0, 0); 15997bb86316SChris Mason BUG_ON(ret); 1600925baeddSChris Mason mutex_unlock(&root->fs_info->alloc_mutex); 16017bb86316SChris Mason btrfs_free_path(back_path); 16027bb86316SChris Mason } 160374123bd7SChris Mason return 0; 160474123bd7SChris Mason } 16055c680ed6SChris Mason 16065c680ed6SChris Mason /* 16075c680ed6SChris Mason * worker function to insert a single pointer in a node. 16085c680ed6SChris Mason * the node should have enough room for the pointer already 160997571fd0SChris Mason * 16105c680ed6SChris Mason * slot and level indicate where you want the key to go, and 16115c680ed6SChris Mason * blocknr is the block the key points to. 1612aa5d6bedSChris Mason * 1613aa5d6bedSChris Mason * returns zero on success and < 0 on any error 16145c680ed6SChris Mason */ 1615e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root 1616e089f05cSChris Mason *root, struct btrfs_path *path, struct btrfs_disk_key 1617db94535dSChris Mason *key, u64 bytenr, int slot, int level) 16185c680ed6SChris Mason { 16195f39d397SChris Mason struct extent_buffer *lower; 16205c680ed6SChris Mason int nritems; 16215c680ed6SChris Mason 16225c680ed6SChris Mason BUG_ON(!path->nodes[level]); 16235f39d397SChris Mason lower = path->nodes[level]; 16245f39d397SChris Mason nritems = btrfs_header_nritems(lower); 162574123bd7SChris Mason if (slot > nritems) 162674123bd7SChris Mason BUG(); 1627123abc88SChris Mason if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root)) 162874123bd7SChris Mason BUG(); 162974123bd7SChris Mason if (slot != nritems) { 16305f39d397SChris Mason memmove_extent_buffer(lower, 16315f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 16325f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 1633123abc88SChris Mason (nritems - slot) * sizeof(struct btrfs_key_ptr)); 163474123bd7SChris Mason } 16355f39d397SChris Mason btrfs_set_node_key(lower, key, slot); 1636db94535dSChris Mason btrfs_set_node_blockptr(lower, slot, bytenr); 163774493f7aSChris Mason WARN_ON(trans->transid == 0); 163874493f7aSChris Mason btrfs_set_node_ptr_generation(lower, slot, trans->transid); 16395f39d397SChris Mason btrfs_set_header_nritems(lower, nritems + 1); 16405f39d397SChris Mason btrfs_mark_buffer_dirty(lower); 164174123bd7SChris Mason return 0; 164274123bd7SChris Mason } 164374123bd7SChris Mason 164497571fd0SChris Mason /* 164597571fd0SChris Mason * split the node at the specified level in path in two. 164697571fd0SChris Mason * The path is corrected to point to the appropriate node after the split 164797571fd0SChris Mason * 164897571fd0SChris Mason * Before splitting this tries to make some room in the node by pushing 164997571fd0SChris Mason * left and right, if either one works, it returns right away. 1650aa5d6bedSChris Mason * 1651aa5d6bedSChris Mason * returns 0 on success and < 0 on failure 165297571fd0SChris Mason */ 1653e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 1654e089f05cSChris Mason *root, struct btrfs_path *path, int level) 1655be0e5c09SChris Mason { 16567bb86316SChris Mason u64 root_gen; 16575f39d397SChris Mason struct extent_buffer *c; 16585f39d397SChris Mason struct extent_buffer *split; 16595f39d397SChris Mason struct btrfs_disk_key disk_key; 1660be0e5c09SChris Mason int mid; 16615c680ed6SChris Mason int ret; 1662aa5d6bedSChris Mason int wret; 16637518a238SChris Mason u32 c_nritems; 1664be0e5c09SChris Mason 16655f39d397SChris Mason c = path->nodes[level]; 16667bb86316SChris Mason WARN_ON(btrfs_header_generation(c) != trans->transid); 16675f39d397SChris Mason if (c == root->node) { 16685c680ed6SChris Mason /* trying to split the root, lets make a new one */ 1669e089f05cSChris Mason ret = insert_new_root(trans, root, path, level + 1); 16705c680ed6SChris Mason if (ret) 16715c680ed6SChris Mason return ret; 1672e66f709bSChris Mason } else { 1673e66f709bSChris Mason ret = push_nodes_for_insert(trans, root, path, level); 16745f39d397SChris Mason c = path->nodes[level]; 16755f39d397SChris Mason if (!ret && btrfs_header_nritems(c) < 1676c448acf0SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) 1677e66f709bSChris Mason return 0; 167854aa1f4dSChris Mason if (ret < 0) 167954aa1f4dSChris Mason return ret; 16805c680ed6SChris Mason } 1681e66f709bSChris Mason 16825f39d397SChris Mason c_nritems = btrfs_header_nritems(c); 16837bb86316SChris Mason if (root->ref_cows) 16847bb86316SChris Mason root_gen = trans->transid; 16857bb86316SChris Mason else 16867bb86316SChris Mason root_gen = 0; 16877bb86316SChris Mason 16887bb86316SChris Mason btrfs_node_key(c, &disk_key, 0); 1689925baeddSChris Mason split = btrfs_alloc_free_block(trans, root, root->nodesize, 16907bb86316SChris Mason root->root_key.objectid, 16917bb86316SChris Mason root_gen, 16927bb86316SChris Mason btrfs_disk_key_objectid(&disk_key), 16937bb86316SChris Mason level, c->start, 0); 16945f39d397SChris Mason if (IS_ERR(split)) 16955f39d397SChris Mason return PTR_ERR(split); 169654aa1f4dSChris Mason 16975f39d397SChris Mason btrfs_set_header_flags(split, btrfs_header_flags(c)); 16985f39d397SChris Mason btrfs_set_header_level(split, btrfs_header_level(c)); 1699db94535dSChris Mason btrfs_set_header_bytenr(split, split->start); 17005f39d397SChris Mason btrfs_set_header_generation(split, trans->transid); 17015f39d397SChris Mason btrfs_set_header_owner(split, root->root_key.objectid); 170263b10fc4SChris Mason btrfs_set_header_flags(split, 0); 17035f39d397SChris Mason write_extent_buffer(split, root->fs_info->fsid, 17045f39d397SChris Mason (unsigned long)btrfs_header_fsid(split), 17055f39d397SChris Mason BTRFS_FSID_SIZE); 1706e17cade2SChris Mason write_extent_buffer(split, root->fs_info->chunk_tree_uuid, 1707e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(split), 1708e17cade2SChris Mason BTRFS_UUID_SIZE); 17095f39d397SChris Mason 17107518a238SChris Mason mid = (c_nritems + 1) / 2; 17115f39d397SChris Mason 17125f39d397SChris Mason copy_extent_buffer(split, c, 17135f39d397SChris Mason btrfs_node_key_ptr_offset(0), 17145f39d397SChris Mason btrfs_node_key_ptr_offset(mid), 1715123abc88SChris Mason (c_nritems - mid) * sizeof(struct btrfs_key_ptr)); 17165f39d397SChris Mason btrfs_set_header_nritems(split, c_nritems - mid); 17175f39d397SChris Mason btrfs_set_header_nritems(c, mid); 1718aa5d6bedSChris Mason ret = 0; 1719aa5d6bedSChris Mason 17205f39d397SChris Mason btrfs_mark_buffer_dirty(c); 17215f39d397SChris Mason btrfs_mark_buffer_dirty(split); 17225f39d397SChris Mason 17235f39d397SChris Mason btrfs_node_key(split, &disk_key, 0); 1724db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, split->start, 17255f39d397SChris Mason path->slots[level + 1] + 1, 1726123abc88SChris Mason level + 1); 1727aa5d6bedSChris Mason if (wret) 1728aa5d6bedSChris Mason ret = wret; 1729aa5d6bedSChris Mason 17305de08d7dSChris Mason if (path->slots[level] >= mid) { 17315c680ed6SChris Mason path->slots[level] -= mid; 1732925baeddSChris Mason btrfs_tree_unlock(c); 17335f39d397SChris Mason free_extent_buffer(c); 17345f39d397SChris Mason path->nodes[level] = split; 17355c680ed6SChris Mason path->slots[level + 1] += 1; 1736eb60ceacSChris Mason } else { 1737925baeddSChris Mason btrfs_tree_unlock(split); 17385f39d397SChris Mason free_extent_buffer(split); 1739be0e5c09SChris Mason } 1740aa5d6bedSChris Mason return ret; 1741be0e5c09SChris Mason } 1742be0e5c09SChris Mason 174374123bd7SChris Mason /* 174474123bd7SChris Mason * how many bytes are required to store the items in a leaf. start 174574123bd7SChris Mason * and nr indicate which items in the leaf to check. This totals up the 174674123bd7SChris Mason * space used both by the item structs and the item data 174774123bd7SChris Mason */ 17485f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr) 1749be0e5c09SChris Mason { 1750be0e5c09SChris Mason int data_len; 17515f39d397SChris Mason int nritems = btrfs_header_nritems(l); 1752d4dbff95SChris Mason int end = min(nritems, start + nr) - 1; 1753be0e5c09SChris Mason 1754be0e5c09SChris Mason if (!nr) 1755be0e5c09SChris Mason return 0; 17565f39d397SChris Mason data_len = btrfs_item_end_nr(l, start); 17575f39d397SChris Mason data_len = data_len - btrfs_item_offset_nr(l, end); 17580783fcfcSChris Mason data_len += sizeof(struct btrfs_item) * nr; 1759d4dbff95SChris Mason WARN_ON(data_len < 0); 1760be0e5c09SChris Mason return data_len; 1761be0e5c09SChris Mason } 1762be0e5c09SChris Mason 176374123bd7SChris Mason /* 1764d4dbff95SChris Mason * The space between the end of the leaf items and 1765d4dbff95SChris Mason * the start of the leaf data. IOW, how much room 1766d4dbff95SChris Mason * the leaf has left for both items and data 1767d4dbff95SChris Mason */ 17685f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf) 1769d4dbff95SChris Mason { 17705f39d397SChris Mason int nritems = btrfs_header_nritems(leaf); 17715f39d397SChris Mason int ret; 17725f39d397SChris Mason ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems); 17735f39d397SChris Mason if (ret < 0) { 17745f39d397SChris Mason printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n", 1775ae2f5411SJens Axboe ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root), 17765f39d397SChris Mason leaf_space_used(leaf, 0, nritems), nritems); 17775f39d397SChris Mason } 17785f39d397SChris Mason return ret; 1779d4dbff95SChris Mason } 1780d4dbff95SChris Mason 1781d4dbff95SChris Mason /* 178200ec4c51SChris Mason * push some data in the path leaf to the right, trying to free up at 178300ec4c51SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 1784aa5d6bedSChris Mason * 1785aa5d6bedSChris Mason * returns 1 if the push failed because the other node didn't have enough 1786aa5d6bedSChris Mason * room, 0 if everything worked out and < 0 if there were major errors. 178700ec4c51SChris Mason */ 1788e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root 178934a38218SChris Mason *root, struct btrfs_path *path, int data_size, 179034a38218SChris Mason int empty) 179100ec4c51SChris Mason { 17925f39d397SChris Mason struct extent_buffer *left = path->nodes[0]; 17935f39d397SChris Mason struct extent_buffer *right; 17945f39d397SChris Mason struct extent_buffer *upper; 17955f39d397SChris Mason struct btrfs_disk_key disk_key; 179600ec4c51SChris Mason int slot; 179734a38218SChris Mason u32 i; 179800ec4c51SChris Mason int free_space; 179900ec4c51SChris Mason int push_space = 0; 180000ec4c51SChris Mason int push_items = 0; 18010783fcfcSChris Mason struct btrfs_item *item; 18027518a238SChris Mason u32 left_nritems; 180334a38218SChris Mason u32 nr; 18047518a238SChris Mason u32 right_nritems; 18055f39d397SChris Mason u32 data_end; 1806db94535dSChris Mason u32 this_item_size; 180754aa1f4dSChris Mason int ret; 180800ec4c51SChris Mason 180900ec4c51SChris Mason slot = path->slots[1]; 181000ec4c51SChris Mason if (!path->nodes[1]) { 181100ec4c51SChris Mason return 1; 181200ec4c51SChris Mason } 181300ec4c51SChris Mason upper = path->nodes[1]; 18145f39d397SChris Mason if (slot >= btrfs_header_nritems(upper) - 1) 181500ec4c51SChris Mason return 1; 18165f39d397SChris Mason 1817*a2135011SChris Mason WARN_ON(!btrfs_tree_locked(path->nodes[1])); 1818*a2135011SChris Mason 1819ca7a79adSChris Mason right = read_node_slot(root, upper, slot + 1); 1820925baeddSChris Mason btrfs_tree_lock(right); 1821123abc88SChris Mason free_space = btrfs_leaf_free_space(root, right); 1822925baeddSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) 1823925baeddSChris Mason goto out_unlock; 182402217ed2SChris Mason 18255f39d397SChris Mason /* cow and double check */ 18265f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, upper, 18275f39d397SChris Mason slot + 1, &right); 1828925baeddSChris Mason if (ret) 1829925baeddSChris Mason goto out_unlock; 1830925baeddSChris Mason 18315f39d397SChris Mason free_space = btrfs_leaf_free_space(root, right); 1832925baeddSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) 1833925baeddSChris Mason goto out_unlock; 18345f39d397SChris Mason 18355f39d397SChris Mason left_nritems = btrfs_header_nritems(left); 1836925baeddSChris Mason if (left_nritems == 0) 1837925baeddSChris Mason goto out_unlock; 18385f39d397SChris Mason 183934a38218SChris Mason if (empty) 184034a38218SChris Mason nr = 0; 184134a38218SChris Mason else 184234a38218SChris Mason nr = 1; 184334a38218SChris Mason 184434a38218SChris Mason i = left_nritems - 1; 184534a38218SChris Mason while (i >= nr) { 18465f39d397SChris Mason item = btrfs_item_nr(left, i); 1847db94535dSChris Mason 184800ec4c51SChris Mason if (path->slots[0] == i) 184900ec4c51SChris Mason push_space += data_size + sizeof(*item); 1850db94535dSChris Mason 1851db94535dSChris Mason if (!left->map_token) { 1852db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 1853db94535dSChris Mason sizeof(struct btrfs_item), 1854db94535dSChris Mason &left->map_token, &left->kaddr, 1855db94535dSChris Mason &left->map_start, &left->map_len, 1856db94535dSChris Mason KM_USER1); 1857db94535dSChris Mason } 1858db94535dSChris Mason 1859db94535dSChris Mason this_item_size = btrfs_item_size(left, item); 1860db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 186100ec4c51SChris Mason break; 186200ec4c51SChris Mason push_items++; 1863db94535dSChris Mason push_space += this_item_size + sizeof(*item); 186434a38218SChris Mason if (i == 0) 186534a38218SChris Mason break; 186634a38218SChris Mason i--; 1867db94535dSChris Mason } 1868db94535dSChris Mason if (left->map_token) { 1869db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 1870db94535dSChris Mason left->map_token = NULL; 187100ec4c51SChris Mason } 18725f39d397SChris Mason 1873925baeddSChris Mason if (push_items == 0) 1874925baeddSChris Mason goto out_unlock; 18755f39d397SChris Mason 187634a38218SChris Mason if (!empty && push_items == left_nritems) 1877a429e513SChris Mason WARN_ON(1); 18785f39d397SChris Mason 187900ec4c51SChris Mason /* push left to right */ 18805f39d397SChris Mason right_nritems = btrfs_header_nritems(right); 188134a38218SChris Mason 18825f39d397SChris Mason push_space = btrfs_item_end_nr(left, left_nritems - push_items); 1883123abc88SChris Mason push_space -= leaf_data_end(root, left); 18845f39d397SChris Mason 188500ec4c51SChris Mason /* make room in the right data area */ 18865f39d397SChris Mason data_end = leaf_data_end(root, right); 18875f39d397SChris Mason memmove_extent_buffer(right, 18885f39d397SChris Mason btrfs_leaf_data(right) + data_end - push_space, 18895f39d397SChris Mason btrfs_leaf_data(right) + data_end, 18905f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root) - data_end); 18915f39d397SChris Mason 189200ec4c51SChris Mason /* copy from the left data area */ 18935f39d397SChris Mason copy_extent_buffer(right, left, btrfs_leaf_data(right) + 1894d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 1895d6025579SChris Mason btrfs_leaf_data(left) + leaf_data_end(root, left), 1896d6025579SChris Mason push_space); 18975f39d397SChris Mason 18985f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(push_items), 18995f39d397SChris Mason btrfs_item_nr_offset(0), 19000783fcfcSChris Mason right_nritems * sizeof(struct btrfs_item)); 19015f39d397SChris Mason 190200ec4c51SChris Mason /* copy the items from left to right */ 19035f39d397SChris Mason copy_extent_buffer(right, left, btrfs_item_nr_offset(0), 19045f39d397SChris Mason btrfs_item_nr_offset(left_nritems - push_items), 19050783fcfcSChris Mason push_items * sizeof(struct btrfs_item)); 190600ec4c51SChris Mason 190700ec4c51SChris Mason /* update the item pointers */ 19087518a238SChris Mason right_nritems += push_items; 19095f39d397SChris Mason btrfs_set_header_nritems(right, right_nritems); 1910123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 19117518a238SChris Mason for (i = 0; i < right_nritems; i++) { 19125f39d397SChris Mason item = btrfs_item_nr(right, i); 1913db94535dSChris Mason if (!right->map_token) { 1914db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1915db94535dSChris Mason sizeof(struct btrfs_item), 1916db94535dSChris Mason &right->map_token, &right->kaddr, 1917db94535dSChris Mason &right->map_start, &right->map_len, 1918db94535dSChris Mason KM_USER1); 1919db94535dSChris Mason } 1920db94535dSChris Mason push_space -= btrfs_item_size(right, item); 1921db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 1922db94535dSChris Mason } 1923db94535dSChris Mason 1924db94535dSChris Mason if (right->map_token) { 1925db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1926db94535dSChris Mason right->map_token = NULL; 192700ec4c51SChris Mason } 19287518a238SChris Mason left_nritems -= push_items; 19295f39d397SChris Mason btrfs_set_header_nritems(left, left_nritems); 193000ec4c51SChris Mason 193134a38218SChris Mason if (left_nritems) 19325f39d397SChris Mason btrfs_mark_buffer_dirty(left); 19335f39d397SChris Mason btrfs_mark_buffer_dirty(right); 1934a429e513SChris Mason 19355f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 19365f39d397SChris Mason btrfs_set_node_key(upper, &disk_key, slot + 1); 1937d6025579SChris Mason btrfs_mark_buffer_dirty(upper); 193802217ed2SChris Mason 193900ec4c51SChris Mason /* then fixup the leaf pointer in the path */ 19407518a238SChris Mason if (path->slots[0] >= left_nritems) { 19417518a238SChris Mason path->slots[0] -= left_nritems; 1942925baeddSChris Mason if (btrfs_header_nritems(path->nodes[0]) == 0) 1943925baeddSChris Mason clean_tree_block(trans, root, path->nodes[0]); 1944925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 19455f39d397SChris Mason free_extent_buffer(path->nodes[0]); 19465f39d397SChris Mason path->nodes[0] = right; 194700ec4c51SChris Mason path->slots[1] += 1; 194800ec4c51SChris Mason } else { 1949925baeddSChris Mason btrfs_tree_unlock(right); 19505f39d397SChris Mason free_extent_buffer(right); 195100ec4c51SChris Mason } 195200ec4c51SChris Mason return 0; 1953925baeddSChris Mason 1954925baeddSChris Mason out_unlock: 1955925baeddSChris Mason btrfs_tree_unlock(right); 1956925baeddSChris Mason free_extent_buffer(right); 1957925baeddSChris Mason return 1; 195800ec4c51SChris Mason } 1959925baeddSChris Mason 196000ec4c51SChris Mason /* 196174123bd7SChris Mason * push some data in the path leaf to the left, trying to free up at 196274123bd7SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 196374123bd7SChris Mason */ 1964e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root 196534a38218SChris Mason *root, struct btrfs_path *path, int data_size, 196634a38218SChris Mason int empty) 1967be0e5c09SChris Mason { 19685f39d397SChris Mason struct btrfs_disk_key disk_key; 19695f39d397SChris Mason struct extent_buffer *right = path->nodes[0]; 19705f39d397SChris Mason struct extent_buffer *left; 1971be0e5c09SChris Mason int slot; 1972be0e5c09SChris Mason int i; 1973be0e5c09SChris Mason int free_space; 1974be0e5c09SChris Mason int push_space = 0; 1975be0e5c09SChris Mason int push_items = 0; 19760783fcfcSChris Mason struct btrfs_item *item; 19777518a238SChris Mason u32 old_left_nritems; 19785f39d397SChris Mason u32 right_nritems; 197934a38218SChris Mason u32 nr; 1980aa5d6bedSChris Mason int ret = 0; 1981aa5d6bedSChris Mason int wret; 1982db94535dSChris Mason u32 this_item_size; 1983db94535dSChris Mason u32 old_left_item_size; 1984be0e5c09SChris Mason 1985be0e5c09SChris Mason slot = path->slots[1]; 19865f39d397SChris Mason if (slot == 0) 1987be0e5c09SChris Mason return 1; 19885f39d397SChris Mason if (!path->nodes[1]) 1989be0e5c09SChris Mason return 1; 19905f39d397SChris Mason 19913685f791SChris Mason right_nritems = btrfs_header_nritems(right); 19923685f791SChris Mason if (right_nritems == 0) { 19933685f791SChris Mason return 1; 19943685f791SChris Mason } 19953685f791SChris Mason 1996*a2135011SChris Mason WARN_ON(!btrfs_tree_locked(path->nodes[1])); 1997*a2135011SChris Mason 1998ca7a79adSChris Mason left = read_node_slot(root, path->nodes[1], slot - 1); 1999925baeddSChris Mason btrfs_tree_lock(left); 2000123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 20010783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 2002925baeddSChris Mason ret = 1; 2003925baeddSChris Mason goto out; 2004be0e5c09SChris Mason } 200502217ed2SChris Mason 200602217ed2SChris Mason /* cow and double check */ 20075f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, 20085f39d397SChris Mason path->nodes[1], slot - 1, &left); 200954aa1f4dSChris Mason if (ret) { 201054aa1f4dSChris Mason /* we hit -ENOSPC, but it isn't fatal here */ 2011925baeddSChris Mason ret = 1; 2012925baeddSChris Mason goto out; 201354aa1f4dSChris Mason } 20143685f791SChris Mason 2015123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 20160783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 2017925baeddSChris Mason ret = 1; 2018925baeddSChris Mason goto out; 201902217ed2SChris Mason } 202002217ed2SChris Mason 202134a38218SChris Mason if (empty) 202234a38218SChris Mason nr = right_nritems; 202334a38218SChris Mason else 202434a38218SChris Mason nr = right_nritems - 1; 202534a38218SChris Mason 202634a38218SChris Mason for (i = 0; i < nr; i++) { 20275f39d397SChris Mason item = btrfs_item_nr(right, i); 2028db94535dSChris Mason if (!right->map_token) { 2029db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2030db94535dSChris Mason sizeof(struct btrfs_item), 2031db94535dSChris Mason &right->map_token, &right->kaddr, 2032db94535dSChris Mason &right->map_start, &right->map_len, 2033db94535dSChris Mason KM_USER1); 2034db94535dSChris Mason } 2035db94535dSChris Mason 2036be0e5c09SChris Mason if (path->slots[0] == i) 2037be0e5c09SChris Mason push_space += data_size + sizeof(*item); 2038db94535dSChris Mason 2039db94535dSChris Mason this_item_size = btrfs_item_size(right, item); 2040db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 2041be0e5c09SChris Mason break; 2042db94535dSChris Mason 2043be0e5c09SChris Mason push_items++; 2044db94535dSChris Mason push_space += this_item_size + sizeof(*item); 2045be0e5c09SChris Mason } 2046db94535dSChris Mason 2047db94535dSChris Mason if (right->map_token) { 2048db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2049db94535dSChris Mason right->map_token = NULL; 2050db94535dSChris Mason } 2051db94535dSChris Mason 2052be0e5c09SChris Mason if (push_items == 0) { 2053925baeddSChris Mason ret = 1; 2054925baeddSChris Mason goto out; 2055be0e5c09SChris Mason } 205634a38218SChris Mason if (!empty && push_items == btrfs_header_nritems(right)) 2057a429e513SChris Mason WARN_ON(1); 20585f39d397SChris Mason 2059be0e5c09SChris Mason /* push data from right to left */ 20605f39d397SChris Mason copy_extent_buffer(left, right, 20615f39d397SChris Mason btrfs_item_nr_offset(btrfs_header_nritems(left)), 20625f39d397SChris Mason btrfs_item_nr_offset(0), 20635f39d397SChris Mason push_items * sizeof(struct btrfs_item)); 20645f39d397SChris Mason 2065123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root) - 20665f39d397SChris Mason btrfs_item_offset_nr(right, push_items -1); 20675f39d397SChris Mason 20685f39d397SChris Mason copy_extent_buffer(left, right, btrfs_leaf_data(left) + 2069d6025579SChris Mason leaf_data_end(root, left) - push_space, 2070123abc88SChris Mason btrfs_leaf_data(right) + 20715f39d397SChris Mason btrfs_item_offset_nr(right, push_items - 1), 2072be0e5c09SChris Mason push_space); 20735f39d397SChris Mason old_left_nritems = btrfs_header_nritems(left); 2074eb60ceacSChris Mason BUG_ON(old_left_nritems < 0); 2075eb60ceacSChris Mason 2076db94535dSChris Mason old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1); 2077be0e5c09SChris Mason for (i = old_left_nritems; i < old_left_nritems + push_items; i++) { 20785f39d397SChris Mason u32 ioff; 2079db94535dSChris Mason 20805f39d397SChris Mason item = btrfs_item_nr(left, i); 2081db94535dSChris Mason if (!left->map_token) { 2082db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 2083db94535dSChris Mason sizeof(struct btrfs_item), 2084db94535dSChris Mason &left->map_token, &left->kaddr, 2085db94535dSChris Mason &left->map_start, &left->map_len, 2086db94535dSChris Mason KM_USER1); 2087db94535dSChris Mason } 2088db94535dSChris Mason 20895f39d397SChris Mason ioff = btrfs_item_offset(left, item); 20905f39d397SChris Mason btrfs_set_item_offset(left, item, 2091db94535dSChris Mason ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size)); 2092be0e5c09SChris Mason } 20935f39d397SChris Mason btrfs_set_header_nritems(left, old_left_nritems + push_items); 2094db94535dSChris Mason if (left->map_token) { 2095db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 2096db94535dSChris Mason left->map_token = NULL; 2097db94535dSChris Mason } 2098be0e5c09SChris Mason 2099be0e5c09SChris Mason /* fixup right node */ 210034a38218SChris Mason if (push_items > right_nritems) { 210134a38218SChris Mason printk("push items %d nr %u\n", push_items, right_nritems); 210234a38218SChris Mason WARN_ON(1); 210334a38218SChris Mason } 210434a38218SChris Mason 210534a38218SChris Mason if (push_items < right_nritems) { 21065f39d397SChris Mason push_space = btrfs_item_offset_nr(right, push_items - 1) - 2107123abc88SChris Mason leaf_data_end(root, right); 21085f39d397SChris Mason memmove_extent_buffer(right, btrfs_leaf_data(right) + 2109d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 2110d6025579SChris Mason btrfs_leaf_data(right) + 2111123abc88SChris Mason leaf_data_end(root, right), push_space); 21125f39d397SChris Mason 21135f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(0), 21145f39d397SChris Mason btrfs_item_nr_offset(push_items), 21155f39d397SChris Mason (btrfs_header_nritems(right) - push_items) * 21160783fcfcSChris Mason sizeof(struct btrfs_item)); 211734a38218SChris Mason } 2118eef1c494SYan right_nritems -= push_items; 2119eef1c494SYan btrfs_set_header_nritems(right, right_nritems); 2120123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 21215f39d397SChris Mason for (i = 0; i < right_nritems; i++) { 21225f39d397SChris Mason item = btrfs_item_nr(right, i); 2123db94535dSChris Mason 2124db94535dSChris Mason if (!right->map_token) { 2125db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2126db94535dSChris Mason sizeof(struct btrfs_item), 2127db94535dSChris Mason &right->map_token, &right->kaddr, 2128db94535dSChris Mason &right->map_start, &right->map_len, 2129db94535dSChris Mason KM_USER1); 2130db94535dSChris Mason } 2131db94535dSChris Mason 2132db94535dSChris Mason push_space = push_space - btrfs_item_size(right, item); 2133db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 2134db94535dSChris Mason } 2135db94535dSChris Mason if (right->map_token) { 2136db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2137db94535dSChris Mason right->map_token = NULL; 2138be0e5c09SChris Mason } 2139eb60ceacSChris Mason 21405f39d397SChris Mason btrfs_mark_buffer_dirty(left); 214134a38218SChris Mason if (right_nritems) 21425f39d397SChris Mason btrfs_mark_buffer_dirty(right); 2143098f59c2SChris Mason 21445f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 21455f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, 1); 2146aa5d6bedSChris Mason if (wret) 2147aa5d6bedSChris Mason ret = wret; 2148be0e5c09SChris Mason 2149be0e5c09SChris Mason /* then fixup the leaf pointer in the path */ 2150be0e5c09SChris Mason if (path->slots[0] < push_items) { 2151be0e5c09SChris Mason path->slots[0] += old_left_nritems; 2152925baeddSChris Mason if (btrfs_header_nritems(path->nodes[0]) == 0) 2153925baeddSChris Mason clean_tree_block(trans, root, path->nodes[0]); 2154925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 21555f39d397SChris Mason free_extent_buffer(path->nodes[0]); 21565f39d397SChris Mason path->nodes[0] = left; 2157be0e5c09SChris Mason path->slots[1] -= 1; 2158be0e5c09SChris Mason } else { 2159925baeddSChris Mason btrfs_tree_unlock(left); 21605f39d397SChris Mason free_extent_buffer(left); 2161be0e5c09SChris Mason path->slots[0] -= push_items; 2162be0e5c09SChris Mason } 2163eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 2164aa5d6bedSChris Mason return ret; 2165925baeddSChris Mason out: 2166925baeddSChris Mason btrfs_tree_unlock(left); 2167925baeddSChris Mason free_extent_buffer(left); 2168925baeddSChris Mason return ret; 2169be0e5c09SChris Mason } 2170be0e5c09SChris Mason 217174123bd7SChris Mason /* 217274123bd7SChris Mason * split the path's leaf in two, making sure there is at least data_size 217374123bd7SChris Mason * available for the resulting leaf level of the path. 2174aa5d6bedSChris Mason * 2175aa5d6bedSChris Mason * returns 0 if all went well and < 0 on failure. 217674123bd7SChris Mason */ 2177e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 2178d4dbff95SChris Mason *root, struct btrfs_key *ins_key, 2179cc0c5538SChris Mason struct btrfs_path *path, int data_size, int extend) 2180be0e5c09SChris Mason { 21817bb86316SChris Mason u64 root_gen; 21825f39d397SChris Mason struct extent_buffer *l; 21837518a238SChris Mason u32 nritems; 2184eb60ceacSChris Mason int mid; 2185eb60ceacSChris Mason int slot; 21865f39d397SChris Mason struct extent_buffer *right; 21870783fcfcSChris Mason int space_needed = data_size + sizeof(struct btrfs_item); 2188be0e5c09SChris Mason int data_copy_size; 2189be0e5c09SChris Mason int rt_data_off; 2190be0e5c09SChris Mason int i; 2191d4dbff95SChris Mason int ret = 0; 2192aa5d6bedSChris Mason int wret; 2193cc0c5538SChris Mason int double_split; 2194cc0c5538SChris Mason int num_doubles = 0; 2195d4dbff95SChris Mason struct btrfs_disk_key disk_key; 2196be0e5c09SChris Mason 2197cc0c5538SChris Mason if (extend) 2198cc0c5538SChris Mason space_needed = data_size; 2199cc0c5538SChris Mason 22007bb86316SChris Mason if (root->ref_cows) 22017bb86316SChris Mason root_gen = trans->transid; 22027bb86316SChris Mason else 22037bb86316SChris Mason root_gen = 0; 22047bb86316SChris Mason 220540689478SChris Mason /* first try to make some room by pushing left and right */ 22063685f791SChris Mason if (ins_key->type != BTRFS_DIR_ITEM_KEY) { 220734a38218SChris Mason wret = push_leaf_right(trans, root, path, data_size, 0); 22083326d1b0SChris Mason if (wret < 0) { 2209eaee50e8SChris Mason return wret; 22103326d1b0SChris Mason } 2211eaee50e8SChris Mason if (wret) { 221234a38218SChris Mason wret = push_leaf_left(trans, root, path, data_size, 0); 2213eaee50e8SChris Mason if (wret < 0) 2214eaee50e8SChris Mason return wret; 2215eaee50e8SChris Mason } 22165f39d397SChris Mason l = path->nodes[0]; 2217aa5d6bedSChris Mason 2218aa5d6bedSChris Mason /* did the pushes work? */ 2219cc0c5538SChris Mason if (btrfs_leaf_free_space(root, l) >= space_needed) 2220be0e5c09SChris Mason return 0; 22213326d1b0SChris Mason } 2222aa5d6bedSChris Mason 22235c680ed6SChris Mason if (!path->nodes[1]) { 2224e089f05cSChris Mason ret = insert_new_root(trans, root, path, 1); 22255c680ed6SChris Mason if (ret) 22265c680ed6SChris Mason return ret; 22275c680ed6SChris Mason } 2228cc0c5538SChris Mason again: 2229cc0c5538SChris Mason double_split = 0; 2230cc0c5538SChris Mason l = path->nodes[0]; 2231eb60ceacSChris Mason slot = path->slots[0]; 22325f39d397SChris Mason nritems = btrfs_header_nritems(l); 2233eb60ceacSChris Mason mid = (nritems + 1)/ 2; 223454aa1f4dSChris Mason 22357bb86316SChris Mason btrfs_item_key(l, &disk_key, 0); 22367bb86316SChris Mason 2237925baeddSChris Mason right = btrfs_alloc_free_block(trans, root, root->leafsize, 22387bb86316SChris Mason root->root_key.objectid, 22397bb86316SChris Mason root_gen, disk_key.objectid, 0, 2240db94535dSChris Mason l->start, 0); 2241cea9e445SChris Mason if (IS_ERR(right)) { 2242cea9e445SChris Mason BUG_ON(1); 22435f39d397SChris Mason return PTR_ERR(right); 2244cea9e445SChris Mason } 224554aa1f4dSChris Mason 22465f39d397SChris Mason memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header)); 2247db94535dSChris Mason btrfs_set_header_bytenr(right, right->start); 22485f39d397SChris Mason btrfs_set_header_generation(right, trans->transid); 22495f39d397SChris Mason btrfs_set_header_owner(right, root->root_key.objectid); 22505f39d397SChris Mason btrfs_set_header_level(right, 0); 22515f39d397SChris Mason write_extent_buffer(right, root->fs_info->fsid, 22525f39d397SChris Mason (unsigned long)btrfs_header_fsid(right), 22535f39d397SChris Mason BTRFS_FSID_SIZE); 2254e17cade2SChris Mason 2255e17cade2SChris Mason write_extent_buffer(right, root->fs_info->chunk_tree_uuid, 2256e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(right), 2257e17cade2SChris Mason BTRFS_UUID_SIZE); 2258d4dbff95SChris Mason if (mid <= slot) { 2259d4dbff95SChris Mason if (nritems == 1 || 2260d4dbff95SChris Mason leaf_space_used(l, mid, nritems - mid) + space_needed > 2261d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 2262d4dbff95SChris Mason if (slot >= nritems) { 2263d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 22645f39d397SChris Mason btrfs_set_header_nritems(right, 0); 2265d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 2266db94535dSChris Mason &disk_key, right->start, 2267d4dbff95SChris Mason path->slots[1] + 1, 1); 2268d4dbff95SChris Mason if (wret) 2269d4dbff95SChris Mason ret = wret; 2270925baeddSChris Mason 2271925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 22725f39d397SChris Mason free_extent_buffer(path->nodes[0]); 22735f39d397SChris Mason path->nodes[0] = right; 2274d4dbff95SChris Mason path->slots[0] = 0; 2275d4dbff95SChris Mason path->slots[1] += 1; 22760ef8b242SChris Mason btrfs_mark_buffer_dirty(right); 2277d4dbff95SChris Mason return ret; 2278d4dbff95SChris Mason } 2279d4dbff95SChris Mason mid = slot; 22803326d1b0SChris Mason if (mid != nritems && 22813326d1b0SChris Mason leaf_space_used(l, mid, nritems - mid) + 22823326d1b0SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 2283d4dbff95SChris Mason double_split = 1; 2284d4dbff95SChris Mason } 22853326d1b0SChris Mason } 2286d4dbff95SChris Mason } else { 2287d4dbff95SChris Mason if (leaf_space_used(l, 0, mid + 1) + space_needed > 2288d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 2289cc0c5538SChris Mason if (!extend && slot == 0) { 2290d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 22915f39d397SChris Mason btrfs_set_header_nritems(right, 0); 2292d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 2293d4dbff95SChris Mason &disk_key, 2294db94535dSChris Mason right->start, 2295098f59c2SChris Mason path->slots[1], 1); 2296d4dbff95SChris Mason if (wret) 2297d4dbff95SChris Mason ret = wret; 2298925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 22995f39d397SChris Mason free_extent_buffer(path->nodes[0]); 23005f39d397SChris Mason path->nodes[0] = right; 2301d4dbff95SChris Mason path->slots[0] = 0; 2302a429e513SChris Mason if (path->slots[1] == 0) { 2303a429e513SChris Mason wret = fixup_low_keys(trans, root, 2304a429e513SChris Mason path, &disk_key, 1); 2305a429e513SChris Mason if (wret) 2306a429e513SChris Mason ret = wret; 2307a429e513SChris Mason } 23080ef8b242SChris Mason btrfs_mark_buffer_dirty(right); 2309d4dbff95SChris Mason return ret; 2310cc0c5538SChris Mason } else if (extend && slot == 0) { 2311cc0c5538SChris Mason mid = 1; 2312cc0c5538SChris Mason } else { 2313d4dbff95SChris Mason mid = slot; 23145ee78ac7SChris Mason if (mid != nritems && 23155ee78ac7SChris Mason leaf_space_used(l, mid, nritems - mid) + 23165ee78ac7SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 2317d4dbff95SChris Mason double_split = 1; 2318d4dbff95SChris Mason } 2319d4dbff95SChris Mason } 23205ee78ac7SChris Mason } 2321cc0c5538SChris Mason } 23225f39d397SChris Mason nritems = nritems - mid; 23235f39d397SChris Mason btrfs_set_header_nritems(right, nritems); 23245f39d397SChris Mason data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l); 23255f39d397SChris Mason 23265f39d397SChris Mason copy_extent_buffer(right, l, btrfs_item_nr_offset(0), 23275f39d397SChris Mason btrfs_item_nr_offset(mid), 23285f39d397SChris Mason nritems * sizeof(struct btrfs_item)); 23295f39d397SChris Mason 23305f39d397SChris Mason copy_extent_buffer(right, l, 2331d6025579SChris Mason btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - 2332123abc88SChris Mason data_copy_size, btrfs_leaf_data(l) + 2333123abc88SChris Mason leaf_data_end(root, l), data_copy_size); 233474123bd7SChris Mason 23355f39d397SChris Mason rt_data_off = BTRFS_LEAF_DATA_SIZE(root) - 23365f39d397SChris Mason btrfs_item_end_nr(l, mid); 23375f39d397SChris Mason 23385f39d397SChris Mason for (i = 0; i < nritems; i++) { 23395f39d397SChris Mason struct btrfs_item *item = btrfs_item_nr(right, i); 2340db94535dSChris Mason u32 ioff; 2341db94535dSChris Mason 2342db94535dSChris Mason if (!right->map_token) { 2343db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2344db94535dSChris Mason sizeof(struct btrfs_item), 2345db94535dSChris Mason &right->map_token, &right->kaddr, 2346db94535dSChris Mason &right->map_start, &right->map_len, 2347db94535dSChris Mason KM_USER1); 2348db94535dSChris Mason } 2349db94535dSChris Mason 2350db94535dSChris Mason ioff = btrfs_item_offset(right, item); 23515f39d397SChris Mason btrfs_set_item_offset(right, item, ioff + rt_data_off); 23520783fcfcSChris Mason } 235374123bd7SChris Mason 2354db94535dSChris Mason if (right->map_token) { 2355db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2356db94535dSChris Mason right->map_token = NULL; 2357db94535dSChris Mason } 2358db94535dSChris Mason 23595f39d397SChris Mason btrfs_set_header_nritems(l, mid); 2360aa5d6bedSChris Mason ret = 0; 23615f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 2362db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, right->start, 2363db94535dSChris Mason path->slots[1] + 1, 1); 2364aa5d6bedSChris Mason if (wret) 2365aa5d6bedSChris Mason ret = wret; 23665f39d397SChris Mason 23675f39d397SChris Mason btrfs_mark_buffer_dirty(right); 23685f39d397SChris Mason btrfs_mark_buffer_dirty(l); 2369eb60ceacSChris Mason BUG_ON(path->slots[0] != slot); 23705f39d397SChris Mason 2371be0e5c09SChris Mason if (mid <= slot) { 2372925baeddSChris Mason btrfs_tree_unlock(path->nodes[0]); 23735f39d397SChris Mason free_extent_buffer(path->nodes[0]); 23745f39d397SChris Mason path->nodes[0] = right; 2375be0e5c09SChris Mason path->slots[0] -= mid; 2376be0e5c09SChris Mason path->slots[1] += 1; 2377925baeddSChris Mason } else { 2378925baeddSChris Mason btrfs_tree_unlock(right); 23795f39d397SChris Mason free_extent_buffer(right); 2380925baeddSChris Mason } 23815f39d397SChris Mason 2382eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 2383d4dbff95SChris Mason 2384cc0c5538SChris Mason if (double_split) { 2385cc0c5538SChris Mason BUG_ON(num_doubles != 0); 2386cc0c5538SChris Mason num_doubles++; 2387cc0c5538SChris Mason goto again; 23883326d1b0SChris Mason } 2389be0e5c09SChris Mason return ret; 2390be0e5c09SChris Mason } 2391be0e5c09SChris Mason 2392b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans, 2393b18c6685SChris Mason struct btrfs_root *root, 2394b18c6685SChris Mason struct btrfs_path *path, 2395179e29e4SChris Mason u32 new_size, int from_end) 2396b18c6685SChris Mason { 2397b18c6685SChris Mason int ret = 0; 2398b18c6685SChris Mason int slot; 2399b18c6685SChris Mason int slot_orig; 24005f39d397SChris Mason struct extent_buffer *leaf; 24015f39d397SChris Mason struct btrfs_item *item; 2402b18c6685SChris Mason u32 nritems; 2403b18c6685SChris Mason unsigned int data_end; 2404b18c6685SChris Mason unsigned int old_data_start; 2405b18c6685SChris Mason unsigned int old_size; 2406b18c6685SChris Mason unsigned int size_diff; 2407b18c6685SChris Mason int i; 2408b18c6685SChris Mason 2409b18c6685SChris Mason slot_orig = path->slots[0]; 24105f39d397SChris Mason leaf = path->nodes[0]; 2411179e29e4SChris Mason slot = path->slots[0]; 2412179e29e4SChris Mason 2413179e29e4SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 2414179e29e4SChris Mason if (old_size == new_size) 2415179e29e4SChris Mason return 0; 2416b18c6685SChris Mason 24175f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2418b18c6685SChris Mason data_end = leaf_data_end(root, leaf); 2419b18c6685SChris Mason 24205f39d397SChris Mason old_data_start = btrfs_item_offset_nr(leaf, slot); 2421179e29e4SChris Mason 2422b18c6685SChris Mason size_diff = old_size - new_size; 2423b18c6685SChris Mason 2424b18c6685SChris Mason BUG_ON(slot < 0); 2425b18c6685SChris Mason BUG_ON(slot >= nritems); 2426b18c6685SChris Mason 2427b18c6685SChris Mason /* 2428b18c6685SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2429b18c6685SChris Mason */ 2430b18c6685SChris Mason /* first correct the data pointers */ 2431b18c6685SChris Mason for (i = slot; i < nritems; i++) { 24325f39d397SChris Mason u32 ioff; 24335f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2434db94535dSChris Mason 2435db94535dSChris Mason if (!leaf->map_token) { 2436db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2437db94535dSChris Mason sizeof(struct btrfs_item), 2438db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2439db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2440db94535dSChris Mason KM_USER1); 2441db94535dSChris Mason } 2442db94535dSChris Mason 24435f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 24445f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + size_diff); 2445b18c6685SChris Mason } 2446db94535dSChris Mason 2447db94535dSChris Mason if (leaf->map_token) { 2448db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2449db94535dSChris Mason leaf->map_token = NULL; 2450db94535dSChris Mason } 2451db94535dSChris Mason 2452b18c6685SChris Mason /* shift the data */ 2453179e29e4SChris Mason if (from_end) { 24545f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2455b18c6685SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 2456b18c6685SChris Mason data_end, old_data_start + new_size - data_end); 2457179e29e4SChris Mason } else { 2458179e29e4SChris Mason struct btrfs_disk_key disk_key; 2459179e29e4SChris Mason u64 offset; 2460179e29e4SChris Mason 2461179e29e4SChris Mason btrfs_item_key(leaf, &disk_key, slot); 2462179e29e4SChris Mason 2463179e29e4SChris Mason if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) { 2464179e29e4SChris Mason unsigned long ptr; 2465179e29e4SChris Mason struct btrfs_file_extent_item *fi; 2466179e29e4SChris Mason 2467179e29e4SChris Mason fi = btrfs_item_ptr(leaf, slot, 2468179e29e4SChris Mason struct btrfs_file_extent_item); 2469179e29e4SChris Mason fi = (struct btrfs_file_extent_item *)( 2470179e29e4SChris Mason (unsigned long)fi - size_diff); 2471179e29e4SChris Mason 2472179e29e4SChris Mason if (btrfs_file_extent_type(leaf, fi) == 2473179e29e4SChris Mason BTRFS_FILE_EXTENT_INLINE) { 2474179e29e4SChris Mason ptr = btrfs_item_ptr_offset(leaf, slot); 2475179e29e4SChris Mason memmove_extent_buffer(leaf, ptr, 2476179e29e4SChris Mason (unsigned long)fi, 2477179e29e4SChris Mason offsetof(struct btrfs_file_extent_item, 2478179e29e4SChris Mason disk_bytenr)); 2479179e29e4SChris Mason } 2480179e29e4SChris Mason } 2481179e29e4SChris Mason 2482179e29e4SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2483179e29e4SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 2484179e29e4SChris Mason data_end, old_data_start - data_end); 2485179e29e4SChris Mason 2486179e29e4SChris Mason offset = btrfs_disk_key_offset(&disk_key); 2487179e29e4SChris Mason btrfs_set_disk_key_offset(&disk_key, offset + size_diff); 2488179e29e4SChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 2489179e29e4SChris Mason if (slot == 0) 2490179e29e4SChris Mason fixup_low_keys(trans, root, path, &disk_key, 1); 2491179e29e4SChris Mason } 24925f39d397SChris Mason 24935f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 24945f39d397SChris Mason btrfs_set_item_size(leaf, item, new_size); 24955f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2496b18c6685SChris Mason 2497b18c6685SChris Mason ret = 0; 24985f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 24995f39d397SChris Mason btrfs_print_leaf(root, leaf); 2500b18c6685SChris Mason BUG(); 25015f39d397SChris Mason } 2502b18c6685SChris Mason return ret; 2503b18c6685SChris Mason } 2504b18c6685SChris Mason 25055f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans, 25065f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 25075f39d397SChris Mason u32 data_size) 25086567e837SChris Mason { 25096567e837SChris Mason int ret = 0; 25106567e837SChris Mason int slot; 25116567e837SChris Mason int slot_orig; 25125f39d397SChris Mason struct extent_buffer *leaf; 25135f39d397SChris Mason struct btrfs_item *item; 25146567e837SChris Mason u32 nritems; 25156567e837SChris Mason unsigned int data_end; 25166567e837SChris Mason unsigned int old_data; 25176567e837SChris Mason unsigned int old_size; 25186567e837SChris Mason int i; 25196567e837SChris Mason 25206567e837SChris Mason slot_orig = path->slots[0]; 25215f39d397SChris Mason leaf = path->nodes[0]; 25226567e837SChris Mason 25235f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 25246567e837SChris Mason data_end = leaf_data_end(root, leaf); 25256567e837SChris Mason 25265f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < data_size) { 25275f39d397SChris Mason btrfs_print_leaf(root, leaf); 25286567e837SChris Mason BUG(); 25295f39d397SChris Mason } 25306567e837SChris Mason slot = path->slots[0]; 25315f39d397SChris Mason old_data = btrfs_item_end_nr(leaf, slot); 25326567e837SChris Mason 25336567e837SChris Mason BUG_ON(slot < 0); 25343326d1b0SChris Mason if (slot >= nritems) { 25353326d1b0SChris Mason btrfs_print_leaf(root, leaf); 25363326d1b0SChris Mason printk("slot %d too large, nritems %d\n", slot, nritems); 25373326d1b0SChris Mason BUG_ON(1); 25383326d1b0SChris Mason } 25396567e837SChris Mason 25406567e837SChris Mason /* 25416567e837SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 25426567e837SChris Mason */ 25436567e837SChris Mason /* first correct the data pointers */ 25446567e837SChris Mason for (i = slot; i < nritems; i++) { 25455f39d397SChris Mason u32 ioff; 25465f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2547db94535dSChris Mason 2548db94535dSChris Mason if (!leaf->map_token) { 2549db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2550db94535dSChris Mason sizeof(struct btrfs_item), 2551db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2552db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2553db94535dSChris Mason KM_USER1); 2554db94535dSChris Mason } 25555f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 25565f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff - data_size); 25576567e837SChris Mason } 25585f39d397SChris Mason 2559db94535dSChris Mason if (leaf->map_token) { 2560db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2561db94535dSChris Mason leaf->map_token = NULL; 2562db94535dSChris Mason } 2563db94535dSChris Mason 25646567e837SChris Mason /* shift the data */ 25655f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 25666567e837SChris Mason data_end - data_size, btrfs_leaf_data(leaf) + 25676567e837SChris Mason data_end, old_data - data_end); 25685f39d397SChris Mason 25696567e837SChris Mason data_end = old_data; 25705f39d397SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 25715f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 25725f39d397SChris Mason btrfs_set_item_size(leaf, item, old_size + data_size); 25735f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 25746567e837SChris Mason 25756567e837SChris Mason ret = 0; 25765f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 25775f39d397SChris Mason btrfs_print_leaf(root, leaf); 25786567e837SChris Mason BUG(); 25795f39d397SChris Mason } 25806567e837SChris Mason return ret; 25816567e837SChris Mason } 25826567e837SChris Mason 258374123bd7SChris Mason /* 258474123bd7SChris Mason * Given a key and some data, insert an item into the tree. 258574123bd7SChris Mason * This does all the path init required, making room in the tree if needed. 258674123bd7SChris Mason */ 25879c58309dSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans, 25885f39d397SChris Mason struct btrfs_root *root, 25895f39d397SChris Mason struct btrfs_path *path, 25909c58309dSChris Mason struct btrfs_key *cpu_key, u32 *data_size, 25919c58309dSChris Mason int nr) 2592be0e5c09SChris Mason { 25935f39d397SChris Mason struct extent_buffer *leaf; 25945f39d397SChris Mason struct btrfs_item *item; 2595aa5d6bedSChris Mason int ret = 0; 2596be0e5c09SChris Mason int slot; 2597eb60ceacSChris Mason int slot_orig; 25989c58309dSChris Mason int i; 25997518a238SChris Mason u32 nritems; 26009c58309dSChris Mason u32 total_size = 0; 26019c58309dSChris Mason u32 total_data = 0; 2602be0e5c09SChris Mason unsigned int data_end; 2603e2fa7227SChris Mason struct btrfs_disk_key disk_key; 2604e2fa7227SChris Mason 26059c58309dSChris Mason for (i = 0; i < nr; i++) { 26069c58309dSChris Mason total_data += data_size[i]; 26079c58309dSChris Mason } 2608be0e5c09SChris Mason 26099c58309dSChris Mason total_size = total_data + (nr - 1) * sizeof(struct btrfs_item); 26109c58309dSChris Mason ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1); 2611eb60ceacSChris Mason if (ret == 0) { 2612f0930a37SChris Mason return -EEXIST; 2613aa5d6bedSChris Mason } 2614ed2ff2cbSChris Mason if (ret < 0) 2615ed2ff2cbSChris Mason goto out; 2616be0e5c09SChris Mason 261762e2749eSChris Mason slot_orig = path->slots[0]; 26185f39d397SChris Mason leaf = path->nodes[0]; 261974123bd7SChris Mason 26205f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2621123abc88SChris Mason data_end = leaf_data_end(root, leaf); 2622eb60ceacSChris Mason 2623123abc88SChris Mason if (btrfs_leaf_free_space(root, leaf) < 26249c58309dSChris Mason sizeof(struct btrfs_item) + total_size) { 26253326d1b0SChris Mason btrfs_print_leaf(root, leaf); 26263326d1b0SChris Mason printk("not enough freespace need %u have %d\n", 26279c58309dSChris Mason total_size, btrfs_leaf_free_space(root, leaf)); 2628be0e5c09SChris Mason BUG(); 2629d4dbff95SChris Mason } 26305f39d397SChris Mason 263162e2749eSChris Mason slot = path->slots[0]; 2632eb60ceacSChris Mason BUG_ON(slot < 0); 26335f39d397SChris Mason 2634be0e5c09SChris Mason if (slot != nritems) { 2635be0e5c09SChris Mason int i; 26365f39d397SChris Mason unsigned int old_data = btrfs_item_end_nr(leaf, slot); 2637be0e5c09SChris Mason 26385f39d397SChris Mason if (old_data < data_end) { 26395f39d397SChris Mason btrfs_print_leaf(root, leaf); 26405f39d397SChris Mason printk("slot %d old_data %d data_end %d\n", 26415f39d397SChris Mason slot, old_data, data_end); 26425f39d397SChris Mason BUG_ON(1); 26435f39d397SChris Mason } 2644be0e5c09SChris Mason /* 2645be0e5c09SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2646be0e5c09SChris Mason */ 2647be0e5c09SChris Mason /* first correct the data pointers */ 2648db94535dSChris Mason WARN_ON(leaf->map_token); 26490783fcfcSChris Mason for (i = slot; i < nritems; i++) { 26505f39d397SChris Mason u32 ioff; 2651db94535dSChris Mason 26525f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2653db94535dSChris Mason if (!leaf->map_token) { 2654db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2655db94535dSChris Mason sizeof(struct btrfs_item), 2656db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2657db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2658db94535dSChris Mason KM_USER1); 2659db94535dSChris Mason } 2660db94535dSChris Mason 26615f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 26629c58309dSChris Mason btrfs_set_item_offset(leaf, item, ioff - total_data); 26630783fcfcSChris Mason } 2664db94535dSChris Mason if (leaf->map_token) { 2665db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2666db94535dSChris Mason leaf->map_token = NULL; 2667db94535dSChris Mason } 2668be0e5c09SChris Mason 2669be0e5c09SChris Mason /* shift the items */ 26709c58309dSChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr), 26715f39d397SChris Mason btrfs_item_nr_offset(slot), 26720783fcfcSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 2673be0e5c09SChris Mason 2674be0e5c09SChris Mason /* shift the data */ 26755f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 26769c58309dSChris Mason data_end - total_data, btrfs_leaf_data(leaf) + 2677be0e5c09SChris Mason data_end, old_data - data_end); 2678be0e5c09SChris Mason data_end = old_data; 2679be0e5c09SChris Mason } 26805f39d397SChris Mason 268162e2749eSChris Mason /* setup the item for the new data */ 26829c58309dSChris Mason for (i = 0; i < nr; i++) { 26839c58309dSChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key + i); 26849c58309dSChris Mason btrfs_set_item_key(leaf, &disk_key, slot + i); 26859c58309dSChris Mason item = btrfs_item_nr(leaf, slot + i); 26869c58309dSChris Mason btrfs_set_item_offset(leaf, item, data_end - data_size[i]); 26879c58309dSChris Mason data_end -= data_size[i]; 26889c58309dSChris Mason btrfs_set_item_size(leaf, item, data_size[i]); 26899c58309dSChris Mason } 26909c58309dSChris Mason btrfs_set_header_nritems(leaf, nritems + nr); 26915f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2692aa5d6bedSChris Mason 2693aa5d6bedSChris Mason ret = 0; 26945a01a2e3SChris Mason if (slot == 0) { 26955a01a2e3SChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key); 2696e089f05cSChris Mason ret = fixup_low_keys(trans, root, path, &disk_key, 1); 26975a01a2e3SChris Mason } 2698aa5d6bedSChris Mason 26995f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 27005f39d397SChris Mason btrfs_print_leaf(root, leaf); 2701be0e5c09SChris Mason BUG(); 27025f39d397SChris Mason } 2703ed2ff2cbSChris Mason out: 270462e2749eSChris Mason return ret; 270562e2749eSChris Mason } 270662e2749eSChris Mason 270762e2749eSChris Mason /* 270862e2749eSChris Mason * Given a key and some data, insert an item into the tree. 270962e2749eSChris Mason * This does all the path init required, making room in the tree if needed. 271062e2749eSChris Mason */ 2711e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root 2712e089f05cSChris Mason *root, struct btrfs_key *cpu_key, void *data, u32 2713e089f05cSChris Mason data_size) 271462e2749eSChris Mason { 271562e2749eSChris Mason int ret = 0; 27162c90e5d6SChris Mason struct btrfs_path *path; 27175f39d397SChris Mason struct extent_buffer *leaf; 27185f39d397SChris Mason unsigned long ptr; 271962e2749eSChris Mason 27202c90e5d6SChris Mason path = btrfs_alloc_path(); 27212c90e5d6SChris Mason BUG_ON(!path); 27222c90e5d6SChris Mason ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size); 272362e2749eSChris Mason if (!ret) { 27245f39d397SChris Mason leaf = path->nodes[0]; 27255f39d397SChris Mason ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); 27265f39d397SChris Mason write_extent_buffer(leaf, data, ptr, data_size); 27275f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 272862e2749eSChris Mason } 27292c90e5d6SChris Mason btrfs_free_path(path); 2730aa5d6bedSChris Mason return ret; 2731be0e5c09SChris Mason } 2732be0e5c09SChris Mason 273374123bd7SChris Mason /* 27345de08d7dSChris Mason * delete the pointer from a given node. 273574123bd7SChris Mason * 273674123bd7SChris Mason * If the delete empties a node, the node is removed from the tree, 273774123bd7SChris Mason * continuing all the way the root if required. The root is converted into 273874123bd7SChris Mason * a leaf if all the nodes are emptied. 273974123bd7SChris Mason */ 2740e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 2741e089f05cSChris Mason struct btrfs_path *path, int level, int slot) 2742be0e5c09SChris Mason { 27435f39d397SChris Mason struct extent_buffer *parent = path->nodes[level]; 27447518a238SChris Mason u32 nritems; 2745aa5d6bedSChris Mason int ret = 0; 2746bb803951SChris Mason int wret; 2747be0e5c09SChris Mason 27485f39d397SChris Mason nritems = btrfs_header_nritems(parent); 2749be0e5c09SChris Mason if (slot != nritems -1) { 27505f39d397SChris Mason memmove_extent_buffer(parent, 27515f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 27525f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 2753d6025579SChris Mason sizeof(struct btrfs_key_ptr) * 2754d6025579SChris Mason (nritems - slot - 1)); 2755be0e5c09SChris Mason } 27567518a238SChris Mason nritems--; 27575f39d397SChris Mason btrfs_set_header_nritems(parent, nritems); 27587518a238SChris Mason if (nritems == 0 && parent == root->node) { 27595f39d397SChris Mason BUG_ON(btrfs_header_level(root->node) != 1); 2760eb60ceacSChris Mason /* just turn the root into a leaf and break */ 27615f39d397SChris Mason btrfs_set_header_level(root->node, 0); 2762bb803951SChris Mason } else if (slot == 0) { 27635f39d397SChris Mason struct btrfs_disk_key disk_key; 27645f39d397SChris Mason 27655f39d397SChris Mason btrfs_node_key(parent, &disk_key, 0); 27665f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, level + 1); 27670f70abe2SChris Mason if (wret) 27680f70abe2SChris Mason ret = wret; 2769be0e5c09SChris Mason } 2770d6025579SChris Mason btrfs_mark_buffer_dirty(parent); 2771aa5d6bedSChris Mason return ret; 2772be0e5c09SChris Mason } 2773be0e5c09SChris Mason 277474123bd7SChris Mason /* 277574123bd7SChris Mason * delete the item at the leaf level in path. If that empties 277674123bd7SChris Mason * the leaf, remove it from the tree 277774123bd7SChris Mason */ 277885e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root, 277985e21bacSChris Mason struct btrfs_path *path, int slot, int nr) 2780be0e5c09SChris Mason { 27815f39d397SChris Mason struct extent_buffer *leaf; 27825f39d397SChris Mason struct btrfs_item *item; 278385e21bacSChris Mason int last_off; 278485e21bacSChris Mason int dsize = 0; 2785aa5d6bedSChris Mason int ret = 0; 2786aa5d6bedSChris Mason int wret; 278785e21bacSChris Mason int i; 27887518a238SChris Mason u32 nritems; 2789be0e5c09SChris Mason 27905f39d397SChris Mason leaf = path->nodes[0]; 279185e21bacSChris Mason last_off = btrfs_item_offset_nr(leaf, slot + nr - 1); 279285e21bacSChris Mason 279385e21bacSChris Mason for (i = 0; i < nr; i++) 279485e21bacSChris Mason dsize += btrfs_item_size_nr(leaf, slot + i); 279585e21bacSChris Mason 27965f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2797be0e5c09SChris Mason 279885e21bacSChris Mason if (slot + nr != nritems) { 2799be0e5c09SChris Mason int i; 2800123abc88SChris Mason int data_end = leaf_data_end(root, leaf); 28015f39d397SChris Mason 28025f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2803d6025579SChris Mason data_end + dsize, 2804123abc88SChris Mason btrfs_leaf_data(leaf) + data_end, 280585e21bacSChris Mason last_off - data_end); 28065f39d397SChris Mason 280785e21bacSChris Mason for (i = slot + nr; i < nritems; i++) { 28085f39d397SChris Mason u32 ioff; 2809db94535dSChris Mason 28105f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2811db94535dSChris Mason if (!leaf->map_token) { 2812db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2813db94535dSChris Mason sizeof(struct btrfs_item), 2814db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2815db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2816db94535dSChris Mason KM_USER1); 2817db94535dSChris Mason } 28185f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 28195f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + dsize); 28200783fcfcSChris Mason } 2821db94535dSChris Mason 2822db94535dSChris Mason if (leaf->map_token) { 2823db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2824db94535dSChris Mason leaf->map_token = NULL; 2825db94535dSChris Mason } 2826db94535dSChris Mason 28275f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot), 282885e21bacSChris Mason btrfs_item_nr_offset(slot + nr), 28290783fcfcSChris Mason sizeof(struct btrfs_item) * 283085e21bacSChris Mason (nritems - slot - nr)); 2831be0e5c09SChris Mason } 283285e21bacSChris Mason btrfs_set_header_nritems(leaf, nritems - nr); 283385e21bacSChris Mason nritems -= nr; 28345f39d397SChris Mason 283574123bd7SChris Mason /* delete the leaf if we've emptied it */ 28367518a238SChris Mason if (nritems == 0) { 28375f39d397SChris Mason if (leaf == root->node) { 28385f39d397SChris Mason btrfs_set_header_level(leaf, 0); 28399a8dd150SChris Mason } else { 28407bb86316SChris Mason u64 root_gen = btrfs_header_generation(path->nodes[1]); 2841e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, path->slots[1]); 2842aa5d6bedSChris Mason if (wret) 2843aa5d6bedSChris Mason ret = wret; 2844e089f05cSChris Mason wret = btrfs_free_extent(trans, root, 28457bb86316SChris Mason leaf->start, leaf->len, 28467bb86316SChris Mason btrfs_header_owner(path->nodes[1]), 28477bb86316SChris Mason root_gen, 0, 0, 1); 28480f70abe2SChris Mason if (wret) 28490f70abe2SChris Mason ret = wret; 28509a8dd150SChris Mason } 2851be0e5c09SChris Mason } else { 28527518a238SChris Mason int used = leaf_space_used(leaf, 0, nritems); 2853aa5d6bedSChris Mason if (slot == 0) { 28545f39d397SChris Mason struct btrfs_disk_key disk_key; 28555f39d397SChris Mason 28565f39d397SChris Mason btrfs_item_key(leaf, &disk_key, 0); 2857e089f05cSChris Mason wret = fixup_low_keys(trans, root, path, 28585f39d397SChris Mason &disk_key, 1); 2859aa5d6bedSChris Mason if (wret) 2860aa5d6bedSChris Mason ret = wret; 2861aa5d6bedSChris Mason } 2862aa5d6bedSChris Mason 286374123bd7SChris Mason /* delete the leaf if it is mostly empty */ 286485e21bacSChris Mason if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) { 2865be0e5c09SChris Mason /* push_leaf_left fixes the path. 2866be0e5c09SChris Mason * make sure the path still points to our leaf 2867be0e5c09SChris Mason * for possible call to del_ptr below 2868be0e5c09SChris Mason */ 28694920c9acSChris Mason slot = path->slots[1]; 28705f39d397SChris Mason extent_buffer_get(leaf); 28715f39d397SChris Mason 287285e21bacSChris Mason wret = push_leaf_left(trans, root, path, 1, 1); 287354aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2874aa5d6bedSChris Mason ret = wret; 28755f39d397SChris Mason 28765f39d397SChris Mason if (path->nodes[0] == leaf && 28775f39d397SChris Mason btrfs_header_nritems(leaf)) { 287885e21bacSChris Mason wret = push_leaf_right(trans, root, path, 1, 1); 287954aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2880aa5d6bedSChris Mason ret = wret; 2881aa5d6bedSChris Mason } 28825f39d397SChris Mason 28835f39d397SChris Mason if (btrfs_header_nritems(leaf) == 0) { 28847bb86316SChris Mason u64 root_gen; 2885db94535dSChris Mason u64 bytenr = leaf->start; 2886db94535dSChris Mason u32 blocksize = leaf->len; 28875f39d397SChris Mason 28887bb86316SChris Mason root_gen = btrfs_header_generation( 28897bb86316SChris Mason path->nodes[1]); 28907bb86316SChris Mason 2891e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, slot); 2892aa5d6bedSChris Mason if (wret) 2893aa5d6bedSChris Mason ret = wret; 28945f39d397SChris Mason 28955f39d397SChris Mason free_extent_buffer(leaf); 2896db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 28977bb86316SChris Mason blocksize, 28987bb86316SChris Mason btrfs_header_owner(path->nodes[1]), 28997bb86316SChris Mason root_gen, 0, 0, 1); 29000f70abe2SChris Mason if (wret) 29010f70abe2SChris Mason ret = wret; 29025de08d7dSChris Mason } else { 2903925baeddSChris Mason /* if we're still in the path, make sure 2904925baeddSChris Mason * we're dirty. Otherwise, one of the 2905925baeddSChris Mason * push_leaf functions must have already 2906925baeddSChris Mason * dirtied this buffer 2907925baeddSChris Mason */ 2908925baeddSChris Mason if (path->nodes[0] == leaf) 29095f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 29105f39d397SChris Mason free_extent_buffer(leaf); 2911be0e5c09SChris Mason } 2912d5719762SChris Mason } else { 29135f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2914be0e5c09SChris Mason } 29159a8dd150SChris Mason } 2916aa5d6bedSChris Mason return ret; 29179a8dd150SChris Mason } 29189a8dd150SChris Mason 291997571fd0SChris Mason /* 2920925baeddSChris Mason * search the tree again to find a leaf with lesser keys 29217bb86316SChris Mason * returns 0 if it found something or 1 if there are no lesser leaves. 29227bb86316SChris Mason * returns < 0 on io errors. 29237bb86316SChris Mason */ 29247bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path) 29257bb86316SChris Mason { 2926925baeddSChris Mason struct btrfs_key key; 2927925baeddSChris Mason struct btrfs_disk_key found_key; 2928925baeddSChris Mason int ret; 29297bb86316SChris Mason 2930925baeddSChris Mason btrfs_item_key_to_cpu(path->nodes[0], &key, 0); 2931925baeddSChris Mason 2932925baeddSChris Mason if (key.offset > 0) 2933925baeddSChris Mason key.offset--; 2934925baeddSChris Mason else if (key.type > 0) 2935925baeddSChris Mason key.type--; 2936925baeddSChris Mason else if (key.objectid > 0) 2937925baeddSChris Mason key.objectid--; 2938925baeddSChris Mason else 29397bb86316SChris Mason return 1; 29407bb86316SChris Mason 2941925baeddSChris Mason btrfs_release_path(root, path); 2942925baeddSChris Mason ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); 2943925baeddSChris Mason if (ret < 0) 2944925baeddSChris Mason return ret; 2945925baeddSChris Mason btrfs_item_key(path->nodes[0], &found_key, 0); 2946925baeddSChris Mason ret = comp_keys(&found_key, &key); 2947925baeddSChris Mason if (ret < 0) 29487bb86316SChris Mason return 0; 2949925baeddSChris Mason return 1; 29507bb86316SChris Mason } 29517bb86316SChris Mason 29527bb86316SChris Mason /* 2953925baeddSChris Mason * search the tree again to find a leaf with greater keys 29540f70abe2SChris Mason * returns 0 if it found something or 1 if there are no greater leaves. 29550f70abe2SChris Mason * returns < 0 on io errors. 295697571fd0SChris Mason */ 2957234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path) 2958d97e63b6SChris Mason { 2959d97e63b6SChris Mason int slot; 2960d97e63b6SChris Mason int level = 1; 29615f39d397SChris Mason struct extent_buffer *c; 29625f39d397SChris Mason struct extent_buffer *next = NULL; 2963925baeddSChris Mason struct btrfs_key key; 2964925baeddSChris Mason u32 nritems; 2965925baeddSChris Mason int ret; 2966925baeddSChris Mason 2967925baeddSChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 2968925baeddSChris Mason if (nritems == 0) { 2969925baeddSChris Mason return 1; 2970925baeddSChris Mason } 2971925baeddSChris Mason 2972925baeddSChris Mason btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1); 2973925baeddSChris Mason 2974925baeddSChris Mason btrfs_release_path(root, path); 2975*a2135011SChris Mason path->keep_locks = 1; 2976925baeddSChris Mason ret = btrfs_search_slot(NULL, root, &key, path, 0, 0); 2977925baeddSChris Mason path->keep_locks = 0; 2978925baeddSChris Mason 2979925baeddSChris Mason if (ret < 0) 2980925baeddSChris Mason return ret; 2981925baeddSChris Mason 2982*a2135011SChris Mason nritems = btrfs_header_nritems(path->nodes[0]); 2983*a2135011SChris Mason if (nritems > 0 && path->slots[0] < nritems - 1) { 2984925baeddSChris Mason goto done; 2985925baeddSChris Mason } 2986d97e63b6SChris Mason 2987234b63a0SChris Mason while(level < BTRFS_MAX_LEVEL) { 2988d97e63b6SChris Mason if (!path->nodes[level]) 29890f70abe2SChris Mason return 1; 29905f39d397SChris Mason 2991d97e63b6SChris Mason slot = path->slots[level] + 1; 2992d97e63b6SChris Mason c = path->nodes[level]; 29935f39d397SChris Mason if (slot >= btrfs_header_nritems(c)) { 2994d97e63b6SChris Mason level++; 2995925baeddSChris Mason if (level == BTRFS_MAX_LEVEL) { 29967bb86316SChris Mason return 1; 2997925baeddSChris Mason } 2998d97e63b6SChris Mason continue; 2999d97e63b6SChris Mason } 30005f39d397SChris Mason 3001925baeddSChris Mason if (next) { 3002925baeddSChris Mason btrfs_tree_unlock(next); 30035f39d397SChris Mason free_extent_buffer(next); 3004925baeddSChris Mason } 30055f39d397SChris Mason 3006925baeddSChris Mason if (level == 1 && path->locks[1] && path->reada) 300701f46658SChris Mason reada_for_search(root, path, level, slot, 0); 30085f39d397SChris Mason 3009ca7a79adSChris Mason next = read_node_slot(root, c, slot); 3010*a2135011SChris Mason if (!path->skip_locking) { 3011*a2135011SChris Mason if (!btrfs_tree_locked(c)) { 3012*a2135011SChris Mason int i; 3013*a2135011SChris Mason WARN_ON(1); 3014*a2135011SChris Mason printk("path %p no lock on level %d\n", path, level); 3015*a2135011SChris Mason for (i = 0; i < BTRFS_MAX_LEVEL; i++) { 3016*a2135011SChris Mason printk("path %p level %d slot %d nritems %d\n", path, i, path->slots[i], btrfs_header_nritems(path->nodes[i])); 3017*a2135011SChris Mason } 3018*a2135011SChris Mason } 3019925baeddSChris Mason btrfs_tree_lock(next); 3020*a2135011SChris Mason } 3021d97e63b6SChris Mason break; 3022d97e63b6SChris Mason } 3023d97e63b6SChris Mason path->slots[level] = slot; 3024d97e63b6SChris Mason while(1) { 3025d97e63b6SChris Mason level--; 3026d97e63b6SChris Mason c = path->nodes[level]; 3027925baeddSChris Mason if (path->locks[level]) 3028925baeddSChris Mason btrfs_tree_unlock(c); 30295f39d397SChris Mason free_extent_buffer(c); 3030d97e63b6SChris Mason path->nodes[level] = next; 3031d97e63b6SChris Mason path->slots[level] = 0; 3032925baeddSChris Mason path->locks[level] = 1; 3033d97e63b6SChris Mason if (!level) 3034d97e63b6SChris Mason break; 3035925baeddSChris Mason if (level == 1 && path->locks[1] && path->reada) 3036925baeddSChris Mason reada_for_search(root, path, level, slot, 0); 3037ca7a79adSChris Mason next = read_node_slot(root, next, 0); 3038*a2135011SChris Mason if (!path->skip_locking) { 3039*a2135011SChris Mason WARN_ON(!btrfs_tree_locked(path->nodes[level])); 3040925baeddSChris Mason btrfs_tree_lock(next); 3041d97e63b6SChris Mason } 3042*a2135011SChris Mason } 3043925baeddSChris Mason done: 3044925baeddSChris Mason unlock_up(path, 0, 1); 3045d97e63b6SChris Mason return 0; 3046d97e63b6SChris Mason } 30470b86a832SChris Mason 30480b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root, 30490b86a832SChris Mason struct btrfs_path *path, u64 min_objectid, 30500b86a832SChris Mason int type) 30510b86a832SChris Mason { 30520b86a832SChris Mason struct btrfs_key found_key; 30530b86a832SChris Mason struct extent_buffer *leaf; 30540b86a832SChris Mason int ret; 30550b86a832SChris Mason 30560b86a832SChris Mason while(1) { 30570b86a832SChris Mason if (path->slots[0] == 0) { 30580b86a832SChris Mason ret = btrfs_prev_leaf(root, path); 30590b86a832SChris Mason if (ret != 0) 30600b86a832SChris Mason return ret; 30610b86a832SChris Mason } else { 30620b86a832SChris Mason path->slots[0]--; 30630b86a832SChris Mason } 30640b86a832SChris Mason leaf = path->nodes[0]; 30650b86a832SChris Mason btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); 30660b86a832SChris Mason if (found_key.type == type) 30670b86a832SChris Mason return 0; 30680b86a832SChris Mason } 30690b86a832SChris Mason return 1; 30700b86a832SChris Mason } 30710b86a832SChris Mason 3072