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" 249a8dd150SChris Mason 25e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 26e089f05cSChris Mason *root, struct btrfs_path *path, int level); 27e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 28d4dbff95SChris Mason *root, struct btrfs_key *ins_key, 29cc0c5538SChris Mason struct btrfs_path *path, int data_size, int extend); 305f39d397SChris Mason static int push_node_left(struct btrfs_trans_handle *trans, 315f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *dst, 32971a1f66SChris Mason struct extent_buffer *src, int empty); 335f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 345f39d397SChris Mason struct btrfs_root *root, 355f39d397SChris Mason struct extent_buffer *dst_buf, 365f39d397SChris Mason struct extent_buffer *src_buf); 37e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 38e089f05cSChris Mason struct btrfs_path *path, int level, int slot); 39d97e63b6SChris Mason 40df24a2b9SChris Mason inline void btrfs_init_path(struct btrfs_path *p) 41df24a2b9SChris Mason { 42df24a2b9SChris Mason memset(p, 0, sizeof(*p)); 43df24a2b9SChris Mason } 44df24a2b9SChris Mason 452c90e5d6SChris Mason struct btrfs_path *btrfs_alloc_path(void) 462c90e5d6SChris Mason { 47df24a2b9SChris Mason struct btrfs_path *path; 48df24a2b9SChris Mason path = kmem_cache_alloc(btrfs_path_cachep, GFP_NOFS); 492cc58cf2SChris Mason if (path) { 50df24a2b9SChris Mason btrfs_init_path(path); 512cc58cf2SChris Mason path->reada = 1; 522cc58cf2SChris Mason } 53df24a2b9SChris Mason return path; 542c90e5d6SChris Mason } 552c90e5d6SChris Mason 562c90e5d6SChris Mason void btrfs_free_path(struct btrfs_path *p) 572c90e5d6SChris Mason { 58df24a2b9SChris Mason btrfs_release_path(NULL, p); 592c90e5d6SChris Mason kmem_cache_free(btrfs_path_cachep, p); 602c90e5d6SChris Mason } 612c90e5d6SChris Mason 62234b63a0SChris Mason void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p) 63eb60ceacSChris Mason { 64eb60ceacSChris Mason int i; 65234b63a0SChris Mason for (i = 0; i < BTRFS_MAX_LEVEL; i++) { 66eb60ceacSChris Mason if (!p->nodes[i]) 67eb60ceacSChris Mason break; 685f39d397SChris Mason free_extent_buffer(p->nodes[i]); 69eb60ceacSChris Mason } 70aa5d6bedSChris Mason memset(p, 0, sizeof(*p)); 71eb60ceacSChris Mason } 72eb60ceacSChris Mason 730b86a832SChris Mason static void add_root_to_dirty_list(struct btrfs_root *root) 740b86a832SChris Mason { 750b86a832SChris Mason if (root->track_dirty && list_empty(&root->dirty_list)) { 760b86a832SChris Mason list_add(&root->dirty_list, 770b86a832SChris Mason &root->fs_info->dirty_cowonly_roots); 780b86a832SChris Mason } 790b86a832SChris Mason } 800b86a832SChris Mason 81be20aa9dSChris Mason int btrfs_copy_root(struct btrfs_trans_handle *trans, 82be20aa9dSChris Mason struct btrfs_root *root, 83be20aa9dSChris Mason struct extent_buffer *buf, 84be20aa9dSChris Mason struct extent_buffer **cow_ret, u64 new_root_objectid) 85be20aa9dSChris Mason { 86be20aa9dSChris Mason struct extent_buffer *cow; 87be20aa9dSChris Mason u32 nritems; 88be20aa9dSChris Mason int ret = 0; 89be20aa9dSChris Mason int level; 90be20aa9dSChris Mason struct btrfs_key first_key; 914aec2b52SChris Mason struct btrfs_root *new_root; 92be20aa9dSChris Mason 934aec2b52SChris Mason new_root = kmalloc(sizeof(*new_root), GFP_NOFS); 944aec2b52SChris Mason if (!new_root) 954aec2b52SChris Mason return -ENOMEM; 964aec2b52SChris Mason 974aec2b52SChris Mason memcpy(new_root, root, sizeof(*new_root)); 984aec2b52SChris Mason new_root->root_key.objectid = new_root_objectid; 99be20aa9dSChris Mason 100be20aa9dSChris Mason WARN_ON(root->ref_cows && trans->transid != 101be20aa9dSChris Mason root->fs_info->running_transaction->transid); 102be20aa9dSChris Mason WARN_ON(root->ref_cows && trans->transid != root->last_trans); 103be20aa9dSChris Mason 104be20aa9dSChris Mason level = btrfs_header_level(buf); 105be20aa9dSChris Mason nritems = btrfs_header_nritems(buf); 106be20aa9dSChris Mason if (nritems) { 107be20aa9dSChris Mason if (level == 0) 108be20aa9dSChris Mason btrfs_item_key_to_cpu(buf, &first_key, 0); 109be20aa9dSChris Mason else 110be20aa9dSChris Mason btrfs_node_key_to_cpu(buf, &first_key, 0); 111be20aa9dSChris Mason } else { 112be20aa9dSChris Mason first_key.objectid = 0; 113be20aa9dSChris Mason } 1144aec2b52SChris Mason cow = __btrfs_alloc_free_block(trans, new_root, buf->len, 115be20aa9dSChris Mason new_root_objectid, 116be20aa9dSChris Mason trans->transid, first_key.objectid, 117be20aa9dSChris Mason level, buf->start, 0); 1184aec2b52SChris Mason if (IS_ERR(cow)) { 1194aec2b52SChris Mason kfree(new_root); 120be20aa9dSChris Mason return PTR_ERR(cow); 1214aec2b52SChris Mason } 122be20aa9dSChris Mason 123be20aa9dSChris Mason copy_extent_buffer(cow, buf, 0, 0, cow->len); 124be20aa9dSChris Mason btrfs_set_header_bytenr(cow, cow->start); 125be20aa9dSChris Mason btrfs_set_header_generation(cow, trans->transid); 126be20aa9dSChris Mason btrfs_set_header_owner(cow, new_root_objectid); 12763b10fc4SChris Mason btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN); 128be20aa9dSChris Mason 129be20aa9dSChris Mason WARN_ON(btrfs_header_generation(buf) > trans->transid); 1304aec2b52SChris Mason ret = btrfs_inc_ref(trans, new_root, buf); 1314aec2b52SChris Mason kfree(new_root); 1324aec2b52SChris Mason 133be20aa9dSChris Mason if (ret) 134be20aa9dSChris Mason return ret; 135be20aa9dSChris Mason 136be20aa9dSChris Mason btrfs_mark_buffer_dirty(cow); 137be20aa9dSChris Mason *cow_ret = cow; 138be20aa9dSChris Mason return 0; 139be20aa9dSChris Mason } 140be20aa9dSChris Mason 141be20aa9dSChris Mason int __btrfs_cow_block(struct btrfs_trans_handle *trans, 1425f39d397SChris Mason struct btrfs_root *root, 1435f39d397SChris Mason struct extent_buffer *buf, 1445f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 1455f39d397SChris Mason struct extent_buffer **cow_ret, 1465f39d397SChris Mason u64 search_start, u64 empty_size) 1476702ed49SChris Mason { 1487bb86316SChris Mason u64 root_gen; 1495f39d397SChris Mason struct extent_buffer *cow; 1507bb86316SChris Mason u32 nritems; 1516702ed49SChris Mason int ret = 0; 1526702ed49SChris Mason int different_trans = 0; 1537bb86316SChris Mason int level; 1547bb86316SChris Mason struct btrfs_key first_key; 1556702ed49SChris Mason 1567bb86316SChris Mason if (root->ref_cows) { 1577bb86316SChris Mason root_gen = trans->transid; 1587bb86316SChris Mason } else { 1597bb86316SChris Mason root_gen = 0; 1607bb86316SChris Mason } 1617bb86316SChris Mason WARN_ON(root->ref_cows && trans->transid != 1627bb86316SChris Mason root->fs_info->running_transaction->transid); 1636702ed49SChris Mason WARN_ON(root->ref_cows && trans->transid != root->last_trans); 1645f39d397SChris Mason 1657bb86316SChris Mason level = btrfs_header_level(buf); 1667bb86316SChris Mason nritems = btrfs_header_nritems(buf); 1677bb86316SChris Mason if (nritems) { 1687bb86316SChris Mason if (level == 0) 1697bb86316SChris Mason btrfs_item_key_to_cpu(buf, &first_key, 0); 1707bb86316SChris Mason else 1717bb86316SChris Mason btrfs_node_key_to_cpu(buf, &first_key, 0); 1727bb86316SChris Mason } else { 1737bb86316SChris Mason first_key.objectid = 0; 1747bb86316SChris Mason } 1757bb86316SChris Mason cow = __btrfs_alloc_free_block(trans, root, buf->len, 1767bb86316SChris Mason root->root_key.objectid, 1777bb86316SChris Mason root_gen, first_key.objectid, level, 178db94535dSChris Mason search_start, empty_size); 1796702ed49SChris Mason if (IS_ERR(cow)) 1806702ed49SChris Mason return PTR_ERR(cow); 1816702ed49SChris Mason 1825f39d397SChris Mason copy_extent_buffer(cow, buf, 0, 0, cow->len); 183db94535dSChris Mason btrfs_set_header_bytenr(cow, cow->start); 1845f39d397SChris Mason btrfs_set_header_generation(cow, trans->transid); 1855f39d397SChris Mason btrfs_set_header_owner(cow, root->root_key.objectid); 18663b10fc4SChris Mason btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN); 1876702ed49SChris Mason 1885f39d397SChris Mason WARN_ON(btrfs_header_generation(buf) > trans->transid); 1895f39d397SChris Mason if (btrfs_header_generation(buf) != trans->transid) { 1906702ed49SChris Mason different_trans = 1; 1916702ed49SChris Mason ret = btrfs_inc_ref(trans, root, buf); 1926702ed49SChris Mason if (ret) 1936702ed49SChris Mason return ret; 1946702ed49SChris Mason } else { 1956702ed49SChris Mason clean_tree_block(trans, root, buf); 1966702ed49SChris Mason } 1976702ed49SChris Mason 1986702ed49SChris Mason if (buf == root->node) { 1997bb86316SChris Mason root_gen = btrfs_header_generation(buf); 2006702ed49SChris Mason root->node = cow; 2015f39d397SChris Mason extent_buffer_get(cow); 2026702ed49SChris Mason if (buf != root->commit_root) { 203db94535dSChris Mason btrfs_free_extent(trans, root, buf->start, 2047bb86316SChris Mason buf->len, root->root_key.objectid, 2057bb86316SChris Mason root_gen, 0, 0, 1); 2066702ed49SChris Mason } 2075f39d397SChris Mason free_extent_buffer(buf); 2080b86a832SChris Mason add_root_to_dirty_list(root); 2096702ed49SChris Mason } else { 2107bb86316SChris Mason root_gen = btrfs_header_generation(parent); 2115f39d397SChris Mason btrfs_set_node_blockptr(parent, parent_slot, 212db94535dSChris Mason cow->start); 21374493f7aSChris Mason WARN_ON(trans->transid == 0); 21474493f7aSChris Mason btrfs_set_node_ptr_generation(parent, parent_slot, 21574493f7aSChris Mason trans->transid); 2166702ed49SChris Mason btrfs_mark_buffer_dirty(parent); 2175f39d397SChris Mason WARN_ON(btrfs_header_generation(parent) != trans->transid); 2187bb86316SChris Mason btrfs_free_extent(trans, root, buf->start, buf->len, 2197bb86316SChris Mason btrfs_header_owner(parent), root_gen, 2207bb86316SChris Mason 0, 0, 1); 2216702ed49SChris Mason } 2225f39d397SChris Mason free_extent_buffer(buf); 2236702ed49SChris Mason btrfs_mark_buffer_dirty(cow); 2246702ed49SChris Mason *cow_ret = cow; 2256702ed49SChris Mason return 0; 2266702ed49SChris Mason } 2276702ed49SChris Mason 2285f39d397SChris Mason int btrfs_cow_block(struct btrfs_trans_handle *trans, 2295f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *buf, 2305f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 2315f39d397SChris Mason struct extent_buffer **cow_ret) 23202217ed2SChris Mason { 2336702ed49SChris Mason u64 search_start; 234dc17ff8fSChris Mason u64 header_trans; 235f510cfecSChris Mason int ret; 236dc17ff8fSChris Mason 237ccd467d6SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 238ccd467d6SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 239ccd467d6SChris Mason root->fs_info->running_transaction->transid); 240ccd467d6SChris Mason WARN_ON(1); 241ccd467d6SChris Mason } 242ccd467d6SChris Mason if (trans->transid != root->fs_info->generation) { 243ccd467d6SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 244ccd467d6SChris Mason root->fs_info->generation); 245ccd467d6SChris Mason WARN_ON(1); 246ccd467d6SChris Mason } 247dc17ff8fSChris Mason 248dc17ff8fSChris Mason header_trans = btrfs_header_generation(buf); 24963b10fc4SChris Mason spin_lock(&root->fs_info->hash_lock); 25063b10fc4SChris Mason if (header_trans == trans->transid && 25163b10fc4SChris Mason !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN)) { 25202217ed2SChris Mason *cow_ret = buf; 25363b10fc4SChris Mason spin_unlock(&root->fs_info->hash_lock); 25402217ed2SChris Mason return 0; 25502217ed2SChris Mason } 25663b10fc4SChris Mason spin_unlock(&root->fs_info->hash_lock); 2570b86a832SChris Mason search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1); 258f510cfecSChris Mason ret = __btrfs_cow_block(trans, root, buf, parent, 2596702ed49SChris Mason parent_slot, cow_ret, search_start, 0); 260f510cfecSChris Mason return ret; 2612c90e5d6SChris Mason } 2626702ed49SChris Mason 2636b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize) 2646702ed49SChris Mason { 2656b80053dSChris Mason if (blocknr < other && other - (blocknr + blocksize) < 32768) 2666702ed49SChris Mason return 1; 2676b80053dSChris Mason if (blocknr > other && blocknr - (other + blocksize) < 32768) 2686702ed49SChris Mason return 1; 26902217ed2SChris Mason return 0; 27002217ed2SChris Mason } 27102217ed2SChris Mason 272081e9573SChris Mason /* 273081e9573SChris Mason * compare two keys in a memcmp fashion 274081e9573SChris Mason */ 275081e9573SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2) 276081e9573SChris Mason { 277081e9573SChris Mason struct btrfs_key k1; 278081e9573SChris Mason 279081e9573SChris Mason btrfs_disk_key_to_cpu(&k1, disk); 280081e9573SChris Mason 281081e9573SChris Mason if (k1.objectid > k2->objectid) 282081e9573SChris Mason return 1; 283081e9573SChris Mason if (k1.objectid < k2->objectid) 284081e9573SChris Mason return -1; 285081e9573SChris Mason if (k1.type > k2->type) 286081e9573SChris Mason return 1; 287081e9573SChris Mason if (k1.type < k2->type) 288081e9573SChris Mason return -1; 289081e9573SChris Mason if (k1.offset > k2->offset) 290081e9573SChris Mason return 1; 291081e9573SChris Mason if (k1.offset < k2->offset) 292081e9573SChris Mason return -1; 293081e9573SChris Mason return 0; 294081e9573SChris Mason } 295081e9573SChris Mason 296081e9573SChris Mason 2976702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans, 2985f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *parent, 299a6b6e75eSChris Mason int start_slot, int cache_only, u64 *last_ret, 300a6b6e75eSChris Mason struct btrfs_key *progress) 3016702ed49SChris Mason { 3026b80053dSChris Mason struct extent_buffer *cur; 3036b80053dSChris Mason struct extent_buffer *tmp; 3046702ed49SChris Mason u64 blocknr; 305e9d0b13bSChris Mason u64 search_start = *last_ret; 306e9d0b13bSChris Mason u64 last_block = 0; 3076702ed49SChris Mason u64 other; 3086702ed49SChris Mason u32 parent_nritems; 3096702ed49SChris Mason int end_slot; 3106702ed49SChris Mason int i; 3116702ed49SChris Mason int err = 0; 312f2183bdeSChris Mason int parent_level; 3136b80053dSChris Mason int uptodate; 3146b80053dSChris Mason u32 blocksize; 315081e9573SChris Mason int progress_passed = 0; 316081e9573SChris Mason struct btrfs_disk_key disk_key; 3176702ed49SChris Mason 3185708b959SChris Mason parent_level = btrfs_header_level(parent); 3195708b959SChris Mason if (cache_only && parent_level != 1) 3205708b959SChris Mason return 0; 3215708b959SChris Mason 3226702ed49SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 3236702ed49SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 3246702ed49SChris Mason root->fs_info->running_transaction->transid); 3256702ed49SChris Mason WARN_ON(1); 3266702ed49SChris Mason } 3276702ed49SChris Mason if (trans->transid != root->fs_info->generation) { 3286702ed49SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 3296702ed49SChris Mason root->fs_info->generation); 3306702ed49SChris Mason WARN_ON(1); 3316702ed49SChris Mason } 33286479a04SChris Mason 3336b80053dSChris Mason parent_nritems = btrfs_header_nritems(parent); 3346b80053dSChris Mason blocksize = btrfs_level_size(root, parent_level - 1); 3356702ed49SChris Mason end_slot = parent_nritems; 3366702ed49SChris Mason 3376702ed49SChris Mason if (parent_nritems == 1) 3386702ed49SChris Mason return 0; 3396702ed49SChris Mason 3406702ed49SChris Mason for (i = start_slot; i < end_slot; i++) { 3416702ed49SChris Mason int close = 1; 342a6b6e75eSChris Mason 3435708b959SChris Mason if (!parent->map_token) { 3445708b959SChris Mason map_extent_buffer(parent, 3455708b959SChris Mason btrfs_node_key_ptr_offset(i), 3465708b959SChris Mason sizeof(struct btrfs_key_ptr), 3475708b959SChris Mason &parent->map_token, &parent->kaddr, 3485708b959SChris Mason &parent->map_start, &parent->map_len, 3495708b959SChris Mason KM_USER1); 3505708b959SChris Mason } 351081e9573SChris Mason btrfs_node_key(parent, &disk_key, i); 352081e9573SChris Mason if (!progress_passed && comp_keys(&disk_key, progress) < 0) 353081e9573SChris Mason continue; 354081e9573SChris Mason 355081e9573SChris Mason progress_passed = 1; 3566b80053dSChris Mason blocknr = btrfs_node_blockptr(parent, i); 357e9d0b13bSChris Mason if (last_block == 0) 358e9d0b13bSChris Mason last_block = blocknr; 3595708b959SChris Mason 3606702ed49SChris Mason if (i > 0) { 3616b80053dSChris Mason other = btrfs_node_blockptr(parent, i - 1); 3626b80053dSChris Mason close = close_blocks(blocknr, other, blocksize); 3636702ed49SChris Mason } 3645708b959SChris Mason if (close && i < end_slot - 2) { 3656b80053dSChris Mason other = btrfs_node_blockptr(parent, i + 1); 3666b80053dSChris Mason close = close_blocks(blocknr, other, blocksize); 3676702ed49SChris Mason } 368e9d0b13bSChris Mason if (close) { 369e9d0b13bSChris Mason last_block = blocknr; 3706702ed49SChris Mason continue; 371e9d0b13bSChris Mason } 3725708b959SChris Mason if (parent->map_token) { 3735708b959SChris Mason unmap_extent_buffer(parent, parent->map_token, 3745708b959SChris Mason KM_USER1); 3755708b959SChris Mason parent->map_token = NULL; 3765708b959SChris Mason } 3776702ed49SChris Mason 3786b80053dSChris Mason cur = btrfs_find_tree_block(root, blocknr, blocksize); 3796b80053dSChris Mason if (cur) 3806b80053dSChris Mason uptodate = btrfs_buffer_uptodate(cur); 3816b80053dSChris Mason else 3826b80053dSChris Mason uptodate = 0; 3835708b959SChris Mason if (!cur || !uptodate) { 3846702ed49SChris Mason if (cache_only) { 3856b80053dSChris Mason free_extent_buffer(cur); 3866702ed49SChris Mason continue; 3876702ed49SChris Mason } 3886b80053dSChris Mason if (!cur) { 3896b80053dSChris Mason cur = read_tree_block(root, blocknr, 3906b80053dSChris Mason blocksize); 3916b80053dSChris Mason } else if (!uptodate) { 3926b80053dSChris Mason btrfs_read_buffer(cur); 3936702ed49SChris Mason } 394f2183bdeSChris Mason } 395e9d0b13bSChris Mason if (search_start == 0) 3966b80053dSChris Mason search_start = last_block; 397e9d0b13bSChris Mason 3980999df54SChris Mason btrfs_verify_block_csum(root, cur); 3996b80053dSChris Mason err = __btrfs_cow_block(trans, root, cur, parent, i, 4006b80053dSChris Mason &tmp, search_start, 4016b80053dSChris Mason min(16 * blocksize, 4026b80053dSChris Mason (end_slot - i) * blocksize)); 403252c38f0SYan if (err) { 4046b80053dSChris Mason free_extent_buffer(cur); 4056702ed49SChris Mason break; 406252c38f0SYan } 4076b80053dSChris Mason search_start = tmp->start; 4085708b959SChris Mason last_block = tmp->start; 409f2183bdeSChris Mason *last_ret = search_start; 410f2183bdeSChris Mason if (parent_level == 1) 4116b80053dSChris Mason btrfs_clear_buffer_defrag(tmp); 4126b80053dSChris Mason free_extent_buffer(tmp); 4136702ed49SChris Mason } 4145708b959SChris Mason if (parent->map_token) { 4155708b959SChris Mason unmap_extent_buffer(parent, parent->map_token, 4165708b959SChris Mason KM_USER1); 4175708b959SChris Mason parent->map_token = NULL; 4185708b959SChris Mason } 4196702ed49SChris Mason return err; 4206702ed49SChris Mason } 4216702ed49SChris Mason 42274123bd7SChris Mason /* 42374123bd7SChris Mason * The leaf data grows from end-to-front in the node. 42474123bd7SChris Mason * this returns the address of the start of the last item, 42574123bd7SChris Mason * which is the stop of the leaf data stack 42674123bd7SChris Mason */ 427123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root, 4285f39d397SChris Mason struct extent_buffer *leaf) 429be0e5c09SChris Mason { 4305f39d397SChris Mason u32 nr = btrfs_header_nritems(leaf); 431be0e5c09SChris Mason if (nr == 0) 432123abc88SChris Mason return BTRFS_LEAF_DATA_SIZE(root); 4335f39d397SChris Mason return btrfs_item_offset_nr(leaf, nr - 1); 434be0e5c09SChris Mason } 435be0e5c09SChris Mason 436123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path, 437123abc88SChris Mason int level) 438aa5d6bedSChris Mason { 4395f39d397SChris Mason struct extent_buffer *parent = NULL; 4405f39d397SChris Mason struct extent_buffer *node = path->nodes[level]; 4415f39d397SChris Mason struct btrfs_disk_key parent_key; 4425f39d397SChris Mason struct btrfs_disk_key node_key; 443aa5d6bedSChris Mason int parent_slot; 4448d7be552SChris Mason int slot; 4458d7be552SChris Mason struct btrfs_key cpukey; 4465f39d397SChris Mason u32 nritems = btrfs_header_nritems(node); 447aa5d6bedSChris Mason 448aa5d6bedSChris Mason if (path->nodes[level + 1]) 4495f39d397SChris Mason parent = path->nodes[level + 1]; 450a1f39630SAneesh 4518d7be552SChris Mason slot = path->slots[level]; 4527518a238SChris Mason BUG_ON(nritems == 0); 4537518a238SChris Mason if (parent) { 454a1f39630SAneesh parent_slot = path->slots[level + 1]; 4555f39d397SChris Mason btrfs_node_key(parent, &parent_key, parent_slot); 4565f39d397SChris Mason btrfs_node_key(node, &node_key, 0); 4575f39d397SChris Mason BUG_ON(memcmp(&parent_key, &node_key, 458e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 4591d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 460db94535dSChris Mason btrfs_header_bytenr(node)); 461aa5d6bedSChris Mason } 462123abc88SChris Mason BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root)); 4638d7be552SChris Mason if (slot != 0) { 4645f39d397SChris Mason btrfs_node_key_to_cpu(node, &cpukey, slot - 1); 4655f39d397SChris Mason btrfs_node_key(node, &node_key, slot); 4665f39d397SChris Mason BUG_ON(comp_keys(&node_key, &cpukey) <= 0); 4678d7be552SChris Mason } 4688d7be552SChris Mason if (slot < nritems - 1) { 4695f39d397SChris Mason btrfs_node_key_to_cpu(node, &cpukey, slot + 1); 4705f39d397SChris Mason btrfs_node_key(node, &node_key, slot); 4715f39d397SChris Mason BUG_ON(comp_keys(&node_key, &cpukey) >= 0); 472aa5d6bedSChris Mason } 473aa5d6bedSChris Mason return 0; 474aa5d6bedSChris Mason } 475aa5d6bedSChris Mason 476123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path, 477123abc88SChris Mason int level) 478aa5d6bedSChris Mason { 4795f39d397SChris Mason struct extent_buffer *leaf = path->nodes[level]; 4805f39d397SChris Mason struct extent_buffer *parent = NULL; 481aa5d6bedSChris Mason int parent_slot; 4828d7be552SChris Mason struct btrfs_key cpukey; 4835f39d397SChris Mason struct btrfs_disk_key parent_key; 4845f39d397SChris Mason struct btrfs_disk_key leaf_key; 4855f39d397SChris Mason int slot = path->slots[0]; 4868d7be552SChris Mason 4875f39d397SChris Mason u32 nritems = btrfs_header_nritems(leaf); 488aa5d6bedSChris Mason 489aa5d6bedSChris Mason if (path->nodes[level + 1]) 4905f39d397SChris Mason parent = path->nodes[level + 1]; 4917518a238SChris Mason 4927518a238SChris Mason if (nritems == 0) 4937518a238SChris Mason return 0; 4947518a238SChris Mason 4957518a238SChris Mason if (parent) { 496a1f39630SAneesh parent_slot = path->slots[level + 1]; 4975f39d397SChris Mason btrfs_node_key(parent, &parent_key, parent_slot); 4985f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, 0); 4996702ed49SChris Mason 5005f39d397SChris Mason BUG_ON(memcmp(&parent_key, &leaf_key, 501e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 5021d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 503db94535dSChris Mason btrfs_header_bytenr(leaf)); 504aa5d6bedSChris Mason } 5055f39d397SChris Mason #if 0 5065f39d397SChris Mason for (i = 0; nritems > 1 && i < nritems - 2; i++) { 5075f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, i + 1); 5085f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, i); 5095f39d397SChris Mason if (comp_keys(&leaf_key, &cpukey) >= 0) { 5105f39d397SChris Mason btrfs_print_leaf(root, leaf); 5115f39d397SChris Mason printk("slot %d offset bad key\n", i); 5125f39d397SChris Mason BUG_ON(1); 5135f39d397SChris Mason } 5145f39d397SChris Mason if (btrfs_item_offset_nr(leaf, i) != 5155f39d397SChris Mason btrfs_item_end_nr(leaf, i + 1)) { 5165f39d397SChris Mason btrfs_print_leaf(root, leaf); 5175f39d397SChris Mason printk("slot %d offset bad\n", i); 5185f39d397SChris Mason BUG_ON(1); 5195f39d397SChris Mason } 5205f39d397SChris Mason if (i == 0) { 5215f39d397SChris Mason if (btrfs_item_offset_nr(leaf, i) + 5225f39d397SChris Mason btrfs_item_size_nr(leaf, i) != 5235f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 5245f39d397SChris Mason btrfs_print_leaf(root, leaf); 5255f39d397SChris Mason printk("slot %d first offset bad\n", i); 5265f39d397SChris Mason BUG_ON(1); 5275f39d397SChris Mason } 5285f39d397SChris Mason } 5295f39d397SChris Mason } 5305f39d397SChris Mason if (nritems > 0) { 5315f39d397SChris Mason if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) { 5325f39d397SChris Mason btrfs_print_leaf(root, leaf); 5335f39d397SChris Mason printk("slot %d bad size \n", nritems - 1); 5345f39d397SChris Mason BUG_ON(1); 5355f39d397SChris Mason } 5365f39d397SChris Mason } 5375f39d397SChris Mason #endif 5385f39d397SChris Mason if (slot != 0 && slot < nritems - 1) { 5395f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, slot); 5405f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1); 5415f39d397SChris Mason if (comp_keys(&leaf_key, &cpukey) <= 0) { 5425f39d397SChris Mason btrfs_print_leaf(root, leaf); 5435f39d397SChris Mason printk("slot %d offset bad key\n", slot); 5445f39d397SChris Mason BUG_ON(1); 5455f39d397SChris Mason } 5465f39d397SChris Mason if (btrfs_item_offset_nr(leaf, slot - 1) != 5475f39d397SChris Mason btrfs_item_end_nr(leaf, slot)) { 5485f39d397SChris Mason btrfs_print_leaf(root, leaf); 5495f39d397SChris Mason printk("slot %d offset bad\n", slot); 5505f39d397SChris Mason BUG_ON(1); 5515f39d397SChris Mason } 552aa5d6bedSChris Mason } 5538d7be552SChris Mason if (slot < nritems - 1) { 5545f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, slot); 5555f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1); 5565f39d397SChris Mason BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0); 5575f39d397SChris Mason if (btrfs_item_offset_nr(leaf, slot) != 5585f39d397SChris Mason btrfs_item_end_nr(leaf, slot + 1)) { 5595f39d397SChris Mason btrfs_print_leaf(root, leaf); 5605f39d397SChris Mason printk("slot %d offset bad\n", slot); 5615f39d397SChris Mason BUG_ON(1); 562aa5d6bedSChris Mason } 5635f39d397SChris Mason } 5645f39d397SChris Mason BUG_ON(btrfs_item_offset_nr(leaf, 0) + 5655f39d397SChris Mason btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root)); 566aa5d6bedSChris Mason return 0; 567aa5d6bedSChris Mason } 568aa5d6bedSChris Mason 56998ed5174SChris Mason static int noinline check_block(struct btrfs_root *root, 57098ed5174SChris Mason struct btrfs_path *path, int level) 571aa5d6bedSChris Mason { 572f188591eSChris Mason u64 found_start; 57385d824c4SChris Mason return 0; 574f188591eSChris Mason if (btrfs_header_level(path->nodes[level]) != level) 575f188591eSChris Mason printk("warning: bad level %Lu wanted %d found %d\n", 576f188591eSChris Mason path->nodes[level]->start, level, 577f188591eSChris Mason btrfs_header_level(path->nodes[level])); 578f188591eSChris Mason found_start = btrfs_header_bytenr(path->nodes[level]); 579f188591eSChris Mason if (found_start != path->nodes[level]->start) { 580f188591eSChris Mason printk("warning: bad bytentr %Lu found %Lu\n", 581f188591eSChris Mason path->nodes[level]->start, found_start); 582f188591eSChris Mason } 583db94535dSChris Mason #if 0 5845f39d397SChris Mason struct extent_buffer *buf = path->nodes[level]; 5855f39d397SChris Mason 586479965d6SChris Mason if (memcmp_extent_buffer(buf, root->fs_info->fsid, 587479965d6SChris Mason (unsigned long)btrfs_header_fsid(buf), 588479965d6SChris Mason BTRFS_FSID_SIZE)) { 5895f39d397SChris Mason printk("warning bad block %Lu\n", buf->start); 590db94535dSChris Mason return 1; 5915f39d397SChris Mason } 592db94535dSChris Mason #endif 593aa5d6bedSChris Mason if (level == 0) 594123abc88SChris Mason return check_leaf(root, path, level); 595123abc88SChris Mason return check_node(root, path, level); 596aa5d6bedSChris Mason } 597aa5d6bedSChris Mason 59874123bd7SChris Mason /* 5995f39d397SChris Mason * search for key in the extent_buffer. The items start at offset p, 6005f39d397SChris Mason * and they are item_size apart. There are 'max' items in p. 6015f39d397SChris Mason * 60274123bd7SChris Mason * the slot in the array is returned via slot, and it points to 60374123bd7SChris Mason * the place where you would insert key if it is not found in 60474123bd7SChris Mason * the array. 60574123bd7SChris Mason * 60674123bd7SChris Mason * slot may point to max if the key is bigger than all of the keys 60774123bd7SChris Mason */ 6085f39d397SChris Mason static int generic_bin_search(struct extent_buffer *eb, unsigned long p, 6095f39d397SChris Mason int item_size, struct btrfs_key *key, 610be0e5c09SChris Mason int max, int *slot) 611be0e5c09SChris Mason { 612be0e5c09SChris Mason int low = 0; 613be0e5c09SChris Mason int high = max; 614be0e5c09SChris Mason int mid; 615be0e5c09SChris Mason int ret; 616479965d6SChris Mason struct btrfs_disk_key *tmp = NULL; 6175f39d397SChris Mason struct btrfs_disk_key unaligned; 6185f39d397SChris Mason unsigned long offset; 6195f39d397SChris Mason char *map_token = NULL; 6205f39d397SChris Mason char *kaddr = NULL; 6215f39d397SChris Mason unsigned long map_start = 0; 6225f39d397SChris Mason unsigned long map_len = 0; 623479965d6SChris Mason int err; 624be0e5c09SChris Mason 625be0e5c09SChris Mason while(low < high) { 626be0e5c09SChris Mason mid = (low + high) / 2; 6275f39d397SChris Mason offset = p + mid * item_size; 6285f39d397SChris Mason 6295f39d397SChris Mason if (!map_token || offset < map_start || 6305f39d397SChris Mason (offset + sizeof(struct btrfs_disk_key)) > 6315f39d397SChris Mason map_start + map_len) { 632479965d6SChris Mason if (map_token) { 6335f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 634479965d6SChris Mason map_token = NULL; 635479965d6SChris Mason } 636479965d6SChris Mason err = map_extent_buffer(eb, offset, 637479965d6SChris Mason sizeof(struct btrfs_disk_key), 638479965d6SChris Mason &map_token, &kaddr, 6395f39d397SChris Mason &map_start, &map_len, KM_USER0); 6405f39d397SChris Mason 641479965d6SChris Mason if (!err) { 642479965d6SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 643479965d6SChris Mason map_start); 644479965d6SChris Mason } else { 6455f39d397SChris Mason read_extent_buffer(eb, &unaligned, 6465f39d397SChris Mason offset, sizeof(unaligned)); 6475f39d397SChris Mason tmp = &unaligned; 648479965d6SChris Mason } 649479965d6SChris Mason 6505f39d397SChris Mason } else { 6515f39d397SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 6525f39d397SChris Mason map_start); 6535f39d397SChris Mason } 654be0e5c09SChris Mason ret = comp_keys(tmp, key); 655be0e5c09SChris Mason 656be0e5c09SChris Mason if (ret < 0) 657be0e5c09SChris Mason low = mid + 1; 658be0e5c09SChris Mason else if (ret > 0) 659be0e5c09SChris Mason high = mid; 660be0e5c09SChris Mason else { 661be0e5c09SChris Mason *slot = mid; 662479965d6SChris Mason if (map_token) 6635f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 664be0e5c09SChris Mason return 0; 665be0e5c09SChris Mason } 666be0e5c09SChris Mason } 667be0e5c09SChris Mason *slot = low; 6685f39d397SChris Mason if (map_token) 6695f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 670be0e5c09SChris Mason return 1; 671be0e5c09SChris Mason } 672be0e5c09SChris Mason 67397571fd0SChris Mason /* 67497571fd0SChris Mason * simple bin_search frontend that does the right thing for 67597571fd0SChris Mason * leaves vs nodes 67697571fd0SChris Mason */ 6775f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key, 6785f39d397SChris Mason int level, int *slot) 679be0e5c09SChris Mason { 6805f39d397SChris Mason if (level == 0) { 6815f39d397SChris Mason return generic_bin_search(eb, 6825f39d397SChris Mason offsetof(struct btrfs_leaf, items), 6830783fcfcSChris Mason sizeof(struct btrfs_item), 6845f39d397SChris Mason key, btrfs_header_nritems(eb), 6857518a238SChris Mason slot); 686be0e5c09SChris Mason } else { 6875f39d397SChris Mason return generic_bin_search(eb, 6885f39d397SChris Mason offsetof(struct btrfs_node, ptrs), 689123abc88SChris Mason sizeof(struct btrfs_key_ptr), 6905f39d397SChris Mason key, btrfs_header_nritems(eb), 6917518a238SChris Mason slot); 692be0e5c09SChris Mason } 693be0e5c09SChris Mason return -1; 694be0e5c09SChris Mason } 695be0e5c09SChris Mason 6965f39d397SChris Mason static struct extent_buffer *read_node_slot(struct btrfs_root *root, 6975f39d397SChris Mason struct extent_buffer *parent, int slot) 698bb803951SChris Mason { 699bb803951SChris Mason if (slot < 0) 700bb803951SChris Mason return NULL; 7015f39d397SChris Mason if (slot >= btrfs_header_nritems(parent)) 702bb803951SChris Mason return NULL; 703db94535dSChris Mason return read_tree_block(root, btrfs_node_blockptr(parent, slot), 704db94535dSChris Mason btrfs_level_size(root, btrfs_header_level(parent) - 1)); 705bb803951SChris Mason } 706bb803951SChris Mason 70798ed5174SChris Mason static int balance_level(struct btrfs_trans_handle *trans, 70898ed5174SChris Mason struct btrfs_root *root, 70998ed5174SChris Mason struct btrfs_path *path, int level) 710bb803951SChris Mason { 7115f39d397SChris Mason struct extent_buffer *right = NULL; 7125f39d397SChris Mason struct extent_buffer *mid; 7135f39d397SChris Mason struct extent_buffer *left = NULL; 7145f39d397SChris Mason struct extent_buffer *parent = NULL; 715bb803951SChris Mason int ret = 0; 716bb803951SChris Mason int wret; 717bb803951SChris Mason int pslot; 718bb803951SChris Mason int orig_slot = path->slots[level]; 71954aa1f4dSChris Mason int err_on_enospc = 0; 72079f95c82SChris Mason u64 orig_ptr; 721bb803951SChris Mason 722bb803951SChris Mason if (level == 0) 723bb803951SChris Mason return 0; 724bb803951SChris Mason 7255f39d397SChris Mason mid = path->nodes[level]; 7267bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 7277bb86316SChris Mason 7281d4f8a0cSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 72979f95c82SChris Mason 730234b63a0SChris Mason if (level < BTRFS_MAX_LEVEL - 1) 7315f39d397SChris Mason parent = path->nodes[level + 1]; 732bb803951SChris Mason pslot = path->slots[level + 1]; 733bb803951SChris Mason 73440689478SChris Mason /* 73540689478SChris Mason * deal with the case where there is only one pointer in the root 73640689478SChris Mason * by promoting the node below to a root 73740689478SChris Mason */ 7385f39d397SChris Mason if (!parent) { 7395f39d397SChris Mason struct extent_buffer *child; 740bb803951SChris Mason 7415f39d397SChris Mason if (btrfs_header_nritems(mid) != 1) 742bb803951SChris Mason return 0; 743bb803951SChris Mason 744bb803951SChris Mason /* promote the child to a root */ 7455f39d397SChris Mason child = read_node_slot(root, mid, 0); 746bb803951SChris Mason BUG_ON(!child); 7472f375ab9SYan ret = btrfs_cow_block(trans, root, child, mid, 0, &child); 7482f375ab9SYan BUG_ON(ret); 7492f375ab9SYan 750bb803951SChris Mason root->node = child; 7510b86a832SChris Mason add_root_to_dirty_list(root); 752bb803951SChris Mason path->nodes[level] = NULL; 7535f39d397SChris Mason clean_tree_block(trans, root, mid); 754bb803951SChris Mason /* once for the path */ 7555f39d397SChris Mason free_extent_buffer(mid); 7567bb86316SChris Mason ret = btrfs_free_extent(trans, root, mid->start, mid->len, 7577bb86316SChris Mason root->root_key.objectid, 7587bb86316SChris Mason btrfs_header_generation(mid), 0, 0, 1); 759bb803951SChris Mason /* once for the root ptr */ 7605f39d397SChris Mason free_extent_buffer(mid); 761db94535dSChris Mason return ret; 762bb803951SChris Mason } 7635f39d397SChris Mason if (btrfs_header_nritems(mid) > 764123abc88SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) / 4) 765bb803951SChris Mason return 0; 766bb803951SChris Mason 7675f39d397SChris Mason if (btrfs_header_nritems(mid) < 2) 76854aa1f4dSChris Mason err_on_enospc = 1; 76954aa1f4dSChris Mason 7705f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 7715f39d397SChris Mason if (left) { 7725f39d397SChris Mason wret = btrfs_cow_block(trans, root, left, 7735f39d397SChris Mason parent, pslot - 1, &left); 77454aa1f4dSChris Mason if (wret) { 77554aa1f4dSChris Mason ret = wret; 77654aa1f4dSChris Mason goto enospc; 77754aa1f4dSChris Mason } 7782cc58cf2SChris Mason } 7795f39d397SChris Mason right = read_node_slot(root, parent, pslot + 1); 7805f39d397SChris Mason if (right) { 7815f39d397SChris Mason wret = btrfs_cow_block(trans, root, right, 7825f39d397SChris Mason parent, pslot + 1, &right); 7832cc58cf2SChris Mason if (wret) { 7842cc58cf2SChris Mason ret = wret; 7852cc58cf2SChris Mason goto enospc; 7862cc58cf2SChris Mason } 7872cc58cf2SChris Mason } 7882cc58cf2SChris Mason 7892cc58cf2SChris Mason /* first, try to make some room in the middle buffer */ 7905f39d397SChris Mason if (left) { 7915f39d397SChris Mason orig_slot += btrfs_header_nritems(left); 792*bce4eae9SChris Mason wret = push_node_left(trans, root, left, mid, 1); 79379f95c82SChris Mason if (wret < 0) 79479f95c82SChris Mason ret = wret; 7955f39d397SChris Mason if (btrfs_header_nritems(mid) < 2) 79654aa1f4dSChris Mason err_on_enospc = 1; 797bb803951SChris Mason } 79879f95c82SChris Mason 79979f95c82SChris Mason /* 80079f95c82SChris Mason * then try to empty the right most buffer into the middle 80179f95c82SChris Mason */ 8025f39d397SChris Mason if (right) { 803971a1f66SChris Mason wret = push_node_left(trans, root, mid, right, 1); 80454aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 80579f95c82SChris Mason ret = wret; 8065f39d397SChris Mason if (btrfs_header_nritems(right) == 0) { 807db94535dSChris Mason u64 bytenr = right->start; 8087bb86316SChris Mason u64 generation = btrfs_header_generation(parent); 809db94535dSChris Mason u32 blocksize = right->len; 810db94535dSChris Mason 8115f39d397SChris Mason clean_tree_block(trans, root, right); 8125f39d397SChris Mason free_extent_buffer(right); 813bb803951SChris Mason right = NULL; 814e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot + 815e089f05cSChris Mason 1); 816bb803951SChris Mason if (wret) 817bb803951SChris Mason ret = wret; 818db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 8197bb86316SChris Mason blocksize, 8207bb86316SChris Mason btrfs_header_owner(parent), 8217bb86316SChris Mason generation, 0, 0, 1); 822bb803951SChris Mason if (wret) 823bb803951SChris Mason ret = wret; 824bb803951SChris Mason } else { 8255f39d397SChris Mason struct btrfs_disk_key right_key; 8265f39d397SChris Mason btrfs_node_key(right, &right_key, 0); 8275f39d397SChris Mason btrfs_set_node_key(parent, &right_key, pslot + 1); 8285f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 829bb803951SChris Mason } 830bb803951SChris Mason } 8315f39d397SChris Mason if (btrfs_header_nritems(mid) == 1) { 83279f95c82SChris Mason /* 83379f95c82SChris Mason * we're not allowed to leave a node with one item in the 83479f95c82SChris Mason * tree during a delete. A deletion from lower in the tree 83579f95c82SChris Mason * could try to delete the only pointer in this node. 83679f95c82SChris Mason * So, pull some keys from the left. 83779f95c82SChris Mason * There has to be a left pointer at this point because 83879f95c82SChris Mason * otherwise we would have pulled some pointers from the 83979f95c82SChris Mason * right 84079f95c82SChris Mason */ 8415f39d397SChris Mason BUG_ON(!left); 8425f39d397SChris Mason wret = balance_node_right(trans, root, mid, left); 84354aa1f4dSChris Mason if (wret < 0) { 84479f95c82SChris Mason ret = wret; 84554aa1f4dSChris Mason goto enospc; 84654aa1f4dSChris Mason } 847*bce4eae9SChris Mason if (wret == 1) { 848*bce4eae9SChris Mason wret = push_node_left(trans, root, left, mid, 1); 849*bce4eae9SChris Mason if (wret < 0) 850*bce4eae9SChris Mason ret = wret; 851*bce4eae9SChris Mason } 85279f95c82SChris Mason BUG_ON(wret == 1); 85379f95c82SChris Mason } 8545f39d397SChris Mason if (btrfs_header_nritems(mid) == 0) { 85579f95c82SChris Mason /* we've managed to empty the middle node, drop it */ 8567bb86316SChris Mason u64 root_gen = btrfs_header_generation(parent); 857db94535dSChris Mason u64 bytenr = mid->start; 858db94535dSChris Mason u32 blocksize = mid->len; 8595f39d397SChris Mason clean_tree_block(trans, root, mid); 8605f39d397SChris Mason free_extent_buffer(mid); 861bb803951SChris Mason mid = NULL; 862e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot); 863bb803951SChris Mason if (wret) 864bb803951SChris Mason ret = wret; 8657bb86316SChris Mason wret = btrfs_free_extent(trans, root, bytenr, blocksize, 8667bb86316SChris Mason btrfs_header_owner(parent), 8677bb86316SChris Mason root_gen, 0, 0, 1); 868bb803951SChris Mason if (wret) 869bb803951SChris Mason ret = wret; 87079f95c82SChris Mason } else { 87179f95c82SChris Mason /* update the parent key to reflect our changes */ 8725f39d397SChris Mason struct btrfs_disk_key mid_key; 8735f39d397SChris Mason btrfs_node_key(mid, &mid_key, 0); 8745f39d397SChris Mason btrfs_set_node_key(parent, &mid_key, pslot); 8755f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 87679f95c82SChris Mason } 877bb803951SChris Mason 87879f95c82SChris Mason /* update the path */ 8795f39d397SChris Mason if (left) { 8805f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 8815f39d397SChris Mason extent_buffer_get(left); 8825f39d397SChris Mason path->nodes[level] = left; 883bb803951SChris Mason path->slots[level + 1] -= 1; 884bb803951SChris Mason path->slots[level] = orig_slot; 8855f39d397SChris Mason if (mid) 8865f39d397SChris Mason free_extent_buffer(mid); 887bb803951SChris Mason } else { 8885f39d397SChris Mason orig_slot -= btrfs_header_nritems(left); 889bb803951SChris Mason path->slots[level] = orig_slot; 890bb803951SChris Mason } 891bb803951SChris Mason } 89279f95c82SChris Mason /* double check we haven't messed things up */ 893123abc88SChris Mason check_block(root, path, level); 894e20d96d6SChris Mason if (orig_ptr != 8955f39d397SChris Mason btrfs_node_blockptr(path->nodes[level], path->slots[level])) 89679f95c82SChris Mason BUG(); 89754aa1f4dSChris Mason enospc: 8985f39d397SChris Mason if (right) 8995f39d397SChris Mason free_extent_buffer(right); 9005f39d397SChris Mason if (left) 9015f39d397SChris Mason free_extent_buffer(left); 902bb803951SChris Mason return ret; 903bb803951SChris Mason } 904bb803951SChris Mason 905e66f709bSChris Mason /* returns zero if the push worked, non-zero otherwise */ 90698ed5174SChris Mason static int noinline push_nodes_for_insert(struct btrfs_trans_handle *trans, 907e66f709bSChris Mason struct btrfs_root *root, 908e66f709bSChris Mason struct btrfs_path *path, int level) 909e66f709bSChris Mason { 9105f39d397SChris Mason struct extent_buffer *right = NULL; 9115f39d397SChris Mason struct extent_buffer *mid; 9125f39d397SChris Mason struct extent_buffer *left = NULL; 9135f39d397SChris Mason struct extent_buffer *parent = NULL; 914e66f709bSChris Mason int ret = 0; 915e66f709bSChris Mason int wret; 916e66f709bSChris Mason int pslot; 917e66f709bSChris Mason int orig_slot = path->slots[level]; 918e66f709bSChris Mason u64 orig_ptr; 919e66f709bSChris Mason 920e66f709bSChris Mason if (level == 0) 921e66f709bSChris Mason return 1; 922e66f709bSChris Mason 9235f39d397SChris Mason mid = path->nodes[level]; 9247bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 925e66f709bSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 926e66f709bSChris Mason 927e66f709bSChris Mason if (level < BTRFS_MAX_LEVEL - 1) 9285f39d397SChris Mason parent = path->nodes[level + 1]; 929e66f709bSChris Mason pslot = path->slots[level + 1]; 930e66f709bSChris Mason 9315f39d397SChris Mason if (!parent) 932e66f709bSChris Mason return 1; 933e66f709bSChris Mason 9345f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 935e66f709bSChris Mason 936e66f709bSChris Mason /* first, try to make some room in the middle buffer */ 9375f39d397SChris Mason if (left) { 938e66f709bSChris Mason u32 left_nr; 9395f39d397SChris Mason left_nr = btrfs_header_nritems(left); 94033ade1f8SChris Mason if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 94133ade1f8SChris Mason wret = 1; 94233ade1f8SChris Mason } else { 9435f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, parent, 9445f39d397SChris Mason pslot - 1, &left); 94554aa1f4dSChris Mason if (ret) 94654aa1f4dSChris Mason wret = 1; 94754aa1f4dSChris Mason else { 94854aa1f4dSChris Mason wret = push_node_left(trans, root, 949971a1f66SChris Mason left, mid, 0); 95054aa1f4dSChris Mason } 95133ade1f8SChris Mason } 952e66f709bSChris Mason if (wret < 0) 953e66f709bSChris Mason ret = wret; 954e66f709bSChris Mason if (wret == 0) { 9555f39d397SChris Mason struct btrfs_disk_key disk_key; 956e66f709bSChris Mason orig_slot += left_nr; 9575f39d397SChris Mason btrfs_node_key(mid, &disk_key, 0); 9585f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot); 9595f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 9605f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 9615f39d397SChris Mason path->nodes[level] = left; 962e66f709bSChris Mason path->slots[level + 1] -= 1; 963e66f709bSChris Mason path->slots[level] = orig_slot; 9645f39d397SChris Mason free_extent_buffer(mid); 965e66f709bSChris Mason } else { 966e66f709bSChris Mason orig_slot -= 9675f39d397SChris Mason btrfs_header_nritems(left); 968e66f709bSChris Mason path->slots[level] = orig_slot; 9695f39d397SChris Mason free_extent_buffer(left); 970e66f709bSChris Mason } 971e66f709bSChris Mason return 0; 972e66f709bSChris Mason } 9735f39d397SChris Mason free_extent_buffer(left); 974e66f709bSChris Mason } 9755f39d397SChris Mason right= read_node_slot(root, parent, pslot + 1); 976e66f709bSChris Mason 977e66f709bSChris Mason /* 978e66f709bSChris Mason * then try to empty the right most buffer into the middle 979e66f709bSChris Mason */ 9805f39d397SChris Mason if (right) { 98133ade1f8SChris Mason u32 right_nr; 9825f39d397SChris Mason right_nr = btrfs_header_nritems(right); 98333ade1f8SChris Mason if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 98433ade1f8SChris Mason wret = 1; 98533ade1f8SChris Mason } else { 9865f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, 9875f39d397SChris Mason parent, pslot + 1, 9885f39d397SChris Mason &right); 98954aa1f4dSChris Mason if (ret) 99054aa1f4dSChris Mason wret = 1; 99154aa1f4dSChris Mason else { 99233ade1f8SChris Mason wret = balance_node_right(trans, root, 9935f39d397SChris Mason right, mid); 99433ade1f8SChris Mason } 99554aa1f4dSChris Mason } 996e66f709bSChris Mason if (wret < 0) 997e66f709bSChris Mason ret = wret; 998e66f709bSChris Mason if (wret == 0) { 9995f39d397SChris Mason struct btrfs_disk_key disk_key; 10005f39d397SChris Mason 10015f39d397SChris Mason btrfs_node_key(right, &disk_key, 0); 10025f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot + 1); 10035f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 10045f39d397SChris Mason 10055f39d397SChris Mason if (btrfs_header_nritems(mid) <= orig_slot) { 10065f39d397SChris Mason path->nodes[level] = right; 1007e66f709bSChris Mason path->slots[level + 1] += 1; 1008e66f709bSChris Mason path->slots[level] = orig_slot - 10095f39d397SChris Mason btrfs_header_nritems(mid); 10105f39d397SChris Mason free_extent_buffer(mid); 1011e66f709bSChris Mason } else { 10125f39d397SChris Mason free_extent_buffer(right); 1013e66f709bSChris Mason } 1014e66f709bSChris Mason return 0; 1015e66f709bSChris Mason } 10165f39d397SChris Mason free_extent_buffer(right); 1017e66f709bSChris Mason } 1018e66f709bSChris Mason return 1; 1019e66f709bSChris Mason } 1020e66f709bSChris Mason 102174123bd7SChris Mason /* 10223c69faecSChris Mason * readahead one full node of leaves 10233c69faecSChris Mason */ 10243c69faecSChris Mason static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path, 102501f46658SChris Mason int level, int slot, u64 objectid) 10263c69faecSChris Mason { 10275f39d397SChris Mason struct extent_buffer *node; 102801f46658SChris Mason struct btrfs_disk_key disk_key; 10293c69faecSChris Mason u32 nritems; 10303c69faecSChris Mason u64 search; 10316b80053dSChris Mason u64 lowest_read; 10326b80053dSChris Mason u64 highest_read; 10336b80053dSChris Mason u64 nread = 0; 10343c69faecSChris Mason int direction = path->reada; 10355f39d397SChris Mason struct extent_buffer *eb; 10366b80053dSChris Mason u32 nr; 10376b80053dSChris Mason u32 blocksize; 10386b80053dSChris Mason u32 nscan = 0; 1039db94535dSChris Mason 1040a6b6e75eSChris Mason if (level != 1) 10413c69faecSChris Mason return; 10423c69faecSChris Mason 10436702ed49SChris Mason if (!path->nodes[level]) 10446702ed49SChris Mason return; 10456702ed49SChris Mason 10465f39d397SChris Mason node = path->nodes[level]; 10473c69faecSChris Mason search = btrfs_node_blockptr(node, slot); 10486b80053dSChris Mason blocksize = btrfs_level_size(root, level - 1); 10496b80053dSChris Mason eb = btrfs_find_tree_block(root, search, blocksize); 10505f39d397SChris Mason if (eb) { 10515f39d397SChris Mason free_extent_buffer(eb); 10523c69faecSChris Mason return; 10533c69faecSChris Mason } 10543c69faecSChris Mason 10556b80053dSChris Mason highest_read = search; 10566b80053dSChris Mason lowest_read = search; 10576b80053dSChris Mason 10585f39d397SChris Mason nritems = btrfs_header_nritems(node); 10596b80053dSChris Mason nr = slot; 10603c69faecSChris Mason while(1) { 10616b80053dSChris Mason if (direction < 0) { 10626b80053dSChris Mason if (nr == 0) 10633c69faecSChris Mason break; 10646b80053dSChris Mason nr--; 10656b80053dSChris Mason } else if (direction > 0) { 10666b80053dSChris Mason nr++; 10676b80053dSChris Mason if (nr >= nritems) 10686b80053dSChris Mason break; 10693c69faecSChris Mason } 107001f46658SChris Mason if (path->reada < 0 && objectid) { 107101f46658SChris Mason btrfs_node_key(node, &disk_key, nr); 107201f46658SChris Mason if (btrfs_disk_key_objectid(&disk_key) != objectid) 107301f46658SChris Mason break; 107401f46658SChris Mason } 10756b80053dSChris Mason search = btrfs_node_blockptr(node, nr); 10766b80053dSChris Mason if ((search >= lowest_read && search <= highest_read) || 10776b80053dSChris Mason (search < lowest_read && lowest_read - search <= 32768) || 10786b80053dSChris Mason (search > highest_read && search - highest_read <= 32768)) { 10796b80053dSChris Mason readahead_tree_block(root, search, blocksize); 10806b80053dSChris Mason nread += blocksize; 10813c69faecSChris Mason } 10826b80053dSChris Mason nscan++; 10836b80053dSChris Mason if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32)) 10846b80053dSChris Mason break; 10856b80053dSChris Mason if(nread > (1024 * 1024) || nscan > 128) 10866b80053dSChris Mason break; 10876b80053dSChris Mason 10886b80053dSChris Mason if (search < lowest_read) 10896b80053dSChris Mason lowest_read = search; 10906b80053dSChris Mason if (search > highest_read) 10916b80053dSChris Mason highest_read = search; 10923c69faecSChris Mason } 10933c69faecSChris Mason } 10943c69faecSChris Mason /* 109574123bd7SChris Mason * look for key in the tree. path is filled in with nodes along the way 109674123bd7SChris Mason * if key is found, we return zero and you can find the item in the leaf 109774123bd7SChris Mason * level of the path (level 0) 109874123bd7SChris Mason * 109974123bd7SChris Mason * If the key isn't found, the path points to the slot where it should 1100aa5d6bedSChris Mason * be inserted, and 1 is returned. If there are other errors during the 1101aa5d6bedSChris Mason * search a negative error number is returned. 110297571fd0SChris Mason * 110397571fd0SChris Mason * if ins_len > 0, nodes and leaves will be split as we walk down the 110497571fd0SChris Mason * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if 110597571fd0SChris Mason * possible) 110674123bd7SChris Mason */ 1107e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root 1108e089f05cSChris Mason *root, struct btrfs_key *key, struct btrfs_path *p, int 1109e089f05cSChris Mason ins_len, int cow) 1110be0e5c09SChris Mason { 11115f39d397SChris Mason struct extent_buffer *b; 1112db94535dSChris Mason u64 bytenr; 111374493f7aSChris Mason u64 ptr_gen; 1114be0e5c09SChris Mason int slot; 1115be0e5c09SChris Mason int ret; 1116be0e5c09SChris Mason int level; 11173c69faecSChris Mason int should_reada = p->reada; 11189f3a7427SChris Mason u8 lowest_level = 0; 11199f3a7427SChris Mason 11206702ed49SChris Mason lowest_level = p->lowest_level; 11216702ed49SChris Mason WARN_ON(lowest_level && ins_len); 112222b0ebdaSChris Mason WARN_ON(p->nodes[0] != NULL); 112322b0ebdaSChris Mason WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex)); 1124bb803951SChris Mason again: 1125bb803951SChris Mason b = root->node; 11265f39d397SChris Mason extent_buffer_get(b); 1127eb60ceacSChris Mason while (b) { 11285f39d397SChris Mason level = btrfs_header_level(b); 112902217ed2SChris Mason if (cow) { 113002217ed2SChris Mason int wret; 1131e20d96d6SChris Mason wret = btrfs_cow_block(trans, root, b, 1132e20d96d6SChris Mason p->nodes[level + 1], 1133e20d96d6SChris Mason p->slots[level + 1], 1134252c38f0SYan &b); 113554aa1f4dSChris Mason if (wret) { 11365f39d397SChris Mason free_extent_buffer(b); 113754aa1f4dSChris Mason return wret; 113854aa1f4dSChris Mason } 113902217ed2SChris Mason } 114002217ed2SChris Mason BUG_ON(!cow && ins_len); 11415f39d397SChris Mason if (level != btrfs_header_level(b)) 11422c90e5d6SChris Mason WARN_ON(1); 11435f39d397SChris Mason level = btrfs_header_level(b); 1144eb60ceacSChris Mason p->nodes[level] = b; 1145123abc88SChris Mason ret = check_block(root, p, level); 1146aa5d6bedSChris Mason if (ret) 1147aa5d6bedSChris Mason return -1; 11485f39d397SChris Mason ret = bin_search(b, key, level, &slot); 11495f39d397SChris Mason if (level != 0) { 1150be0e5c09SChris Mason if (ret && slot > 0) 1151be0e5c09SChris Mason slot -= 1; 1152be0e5c09SChris Mason p->slots[level] = slot; 11535f39d397SChris Mason if (ins_len > 0 && btrfs_header_nritems(b) >= 11541514794eSChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) { 1155e089f05cSChris Mason int sret = split_node(trans, root, p, level); 11565c680ed6SChris Mason BUG_ON(sret > 0); 11575c680ed6SChris Mason if (sret) 11585c680ed6SChris Mason return sret; 11595c680ed6SChris Mason b = p->nodes[level]; 11605c680ed6SChris Mason slot = p->slots[level]; 1161bb803951SChris Mason } else if (ins_len < 0) { 1162e089f05cSChris Mason int sret = balance_level(trans, root, p, 1163e089f05cSChris Mason level); 1164bb803951SChris Mason if (sret) 1165bb803951SChris Mason return sret; 1166bb803951SChris Mason b = p->nodes[level]; 1167f510cfecSChris Mason if (!b) { 1168f510cfecSChris Mason btrfs_release_path(NULL, p); 1169bb803951SChris Mason goto again; 1170f510cfecSChris Mason } 1171bb803951SChris Mason slot = p->slots[level]; 11725f39d397SChris Mason BUG_ON(btrfs_header_nritems(b) == 1); 11735c680ed6SChris Mason } 11749f3a7427SChris Mason /* this is only true while dropping a snapshot */ 11759f3a7427SChris Mason if (level == lowest_level) 11769f3a7427SChris Mason break; 1177db94535dSChris Mason bytenr = btrfs_node_blockptr(b, slot); 117874493f7aSChris Mason ptr_gen = btrfs_node_ptr_generation(b, slot); 11796702ed49SChris Mason if (should_reada) 118001f46658SChris Mason reada_for_search(root, p, level, slot, 118101f46658SChris Mason key->objectid); 1182db94535dSChris Mason b = read_tree_block(root, bytenr, 1183db94535dSChris Mason btrfs_level_size(root, level - 1)); 118474493f7aSChris Mason if (ptr_gen != btrfs_header_generation(b)) { 118574493f7aSChris Mason printk("block %llu bad gen wanted %llu " 118674493f7aSChris Mason "found %llu\n", 118774493f7aSChris Mason (unsigned long long)b->start, 118874493f7aSChris Mason (unsigned long long)ptr_gen, 118974493f7aSChris Mason (unsigned long long)btrfs_header_generation(b)); 119074493f7aSChris Mason } 1191be0e5c09SChris Mason } else { 1192be0e5c09SChris Mason p->slots[level] = slot; 11935f39d397SChris Mason if (ins_len > 0 && btrfs_leaf_free_space(root, b) < 11940783fcfcSChris Mason sizeof(struct btrfs_item) + ins_len) { 1195d4dbff95SChris Mason int sret = split_leaf(trans, root, key, 1196cc0c5538SChris Mason p, ins_len, ret == 0); 11975c680ed6SChris Mason BUG_ON(sret > 0); 11985c680ed6SChris Mason if (sret) 11995c680ed6SChris Mason return sret; 12005c680ed6SChris Mason } 1201be0e5c09SChris Mason return ret; 1202be0e5c09SChris Mason } 1203be0e5c09SChris Mason } 1204aa5d6bedSChris Mason return 1; 1205be0e5c09SChris Mason } 1206be0e5c09SChris Mason 120774123bd7SChris Mason /* 120874123bd7SChris Mason * adjust the pointers going up the tree, starting at level 120974123bd7SChris Mason * making sure the right key of each node is points to 'key'. 121074123bd7SChris Mason * This is used after shifting pointers to the left, so it stops 121174123bd7SChris Mason * fixing up pointers when a given leaf/node is not in slot 0 of the 121274123bd7SChris Mason * higher levels 1213aa5d6bedSChris Mason * 1214aa5d6bedSChris Mason * If this fails to write a tree block, it returns -1, but continues 1215aa5d6bedSChris Mason * fixing up the blocks in ram so the tree is consistent. 121674123bd7SChris Mason */ 12175f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans, 12185f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 12195f39d397SChris Mason struct btrfs_disk_key *key, int level) 1220be0e5c09SChris Mason { 1221be0e5c09SChris Mason int i; 1222aa5d6bedSChris Mason int ret = 0; 12235f39d397SChris Mason struct extent_buffer *t; 12245f39d397SChris Mason 1225234b63a0SChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 1226be0e5c09SChris Mason int tslot = path->slots[i]; 1227eb60ceacSChris Mason if (!path->nodes[i]) 1228be0e5c09SChris Mason break; 12295f39d397SChris Mason t = path->nodes[i]; 12305f39d397SChris Mason btrfs_set_node_key(t, key, tslot); 1231d6025579SChris Mason btrfs_mark_buffer_dirty(path->nodes[i]); 1232be0e5c09SChris Mason if (tslot != 0) 1233be0e5c09SChris Mason break; 1234be0e5c09SChris Mason } 1235aa5d6bedSChris Mason return ret; 1236be0e5c09SChris Mason } 1237be0e5c09SChris Mason 123874123bd7SChris Mason /* 123974123bd7SChris Mason * try to push data from one node into the next node left in the 124079f95c82SChris Mason * tree. 1241aa5d6bedSChris Mason * 1242aa5d6bedSChris Mason * returns 0 if some ptrs were pushed left, < 0 if there was some horrible 1243aa5d6bedSChris Mason * error, and > 0 if there was no room in the left hand block. 124474123bd7SChris Mason */ 124598ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans, 124698ed5174SChris Mason struct btrfs_root *root, struct extent_buffer *dst, 1247971a1f66SChris Mason struct extent_buffer *src, int empty) 1248be0e5c09SChris Mason { 1249be0e5c09SChris Mason int push_items = 0; 1250bb803951SChris Mason int src_nritems; 1251bb803951SChris Mason int dst_nritems; 1252aa5d6bedSChris Mason int ret = 0; 1253be0e5c09SChris Mason 12545f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 12555f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1256123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 12577bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 12587bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 125954aa1f4dSChris Mason 1260*bce4eae9SChris Mason if (!empty && src_nritems <= 8) 1261971a1f66SChris Mason return 1; 1262971a1f66SChris Mason 1263eb60ceacSChris Mason if (push_items <= 0) { 1264be0e5c09SChris Mason return 1; 1265eb60ceacSChris Mason } 1266be0e5c09SChris Mason 1267*bce4eae9SChris Mason if (empty) { 1268971a1f66SChris Mason push_items = min(src_nritems, push_items); 1269*bce4eae9SChris Mason if (push_items < src_nritems) { 1270*bce4eae9SChris Mason /* leave at least 8 pointers in the node if 1271*bce4eae9SChris Mason * we aren't going to empty it 1272*bce4eae9SChris Mason */ 1273*bce4eae9SChris Mason if (src_nritems - push_items < 8) { 1274*bce4eae9SChris Mason if (push_items <= 8) 1275*bce4eae9SChris Mason return 1; 1276*bce4eae9SChris Mason push_items -= 8; 1277*bce4eae9SChris Mason } 1278*bce4eae9SChris Mason } 1279*bce4eae9SChris Mason } else 1280*bce4eae9SChris Mason push_items = min(src_nritems - 8, push_items); 128179f95c82SChris Mason 12825f39d397SChris Mason copy_extent_buffer(dst, src, 12835f39d397SChris Mason btrfs_node_key_ptr_offset(dst_nritems), 12845f39d397SChris Mason btrfs_node_key_ptr_offset(0), 1285123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 12865f39d397SChris Mason 1287bb803951SChris Mason if (push_items < src_nritems) { 12885f39d397SChris Mason memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0), 12895f39d397SChris Mason btrfs_node_key_ptr_offset(push_items), 1290e2fa7227SChris Mason (src_nritems - push_items) * 1291123abc88SChris Mason sizeof(struct btrfs_key_ptr)); 1292bb803951SChris Mason } 12935f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 12945f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 12955f39d397SChris Mason btrfs_mark_buffer_dirty(src); 12965f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 1297bb803951SChris Mason return ret; 1298be0e5c09SChris Mason } 1299be0e5c09SChris Mason 130097571fd0SChris Mason /* 130179f95c82SChris Mason * try to push data from one node into the next node right in the 130279f95c82SChris Mason * tree. 130379f95c82SChris Mason * 130479f95c82SChris Mason * returns 0 if some ptrs were pushed, < 0 if there was some horrible 130579f95c82SChris Mason * error, and > 0 if there was no room in the right hand block. 130679f95c82SChris Mason * 130779f95c82SChris Mason * this will only push up to 1/2 the contents of the left node over 130879f95c82SChris Mason */ 13095f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 13105f39d397SChris Mason struct btrfs_root *root, 13115f39d397SChris Mason struct extent_buffer *dst, 13125f39d397SChris Mason struct extent_buffer *src) 131379f95c82SChris Mason { 131479f95c82SChris Mason int push_items = 0; 131579f95c82SChris Mason int max_push; 131679f95c82SChris Mason int src_nritems; 131779f95c82SChris Mason int dst_nritems; 131879f95c82SChris Mason int ret = 0; 131979f95c82SChris Mason 13207bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 13217bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 13227bb86316SChris Mason 13235f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 13245f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1325123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 1326*bce4eae9SChris Mason if (push_items <= 0) { 132779f95c82SChris Mason return 1; 1328*bce4eae9SChris Mason } 1329*bce4eae9SChris Mason 1330*bce4eae9SChris Mason if (src_nritems < 4) { 1331*bce4eae9SChris Mason return 1; 1332*bce4eae9SChris Mason } 133379f95c82SChris Mason 133479f95c82SChris Mason max_push = src_nritems / 2 + 1; 133579f95c82SChris Mason /* don't try to empty the node */ 1336*bce4eae9SChris Mason if (max_push >= src_nritems) { 133779f95c82SChris Mason return 1; 1338*bce4eae9SChris Mason } 1339252c38f0SYan 134079f95c82SChris Mason if (max_push < push_items) 134179f95c82SChris Mason push_items = max_push; 134279f95c82SChris Mason 13435f39d397SChris Mason memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items), 13445f39d397SChris Mason btrfs_node_key_ptr_offset(0), 13455f39d397SChris Mason (dst_nritems) * 13465f39d397SChris Mason sizeof(struct btrfs_key_ptr)); 1347d6025579SChris Mason 13485f39d397SChris Mason copy_extent_buffer(dst, src, 13495f39d397SChris Mason btrfs_node_key_ptr_offset(0), 13505f39d397SChris Mason btrfs_node_key_ptr_offset(src_nritems - push_items), 1351123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 135279f95c82SChris Mason 13535f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 13545f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 135579f95c82SChris Mason 13565f39d397SChris Mason btrfs_mark_buffer_dirty(src); 13575f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 135879f95c82SChris Mason return ret; 135979f95c82SChris Mason } 136079f95c82SChris Mason 136179f95c82SChris Mason /* 136297571fd0SChris Mason * helper function to insert a new root level in the tree. 136397571fd0SChris Mason * A new node is allocated, and a single item is inserted to 136497571fd0SChris Mason * point to the existing root 1365aa5d6bedSChris Mason * 1366aa5d6bedSChris Mason * returns zero on success or < 0 on failure. 136797571fd0SChris Mason */ 136898ed5174SChris Mason static int noinline insert_new_root(struct btrfs_trans_handle *trans, 13695f39d397SChris Mason struct btrfs_root *root, 13705f39d397SChris Mason struct btrfs_path *path, int level) 137174123bd7SChris Mason { 13727bb86316SChris Mason u64 root_gen; 13737bb86316SChris Mason u64 lower_gen; 13745f39d397SChris Mason struct extent_buffer *lower; 13755f39d397SChris Mason struct extent_buffer *c; 13765f39d397SChris Mason struct btrfs_disk_key lower_key; 13775c680ed6SChris Mason 13785c680ed6SChris Mason BUG_ON(path->nodes[level]); 13795c680ed6SChris Mason BUG_ON(path->nodes[level-1] != root->node); 13805c680ed6SChris Mason 13817bb86316SChris Mason if (root->ref_cows) 13827bb86316SChris Mason root_gen = trans->transid; 13837bb86316SChris Mason else 13847bb86316SChris Mason root_gen = 0; 13857bb86316SChris Mason 13867bb86316SChris Mason lower = path->nodes[level-1]; 13877bb86316SChris Mason if (level == 1) 13887bb86316SChris Mason btrfs_item_key(lower, &lower_key, 0); 13897bb86316SChris Mason else 13907bb86316SChris Mason btrfs_node_key(lower, &lower_key, 0); 13917bb86316SChris Mason 13927bb86316SChris Mason c = __btrfs_alloc_free_block(trans, root, root->nodesize, 13937bb86316SChris Mason root->root_key.objectid, 13947bb86316SChris Mason root_gen, lower_key.objectid, level, 1395db94535dSChris Mason root->node->start, 0); 13965f39d397SChris Mason if (IS_ERR(c)) 13975f39d397SChris Mason return PTR_ERR(c); 13985f39d397SChris Mason memset_extent_buffer(c, 0, 0, root->nodesize); 13995f39d397SChris Mason btrfs_set_header_nritems(c, 1); 14005f39d397SChris Mason btrfs_set_header_level(c, level); 1401db94535dSChris Mason btrfs_set_header_bytenr(c, c->start); 14025f39d397SChris Mason btrfs_set_header_generation(c, trans->transid); 14035f39d397SChris Mason btrfs_set_header_owner(c, root->root_key.objectid); 1404d5719762SChris Mason 14055f39d397SChris Mason write_extent_buffer(c, root->fs_info->fsid, 14065f39d397SChris Mason (unsigned long)btrfs_header_fsid(c), 14075f39d397SChris Mason BTRFS_FSID_SIZE); 1408e17cade2SChris Mason 1409e17cade2SChris Mason write_extent_buffer(c, root->fs_info->chunk_tree_uuid, 1410e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(c), 1411e17cade2SChris Mason BTRFS_UUID_SIZE); 1412e17cade2SChris Mason 14135f39d397SChris Mason btrfs_set_node_key(c, &lower_key, 0); 1414db94535dSChris Mason btrfs_set_node_blockptr(c, 0, lower->start); 14157bb86316SChris Mason lower_gen = btrfs_header_generation(lower); 14167bb86316SChris Mason WARN_ON(lower_gen == 0); 14177bb86316SChris Mason 14187bb86316SChris Mason btrfs_set_node_ptr_generation(c, 0, lower_gen); 14195f39d397SChris Mason 14205f39d397SChris Mason btrfs_mark_buffer_dirty(c); 1421d5719762SChris Mason 1422cfaa7295SChris Mason /* the super has an extra ref to root->node */ 14235f39d397SChris Mason free_extent_buffer(root->node); 14245f39d397SChris Mason root->node = c; 14250b86a832SChris Mason add_root_to_dirty_list(root); 14265f39d397SChris Mason extent_buffer_get(c); 14275f39d397SChris Mason path->nodes[level] = c; 142874123bd7SChris Mason path->slots[level] = 0; 14297bb86316SChris Mason 14307bb86316SChris Mason if (root->ref_cows && lower_gen != trans->transid) { 14317bb86316SChris Mason struct btrfs_path *back_path = btrfs_alloc_path(); 14327bb86316SChris Mason int ret; 14337bb86316SChris Mason ret = btrfs_insert_extent_backref(trans, 14347bb86316SChris Mason root->fs_info->extent_root, 14357bb86316SChris Mason path, lower->start, 14367bb86316SChris Mason root->root_key.objectid, 14377bb86316SChris Mason trans->transid, 0, 0); 14387bb86316SChris Mason BUG_ON(ret); 14397bb86316SChris Mason btrfs_free_path(back_path); 14407bb86316SChris Mason } 144174123bd7SChris Mason return 0; 144274123bd7SChris Mason } 14435c680ed6SChris Mason 14445c680ed6SChris Mason /* 14455c680ed6SChris Mason * worker function to insert a single pointer in a node. 14465c680ed6SChris Mason * the node should have enough room for the pointer already 144797571fd0SChris Mason * 14485c680ed6SChris Mason * slot and level indicate where you want the key to go, and 14495c680ed6SChris Mason * blocknr is the block the key points to. 1450aa5d6bedSChris Mason * 1451aa5d6bedSChris Mason * returns zero on success and < 0 on any error 14525c680ed6SChris Mason */ 1453e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root 1454e089f05cSChris Mason *root, struct btrfs_path *path, struct btrfs_disk_key 1455db94535dSChris Mason *key, u64 bytenr, int slot, int level) 14565c680ed6SChris Mason { 14575f39d397SChris Mason struct extent_buffer *lower; 14585c680ed6SChris Mason int nritems; 14595c680ed6SChris Mason 14605c680ed6SChris Mason BUG_ON(!path->nodes[level]); 14615f39d397SChris Mason lower = path->nodes[level]; 14625f39d397SChris Mason nritems = btrfs_header_nritems(lower); 146374123bd7SChris Mason if (slot > nritems) 146474123bd7SChris Mason BUG(); 1465123abc88SChris Mason if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root)) 146674123bd7SChris Mason BUG(); 146774123bd7SChris Mason if (slot != nritems) { 14685f39d397SChris Mason memmove_extent_buffer(lower, 14695f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 14705f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 1471123abc88SChris Mason (nritems - slot) * sizeof(struct btrfs_key_ptr)); 147274123bd7SChris Mason } 14735f39d397SChris Mason btrfs_set_node_key(lower, key, slot); 1474db94535dSChris Mason btrfs_set_node_blockptr(lower, slot, bytenr); 147574493f7aSChris Mason WARN_ON(trans->transid == 0); 147674493f7aSChris Mason btrfs_set_node_ptr_generation(lower, slot, trans->transid); 14775f39d397SChris Mason btrfs_set_header_nritems(lower, nritems + 1); 14785f39d397SChris Mason btrfs_mark_buffer_dirty(lower); 147974123bd7SChris Mason return 0; 148074123bd7SChris Mason } 148174123bd7SChris Mason 148297571fd0SChris Mason /* 148397571fd0SChris Mason * split the node at the specified level in path in two. 148497571fd0SChris Mason * The path is corrected to point to the appropriate node after the split 148597571fd0SChris Mason * 148697571fd0SChris Mason * Before splitting this tries to make some room in the node by pushing 148797571fd0SChris Mason * left and right, if either one works, it returns right away. 1488aa5d6bedSChris Mason * 1489aa5d6bedSChris Mason * returns 0 on success and < 0 on failure 149097571fd0SChris Mason */ 1491e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 1492e089f05cSChris Mason *root, struct btrfs_path *path, int level) 1493be0e5c09SChris Mason { 14947bb86316SChris Mason u64 root_gen; 14955f39d397SChris Mason struct extent_buffer *c; 14965f39d397SChris Mason struct extent_buffer *split; 14975f39d397SChris Mason struct btrfs_disk_key disk_key; 1498be0e5c09SChris Mason int mid; 14995c680ed6SChris Mason int ret; 1500aa5d6bedSChris Mason int wret; 15017518a238SChris Mason u32 c_nritems; 1502be0e5c09SChris Mason 15035f39d397SChris Mason c = path->nodes[level]; 15047bb86316SChris Mason WARN_ON(btrfs_header_generation(c) != trans->transid); 15055f39d397SChris Mason if (c == root->node) { 15065c680ed6SChris Mason /* trying to split the root, lets make a new one */ 1507e089f05cSChris Mason ret = insert_new_root(trans, root, path, level + 1); 15085c680ed6SChris Mason if (ret) 15095c680ed6SChris Mason return ret; 1510e66f709bSChris Mason } else { 1511e66f709bSChris Mason ret = push_nodes_for_insert(trans, root, path, level); 15125f39d397SChris Mason c = path->nodes[level]; 15135f39d397SChris Mason if (!ret && btrfs_header_nritems(c) < 1514c448acf0SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) 1515e66f709bSChris Mason return 0; 151654aa1f4dSChris Mason if (ret < 0) 151754aa1f4dSChris Mason return ret; 15185c680ed6SChris Mason } 1519e66f709bSChris Mason 15205f39d397SChris Mason c_nritems = btrfs_header_nritems(c); 15217bb86316SChris Mason if (root->ref_cows) 15227bb86316SChris Mason root_gen = trans->transid; 15237bb86316SChris Mason else 15247bb86316SChris Mason root_gen = 0; 15257bb86316SChris Mason 15267bb86316SChris Mason btrfs_node_key(c, &disk_key, 0); 15277bb86316SChris Mason split = __btrfs_alloc_free_block(trans, root, root->nodesize, 15287bb86316SChris Mason root->root_key.objectid, 15297bb86316SChris Mason root_gen, 15307bb86316SChris Mason btrfs_disk_key_objectid(&disk_key), 15317bb86316SChris Mason level, c->start, 0); 15325f39d397SChris Mason if (IS_ERR(split)) 15335f39d397SChris Mason return PTR_ERR(split); 153454aa1f4dSChris Mason 15355f39d397SChris Mason btrfs_set_header_flags(split, btrfs_header_flags(c)); 15365f39d397SChris Mason btrfs_set_header_level(split, btrfs_header_level(c)); 1537db94535dSChris Mason btrfs_set_header_bytenr(split, split->start); 15385f39d397SChris Mason btrfs_set_header_generation(split, trans->transid); 15395f39d397SChris Mason btrfs_set_header_owner(split, root->root_key.objectid); 154063b10fc4SChris Mason btrfs_set_header_flags(split, 0); 15415f39d397SChris Mason write_extent_buffer(split, root->fs_info->fsid, 15425f39d397SChris Mason (unsigned long)btrfs_header_fsid(split), 15435f39d397SChris Mason BTRFS_FSID_SIZE); 1544e17cade2SChris Mason write_extent_buffer(split, root->fs_info->chunk_tree_uuid, 1545e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(split), 1546e17cade2SChris Mason BTRFS_UUID_SIZE); 15475f39d397SChris Mason 15487518a238SChris Mason mid = (c_nritems + 1) / 2; 15495f39d397SChris Mason 15505f39d397SChris Mason copy_extent_buffer(split, c, 15515f39d397SChris Mason btrfs_node_key_ptr_offset(0), 15525f39d397SChris Mason btrfs_node_key_ptr_offset(mid), 1553123abc88SChris Mason (c_nritems - mid) * sizeof(struct btrfs_key_ptr)); 15545f39d397SChris Mason btrfs_set_header_nritems(split, c_nritems - mid); 15555f39d397SChris Mason btrfs_set_header_nritems(c, mid); 1556aa5d6bedSChris Mason ret = 0; 1557aa5d6bedSChris Mason 15585f39d397SChris Mason btrfs_mark_buffer_dirty(c); 15595f39d397SChris Mason btrfs_mark_buffer_dirty(split); 15605f39d397SChris Mason 15615f39d397SChris Mason btrfs_node_key(split, &disk_key, 0); 1562db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, split->start, 15635f39d397SChris Mason path->slots[level + 1] + 1, 1564123abc88SChris Mason level + 1); 1565aa5d6bedSChris Mason if (wret) 1566aa5d6bedSChris Mason ret = wret; 1567aa5d6bedSChris Mason 15685de08d7dSChris Mason if (path->slots[level] >= mid) { 15695c680ed6SChris Mason path->slots[level] -= mid; 15705f39d397SChris Mason free_extent_buffer(c); 15715f39d397SChris Mason path->nodes[level] = split; 15725c680ed6SChris Mason path->slots[level + 1] += 1; 1573eb60ceacSChris Mason } else { 15745f39d397SChris Mason free_extent_buffer(split); 1575be0e5c09SChris Mason } 1576aa5d6bedSChris Mason return ret; 1577be0e5c09SChris Mason } 1578be0e5c09SChris Mason 157974123bd7SChris Mason /* 158074123bd7SChris Mason * how many bytes are required to store the items in a leaf. start 158174123bd7SChris Mason * and nr indicate which items in the leaf to check. This totals up the 158274123bd7SChris Mason * space used both by the item structs and the item data 158374123bd7SChris Mason */ 15845f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr) 1585be0e5c09SChris Mason { 1586be0e5c09SChris Mason int data_len; 15875f39d397SChris Mason int nritems = btrfs_header_nritems(l); 1588d4dbff95SChris Mason int end = min(nritems, start + nr) - 1; 1589be0e5c09SChris Mason 1590be0e5c09SChris Mason if (!nr) 1591be0e5c09SChris Mason return 0; 15925f39d397SChris Mason data_len = btrfs_item_end_nr(l, start); 15935f39d397SChris Mason data_len = data_len - btrfs_item_offset_nr(l, end); 15940783fcfcSChris Mason data_len += sizeof(struct btrfs_item) * nr; 1595d4dbff95SChris Mason WARN_ON(data_len < 0); 1596be0e5c09SChris Mason return data_len; 1597be0e5c09SChris Mason } 1598be0e5c09SChris Mason 159974123bd7SChris Mason /* 1600d4dbff95SChris Mason * The space between the end of the leaf items and 1601d4dbff95SChris Mason * the start of the leaf data. IOW, how much room 1602d4dbff95SChris Mason * the leaf has left for both items and data 1603d4dbff95SChris Mason */ 16045f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf) 1605d4dbff95SChris Mason { 16065f39d397SChris Mason int nritems = btrfs_header_nritems(leaf); 16075f39d397SChris Mason int ret; 16085f39d397SChris Mason ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems); 16095f39d397SChris Mason if (ret < 0) { 16105f39d397SChris Mason printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n", 1611ae2f5411SJens Axboe ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root), 16125f39d397SChris Mason leaf_space_used(leaf, 0, nritems), nritems); 16135f39d397SChris Mason } 16145f39d397SChris Mason return ret; 1615d4dbff95SChris Mason } 1616d4dbff95SChris Mason 1617d4dbff95SChris Mason /* 161800ec4c51SChris Mason * push some data in the path leaf to the right, trying to free up at 161900ec4c51SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 1620aa5d6bedSChris Mason * 1621aa5d6bedSChris Mason * returns 1 if the push failed because the other node didn't have enough 1622aa5d6bedSChris Mason * room, 0 if everything worked out and < 0 if there were major errors. 162300ec4c51SChris Mason */ 1624e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root 162534a38218SChris Mason *root, struct btrfs_path *path, int data_size, 162634a38218SChris Mason int empty) 162700ec4c51SChris Mason { 16285f39d397SChris Mason struct extent_buffer *left = path->nodes[0]; 16295f39d397SChris Mason struct extent_buffer *right; 16305f39d397SChris Mason struct extent_buffer *upper; 16315f39d397SChris Mason struct btrfs_disk_key disk_key; 163200ec4c51SChris Mason int slot; 163334a38218SChris Mason u32 i; 163400ec4c51SChris Mason int free_space; 163500ec4c51SChris Mason int push_space = 0; 163600ec4c51SChris Mason int push_items = 0; 16370783fcfcSChris Mason struct btrfs_item *item; 16387518a238SChris Mason u32 left_nritems; 163934a38218SChris Mason u32 nr; 16407518a238SChris Mason u32 right_nritems; 16415f39d397SChris Mason u32 data_end; 1642db94535dSChris Mason u32 this_item_size; 164354aa1f4dSChris Mason int ret; 164400ec4c51SChris Mason 164500ec4c51SChris Mason slot = path->slots[1]; 164600ec4c51SChris Mason if (!path->nodes[1]) { 164700ec4c51SChris Mason return 1; 164800ec4c51SChris Mason } 164900ec4c51SChris Mason upper = path->nodes[1]; 16505f39d397SChris Mason if (slot >= btrfs_header_nritems(upper) - 1) 165100ec4c51SChris Mason return 1; 16525f39d397SChris Mason 1653db94535dSChris Mason right = read_tree_block(root, btrfs_node_blockptr(upper, slot + 1), 1654db94535dSChris Mason root->leafsize); 1655123abc88SChris Mason free_space = btrfs_leaf_free_space(root, right); 16560783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 16575f39d397SChris Mason free_extent_buffer(right); 165802217ed2SChris Mason return 1; 165902217ed2SChris Mason } 166002217ed2SChris Mason 16615f39d397SChris Mason /* cow and double check */ 16625f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, upper, 16635f39d397SChris Mason slot + 1, &right); 16645f39d397SChris Mason if (ret) { 16655f39d397SChris Mason free_extent_buffer(right); 1666a429e513SChris Mason return 1; 1667a429e513SChris Mason } 16685f39d397SChris Mason free_space = btrfs_leaf_free_space(root, right); 16695f39d397SChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 16705f39d397SChris Mason free_extent_buffer(right); 16715f39d397SChris Mason return 1; 16725f39d397SChris Mason } 16735f39d397SChris Mason 16745f39d397SChris Mason left_nritems = btrfs_header_nritems(left); 16755f39d397SChris Mason if (left_nritems == 0) { 16765f39d397SChris Mason free_extent_buffer(right); 16775f39d397SChris Mason return 1; 16785f39d397SChris Mason } 16795f39d397SChris Mason 168034a38218SChris Mason if (empty) 168134a38218SChris Mason nr = 0; 168234a38218SChris Mason else 168334a38218SChris Mason nr = 1; 168434a38218SChris Mason 168534a38218SChris Mason i = left_nritems - 1; 168634a38218SChris Mason while (i >= nr) { 16875f39d397SChris Mason item = btrfs_item_nr(left, i); 1688db94535dSChris Mason 168900ec4c51SChris Mason if (path->slots[0] == i) 169000ec4c51SChris Mason push_space += data_size + sizeof(*item); 1691db94535dSChris Mason 1692db94535dSChris Mason if (!left->map_token) { 1693db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 1694db94535dSChris Mason sizeof(struct btrfs_item), 1695db94535dSChris Mason &left->map_token, &left->kaddr, 1696db94535dSChris Mason &left->map_start, &left->map_len, 1697db94535dSChris Mason KM_USER1); 1698db94535dSChris Mason } 1699db94535dSChris Mason 1700db94535dSChris Mason this_item_size = btrfs_item_size(left, item); 1701db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 170200ec4c51SChris Mason break; 170300ec4c51SChris Mason push_items++; 1704db94535dSChris Mason push_space += this_item_size + sizeof(*item); 170534a38218SChris Mason if (i == 0) 170634a38218SChris Mason break; 170734a38218SChris Mason i--; 1708db94535dSChris Mason } 1709db94535dSChris Mason if (left->map_token) { 1710db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 1711db94535dSChris Mason left->map_token = NULL; 171200ec4c51SChris Mason } 17135f39d397SChris Mason 171400ec4c51SChris Mason if (push_items == 0) { 17155f39d397SChris Mason free_extent_buffer(right); 171600ec4c51SChris Mason return 1; 171700ec4c51SChris Mason } 17185f39d397SChris Mason 171934a38218SChris Mason if (!empty && push_items == left_nritems) 1720a429e513SChris Mason WARN_ON(1); 17215f39d397SChris Mason 172200ec4c51SChris Mason /* push left to right */ 17235f39d397SChris Mason right_nritems = btrfs_header_nritems(right); 172434a38218SChris Mason 17255f39d397SChris Mason push_space = btrfs_item_end_nr(left, left_nritems - push_items); 1726123abc88SChris Mason push_space -= leaf_data_end(root, left); 17275f39d397SChris Mason 172800ec4c51SChris Mason /* make room in the right data area */ 17295f39d397SChris Mason data_end = leaf_data_end(root, right); 17305f39d397SChris Mason memmove_extent_buffer(right, 17315f39d397SChris Mason btrfs_leaf_data(right) + data_end - push_space, 17325f39d397SChris Mason btrfs_leaf_data(right) + data_end, 17335f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root) - data_end); 17345f39d397SChris Mason 173500ec4c51SChris Mason /* copy from the left data area */ 17365f39d397SChris Mason copy_extent_buffer(right, left, btrfs_leaf_data(right) + 1737d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 1738d6025579SChris Mason btrfs_leaf_data(left) + leaf_data_end(root, left), 1739d6025579SChris Mason push_space); 17405f39d397SChris Mason 17415f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(push_items), 17425f39d397SChris Mason btrfs_item_nr_offset(0), 17430783fcfcSChris Mason right_nritems * sizeof(struct btrfs_item)); 17445f39d397SChris Mason 174500ec4c51SChris Mason /* copy the items from left to right */ 17465f39d397SChris Mason copy_extent_buffer(right, left, btrfs_item_nr_offset(0), 17475f39d397SChris Mason btrfs_item_nr_offset(left_nritems - push_items), 17480783fcfcSChris Mason push_items * sizeof(struct btrfs_item)); 174900ec4c51SChris Mason 175000ec4c51SChris Mason /* update the item pointers */ 17517518a238SChris Mason right_nritems += push_items; 17525f39d397SChris Mason btrfs_set_header_nritems(right, right_nritems); 1753123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 17547518a238SChris Mason for (i = 0; i < right_nritems; i++) { 17555f39d397SChris Mason item = btrfs_item_nr(right, i); 1756db94535dSChris Mason if (!right->map_token) { 1757db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1758db94535dSChris Mason sizeof(struct btrfs_item), 1759db94535dSChris Mason &right->map_token, &right->kaddr, 1760db94535dSChris Mason &right->map_start, &right->map_len, 1761db94535dSChris Mason KM_USER1); 1762db94535dSChris Mason } 1763db94535dSChris Mason push_space -= btrfs_item_size(right, item); 1764db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 1765db94535dSChris Mason } 1766db94535dSChris Mason 1767db94535dSChris Mason if (right->map_token) { 1768db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1769db94535dSChris Mason right->map_token = NULL; 177000ec4c51SChris Mason } 17717518a238SChris Mason left_nritems -= push_items; 17725f39d397SChris Mason btrfs_set_header_nritems(left, left_nritems); 177300ec4c51SChris Mason 177434a38218SChris Mason if (left_nritems) 17755f39d397SChris Mason btrfs_mark_buffer_dirty(left); 17765f39d397SChris Mason btrfs_mark_buffer_dirty(right); 1777a429e513SChris Mason 17785f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 17795f39d397SChris Mason btrfs_set_node_key(upper, &disk_key, slot + 1); 1780d6025579SChris Mason btrfs_mark_buffer_dirty(upper); 178102217ed2SChris Mason 178200ec4c51SChris Mason /* then fixup the leaf pointer in the path */ 17837518a238SChris Mason if (path->slots[0] >= left_nritems) { 17847518a238SChris Mason path->slots[0] -= left_nritems; 17855f39d397SChris Mason free_extent_buffer(path->nodes[0]); 17865f39d397SChris Mason path->nodes[0] = right; 178700ec4c51SChris Mason path->slots[1] += 1; 178800ec4c51SChris Mason } else { 17895f39d397SChris Mason free_extent_buffer(right); 179000ec4c51SChris Mason } 179100ec4c51SChris Mason return 0; 179200ec4c51SChris Mason } 179300ec4c51SChris Mason /* 179474123bd7SChris Mason * push some data in the path leaf to the left, trying to free up at 179574123bd7SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 179674123bd7SChris Mason */ 1797e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root 179834a38218SChris Mason *root, struct btrfs_path *path, int data_size, 179934a38218SChris Mason int empty) 1800be0e5c09SChris Mason { 18015f39d397SChris Mason struct btrfs_disk_key disk_key; 18025f39d397SChris Mason struct extent_buffer *right = path->nodes[0]; 18035f39d397SChris Mason struct extent_buffer *left; 1804be0e5c09SChris Mason int slot; 1805be0e5c09SChris Mason int i; 1806be0e5c09SChris Mason int free_space; 1807be0e5c09SChris Mason int push_space = 0; 1808be0e5c09SChris Mason int push_items = 0; 18090783fcfcSChris Mason struct btrfs_item *item; 18107518a238SChris Mason u32 old_left_nritems; 18115f39d397SChris Mason u32 right_nritems; 181234a38218SChris Mason u32 nr; 1813aa5d6bedSChris Mason int ret = 0; 1814aa5d6bedSChris Mason int wret; 1815db94535dSChris Mason u32 this_item_size; 1816db94535dSChris Mason u32 old_left_item_size; 1817be0e5c09SChris Mason 1818be0e5c09SChris Mason slot = path->slots[1]; 18195f39d397SChris Mason if (slot == 0) 1820be0e5c09SChris Mason return 1; 18215f39d397SChris Mason if (!path->nodes[1]) 1822be0e5c09SChris Mason return 1; 18235f39d397SChris Mason 18243685f791SChris Mason right_nritems = btrfs_header_nritems(right); 18253685f791SChris Mason if (right_nritems == 0) { 18263685f791SChris Mason return 1; 18273685f791SChris Mason } 18283685f791SChris Mason 18295f39d397SChris Mason left = read_tree_block(root, btrfs_node_blockptr(path->nodes[1], 1830db94535dSChris Mason slot - 1), root->leafsize); 1831123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 18320783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 18335f39d397SChris Mason free_extent_buffer(left); 1834be0e5c09SChris Mason return 1; 1835be0e5c09SChris Mason } 183602217ed2SChris Mason 183702217ed2SChris Mason /* cow and double check */ 18385f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, 18395f39d397SChris Mason path->nodes[1], slot - 1, &left); 184054aa1f4dSChris Mason if (ret) { 184154aa1f4dSChris Mason /* we hit -ENOSPC, but it isn't fatal here */ 18425f39d397SChris Mason free_extent_buffer(left); 184354aa1f4dSChris Mason return 1; 184454aa1f4dSChris Mason } 18453685f791SChris Mason 1846123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 18470783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 18485f39d397SChris Mason free_extent_buffer(left); 184902217ed2SChris Mason return 1; 185002217ed2SChris Mason } 185102217ed2SChris Mason 185234a38218SChris Mason if (empty) 185334a38218SChris Mason nr = right_nritems; 185434a38218SChris Mason else 185534a38218SChris Mason nr = right_nritems - 1; 185634a38218SChris Mason 185734a38218SChris Mason for (i = 0; i < nr; i++) { 18585f39d397SChris Mason item = btrfs_item_nr(right, i); 1859db94535dSChris Mason if (!right->map_token) { 1860db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1861db94535dSChris Mason sizeof(struct btrfs_item), 1862db94535dSChris Mason &right->map_token, &right->kaddr, 1863db94535dSChris Mason &right->map_start, &right->map_len, 1864db94535dSChris Mason KM_USER1); 1865db94535dSChris Mason } 1866db94535dSChris Mason 1867be0e5c09SChris Mason if (path->slots[0] == i) 1868be0e5c09SChris Mason push_space += data_size + sizeof(*item); 1869db94535dSChris Mason 1870db94535dSChris Mason this_item_size = btrfs_item_size(right, item); 1871db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 1872be0e5c09SChris Mason break; 1873db94535dSChris Mason 1874be0e5c09SChris Mason push_items++; 1875db94535dSChris Mason push_space += this_item_size + sizeof(*item); 1876be0e5c09SChris Mason } 1877db94535dSChris Mason 1878db94535dSChris Mason if (right->map_token) { 1879db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1880db94535dSChris Mason right->map_token = NULL; 1881db94535dSChris Mason } 1882db94535dSChris Mason 1883be0e5c09SChris Mason if (push_items == 0) { 18845f39d397SChris Mason free_extent_buffer(left); 1885be0e5c09SChris Mason return 1; 1886be0e5c09SChris Mason } 188734a38218SChris Mason if (!empty && push_items == btrfs_header_nritems(right)) 1888a429e513SChris Mason WARN_ON(1); 18895f39d397SChris Mason 1890be0e5c09SChris Mason /* push data from right to left */ 18915f39d397SChris Mason copy_extent_buffer(left, right, 18925f39d397SChris Mason btrfs_item_nr_offset(btrfs_header_nritems(left)), 18935f39d397SChris Mason btrfs_item_nr_offset(0), 18945f39d397SChris Mason push_items * sizeof(struct btrfs_item)); 18955f39d397SChris Mason 1896123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root) - 18975f39d397SChris Mason btrfs_item_offset_nr(right, push_items -1); 18985f39d397SChris Mason 18995f39d397SChris Mason copy_extent_buffer(left, right, btrfs_leaf_data(left) + 1900d6025579SChris Mason leaf_data_end(root, left) - push_space, 1901123abc88SChris Mason btrfs_leaf_data(right) + 19025f39d397SChris Mason btrfs_item_offset_nr(right, push_items - 1), 1903be0e5c09SChris Mason push_space); 19045f39d397SChris Mason old_left_nritems = btrfs_header_nritems(left); 1905eb60ceacSChris Mason BUG_ON(old_left_nritems < 0); 1906eb60ceacSChris Mason 1907db94535dSChris Mason old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1); 1908be0e5c09SChris Mason for (i = old_left_nritems; i < old_left_nritems + push_items; i++) { 19095f39d397SChris Mason u32 ioff; 1910db94535dSChris Mason 19115f39d397SChris Mason item = btrfs_item_nr(left, i); 1912db94535dSChris Mason if (!left->map_token) { 1913db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 1914db94535dSChris Mason sizeof(struct btrfs_item), 1915db94535dSChris Mason &left->map_token, &left->kaddr, 1916db94535dSChris Mason &left->map_start, &left->map_len, 1917db94535dSChris Mason KM_USER1); 1918db94535dSChris Mason } 1919db94535dSChris Mason 19205f39d397SChris Mason ioff = btrfs_item_offset(left, item); 19215f39d397SChris Mason btrfs_set_item_offset(left, item, 1922db94535dSChris Mason ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size)); 1923be0e5c09SChris Mason } 19245f39d397SChris Mason btrfs_set_header_nritems(left, old_left_nritems + push_items); 1925db94535dSChris Mason if (left->map_token) { 1926db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 1927db94535dSChris Mason left->map_token = NULL; 1928db94535dSChris Mason } 1929be0e5c09SChris Mason 1930be0e5c09SChris Mason /* fixup right node */ 193134a38218SChris Mason if (push_items > right_nritems) { 193234a38218SChris Mason printk("push items %d nr %u\n", push_items, right_nritems); 193334a38218SChris Mason WARN_ON(1); 193434a38218SChris Mason } 193534a38218SChris Mason 193634a38218SChris Mason if (push_items < right_nritems) { 19375f39d397SChris Mason push_space = btrfs_item_offset_nr(right, push_items - 1) - 1938123abc88SChris Mason leaf_data_end(root, right); 19395f39d397SChris Mason memmove_extent_buffer(right, btrfs_leaf_data(right) + 1940d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 1941d6025579SChris Mason btrfs_leaf_data(right) + 1942123abc88SChris Mason leaf_data_end(root, right), push_space); 19435f39d397SChris Mason 19445f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(0), 19455f39d397SChris Mason btrfs_item_nr_offset(push_items), 19465f39d397SChris Mason (btrfs_header_nritems(right) - push_items) * 19470783fcfcSChris Mason sizeof(struct btrfs_item)); 194834a38218SChris Mason } 1949eef1c494SYan right_nritems -= push_items; 1950eef1c494SYan btrfs_set_header_nritems(right, right_nritems); 1951123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 19525f39d397SChris Mason for (i = 0; i < right_nritems; i++) { 19535f39d397SChris Mason item = btrfs_item_nr(right, i); 1954db94535dSChris Mason 1955db94535dSChris Mason if (!right->map_token) { 1956db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1957db94535dSChris Mason sizeof(struct btrfs_item), 1958db94535dSChris Mason &right->map_token, &right->kaddr, 1959db94535dSChris Mason &right->map_start, &right->map_len, 1960db94535dSChris Mason KM_USER1); 1961db94535dSChris Mason } 1962db94535dSChris Mason 1963db94535dSChris Mason push_space = push_space - btrfs_item_size(right, item); 1964db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 1965db94535dSChris Mason } 1966db94535dSChris Mason if (right->map_token) { 1967db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1968db94535dSChris Mason right->map_token = NULL; 1969be0e5c09SChris Mason } 1970eb60ceacSChris Mason 19715f39d397SChris Mason btrfs_mark_buffer_dirty(left); 197234a38218SChris Mason if (right_nritems) 19735f39d397SChris Mason btrfs_mark_buffer_dirty(right); 1974098f59c2SChris Mason 19755f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 19765f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, 1); 1977aa5d6bedSChris Mason if (wret) 1978aa5d6bedSChris Mason ret = wret; 1979be0e5c09SChris Mason 1980be0e5c09SChris Mason /* then fixup the leaf pointer in the path */ 1981be0e5c09SChris Mason if (path->slots[0] < push_items) { 1982be0e5c09SChris Mason path->slots[0] += old_left_nritems; 19835f39d397SChris Mason free_extent_buffer(path->nodes[0]); 19845f39d397SChris Mason path->nodes[0] = left; 1985be0e5c09SChris Mason path->slots[1] -= 1; 1986be0e5c09SChris Mason } else { 19875f39d397SChris Mason free_extent_buffer(left); 1988be0e5c09SChris Mason path->slots[0] -= push_items; 1989be0e5c09SChris Mason } 1990eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 1991aa5d6bedSChris Mason return ret; 1992be0e5c09SChris Mason } 1993be0e5c09SChris Mason 199474123bd7SChris Mason /* 199574123bd7SChris Mason * split the path's leaf in two, making sure there is at least data_size 199674123bd7SChris Mason * available for the resulting leaf level of the path. 1997aa5d6bedSChris Mason * 1998aa5d6bedSChris Mason * returns 0 if all went well and < 0 on failure. 199974123bd7SChris Mason */ 2000e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 2001d4dbff95SChris Mason *root, struct btrfs_key *ins_key, 2002cc0c5538SChris Mason struct btrfs_path *path, int data_size, int extend) 2003be0e5c09SChris Mason { 20047bb86316SChris Mason u64 root_gen; 20055f39d397SChris Mason struct extent_buffer *l; 20067518a238SChris Mason u32 nritems; 2007eb60ceacSChris Mason int mid; 2008eb60ceacSChris Mason int slot; 20095f39d397SChris Mason struct extent_buffer *right; 20100783fcfcSChris Mason int space_needed = data_size + sizeof(struct btrfs_item); 2011be0e5c09SChris Mason int data_copy_size; 2012be0e5c09SChris Mason int rt_data_off; 2013be0e5c09SChris Mason int i; 2014d4dbff95SChris Mason int ret = 0; 2015aa5d6bedSChris Mason int wret; 2016cc0c5538SChris Mason int double_split; 2017cc0c5538SChris Mason int num_doubles = 0; 2018d4dbff95SChris Mason struct btrfs_disk_key disk_key; 2019be0e5c09SChris Mason 2020cc0c5538SChris Mason if (extend) 2021cc0c5538SChris Mason space_needed = data_size; 2022cc0c5538SChris Mason 20237bb86316SChris Mason if (root->ref_cows) 20247bb86316SChris Mason root_gen = trans->transid; 20257bb86316SChris Mason else 20267bb86316SChris Mason root_gen = 0; 20277bb86316SChris Mason 202840689478SChris Mason /* first try to make some room by pushing left and right */ 20293685f791SChris Mason if (ins_key->type != BTRFS_DIR_ITEM_KEY) { 203034a38218SChris Mason wret = push_leaf_right(trans, root, path, data_size, 0); 20313326d1b0SChris Mason if (wret < 0) { 2032eaee50e8SChris Mason return wret; 20333326d1b0SChris Mason } 2034eaee50e8SChris Mason if (wret) { 203534a38218SChris Mason wret = push_leaf_left(trans, root, path, data_size, 0); 2036eaee50e8SChris Mason if (wret < 0) 2037eaee50e8SChris Mason return wret; 2038eaee50e8SChris Mason } 20395f39d397SChris Mason l = path->nodes[0]; 2040aa5d6bedSChris Mason 2041aa5d6bedSChris Mason /* did the pushes work? */ 2042cc0c5538SChris Mason if (btrfs_leaf_free_space(root, l) >= space_needed) 2043be0e5c09SChris Mason return 0; 20443326d1b0SChris Mason } 2045aa5d6bedSChris Mason 20465c680ed6SChris Mason if (!path->nodes[1]) { 2047e089f05cSChris Mason ret = insert_new_root(trans, root, path, 1); 20485c680ed6SChris Mason if (ret) 20495c680ed6SChris Mason return ret; 20505c680ed6SChris Mason } 2051cc0c5538SChris Mason again: 2052cc0c5538SChris Mason double_split = 0; 2053cc0c5538SChris Mason l = path->nodes[0]; 2054eb60ceacSChris Mason slot = path->slots[0]; 20555f39d397SChris Mason nritems = btrfs_header_nritems(l); 2056eb60ceacSChris Mason mid = (nritems + 1)/ 2; 205754aa1f4dSChris Mason 20587bb86316SChris Mason btrfs_item_key(l, &disk_key, 0); 20597bb86316SChris Mason 20607bb86316SChris Mason right = __btrfs_alloc_free_block(trans, root, root->leafsize, 20617bb86316SChris Mason root->root_key.objectid, 20627bb86316SChris Mason root_gen, disk_key.objectid, 0, 2063db94535dSChris Mason l->start, 0); 2064cea9e445SChris Mason if (IS_ERR(right)) { 2065cea9e445SChris Mason BUG_ON(1); 20665f39d397SChris Mason return PTR_ERR(right); 2067cea9e445SChris Mason } 206854aa1f4dSChris Mason 20695f39d397SChris Mason memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header)); 2070db94535dSChris Mason btrfs_set_header_bytenr(right, right->start); 20715f39d397SChris Mason btrfs_set_header_generation(right, trans->transid); 20725f39d397SChris Mason btrfs_set_header_owner(right, root->root_key.objectid); 20735f39d397SChris Mason btrfs_set_header_level(right, 0); 20745f39d397SChris Mason write_extent_buffer(right, root->fs_info->fsid, 20755f39d397SChris Mason (unsigned long)btrfs_header_fsid(right), 20765f39d397SChris Mason BTRFS_FSID_SIZE); 2077e17cade2SChris Mason 2078e17cade2SChris Mason write_extent_buffer(right, root->fs_info->chunk_tree_uuid, 2079e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(right), 2080e17cade2SChris Mason BTRFS_UUID_SIZE); 2081d4dbff95SChris Mason if (mid <= slot) { 2082d4dbff95SChris Mason if (nritems == 1 || 2083d4dbff95SChris Mason leaf_space_used(l, mid, nritems - mid) + space_needed > 2084d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 2085d4dbff95SChris Mason if (slot >= nritems) { 2086d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 20875f39d397SChris Mason btrfs_set_header_nritems(right, 0); 2088d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 2089db94535dSChris Mason &disk_key, right->start, 2090d4dbff95SChris Mason path->slots[1] + 1, 1); 2091d4dbff95SChris Mason if (wret) 2092d4dbff95SChris Mason ret = wret; 20935f39d397SChris Mason free_extent_buffer(path->nodes[0]); 20945f39d397SChris Mason path->nodes[0] = right; 2095d4dbff95SChris Mason path->slots[0] = 0; 2096d4dbff95SChris Mason path->slots[1] += 1; 20970ef8b242SChris Mason btrfs_mark_buffer_dirty(right); 2098d4dbff95SChris Mason return ret; 2099d4dbff95SChris Mason } 2100d4dbff95SChris Mason mid = slot; 21013326d1b0SChris Mason if (mid != nritems && 21023326d1b0SChris Mason leaf_space_used(l, mid, nritems - mid) + 21033326d1b0SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 2104d4dbff95SChris Mason double_split = 1; 2105d4dbff95SChris Mason } 21063326d1b0SChris Mason } 2107d4dbff95SChris Mason } else { 2108d4dbff95SChris Mason if (leaf_space_used(l, 0, mid + 1) + space_needed > 2109d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 2110cc0c5538SChris Mason if (!extend && slot == 0) { 2111d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 21125f39d397SChris Mason btrfs_set_header_nritems(right, 0); 2113d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 2114d4dbff95SChris Mason &disk_key, 2115db94535dSChris Mason right->start, 2116098f59c2SChris Mason path->slots[1], 1); 2117d4dbff95SChris Mason if (wret) 2118d4dbff95SChris Mason ret = wret; 21195f39d397SChris Mason free_extent_buffer(path->nodes[0]); 21205f39d397SChris Mason path->nodes[0] = right; 2121d4dbff95SChris Mason path->slots[0] = 0; 2122a429e513SChris Mason if (path->slots[1] == 0) { 2123a429e513SChris Mason wret = fixup_low_keys(trans, root, 2124a429e513SChris Mason path, &disk_key, 1); 2125a429e513SChris Mason if (wret) 2126a429e513SChris Mason ret = wret; 2127a429e513SChris Mason } 21280ef8b242SChris Mason btrfs_mark_buffer_dirty(right); 2129d4dbff95SChris Mason return ret; 2130cc0c5538SChris Mason } else if (extend && slot == 0) { 2131cc0c5538SChris Mason mid = 1; 2132cc0c5538SChris Mason } else { 2133d4dbff95SChris Mason mid = slot; 21345ee78ac7SChris Mason if (mid != nritems && 21355ee78ac7SChris Mason leaf_space_used(l, mid, nritems - mid) + 21365ee78ac7SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 2137d4dbff95SChris Mason double_split = 1; 2138d4dbff95SChris Mason } 2139d4dbff95SChris Mason } 21405ee78ac7SChris Mason } 2141cc0c5538SChris Mason } 21425f39d397SChris Mason nritems = nritems - mid; 21435f39d397SChris Mason btrfs_set_header_nritems(right, nritems); 21445f39d397SChris Mason data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l); 21455f39d397SChris Mason 21465f39d397SChris Mason copy_extent_buffer(right, l, btrfs_item_nr_offset(0), 21475f39d397SChris Mason btrfs_item_nr_offset(mid), 21485f39d397SChris Mason nritems * sizeof(struct btrfs_item)); 21495f39d397SChris Mason 21505f39d397SChris Mason copy_extent_buffer(right, l, 2151d6025579SChris Mason btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - 2152123abc88SChris Mason data_copy_size, btrfs_leaf_data(l) + 2153123abc88SChris Mason leaf_data_end(root, l), data_copy_size); 215474123bd7SChris Mason 21555f39d397SChris Mason rt_data_off = BTRFS_LEAF_DATA_SIZE(root) - 21565f39d397SChris Mason btrfs_item_end_nr(l, mid); 21575f39d397SChris Mason 21585f39d397SChris Mason for (i = 0; i < nritems; i++) { 21595f39d397SChris Mason struct btrfs_item *item = btrfs_item_nr(right, i); 2160db94535dSChris Mason u32 ioff; 2161db94535dSChris Mason 2162db94535dSChris Mason if (!right->map_token) { 2163db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2164db94535dSChris Mason sizeof(struct btrfs_item), 2165db94535dSChris Mason &right->map_token, &right->kaddr, 2166db94535dSChris Mason &right->map_start, &right->map_len, 2167db94535dSChris Mason KM_USER1); 2168db94535dSChris Mason } 2169db94535dSChris Mason 2170db94535dSChris Mason ioff = btrfs_item_offset(right, item); 21715f39d397SChris Mason btrfs_set_item_offset(right, item, ioff + rt_data_off); 21720783fcfcSChris Mason } 217374123bd7SChris Mason 2174db94535dSChris Mason if (right->map_token) { 2175db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2176db94535dSChris Mason right->map_token = NULL; 2177db94535dSChris Mason } 2178db94535dSChris Mason 21795f39d397SChris Mason btrfs_set_header_nritems(l, mid); 2180aa5d6bedSChris Mason ret = 0; 21815f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 2182db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, right->start, 2183db94535dSChris Mason path->slots[1] + 1, 1); 2184aa5d6bedSChris Mason if (wret) 2185aa5d6bedSChris Mason ret = wret; 21865f39d397SChris Mason 21875f39d397SChris Mason btrfs_mark_buffer_dirty(right); 21885f39d397SChris Mason btrfs_mark_buffer_dirty(l); 2189eb60ceacSChris Mason BUG_ON(path->slots[0] != slot); 21905f39d397SChris Mason 2191be0e5c09SChris Mason if (mid <= slot) { 21925f39d397SChris Mason free_extent_buffer(path->nodes[0]); 21935f39d397SChris Mason path->nodes[0] = right; 2194be0e5c09SChris Mason path->slots[0] -= mid; 2195be0e5c09SChris Mason path->slots[1] += 1; 2196eb60ceacSChris Mason } else 21975f39d397SChris Mason free_extent_buffer(right); 21985f39d397SChris Mason 2199eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 2200d4dbff95SChris Mason 2201cc0c5538SChris Mason if (double_split) { 2202cc0c5538SChris Mason BUG_ON(num_doubles != 0); 2203cc0c5538SChris Mason num_doubles++; 2204cc0c5538SChris Mason goto again; 22053326d1b0SChris Mason } 2206be0e5c09SChris Mason return ret; 2207be0e5c09SChris Mason } 2208be0e5c09SChris Mason 2209b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans, 2210b18c6685SChris Mason struct btrfs_root *root, 2211b18c6685SChris Mason struct btrfs_path *path, 2212179e29e4SChris Mason u32 new_size, int from_end) 2213b18c6685SChris Mason { 2214b18c6685SChris Mason int ret = 0; 2215b18c6685SChris Mason int slot; 2216b18c6685SChris Mason int slot_orig; 22175f39d397SChris Mason struct extent_buffer *leaf; 22185f39d397SChris Mason struct btrfs_item *item; 2219b18c6685SChris Mason u32 nritems; 2220b18c6685SChris Mason unsigned int data_end; 2221b18c6685SChris Mason unsigned int old_data_start; 2222b18c6685SChris Mason unsigned int old_size; 2223b18c6685SChris Mason unsigned int size_diff; 2224b18c6685SChris Mason int i; 2225b18c6685SChris Mason 2226b18c6685SChris Mason slot_orig = path->slots[0]; 22275f39d397SChris Mason leaf = path->nodes[0]; 2228179e29e4SChris Mason slot = path->slots[0]; 2229179e29e4SChris Mason 2230179e29e4SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 2231179e29e4SChris Mason if (old_size == new_size) 2232179e29e4SChris Mason return 0; 2233b18c6685SChris Mason 22345f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2235b18c6685SChris Mason data_end = leaf_data_end(root, leaf); 2236b18c6685SChris Mason 22375f39d397SChris Mason old_data_start = btrfs_item_offset_nr(leaf, slot); 2238179e29e4SChris Mason 2239b18c6685SChris Mason size_diff = old_size - new_size; 2240b18c6685SChris Mason 2241b18c6685SChris Mason BUG_ON(slot < 0); 2242b18c6685SChris Mason BUG_ON(slot >= nritems); 2243b18c6685SChris Mason 2244b18c6685SChris Mason /* 2245b18c6685SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2246b18c6685SChris Mason */ 2247b18c6685SChris Mason /* first correct the data pointers */ 2248b18c6685SChris Mason for (i = slot; i < nritems; i++) { 22495f39d397SChris Mason u32 ioff; 22505f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2251db94535dSChris Mason 2252db94535dSChris Mason if (!leaf->map_token) { 2253db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2254db94535dSChris Mason sizeof(struct btrfs_item), 2255db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2256db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2257db94535dSChris Mason KM_USER1); 2258db94535dSChris Mason } 2259db94535dSChris Mason 22605f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 22615f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + size_diff); 2262b18c6685SChris Mason } 2263db94535dSChris Mason 2264db94535dSChris Mason if (leaf->map_token) { 2265db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2266db94535dSChris Mason leaf->map_token = NULL; 2267db94535dSChris Mason } 2268db94535dSChris Mason 2269b18c6685SChris Mason /* shift the data */ 2270179e29e4SChris Mason if (from_end) { 22715f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2272b18c6685SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 2273b18c6685SChris Mason data_end, old_data_start + new_size - data_end); 2274179e29e4SChris Mason } else { 2275179e29e4SChris Mason struct btrfs_disk_key disk_key; 2276179e29e4SChris Mason u64 offset; 2277179e29e4SChris Mason 2278179e29e4SChris Mason btrfs_item_key(leaf, &disk_key, slot); 2279179e29e4SChris Mason 2280179e29e4SChris Mason if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) { 2281179e29e4SChris Mason unsigned long ptr; 2282179e29e4SChris Mason struct btrfs_file_extent_item *fi; 2283179e29e4SChris Mason 2284179e29e4SChris Mason fi = btrfs_item_ptr(leaf, slot, 2285179e29e4SChris Mason struct btrfs_file_extent_item); 2286179e29e4SChris Mason fi = (struct btrfs_file_extent_item *)( 2287179e29e4SChris Mason (unsigned long)fi - size_diff); 2288179e29e4SChris Mason 2289179e29e4SChris Mason if (btrfs_file_extent_type(leaf, fi) == 2290179e29e4SChris Mason BTRFS_FILE_EXTENT_INLINE) { 2291179e29e4SChris Mason ptr = btrfs_item_ptr_offset(leaf, slot); 2292179e29e4SChris Mason memmove_extent_buffer(leaf, ptr, 2293179e29e4SChris Mason (unsigned long)fi, 2294179e29e4SChris Mason offsetof(struct btrfs_file_extent_item, 2295179e29e4SChris Mason disk_bytenr)); 2296179e29e4SChris Mason } 2297179e29e4SChris Mason } 2298179e29e4SChris Mason 2299179e29e4SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2300179e29e4SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 2301179e29e4SChris Mason data_end, old_data_start - data_end); 2302179e29e4SChris Mason 2303179e29e4SChris Mason offset = btrfs_disk_key_offset(&disk_key); 2304179e29e4SChris Mason btrfs_set_disk_key_offset(&disk_key, offset + size_diff); 2305179e29e4SChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 2306179e29e4SChris Mason if (slot == 0) 2307179e29e4SChris Mason fixup_low_keys(trans, root, path, &disk_key, 1); 2308179e29e4SChris Mason } 23095f39d397SChris Mason 23105f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 23115f39d397SChris Mason btrfs_set_item_size(leaf, item, new_size); 23125f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2313b18c6685SChris Mason 2314b18c6685SChris Mason ret = 0; 23155f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 23165f39d397SChris Mason btrfs_print_leaf(root, leaf); 2317b18c6685SChris Mason BUG(); 23185f39d397SChris Mason } 2319b18c6685SChris Mason return ret; 2320b18c6685SChris Mason } 2321b18c6685SChris Mason 23225f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans, 23235f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 23245f39d397SChris Mason u32 data_size) 23256567e837SChris Mason { 23266567e837SChris Mason int ret = 0; 23276567e837SChris Mason int slot; 23286567e837SChris Mason int slot_orig; 23295f39d397SChris Mason struct extent_buffer *leaf; 23305f39d397SChris Mason struct btrfs_item *item; 23316567e837SChris Mason u32 nritems; 23326567e837SChris Mason unsigned int data_end; 23336567e837SChris Mason unsigned int old_data; 23346567e837SChris Mason unsigned int old_size; 23356567e837SChris Mason int i; 23366567e837SChris Mason 23376567e837SChris Mason slot_orig = path->slots[0]; 23385f39d397SChris Mason leaf = path->nodes[0]; 23396567e837SChris Mason 23405f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 23416567e837SChris Mason data_end = leaf_data_end(root, leaf); 23426567e837SChris Mason 23435f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < data_size) { 23445f39d397SChris Mason btrfs_print_leaf(root, leaf); 23456567e837SChris Mason BUG(); 23465f39d397SChris Mason } 23476567e837SChris Mason slot = path->slots[0]; 23485f39d397SChris Mason old_data = btrfs_item_end_nr(leaf, slot); 23496567e837SChris Mason 23506567e837SChris Mason BUG_ON(slot < 0); 23513326d1b0SChris Mason if (slot >= nritems) { 23523326d1b0SChris Mason btrfs_print_leaf(root, leaf); 23533326d1b0SChris Mason printk("slot %d too large, nritems %d\n", slot, nritems); 23543326d1b0SChris Mason BUG_ON(1); 23553326d1b0SChris Mason } 23566567e837SChris Mason 23576567e837SChris Mason /* 23586567e837SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 23596567e837SChris Mason */ 23606567e837SChris Mason /* first correct the data pointers */ 23616567e837SChris Mason for (i = slot; i < nritems; i++) { 23625f39d397SChris Mason u32 ioff; 23635f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2364db94535dSChris Mason 2365db94535dSChris Mason if (!leaf->map_token) { 2366db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2367db94535dSChris Mason sizeof(struct btrfs_item), 2368db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2369db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2370db94535dSChris Mason KM_USER1); 2371db94535dSChris Mason } 23725f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 23735f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff - data_size); 23746567e837SChris Mason } 23755f39d397SChris Mason 2376db94535dSChris Mason if (leaf->map_token) { 2377db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2378db94535dSChris Mason leaf->map_token = NULL; 2379db94535dSChris Mason } 2380db94535dSChris Mason 23816567e837SChris Mason /* shift the data */ 23825f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 23836567e837SChris Mason data_end - data_size, btrfs_leaf_data(leaf) + 23846567e837SChris Mason data_end, old_data - data_end); 23855f39d397SChris Mason 23866567e837SChris Mason data_end = old_data; 23875f39d397SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 23885f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 23895f39d397SChris Mason btrfs_set_item_size(leaf, item, old_size + data_size); 23905f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 23916567e837SChris Mason 23926567e837SChris Mason ret = 0; 23935f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 23945f39d397SChris Mason btrfs_print_leaf(root, leaf); 23956567e837SChris Mason BUG(); 23965f39d397SChris Mason } 23976567e837SChris Mason return ret; 23986567e837SChris Mason } 23996567e837SChris Mason 240074123bd7SChris Mason /* 240174123bd7SChris Mason * Given a key and some data, insert an item into the tree. 240274123bd7SChris Mason * This does all the path init required, making room in the tree if needed. 240374123bd7SChris Mason */ 24049c58309dSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans, 24055f39d397SChris Mason struct btrfs_root *root, 24065f39d397SChris Mason struct btrfs_path *path, 24079c58309dSChris Mason struct btrfs_key *cpu_key, u32 *data_size, 24089c58309dSChris Mason int nr) 2409be0e5c09SChris Mason { 24105f39d397SChris Mason struct extent_buffer *leaf; 24115f39d397SChris Mason struct btrfs_item *item; 2412aa5d6bedSChris Mason int ret = 0; 2413be0e5c09SChris Mason int slot; 2414eb60ceacSChris Mason int slot_orig; 24159c58309dSChris Mason int i; 24167518a238SChris Mason u32 nritems; 24179c58309dSChris Mason u32 total_size = 0; 24189c58309dSChris Mason u32 total_data = 0; 2419be0e5c09SChris Mason unsigned int data_end; 2420e2fa7227SChris Mason struct btrfs_disk_key disk_key; 2421e2fa7227SChris Mason 24229c58309dSChris Mason for (i = 0; i < nr; i++) { 24239c58309dSChris Mason total_data += data_size[i]; 24249c58309dSChris Mason } 2425be0e5c09SChris Mason 242674123bd7SChris Mason /* create a root if there isn't one */ 24275c680ed6SChris Mason if (!root->node) 2428cfaa7295SChris Mason BUG(); 24295f39d397SChris Mason 24309c58309dSChris Mason total_size = total_data + (nr - 1) * sizeof(struct btrfs_item); 24319c58309dSChris Mason ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1); 2432eb60ceacSChris Mason if (ret == 0) { 2433f0930a37SChris Mason return -EEXIST; 2434aa5d6bedSChris Mason } 2435ed2ff2cbSChris Mason if (ret < 0) 2436ed2ff2cbSChris Mason goto out; 2437be0e5c09SChris Mason 243862e2749eSChris Mason slot_orig = path->slots[0]; 24395f39d397SChris Mason leaf = path->nodes[0]; 244074123bd7SChris Mason 24415f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2442123abc88SChris Mason data_end = leaf_data_end(root, leaf); 2443eb60ceacSChris Mason 2444123abc88SChris Mason if (btrfs_leaf_free_space(root, leaf) < 24459c58309dSChris Mason sizeof(struct btrfs_item) + total_size) { 24463326d1b0SChris Mason btrfs_print_leaf(root, leaf); 24473326d1b0SChris Mason printk("not enough freespace need %u have %d\n", 24489c58309dSChris Mason total_size, btrfs_leaf_free_space(root, leaf)); 2449be0e5c09SChris Mason BUG(); 2450d4dbff95SChris Mason } 24515f39d397SChris Mason 245262e2749eSChris Mason slot = path->slots[0]; 2453eb60ceacSChris Mason BUG_ON(slot < 0); 24545f39d397SChris Mason 2455be0e5c09SChris Mason if (slot != nritems) { 2456be0e5c09SChris Mason int i; 24575f39d397SChris Mason unsigned int old_data = btrfs_item_end_nr(leaf, slot); 2458be0e5c09SChris Mason 24595f39d397SChris Mason if (old_data < data_end) { 24605f39d397SChris Mason btrfs_print_leaf(root, leaf); 24615f39d397SChris Mason printk("slot %d old_data %d data_end %d\n", 24625f39d397SChris Mason slot, old_data, data_end); 24635f39d397SChris Mason BUG_ON(1); 24645f39d397SChris Mason } 2465be0e5c09SChris Mason /* 2466be0e5c09SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2467be0e5c09SChris Mason */ 2468be0e5c09SChris Mason /* first correct the data pointers */ 2469db94535dSChris Mason WARN_ON(leaf->map_token); 24700783fcfcSChris Mason for (i = slot; i < nritems; i++) { 24715f39d397SChris Mason u32 ioff; 2472db94535dSChris Mason 24735f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2474db94535dSChris Mason if (!leaf->map_token) { 2475db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2476db94535dSChris Mason sizeof(struct btrfs_item), 2477db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2478db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2479db94535dSChris Mason KM_USER1); 2480db94535dSChris Mason } 2481db94535dSChris Mason 24825f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 24839c58309dSChris Mason btrfs_set_item_offset(leaf, item, ioff - total_data); 24840783fcfcSChris Mason } 2485db94535dSChris Mason if (leaf->map_token) { 2486db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2487db94535dSChris Mason leaf->map_token = NULL; 2488db94535dSChris Mason } 2489be0e5c09SChris Mason 2490be0e5c09SChris Mason /* shift the items */ 24919c58309dSChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr), 24925f39d397SChris Mason btrfs_item_nr_offset(slot), 24930783fcfcSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 2494be0e5c09SChris Mason 2495be0e5c09SChris Mason /* shift the data */ 24965f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 24979c58309dSChris Mason data_end - total_data, btrfs_leaf_data(leaf) + 2498be0e5c09SChris Mason data_end, old_data - data_end); 2499be0e5c09SChris Mason data_end = old_data; 2500be0e5c09SChris Mason } 25015f39d397SChris Mason 250262e2749eSChris Mason /* setup the item for the new data */ 25039c58309dSChris Mason for (i = 0; i < nr; i++) { 25049c58309dSChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key + i); 25059c58309dSChris Mason btrfs_set_item_key(leaf, &disk_key, slot + i); 25069c58309dSChris Mason item = btrfs_item_nr(leaf, slot + i); 25079c58309dSChris Mason btrfs_set_item_offset(leaf, item, data_end - data_size[i]); 25089c58309dSChris Mason data_end -= data_size[i]; 25099c58309dSChris Mason btrfs_set_item_size(leaf, item, data_size[i]); 25109c58309dSChris Mason } 25119c58309dSChris Mason btrfs_set_header_nritems(leaf, nritems + nr); 25125f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2513aa5d6bedSChris Mason 2514aa5d6bedSChris Mason ret = 0; 25155a01a2e3SChris Mason if (slot == 0) { 25165a01a2e3SChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key); 2517e089f05cSChris Mason ret = fixup_low_keys(trans, root, path, &disk_key, 1); 25185a01a2e3SChris Mason } 2519aa5d6bedSChris Mason 25205f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 25215f39d397SChris Mason btrfs_print_leaf(root, leaf); 2522be0e5c09SChris Mason BUG(); 25235f39d397SChris Mason } 25245a01a2e3SChris Mason 2525ed2ff2cbSChris Mason out: 252662e2749eSChris Mason return ret; 252762e2749eSChris Mason } 252862e2749eSChris Mason 252962e2749eSChris Mason /* 253062e2749eSChris Mason * Given a key and some data, insert an item into the tree. 253162e2749eSChris Mason * This does all the path init required, making room in the tree if needed. 253262e2749eSChris Mason */ 2533e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root 2534e089f05cSChris Mason *root, struct btrfs_key *cpu_key, void *data, u32 2535e089f05cSChris Mason data_size) 253662e2749eSChris Mason { 253762e2749eSChris Mason int ret = 0; 25382c90e5d6SChris Mason struct btrfs_path *path; 25395f39d397SChris Mason struct extent_buffer *leaf; 25405f39d397SChris Mason unsigned long ptr; 254162e2749eSChris Mason 25422c90e5d6SChris Mason path = btrfs_alloc_path(); 25432c90e5d6SChris Mason BUG_ON(!path); 25442c90e5d6SChris Mason ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size); 254562e2749eSChris Mason if (!ret) { 25465f39d397SChris Mason leaf = path->nodes[0]; 25475f39d397SChris Mason ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); 25485f39d397SChris Mason write_extent_buffer(leaf, data, ptr, data_size); 25495f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 255062e2749eSChris Mason } 25512c90e5d6SChris Mason btrfs_free_path(path); 2552aa5d6bedSChris Mason return ret; 2553be0e5c09SChris Mason } 2554be0e5c09SChris Mason 255574123bd7SChris Mason /* 25565de08d7dSChris Mason * delete the pointer from a given node. 255774123bd7SChris Mason * 255874123bd7SChris Mason * If the delete empties a node, the node is removed from the tree, 255974123bd7SChris Mason * continuing all the way the root if required. The root is converted into 256074123bd7SChris Mason * a leaf if all the nodes are emptied. 256174123bd7SChris Mason */ 2562e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 2563e089f05cSChris Mason struct btrfs_path *path, int level, int slot) 2564be0e5c09SChris Mason { 25655f39d397SChris Mason struct extent_buffer *parent = path->nodes[level]; 25667518a238SChris Mason u32 nritems; 2567aa5d6bedSChris Mason int ret = 0; 2568bb803951SChris Mason int wret; 2569be0e5c09SChris Mason 25705f39d397SChris Mason nritems = btrfs_header_nritems(parent); 2571be0e5c09SChris Mason if (slot != nritems -1) { 25725f39d397SChris Mason memmove_extent_buffer(parent, 25735f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 25745f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 2575d6025579SChris Mason sizeof(struct btrfs_key_ptr) * 2576d6025579SChris Mason (nritems - slot - 1)); 2577be0e5c09SChris Mason } 25787518a238SChris Mason nritems--; 25795f39d397SChris Mason btrfs_set_header_nritems(parent, nritems); 25807518a238SChris Mason if (nritems == 0 && parent == root->node) { 25815f39d397SChris Mason BUG_ON(btrfs_header_level(root->node) != 1); 2582eb60ceacSChris Mason /* just turn the root into a leaf and break */ 25835f39d397SChris Mason btrfs_set_header_level(root->node, 0); 2584bb803951SChris Mason } else if (slot == 0) { 25855f39d397SChris Mason struct btrfs_disk_key disk_key; 25865f39d397SChris Mason 25875f39d397SChris Mason btrfs_node_key(parent, &disk_key, 0); 25885f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, level + 1); 25890f70abe2SChris Mason if (wret) 25900f70abe2SChris Mason ret = wret; 2591be0e5c09SChris Mason } 2592d6025579SChris Mason btrfs_mark_buffer_dirty(parent); 2593aa5d6bedSChris Mason return ret; 2594be0e5c09SChris Mason } 2595be0e5c09SChris Mason 259674123bd7SChris Mason /* 259774123bd7SChris Mason * delete the item at the leaf level in path. If that empties 259874123bd7SChris Mason * the leaf, remove it from the tree 259974123bd7SChris Mason */ 260085e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root, 260185e21bacSChris Mason struct btrfs_path *path, int slot, int nr) 2602be0e5c09SChris Mason { 26035f39d397SChris Mason struct extent_buffer *leaf; 26045f39d397SChris Mason struct btrfs_item *item; 260585e21bacSChris Mason int last_off; 260685e21bacSChris Mason int dsize = 0; 2607aa5d6bedSChris Mason int ret = 0; 2608aa5d6bedSChris Mason int wret; 260985e21bacSChris Mason int i; 26107518a238SChris Mason u32 nritems; 2611be0e5c09SChris Mason 26125f39d397SChris Mason leaf = path->nodes[0]; 261385e21bacSChris Mason last_off = btrfs_item_offset_nr(leaf, slot + nr - 1); 261485e21bacSChris Mason 261585e21bacSChris Mason for (i = 0; i < nr; i++) 261685e21bacSChris Mason dsize += btrfs_item_size_nr(leaf, slot + i); 261785e21bacSChris Mason 26185f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2619be0e5c09SChris Mason 262085e21bacSChris Mason if (slot + nr != nritems) { 2621be0e5c09SChris Mason int i; 2622123abc88SChris Mason int data_end = leaf_data_end(root, leaf); 26235f39d397SChris Mason 26245f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2625d6025579SChris Mason data_end + dsize, 2626123abc88SChris Mason btrfs_leaf_data(leaf) + data_end, 262785e21bacSChris Mason last_off - data_end); 26285f39d397SChris Mason 262985e21bacSChris Mason for (i = slot + nr; i < nritems; i++) { 26305f39d397SChris Mason u32 ioff; 2631db94535dSChris Mason 26325f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2633db94535dSChris Mason if (!leaf->map_token) { 2634db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2635db94535dSChris Mason sizeof(struct btrfs_item), 2636db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2637db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2638db94535dSChris Mason KM_USER1); 2639db94535dSChris Mason } 26405f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 26415f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + dsize); 26420783fcfcSChris Mason } 2643db94535dSChris Mason 2644db94535dSChris Mason if (leaf->map_token) { 2645db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2646db94535dSChris Mason leaf->map_token = NULL; 2647db94535dSChris Mason } 2648db94535dSChris Mason 26495f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot), 265085e21bacSChris Mason btrfs_item_nr_offset(slot + nr), 26510783fcfcSChris Mason sizeof(struct btrfs_item) * 265285e21bacSChris Mason (nritems - slot - nr)); 2653be0e5c09SChris Mason } 265485e21bacSChris Mason btrfs_set_header_nritems(leaf, nritems - nr); 265585e21bacSChris Mason nritems -= nr; 26565f39d397SChris Mason 265774123bd7SChris Mason /* delete the leaf if we've emptied it */ 26587518a238SChris Mason if (nritems == 0) { 26595f39d397SChris Mason if (leaf == root->node) { 26605f39d397SChris Mason btrfs_set_header_level(leaf, 0); 26619a8dd150SChris Mason } else { 26627bb86316SChris Mason u64 root_gen = btrfs_header_generation(path->nodes[1]); 26635f39d397SChris Mason clean_tree_block(trans, root, leaf); 2664e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, path->slots[1]); 2665aa5d6bedSChris Mason if (wret) 2666aa5d6bedSChris Mason ret = wret; 2667e089f05cSChris Mason wret = btrfs_free_extent(trans, root, 26687bb86316SChris Mason leaf->start, leaf->len, 26697bb86316SChris Mason btrfs_header_owner(path->nodes[1]), 26707bb86316SChris Mason root_gen, 0, 0, 1); 26710f70abe2SChris Mason if (wret) 26720f70abe2SChris Mason ret = wret; 26739a8dd150SChris Mason } 2674be0e5c09SChris Mason } else { 26757518a238SChris Mason int used = leaf_space_used(leaf, 0, nritems); 2676aa5d6bedSChris Mason if (slot == 0) { 26775f39d397SChris Mason struct btrfs_disk_key disk_key; 26785f39d397SChris Mason 26795f39d397SChris Mason btrfs_item_key(leaf, &disk_key, 0); 2680e089f05cSChris Mason wret = fixup_low_keys(trans, root, path, 26815f39d397SChris Mason &disk_key, 1); 2682aa5d6bedSChris Mason if (wret) 2683aa5d6bedSChris Mason ret = wret; 2684aa5d6bedSChris Mason } 2685aa5d6bedSChris Mason 268674123bd7SChris Mason /* delete the leaf if it is mostly empty */ 268785e21bacSChris Mason if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) { 2688be0e5c09SChris Mason /* push_leaf_left fixes the path. 2689be0e5c09SChris Mason * make sure the path still points to our leaf 2690be0e5c09SChris Mason * for possible call to del_ptr below 2691be0e5c09SChris Mason */ 26924920c9acSChris Mason slot = path->slots[1]; 26935f39d397SChris Mason extent_buffer_get(leaf); 26945f39d397SChris Mason 269585e21bacSChris Mason wret = push_leaf_left(trans, root, path, 1, 1); 269654aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2697aa5d6bedSChris Mason ret = wret; 26985f39d397SChris Mason 26995f39d397SChris Mason if (path->nodes[0] == leaf && 27005f39d397SChris Mason btrfs_header_nritems(leaf)) { 270185e21bacSChris Mason wret = push_leaf_right(trans, root, path, 1, 1); 270254aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2703aa5d6bedSChris Mason ret = wret; 2704aa5d6bedSChris Mason } 27055f39d397SChris Mason 27065f39d397SChris Mason if (btrfs_header_nritems(leaf) == 0) { 27077bb86316SChris Mason u64 root_gen; 2708db94535dSChris Mason u64 bytenr = leaf->start; 2709db94535dSChris Mason u32 blocksize = leaf->len; 27105f39d397SChris Mason 27117bb86316SChris Mason root_gen = btrfs_header_generation( 27127bb86316SChris Mason path->nodes[1]); 27137bb86316SChris Mason 27145f39d397SChris Mason clean_tree_block(trans, root, leaf); 27155f39d397SChris Mason 2716e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, slot); 2717aa5d6bedSChris Mason if (wret) 2718aa5d6bedSChris Mason ret = wret; 27195f39d397SChris Mason 27205f39d397SChris Mason free_extent_buffer(leaf); 2721db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 27227bb86316SChris Mason blocksize, 27237bb86316SChris Mason btrfs_header_owner(path->nodes[1]), 27247bb86316SChris Mason root_gen, 0, 0, 1); 27250f70abe2SChris Mason if (wret) 27260f70abe2SChris Mason ret = wret; 27275de08d7dSChris Mason } else { 27285f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 27295f39d397SChris Mason free_extent_buffer(leaf); 2730be0e5c09SChris Mason } 2731d5719762SChris Mason } else { 27325f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2733be0e5c09SChris Mason } 27349a8dd150SChris Mason } 2735aa5d6bedSChris Mason return ret; 27369a8dd150SChris Mason } 27379a8dd150SChris Mason 273897571fd0SChris Mason /* 27397bb86316SChris Mason * walk up the tree as far as required to find the previous leaf. 27407bb86316SChris Mason * returns 0 if it found something or 1 if there are no lesser leaves. 27417bb86316SChris Mason * returns < 0 on io errors. 27427bb86316SChris Mason */ 27437bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path) 27447bb86316SChris Mason { 27458f662a76SChris Mason u64 bytenr; 27467bb86316SChris Mason int slot; 27477bb86316SChris Mason int level = 1; 27487bb86316SChris Mason struct extent_buffer *c; 27497bb86316SChris Mason struct extent_buffer *next = NULL; 27507bb86316SChris Mason 27517bb86316SChris Mason while(level < BTRFS_MAX_LEVEL) { 27527bb86316SChris Mason if (!path->nodes[level]) 27537bb86316SChris Mason return 1; 27547bb86316SChris Mason 27557bb86316SChris Mason slot = path->slots[level]; 27567bb86316SChris Mason c = path->nodes[level]; 27577bb86316SChris Mason if (slot == 0) { 27587bb86316SChris Mason level++; 27597bb86316SChris Mason if (level == BTRFS_MAX_LEVEL) 27607bb86316SChris Mason return 1; 27617bb86316SChris Mason continue; 27627bb86316SChris Mason } 27637bb86316SChris Mason slot--; 27647bb86316SChris Mason 27657bb86316SChris Mason bytenr = btrfs_node_blockptr(c, slot); 27667bb86316SChris Mason if (next) 27677bb86316SChris Mason free_extent_buffer(next); 27687bb86316SChris Mason 27697bb86316SChris Mason next = read_tree_block(root, bytenr, 27707bb86316SChris Mason btrfs_level_size(root, level - 1)); 27717bb86316SChris Mason break; 27727bb86316SChris Mason } 27737bb86316SChris Mason path->slots[level] = slot; 27747bb86316SChris Mason while(1) { 27757bb86316SChris Mason level--; 27767bb86316SChris Mason c = path->nodes[level]; 27777bb86316SChris Mason free_extent_buffer(c); 27788f662a76SChris Mason slot = btrfs_header_nritems(next); 27798f662a76SChris Mason if (slot != 0) 27808f662a76SChris Mason slot--; 27817bb86316SChris Mason path->nodes[level] = next; 27828f662a76SChris Mason path->slots[level] = slot; 27837bb86316SChris Mason if (!level) 27847bb86316SChris Mason break; 27858f662a76SChris Mason next = read_tree_block(root, btrfs_node_blockptr(next, slot), 27867bb86316SChris Mason btrfs_level_size(root, level - 1)); 27877bb86316SChris Mason } 27887bb86316SChris Mason return 0; 27897bb86316SChris Mason } 27907bb86316SChris Mason 27917bb86316SChris Mason /* 279297571fd0SChris Mason * walk up the tree as far as required to find the next leaf. 27930f70abe2SChris Mason * returns 0 if it found something or 1 if there are no greater leaves. 27940f70abe2SChris Mason * returns < 0 on io errors. 279597571fd0SChris Mason */ 2796234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path) 2797d97e63b6SChris Mason { 2798d97e63b6SChris Mason int slot; 2799d97e63b6SChris Mason int level = 1; 2800db94535dSChris Mason u64 bytenr; 28015f39d397SChris Mason struct extent_buffer *c; 28025f39d397SChris Mason struct extent_buffer *next = NULL; 2803d97e63b6SChris Mason 2804234b63a0SChris Mason while(level < BTRFS_MAX_LEVEL) { 2805d97e63b6SChris Mason if (!path->nodes[level]) 28060f70abe2SChris Mason return 1; 28075f39d397SChris Mason 2808d97e63b6SChris Mason slot = path->slots[level] + 1; 2809d97e63b6SChris Mason c = path->nodes[level]; 28105f39d397SChris Mason if (slot >= btrfs_header_nritems(c)) { 2811d97e63b6SChris Mason level++; 28127bb86316SChris Mason if (level == BTRFS_MAX_LEVEL) 28137bb86316SChris Mason return 1; 2814d97e63b6SChris Mason continue; 2815d97e63b6SChris Mason } 28165f39d397SChris Mason 2817db94535dSChris Mason bytenr = btrfs_node_blockptr(c, slot); 2818cfaa7295SChris Mason if (next) 28195f39d397SChris Mason free_extent_buffer(next); 28205f39d397SChris Mason 28216702ed49SChris Mason if (path->reada) 282201f46658SChris Mason reada_for_search(root, path, level, slot, 0); 28235f39d397SChris Mason 2824db94535dSChris Mason next = read_tree_block(root, bytenr, 2825db94535dSChris Mason btrfs_level_size(root, level -1)); 2826d97e63b6SChris Mason break; 2827d97e63b6SChris Mason } 2828d97e63b6SChris Mason path->slots[level] = slot; 2829d97e63b6SChris Mason while(1) { 2830d97e63b6SChris Mason level--; 2831d97e63b6SChris Mason c = path->nodes[level]; 28325f39d397SChris Mason free_extent_buffer(c); 2833d97e63b6SChris Mason path->nodes[level] = next; 2834d97e63b6SChris Mason path->slots[level] = 0; 2835d97e63b6SChris Mason if (!level) 2836d97e63b6SChris Mason break; 28376702ed49SChris Mason if (path->reada) 283801f46658SChris Mason reada_for_search(root, path, level, 0, 0); 2839db94535dSChris Mason next = read_tree_block(root, btrfs_node_blockptr(next, 0), 2840db94535dSChris Mason btrfs_level_size(root, level - 1)); 2841d97e63b6SChris Mason } 2842d97e63b6SChris Mason return 0; 2843d97e63b6SChris Mason } 28440b86a832SChris Mason 28450b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root, 28460b86a832SChris Mason struct btrfs_path *path, u64 min_objectid, 28470b86a832SChris Mason int type) 28480b86a832SChris Mason { 28490b86a832SChris Mason struct btrfs_key found_key; 28500b86a832SChris Mason struct extent_buffer *leaf; 28510b86a832SChris Mason int ret; 28520b86a832SChris Mason 28530b86a832SChris Mason while(1) { 28540b86a832SChris Mason if (path->slots[0] == 0) { 28550b86a832SChris Mason ret = btrfs_prev_leaf(root, path); 28560b86a832SChris Mason if (ret != 0) 28570b86a832SChris Mason return ret; 28580b86a832SChris Mason } else { 28590b86a832SChris Mason path->slots[0]--; 28600b86a832SChris Mason } 28610b86a832SChris Mason leaf = path->nodes[0]; 28620b86a832SChris Mason btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); 28630b86a832SChris Mason if (found_key.type == type) 28640b86a832SChris Mason return 0; 28650b86a832SChris Mason } 28660b86a832SChris Mason return 1; 28670b86a832SChris Mason } 28680b86a832SChris Mason 2869