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, 325f39d397SChris Mason struct extent_buffer *src); 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); 7925f39d397SChris Mason wret = push_node_left(trans, root, left, mid); 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) { 8035f39d397SChris Mason wret = push_node_left(trans, root, mid, right); 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 } 84779f95c82SChris Mason BUG_ON(wret == 1); 84879f95c82SChris Mason } 8495f39d397SChris Mason if (btrfs_header_nritems(mid) == 0) { 85079f95c82SChris Mason /* we've managed to empty the middle node, drop it */ 8517bb86316SChris Mason u64 root_gen = btrfs_header_generation(parent); 852db94535dSChris Mason u64 bytenr = mid->start; 853db94535dSChris Mason u32 blocksize = mid->len; 8545f39d397SChris Mason clean_tree_block(trans, root, mid); 8555f39d397SChris Mason free_extent_buffer(mid); 856bb803951SChris Mason mid = NULL; 857e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot); 858bb803951SChris Mason if (wret) 859bb803951SChris Mason ret = wret; 8607bb86316SChris Mason wret = btrfs_free_extent(trans, root, bytenr, blocksize, 8617bb86316SChris Mason btrfs_header_owner(parent), 8627bb86316SChris Mason root_gen, 0, 0, 1); 863bb803951SChris Mason if (wret) 864bb803951SChris Mason ret = wret; 86579f95c82SChris Mason } else { 86679f95c82SChris Mason /* update the parent key to reflect our changes */ 8675f39d397SChris Mason struct btrfs_disk_key mid_key; 8685f39d397SChris Mason btrfs_node_key(mid, &mid_key, 0); 8695f39d397SChris Mason btrfs_set_node_key(parent, &mid_key, pslot); 8705f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 87179f95c82SChris Mason } 872bb803951SChris Mason 87379f95c82SChris Mason /* update the path */ 8745f39d397SChris Mason if (left) { 8755f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 8765f39d397SChris Mason extent_buffer_get(left); 8775f39d397SChris Mason path->nodes[level] = left; 878bb803951SChris Mason path->slots[level + 1] -= 1; 879bb803951SChris Mason path->slots[level] = orig_slot; 8805f39d397SChris Mason if (mid) 8815f39d397SChris Mason free_extent_buffer(mid); 882bb803951SChris Mason } else { 8835f39d397SChris Mason orig_slot -= btrfs_header_nritems(left); 884bb803951SChris Mason path->slots[level] = orig_slot; 885bb803951SChris Mason } 886bb803951SChris Mason } 88779f95c82SChris Mason /* double check we haven't messed things up */ 888123abc88SChris Mason check_block(root, path, level); 889e20d96d6SChris Mason if (orig_ptr != 8905f39d397SChris Mason btrfs_node_blockptr(path->nodes[level], path->slots[level])) 89179f95c82SChris Mason BUG(); 89254aa1f4dSChris Mason enospc: 8935f39d397SChris Mason if (right) 8945f39d397SChris Mason free_extent_buffer(right); 8955f39d397SChris Mason if (left) 8965f39d397SChris Mason free_extent_buffer(left); 897bb803951SChris Mason return ret; 898bb803951SChris Mason } 899bb803951SChris Mason 900e66f709bSChris Mason /* returns zero if the push worked, non-zero otherwise */ 90198ed5174SChris Mason static int noinline push_nodes_for_insert(struct btrfs_trans_handle *trans, 902e66f709bSChris Mason struct btrfs_root *root, 903e66f709bSChris Mason struct btrfs_path *path, int level) 904e66f709bSChris Mason { 9055f39d397SChris Mason struct extent_buffer *right = NULL; 9065f39d397SChris Mason struct extent_buffer *mid; 9075f39d397SChris Mason struct extent_buffer *left = NULL; 9085f39d397SChris Mason struct extent_buffer *parent = NULL; 909e66f709bSChris Mason int ret = 0; 910e66f709bSChris Mason int wret; 911e66f709bSChris Mason int pslot; 912e66f709bSChris Mason int orig_slot = path->slots[level]; 913e66f709bSChris Mason u64 orig_ptr; 914e66f709bSChris Mason 915e66f709bSChris Mason if (level == 0) 916e66f709bSChris Mason return 1; 917e66f709bSChris Mason 9185f39d397SChris Mason mid = path->nodes[level]; 9197bb86316SChris Mason WARN_ON(btrfs_header_generation(mid) != trans->transid); 920e66f709bSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 921e66f709bSChris Mason 922e66f709bSChris Mason if (level < BTRFS_MAX_LEVEL - 1) 9235f39d397SChris Mason parent = path->nodes[level + 1]; 924e66f709bSChris Mason pslot = path->slots[level + 1]; 925e66f709bSChris Mason 9265f39d397SChris Mason if (!parent) 927e66f709bSChris Mason return 1; 928e66f709bSChris Mason 9295f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 930e66f709bSChris Mason 931e66f709bSChris Mason /* first, try to make some room in the middle buffer */ 9325f39d397SChris Mason if (left) { 933e66f709bSChris Mason u32 left_nr; 9345f39d397SChris Mason left_nr = btrfs_header_nritems(left); 93533ade1f8SChris Mason if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 93633ade1f8SChris Mason wret = 1; 93733ade1f8SChris Mason } else { 9385f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, parent, 9395f39d397SChris Mason pslot - 1, &left); 94054aa1f4dSChris Mason if (ret) 94154aa1f4dSChris Mason wret = 1; 94254aa1f4dSChris Mason else { 94354aa1f4dSChris Mason wret = push_node_left(trans, root, 9445f39d397SChris Mason left, mid); 94554aa1f4dSChris Mason } 94633ade1f8SChris Mason } 947e66f709bSChris Mason if (wret < 0) 948e66f709bSChris Mason ret = wret; 949e66f709bSChris Mason if (wret == 0) { 9505f39d397SChris Mason struct btrfs_disk_key disk_key; 951e66f709bSChris Mason orig_slot += left_nr; 9525f39d397SChris Mason btrfs_node_key(mid, &disk_key, 0); 9535f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot); 9545f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 9555f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 9565f39d397SChris Mason path->nodes[level] = left; 957e66f709bSChris Mason path->slots[level + 1] -= 1; 958e66f709bSChris Mason path->slots[level] = orig_slot; 9595f39d397SChris Mason free_extent_buffer(mid); 960e66f709bSChris Mason } else { 961e66f709bSChris Mason orig_slot -= 9625f39d397SChris Mason btrfs_header_nritems(left); 963e66f709bSChris Mason path->slots[level] = orig_slot; 9645f39d397SChris Mason free_extent_buffer(left); 965e66f709bSChris Mason } 966e66f709bSChris Mason return 0; 967e66f709bSChris Mason } 9685f39d397SChris Mason free_extent_buffer(left); 969e66f709bSChris Mason } 9705f39d397SChris Mason right= read_node_slot(root, parent, pslot + 1); 971e66f709bSChris Mason 972e66f709bSChris Mason /* 973e66f709bSChris Mason * then try to empty the right most buffer into the middle 974e66f709bSChris Mason */ 9755f39d397SChris Mason if (right) { 97633ade1f8SChris Mason u32 right_nr; 9775f39d397SChris Mason right_nr = btrfs_header_nritems(right); 97833ade1f8SChris Mason if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 97933ade1f8SChris Mason wret = 1; 98033ade1f8SChris Mason } else { 9815f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, 9825f39d397SChris Mason parent, pslot + 1, 9835f39d397SChris Mason &right); 98454aa1f4dSChris Mason if (ret) 98554aa1f4dSChris Mason wret = 1; 98654aa1f4dSChris Mason else { 98733ade1f8SChris Mason wret = balance_node_right(trans, root, 9885f39d397SChris Mason right, mid); 98933ade1f8SChris Mason } 99054aa1f4dSChris Mason } 991e66f709bSChris Mason if (wret < 0) 992e66f709bSChris Mason ret = wret; 993e66f709bSChris Mason if (wret == 0) { 9945f39d397SChris Mason struct btrfs_disk_key disk_key; 9955f39d397SChris Mason 9965f39d397SChris Mason btrfs_node_key(right, &disk_key, 0); 9975f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot + 1); 9985f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 9995f39d397SChris Mason 10005f39d397SChris Mason if (btrfs_header_nritems(mid) <= orig_slot) { 10015f39d397SChris Mason path->nodes[level] = right; 1002e66f709bSChris Mason path->slots[level + 1] += 1; 1003e66f709bSChris Mason path->slots[level] = orig_slot - 10045f39d397SChris Mason btrfs_header_nritems(mid); 10055f39d397SChris Mason free_extent_buffer(mid); 1006e66f709bSChris Mason } else { 10075f39d397SChris Mason free_extent_buffer(right); 1008e66f709bSChris Mason } 1009e66f709bSChris Mason return 0; 1010e66f709bSChris Mason } 10115f39d397SChris Mason free_extent_buffer(right); 1012e66f709bSChris Mason } 1013e66f709bSChris Mason return 1; 1014e66f709bSChris Mason } 1015e66f709bSChris Mason 101674123bd7SChris Mason /* 10173c69faecSChris Mason * readahead one full node of leaves 10183c69faecSChris Mason */ 10193c69faecSChris Mason static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path, 102001f46658SChris Mason int level, int slot, u64 objectid) 10213c69faecSChris Mason { 10225f39d397SChris Mason struct extent_buffer *node; 102301f46658SChris Mason struct btrfs_disk_key disk_key; 10243c69faecSChris Mason u32 nritems; 10253c69faecSChris Mason u64 search; 10266b80053dSChris Mason u64 lowest_read; 10276b80053dSChris Mason u64 highest_read; 10286b80053dSChris Mason u64 nread = 0; 10293c69faecSChris Mason int direction = path->reada; 10305f39d397SChris Mason struct extent_buffer *eb; 10316b80053dSChris Mason u32 nr; 10326b80053dSChris Mason u32 blocksize; 10336b80053dSChris Mason u32 nscan = 0; 1034db94535dSChris Mason 1035a6b6e75eSChris Mason if (level != 1) 10363c69faecSChris Mason return; 10373c69faecSChris Mason 10386702ed49SChris Mason if (!path->nodes[level]) 10396702ed49SChris Mason return; 10406702ed49SChris Mason 10415f39d397SChris Mason node = path->nodes[level]; 10423c69faecSChris Mason search = btrfs_node_blockptr(node, slot); 10436b80053dSChris Mason blocksize = btrfs_level_size(root, level - 1); 10446b80053dSChris Mason eb = btrfs_find_tree_block(root, search, blocksize); 10455f39d397SChris Mason if (eb) { 10465f39d397SChris Mason free_extent_buffer(eb); 10473c69faecSChris Mason return; 10483c69faecSChris Mason } 10493c69faecSChris Mason 10506b80053dSChris Mason highest_read = search; 10516b80053dSChris Mason lowest_read = search; 10526b80053dSChris Mason 10535f39d397SChris Mason nritems = btrfs_header_nritems(node); 10546b80053dSChris Mason nr = slot; 10553c69faecSChris Mason while(1) { 10566b80053dSChris Mason if (direction < 0) { 10576b80053dSChris Mason if (nr == 0) 10583c69faecSChris Mason break; 10596b80053dSChris Mason nr--; 10606b80053dSChris Mason } else if (direction > 0) { 10616b80053dSChris Mason nr++; 10626b80053dSChris Mason if (nr >= nritems) 10636b80053dSChris Mason break; 10643c69faecSChris Mason } 106501f46658SChris Mason if (path->reada < 0 && objectid) { 106601f46658SChris Mason btrfs_node_key(node, &disk_key, nr); 106701f46658SChris Mason if (btrfs_disk_key_objectid(&disk_key) != objectid) 106801f46658SChris Mason break; 106901f46658SChris Mason } 10706b80053dSChris Mason search = btrfs_node_blockptr(node, nr); 10716b80053dSChris Mason if ((search >= lowest_read && search <= highest_read) || 10726b80053dSChris Mason (search < lowest_read && lowest_read - search <= 32768) || 10736b80053dSChris Mason (search > highest_read && search - highest_read <= 32768)) { 10746b80053dSChris Mason readahead_tree_block(root, search, blocksize); 10756b80053dSChris Mason nread += blocksize; 10763c69faecSChris Mason } 10776b80053dSChris Mason nscan++; 10786b80053dSChris Mason if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32)) 10796b80053dSChris Mason break; 10806b80053dSChris Mason if(nread > (1024 * 1024) || nscan > 128) 10816b80053dSChris Mason break; 10826b80053dSChris Mason 10836b80053dSChris Mason if (search < lowest_read) 10846b80053dSChris Mason lowest_read = search; 10856b80053dSChris Mason if (search > highest_read) 10866b80053dSChris Mason highest_read = search; 10873c69faecSChris Mason } 10883c69faecSChris Mason } 10893c69faecSChris Mason /* 109074123bd7SChris Mason * look for key in the tree. path is filled in with nodes along the way 109174123bd7SChris Mason * if key is found, we return zero and you can find the item in the leaf 109274123bd7SChris Mason * level of the path (level 0) 109374123bd7SChris Mason * 109474123bd7SChris Mason * If the key isn't found, the path points to the slot where it should 1095aa5d6bedSChris Mason * be inserted, and 1 is returned. If there are other errors during the 1096aa5d6bedSChris Mason * search a negative error number is returned. 109797571fd0SChris Mason * 109897571fd0SChris Mason * if ins_len > 0, nodes and leaves will be split as we walk down the 109997571fd0SChris Mason * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if 110097571fd0SChris Mason * possible) 110174123bd7SChris Mason */ 1102e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root 1103e089f05cSChris Mason *root, struct btrfs_key *key, struct btrfs_path *p, int 1104e089f05cSChris Mason ins_len, int cow) 1105be0e5c09SChris Mason { 11065f39d397SChris Mason struct extent_buffer *b; 1107db94535dSChris Mason u64 bytenr; 110874493f7aSChris Mason u64 ptr_gen; 1109be0e5c09SChris Mason int slot; 1110be0e5c09SChris Mason int ret; 1111be0e5c09SChris Mason int level; 11123c69faecSChris Mason int should_reada = p->reada; 11139f3a7427SChris Mason u8 lowest_level = 0; 11149f3a7427SChris Mason 11156702ed49SChris Mason lowest_level = p->lowest_level; 11166702ed49SChris Mason WARN_ON(lowest_level && ins_len); 111722b0ebdaSChris Mason WARN_ON(p->nodes[0] != NULL); 111822b0ebdaSChris Mason WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex)); 1119bb803951SChris Mason again: 1120bb803951SChris Mason b = root->node; 11215f39d397SChris Mason extent_buffer_get(b); 1122eb60ceacSChris Mason while (b) { 11235f39d397SChris Mason level = btrfs_header_level(b); 112402217ed2SChris Mason if (cow) { 112502217ed2SChris Mason int wret; 1126e20d96d6SChris Mason wret = btrfs_cow_block(trans, root, b, 1127e20d96d6SChris Mason p->nodes[level + 1], 1128e20d96d6SChris Mason p->slots[level + 1], 1129252c38f0SYan &b); 113054aa1f4dSChris Mason if (wret) { 11315f39d397SChris Mason free_extent_buffer(b); 113254aa1f4dSChris Mason return wret; 113354aa1f4dSChris Mason } 113402217ed2SChris Mason } 113502217ed2SChris Mason BUG_ON(!cow && ins_len); 11365f39d397SChris Mason if (level != btrfs_header_level(b)) 11372c90e5d6SChris Mason WARN_ON(1); 11385f39d397SChris Mason level = btrfs_header_level(b); 1139eb60ceacSChris Mason p->nodes[level] = b; 1140123abc88SChris Mason ret = check_block(root, p, level); 1141aa5d6bedSChris Mason if (ret) 1142aa5d6bedSChris Mason return -1; 11435f39d397SChris Mason ret = bin_search(b, key, level, &slot); 11445f39d397SChris Mason if (level != 0) { 1145be0e5c09SChris Mason if (ret && slot > 0) 1146be0e5c09SChris Mason slot -= 1; 1147be0e5c09SChris Mason p->slots[level] = slot; 11485f39d397SChris Mason if (ins_len > 0 && btrfs_header_nritems(b) >= 1149*1514794eSChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 3) { 1150e089f05cSChris Mason int sret = split_node(trans, root, p, level); 11515c680ed6SChris Mason BUG_ON(sret > 0); 11525c680ed6SChris Mason if (sret) 11535c680ed6SChris Mason return sret; 11545c680ed6SChris Mason b = p->nodes[level]; 11555c680ed6SChris Mason slot = p->slots[level]; 1156bb803951SChris Mason } else if (ins_len < 0) { 1157e089f05cSChris Mason int sret = balance_level(trans, root, p, 1158e089f05cSChris Mason level); 1159bb803951SChris Mason if (sret) 1160bb803951SChris Mason return sret; 1161bb803951SChris Mason b = p->nodes[level]; 1162f510cfecSChris Mason if (!b) { 1163f510cfecSChris Mason btrfs_release_path(NULL, p); 1164bb803951SChris Mason goto again; 1165f510cfecSChris Mason } 1166bb803951SChris Mason slot = p->slots[level]; 11675f39d397SChris Mason BUG_ON(btrfs_header_nritems(b) == 1); 11685c680ed6SChris Mason } 11699f3a7427SChris Mason /* this is only true while dropping a snapshot */ 11709f3a7427SChris Mason if (level == lowest_level) 11719f3a7427SChris Mason break; 1172db94535dSChris Mason bytenr = btrfs_node_blockptr(b, slot); 117374493f7aSChris Mason ptr_gen = btrfs_node_ptr_generation(b, slot); 11746702ed49SChris Mason if (should_reada) 117501f46658SChris Mason reada_for_search(root, p, level, slot, 117601f46658SChris Mason key->objectid); 1177db94535dSChris Mason b = read_tree_block(root, bytenr, 1178db94535dSChris Mason btrfs_level_size(root, level - 1)); 117974493f7aSChris Mason if (ptr_gen != btrfs_header_generation(b)) { 118074493f7aSChris Mason printk("block %llu bad gen wanted %llu " 118174493f7aSChris Mason "found %llu\n", 118274493f7aSChris Mason (unsigned long long)b->start, 118374493f7aSChris Mason (unsigned long long)ptr_gen, 118474493f7aSChris Mason (unsigned long long)btrfs_header_generation(b)); 118574493f7aSChris Mason } 1186be0e5c09SChris Mason } else { 1187be0e5c09SChris Mason p->slots[level] = slot; 11885f39d397SChris Mason if (ins_len > 0 && btrfs_leaf_free_space(root, b) < 11890783fcfcSChris Mason sizeof(struct btrfs_item) + ins_len) { 1190d4dbff95SChris Mason int sret = split_leaf(trans, root, key, 1191cc0c5538SChris Mason p, ins_len, ret == 0); 11925c680ed6SChris Mason BUG_ON(sret > 0); 11935c680ed6SChris Mason if (sret) 11945c680ed6SChris Mason return sret; 11955c680ed6SChris Mason } 1196be0e5c09SChris Mason return ret; 1197be0e5c09SChris Mason } 1198be0e5c09SChris Mason } 1199aa5d6bedSChris Mason return 1; 1200be0e5c09SChris Mason } 1201be0e5c09SChris Mason 120274123bd7SChris Mason /* 120374123bd7SChris Mason * adjust the pointers going up the tree, starting at level 120474123bd7SChris Mason * making sure the right key of each node is points to 'key'. 120574123bd7SChris Mason * This is used after shifting pointers to the left, so it stops 120674123bd7SChris Mason * fixing up pointers when a given leaf/node is not in slot 0 of the 120774123bd7SChris Mason * higher levels 1208aa5d6bedSChris Mason * 1209aa5d6bedSChris Mason * If this fails to write a tree block, it returns -1, but continues 1210aa5d6bedSChris Mason * fixing up the blocks in ram so the tree is consistent. 121174123bd7SChris Mason */ 12125f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans, 12135f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 12145f39d397SChris Mason struct btrfs_disk_key *key, int level) 1215be0e5c09SChris Mason { 1216be0e5c09SChris Mason int i; 1217aa5d6bedSChris Mason int ret = 0; 12185f39d397SChris Mason struct extent_buffer *t; 12195f39d397SChris Mason 1220234b63a0SChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 1221be0e5c09SChris Mason int tslot = path->slots[i]; 1222eb60ceacSChris Mason if (!path->nodes[i]) 1223be0e5c09SChris Mason break; 12245f39d397SChris Mason t = path->nodes[i]; 12255f39d397SChris Mason btrfs_set_node_key(t, key, tslot); 1226d6025579SChris Mason btrfs_mark_buffer_dirty(path->nodes[i]); 1227be0e5c09SChris Mason if (tslot != 0) 1228be0e5c09SChris Mason break; 1229be0e5c09SChris Mason } 1230aa5d6bedSChris Mason return ret; 1231be0e5c09SChris Mason } 1232be0e5c09SChris Mason 123374123bd7SChris Mason /* 123474123bd7SChris Mason * try to push data from one node into the next node left in the 123579f95c82SChris Mason * tree. 1236aa5d6bedSChris Mason * 1237aa5d6bedSChris Mason * returns 0 if some ptrs were pushed left, < 0 if there was some horrible 1238aa5d6bedSChris Mason * error, and > 0 if there was no room in the left hand block. 123974123bd7SChris Mason */ 124098ed5174SChris Mason static int push_node_left(struct btrfs_trans_handle *trans, 124198ed5174SChris Mason struct btrfs_root *root, struct extent_buffer *dst, 12425f39d397SChris Mason struct extent_buffer *src) 1243be0e5c09SChris Mason { 1244be0e5c09SChris Mason int push_items = 0; 1245bb803951SChris Mason int src_nritems; 1246bb803951SChris Mason int dst_nritems; 1247aa5d6bedSChris Mason int ret = 0; 1248be0e5c09SChris Mason 12495f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 12505f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1251123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 12527bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 12537bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 125454aa1f4dSChris Mason 1255eb60ceacSChris Mason if (push_items <= 0) { 1256be0e5c09SChris Mason return 1; 1257eb60ceacSChris Mason } 1258be0e5c09SChris Mason 1259be0e5c09SChris Mason if (src_nritems < push_items) 1260be0e5c09SChris Mason push_items = src_nritems; 126179f95c82SChris Mason 12625f39d397SChris Mason copy_extent_buffer(dst, src, 12635f39d397SChris Mason btrfs_node_key_ptr_offset(dst_nritems), 12645f39d397SChris Mason btrfs_node_key_ptr_offset(0), 1265123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 12665f39d397SChris Mason 1267bb803951SChris Mason if (push_items < src_nritems) { 12685f39d397SChris Mason memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0), 12695f39d397SChris Mason btrfs_node_key_ptr_offset(push_items), 1270e2fa7227SChris Mason (src_nritems - push_items) * 1271123abc88SChris Mason sizeof(struct btrfs_key_ptr)); 1272bb803951SChris Mason } 12735f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 12745f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 12755f39d397SChris Mason btrfs_mark_buffer_dirty(src); 12765f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 1277bb803951SChris Mason return ret; 1278be0e5c09SChris Mason } 1279be0e5c09SChris Mason 128097571fd0SChris Mason /* 128179f95c82SChris Mason * try to push data from one node into the next node right in the 128279f95c82SChris Mason * tree. 128379f95c82SChris Mason * 128479f95c82SChris Mason * returns 0 if some ptrs were pushed, < 0 if there was some horrible 128579f95c82SChris Mason * error, and > 0 if there was no room in the right hand block. 128679f95c82SChris Mason * 128779f95c82SChris Mason * this will only push up to 1/2 the contents of the left node over 128879f95c82SChris Mason */ 12895f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 12905f39d397SChris Mason struct btrfs_root *root, 12915f39d397SChris Mason struct extent_buffer *dst, 12925f39d397SChris Mason struct extent_buffer *src) 129379f95c82SChris Mason { 129479f95c82SChris Mason int push_items = 0; 129579f95c82SChris Mason int max_push; 129679f95c82SChris Mason int src_nritems; 129779f95c82SChris Mason int dst_nritems; 129879f95c82SChris Mason int ret = 0; 129979f95c82SChris Mason 13007bb86316SChris Mason WARN_ON(btrfs_header_generation(src) != trans->transid); 13017bb86316SChris Mason WARN_ON(btrfs_header_generation(dst) != trans->transid); 13027bb86316SChris Mason 13035f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 13045f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1305123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 13065f39d397SChris Mason if (push_items <= 0) 130779f95c82SChris Mason return 1; 130879f95c82SChris Mason 130979f95c82SChris Mason max_push = src_nritems / 2 + 1; 131079f95c82SChris Mason /* don't try to empty the node */ 1311252c38f0SYan if (max_push >= src_nritems) 131279f95c82SChris Mason return 1; 1313252c38f0SYan 131479f95c82SChris Mason if (max_push < push_items) 131579f95c82SChris Mason push_items = max_push; 131679f95c82SChris Mason 13175f39d397SChris Mason memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items), 13185f39d397SChris Mason btrfs_node_key_ptr_offset(0), 13195f39d397SChris Mason (dst_nritems) * 13205f39d397SChris Mason sizeof(struct btrfs_key_ptr)); 1321d6025579SChris Mason 13225f39d397SChris Mason copy_extent_buffer(dst, src, 13235f39d397SChris Mason btrfs_node_key_ptr_offset(0), 13245f39d397SChris Mason btrfs_node_key_ptr_offset(src_nritems - push_items), 1325123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 132679f95c82SChris Mason 13275f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 13285f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 132979f95c82SChris Mason 13305f39d397SChris Mason btrfs_mark_buffer_dirty(src); 13315f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 133279f95c82SChris Mason return ret; 133379f95c82SChris Mason } 133479f95c82SChris Mason 133579f95c82SChris Mason /* 133697571fd0SChris Mason * helper function to insert a new root level in the tree. 133797571fd0SChris Mason * A new node is allocated, and a single item is inserted to 133897571fd0SChris Mason * point to the existing root 1339aa5d6bedSChris Mason * 1340aa5d6bedSChris Mason * returns zero on success or < 0 on failure. 134197571fd0SChris Mason */ 134298ed5174SChris Mason static int noinline insert_new_root(struct btrfs_trans_handle *trans, 13435f39d397SChris Mason struct btrfs_root *root, 13445f39d397SChris Mason struct btrfs_path *path, int level) 134574123bd7SChris Mason { 13467bb86316SChris Mason u64 root_gen; 13477bb86316SChris Mason u64 lower_gen; 13485f39d397SChris Mason struct extent_buffer *lower; 13495f39d397SChris Mason struct extent_buffer *c; 13505f39d397SChris Mason struct btrfs_disk_key lower_key; 13515c680ed6SChris Mason 13525c680ed6SChris Mason BUG_ON(path->nodes[level]); 13535c680ed6SChris Mason BUG_ON(path->nodes[level-1] != root->node); 13545c680ed6SChris Mason 13557bb86316SChris Mason if (root->ref_cows) 13567bb86316SChris Mason root_gen = trans->transid; 13577bb86316SChris Mason else 13587bb86316SChris Mason root_gen = 0; 13597bb86316SChris Mason 13607bb86316SChris Mason lower = path->nodes[level-1]; 13617bb86316SChris Mason if (level == 1) 13627bb86316SChris Mason btrfs_item_key(lower, &lower_key, 0); 13637bb86316SChris Mason else 13647bb86316SChris Mason btrfs_node_key(lower, &lower_key, 0); 13657bb86316SChris Mason 13667bb86316SChris Mason c = __btrfs_alloc_free_block(trans, root, root->nodesize, 13677bb86316SChris Mason root->root_key.objectid, 13687bb86316SChris Mason root_gen, lower_key.objectid, level, 1369db94535dSChris Mason root->node->start, 0); 13705f39d397SChris Mason if (IS_ERR(c)) 13715f39d397SChris Mason return PTR_ERR(c); 13725f39d397SChris Mason memset_extent_buffer(c, 0, 0, root->nodesize); 13735f39d397SChris Mason btrfs_set_header_nritems(c, 1); 13745f39d397SChris Mason btrfs_set_header_level(c, level); 1375db94535dSChris Mason btrfs_set_header_bytenr(c, c->start); 13765f39d397SChris Mason btrfs_set_header_generation(c, trans->transid); 13775f39d397SChris Mason btrfs_set_header_owner(c, root->root_key.objectid); 1378d5719762SChris Mason 13795f39d397SChris Mason write_extent_buffer(c, root->fs_info->fsid, 13805f39d397SChris Mason (unsigned long)btrfs_header_fsid(c), 13815f39d397SChris Mason BTRFS_FSID_SIZE); 1382e17cade2SChris Mason 1383e17cade2SChris Mason write_extent_buffer(c, root->fs_info->chunk_tree_uuid, 1384e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(c), 1385e17cade2SChris Mason BTRFS_UUID_SIZE); 1386e17cade2SChris Mason 13875f39d397SChris Mason btrfs_set_node_key(c, &lower_key, 0); 1388db94535dSChris Mason btrfs_set_node_blockptr(c, 0, lower->start); 13897bb86316SChris Mason lower_gen = btrfs_header_generation(lower); 13907bb86316SChris Mason WARN_ON(lower_gen == 0); 13917bb86316SChris Mason 13927bb86316SChris Mason btrfs_set_node_ptr_generation(c, 0, lower_gen); 13935f39d397SChris Mason 13945f39d397SChris Mason btrfs_mark_buffer_dirty(c); 1395d5719762SChris Mason 1396cfaa7295SChris Mason /* the super has an extra ref to root->node */ 13975f39d397SChris Mason free_extent_buffer(root->node); 13985f39d397SChris Mason root->node = c; 13990b86a832SChris Mason add_root_to_dirty_list(root); 14005f39d397SChris Mason extent_buffer_get(c); 14015f39d397SChris Mason path->nodes[level] = c; 140274123bd7SChris Mason path->slots[level] = 0; 14037bb86316SChris Mason 14047bb86316SChris Mason if (root->ref_cows && lower_gen != trans->transid) { 14057bb86316SChris Mason struct btrfs_path *back_path = btrfs_alloc_path(); 14067bb86316SChris Mason int ret; 14077bb86316SChris Mason ret = btrfs_insert_extent_backref(trans, 14087bb86316SChris Mason root->fs_info->extent_root, 14097bb86316SChris Mason path, lower->start, 14107bb86316SChris Mason root->root_key.objectid, 14117bb86316SChris Mason trans->transid, 0, 0); 14127bb86316SChris Mason BUG_ON(ret); 14137bb86316SChris Mason btrfs_free_path(back_path); 14147bb86316SChris Mason } 141574123bd7SChris Mason return 0; 141674123bd7SChris Mason } 14175c680ed6SChris Mason 14185c680ed6SChris Mason /* 14195c680ed6SChris Mason * worker function to insert a single pointer in a node. 14205c680ed6SChris Mason * the node should have enough room for the pointer already 142197571fd0SChris Mason * 14225c680ed6SChris Mason * slot and level indicate where you want the key to go, and 14235c680ed6SChris Mason * blocknr is the block the key points to. 1424aa5d6bedSChris Mason * 1425aa5d6bedSChris Mason * returns zero on success and < 0 on any error 14265c680ed6SChris Mason */ 1427e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root 1428e089f05cSChris Mason *root, struct btrfs_path *path, struct btrfs_disk_key 1429db94535dSChris Mason *key, u64 bytenr, int slot, int level) 14305c680ed6SChris Mason { 14315f39d397SChris Mason struct extent_buffer *lower; 14325c680ed6SChris Mason int nritems; 14335c680ed6SChris Mason 14345c680ed6SChris Mason BUG_ON(!path->nodes[level]); 14355f39d397SChris Mason lower = path->nodes[level]; 14365f39d397SChris Mason nritems = btrfs_header_nritems(lower); 143774123bd7SChris Mason if (slot > nritems) 143874123bd7SChris Mason BUG(); 1439123abc88SChris Mason if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root)) 144074123bd7SChris Mason BUG(); 144174123bd7SChris Mason if (slot != nritems) { 14425f39d397SChris Mason memmove_extent_buffer(lower, 14435f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 14445f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 1445123abc88SChris Mason (nritems - slot) * sizeof(struct btrfs_key_ptr)); 144674123bd7SChris Mason } 14475f39d397SChris Mason btrfs_set_node_key(lower, key, slot); 1448db94535dSChris Mason btrfs_set_node_blockptr(lower, slot, bytenr); 144974493f7aSChris Mason WARN_ON(trans->transid == 0); 145074493f7aSChris Mason btrfs_set_node_ptr_generation(lower, slot, trans->transid); 14515f39d397SChris Mason btrfs_set_header_nritems(lower, nritems + 1); 14525f39d397SChris Mason btrfs_mark_buffer_dirty(lower); 145374123bd7SChris Mason return 0; 145474123bd7SChris Mason } 145574123bd7SChris Mason 145697571fd0SChris Mason /* 145797571fd0SChris Mason * split the node at the specified level in path in two. 145897571fd0SChris Mason * The path is corrected to point to the appropriate node after the split 145997571fd0SChris Mason * 146097571fd0SChris Mason * Before splitting this tries to make some room in the node by pushing 146197571fd0SChris Mason * left and right, if either one works, it returns right away. 1462aa5d6bedSChris Mason * 1463aa5d6bedSChris Mason * returns 0 on success and < 0 on failure 146497571fd0SChris Mason */ 1465e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 1466e089f05cSChris Mason *root, struct btrfs_path *path, int level) 1467be0e5c09SChris Mason { 14687bb86316SChris Mason u64 root_gen; 14695f39d397SChris Mason struct extent_buffer *c; 14705f39d397SChris Mason struct extent_buffer *split; 14715f39d397SChris Mason struct btrfs_disk_key disk_key; 1472be0e5c09SChris Mason int mid; 14735c680ed6SChris Mason int ret; 1474aa5d6bedSChris Mason int wret; 14757518a238SChris Mason u32 c_nritems; 1476be0e5c09SChris Mason 14775f39d397SChris Mason c = path->nodes[level]; 14787bb86316SChris Mason WARN_ON(btrfs_header_generation(c) != trans->transid); 14795f39d397SChris Mason if (c == root->node) { 14805c680ed6SChris Mason /* trying to split the root, lets make a new one */ 1481e089f05cSChris Mason ret = insert_new_root(trans, root, path, level + 1); 14825c680ed6SChris Mason if (ret) 14835c680ed6SChris Mason return ret; 1484e66f709bSChris Mason } else { 1485e66f709bSChris Mason ret = push_nodes_for_insert(trans, root, path, level); 14865f39d397SChris Mason c = path->nodes[level]; 14875f39d397SChris Mason if (!ret && btrfs_header_nritems(c) < 1488e66f709bSChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 1) 1489e66f709bSChris Mason return 0; 149054aa1f4dSChris Mason if (ret < 0) 149154aa1f4dSChris Mason return ret; 14925c680ed6SChris Mason } 1493e66f709bSChris Mason 14945f39d397SChris Mason c_nritems = btrfs_header_nritems(c); 14957bb86316SChris Mason if (root->ref_cows) 14967bb86316SChris Mason root_gen = trans->transid; 14977bb86316SChris Mason else 14987bb86316SChris Mason root_gen = 0; 14997bb86316SChris Mason 15007bb86316SChris Mason btrfs_node_key(c, &disk_key, 0); 15017bb86316SChris Mason split = __btrfs_alloc_free_block(trans, root, root->nodesize, 15027bb86316SChris Mason root->root_key.objectid, 15037bb86316SChris Mason root_gen, 15047bb86316SChris Mason btrfs_disk_key_objectid(&disk_key), 15057bb86316SChris Mason level, c->start, 0); 15065f39d397SChris Mason if (IS_ERR(split)) 15075f39d397SChris Mason return PTR_ERR(split); 150854aa1f4dSChris Mason 15095f39d397SChris Mason btrfs_set_header_flags(split, btrfs_header_flags(c)); 15105f39d397SChris Mason btrfs_set_header_level(split, btrfs_header_level(c)); 1511db94535dSChris Mason btrfs_set_header_bytenr(split, split->start); 15125f39d397SChris Mason btrfs_set_header_generation(split, trans->transid); 15135f39d397SChris Mason btrfs_set_header_owner(split, root->root_key.objectid); 151463b10fc4SChris Mason btrfs_set_header_flags(split, 0); 15155f39d397SChris Mason write_extent_buffer(split, root->fs_info->fsid, 15165f39d397SChris Mason (unsigned long)btrfs_header_fsid(split), 15175f39d397SChris Mason BTRFS_FSID_SIZE); 1518e17cade2SChris Mason write_extent_buffer(split, root->fs_info->chunk_tree_uuid, 1519e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(split), 1520e17cade2SChris Mason BTRFS_UUID_SIZE); 15215f39d397SChris Mason 15227518a238SChris Mason mid = (c_nritems + 1) / 2; 15235f39d397SChris Mason 15245f39d397SChris Mason copy_extent_buffer(split, c, 15255f39d397SChris Mason btrfs_node_key_ptr_offset(0), 15265f39d397SChris Mason btrfs_node_key_ptr_offset(mid), 1527123abc88SChris Mason (c_nritems - mid) * sizeof(struct btrfs_key_ptr)); 15285f39d397SChris Mason btrfs_set_header_nritems(split, c_nritems - mid); 15295f39d397SChris Mason btrfs_set_header_nritems(c, mid); 1530aa5d6bedSChris Mason ret = 0; 1531aa5d6bedSChris Mason 15325f39d397SChris Mason btrfs_mark_buffer_dirty(c); 15335f39d397SChris Mason btrfs_mark_buffer_dirty(split); 15345f39d397SChris Mason 15355f39d397SChris Mason btrfs_node_key(split, &disk_key, 0); 1536db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, split->start, 15375f39d397SChris Mason path->slots[level + 1] + 1, 1538123abc88SChris Mason level + 1); 1539aa5d6bedSChris Mason if (wret) 1540aa5d6bedSChris Mason ret = wret; 1541aa5d6bedSChris Mason 15425de08d7dSChris Mason if (path->slots[level] >= mid) { 15435c680ed6SChris Mason path->slots[level] -= mid; 15445f39d397SChris Mason free_extent_buffer(c); 15455f39d397SChris Mason path->nodes[level] = split; 15465c680ed6SChris Mason path->slots[level + 1] += 1; 1547eb60ceacSChris Mason } else { 15485f39d397SChris Mason free_extent_buffer(split); 1549be0e5c09SChris Mason } 1550aa5d6bedSChris Mason return ret; 1551be0e5c09SChris Mason } 1552be0e5c09SChris Mason 155374123bd7SChris Mason /* 155474123bd7SChris Mason * how many bytes are required to store the items in a leaf. start 155574123bd7SChris Mason * and nr indicate which items in the leaf to check. This totals up the 155674123bd7SChris Mason * space used both by the item structs and the item data 155774123bd7SChris Mason */ 15585f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr) 1559be0e5c09SChris Mason { 1560be0e5c09SChris Mason int data_len; 15615f39d397SChris Mason int nritems = btrfs_header_nritems(l); 1562d4dbff95SChris Mason int end = min(nritems, start + nr) - 1; 1563be0e5c09SChris Mason 1564be0e5c09SChris Mason if (!nr) 1565be0e5c09SChris Mason return 0; 15665f39d397SChris Mason data_len = btrfs_item_end_nr(l, start); 15675f39d397SChris Mason data_len = data_len - btrfs_item_offset_nr(l, end); 15680783fcfcSChris Mason data_len += sizeof(struct btrfs_item) * nr; 1569d4dbff95SChris Mason WARN_ON(data_len < 0); 1570be0e5c09SChris Mason return data_len; 1571be0e5c09SChris Mason } 1572be0e5c09SChris Mason 157374123bd7SChris Mason /* 1574d4dbff95SChris Mason * The space between the end of the leaf items and 1575d4dbff95SChris Mason * the start of the leaf data. IOW, how much room 1576d4dbff95SChris Mason * the leaf has left for both items and data 1577d4dbff95SChris Mason */ 15785f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf) 1579d4dbff95SChris Mason { 15805f39d397SChris Mason int nritems = btrfs_header_nritems(leaf); 15815f39d397SChris Mason int ret; 15825f39d397SChris Mason ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems); 15835f39d397SChris Mason if (ret < 0) { 15845f39d397SChris Mason printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n", 1585ae2f5411SJens Axboe ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root), 15865f39d397SChris Mason leaf_space_used(leaf, 0, nritems), nritems); 15875f39d397SChris Mason } 15885f39d397SChris Mason return ret; 1589d4dbff95SChris Mason } 1590d4dbff95SChris Mason 1591d4dbff95SChris Mason /* 159200ec4c51SChris Mason * push some data in the path leaf to the right, trying to free up at 159300ec4c51SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 1594aa5d6bedSChris Mason * 1595aa5d6bedSChris Mason * returns 1 if the push failed because the other node didn't have enough 1596aa5d6bedSChris Mason * room, 0 if everything worked out and < 0 if there were major errors. 159700ec4c51SChris Mason */ 1598e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root 159934a38218SChris Mason *root, struct btrfs_path *path, int data_size, 160034a38218SChris Mason int empty) 160100ec4c51SChris Mason { 16025f39d397SChris Mason struct extent_buffer *left = path->nodes[0]; 16035f39d397SChris Mason struct extent_buffer *right; 16045f39d397SChris Mason struct extent_buffer *upper; 16055f39d397SChris Mason struct btrfs_disk_key disk_key; 160600ec4c51SChris Mason int slot; 160734a38218SChris Mason u32 i; 160800ec4c51SChris Mason int free_space; 160900ec4c51SChris Mason int push_space = 0; 161000ec4c51SChris Mason int push_items = 0; 16110783fcfcSChris Mason struct btrfs_item *item; 16127518a238SChris Mason u32 left_nritems; 161334a38218SChris Mason u32 nr; 16147518a238SChris Mason u32 right_nritems; 16155f39d397SChris Mason u32 data_end; 1616db94535dSChris Mason u32 this_item_size; 161754aa1f4dSChris Mason int ret; 161800ec4c51SChris Mason 161900ec4c51SChris Mason slot = path->slots[1]; 162000ec4c51SChris Mason if (!path->nodes[1]) { 162100ec4c51SChris Mason return 1; 162200ec4c51SChris Mason } 162300ec4c51SChris Mason upper = path->nodes[1]; 16245f39d397SChris Mason if (slot >= btrfs_header_nritems(upper) - 1) 162500ec4c51SChris Mason return 1; 16265f39d397SChris Mason 1627db94535dSChris Mason right = read_tree_block(root, btrfs_node_blockptr(upper, slot + 1), 1628db94535dSChris Mason root->leafsize); 1629123abc88SChris Mason free_space = btrfs_leaf_free_space(root, right); 16300783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 16315f39d397SChris Mason free_extent_buffer(right); 163202217ed2SChris Mason return 1; 163302217ed2SChris Mason } 163402217ed2SChris Mason 16355f39d397SChris Mason /* cow and double check */ 16365f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, upper, 16375f39d397SChris Mason slot + 1, &right); 16385f39d397SChris Mason if (ret) { 16395f39d397SChris Mason free_extent_buffer(right); 1640a429e513SChris Mason return 1; 1641a429e513SChris Mason } 16425f39d397SChris Mason free_space = btrfs_leaf_free_space(root, right); 16435f39d397SChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 16445f39d397SChris Mason free_extent_buffer(right); 16455f39d397SChris Mason return 1; 16465f39d397SChris Mason } 16475f39d397SChris Mason 16485f39d397SChris Mason left_nritems = btrfs_header_nritems(left); 16495f39d397SChris Mason if (left_nritems == 0) { 16505f39d397SChris Mason free_extent_buffer(right); 16515f39d397SChris Mason return 1; 16525f39d397SChris Mason } 16535f39d397SChris Mason 165434a38218SChris Mason if (empty) 165534a38218SChris Mason nr = 0; 165634a38218SChris Mason else 165734a38218SChris Mason nr = 1; 165834a38218SChris Mason 165934a38218SChris Mason i = left_nritems - 1; 166034a38218SChris Mason while (i >= nr) { 16615f39d397SChris Mason item = btrfs_item_nr(left, i); 1662db94535dSChris Mason 166300ec4c51SChris Mason if (path->slots[0] == i) 166400ec4c51SChris Mason push_space += data_size + sizeof(*item); 1665db94535dSChris Mason 1666db94535dSChris Mason if (!left->map_token) { 1667db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 1668db94535dSChris Mason sizeof(struct btrfs_item), 1669db94535dSChris Mason &left->map_token, &left->kaddr, 1670db94535dSChris Mason &left->map_start, &left->map_len, 1671db94535dSChris Mason KM_USER1); 1672db94535dSChris Mason } 1673db94535dSChris Mason 1674db94535dSChris Mason this_item_size = btrfs_item_size(left, item); 1675db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 167600ec4c51SChris Mason break; 167700ec4c51SChris Mason push_items++; 1678db94535dSChris Mason push_space += this_item_size + sizeof(*item); 167934a38218SChris Mason if (i == 0) 168034a38218SChris Mason break; 168134a38218SChris Mason i--; 1682db94535dSChris Mason } 1683db94535dSChris Mason if (left->map_token) { 1684db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 1685db94535dSChris Mason left->map_token = NULL; 168600ec4c51SChris Mason } 16875f39d397SChris Mason 168800ec4c51SChris Mason if (push_items == 0) { 16895f39d397SChris Mason free_extent_buffer(right); 169000ec4c51SChris Mason return 1; 169100ec4c51SChris Mason } 16925f39d397SChris Mason 169334a38218SChris Mason if (!empty && push_items == left_nritems) 1694a429e513SChris Mason WARN_ON(1); 16955f39d397SChris Mason 169600ec4c51SChris Mason /* push left to right */ 16975f39d397SChris Mason right_nritems = btrfs_header_nritems(right); 169834a38218SChris Mason 16995f39d397SChris Mason push_space = btrfs_item_end_nr(left, left_nritems - push_items); 1700123abc88SChris Mason push_space -= leaf_data_end(root, left); 17015f39d397SChris Mason 170200ec4c51SChris Mason /* make room in the right data area */ 17035f39d397SChris Mason data_end = leaf_data_end(root, right); 17045f39d397SChris Mason memmove_extent_buffer(right, 17055f39d397SChris Mason btrfs_leaf_data(right) + data_end - push_space, 17065f39d397SChris Mason btrfs_leaf_data(right) + data_end, 17075f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root) - data_end); 17085f39d397SChris Mason 170900ec4c51SChris Mason /* copy from the left data area */ 17105f39d397SChris Mason copy_extent_buffer(right, left, btrfs_leaf_data(right) + 1711d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 1712d6025579SChris Mason btrfs_leaf_data(left) + leaf_data_end(root, left), 1713d6025579SChris Mason push_space); 17145f39d397SChris Mason 17155f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(push_items), 17165f39d397SChris Mason btrfs_item_nr_offset(0), 17170783fcfcSChris Mason right_nritems * sizeof(struct btrfs_item)); 17185f39d397SChris Mason 171900ec4c51SChris Mason /* copy the items from left to right */ 17205f39d397SChris Mason copy_extent_buffer(right, left, btrfs_item_nr_offset(0), 17215f39d397SChris Mason btrfs_item_nr_offset(left_nritems - push_items), 17220783fcfcSChris Mason push_items * sizeof(struct btrfs_item)); 172300ec4c51SChris Mason 172400ec4c51SChris Mason /* update the item pointers */ 17257518a238SChris Mason right_nritems += push_items; 17265f39d397SChris Mason btrfs_set_header_nritems(right, right_nritems); 1727123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 17287518a238SChris Mason for (i = 0; i < right_nritems; i++) { 17295f39d397SChris Mason item = btrfs_item_nr(right, i); 1730db94535dSChris Mason if (!right->map_token) { 1731db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1732db94535dSChris Mason sizeof(struct btrfs_item), 1733db94535dSChris Mason &right->map_token, &right->kaddr, 1734db94535dSChris Mason &right->map_start, &right->map_len, 1735db94535dSChris Mason KM_USER1); 1736db94535dSChris Mason } 1737db94535dSChris Mason push_space -= btrfs_item_size(right, item); 1738db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 1739db94535dSChris Mason } 1740db94535dSChris Mason 1741db94535dSChris Mason if (right->map_token) { 1742db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1743db94535dSChris Mason right->map_token = NULL; 174400ec4c51SChris Mason } 17457518a238SChris Mason left_nritems -= push_items; 17465f39d397SChris Mason btrfs_set_header_nritems(left, left_nritems); 174700ec4c51SChris Mason 174834a38218SChris Mason if (left_nritems) 17495f39d397SChris Mason btrfs_mark_buffer_dirty(left); 17505f39d397SChris Mason btrfs_mark_buffer_dirty(right); 1751a429e513SChris Mason 17525f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 17535f39d397SChris Mason btrfs_set_node_key(upper, &disk_key, slot + 1); 1754d6025579SChris Mason btrfs_mark_buffer_dirty(upper); 175502217ed2SChris Mason 175600ec4c51SChris Mason /* then fixup the leaf pointer in the path */ 17577518a238SChris Mason if (path->slots[0] >= left_nritems) { 17587518a238SChris Mason path->slots[0] -= left_nritems; 17595f39d397SChris Mason free_extent_buffer(path->nodes[0]); 17605f39d397SChris Mason path->nodes[0] = right; 176100ec4c51SChris Mason path->slots[1] += 1; 176200ec4c51SChris Mason } else { 17635f39d397SChris Mason free_extent_buffer(right); 176400ec4c51SChris Mason } 176500ec4c51SChris Mason return 0; 176600ec4c51SChris Mason } 176700ec4c51SChris Mason /* 176874123bd7SChris Mason * push some data in the path leaf to the left, trying to free up at 176974123bd7SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 177074123bd7SChris Mason */ 1771e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root 177234a38218SChris Mason *root, struct btrfs_path *path, int data_size, 177334a38218SChris Mason int empty) 1774be0e5c09SChris Mason { 17755f39d397SChris Mason struct btrfs_disk_key disk_key; 17765f39d397SChris Mason struct extent_buffer *right = path->nodes[0]; 17775f39d397SChris Mason struct extent_buffer *left; 1778be0e5c09SChris Mason int slot; 1779be0e5c09SChris Mason int i; 1780be0e5c09SChris Mason int free_space; 1781be0e5c09SChris Mason int push_space = 0; 1782be0e5c09SChris Mason int push_items = 0; 17830783fcfcSChris Mason struct btrfs_item *item; 17847518a238SChris Mason u32 old_left_nritems; 17855f39d397SChris Mason u32 right_nritems; 178634a38218SChris Mason u32 nr; 1787aa5d6bedSChris Mason int ret = 0; 1788aa5d6bedSChris Mason int wret; 1789db94535dSChris Mason u32 this_item_size; 1790db94535dSChris Mason u32 old_left_item_size; 1791be0e5c09SChris Mason 1792be0e5c09SChris Mason slot = path->slots[1]; 17935f39d397SChris Mason if (slot == 0) 1794be0e5c09SChris Mason return 1; 17955f39d397SChris Mason if (!path->nodes[1]) 1796be0e5c09SChris Mason return 1; 17975f39d397SChris Mason 17983685f791SChris Mason right_nritems = btrfs_header_nritems(right); 17993685f791SChris Mason if (right_nritems == 0) { 18003685f791SChris Mason return 1; 18013685f791SChris Mason } 18023685f791SChris Mason 18035f39d397SChris Mason left = read_tree_block(root, btrfs_node_blockptr(path->nodes[1], 1804db94535dSChris Mason slot - 1), root->leafsize); 1805123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 18060783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 18075f39d397SChris Mason free_extent_buffer(left); 1808be0e5c09SChris Mason return 1; 1809be0e5c09SChris Mason } 181002217ed2SChris Mason 181102217ed2SChris Mason /* cow and double check */ 18125f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, 18135f39d397SChris Mason path->nodes[1], slot - 1, &left); 181454aa1f4dSChris Mason if (ret) { 181554aa1f4dSChris Mason /* we hit -ENOSPC, but it isn't fatal here */ 18165f39d397SChris Mason free_extent_buffer(left); 181754aa1f4dSChris Mason return 1; 181854aa1f4dSChris Mason } 18193685f791SChris Mason 1820123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 18210783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 18225f39d397SChris Mason free_extent_buffer(left); 182302217ed2SChris Mason return 1; 182402217ed2SChris Mason } 182502217ed2SChris Mason 182634a38218SChris Mason if (empty) 182734a38218SChris Mason nr = right_nritems; 182834a38218SChris Mason else 182934a38218SChris Mason nr = right_nritems - 1; 183034a38218SChris Mason 183134a38218SChris Mason for (i = 0; i < nr; i++) { 18325f39d397SChris Mason item = btrfs_item_nr(right, i); 1833db94535dSChris Mason if (!right->map_token) { 1834db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1835db94535dSChris Mason sizeof(struct btrfs_item), 1836db94535dSChris Mason &right->map_token, &right->kaddr, 1837db94535dSChris Mason &right->map_start, &right->map_len, 1838db94535dSChris Mason KM_USER1); 1839db94535dSChris Mason } 1840db94535dSChris Mason 1841be0e5c09SChris Mason if (path->slots[0] == i) 1842be0e5c09SChris Mason push_space += data_size + sizeof(*item); 1843db94535dSChris Mason 1844db94535dSChris Mason this_item_size = btrfs_item_size(right, item); 1845db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 1846be0e5c09SChris Mason break; 1847db94535dSChris Mason 1848be0e5c09SChris Mason push_items++; 1849db94535dSChris Mason push_space += this_item_size + sizeof(*item); 1850be0e5c09SChris Mason } 1851db94535dSChris Mason 1852db94535dSChris Mason if (right->map_token) { 1853db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1854db94535dSChris Mason right->map_token = NULL; 1855db94535dSChris Mason } 1856db94535dSChris Mason 1857be0e5c09SChris Mason if (push_items == 0) { 18585f39d397SChris Mason free_extent_buffer(left); 1859be0e5c09SChris Mason return 1; 1860be0e5c09SChris Mason } 186134a38218SChris Mason if (!empty && push_items == btrfs_header_nritems(right)) 1862a429e513SChris Mason WARN_ON(1); 18635f39d397SChris Mason 1864be0e5c09SChris Mason /* push data from right to left */ 18655f39d397SChris Mason copy_extent_buffer(left, right, 18665f39d397SChris Mason btrfs_item_nr_offset(btrfs_header_nritems(left)), 18675f39d397SChris Mason btrfs_item_nr_offset(0), 18685f39d397SChris Mason push_items * sizeof(struct btrfs_item)); 18695f39d397SChris Mason 1870123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root) - 18715f39d397SChris Mason btrfs_item_offset_nr(right, push_items -1); 18725f39d397SChris Mason 18735f39d397SChris Mason copy_extent_buffer(left, right, btrfs_leaf_data(left) + 1874d6025579SChris Mason leaf_data_end(root, left) - push_space, 1875123abc88SChris Mason btrfs_leaf_data(right) + 18765f39d397SChris Mason btrfs_item_offset_nr(right, push_items - 1), 1877be0e5c09SChris Mason push_space); 18785f39d397SChris Mason old_left_nritems = btrfs_header_nritems(left); 1879eb60ceacSChris Mason BUG_ON(old_left_nritems < 0); 1880eb60ceacSChris Mason 1881db94535dSChris Mason old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1); 1882be0e5c09SChris Mason for (i = old_left_nritems; i < old_left_nritems + push_items; i++) { 18835f39d397SChris Mason u32 ioff; 1884db94535dSChris Mason 18855f39d397SChris Mason item = btrfs_item_nr(left, i); 1886db94535dSChris Mason if (!left->map_token) { 1887db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 1888db94535dSChris Mason sizeof(struct btrfs_item), 1889db94535dSChris Mason &left->map_token, &left->kaddr, 1890db94535dSChris Mason &left->map_start, &left->map_len, 1891db94535dSChris Mason KM_USER1); 1892db94535dSChris Mason } 1893db94535dSChris Mason 18945f39d397SChris Mason ioff = btrfs_item_offset(left, item); 18955f39d397SChris Mason btrfs_set_item_offset(left, item, 1896db94535dSChris Mason ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size)); 1897be0e5c09SChris Mason } 18985f39d397SChris Mason btrfs_set_header_nritems(left, old_left_nritems + push_items); 1899db94535dSChris Mason if (left->map_token) { 1900db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 1901db94535dSChris Mason left->map_token = NULL; 1902db94535dSChris Mason } 1903be0e5c09SChris Mason 1904be0e5c09SChris Mason /* fixup right node */ 190534a38218SChris Mason if (push_items > right_nritems) { 190634a38218SChris Mason printk("push items %d nr %u\n", push_items, right_nritems); 190734a38218SChris Mason WARN_ON(1); 190834a38218SChris Mason } 190934a38218SChris Mason 191034a38218SChris Mason if (push_items < right_nritems) { 19115f39d397SChris Mason push_space = btrfs_item_offset_nr(right, push_items - 1) - 1912123abc88SChris Mason leaf_data_end(root, right); 19135f39d397SChris Mason memmove_extent_buffer(right, btrfs_leaf_data(right) + 1914d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 1915d6025579SChris Mason btrfs_leaf_data(right) + 1916123abc88SChris Mason leaf_data_end(root, right), push_space); 19175f39d397SChris Mason 19185f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(0), 19195f39d397SChris Mason btrfs_item_nr_offset(push_items), 19205f39d397SChris Mason (btrfs_header_nritems(right) - push_items) * 19210783fcfcSChris Mason sizeof(struct btrfs_item)); 192234a38218SChris Mason } 1923eef1c494SYan right_nritems -= push_items; 1924eef1c494SYan btrfs_set_header_nritems(right, right_nritems); 1925123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 19265f39d397SChris Mason for (i = 0; i < right_nritems; i++) { 19275f39d397SChris Mason item = btrfs_item_nr(right, i); 1928db94535dSChris Mason 1929db94535dSChris Mason if (!right->map_token) { 1930db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1931db94535dSChris Mason sizeof(struct btrfs_item), 1932db94535dSChris Mason &right->map_token, &right->kaddr, 1933db94535dSChris Mason &right->map_start, &right->map_len, 1934db94535dSChris Mason KM_USER1); 1935db94535dSChris Mason } 1936db94535dSChris Mason 1937db94535dSChris Mason push_space = push_space - btrfs_item_size(right, item); 1938db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 1939db94535dSChris Mason } 1940db94535dSChris Mason if (right->map_token) { 1941db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1942db94535dSChris Mason right->map_token = NULL; 1943be0e5c09SChris Mason } 1944eb60ceacSChris Mason 19455f39d397SChris Mason btrfs_mark_buffer_dirty(left); 194634a38218SChris Mason if (right_nritems) 19475f39d397SChris Mason btrfs_mark_buffer_dirty(right); 1948098f59c2SChris Mason 19495f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 19505f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, 1); 1951aa5d6bedSChris Mason if (wret) 1952aa5d6bedSChris Mason ret = wret; 1953be0e5c09SChris Mason 1954be0e5c09SChris Mason /* then fixup the leaf pointer in the path */ 1955be0e5c09SChris Mason if (path->slots[0] < push_items) { 1956be0e5c09SChris Mason path->slots[0] += old_left_nritems; 19575f39d397SChris Mason free_extent_buffer(path->nodes[0]); 19585f39d397SChris Mason path->nodes[0] = left; 1959be0e5c09SChris Mason path->slots[1] -= 1; 1960be0e5c09SChris Mason } else { 19615f39d397SChris Mason free_extent_buffer(left); 1962be0e5c09SChris Mason path->slots[0] -= push_items; 1963be0e5c09SChris Mason } 1964eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 1965aa5d6bedSChris Mason return ret; 1966be0e5c09SChris Mason } 1967be0e5c09SChris Mason 196874123bd7SChris Mason /* 196974123bd7SChris Mason * split the path's leaf in two, making sure there is at least data_size 197074123bd7SChris Mason * available for the resulting leaf level of the path. 1971aa5d6bedSChris Mason * 1972aa5d6bedSChris Mason * returns 0 if all went well and < 0 on failure. 197374123bd7SChris Mason */ 1974e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 1975d4dbff95SChris Mason *root, struct btrfs_key *ins_key, 1976cc0c5538SChris Mason struct btrfs_path *path, int data_size, int extend) 1977be0e5c09SChris Mason { 19787bb86316SChris Mason u64 root_gen; 19795f39d397SChris Mason struct extent_buffer *l; 19807518a238SChris Mason u32 nritems; 1981eb60ceacSChris Mason int mid; 1982eb60ceacSChris Mason int slot; 19835f39d397SChris Mason struct extent_buffer *right; 19840783fcfcSChris Mason int space_needed = data_size + sizeof(struct btrfs_item); 1985be0e5c09SChris Mason int data_copy_size; 1986be0e5c09SChris Mason int rt_data_off; 1987be0e5c09SChris Mason int i; 1988d4dbff95SChris Mason int ret = 0; 1989aa5d6bedSChris Mason int wret; 1990cc0c5538SChris Mason int double_split; 1991cc0c5538SChris Mason int num_doubles = 0; 1992d4dbff95SChris Mason struct btrfs_disk_key disk_key; 1993be0e5c09SChris Mason 1994cc0c5538SChris Mason if (extend) 1995cc0c5538SChris Mason space_needed = data_size; 1996cc0c5538SChris Mason 19977bb86316SChris Mason if (root->ref_cows) 19987bb86316SChris Mason root_gen = trans->transid; 19997bb86316SChris Mason else 20007bb86316SChris Mason root_gen = 0; 20017bb86316SChris Mason 200240689478SChris Mason /* first try to make some room by pushing left and right */ 20033685f791SChris Mason if (ins_key->type != BTRFS_DIR_ITEM_KEY) { 200434a38218SChris Mason wret = push_leaf_right(trans, root, path, data_size, 0); 20053326d1b0SChris Mason if (wret < 0) { 2006eaee50e8SChris Mason return wret; 20073326d1b0SChris Mason } 2008eaee50e8SChris Mason if (wret) { 200934a38218SChris Mason wret = push_leaf_left(trans, root, path, data_size, 0); 2010eaee50e8SChris Mason if (wret < 0) 2011eaee50e8SChris Mason return wret; 2012eaee50e8SChris Mason } 20135f39d397SChris Mason l = path->nodes[0]; 2014aa5d6bedSChris Mason 2015aa5d6bedSChris Mason /* did the pushes work? */ 2016cc0c5538SChris Mason if (btrfs_leaf_free_space(root, l) >= space_needed) 2017be0e5c09SChris Mason return 0; 20183326d1b0SChris Mason } 2019aa5d6bedSChris Mason 20205c680ed6SChris Mason if (!path->nodes[1]) { 2021e089f05cSChris Mason ret = insert_new_root(trans, root, path, 1); 20225c680ed6SChris Mason if (ret) 20235c680ed6SChris Mason return ret; 20245c680ed6SChris Mason } 2025cc0c5538SChris Mason again: 2026cc0c5538SChris Mason double_split = 0; 2027cc0c5538SChris Mason l = path->nodes[0]; 2028eb60ceacSChris Mason slot = path->slots[0]; 20295f39d397SChris Mason nritems = btrfs_header_nritems(l); 2030eb60ceacSChris Mason mid = (nritems + 1)/ 2; 203154aa1f4dSChris Mason 20327bb86316SChris Mason btrfs_item_key(l, &disk_key, 0); 20337bb86316SChris Mason 20347bb86316SChris Mason right = __btrfs_alloc_free_block(trans, root, root->leafsize, 20357bb86316SChris Mason root->root_key.objectid, 20367bb86316SChris Mason root_gen, disk_key.objectid, 0, 2037db94535dSChris Mason l->start, 0); 2038cea9e445SChris Mason if (IS_ERR(right)) { 2039cea9e445SChris Mason BUG_ON(1); 20405f39d397SChris Mason return PTR_ERR(right); 2041cea9e445SChris Mason } 204254aa1f4dSChris Mason 20435f39d397SChris Mason memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header)); 2044db94535dSChris Mason btrfs_set_header_bytenr(right, right->start); 20455f39d397SChris Mason btrfs_set_header_generation(right, trans->transid); 20465f39d397SChris Mason btrfs_set_header_owner(right, root->root_key.objectid); 20475f39d397SChris Mason btrfs_set_header_level(right, 0); 20485f39d397SChris Mason write_extent_buffer(right, root->fs_info->fsid, 20495f39d397SChris Mason (unsigned long)btrfs_header_fsid(right), 20505f39d397SChris Mason BTRFS_FSID_SIZE); 2051e17cade2SChris Mason 2052e17cade2SChris Mason write_extent_buffer(right, root->fs_info->chunk_tree_uuid, 2053e17cade2SChris Mason (unsigned long)btrfs_header_chunk_tree_uuid(right), 2054e17cade2SChris Mason BTRFS_UUID_SIZE); 2055d4dbff95SChris Mason if (mid <= slot) { 2056d4dbff95SChris Mason if (nritems == 1 || 2057d4dbff95SChris Mason leaf_space_used(l, mid, nritems - mid) + space_needed > 2058d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 2059d4dbff95SChris Mason if (slot >= nritems) { 2060d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 20615f39d397SChris Mason btrfs_set_header_nritems(right, 0); 2062d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 2063db94535dSChris Mason &disk_key, right->start, 2064d4dbff95SChris Mason path->slots[1] + 1, 1); 2065d4dbff95SChris Mason if (wret) 2066d4dbff95SChris Mason ret = wret; 20675f39d397SChris Mason free_extent_buffer(path->nodes[0]); 20685f39d397SChris Mason path->nodes[0] = right; 2069d4dbff95SChris Mason path->slots[0] = 0; 2070d4dbff95SChris Mason path->slots[1] += 1; 20710ef8b242SChris Mason btrfs_mark_buffer_dirty(right); 2072d4dbff95SChris Mason return ret; 2073d4dbff95SChris Mason } 2074d4dbff95SChris Mason mid = slot; 20753326d1b0SChris Mason if (mid != nritems && 20763326d1b0SChris Mason leaf_space_used(l, mid, nritems - mid) + 20773326d1b0SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 2078d4dbff95SChris Mason double_split = 1; 2079d4dbff95SChris Mason } 20803326d1b0SChris Mason } 2081d4dbff95SChris Mason } else { 2082d4dbff95SChris Mason if (leaf_space_used(l, 0, mid + 1) + space_needed > 2083d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 2084cc0c5538SChris Mason if (!extend && slot == 0) { 2085d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 20865f39d397SChris Mason btrfs_set_header_nritems(right, 0); 2087d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 2088d4dbff95SChris Mason &disk_key, 2089db94535dSChris Mason right->start, 2090098f59c2SChris Mason path->slots[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; 2096a429e513SChris Mason if (path->slots[1] == 0) { 2097a429e513SChris Mason wret = fixup_low_keys(trans, root, 2098a429e513SChris Mason path, &disk_key, 1); 2099a429e513SChris Mason if (wret) 2100a429e513SChris Mason ret = wret; 2101a429e513SChris Mason } 21020ef8b242SChris Mason btrfs_mark_buffer_dirty(right); 2103d4dbff95SChris Mason return ret; 2104cc0c5538SChris Mason } else if (extend && slot == 0) { 2105cc0c5538SChris Mason mid = 1; 2106cc0c5538SChris Mason } else { 2107d4dbff95SChris Mason mid = slot; 21085ee78ac7SChris Mason if (mid != nritems && 21095ee78ac7SChris Mason leaf_space_used(l, mid, nritems - mid) + 21105ee78ac7SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 2111d4dbff95SChris Mason double_split = 1; 2112d4dbff95SChris Mason } 2113d4dbff95SChris Mason } 21145ee78ac7SChris Mason } 2115cc0c5538SChris Mason } 21165f39d397SChris Mason nritems = nritems - mid; 21175f39d397SChris Mason btrfs_set_header_nritems(right, nritems); 21185f39d397SChris Mason data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l); 21195f39d397SChris Mason 21205f39d397SChris Mason copy_extent_buffer(right, l, btrfs_item_nr_offset(0), 21215f39d397SChris Mason btrfs_item_nr_offset(mid), 21225f39d397SChris Mason nritems * sizeof(struct btrfs_item)); 21235f39d397SChris Mason 21245f39d397SChris Mason copy_extent_buffer(right, l, 2125d6025579SChris Mason btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - 2126123abc88SChris Mason data_copy_size, btrfs_leaf_data(l) + 2127123abc88SChris Mason leaf_data_end(root, l), data_copy_size); 212874123bd7SChris Mason 21295f39d397SChris Mason rt_data_off = BTRFS_LEAF_DATA_SIZE(root) - 21305f39d397SChris Mason btrfs_item_end_nr(l, mid); 21315f39d397SChris Mason 21325f39d397SChris Mason for (i = 0; i < nritems; i++) { 21335f39d397SChris Mason struct btrfs_item *item = btrfs_item_nr(right, i); 2134db94535dSChris Mason u32 ioff; 2135db94535dSChris Mason 2136db94535dSChris Mason if (!right->map_token) { 2137db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 2138db94535dSChris Mason sizeof(struct btrfs_item), 2139db94535dSChris Mason &right->map_token, &right->kaddr, 2140db94535dSChris Mason &right->map_start, &right->map_len, 2141db94535dSChris Mason KM_USER1); 2142db94535dSChris Mason } 2143db94535dSChris Mason 2144db94535dSChris Mason ioff = btrfs_item_offset(right, item); 21455f39d397SChris Mason btrfs_set_item_offset(right, item, ioff + rt_data_off); 21460783fcfcSChris Mason } 214774123bd7SChris Mason 2148db94535dSChris Mason if (right->map_token) { 2149db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 2150db94535dSChris Mason right->map_token = NULL; 2151db94535dSChris Mason } 2152db94535dSChris Mason 21535f39d397SChris Mason btrfs_set_header_nritems(l, mid); 2154aa5d6bedSChris Mason ret = 0; 21555f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 2156db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, right->start, 2157db94535dSChris Mason path->slots[1] + 1, 1); 2158aa5d6bedSChris Mason if (wret) 2159aa5d6bedSChris Mason ret = wret; 21605f39d397SChris Mason 21615f39d397SChris Mason btrfs_mark_buffer_dirty(right); 21625f39d397SChris Mason btrfs_mark_buffer_dirty(l); 2163eb60ceacSChris Mason BUG_ON(path->slots[0] != slot); 21645f39d397SChris Mason 2165be0e5c09SChris Mason if (mid <= slot) { 21665f39d397SChris Mason free_extent_buffer(path->nodes[0]); 21675f39d397SChris Mason path->nodes[0] = right; 2168be0e5c09SChris Mason path->slots[0] -= mid; 2169be0e5c09SChris Mason path->slots[1] += 1; 2170eb60ceacSChris Mason } else 21715f39d397SChris Mason free_extent_buffer(right); 21725f39d397SChris Mason 2173eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 2174d4dbff95SChris Mason 2175cc0c5538SChris Mason if (double_split) { 2176cc0c5538SChris Mason BUG_ON(num_doubles != 0); 2177cc0c5538SChris Mason num_doubles++; 2178cc0c5538SChris Mason goto again; 21793326d1b0SChris Mason } 2180be0e5c09SChris Mason return ret; 2181be0e5c09SChris Mason } 2182be0e5c09SChris Mason 2183b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans, 2184b18c6685SChris Mason struct btrfs_root *root, 2185b18c6685SChris Mason struct btrfs_path *path, 2186179e29e4SChris Mason u32 new_size, int from_end) 2187b18c6685SChris Mason { 2188b18c6685SChris Mason int ret = 0; 2189b18c6685SChris Mason int slot; 2190b18c6685SChris Mason int slot_orig; 21915f39d397SChris Mason struct extent_buffer *leaf; 21925f39d397SChris Mason struct btrfs_item *item; 2193b18c6685SChris Mason u32 nritems; 2194b18c6685SChris Mason unsigned int data_end; 2195b18c6685SChris Mason unsigned int old_data_start; 2196b18c6685SChris Mason unsigned int old_size; 2197b18c6685SChris Mason unsigned int size_diff; 2198b18c6685SChris Mason int i; 2199b18c6685SChris Mason 2200b18c6685SChris Mason slot_orig = path->slots[0]; 22015f39d397SChris Mason leaf = path->nodes[0]; 2202179e29e4SChris Mason slot = path->slots[0]; 2203179e29e4SChris Mason 2204179e29e4SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 2205179e29e4SChris Mason if (old_size == new_size) 2206179e29e4SChris Mason return 0; 2207b18c6685SChris Mason 22085f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2209b18c6685SChris Mason data_end = leaf_data_end(root, leaf); 2210b18c6685SChris Mason 22115f39d397SChris Mason old_data_start = btrfs_item_offset_nr(leaf, slot); 2212179e29e4SChris Mason 2213b18c6685SChris Mason size_diff = old_size - new_size; 2214b18c6685SChris Mason 2215b18c6685SChris Mason BUG_ON(slot < 0); 2216b18c6685SChris Mason BUG_ON(slot >= nritems); 2217b18c6685SChris Mason 2218b18c6685SChris Mason /* 2219b18c6685SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2220b18c6685SChris Mason */ 2221b18c6685SChris Mason /* first correct the data pointers */ 2222b18c6685SChris Mason for (i = slot; i < nritems; i++) { 22235f39d397SChris Mason u32 ioff; 22245f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2225db94535dSChris Mason 2226db94535dSChris Mason if (!leaf->map_token) { 2227db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2228db94535dSChris Mason sizeof(struct btrfs_item), 2229db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2230db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2231db94535dSChris Mason KM_USER1); 2232db94535dSChris Mason } 2233db94535dSChris Mason 22345f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 22355f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + size_diff); 2236b18c6685SChris Mason } 2237db94535dSChris Mason 2238db94535dSChris Mason if (leaf->map_token) { 2239db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2240db94535dSChris Mason leaf->map_token = NULL; 2241db94535dSChris Mason } 2242db94535dSChris Mason 2243b18c6685SChris Mason /* shift the data */ 2244179e29e4SChris Mason if (from_end) { 22455f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2246b18c6685SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 2247b18c6685SChris Mason data_end, old_data_start + new_size - data_end); 2248179e29e4SChris Mason } else { 2249179e29e4SChris Mason struct btrfs_disk_key disk_key; 2250179e29e4SChris Mason u64 offset; 2251179e29e4SChris Mason 2252179e29e4SChris Mason btrfs_item_key(leaf, &disk_key, slot); 2253179e29e4SChris Mason 2254179e29e4SChris Mason if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) { 2255179e29e4SChris Mason unsigned long ptr; 2256179e29e4SChris Mason struct btrfs_file_extent_item *fi; 2257179e29e4SChris Mason 2258179e29e4SChris Mason fi = btrfs_item_ptr(leaf, slot, 2259179e29e4SChris Mason struct btrfs_file_extent_item); 2260179e29e4SChris Mason fi = (struct btrfs_file_extent_item *)( 2261179e29e4SChris Mason (unsigned long)fi - size_diff); 2262179e29e4SChris Mason 2263179e29e4SChris Mason if (btrfs_file_extent_type(leaf, fi) == 2264179e29e4SChris Mason BTRFS_FILE_EXTENT_INLINE) { 2265179e29e4SChris Mason ptr = btrfs_item_ptr_offset(leaf, slot); 2266179e29e4SChris Mason memmove_extent_buffer(leaf, ptr, 2267179e29e4SChris Mason (unsigned long)fi, 2268179e29e4SChris Mason offsetof(struct btrfs_file_extent_item, 2269179e29e4SChris Mason disk_bytenr)); 2270179e29e4SChris Mason } 2271179e29e4SChris Mason } 2272179e29e4SChris Mason 2273179e29e4SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2274179e29e4SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 2275179e29e4SChris Mason data_end, old_data_start - data_end); 2276179e29e4SChris Mason 2277179e29e4SChris Mason offset = btrfs_disk_key_offset(&disk_key); 2278179e29e4SChris Mason btrfs_set_disk_key_offset(&disk_key, offset + size_diff); 2279179e29e4SChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 2280179e29e4SChris Mason if (slot == 0) 2281179e29e4SChris Mason fixup_low_keys(trans, root, path, &disk_key, 1); 2282179e29e4SChris Mason } 22835f39d397SChris Mason 22845f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 22855f39d397SChris Mason btrfs_set_item_size(leaf, item, new_size); 22865f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2287b18c6685SChris Mason 2288b18c6685SChris Mason ret = 0; 22895f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 22905f39d397SChris Mason btrfs_print_leaf(root, leaf); 2291b18c6685SChris Mason BUG(); 22925f39d397SChris Mason } 2293b18c6685SChris Mason return ret; 2294b18c6685SChris Mason } 2295b18c6685SChris Mason 22965f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans, 22975f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 22985f39d397SChris Mason u32 data_size) 22996567e837SChris Mason { 23006567e837SChris Mason int ret = 0; 23016567e837SChris Mason int slot; 23026567e837SChris Mason int slot_orig; 23035f39d397SChris Mason struct extent_buffer *leaf; 23045f39d397SChris Mason struct btrfs_item *item; 23056567e837SChris Mason u32 nritems; 23066567e837SChris Mason unsigned int data_end; 23076567e837SChris Mason unsigned int old_data; 23086567e837SChris Mason unsigned int old_size; 23096567e837SChris Mason int i; 23106567e837SChris Mason 23116567e837SChris Mason slot_orig = path->slots[0]; 23125f39d397SChris Mason leaf = path->nodes[0]; 23136567e837SChris Mason 23145f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 23156567e837SChris Mason data_end = leaf_data_end(root, leaf); 23166567e837SChris Mason 23175f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < data_size) { 23185f39d397SChris Mason btrfs_print_leaf(root, leaf); 23196567e837SChris Mason BUG(); 23205f39d397SChris Mason } 23216567e837SChris Mason slot = path->slots[0]; 23225f39d397SChris Mason old_data = btrfs_item_end_nr(leaf, slot); 23236567e837SChris Mason 23246567e837SChris Mason BUG_ON(slot < 0); 23253326d1b0SChris Mason if (slot >= nritems) { 23263326d1b0SChris Mason btrfs_print_leaf(root, leaf); 23273326d1b0SChris Mason printk("slot %d too large, nritems %d\n", slot, nritems); 23283326d1b0SChris Mason BUG_ON(1); 23293326d1b0SChris Mason } 23306567e837SChris Mason 23316567e837SChris Mason /* 23326567e837SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 23336567e837SChris Mason */ 23346567e837SChris Mason /* first correct the data pointers */ 23356567e837SChris Mason for (i = slot; i < nritems; i++) { 23365f39d397SChris Mason u32 ioff; 23375f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2338db94535dSChris Mason 2339db94535dSChris Mason if (!leaf->map_token) { 2340db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2341db94535dSChris Mason sizeof(struct btrfs_item), 2342db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2343db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2344db94535dSChris Mason KM_USER1); 2345db94535dSChris Mason } 23465f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 23475f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff - data_size); 23486567e837SChris Mason } 23495f39d397SChris Mason 2350db94535dSChris Mason if (leaf->map_token) { 2351db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2352db94535dSChris Mason leaf->map_token = NULL; 2353db94535dSChris Mason } 2354db94535dSChris Mason 23556567e837SChris Mason /* shift the data */ 23565f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 23576567e837SChris Mason data_end - data_size, btrfs_leaf_data(leaf) + 23586567e837SChris Mason data_end, old_data - data_end); 23595f39d397SChris Mason 23606567e837SChris Mason data_end = old_data; 23615f39d397SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 23625f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 23635f39d397SChris Mason btrfs_set_item_size(leaf, item, old_size + data_size); 23645f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 23656567e837SChris Mason 23666567e837SChris Mason ret = 0; 23675f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 23685f39d397SChris Mason btrfs_print_leaf(root, leaf); 23696567e837SChris Mason BUG(); 23705f39d397SChris Mason } 23716567e837SChris Mason return ret; 23726567e837SChris Mason } 23736567e837SChris Mason 237474123bd7SChris Mason /* 237574123bd7SChris Mason * Given a key and some data, insert an item into the tree. 237674123bd7SChris Mason * This does all the path init required, making room in the tree if needed. 237774123bd7SChris Mason */ 23789c58309dSChris Mason int btrfs_insert_empty_items(struct btrfs_trans_handle *trans, 23795f39d397SChris Mason struct btrfs_root *root, 23805f39d397SChris Mason struct btrfs_path *path, 23819c58309dSChris Mason struct btrfs_key *cpu_key, u32 *data_size, 23829c58309dSChris Mason int nr) 2383be0e5c09SChris Mason { 23845f39d397SChris Mason struct extent_buffer *leaf; 23855f39d397SChris Mason struct btrfs_item *item; 2386aa5d6bedSChris Mason int ret = 0; 2387be0e5c09SChris Mason int slot; 2388eb60ceacSChris Mason int slot_orig; 23899c58309dSChris Mason int i; 23907518a238SChris Mason u32 nritems; 23919c58309dSChris Mason u32 total_size = 0; 23929c58309dSChris Mason u32 total_data = 0; 2393be0e5c09SChris Mason unsigned int data_end; 2394e2fa7227SChris Mason struct btrfs_disk_key disk_key; 2395e2fa7227SChris Mason 23969c58309dSChris Mason for (i = 0; i < nr; i++) { 23979c58309dSChris Mason total_data += data_size[i]; 23989c58309dSChris Mason } 2399be0e5c09SChris Mason 240074123bd7SChris Mason /* create a root if there isn't one */ 24015c680ed6SChris Mason if (!root->node) 2402cfaa7295SChris Mason BUG(); 24035f39d397SChris Mason 24049c58309dSChris Mason total_size = total_data + (nr - 1) * sizeof(struct btrfs_item); 24059c58309dSChris Mason ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1); 2406eb60ceacSChris Mason if (ret == 0) { 2407f0930a37SChris Mason return -EEXIST; 2408aa5d6bedSChris Mason } 2409ed2ff2cbSChris Mason if (ret < 0) 2410ed2ff2cbSChris Mason goto out; 2411be0e5c09SChris Mason 241262e2749eSChris Mason slot_orig = path->slots[0]; 24135f39d397SChris Mason leaf = path->nodes[0]; 241474123bd7SChris Mason 24155f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2416123abc88SChris Mason data_end = leaf_data_end(root, leaf); 2417eb60ceacSChris Mason 2418123abc88SChris Mason if (btrfs_leaf_free_space(root, leaf) < 24199c58309dSChris Mason sizeof(struct btrfs_item) + total_size) { 24203326d1b0SChris Mason btrfs_print_leaf(root, leaf); 24213326d1b0SChris Mason printk("not enough freespace need %u have %d\n", 24229c58309dSChris Mason total_size, btrfs_leaf_free_space(root, leaf)); 2423be0e5c09SChris Mason BUG(); 2424d4dbff95SChris Mason } 24255f39d397SChris Mason 242662e2749eSChris Mason slot = path->slots[0]; 2427eb60ceacSChris Mason BUG_ON(slot < 0); 24285f39d397SChris Mason 2429be0e5c09SChris Mason if (slot != nritems) { 2430be0e5c09SChris Mason int i; 24315f39d397SChris Mason unsigned int old_data = btrfs_item_end_nr(leaf, slot); 2432be0e5c09SChris Mason 24335f39d397SChris Mason if (old_data < data_end) { 24345f39d397SChris Mason btrfs_print_leaf(root, leaf); 24355f39d397SChris Mason printk("slot %d old_data %d data_end %d\n", 24365f39d397SChris Mason slot, old_data, data_end); 24375f39d397SChris Mason BUG_ON(1); 24385f39d397SChris Mason } 2439be0e5c09SChris Mason /* 2440be0e5c09SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2441be0e5c09SChris Mason */ 2442be0e5c09SChris Mason /* first correct the data pointers */ 2443db94535dSChris Mason WARN_ON(leaf->map_token); 24440783fcfcSChris Mason for (i = slot; i < nritems; i++) { 24455f39d397SChris Mason u32 ioff; 2446db94535dSChris Mason 24475f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2448db94535dSChris Mason if (!leaf->map_token) { 2449db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2450db94535dSChris Mason sizeof(struct btrfs_item), 2451db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2452db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2453db94535dSChris Mason KM_USER1); 2454db94535dSChris Mason } 2455db94535dSChris Mason 24565f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 24579c58309dSChris Mason btrfs_set_item_offset(leaf, item, ioff - total_data); 24580783fcfcSChris Mason } 2459db94535dSChris Mason if (leaf->map_token) { 2460db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2461db94535dSChris Mason leaf->map_token = NULL; 2462db94535dSChris Mason } 2463be0e5c09SChris Mason 2464be0e5c09SChris Mason /* shift the items */ 24659c58309dSChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr), 24665f39d397SChris Mason btrfs_item_nr_offset(slot), 24670783fcfcSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 2468be0e5c09SChris Mason 2469be0e5c09SChris Mason /* shift the data */ 24705f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 24719c58309dSChris Mason data_end - total_data, btrfs_leaf_data(leaf) + 2472be0e5c09SChris Mason data_end, old_data - data_end); 2473be0e5c09SChris Mason data_end = old_data; 2474be0e5c09SChris Mason } 24755f39d397SChris Mason 247662e2749eSChris Mason /* setup the item for the new data */ 24779c58309dSChris Mason for (i = 0; i < nr; i++) { 24789c58309dSChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key + i); 24799c58309dSChris Mason btrfs_set_item_key(leaf, &disk_key, slot + i); 24809c58309dSChris Mason item = btrfs_item_nr(leaf, slot + i); 24819c58309dSChris Mason btrfs_set_item_offset(leaf, item, data_end - data_size[i]); 24829c58309dSChris Mason data_end -= data_size[i]; 24839c58309dSChris Mason btrfs_set_item_size(leaf, item, data_size[i]); 24849c58309dSChris Mason } 24859c58309dSChris Mason btrfs_set_header_nritems(leaf, nritems + nr); 24865f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2487aa5d6bedSChris Mason 2488aa5d6bedSChris Mason ret = 0; 24895a01a2e3SChris Mason if (slot == 0) { 24905a01a2e3SChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key); 2491e089f05cSChris Mason ret = fixup_low_keys(trans, root, path, &disk_key, 1); 24925a01a2e3SChris Mason } 2493aa5d6bedSChris Mason 24945f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 24955f39d397SChris Mason btrfs_print_leaf(root, leaf); 2496be0e5c09SChris Mason BUG(); 24975f39d397SChris Mason } 24985a01a2e3SChris Mason 2499ed2ff2cbSChris Mason out: 250062e2749eSChris Mason return ret; 250162e2749eSChris Mason } 250262e2749eSChris Mason 250362e2749eSChris Mason /* 250462e2749eSChris Mason * Given a key and some data, insert an item into the tree. 250562e2749eSChris Mason * This does all the path init required, making room in the tree if needed. 250662e2749eSChris Mason */ 2507e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root 2508e089f05cSChris Mason *root, struct btrfs_key *cpu_key, void *data, u32 2509e089f05cSChris Mason data_size) 251062e2749eSChris Mason { 251162e2749eSChris Mason int ret = 0; 25122c90e5d6SChris Mason struct btrfs_path *path; 25135f39d397SChris Mason struct extent_buffer *leaf; 25145f39d397SChris Mason unsigned long ptr; 251562e2749eSChris Mason 25162c90e5d6SChris Mason path = btrfs_alloc_path(); 25172c90e5d6SChris Mason BUG_ON(!path); 25182c90e5d6SChris Mason ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size); 251962e2749eSChris Mason if (!ret) { 25205f39d397SChris Mason leaf = path->nodes[0]; 25215f39d397SChris Mason ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); 25225f39d397SChris Mason write_extent_buffer(leaf, data, ptr, data_size); 25235f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 252462e2749eSChris Mason } 25252c90e5d6SChris Mason btrfs_free_path(path); 2526aa5d6bedSChris Mason return ret; 2527be0e5c09SChris Mason } 2528be0e5c09SChris Mason 252974123bd7SChris Mason /* 25305de08d7dSChris Mason * delete the pointer from a given node. 253174123bd7SChris Mason * 253274123bd7SChris Mason * If the delete empties a node, the node is removed from the tree, 253374123bd7SChris Mason * continuing all the way the root if required. The root is converted into 253474123bd7SChris Mason * a leaf if all the nodes are emptied. 253574123bd7SChris Mason */ 2536e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 2537e089f05cSChris Mason struct btrfs_path *path, int level, int slot) 2538be0e5c09SChris Mason { 25395f39d397SChris Mason struct extent_buffer *parent = path->nodes[level]; 25407518a238SChris Mason u32 nritems; 2541aa5d6bedSChris Mason int ret = 0; 2542bb803951SChris Mason int wret; 2543be0e5c09SChris Mason 25445f39d397SChris Mason nritems = btrfs_header_nritems(parent); 2545be0e5c09SChris Mason if (slot != nritems -1) { 25465f39d397SChris Mason memmove_extent_buffer(parent, 25475f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 25485f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 2549d6025579SChris Mason sizeof(struct btrfs_key_ptr) * 2550d6025579SChris Mason (nritems - slot - 1)); 2551be0e5c09SChris Mason } 25527518a238SChris Mason nritems--; 25535f39d397SChris Mason btrfs_set_header_nritems(parent, nritems); 25547518a238SChris Mason if (nritems == 0 && parent == root->node) { 25555f39d397SChris Mason BUG_ON(btrfs_header_level(root->node) != 1); 2556eb60ceacSChris Mason /* just turn the root into a leaf and break */ 25575f39d397SChris Mason btrfs_set_header_level(root->node, 0); 2558bb803951SChris Mason } else if (slot == 0) { 25595f39d397SChris Mason struct btrfs_disk_key disk_key; 25605f39d397SChris Mason 25615f39d397SChris Mason btrfs_node_key(parent, &disk_key, 0); 25625f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, level + 1); 25630f70abe2SChris Mason if (wret) 25640f70abe2SChris Mason ret = wret; 2565be0e5c09SChris Mason } 2566d6025579SChris Mason btrfs_mark_buffer_dirty(parent); 2567aa5d6bedSChris Mason return ret; 2568be0e5c09SChris Mason } 2569be0e5c09SChris Mason 257074123bd7SChris Mason /* 257174123bd7SChris Mason * delete the item at the leaf level in path. If that empties 257274123bd7SChris Mason * the leaf, remove it from the tree 257374123bd7SChris Mason */ 257485e21bacSChris Mason int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root, 257585e21bacSChris Mason struct btrfs_path *path, int slot, int nr) 2576be0e5c09SChris Mason { 25775f39d397SChris Mason struct extent_buffer *leaf; 25785f39d397SChris Mason struct btrfs_item *item; 257985e21bacSChris Mason int last_off; 258085e21bacSChris Mason int dsize = 0; 2581aa5d6bedSChris Mason int ret = 0; 2582aa5d6bedSChris Mason int wret; 258385e21bacSChris Mason int i; 25847518a238SChris Mason u32 nritems; 2585be0e5c09SChris Mason 25865f39d397SChris Mason leaf = path->nodes[0]; 258785e21bacSChris Mason last_off = btrfs_item_offset_nr(leaf, slot + nr - 1); 258885e21bacSChris Mason 258985e21bacSChris Mason for (i = 0; i < nr; i++) 259085e21bacSChris Mason dsize += btrfs_item_size_nr(leaf, slot + i); 259185e21bacSChris Mason 25925f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2593be0e5c09SChris Mason 259485e21bacSChris Mason if (slot + nr != nritems) { 2595be0e5c09SChris Mason int i; 2596123abc88SChris Mason int data_end = leaf_data_end(root, leaf); 25975f39d397SChris Mason 25985f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2599d6025579SChris Mason data_end + dsize, 2600123abc88SChris Mason btrfs_leaf_data(leaf) + data_end, 260185e21bacSChris Mason last_off - data_end); 26025f39d397SChris Mason 260385e21bacSChris Mason for (i = slot + nr; i < nritems; i++) { 26045f39d397SChris Mason u32 ioff; 2605db94535dSChris Mason 26065f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2607db94535dSChris Mason if (!leaf->map_token) { 2608db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2609db94535dSChris Mason sizeof(struct btrfs_item), 2610db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2611db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2612db94535dSChris Mason KM_USER1); 2613db94535dSChris Mason } 26145f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 26155f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + dsize); 26160783fcfcSChris Mason } 2617db94535dSChris Mason 2618db94535dSChris Mason if (leaf->map_token) { 2619db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2620db94535dSChris Mason leaf->map_token = NULL; 2621db94535dSChris Mason } 2622db94535dSChris Mason 26235f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot), 262485e21bacSChris Mason btrfs_item_nr_offset(slot + nr), 26250783fcfcSChris Mason sizeof(struct btrfs_item) * 262685e21bacSChris Mason (nritems - slot - nr)); 2627be0e5c09SChris Mason } 262885e21bacSChris Mason btrfs_set_header_nritems(leaf, nritems - nr); 262985e21bacSChris Mason nritems -= nr; 26305f39d397SChris Mason 263174123bd7SChris Mason /* delete the leaf if we've emptied it */ 26327518a238SChris Mason if (nritems == 0) { 26335f39d397SChris Mason if (leaf == root->node) { 26345f39d397SChris Mason btrfs_set_header_level(leaf, 0); 26359a8dd150SChris Mason } else { 26367bb86316SChris Mason u64 root_gen = btrfs_header_generation(path->nodes[1]); 26375f39d397SChris Mason clean_tree_block(trans, root, leaf); 2638e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, path->slots[1]); 2639aa5d6bedSChris Mason if (wret) 2640aa5d6bedSChris Mason ret = wret; 2641e089f05cSChris Mason wret = btrfs_free_extent(trans, root, 26427bb86316SChris Mason leaf->start, leaf->len, 26437bb86316SChris Mason btrfs_header_owner(path->nodes[1]), 26447bb86316SChris Mason root_gen, 0, 0, 1); 26450f70abe2SChris Mason if (wret) 26460f70abe2SChris Mason ret = wret; 26479a8dd150SChris Mason } 2648be0e5c09SChris Mason } else { 26497518a238SChris Mason int used = leaf_space_used(leaf, 0, nritems); 2650aa5d6bedSChris Mason if (slot == 0) { 26515f39d397SChris Mason struct btrfs_disk_key disk_key; 26525f39d397SChris Mason 26535f39d397SChris Mason btrfs_item_key(leaf, &disk_key, 0); 2654e089f05cSChris Mason wret = fixup_low_keys(trans, root, path, 26555f39d397SChris Mason &disk_key, 1); 2656aa5d6bedSChris Mason if (wret) 2657aa5d6bedSChris Mason ret = wret; 2658aa5d6bedSChris Mason } 2659aa5d6bedSChris Mason 266074123bd7SChris Mason /* delete the leaf if it is mostly empty */ 266185e21bacSChris Mason if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) { 2662be0e5c09SChris Mason /* push_leaf_left fixes the path. 2663be0e5c09SChris Mason * make sure the path still points to our leaf 2664be0e5c09SChris Mason * for possible call to del_ptr below 2665be0e5c09SChris Mason */ 26664920c9acSChris Mason slot = path->slots[1]; 26675f39d397SChris Mason extent_buffer_get(leaf); 26685f39d397SChris Mason 266985e21bacSChris Mason wret = push_leaf_left(trans, root, path, 1, 1); 267054aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2671aa5d6bedSChris Mason ret = wret; 26725f39d397SChris Mason 26735f39d397SChris Mason if (path->nodes[0] == leaf && 26745f39d397SChris Mason btrfs_header_nritems(leaf)) { 267585e21bacSChris Mason wret = push_leaf_right(trans, root, path, 1, 1); 267654aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2677aa5d6bedSChris Mason ret = wret; 2678aa5d6bedSChris Mason } 26795f39d397SChris Mason 26805f39d397SChris Mason if (btrfs_header_nritems(leaf) == 0) { 26817bb86316SChris Mason u64 root_gen; 2682db94535dSChris Mason u64 bytenr = leaf->start; 2683db94535dSChris Mason u32 blocksize = leaf->len; 26845f39d397SChris Mason 26857bb86316SChris Mason root_gen = btrfs_header_generation( 26867bb86316SChris Mason path->nodes[1]); 26877bb86316SChris Mason 26885f39d397SChris Mason clean_tree_block(trans, root, leaf); 26895f39d397SChris Mason 2690e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, slot); 2691aa5d6bedSChris Mason if (wret) 2692aa5d6bedSChris Mason ret = wret; 26935f39d397SChris Mason 26945f39d397SChris Mason free_extent_buffer(leaf); 2695db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 26967bb86316SChris Mason blocksize, 26977bb86316SChris Mason btrfs_header_owner(path->nodes[1]), 26987bb86316SChris Mason root_gen, 0, 0, 1); 26990f70abe2SChris Mason if (wret) 27000f70abe2SChris Mason ret = wret; 27015de08d7dSChris Mason } else { 27025f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 27035f39d397SChris Mason free_extent_buffer(leaf); 2704be0e5c09SChris Mason } 2705d5719762SChris Mason } else { 27065f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2707be0e5c09SChris Mason } 27089a8dd150SChris Mason } 2709aa5d6bedSChris Mason return ret; 27109a8dd150SChris Mason } 27119a8dd150SChris Mason 271297571fd0SChris Mason /* 27137bb86316SChris Mason * walk up the tree as far as required to find the previous leaf. 27147bb86316SChris Mason * returns 0 if it found something or 1 if there are no lesser leaves. 27157bb86316SChris Mason * returns < 0 on io errors. 27167bb86316SChris Mason */ 27177bb86316SChris Mason int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path) 27187bb86316SChris Mason { 27198f662a76SChris Mason u64 bytenr; 27207bb86316SChris Mason int slot; 27217bb86316SChris Mason int level = 1; 27227bb86316SChris Mason struct extent_buffer *c; 27237bb86316SChris Mason struct extent_buffer *next = NULL; 27247bb86316SChris Mason 27257bb86316SChris Mason while(level < BTRFS_MAX_LEVEL) { 27267bb86316SChris Mason if (!path->nodes[level]) 27277bb86316SChris Mason return 1; 27287bb86316SChris Mason 27297bb86316SChris Mason slot = path->slots[level]; 27307bb86316SChris Mason c = path->nodes[level]; 27317bb86316SChris Mason if (slot == 0) { 27327bb86316SChris Mason level++; 27337bb86316SChris Mason if (level == BTRFS_MAX_LEVEL) 27347bb86316SChris Mason return 1; 27357bb86316SChris Mason continue; 27367bb86316SChris Mason } 27377bb86316SChris Mason slot--; 27387bb86316SChris Mason 27397bb86316SChris Mason bytenr = btrfs_node_blockptr(c, slot); 27407bb86316SChris Mason if (next) 27417bb86316SChris Mason free_extent_buffer(next); 27427bb86316SChris Mason 27437bb86316SChris Mason next = read_tree_block(root, bytenr, 27447bb86316SChris Mason btrfs_level_size(root, level - 1)); 27457bb86316SChris Mason break; 27467bb86316SChris Mason } 27477bb86316SChris Mason path->slots[level] = slot; 27487bb86316SChris Mason while(1) { 27497bb86316SChris Mason level--; 27507bb86316SChris Mason c = path->nodes[level]; 27517bb86316SChris Mason free_extent_buffer(c); 27528f662a76SChris Mason slot = btrfs_header_nritems(next); 27538f662a76SChris Mason if (slot != 0) 27548f662a76SChris Mason slot--; 27557bb86316SChris Mason path->nodes[level] = next; 27568f662a76SChris Mason path->slots[level] = slot; 27577bb86316SChris Mason if (!level) 27587bb86316SChris Mason break; 27598f662a76SChris Mason next = read_tree_block(root, btrfs_node_blockptr(next, slot), 27607bb86316SChris Mason btrfs_level_size(root, level - 1)); 27617bb86316SChris Mason } 27627bb86316SChris Mason return 0; 27637bb86316SChris Mason } 27647bb86316SChris Mason 27657bb86316SChris Mason /* 276697571fd0SChris Mason * walk up the tree as far as required to find the next leaf. 27670f70abe2SChris Mason * returns 0 if it found something or 1 if there are no greater leaves. 27680f70abe2SChris Mason * returns < 0 on io errors. 276997571fd0SChris Mason */ 2770234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path) 2771d97e63b6SChris Mason { 2772d97e63b6SChris Mason int slot; 2773d97e63b6SChris Mason int level = 1; 2774db94535dSChris Mason u64 bytenr; 27755f39d397SChris Mason struct extent_buffer *c; 27765f39d397SChris Mason struct extent_buffer *next = NULL; 2777d97e63b6SChris Mason 2778234b63a0SChris Mason while(level < BTRFS_MAX_LEVEL) { 2779d97e63b6SChris Mason if (!path->nodes[level]) 27800f70abe2SChris Mason return 1; 27815f39d397SChris Mason 2782d97e63b6SChris Mason slot = path->slots[level] + 1; 2783d97e63b6SChris Mason c = path->nodes[level]; 27845f39d397SChris Mason if (slot >= btrfs_header_nritems(c)) { 2785d97e63b6SChris Mason level++; 27867bb86316SChris Mason if (level == BTRFS_MAX_LEVEL) 27877bb86316SChris Mason return 1; 2788d97e63b6SChris Mason continue; 2789d97e63b6SChris Mason } 27905f39d397SChris Mason 2791db94535dSChris Mason bytenr = btrfs_node_blockptr(c, slot); 2792cfaa7295SChris Mason if (next) 27935f39d397SChris Mason free_extent_buffer(next); 27945f39d397SChris Mason 27956702ed49SChris Mason if (path->reada) 279601f46658SChris Mason reada_for_search(root, path, level, slot, 0); 27975f39d397SChris Mason 2798db94535dSChris Mason next = read_tree_block(root, bytenr, 2799db94535dSChris Mason btrfs_level_size(root, level -1)); 2800d97e63b6SChris Mason break; 2801d97e63b6SChris Mason } 2802d97e63b6SChris Mason path->slots[level] = slot; 2803d97e63b6SChris Mason while(1) { 2804d97e63b6SChris Mason level--; 2805d97e63b6SChris Mason c = path->nodes[level]; 28065f39d397SChris Mason free_extent_buffer(c); 2807d97e63b6SChris Mason path->nodes[level] = next; 2808d97e63b6SChris Mason path->slots[level] = 0; 2809d97e63b6SChris Mason if (!level) 2810d97e63b6SChris Mason break; 28116702ed49SChris Mason if (path->reada) 281201f46658SChris Mason reada_for_search(root, path, level, 0, 0); 2813db94535dSChris Mason next = read_tree_block(root, btrfs_node_blockptr(next, 0), 2814db94535dSChris Mason btrfs_level_size(root, level - 1)); 2815d97e63b6SChris Mason } 2816d97e63b6SChris Mason return 0; 2817d97e63b6SChris Mason } 28180b86a832SChris Mason 28190b86a832SChris Mason int btrfs_previous_item(struct btrfs_root *root, 28200b86a832SChris Mason struct btrfs_path *path, u64 min_objectid, 28210b86a832SChris Mason int type) 28220b86a832SChris Mason { 28230b86a832SChris Mason struct btrfs_key found_key; 28240b86a832SChris Mason struct extent_buffer *leaf; 28250b86a832SChris Mason int ret; 28260b86a832SChris Mason 28270b86a832SChris Mason while(1) { 28280b86a832SChris Mason if (path->slots[0] == 0) { 28290b86a832SChris Mason ret = btrfs_prev_leaf(root, path); 28300b86a832SChris Mason if (ret != 0) 28310b86a832SChris Mason return ret; 28320b86a832SChris Mason } else { 28330b86a832SChris Mason path->slots[0]--; 28340b86a832SChris Mason } 28350b86a832SChris Mason leaf = path->nodes[0]; 28360b86a832SChris Mason btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]); 28370b86a832SChris Mason if (found_key.type == type) 28380b86a832SChris Mason return 0; 28390b86a832SChris Mason } 28400b86a832SChris Mason return 1; 28410b86a832SChris Mason } 28420b86a832SChris Mason 2843