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 195f39d397SChris Mason #include <linux/highmem.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, 29d4dbff95SChris Mason struct btrfs_path *path, int data_size); 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 735f39d397SChris Mason static int __btrfs_cow_block(struct btrfs_trans_handle *trans, 745f39d397SChris Mason struct btrfs_root *root, 755f39d397SChris Mason struct extent_buffer *buf, 765f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 775f39d397SChris Mason struct extent_buffer **cow_ret, 785f39d397SChris Mason u64 search_start, u64 empty_size) 796702ed49SChris Mason { 805f39d397SChris Mason struct extent_buffer *cow; 816702ed49SChris Mason int ret = 0; 826702ed49SChris Mason int different_trans = 0; 836702ed49SChris Mason 846702ed49SChris Mason WARN_ON(root->ref_cows && trans->transid != root->last_trans); 855f39d397SChris Mason 86*db94535dSChris Mason cow = btrfs_alloc_free_block(trans, root, buf->len, 87*db94535dSChris Mason search_start, empty_size); 886702ed49SChris Mason if (IS_ERR(cow)) 896702ed49SChris Mason return PTR_ERR(cow); 906702ed49SChris Mason 91f510cfecSChris Mason cow->alloc_addr = (unsigned long)__builtin_return_address(0); 926702ed49SChris Mason 935f39d397SChris Mason copy_extent_buffer(cow, buf, 0, 0, cow->len); 94*db94535dSChris Mason btrfs_set_header_bytenr(cow, cow->start); 955f39d397SChris Mason btrfs_set_header_generation(cow, trans->transid); 965f39d397SChris Mason btrfs_set_header_owner(cow, root->root_key.objectid); 976702ed49SChris Mason 985f39d397SChris Mason WARN_ON(btrfs_header_generation(buf) > trans->transid); 995f39d397SChris Mason if (btrfs_header_generation(buf) != trans->transid) { 1006702ed49SChris Mason different_trans = 1; 1016702ed49SChris Mason ret = btrfs_inc_ref(trans, root, buf); 1026702ed49SChris Mason if (ret) 1036702ed49SChris Mason return ret; 1046702ed49SChris Mason } else { 1056702ed49SChris Mason clean_tree_block(trans, root, buf); 1066702ed49SChris Mason } 1076702ed49SChris Mason 1086702ed49SChris Mason if (buf == root->node) { 1096702ed49SChris Mason root->node = cow; 1105f39d397SChris Mason extent_buffer_get(cow); 1116702ed49SChris Mason if (buf != root->commit_root) { 112*db94535dSChris Mason btrfs_free_extent(trans, root, buf->start, 113*db94535dSChris Mason buf->len, 1); 1146702ed49SChris Mason } 1155f39d397SChris Mason free_extent_buffer(buf); 1166702ed49SChris Mason } else { 1175f39d397SChris Mason btrfs_set_node_blockptr(parent, parent_slot, 118*db94535dSChris Mason cow->start); 1196702ed49SChris Mason btrfs_mark_buffer_dirty(parent); 1205f39d397SChris Mason WARN_ON(btrfs_header_generation(parent) != trans->transid); 121*db94535dSChris Mason btrfs_free_extent(trans, root, buf->start, buf->len, 1); 1226702ed49SChris Mason } 1235f39d397SChris Mason free_extent_buffer(buf); 1246702ed49SChris Mason btrfs_mark_buffer_dirty(cow); 1256702ed49SChris Mason *cow_ret = cow; 1266702ed49SChris Mason return 0; 1276702ed49SChris Mason } 1286702ed49SChris Mason 1295f39d397SChris Mason int btrfs_cow_block(struct btrfs_trans_handle *trans, 1305f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *buf, 1315f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 1325f39d397SChris Mason struct extent_buffer **cow_ret) 13302217ed2SChris Mason { 1346702ed49SChris Mason u64 search_start; 135f510cfecSChris Mason int ret; 136ccd467d6SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 137ccd467d6SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 138ccd467d6SChris Mason root->fs_info->running_transaction->transid); 139ccd467d6SChris Mason WARN_ON(1); 140ccd467d6SChris Mason } 141ccd467d6SChris Mason if (trans->transid != root->fs_info->generation) { 142ccd467d6SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 143ccd467d6SChris Mason root->fs_info->generation); 144ccd467d6SChris Mason WARN_ON(1); 145ccd467d6SChris Mason } 1465f39d397SChris Mason if (btrfs_header_generation(buf) == trans->transid) { 14702217ed2SChris Mason *cow_ret = buf; 14802217ed2SChris Mason return 0; 14902217ed2SChris Mason } 1506702ed49SChris Mason 151*db94535dSChris Mason search_start = buf->start & ~((u64)BTRFS_BLOCK_GROUP_SIZE - 1); 152f510cfecSChris Mason ret = __btrfs_cow_block(trans, root, buf, parent, 1536702ed49SChris Mason parent_slot, cow_ret, search_start, 0); 154f510cfecSChris Mason (*cow_ret)->alloc_addr = (unsigned long)__builtin_return_address(0); 155f510cfecSChris Mason return ret; 1562c90e5d6SChris Mason } 1576702ed49SChris Mason 158*db94535dSChris Mason #if 0 1596702ed49SChris Mason static int close_blocks(u64 blocknr, u64 other) 1606702ed49SChris Mason { 1616702ed49SChris Mason if (blocknr < other && other - blocknr < 8) 1626702ed49SChris Mason return 1; 1636702ed49SChris Mason if (blocknr > other && blocknr - other < 8) 1646702ed49SChris Mason return 1; 16502217ed2SChris Mason return 0; 16602217ed2SChris Mason } 16702217ed2SChris Mason 1685f39d397SChris Mason static int should_defrag_leaf(struct extent_buffer *eb) 1692cc58cf2SChris Mason { 1705f39d397SChris Mason return 0; 1715f39d397SChris Mason struct btrfs_leaf *leaf = btrfs_buffer_leaf(eb); 1722cc58cf2SChris Mason struct btrfs_disk_key *key; 1732cc58cf2SChris Mason u32 nritems; 1742cc58cf2SChris Mason 1752cc58cf2SChris Mason if (buffer_defrag(bh)) 1762cc58cf2SChris Mason return 1; 1772cc58cf2SChris Mason 1782cc58cf2SChris Mason nritems = btrfs_header_nritems(&leaf->header); 1792cc58cf2SChris Mason if (nritems == 0) 1802cc58cf2SChris Mason return 0; 1812cc58cf2SChris Mason 1822cc58cf2SChris Mason key = &leaf->items[0].key; 1832cc58cf2SChris Mason if (btrfs_disk_key_type(key) == BTRFS_DIR_ITEM_KEY) 1842cc58cf2SChris Mason return 1; 1852cc58cf2SChris Mason 1862cc58cf2SChris Mason key = &leaf->items[nritems-1].key; 1872cc58cf2SChris Mason if (btrfs_disk_key_type(key) == BTRFS_DIR_ITEM_KEY) 1882cc58cf2SChris Mason return 1; 1892cc58cf2SChris Mason if (nritems > 4) { 1902cc58cf2SChris Mason key = &leaf->items[nritems/2].key; 1912cc58cf2SChris Mason if (btrfs_disk_key_type(key) == BTRFS_DIR_ITEM_KEY) 1922cc58cf2SChris Mason return 1; 1932cc58cf2SChris Mason } 1942cc58cf2SChris Mason return 0; 1952cc58cf2SChris Mason } 1965f39d397SChris Mason #endif 1972cc58cf2SChris Mason 1986702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans, 1995f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *parent, 200e9d0b13bSChris Mason int cache_only, u64 *last_ret) 2016702ed49SChris Mason { 2025f39d397SChris Mason return 0; 2035f39d397SChris Mason #if 0 2046702ed49SChris Mason struct btrfs_node *parent_node; 2055f39d397SChris Mason struct extent_buffer *cur_eb; 2065f39d397SChris Mason struct extent_buffer *tmp_eb; 2076702ed49SChris Mason u64 blocknr; 208e9d0b13bSChris Mason u64 search_start = *last_ret; 209e9d0b13bSChris Mason u64 last_block = 0; 2106702ed49SChris Mason u64 other; 2116702ed49SChris Mason u32 parent_nritems; 2126702ed49SChris Mason int start_slot; 2136702ed49SChris Mason int end_slot; 2146702ed49SChris Mason int i; 2156702ed49SChris Mason int err = 0; 216f2183bdeSChris Mason int parent_level; 2176702ed49SChris Mason 2186702ed49SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 2196702ed49SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 2206702ed49SChris Mason root->fs_info->running_transaction->transid); 2216702ed49SChris Mason WARN_ON(1); 2226702ed49SChris Mason } 2236702ed49SChris Mason if (trans->transid != root->fs_info->generation) { 2246702ed49SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 2256702ed49SChris Mason root->fs_info->generation); 2266702ed49SChris Mason WARN_ON(1); 2276702ed49SChris Mason } 22886479a04SChris Mason if (buffer_defrag_done(parent)) 22986479a04SChris Mason return 0; 23086479a04SChris Mason 2316702ed49SChris Mason parent_node = btrfs_buffer_node(parent); 2326702ed49SChris Mason parent_nritems = btrfs_header_nritems(&parent_node->header); 233f2183bdeSChris Mason parent_level = btrfs_header_level(&parent_node->header); 2346702ed49SChris Mason 2356702ed49SChris Mason start_slot = 0; 2366702ed49SChris Mason end_slot = parent_nritems; 2376702ed49SChris Mason 2386702ed49SChris Mason if (parent_nritems == 1) 2396702ed49SChris Mason return 0; 2406702ed49SChris Mason 2416702ed49SChris Mason for (i = start_slot; i < end_slot; i++) { 2426702ed49SChris Mason int close = 1; 2436702ed49SChris Mason blocknr = btrfs_node_blockptr(parent_node, i); 244e9d0b13bSChris Mason if (last_block == 0) 245e9d0b13bSChris Mason last_block = blocknr; 2466702ed49SChris Mason if (i > 0) { 2476702ed49SChris Mason other = btrfs_node_blockptr(parent_node, i - 1); 2486702ed49SChris Mason close = close_blocks(blocknr, other); 2496702ed49SChris Mason } 2506702ed49SChris Mason if (close && i < end_slot - 1) { 2516702ed49SChris Mason other = btrfs_node_blockptr(parent_node, i + 1); 2526702ed49SChris Mason close = close_blocks(blocknr, other); 2536702ed49SChris Mason } 254e9d0b13bSChris Mason if (close) { 255e9d0b13bSChris Mason last_block = blocknr; 2566702ed49SChris Mason continue; 257e9d0b13bSChris Mason } 2586702ed49SChris Mason 2596702ed49SChris Mason cur_bh = btrfs_find_tree_block(root, blocknr); 2606702ed49SChris Mason if (!cur_bh || !buffer_uptodate(cur_bh) || 2612cc58cf2SChris Mason buffer_locked(cur_bh) || 2622cc58cf2SChris Mason (parent_level != 1 && !buffer_defrag(cur_bh)) || 2632cc58cf2SChris Mason (parent_level == 1 && !should_defrag_leaf(cur_bh))) { 2646702ed49SChris Mason if (cache_only) { 2656702ed49SChris Mason brelse(cur_bh); 2666702ed49SChris Mason continue; 2676702ed49SChris Mason } 268f2183bdeSChris Mason if (!cur_bh || !buffer_uptodate(cur_bh) || 269f2183bdeSChris Mason buffer_locked(cur_bh)) { 2706702ed49SChris Mason brelse(cur_bh); 2716702ed49SChris Mason cur_bh = read_tree_block(root, blocknr); 2726702ed49SChris Mason } 273f2183bdeSChris Mason } 274e9d0b13bSChris Mason if (search_start == 0) 275e9d0b13bSChris Mason search_start = last_block & ~((u64)65535); 276e9d0b13bSChris Mason 2776702ed49SChris Mason err = __btrfs_cow_block(trans, root, cur_bh, parent, i, 2786702ed49SChris Mason &tmp_bh, search_start, 2796702ed49SChris Mason min(8, end_slot - i)); 280252c38f0SYan if (err) { 281252c38f0SYan brelse(cur_bh); 2826702ed49SChris Mason break; 283252c38f0SYan } 2846702ed49SChris Mason search_start = bh_blocknr(tmp_bh); 285f2183bdeSChris Mason *last_ret = search_start; 286f2183bdeSChris Mason if (parent_level == 1) 287f2183bdeSChris Mason clear_buffer_defrag(tmp_bh); 28886479a04SChris Mason set_buffer_defrag_done(tmp_bh); 2896702ed49SChris Mason brelse(tmp_bh); 2906702ed49SChris Mason } 2916702ed49SChris Mason return err; 2925f39d397SChris Mason #endif 2936702ed49SChris Mason } 2946702ed49SChris Mason 29574123bd7SChris Mason /* 29674123bd7SChris Mason * The leaf data grows from end-to-front in the node. 29774123bd7SChris Mason * this returns the address of the start of the last item, 29874123bd7SChris Mason * which is the stop of the leaf data stack 29974123bd7SChris Mason */ 300123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root, 3015f39d397SChris Mason struct extent_buffer *leaf) 302be0e5c09SChris Mason { 3035f39d397SChris Mason u32 nr = btrfs_header_nritems(leaf); 304be0e5c09SChris Mason if (nr == 0) 305123abc88SChris Mason return BTRFS_LEAF_DATA_SIZE(root); 3065f39d397SChris Mason return btrfs_item_offset_nr(leaf, nr - 1); 307be0e5c09SChris Mason } 308be0e5c09SChris Mason 30974123bd7SChris Mason /* 31074123bd7SChris Mason * compare two keys in a memcmp fashion 31174123bd7SChris Mason */ 3129aca1d51SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2) 313be0e5c09SChris Mason { 314e2fa7227SChris Mason struct btrfs_key k1; 315e2fa7227SChris Mason 316e2fa7227SChris Mason btrfs_disk_key_to_cpu(&k1, disk); 317e2fa7227SChris Mason 318e2fa7227SChris Mason if (k1.objectid > k2->objectid) 319be0e5c09SChris Mason return 1; 320e2fa7227SChris Mason if (k1.objectid < k2->objectid) 321be0e5c09SChris Mason return -1; 3225f39d397SChris Mason if (k1.type > k2->type) 323f254e52cSChris Mason return 1; 3245f39d397SChris Mason if (k1.type < k2->type) 325f254e52cSChris Mason return -1; 32670b2befdSChris Mason if (k1.offset > k2->offset) 32770b2befdSChris Mason return 1; 32870b2befdSChris Mason if (k1.offset < k2->offset) 32970b2befdSChris Mason return -1; 330be0e5c09SChris Mason return 0; 331be0e5c09SChris Mason } 33274123bd7SChris Mason 333123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path, 334123abc88SChris Mason int level) 335aa5d6bedSChris Mason { 3365f39d397SChris Mason struct extent_buffer *parent = NULL; 3375f39d397SChris Mason struct extent_buffer *node = path->nodes[level]; 3385f39d397SChris Mason struct btrfs_disk_key parent_key; 3395f39d397SChris Mason struct btrfs_disk_key node_key; 340aa5d6bedSChris Mason int parent_slot; 3418d7be552SChris Mason int slot; 3428d7be552SChris Mason struct btrfs_key cpukey; 3435f39d397SChris Mason u32 nritems = btrfs_header_nritems(node); 344aa5d6bedSChris Mason 345aa5d6bedSChris Mason if (path->nodes[level + 1]) 3465f39d397SChris Mason parent = path->nodes[level + 1]; 347a1f39630SAneesh 3488d7be552SChris Mason slot = path->slots[level]; 3497518a238SChris Mason BUG_ON(nritems == 0); 3507518a238SChris Mason if (parent) { 351a1f39630SAneesh parent_slot = path->slots[level + 1]; 3525f39d397SChris Mason btrfs_node_key(parent, &parent_key, parent_slot); 3535f39d397SChris Mason btrfs_node_key(node, &node_key, 0); 3545f39d397SChris Mason BUG_ON(memcmp(&parent_key, &node_key, 355e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 3561d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 357*db94535dSChris Mason btrfs_header_bytenr(node)); 358aa5d6bedSChris Mason } 359123abc88SChris Mason BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root)); 3608d7be552SChris Mason if (slot != 0) { 3615f39d397SChris Mason btrfs_node_key_to_cpu(node, &cpukey, slot - 1); 3625f39d397SChris Mason btrfs_node_key(node, &node_key, slot); 3635f39d397SChris Mason BUG_ON(comp_keys(&node_key, &cpukey) <= 0); 3648d7be552SChris Mason } 3658d7be552SChris Mason if (slot < nritems - 1) { 3665f39d397SChris Mason btrfs_node_key_to_cpu(node, &cpukey, slot + 1); 3675f39d397SChris Mason btrfs_node_key(node, &node_key, slot); 3685f39d397SChris Mason BUG_ON(comp_keys(&node_key, &cpukey) >= 0); 369aa5d6bedSChris Mason } 370aa5d6bedSChris Mason return 0; 371aa5d6bedSChris Mason } 372aa5d6bedSChris Mason 373123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path, 374123abc88SChris Mason int level) 375aa5d6bedSChris Mason { 3765f39d397SChris Mason struct extent_buffer *leaf = path->nodes[level]; 3775f39d397SChris Mason struct extent_buffer *parent = NULL; 378aa5d6bedSChris Mason int parent_slot; 3798d7be552SChris Mason struct btrfs_key cpukey; 3805f39d397SChris Mason struct btrfs_disk_key parent_key; 3815f39d397SChris Mason struct btrfs_disk_key leaf_key; 3825f39d397SChris Mason int slot = path->slots[0]; 3838d7be552SChris Mason 3845f39d397SChris Mason u32 nritems = btrfs_header_nritems(leaf); 385aa5d6bedSChris Mason 386aa5d6bedSChris Mason if (path->nodes[level + 1]) 3875f39d397SChris Mason parent = path->nodes[level + 1]; 3887518a238SChris Mason 3897518a238SChris Mason if (nritems == 0) 3907518a238SChris Mason return 0; 3917518a238SChris Mason 3927518a238SChris Mason if (parent) { 393a1f39630SAneesh parent_slot = path->slots[level + 1]; 3945f39d397SChris Mason btrfs_node_key(parent, &parent_key, parent_slot); 3955f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, 0); 3966702ed49SChris Mason 3975f39d397SChris Mason BUG_ON(memcmp(&parent_key, &leaf_key, 398e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 3991d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 400*db94535dSChris Mason btrfs_header_bytenr(leaf)); 401aa5d6bedSChris Mason } 4025f39d397SChris Mason #if 0 4035f39d397SChris Mason for (i = 0; nritems > 1 && i < nritems - 2; i++) { 4045f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, i + 1); 4055f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, i); 4065f39d397SChris Mason if (comp_keys(&leaf_key, &cpukey) >= 0) { 4075f39d397SChris Mason btrfs_print_leaf(root, leaf); 4085f39d397SChris Mason printk("slot %d offset bad key\n", i); 4095f39d397SChris Mason BUG_ON(1); 4105f39d397SChris Mason } 4115f39d397SChris Mason if (btrfs_item_offset_nr(leaf, i) != 4125f39d397SChris Mason btrfs_item_end_nr(leaf, i + 1)) { 4135f39d397SChris Mason btrfs_print_leaf(root, leaf); 4145f39d397SChris Mason printk("slot %d offset bad\n", i); 4155f39d397SChris Mason BUG_ON(1); 4165f39d397SChris Mason } 4175f39d397SChris Mason if (i == 0) { 4185f39d397SChris Mason if (btrfs_item_offset_nr(leaf, i) + 4195f39d397SChris Mason btrfs_item_size_nr(leaf, i) != 4205f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 4215f39d397SChris Mason btrfs_print_leaf(root, leaf); 4225f39d397SChris Mason printk("slot %d first offset bad\n", i); 4235f39d397SChris Mason BUG_ON(1); 4245f39d397SChris Mason } 4255f39d397SChris Mason } 4265f39d397SChris Mason } 4275f39d397SChris Mason if (nritems > 0) { 4285f39d397SChris Mason if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) { 4295f39d397SChris Mason btrfs_print_leaf(root, leaf); 4305f39d397SChris Mason printk("slot %d bad size \n", nritems - 1); 4315f39d397SChris Mason BUG_ON(1); 4325f39d397SChris Mason } 4335f39d397SChris Mason } 4345f39d397SChris Mason #endif 4355f39d397SChris Mason if (slot != 0 && slot < nritems - 1) { 4365f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, slot); 4375f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1); 4385f39d397SChris Mason if (comp_keys(&leaf_key, &cpukey) <= 0) { 4395f39d397SChris Mason btrfs_print_leaf(root, leaf); 4405f39d397SChris Mason printk("slot %d offset bad key\n", slot); 4415f39d397SChris Mason BUG_ON(1); 4425f39d397SChris Mason } 4435f39d397SChris Mason if (btrfs_item_offset_nr(leaf, slot - 1) != 4445f39d397SChris Mason btrfs_item_end_nr(leaf, slot)) { 4455f39d397SChris Mason btrfs_print_leaf(root, leaf); 4465f39d397SChris Mason printk("slot %d offset bad\n", slot); 4475f39d397SChris Mason BUG_ON(1); 4485f39d397SChris Mason } 449aa5d6bedSChris Mason } 4508d7be552SChris Mason if (slot < nritems - 1) { 4515f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, slot); 4525f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1); 4535f39d397SChris Mason BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0); 4545f39d397SChris Mason if (btrfs_item_offset_nr(leaf, slot) != 4555f39d397SChris Mason btrfs_item_end_nr(leaf, slot + 1)) { 4565f39d397SChris Mason btrfs_print_leaf(root, leaf); 4575f39d397SChris Mason printk("slot %d offset bad\n", slot); 4585f39d397SChris Mason BUG_ON(1); 459aa5d6bedSChris Mason } 4605f39d397SChris Mason } 4615f39d397SChris Mason BUG_ON(btrfs_item_offset_nr(leaf, 0) + 4625f39d397SChris Mason btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root)); 463aa5d6bedSChris Mason return 0; 464aa5d6bedSChris Mason } 465aa5d6bedSChris Mason 466123abc88SChris Mason static int check_block(struct btrfs_root *root, struct btrfs_path *path, 467123abc88SChris Mason int level) 468aa5d6bedSChris Mason { 469*db94535dSChris Mason #if 0 4705f39d397SChris Mason struct extent_buffer *buf = path->nodes[level]; 4715f39d397SChris Mason 472479965d6SChris Mason if (memcmp_extent_buffer(buf, root->fs_info->fsid, 473479965d6SChris Mason (unsigned long)btrfs_header_fsid(buf), 474479965d6SChris Mason BTRFS_FSID_SIZE)) { 4755f39d397SChris Mason printk("warning bad block %Lu\n", buf->start); 476*db94535dSChris Mason return 1; 4775f39d397SChris Mason } 478*db94535dSChris Mason #endif 479aa5d6bedSChris Mason if (level == 0) 480123abc88SChris Mason return check_leaf(root, path, level); 481123abc88SChris Mason return check_node(root, path, level); 482aa5d6bedSChris Mason } 483aa5d6bedSChris Mason 48474123bd7SChris Mason /* 4855f39d397SChris Mason * search for key in the extent_buffer. The items start at offset p, 4865f39d397SChris Mason * and they are item_size apart. There are 'max' items in p. 4875f39d397SChris Mason * 48874123bd7SChris Mason * the slot in the array is returned via slot, and it points to 48974123bd7SChris Mason * the place where you would insert key if it is not found in 49074123bd7SChris Mason * the array. 49174123bd7SChris Mason * 49274123bd7SChris Mason * slot may point to max if the key is bigger than all of the keys 49374123bd7SChris Mason */ 4945f39d397SChris Mason static int generic_bin_search(struct extent_buffer *eb, unsigned long p, 4955f39d397SChris Mason int item_size, struct btrfs_key *key, 496be0e5c09SChris Mason int max, int *slot) 497be0e5c09SChris Mason { 498be0e5c09SChris Mason int low = 0; 499be0e5c09SChris Mason int high = max; 500be0e5c09SChris Mason int mid; 501be0e5c09SChris Mason int ret; 502479965d6SChris Mason struct btrfs_disk_key *tmp = NULL; 5035f39d397SChris Mason struct btrfs_disk_key unaligned; 5045f39d397SChris Mason unsigned long offset; 5055f39d397SChris Mason char *map_token = NULL; 5065f39d397SChris Mason char *kaddr = NULL; 5075f39d397SChris Mason unsigned long map_start = 0; 5085f39d397SChris Mason unsigned long map_len = 0; 509479965d6SChris Mason int err; 510be0e5c09SChris Mason 511be0e5c09SChris Mason while(low < high) { 512be0e5c09SChris Mason mid = (low + high) / 2; 5135f39d397SChris Mason offset = p + mid * item_size; 5145f39d397SChris Mason 5155f39d397SChris Mason if (!map_token || offset < map_start || 5165f39d397SChris Mason (offset + sizeof(struct btrfs_disk_key)) > 5175f39d397SChris Mason map_start + map_len) { 518479965d6SChris Mason if (map_token) { 5195f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 520479965d6SChris Mason map_token = NULL; 521479965d6SChris Mason } 522479965d6SChris Mason err = map_extent_buffer(eb, offset, 523479965d6SChris Mason sizeof(struct btrfs_disk_key), 524479965d6SChris Mason &map_token, &kaddr, 5255f39d397SChris Mason &map_start, &map_len, KM_USER0); 5265f39d397SChris Mason 527479965d6SChris Mason if (!err) { 528479965d6SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 529479965d6SChris Mason map_start); 530479965d6SChris Mason } else { 5315f39d397SChris Mason read_extent_buffer(eb, &unaligned, 5325f39d397SChris Mason offset, sizeof(unaligned)); 5335f39d397SChris Mason tmp = &unaligned; 534479965d6SChris Mason } 535479965d6SChris Mason 5365f39d397SChris Mason } else { 5375f39d397SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 5385f39d397SChris Mason map_start); 5395f39d397SChris Mason } 540be0e5c09SChris Mason ret = comp_keys(tmp, key); 541be0e5c09SChris Mason 542be0e5c09SChris Mason if (ret < 0) 543be0e5c09SChris Mason low = mid + 1; 544be0e5c09SChris Mason else if (ret > 0) 545be0e5c09SChris Mason high = mid; 546be0e5c09SChris Mason else { 547be0e5c09SChris Mason *slot = mid; 548479965d6SChris Mason if (map_token) 5495f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 550be0e5c09SChris Mason return 0; 551be0e5c09SChris Mason } 552be0e5c09SChris Mason } 553be0e5c09SChris Mason *slot = low; 5545f39d397SChris Mason if (map_token) 5555f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 556be0e5c09SChris Mason return 1; 557be0e5c09SChris Mason } 558be0e5c09SChris Mason 55997571fd0SChris Mason /* 56097571fd0SChris Mason * simple bin_search frontend that does the right thing for 56197571fd0SChris Mason * leaves vs nodes 56297571fd0SChris Mason */ 5635f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key, 5645f39d397SChris Mason int level, int *slot) 565be0e5c09SChris Mason { 5665f39d397SChris Mason if (level == 0) { 5675f39d397SChris Mason return generic_bin_search(eb, 5685f39d397SChris Mason offsetof(struct btrfs_leaf, items), 5690783fcfcSChris Mason sizeof(struct btrfs_item), 5705f39d397SChris Mason key, btrfs_header_nritems(eb), 5717518a238SChris Mason slot); 572be0e5c09SChris Mason } else { 5735f39d397SChris Mason return generic_bin_search(eb, 5745f39d397SChris Mason offsetof(struct btrfs_node, ptrs), 575123abc88SChris Mason sizeof(struct btrfs_key_ptr), 5765f39d397SChris Mason key, btrfs_header_nritems(eb), 5777518a238SChris Mason slot); 578be0e5c09SChris Mason } 579be0e5c09SChris Mason return -1; 580be0e5c09SChris Mason } 581be0e5c09SChris Mason 5825f39d397SChris Mason static struct extent_buffer *read_node_slot(struct btrfs_root *root, 5835f39d397SChris Mason struct extent_buffer *parent, int slot) 584bb803951SChris Mason { 585bb803951SChris Mason if (slot < 0) 586bb803951SChris Mason return NULL; 5875f39d397SChris Mason if (slot >= btrfs_header_nritems(parent)) 588bb803951SChris Mason return NULL; 589*db94535dSChris Mason return read_tree_block(root, btrfs_node_blockptr(parent, slot), 590*db94535dSChris Mason btrfs_level_size(root, btrfs_header_level(parent) - 1)); 591bb803951SChris Mason } 592bb803951SChris Mason 593e089f05cSChris Mason static int balance_level(struct btrfs_trans_handle *trans, struct btrfs_root 594e089f05cSChris Mason *root, struct btrfs_path *path, int level) 595bb803951SChris Mason { 5965f39d397SChris Mason struct extent_buffer *right = NULL; 5975f39d397SChris Mason struct extent_buffer *mid; 5985f39d397SChris Mason struct extent_buffer *left = NULL; 5995f39d397SChris Mason struct extent_buffer *parent = NULL; 600bb803951SChris Mason int ret = 0; 601bb803951SChris Mason int wret; 602bb803951SChris Mason int pslot; 603bb803951SChris Mason int orig_slot = path->slots[level]; 60454aa1f4dSChris Mason int err_on_enospc = 0; 60579f95c82SChris Mason u64 orig_ptr; 606bb803951SChris Mason 607bb803951SChris Mason if (level == 0) 608bb803951SChris Mason return 0; 609bb803951SChris Mason 6105f39d397SChris Mason mid = path->nodes[level]; 6111d4f8a0cSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 61279f95c82SChris Mason 613234b63a0SChris Mason if (level < BTRFS_MAX_LEVEL - 1) 6145f39d397SChris Mason parent = path->nodes[level + 1]; 615bb803951SChris Mason pslot = path->slots[level + 1]; 616bb803951SChris Mason 61740689478SChris Mason /* 61840689478SChris Mason * deal with the case where there is only one pointer in the root 61940689478SChris Mason * by promoting the node below to a root 62040689478SChris Mason */ 6215f39d397SChris Mason if (!parent) { 6225f39d397SChris Mason struct extent_buffer *child; 623bb803951SChris Mason 6245f39d397SChris Mason if (btrfs_header_nritems(mid) != 1) 625bb803951SChris Mason return 0; 626bb803951SChris Mason 627bb803951SChris Mason /* promote the child to a root */ 6285f39d397SChris Mason child = read_node_slot(root, mid, 0); 629bb803951SChris Mason BUG_ON(!child); 630bb803951SChris Mason root->node = child; 631bb803951SChris Mason path->nodes[level] = NULL; 6325f39d397SChris Mason clean_tree_block(trans, root, mid); 6335f39d397SChris Mason wait_on_tree_block_writeback(root, mid); 634bb803951SChris Mason /* once for the path */ 6355f39d397SChris Mason free_extent_buffer(mid); 636*db94535dSChris Mason ret = btrfs_free_extent(trans, root, mid->start, mid->len, 1); 637bb803951SChris Mason /* once for the root ptr */ 6385f39d397SChris Mason free_extent_buffer(mid); 639*db94535dSChris Mason return ret; 640bb803951SChris Mason } 6415f39d397SChris Mason if (btrfs_header_nritems(mid) > 642123abc88SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) / 4) 643bb803951SChris Mason return 0; 644bb803951SChris Mason 6455f39d397SChris Mason if (btrfs_header_nritems(mid) < 2) 64654aa1f4dSChris Mason err_on_enospc = 1; 64754aa1f4dSChris Mason 6485f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 6495f39d397SChris Mason if (left) { 6505f39d397SChris Mason wret = btrfs_cow_block(trans, root, left, 6515f39d397SChris Mason parent, pslot - 1, &left); 65254aa1f4dSChris Mason if (wret) { 65354aa1f4dSChris Mason ret = wret; 65454aa1f4dSChris Mason goto enospc; 65554aa1f4dSChris Mason } 6562cc58cf2SChris Mason } 6575f39d397SChris Mason right = read_node_slot(root, parent, pslot + 1); 6585f39d397SChris Mason if (right) { 6595f39d397SChris Mason wret = btrfs_cow_block(trans, root, right, 6605f39d397SChris Mason parent, pslot + 1, &right); 6612cc58cf2SChris Mason if (wret) { 6622cc58cf2SChris Mason ret = wret; 6632cc58cf2SChris Mason goto enospc; 6642cc58cf2SChris Mason } 6652cc58cf2SChris Mason } 6662cc58cf2SChris Mason 6672cc58cf2SChris Mason /* first, try to make some room in the middle buffer */ 6685f39d397SChris Mason if (left) { 6695f39d397SChris Mason orig_slot += btrfs_header_nritems(left); 6705f39d397SChris Mason wret = push_node_left(trans, root, left, mid); 67179f95c82SChris Mason if (wret < 0) 67279f95c82SChris Mason ret = wret; 6735f39d397SChris Mason if (btrfs_header_nritems(mid) < 2) 67454aa1f4dSChris Mason err_on_enospc = 1; 675bb803951SChris Mason } 67679f95c82SChris Mason 67779f95c82SChris Mason /* 67879f95c82SChris Mason * then try to empty the right most buffer into the middle 67979f95c82SChris Mason */ 6805f39d397SChris Mason if (right) { 6815f39d397SChris Mason wret = push_node_left(trans, root, mid, right); 68254aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 68379f95c82SChris Mason ret = wret; 6845f39d397SChris Mason if (btrfs_header_nritems(right) == 0) { 685*db94535dSChris Mason u64 bytenr = right->start; 686*db94535dSChris Mason u32 blocksize = right->len; 687*db94535dSChris Mason 6885f39d397SChris Mason clean_tree_block(trans, root, right); 6895f39d397SChris Mason wait_on_tree_block_writeback(root, right); 6905f39d397SChris Mason free_extent_buffer(right); 691bb803951SChris Mason right = NULL; 692e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot + 693e089f05cSChris Mason 1); 694bb803951SChris Mason if (wret) 695bb803951SChris Mason ret = wret; 696*db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 697*db94535dSChris Mason blocksize, 1); 698bb803951SChris Mason if (wret) 699bb803951SChris Mason ret = wret; 700bb803951SChris Mason } else { 7015f39d397SChris Mason struct btrfs_disk_key right_key; 7025f39d397SChris Mason btrfs_node_key(right, &right_key, 0); 7035f39d397SChris Mason btrfs_set_node_key(parent, &right_key, pslot + 1); 7045f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 705bb803951SChris Mason } 706bb803951SChris Mason } 7075f39d397SChris Mason if (btrfs_header_nritems(mid) == 1) { 70879f95c82SChris Mason /* 70979f95c82SChris Mason * we're not allowed to leave a node with one item in the 71079f95c82SChris Mason * tree during a delete. A deletion from lower in the tree 71179f95c82SChris Mason * could try to delete the only pointer in this node. 71279f95c82SChris Mason * So, pull some keys from the left. 71379f95c82SChris Mason * There has to be a left pointer at this point because 71479f95c82SChris Mason * otherwise we would have pulled some pointers from the 71579f95c82SChris Mason * right 71679f95c82SChris Mason */ 7175f39d397SChris Mason BUG_ON(!left); 7185f39d397SChris Mason wret = balance_node_right(trans, root, mid, left); 71954aa1f4dSChris Mason if (wret < 0) { 72079f95c82SChris Mason ret = wret; 72154aa1f4dSChris Mason goto enospc; 72254aa1f4dSChris Mason } 72379f95c82SChris Mason BUG_ON(wret == 1); 72479f95c82SChris Mason } 7255f39d397SChris Mason if (btrfs_header_nritems(mid) == 0) { 72679f95c82SChris Mason /* we've managed to empty the middle node, drop it */ 727*db94535dSChris Mason u64 bytenr = mid->start; 728*db94535dSChris Mason u32 blocksize = mid->len; 7295f39d397SChris Mason clean_tree_block(trans, root, mid); 7305f39d397SChris Mason wait_on_tree_block_writeback(root, mid); 7315f39d397SChris Mason free_extent_buffer(mid); 732bb803951SChris Mason mid = NULL; 733e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot); 734bb803951SChris Mason if (wret) 735bb803951SChris Mason ret = wret; 736*db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, blocksize, 1); 737bb803951SChris Mason if (wret) 738bb803951SChris Mason ret = wret; 73979f95c82SChris Mason } else { 74079f95c82SChris Mason /* update the parent key to reflect our changes */ 7415f39d397SChris Mason struct btrfs_disk_key mid_key; 7425f39d397SChris Mason btrfs_node_key(mid, &mid_key, 0); 7435f39d397SChris Mason btrfs_set_node_key(parent, &mid_key, pslot); 7445f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 74579f95c82SChris Mason } 746bb803951SChris Mason 74779f95c82SChris Mason /* update the path */ 7485f39d397SChris Mason if (left) { 7495f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 7505f39d397SChris Mason extent_buffer_get(left); 7515f39d397SChris Mason path->nodes[level] = left; 752bb803951SChris Mason path->slots[level + 1] -= 1; 753bb803951SChris Mason path->slots[level] = orig_slot; 7545f39d397SChris Mason if (mid) 7555f39d397SChris Mason free_extent_buffer(mid); 756bb803951SChris Mason } else { 7575f39d397SChris Mason orig_slot -= btrfs_header_nritems(left); 758bb803951SChris Mason path->slots[level] = orig_slot; 759bb803951SChris Mason } 760bb803951SChris Mason } 76179f95c82SChris Mason /* double check we haven't messed things up */ 762123abc88SChris Mason check_block(root, path, level); 763e20d96d6SChris Mason if (orig_ptr != 7645f39d397SChris Mason btrfs_node_blockptr(path->nodes[level], path->slots[level])) 76579f95c82SChris Mason BUG(); 76654aa1f4dSChris Mason enospc: 7675f39d397SChris Mason if (right) 7685f39d397SChris Mason free_extent_buffer(right); 7695f39d397SChris Mason if (left) 7705f39d397SChris Mason free_extent_buffer(left); 771bb803951SChris Mason return ret; 772bb803951SChris Mason } 773bb803951SChris Mason 774e66f709bSChris Mason /* returns zero if the push worked, non-zero otherwise */ 775e66f709bSChris Mason static int push_nodes_for_insert(struct btrfs_trans_handle *trans, 776e66f709bSChris Mason struct btrfs_root *root, 777e66f709bSChris Mason struct btrfs_path *path, int level) 778e66f709bSChris Mason { 7795f39d397SChris Mason struct extent_buffer *right = NULL; 7805f39d397SChris Mason struct extent_buffer *mid; 7815f39d397SChris Mason struct extent_buffer *left = NULL; 7825f39d397SChris Mason struct extent_buffer *parent = NULL; 783e66f709bSChris Mason int ret = 0; 784e66f709bSChris Mason int wret; 785e66f709bSChris Mason int pslot; 786e66f709bSChris Mason int orig_slot = path->slots[level]; 787e66f709bSChris Mason u64 orig_ptr; 788e66f709bSChris Mason 789e66f709bSChris Mason if (level == 0) 790e66f709bSChris Mason return 1; 791e66f709bSChris Mason 7925f39d397SChris Mason mid = path->nodes[level]; 793e66f709bSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 794e66f709bSChris Mason 795e66f709bSChris Mason if (level < BTRFS_MAX_LEVEL - 1) 7965f39d397SChris Mason parent = path->nodes[level + 1]; 797e66f709bSChris Mason pslot = path->slots[level + 1]; 798e66f709bSChris Mason 7995f39d397SChris Mason if (!parent) 800e66f709bSChris Mason return 1; 801e66f709bSChris Mason 8025f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 803e66f709bSChris Mason 804e66f709bSChris Mason /* first, try to make some room in the middle buffer */ 8055f39d397SChris Mason if (left) { 806e66f709bSChris Mason u32 left_nr; 8075f39d397SChris Mason left_nr = btrfs_header_nritems(left); 80833ade1f8SChris Mason if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 80933ade1f8SChris Mason wret = 1; 81033ade1f8SChris Mason } else { 8115f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, parent, 8125f39d397SChris Mason pslot - 1, &left); 81354aa1f4dSChris Mason if (ret) 81454aa1f4dSChris Mason wret = 1; 81554aa1f4dSChris Mason else { 81654aa1f4dSChris Mason wret = push_node_left(trans, root, 8175f39d397SChris Mason left, mid); 81854aa1f4dSChris Mason } 81933ade1f8SChris Mason } 820e66f709bSChris Mason if (wret < 0) 821e66f709bSChris Mason ret = wret; 822e66f709bSChris Mason if (wret == 0) { 8235f39d397SChris Mason struct btrfs_disk_key disk_key; 824e66f709bSChris Mason orig_slot += left_nr; 8255f39d397SChris Mason btrfs_node_key(mid, &disk_key, 0); 8265f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot); 8275f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 8285f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 8295f39d397SChris Mason path->nodes[level] = left; 830e66f709bSChris Mason path->slots[level + 1] -= 1; 831e66f709bSChris Mason path->slots[level] = orig_slot; 8325f39d397SChris Mason free_extent_buffer(mid); 833e66f709bSChris Mason } else { 834e66f709bSChris Mason orig_slot -= 8355f39d397SChris Mason btrfs_header_nritems(left); 836e66f709bSChris Mason path->slots[level] = orig_slot; 8375f39d397SChris Mason free_extent_buffer(left); 838e66f709bSChris Mason } 839e66f709bSChris Mason return 0; 840e66f709bSChris Mason } 8415f39d397SChris Mason free_extent_buffer(left); 842e66f709bSChris Mason } 8435f39d397SChris Mason right= read_node_slot(root, parent, pslot + 1); 844e66f709bSChris Mason 845e66f709bSChris Mason /* 846e66f709bSChris Mason * then try to empty the right most buffer into the middle 847e66f709bSChris Mason */ 8485f39d397SChris Mason if (right) { 84933ade1f8SChris Mason u32 right_nr; 8505f39d397SChris Mason right_nr = btrfs_header_nritems(right); 85133ade1f8SChris Mason if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 85233ade1f8SChris Mason wret = 1; 85333ade1f8SChris Mason } else { 8545f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, 8555f39d397SChris Mason parent, pslot + 1, 8565f39d397SChris Mason &right); 85754aa1f4dSChris Mason if (ret) 85854aa1f4dSChris Mason wret = 1; 85954aa1f4dSChris Mason else { 86033ade1f8SChris Mason wret = balance_node_right(trans, root, 8615f39d397SChris Mason right, mid); 86233ade1f8SChris Mason } 86354aa1f4dSChris Mason } 864e66f709bSChris Mason if (wret < 0) 865e66f709bSChris Mason ret = wret; 866e66f709bSChris Mason if (wret == 0) { 8675f39d397SChris Mason struct btrfs_disk_key disk_key; 8685f39d397SChris Mason 8695f39d397SChris Mason btrfs_node_key(right, &disk_key, 0); 8705f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot + 1); 8715f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 8725f39d397SChris Mason 8735f39d397SChris Mason if (btrfs_header_nritems(mid) <= orig_slot) { 8745f39d397SChris Mason path->nodes[level] = right; 875e66f709bSChris Mason path->slots[level + 1] += 1; 876e66f709bSChris Mason path->slots[level] = orig_slot - 8775f39d397SChris Mason btrfs_header_nritems(mid); 8785f39d397SChris Mason free_extent_buffer(mid); 879e66f709bSChris Mason } else { 8805f39d397SChris Mason free_extent_buffer(right); 881e66f709bSChris Mason } 882e66f709bSChris Mason return 0; 883e66f709bSChris Mason } 8845f39d397SChris Mason free_extent_buffer(right); 885e66f709bSChris Mason } 886e66f709bSChris Mason return 1; 887e66f709bSChris Mason } 888e66f709bSChris Mason 88974123bd7SChris Mason /* 8903c69faecSChris Mason * readahead one full node of leaves 8913c69faecSChris Mason */ 8923c69faecSChris Mason static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path, 8936702ed49SChris Mason int level, int slot) 8943c69faecSChris Mason { 895*db94535dSChris Mason return; 896*db94535dSChris Mason #if 0 8975f39d397SChris Mason struct extent_buffer *node; 8983c69faecSChris Mason int i; 8993c69faecSChris Mason u32 nritems; 900*db94535dSChris Mason u64 bytenr; 9013c69faecSChris Mason u64 search; 9023c69faecSChris Mason u64 cluster_start; 9033c69faecSChris Mason int ret; 9043c69faecSChris Mason int nread = 0; 9053c69faecSChris Mason int direction = path->reada; 906*db94535dSChris Mason int level; 9073c69faecSChris Mason struct radix_tree_root found; 9083c69faecSChris Mason unsigned long gang[8]; 9095f39d397SChris Mason struct extent_buffer *eb; 9103c69faecSChris Mason 911*db94535dSChris Mason 9126702ed49SChris Mason if (level == 0) 9133c69faecSChris Mason return; 9143c69faecSChris Mason 9156702ed49SChris Mason if (!path->nodes[level]) 9166702ed49SChris Mason return; 9176702ed49SChris Mason 9185f39d397SChris Mason node = path->nodes[level]; 9193c69faecSChris Mason search = btrfs_node_blockptr(node, slot); 9205f39d397SChris Mason eb = btrfs_find_tree_block(root, search); 9215f39d397SChris Mason if (eb) { 9225f39d397SChris Mason free_extent_buffer(eb); 9233c69faecSChris Mason return; 9243c69faecSChris Mason } 9253c69faecSChris Mason 9263c69faecSChris Mason init_bit_radix(&found); 9275f39d397SChris Mason nritems = btrfs_header_nritems(node); 928*db94535dSChris Mason level = btrfs_header_level(node) - 1; 9293c69faecSChris Mason for (i = slot; i < nritems; i++) { 930*db94535dSChris Mason bytenr = btrfs_node_blockptr(node, i); 9313c69faecSChris Mason set_radix_bit(&found, blocknr); 9323c69faecSChris Mason } 9333c69faecSChris Mason if (direction > 0) { 9343c69faecSChris Mason cluster_start = search - 4; 9353c69faecSChris Mason if (cluster_start > search) 9363c69faecSChris Mason cluster_start = 0; 9373c69faecSChris Mason } else 9383c69faecSChris Mason cluster_start = search + 4; 9393c69faecSChris Mason while(1) { 9403c69faecSChris Mason ret = find_first_radix_bit(&found, gang, 0, ARRAY_SIZE(gang)); 9413c69faecSChris Mason if (!ret) 9423c69faecSChris Mason break; 9433c69faecSChris Mason for (i = 0; i < ret; i++) { 9443c69faecSChris Mason blocknr = gang[i]; 9453c69faecSChris Mason clear_radix_bit(&found, blocknr); 9462cc58cf2SChris Mason if (path->reada == 1 && nread > 16) 9473c69faecSChris Mason continue; 948f2183bdeSChris Mason if (close_blocks(cluster_start, blocknr)) { 9493c69faecSChris Mason readahead_tree_block(root, blocknr); 9503c69faecSChris Mason nread++; 9513c69faecSChris Mason cluster_start = blocknr; 9523c69faecSChris Mason } 9533c69faecSChris Mason } 9543c69faecSChris Mason } 955*db94535dSChris Mason #endif 9563c69faecSChris Mason } 9573c69faecSChris Mason /* 95874123bd7SChris Mason * look for key in the tree. path is filled in with nodes along the way 95974123bd7SChris Mason * if key is found, we return zero and you can find the item in the leaf 96074123bd7SChris Mason * level of the path (level 0) 96174123bd7SChris Mason * 96274123bd7SChris Mason * If the key isn't found, the path points to the slot where it should 963aa5d6bedSChris Mason * be inserted, and 1 is returned. If there are other errors during the 964aa5d6bedSChris Mason * search a negative error number is returned. 96597571fd0SChris Mason * 96697571fd0SChris Mason * if ins_len > 0, nodes and leaves will be split as we walk down the 96797571fd0SChris Mason * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if 96897571fd0SChris Mason * possible) 96974123bd7SChris Mason */ 970e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root 971e089f05cSChris Mason *root, struct btrfs_key *key, struct btrfs_path *p, int 972e089f05cSChris Mason ins_len, int cow) 973be0e5c09SChris Mason { 9745f39d397SChris Mason struct extent_buffer *b; 975*db94535dSChris Mason u64 bytenr; 976be0e5c09SChris Mason int slot; 977be0e5c09SChris Mason int ret; 978be0e5c09SChris Mason int level; 9793c69faecSChris Mason int should_reada = p->reada; 9809f3a7427SChris Mason u8 lowest_level = 0; 9819f3a7427SChris Mason 9826702ed49SChris Mason lowest_level = p->lowest_level; 9836702ed49SChris Mason WARN_ON(lowest_level && ins_len); 98422b0ebdaSChris Mason WARN_ON(p->nodes[0] != NULL); 98522b0ebdaSChris Mason WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex)); 986bb803951SChris Mason again: 987bb803951SChris Mason b = root->node; 9885f39d397SChris Mason extent_buffer_get(b); 989eb60ceacSChris Mason while (b) { 9905f39d397SChris Mason level = btrfs_header_level(b); 99102217ed2SChris Mason if (cow) { 99202217ed2SChris Mason int wret; 993e20d96d6SChris Mason wret = btrfs_cow_block(trans, root, b, 994e20d96d6SChris Mason p->nodes[level + 1], 995e20d96d6SChris Mason p->slots[level + 1], 996252c38f0SYan &b); 99754aa1f4dSChris Mason if (wret) { 9985f39d397SChris Mason free_extent_buffer(b); 99954aa1f4dSChris Mason return wret; 100054aa1f4dSChris Mason } 100102217ed2SChris Mason } 100202217ed2SChris Mason BUG_ON(!cow && ins_len); 10035f39d397SChris Mason if (level != btrfs_header_level(b)) 10042c90e5d6SChris Mason WARN_ON(1); 10055f39d397SChris Mason level = btrfs_header_level(b); 1006eb60ceacSChris Mason p->nodes[level] = b; 1007123abc88SChris Mason ret = check_block(root, p, level); 1008aa5d6bedSChris Mason if (ret) 1009aa5d6bedSChris Mason return -1; 10105f39d397SChris Mason ret = bin_search(b, key, level, &slot); 10115f39d397SChris Mason if (level != 0) { 1012be0e5c09SChris Mason if (ret && slot > 0) 1013be0e5c09SChris Mason slot -= 1; 1014be0e5c09SChris Mason p->slots[level] = slot; 10155f39d397SChris Mason if (ins_len > 0 && btrfs_header_nritems(b) >= 1016d4dbff95SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 1017e089f05cSChris Mason int sret = split_node(trans, root, p, level); 10185c680ed6SChris Mason BUG_ON(sret > 0); 10195c680ed6SChris Mason if (sret) 10205c680ed6SChris Mason return sret; 10215c680ed6SChris Mason b = p->nodes[level]; 10225c680ed6SChris Mason slot = p->slots[level]; 1023bb803951SChris Mason } else if (ins_len < 0) { 1024e089f05cSChris Mason int sret = balance_level(trans, root, p, 1025e089f05cSChris Mason level); 1026bb803951SChris Mason if (sret) 1027bb803951SChris Mason return sret; 1028bb803951SChris Mason b = p->nodes[level]; 1029f510cfecSChris Mason if (!b) { 1030f510cfecSChris Mason btrfs_release_path(NULL, p); 1031bb803951SChris Mason goto again; 1032f510cfecSChris Mason } 1033bb803951SChris Mason slot = p->slots[level]; 10345f39d397SChris Mason BUG_ON(btrfs_header_nritems(b) == 1); 10355c680ed6SChris Mason } 10369f3a7427SChris Mason /* this is only true while dropping a snapshot */ 10379f3a7427SChris Mason if (level == lowest_level) 10389f3a7427SChris Mason break; 1039*db94535dSChris Mason bytenr = btrfs_node_blockptr(b, slot); 10406702ed49SChris Mason if (should_reada) 10416702ed49SChris Mason reada_for_search(root, p, level, slot); 1042*db94535dSChris Mason b = read_tree_block(root, bytenr, 1043*db94535dSChris Mason btrfs_level_size(root, level - 1)); 1044be0e5c09SChris Mason } else { 1045be0e5c09SChris Mason p->slots[level] = slot; 10465f39d397SChris Mason if (ins_len > 0 && btrfs_leaf_free_space(root, b) < 10470783fcfcSChris Mason sizeof(struct btrfs_item) + ins_len) { 1048d4dbff95SChris Mason int sret = split_leaf(trans, root, key, 1049d4dbff95SChris Mason p, ins_len); 10505c680ed6SChris Mason BUG_ON(sret > 0); 10515c680ed6SChris Mason if (sret) 10525c680ed6SChris Mason return sret; 10535c680ed6SChris Mason } 1054be0e5c09SChris Mason return ret; 1055be0e5c09SChris Mason } 1056be0e5c09SChris Mason } 1057aa5d6bedSChris Mason return 1; 1058be0e5c09SChris Mason } 1059be0e5c09SChris Mason 106074123bd7SChris Mason /* 106174123bd7SChris Mason * adjust the pointers going up the tree, starting at level 106274123bd7SChris Mason * making sure the right key of each node is points to 'key'. 106374123bd7SChris Mason * This is used after shifting pointers to the left, so it stops 106474123bd7SChris Mason * fixing up pointers when a given leaf/node is not in slot 0 of the 106574123bd7SChris Mason * higher levels 1066aa5d6bedSChris Mason * 1067aa5d6bedSChris Mason * If this fails to write a tree block, it returns -1, but continues 1068aa5d6bedSChris Mason * fixing up the blocks in ram so the tree is consistent. 106974123bd7SChris Mason */ 10705f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans, 10715f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 10725f39d397SChris Mason struct btrfs_disk_key *key, int level) 1073be0e5c09SChris Mason { 1074be0e5c09SChris Mason int i; 1075aa5d6bedSChris Mason int ret = 0; 10765f39d397SChris Mason struct extent_buffer *t; 10775f39d397SChris Mason 1078234b63a0SChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 1079be0e5c09SChris Mason int tslot = path->slots[i]; 1080eb60ceacSChris Mason if (!path->nodes[i]) 1081be0e5c09SChris Mason break; 10825f39d397SChris Mason t = path->nodes[i]; 10835f39d397SChris Mason btrfs_set_node_key(t, key, tslot); 1084d6025579SChris Mason btrfs_mark_buffer_dirty(path->nodes[i]); 1085be0e5c09SChris Mason if (tslot != 0) 1086be0e5c09SChris Mason break; 1087be0e5c09SChris Mason } 1088aa5d6bedSChris Mason return ret; 1089be0e5c09SChris Mason } 1090be0e5c09SChris Mason 109174123bd7SChris Mason /* 109274123bd7SChris Mason * try to push data from one node into the next node left in the 109379f95c82SChris Mason * tree. 1094aa5d6bedSChris Mason * 1095aa5d6bedSChris Mason * returns 0 if some ptrs were pushed left, < 0 if there was some horrible 1096aa5d6bedSChris Mason * error, and > 0 if there was no room in the left hand block. 109774123bd7SChris Mason */ 1098e089f05cSChris Mason static int push_node_left(struct btrfs_trans_handle *trans, struct btrfs_root 10995f39d397SChris Mason *root, struct extent_buffer *dst, 11005f39d397SChris Mason struct extent_buffer *src) 1101be0e5c09SChris Mason { 1102be0e5c09SChris Mason int push_items = 0; 1103bb803951SChris Mason int src_nritems; 1104bb803951SChris Mason int dst_nritems; 1105aa5d6bedSChris Mason int ret = 0; 1106be0e5c09SChris Mason 11075f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 11085f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1109123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 111054aa1f4dSChris Mason 1111eb60ceacSChris Mason if (push_items <= 0) { 1112be0e5c09SChris Mason return 1; 1113eb60ceacSChris Mason } 1114be0e5c09SChris Mason 1115be0e5c09SChris Mason if (src_nritems < push_items) 1116be0e5c09SChris Mason push_items = src_nritems; 111779f95c82SChris Mason 11185f39d397SChris Mason copy_extent_buffer(dst, src, 11195f39d397SChris Mason btrfs_node_key_ptr_offset(dst_nritems), 11205f39d397SChris Mason btrfs_node_key_ptr_offset(0), 1121123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 11225f39d397SChris Mason 1123bb803951SChris Mason if (push_items < src_nritems) { 11245f39d397SChris Mason memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0), 11255f39d397SChris Mason btrfs_node_key_ptr_offset(push_items), 1126e2fa7227SChris Mason (src_nritems - push_items) * 1127123abc88SChris Mason sizeof(struct btrfs_key_ptr)); 1128bb803951SChris Mason } 11295f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 11305f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 11315f39d397SChris Mason btrfs_mark_buffer_dirty(src); 11325f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 1133bb803951SChris Mason return ret; 1134be0e5c09SChris Mason } 1135be0e5c09SChris Mason 113697571fd0SChris Mason /* 113779f95c82SChris Mason * try to push data from one node into the next node right in the 113879f95c82SChris Mason * tree. 113979f95c82SChris Mason * 114079f95c82SChris Mason * returns 0 if some ptrs were pushed, < 0 if there was some horrible 114179f95c82SChris Mason * error, and > 0 if there was no room in the right hand block. 114279f95c82SChris Mason * 114379f95c82SChris Mason * this will only push up to 1/2 the contents of the left node over 114479f95c82SChris Mason */ 11455f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 11465f39d397SChris Mason struct btrfs_root *root, 11475f39d397SChris Mason struct extent_buffer *dst, 11485f39d397SChris Mason struct extent_buffer *src) 114979f95c82SChris Mason { 115079f95c82SChris Mason int push_items = 0; 115179f95c82SChris Mason int max_push; 115279f95c82SChris Mason int src_nritems; 115379f95c82SChris Mason int dst_nritems; 115479f95c82SChris Mason int ret = 0; 115579f95c82SChris Mason 11565f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 11575f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1158123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 11595f39d397SChris Mason if (push_items <= 0) 116079f95c82SChris Mason return 1; 116179f95c82SChris Mason 116279f95c82SChris Mason max_push = src_nritems / 2 + 1; 116379f95c82SChris Mason /* don't try to empty the node */ 1164252c38f0SYan if (max_push >= src_nritems) 116579f95c82SChris Mason return 1; 1166252c38f0SYan 116779f95c82SChris Mason if (max_push < push_items) 116879f95c82SChris Mason push_items = max_push; 116979f95c82SChris Mason 11705f39d397SChris Mason memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items), 11715f39d397SChris Mason btrfs_node_key_ptr_offset(0), 11725f39d397SChris Mason (dst_nritems) * 11735f39d397SChris Mason sizeof(struct btrfs_key_ptr)); 1174d6025579SChris Mason 11755f39d397SChris Mason copy_extent_buffer(dst, src, 11765f39d397SChris Mason btrfs_node_key_ptr_offset(0), 11775f39d397SChris Mason btrfs_node_key_ptr_offset(src_nritems - push_items), 1178123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 117979f95c82SChris Mason 11805f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 11815f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 118279f95c82SChris Mason 11835f39d397SChris Mason btrfs_mark_buffer_dirty(src); 11845f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 118579f95c82SChris Mason return ret; 118679f95c82SChris Mason } 118779f95c82SChris Mason 118879f95c82SChris Mason /* 118997571fd0SChris Mason * helper function to insert a new root level in the tree. 119097571fd0SChris Mason * A new node is allocated, and a single item is inserted to 119197571fd0SChris Mason * point to the existing root 1192aa5d6bedSChris Mason * 1193aa5d6bedSChris Mason * returns zero on success or < 0 on failure. 119497571fd0SChris Mason */ 11955f39d397SChris Mason static int insert_new_root(struct btrfs_trans_handle *trans, 11965f39d397SChris Mason struct btrfs_root *root, 11975f39d397SChris Mason struct btrfs_path *path, int level) 119874123bd7SChris Mason { 11995f39d397SChris Mason struct extent_buffer *lower; 12005f39d397SChris Mason struct extent_buffer *c; 12015f39d397SChris Mason struct btrfs_disk_key lower_key; 12025c680ed6SChris Mason 12035c680ed6SChris Mason BUG_ON(path->nodes[level]); 12045c680ed6SChris Mason BUG_ON(path->nodes[level-1] != root->node); 12055c680ed6SChris Mason 1206*db94535dSChris Mason c = btrfs_alloc_free_block(trans, root, root->nodesize, 1207*db94535dSChris Mason root->node->start, 0); 12085f39d397SChris Mason if (IS_ERR(c)) 12095f39d397SChris Mason return PTR_ERR(c); 12105f39d397SChris Mason memset_extent_buffer(c, 0, 0, root->nodesize); 12115f39d397SChris Mason btrfs_set_header_nritems(c, 1); 12125f39d397SChris Mason btrfs_set_header_level(c, level); 1213*db94535dSChris Mason btrfs_set_header_bytenr(c, c->start); 12145f39d397SChris Mason btrfs_set_header_generation(c, trans->transid); 12155f39d397SChris Mason btrfs_set_header_owner(c, root->root_key.objectid); 12165f39d397SChris Mason lower = path->nodes[level-1]; 1217d5719762SChris Mason 12185f39d397SChris Mason write_extent_buffer(c, root->fs_info->fsid, 12195f39d397SChris Mason (unsigned long)btrfs_header_fsid(c), 12205f39d397SChris Mason BTRFS_FSID_SIZE); 12215f39d397SChris Mason if (level == 1) 12225f39d397SChris Mason btrfs_item_key(lower, &lower_key, 0); 12235f39d397SChris Mason else 12245f39d397SChris Mason btrfs_node_key(lower, &lower_key, 0); 12255f39d397SChris Mason btrfs_set_node_key(c, &lower_key, 0); 1226*db94535dSChris Mason btrfs_set_node_blockptr(c, 0, lower->start); 12275f39d397SChris Mason 12285f39d397SChris Mason btrfs_mark_buffer_dirty(c); 1229d5719762SChris Mason 1230cfaa7295SChris Mason /* the super has an extra ref to root->node */ 12315f39d397SChris Mason free_extent_buffer(root->node); 12325f39d397SChris Mason root->node = c; 12335f39d397SChris Mason extent_buffer_get(c); 12345f39d397SChris Mason path->nodes[level] = c; 123574123bd7SChris Mason path->slots[level] = 0; 123674123bd7SChris Mason return 0; 123774123bd7SChris Mason } 12385c680ed6SChris Mason 12395c680ed6SChris Mason /* 12405c680ed6SChris Mason * worker function to insert a single pointer in a node. 12415c680ed6SChris Mason * the node should have enough room for the pointer already 124297571fd0SChris Mason * 12435c680ed6SChris Mason * slot and level indicate where you want the key to go, and 12445c680ed6SChris Mason * blocknr is the block the key points to. 1245aa5d6bedSChris Mason * 1246aa5d6bedSChris Mason * returns zero on success and < 0 on any error 12475c680ed6SChris Mason */ 1248e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root 1249e089f05cSChris Mason *root, struct btrfs_path *path, struct btrfs_disk_key 1250*db94535dSChris Mason *key, u64 bytenr, int slot, int level) 12515c680ed6SChris Mason { 12525f39d397SChris Mason struct extent_buffer *lower; 12535c680ed6SChris Mason int nritems; 12545c680ed6SChris Mason 12555c680ed6SChris Mason BUG_ON(!path->nodes[level]); 12565f39d397SChris Mason lower = path->nodes[level]; 12575f39d397SChris Mason nritems = btrfs_header_nritems(lower); 125874123bd7SChris Mason if (slot > nritems) 125974123bd7SChris Mason BUG(); 1260123abc88SChris Mason if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root)) 126174123bd7SChris Mason BUG(); 126274123bd7SChris Mason if (slot != nritems) { 12635f39d397SChris Mason memmove_extent_buffer(lower, 12645f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 12655f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 1266123abc88SChris Mason (nritems - slot) * sizeof(struct btrfs_key_ptr)); 126774123bd7SChris Mason } 12685f39d397SChris Mason btrfs_set_node_key(lower, key, slot); 1269*db94535dSChris Mason btrfs_set_node_blockptr(lower, slot, bytenr); 12705f39d397SChris Mason btrfs_set_header_nritems(lower, nritems + 1); 12715f39d397SChris Mason btrfs_mark_buffer_dirty(lower); 127274123bd7SChris Mason return 0; 127374123bd7SChris Mason } 127474123bd7SChris Mason 127597571fd0SChris Mason /* 127697571fd0SChris Mason * split the node at the specified level in path in two. 127797571fd0SChris Mason * The path is corrected to point to the appropriate node after the split 127897571fd0SChris Mason * 127997571fd0SChris Mason * Before splitting this tries to make some room in the node by pushing 128097571fd0SChris Mason * left and right, if either one works, it returns right away. 1281aa5d6bedSChris Mason * 1282aa5d6bedSChris Mason * returns 0 on success and < 0 on failure 128397571fd0SChris Mason */ 1284e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 1285e089f05cSChris Mason *root, struct btrfs_path *path, int level) 1286be0e5c09SChris Mason { 12875f39d397SChris Mason struct extent_buffer *c; 12885f39d397SChris Mason struct extent_buffer *split; 12895f39d397SChris Mason struct btrfs_disk_key disk_key; 1290be0e5c09SChris Mason int mid; 12915c680ed6SChris Mason int ret; 1292aa5d6bedSChris Mason int wret; 12937518a238SChris Mason u32 c_nritems; 1294be0e5c09SChris Mason 12955f39d397SChris Mason c = path->nodes[level]; 12965f39d397SChris Mason if (c == root->node) { 12975c680ed6SChris Mason /* trying to split the root, lets make a new one */ 1298e089f05cSChris Mason ret = insert_new_root(trans, root, path, level + 1); 12995c680ed6SChris Mason if (ret) 13005c680ed6SChris Mason return ret; 1301e66f709bSChris Mason } else { 1302e66f709bSChris Mason ret = push_nodes_for_insert(trans, root, path, level); 13035f39d397SChris Mason c = path->nodes[level]; 13045f39d397SChris Mason if (!ret && btrfs_header_nritems(c) < 1305e66f709bSChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 1) 1306e66f709bSChris Mason return 0; 130754aa1f4dSChris Mason if (ret < 0) 130854aa1f4dSChris Mason return ret; 13095c680ed6SChris Mason } 1310e66f709bSChris Mason 13115f39d397SChris Mason c_nritems = btrfs_header_nritems(c); 1312*db94535dSChris Mason split = btrfs_alloc_free_block(trans, root, root->nodesize, 1313*db94535dSChris Mason c->start, 0); 13145f39d397SChris Mason if (IS_ERR(split)) 13155f39d397SChris Mason return PTR_ERR(split); 131654aa1f4dSChris Mason 13175f39d397SChris Mason btrfs_set_header_flags(split, btrfs_header_flags(c)); 13185f39d397SChris Mason btrfs_set_header_level(split, btrfs_header_level(c)); 1319*db94535dSChris Mason btrfs_set_header_bytenr(split, split->start); 13205f39d397SChris Mason btrfs_set_header_generation(split, trans->transid); 13215f39d397SChris Mason btrfs_set_header_owner(split, root->root_key.objectid); 13225f39d397SChris Mason write_extent_buffer(split, root->fs_info->fsid, 13235f39d397SChris Mason (unsigned long)btrfs_header_fsid(split), 13245f39d397SChris Mason BTRFS_FSID_SIZE); 13255f39d397SChris Mason 13267518a238SChris Mason mid = (c_nritems + 1) / 2; 13275f39d397SChris Mason 13285f39d397SChris Mason copy_extent_buffer(split, c, 13295f39d397SChris Mason btrfs_node_key_ptr_offset(0), 13305f39d397SChris Mason btrfs_node_key_ptr_offset(mid), 1331123abc88SChris Mason (c_nritems - mid) * sizeof(struct btrfs_key_ptr)); 13325f39d397SChris Mason btrfs_set_header_nritems(split, c_nritems - mid); 13335f39d397SChris Mason btrfs_set_header_nritems(c, mid); 1334aa5d6bedSChris Mason ret = 0; 1335aa5d6bedSChris Mason 13365f39d397SChris Mason btrfs_mark_buffer_dirty(c); 13375f39d397SChris Mason btrfs_mark_buffer_dirty(split); 13385f39d397SChris Mason 13395f39d397SChris Mason btrfs_node_key(split, &disk_key, 0); 1340*db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, split->start, 13415f39d397SChris Mason path->slots[level + 1] + 1, 1342123abc88SChris Mason level + 1); 1343aa5d6bedSChris Mason if (wret) 1344aa5d6bedSChris Mason ret = wret; 1345aa5d6bedSChris Mason 13465de08d7dSChris Mason if (path->slots[level] >= mid) { 13475c680ed6SChris Mason path->slots[level] -= mid; 13485f39d397SChris Mason free_extent_buffer(c); 13495f39d397SChris Mason path->nodes[level] = split; 13505c680ed6SChris Mason path->slots[level + 1] += 1; 1351eb60ceacSChris Mason } else { 13525f39d397SChris Mason free_extent_buffer(split); 1353be0e5c09SChris Mason } 1354aa5d6bedSChris Mason return ret; 1355be0e5c09SChris Mason } 1356be0e5c09SChris Mason 135774123bd7SChris Mason /* 135874123bd7SChris Mason * how many bytes are required to store the items in a leaf. start 135974123bd7SChris Mason * and nr indicate which items in the leaf to check. This totals up the 136074123bd7SChris Mason * space used both by the item structs and the item data 136174123bd7SChris Mason */ 13625f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr) 1363be0e5c09SChris Mason { 1364be0e5c09SChris Mason int data_len; 13655f39d397SChris Mason int nritems = btrfs_header_nritems(l); 1366d4dbff95SChris Mason int end = min(nritems, start + nr) - 1; 1367be0e5c09SChris Mason 1368be0e5c09SChris Mason if (!nr) 1369be0e5c09SChris Mason return 0; 13705f39d397SChris Mason data_len = btrfs_item_end_nr(l, start); 13715f39d397SChris Mason data_len = data_len - btrfs_item_offset_nr(l, end); 13720783fcfcSChris Mason data_len += sizeof(struct btrfs_item) * nr; 1373d4dbff95SChris Mason WARN_ON(data_len < 0); 1374be0e5c09SChris Mason return data_len; 1375be0e5c09SChris Mason } 1376be0e5c09SChris Mason 137774123bd7SChris Mason /* 1378d4dbff95SChris Mason * The space between the end of the leaf items and 1379d4dbff95SChris Mason * the start of the leaf data. IOW, how much room 1380d4dbff95SChris Mason * the leaf has left for both items and data 1381d4dbff95SChris Mason */ 13825f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf) 1383d4dbff95SChris Mason { 13845f39d397SChris Mason int nritems = btrfs_header_nritems(leaf); 13855f39d397SChris Mason int ret; 13865f39d397SChris Mason ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems); 13875f39d397SChris Mason if (ret < 0) { 13885f39d397SChris Mason printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n", 13895f39d397SChris Mason ret, BTRFS_LEAF_DATA_SIZE(root), 13905f39d397SChris Mason leaf_space_used(leaf, 0, nritems), nritems); 13915f39d397SChris Mason } 13925f39d397SChris Mason return ret; 1393d4dbff95SChris Mason } 1394d4dbff95SChris Mason 1395d4dbff95SChris Mason /* 139600ec4c51SChris Mason * push some data in the path leaf to the right, trying to free up at 139700ec4c51SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 1398aa5d6bedSChris Mason * 1399aa5d6bedSChris Mason * returns 1 if the push failed because the other node didn't have enough 1400aa5d6bedSChris Mason * room, 0 if everything worked out and < 0 if there were major errors. 140100ec4c51SChris Mason */ 1402e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root 1403e089f05cSChris Mason *root, struct btrfs_path *path, int data_size) 140400ec4c51SChris Mason { 14055f39d397SChris Mason struct extent_buffer *left = path->nodes[0]; 14065f39d397SChris Mason struct extent_buffer *right; 14075f39d397SChris Mason struct extent_buffer *upper; 14085f39d397SChris Mason struct btrfs_disk_key disk_key; 140900ec4c51SChris Mason int slot; 141000ec4c51SChris Mason int i; 141100ec4c51SChris Mason int free_space; 141200ec4c51SChris Mason int push_space = 0; 141300ec4c51SChris Mason int push_items = 0; 14140783fcfcSChris Mason struct btrfs_item *item; 14157518a238SChris Mason u32 left_nritems; 14167518a238SChris Mason u32 right_nritems; 14175f39d397SChris Mason u32 data_end; 1418*db94535dSChris Mason u32 this_item_size; 141954aa1f4dSChris Mason int ret; 142000ec4c51SChris Mason 142100ec4c51SChris Mason slot = path->slots[1]; 142200ec4c51SChris Mason if (!path->nodes[1]) { 142300ec4c51SChris Mason return 1; 142400ec4c51SChris Mason } 142500ec4c51SChris Mason upper = path->nodes[1]; 14265f39d397SChris Mason if (slot >= btrfs_header_nritems(upper) - 1) 142700ec4c51SChris Mason return 1; 14285f39d397SChris Mason 1429*db94535dSChris Mason right = read_tree_block(root, btrfs_node_blockptr(upper, slot + 1), 1430*db94535dSChris Mason root->leafsize); 1431123abc88SChris Mason free_space = btrfs_leaf_free_space(root, right); 14320783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 14335f39d397SChris Mason free_extent_buffer(right); 143402217ed2SChris Mason return 1; 143502217ed2SChris Mason } 143602217ed2SChris Mason 14375f39d397SChris Mason /* cow and double check */ 14385f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, upper, 14395f39d397SChris Mason slot + 1, &right); 14405f39d397SChris Mason if (ret) { 14415f39d397SChris Mason free_extent_buffer(right); 1442a429e513SChris Mason return 1; 1443a429e513SChris Mason } 14445f39d397SChris Mason free_space = btrfs_leaf_free_space(root, right); 14455f39d397SChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 14465f39d397SChris Mason free_extent_buffer(right); 14475f39d397SChris Mason return 1; 14485f39d397SChris Mason } 14495f39d397SChris Mason 14505f39d397SChris Mason left_nritems = btrfs_header_nritems(left); 14515f39d397SChris Mason if (left_nritems == 0) { 14525f39d397SChris Mason free_extent_buffer(right); 14535f39d397SChris Mason return 1; 14545f39d397SChris Mason } 14555f39d397SChris Mason 1456a429e513SChris Mason for (i = left_nritems - 1; i >= 1; i--) { 14575f39d397SChris Mason item = btrfs_item_nr(left, i); 1458*db94535dSChris Mason 145900ec4c51SChris Mason if (path->slots[0] == i) 146000ec4c51SChris Mason push_space += data_size + sizeof(*item); 1461*db94535dSChris Mason 1462*db94535dSChris Mason if (!left->map_token) { 1463*db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 1464*db94535dSChris Mason sizeof(struct btrfs_item), 1465*db94535dSChris Mason &left->map_token, &left->kaddr, 1466*db94535dSChris Mason &left->map_start, &left->map_len, 1467*db94535dSChris Mason KM_USER1); 1468*db94535dSChris Mason } 1469*db94535dSChris Mason 1470*db94535dSChris Mason this_item_size = btrfs_item_size(left, item); 1471*db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 147200ec4c51SChris Mason break; 147300ec4c51SChris Mason push_items++; 1474*db94535dSChris Mason push_space += this_item_size + sizeof(*item); 1475*db94535dSChris Mason } 1476*db94535dSChris Mason if (left->map_token) { 1477*db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 1478*db94535dSChris Mason left->map_token = NULL; 147900ec4c51SChris Mason } 14805f39d397SChris Mason 148100ec4c51SChris Mason if (push_items == 0) { 14825f39d397SChris Mason free_extent_buffer(right); 148300ec4c51SChris Mason return 1; 148400ec4c51SChris Mason } 14855f39d397SChris Mason 1486a429e513SChris Mason if (push_items == left_nritems) 1487a429e513SChris Mason WARN_ON(1); 14885f39d397SChris Mason 148900ec4c51SChris Mason /* push left to right */ 14905f39d397SChris Mason right_nritems = btrfs_header_nritems(right); 14915f39d397SChris Mason push_space = btrfs_item_end_nr(left, left_nritems - push_items); 1492123abc88SChris Mason push_space -= leaf_data_end(root, left); 14935f39d397SChris Mason 149400ec4c51SChris Mason /* make room in the right data area */ 14955f39d397SChris Mason data_end = leaf_data_end(root, right); 14965f39d397SChris Mason memmove_extent_buffer(right, 14975f39d397SChris Mason btrfs_leaf_data(right) + data_end - push_space, 14985f39d397SChris Mason btrfs_leaf_data(right) + data_end, 14995f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root) - data_end); 15005f39d397SChris Mason 150100ec4c51SChris Mason /* copy from the left data area */ 15025f39d397SChris Mason copy_extent_buffer(right, left, btrfs_leaf_data(right) + 1503d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 1504d6025579SChris Mason btrfs_leaf_data(left) + leaf_data_end(root, left), 1505d6025579SChris Mason push_space); 15065f39d397SChris Mason 15075f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(push_items), 15085f39d397SChris Mason btrfs_item_nr_offset(0), 15090783fcfcSChris Mason right_nritems * sizeof(struct btrfs_item)); 15105f39d397SChris Mason 151100ec4c51SChris Mason /* copy the items from left to right */ 15125f39d397SChris Mason copy_extent_buffer(right, left, btrfs_item_nr_offset(0), 15135f39d397SChris Mason btrfs_item_nr_offset(left_nritems - push_items), 15140783fcfcSChris Mason push_items * sizeof(struct btrfs_item)); 151500ec4c51SChris Mason 151600ec4c51SChris Mason /* update the item pointers */ 15177518a238SChris Mason right_nritems += push_items; 15185f39d397SChris Mason btrfs_set_header_nritems(right, right_nritems); 1519123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 1520*db94535dSChris Mason 15217518a238SChris Mason for (i = 0; i < right_nritems; i++) { 15225f39d397SChris Mason item = btrfs_item_nr(right, i); 1523*db94535dSChris Mason if (!right->map_token) { 1524*db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1525*db94535dSChris Mason sizeof(struct btrfs_item), 1526*db94535dSChris Mason &right->map_token, &right->kaddr, 1527*db94535dSChris Mason &right->map_start, &right->map_len, 1528*db94535dSChris Mason KM_USER1); 1529*db94535dSChris Mason } 1530*db94535dSChris Mason push_space -= btrfs_item_size(right, item); 1531*db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 1532*db94535dSChris Mason } 1533*db94535dSChris Mason 1534*db94535dSChris Mason if (right->map_token) { 1535*db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1536*db94535dSChris Mason right->map_token = NULL; 153700ec4c51SChris Mason } 15387518a238SChris Mason left_nritems -= push_items; 15395f39d397SChris Mason btrfs_set_header_nritems(left, left_nritems); 154000ec4c51SChris Mason 15415f39d397SChris Mason btrfs_mark_buffer_dirty(left); 15425f39d397SChris Mason btrfs_mark_buffer_dirty(right); 1543a429e513SChris Mason 15445f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 15455f39d397SChris Mason btrfs_set_node_key(upper, &disk_key, slot + 1); 1546d6025579SChris Mason btrfs_mark_buffer_dirty(upper); 154702217ed2SChris Mason 154800ec4c51SChris Mason /* then fixup the leaf pointer in the path */ 15497518a238SChris Mason if (path->slots[0] >= left_nritems) { 15507518a238SChris Mason path->slots[0] -= left_nritems; 15515f39d397SChris Mason free_extent_buffer(path->nodes[0]); 15525f39d397SChris Mason path->nodes[0] = right; 155300ec4c51SChris Mason path->slots[1] += 1; 155400ec4c51SChris Mason } else { 15555f39d397SChris Mason free_extent_buffer(right); 155600ec4c51SChris Mason } 155700ec4c51SChris Mason return 0; 155800ec4c51SChris Mason } 155900ec4c51SChris Mason /* 156074123bd7SChris Mason * push some data in the path leaf to the left, trying to free up at 156174123bd7SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 156274123bd7SChris Mason */ 1563e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root 1564e089f05cSChris Mason *root, struct btrfs_path *path, int data_size) 1565be0e5c09SChris Mason { 15665f39d397SChris Mason struct btrfs_disk_key disk_key; 15675f39d397SChris Mason struct extent_buffer *right = path->nodes[0]; 15685f39d397SChris Mason struct extent_buffer *left; 1569be0e5c09SChris Mason int slot; 1570be0e5c09SChris Mason int i; 1571be0e5c09SChris Mason int free_space; 1572be0e5c09SChris Mason int push_space = 0; 1573be0e5c09SChris Mason int push_items = 0; 15740783fcfcSChris Mason struct btrfs_item *item; 15757518a238SChris Mason u32 old_left_nritems; 15765f39d397SChris Mason u32 right_nritems; 1577aa5d6bedSChris Mason int ret = 0; 1578aa5d6bedSChris Mason int wret; 1579*db94535dSChris Mason u32 this_item_size; 1580*db94535dSChris Mason u32 old_left_item_size; 1581be0e5c09SChris Mason 1582be0e5c09SChris Mason slot = path->slots[1]; 15835f39d397SChris Mason if (slot == 0) 1584be0e5c09SChris Mason return 1; 15855f39d397SChris Mason if (!path->nodes[1]) 1586be0e5c09SChris Mason return 1; 15875f39d397SChris Mason 15885f39d397SChris Mason left = read_tree_block(root, btrfs_node_blockptr(path->nodes[1], 1589*db94535dSChris Mason slot - 1), root->leafsize); 1590123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 15910783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 15925f39d397SChris Mason free_extent_buffer(left); 1593be0e5c09SChris Mason return 1; 1594be0e5c09SChris Mason } 159502217ed2SChris Mason 159602217ed2SChris Mason /* cow and double check */ 15975f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, 15985f39d397SChris Mason path->nodes[1], slot - 1, &left); 159954aa1f4dSChris Mason if (ret) { 160054aa1f4dSChris Mason /* we hit -ENOSPC, but it isn't fatal here */ 16015f39d397SChris Mason free_extent_buffer(left); 160254aa1f4dSChris Mason return 1; 160354aa1f4dSChris Mason } 1604123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 16050783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 16065f39d397SChris Mason free_extent_buffer(left); 160702217ed2SChris Mason return 1; 160802217ed2SChris Mason } 160902217ed2SChris Mason 16105f39d397SChris Mason right_nritems = btrfs_header_nritems(right); 16115f39d397SChris Mason if (right_nritems == 0) { 16125f39d397SChris Mason free_extent_buffer(left); 1613a429e513SChris Mason return 1; 1614a429e513SChris Mason } 1615a429e513SChris Mason 16165f39d397SChris Mason for (i = 0; i < right_nritems - 1; i++) { 16175f39d397SChris Mason item = btrfs_item_nr(right, i); 1618*db94535dSChris Mason if (!right->map_token) { 1619*db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1620*db94535dSChris Mason sizeof(struct btrfs_item), 1621*db94535dSChris Mason &right->map_token, &right->kaddr, 1622*db94535dSChris Mason &right->map_start, &right->map_len, 1623*db94535dSChris Mason KM_USER1); 1624*db94535dSChris Mason } 1625*db94535dSChris Mason 1626be0e5c09SChris Mason if (path->slots[0] == i) 1627be0e5c09SChris Mason push_space += data_size + sizeof(*item); 1628*db94535dSChris Mason 1629*db94535dSChris Mason this_item_size = btrfs_item_size(right, item); 1630*db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 1631be0e5c09SChris Mason break; 1632*db94535dSChris Mason 1633be0e5c09SChris Mason push_items++; 1634*db94535dSChris Mason push_space += this_item_size + sizeof(*item); 1635be0e5c09SChris Mason } 1636*db94535dSChris Mason 1637*db94535dSChris Mason if (right->map_token) { 1638*db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1639*db94535dSChris Mason right->map_token = NULL; 1640*db94535dSChris Mason } 1641*db94535dSChris Mason 1642be0e5c09SChris Mason if (push_items == 0) { 16435f39d397SChris Mason free_extent_buffer(left); 1644be0e5c09SChris Mason return 1; 1645be0e5c09SChris Mason } 16465f39d397SChris Mason if (push_items == btrfs_header_nritems(right)) 1647a429e513SChris Mason WARN_ON(1); 16485f39d397SChris Mason 1649be0e5c09SChris Mason /* push data from right to left */ 16505f39d397SChris Mason copy_extent_buffer(left, right, 16515f39d397SChris Mason btrfs_item_nr_offset(btrfs_header_nritems(left)), 16525f39d397SChris Mason btrfs_item_nr_offset(0), 16535f39d397SChris Mason push_items * sizeof(struct btrfs_item)); 16545f39d397SChris Mason 1655123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root) - 16565f39d397SChris Mason btrfs_item_offset_nr(right, push_items -1); 16575f39d397SChris Mason 16585f39d397SChris Mason copy_extent_buffer(left, right, btrfs_leaf_data(left) + 1659d6025579SChris Mason leaf_data_end(root, left) - push_space, 1660123abc88SChris Mason btrfs_leaf_data(right) + 16615f39d397SChris Mason btrfs_item_offset_nr(right, push_items - 1), 1662be0e5c09SChris Mason push_space); 16635f39d397SChris Mason old_left_nritems = btrfs_header_nritems(left); 1664eb60ceacSChris Mason BUG_ON(old_left_nritems < 0); 1665eb60ceacSChris Mason 1666*db94535dSChris Mason old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1); 1667be0e5c09SChris Mason for (i = old_left_nritems; i < old_left_nritems + push_items; i++) { 16685f39d397SChris Mason u32 ioff; 1669*db94535dSChris Mason 16705f39d397SChris Mason item = btrfs_item_nr(left, i); 1671*db94535dSChris Mason if (!left->map_token) { 1672*db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 1673*db94535dSChris Mason sizeof(struct btrfs_item), 1674*db94535dSChris Mason &left->map_token, &left->kaddr, 1675*db94535dSChris Mason &left->map_start, &left->map_len, 1676*db94535dSChris Mason KM_USER1); 1677*db94535dSChris Mason } 1678*db94535dSChris Mason 16795f39d397SChris Mason ioff = btrfs_item_offset(left, item); 16805f39d397SChris Mason btrfs_set_item_offset(left, item, 1681*db94535dSChris Mason ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size)); 1682be0e5c09SChris Mason } 16835f39d397SChris Mason btrfs_set_header_nritems(left, old_left_nritems + push_items); 1684*db94535dSChris Mason if (left->map_token) { 1685*db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 1686*db94535dSChris Mason left->map_token = NULL; 1687*db94535dSChris Mason } 1688be0e5c09SChris Mason 1689be0e5c09SChris Mason /* fixup right node */ 16905f39d397SChris Mason push_space = btrfs_item_offset_nr(right, push_items - 1) - 1691123abc88SChris Mason leaf_data_end(root, right); 16925f39d397SChris Mason memmove_extent_buffer(right, btrfs_leaf_data(right) + 1693d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 1694d6025579SChris Mason btrfs_leaf_data(right) + 1695123abc88SChris Mason leaf_data_end(root, right), push_space); 16965f39d397SChris Mason 16975f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(0), 16985f39d397SChris Mason btrfs_item_nr_offset(push_items), 16995f39d397SChris Mason (btrfs_header_nritems(right) - push_items) * 17000783fcfcSChris Mason sizeof(struct btrfs_item)); 17015f39d397SChris Mason 17025f39d397SChris Mason right_nritems = btrfs_header_nritems(right) - push_items; 17035f39d397SChris Mason btrfs_set_header_nritems(right, right_nritems); 1704123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 1705eb60ceacSChris Mason 17065f39d397SChris Mason for (i = 0; i < right_nritems; i++) { 17075f39d397SChris Mason item = btrfs_item_nr(right, i); 1708*db94535dSChris Mason 1709*db94535dSChris Mason if (!right->map_token) { 1710*db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1711*db94535dSChris Mason sizeof(struct btrfs_item), 1712*db94535dSChris Mason &right->map_token, &right->kaddr, 1713*db94535dSChris Mason &right->map_start, &right->map_len, 1714*db94535dSChris Mason KM_USER1); 1715*db94535dSChris Mason } 1716*db94535dSChris Mason 1717*db94535dSChris Mason push_space = push_space - btrfs_item_size(right, item); 1718*db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 1719*db94535dSChris Mason } 1720*db94535dSChris Mason if (right->map_token) { 1721*db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1722*db94535dSChris Mason right->map_token = NULL; 1723be0e5c09SChris Mason } 1724eb60ceacSChris Mason 17255f39d397SChris Mason btrfs_mark_buffer_dirty(left); 17265f39d397SChris Mason btrfs_mark_buffer_dirty(right); 1727098f59c2SChris Mason 17285f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 17295f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, 1); 1730aa5d6bedSChris Mason if (wret) 1731aa5d6bedSChris Mason ret = wret; 1732be0e5c09SChris Mason 1733be0e5c09SChris Mason /* then fixup the leaf pointer in the path */ 1734be0e5c09SChris Mason if (path->slots[0] < push_items) { 1735be0e5c09SChris Mason path->slots[0] += old_left_nritems; 17365f39d397SChris Mason free_extent_buffer(path->nodes[0]); 17375f39d397SChris Mason path->nodes[0] = left; 1738be0e5c09SChris Mason path->slots[1] -= 1; 1739be0e5c09SChris Mason } else { 17405f39d397SChris Mason free_extent_buffer(left); 1741be0e5c09SChris Mason path->slots[0] -= push_items; 1742be0e5c09SChris Mason } 1743eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 1744aa5d6bedSChris Mason return ret; 1745be0e5c09SChris Mason } 1746be0e5c09SChris Mason 174774123bd7SChris Mason /* 174874123bd7SChris Mason * split the path's leaf in two, making sure there is at least data_size 174974123bd7SChris Mason * available for the resulting leaf level of the path. 1750aa5d6bedSChris Mason * 1751aa5d6bedSChris Mason * returns 0 if all went well and < 0 on failure. 175274123bd7SChris Mason */ 1753e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 1754d4dbff95SChris Mason *root, struct btrfs_key *ins_key, 1755d4dbff95SChris Mason struct btrfs_path *path, int data_size) 1756be0e5c09SChris Mason { 17575f39d397SChris Mason struct extent_buffer *l; 17587518a238SChris Mason u32 nritems; 1759eb60ceacSChris Mason int mid; 1760eb60ceacSChris Mason int slot; 17615f39d397SChris Mason struct extent_buffer *right; 17620783fcfcSChris Mason int space_needed = data_size + sizeof(struct btrfs_item); 1763be0e5c09SChris Mason int data_copy_size; 1764be0e5c09SChris Mason int rt_data_off; 1765be0e5c09SChris Mason int i; 1766d4dbff95SChris Mason int ret = 0; 1767aa5d6bedSChris Mason int wret; 1768d4dbff95SChris Mason int double_split = 0; 1769d4dbff95SChris Mason struct btrfs_disk_key disk_key; 1770be0e5c09SChris Mason 177140689478SChris Mason /* first try to make some room by pushing left and right */ 1772e089f05cSChris Mason wret = push_leaf_left(trans, root, path, data_size); 1773eaee50e8SChris Mason if (wret < 0) 1774eaee50e8SChris Mason return wret; 1775eaee50e8SChris Mason if (wret) { 1776e089f05cSChris Mason wret = push_leaf_right(trans, root, path, data_size); 1777eaee50e8SChris Mason if (wret < 0) 1778eaee50e8SChris Mason return wret; 1779eaee50e8SChris Mason } 17805f39d397SChris Mason l = path->nodes[0]; 1781aa5d6bedSChris Mason 1782aa5d6bedSChris Mason /* did the pushes work? */ 1783123abc88SChris Mason if (btrfs_leaf_free_space(root, l) >= 1784123abc88SChris Mason sizeof(struct btrfs_item) + data_size) 1785be0e5c09SChris Mason return 0; 1786aa5d6bedSChris Mason 17875c680ed6SChris Mason if (!path->nodes[1]) { 1788e089f05cSChris Mason ret = insert_new_root(trans, root, path, 1); 17895c680ed6SChris Mason if (ret) 17905c680ed6SChris Mason return ret; 17915c680ed6SChris Mason } 1792eb60ceacSChris Mason slot = path->slots[0]; 17935f39d397SChris Mason nritems = btrfs_header_nritems(l); 1794eb60ceacSChris Mason mid = (nritems + 1)/ 2; 179554aa1f4dSChris Mason 1796*db94535dSChris Mason right = btrfs_alloc_free_block(trans, root, root->leafsize, 1797*db94535dSChris Mason l->start, 0); 17985f39d397SChris Mason if (IS_ERR(right)) 17995f39d397SChris Mason return PTR_ERR(right); 180054aa1f4dSChris Mason 18015f39d397SChris Mason memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header)); 1802*db94535dSChris Mason btrfs_set_header_bytenr(right, right->start); 18035f39d397SChris Mason btrfs_set_header_generation(right, trans->transid); 18045f39d397SChris Mason btrfs_set_header_owner(right, root->root_key.objectid); 18055f39d397SChris Mason btrfs_set_header_level(right, 0); 18065f39d397SChris Mason write_extent_buffer(right, root->fs_info->fsid, 18075f39d397SChris Mason (unsigned long)btrfs_header_fsid(right), 18085f39d397SChris Mason BTRFS_FSID_SIZE); 18095f39d397SChris Mason 1810d4dbff95SChris Mason if (mid <= slot) { 1811d4dbff95SChris Mason if (nritems == 1 || 1812d4dbff95SChris Mason leaf_space_used(l, mid, nritems - mid) + space_needed > 1813d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 1814d4dbff95SChris Mason if (slot >= nritems) { 1815d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 18165f39d397SChris Mason btrfs_set_header_nritems(right, 0); 1817d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 1818*db94535dSChris Mason &disk_key, right->start, 1819d4dbff95SChris Mason path->slots[1] + 1, 1); 1820d4dbff95SChris Mason if (wret) 1821d4dbff95SChris Mason ret = wret; 18225f39d397SChris Mason free_extent_buffer(path->nodes[0]); 18235f39d397SChris Mason path->nodes[0] = right; 1824d4dbff95SChris Mason path->slots[0] = 0; 1825d4dbff95SChris Mason path->slots[1] += 1; 1826d4dbff95SChris Mason return ret; 1827d4dbff95SChris Mason } 1828d4dbff95SChris Mason mid = slot; 1829d4dbff95SChris Mason double_split = 1; 1830d4dbff95SChris Mason } 1831d4dbff95SChris Mason } else { 1832d4dbff95SChris Mason if (leaf_space_used(l, 0, mid + 1) + space_needed > 1833d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 1834d4dbff95SChris Mason if (slot == 0) { 1835d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 18365f39d397SChris Mason btrfs_set_header_nritems(right, 0); 1837d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 1838d4dbff95SChris Mason &disk_key, 1839*db94535dSChris Mason right->start, 1840098f59c2SChris Mason path->slots[1], 1); 1841d4dbff95SChris Mason if (wret) 1842d4dbff95SChris Mason ret = wret; 18435f39d397SChris Mason free_extent_buffer(path->nodes[0]); 18445f39d397SChris Mason path->nodes[0] = right; 1845d4dbff95SChris Mason path->slots[0] = 0; 1846a429e513SChris Mason if (path->slots[1] == 0) { 1847a429e513SChris Mason wret = fixup_low_keys(trans, root, 1848a429e513SChris Mason path, &disk_key, 1); 1849a429e513SChris Mason if (wret) 1850a429e513SChris Mason ret = wret; 1851a429e513SChris Mason } 1852d4dbff95SChris Mason return ret; 1853d4dbff95SChris Mason } 1854d4dbff95SChris Mason mid = slot; 1855d4dbff95SChris Mason double_split = 1; 1856d4dbff95SChris Mason } 1857d4dbff95SChris Mason } 18585f39d397SChris Mason nritems = nritems - mid; 18595f39d397SChris Mason btrfs_set_header_nritems(right, nritems); 18605f39d397SChris Mason data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l); 18615f39d397SChris Mason 18625f39d397SChris Mason copy_extent_buffer(right, l, btrfs_item_nr_offset(0), 18635f39d397SChris Mason btrfs_item_nr_offset(mid), 18645f39d397SChris Mason nritems * sizeof(struct btrfs_item)); 18655f39d397SChris Mason 18665f39d397SChris Mason copy_extent_buffer(right, l, 1867d6025579SChris Mason btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - 1868123abc88SChris Mason data_copy_size, btrfs_leaf_data(l) + 1869123abc88SChris Mason leaf_data_end(root, l), data_copy_size); 187074123bd7SChris Mason 18715f39d397SChris Mason rt_data_off = BTRFS_LEAF_DATA_SIZE(root) - 18725f39d397SChris Mason btrfs_item_end_nr(l, mid); 18735f39d397SChris Mason 18745f39d397SChris Mason for (i = 0; i < nritems; i++) { 18755f39d397SChris Mason struct btrfs_item *item = btrfs_item_nr(right, i); 1876*db94535dSChris Mason u32 ioff; 1877*db94535dSChris Mason 1878*db94535dSChris Mason if (!right->map_token) { 1879*db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1880*db94535dSChris Mason sizeof(struct btrfs_item), 1881*db94535dSChris Mason &right->map_token, &right->kaddr, 1882*db94535dSChris Mason &right->map_start, &right->map_len, 1883*db94535dSChris Mason KM_USER1); 1884*db94535dSChris Mason } 1885*db94535dSChris Mason 1886*db94535dSChris Mason ioff = btrfs_item_offset(right, item); 18875f39d397SChris Mason btrfs_set_item_offset(right, item, ioff + rt_data_off); 18880783fcfcSChris Mason } 188974123bd7SChris Mason 1890*db94535dSChris Mason if (right->map_token) { 1891*db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1892*db94535dSChris Mason right->map_token = NULL; 1893*db94535dSChris Mason } 1894*db94535dSChris Mason 18955f39d397SChris Mason btrfs_set_header_nritems(l, mid); 1896aa5d6bedSChris Mason ret = 0; 18975f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 1898*db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, right->start, 1899*db94535dSChris Mason path->slots[1] + 1, 1); 1900aa5d6bedSChris Mason if (wret) 1901aa5d6bedSChris Mason ret = wret; 19025f39d397SChris Mason 19035f39d397SChris Mason btrfs_mark_buffer_dirty(right); 19045f39d397SChris Mason btrfs_mark_buffer_dirty(l); 1905eb60ceacSChris Mason BUG_ON(path->slots[0] != slot); 19065f39d397SChris Mason 1907be0e5c09SChris Mason if (mid <= slot) { 19085f39d397SChris Mason free_extent_buffer(path->nodes[0]); 19095f39d397SChris Mason path->nodes[0] = right; 1910be0e5c09SChris Mason path->slots[0] -= mid; 1911be0e5c09SChris Mason path->slots[1] += 1; 1912eb60ceacSChris Mason } else 19135f39d397SChris Mason free_extent_buffer(right); 19145f39d397SChris Mason 1915eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 1916d4dbff95SChris Mason 1917d4dbff95SChris Mason if (!double_split) 1918d4dbff95SChris Mason return ret; 191954aa1f4dSChris Mason 1920*db94535dSChris Mason right = btrfs_alloc_free_block(trans, root, root->leafsize, 1921*db94535dSChris Mason l->start, 0); 19225f39d397SChris Mason if (IS_ERR(right)) 19235f39d397SChris Mason return PTR_ERR(right); 19245f39d397SChris Mason 19255f39d397SChris Mason memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header)); 1926*db94535dSChris Mason btrfs_set_header_bytenr(right, right->start); 19275f39d397SChris Mason btrfs_set_header_generation(right, trans->transid); 19285f39d397SChris Mason btrfs_set_header_owner(right, root->root_key.objectid); 19295f39d397SChris Mason btrfs_set_header_level(right, 0); 19305f39d397SChris Mason write_extent_buffer(right, root->fs_info->fsid, 19315f39d397SChris Mason (unsigned long)btrfs_header_fsid(right), 19325f39d397SChris Mason BTRFS_FSID_SIZE); 19335f39d397SChris Mason 1934d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 19355f39d397SChris Mason btrfs_set_header_nritems(right, 0); 1936d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 1937*db94535dSChris Mason &disk_key, right->start, 1938d4dbff95SChris Mason path->slots[1], 1); 1939d4dbff95SChris Mason if (wret) 1940d4dbff95SChris Mason ret = wret; 1941a429e513SChris Mason if (path->slots[1] == 0) { 1942a429e513SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, 1); 1943a429e513SChris Mason if (wret) 1944a429e513SChris Mason ret = wret; 1945a429e513SChris Mason } 19465f39d397SChris Mason free_extent_buffer(path->nodes[0]); 19475f39d397SChris Mason path->nodes[0] = right; 1948d4dbff95SChris Mason path->slots[0] = 0; 1949be0e5c09SChris Mason return ret; 1950be0e5c09SChris Mason } 1951be0e5c09SChris Mason 1952b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans, 1953b18c6685SChris Mason struct btrfs_root *root, 1954b18c6685SChris Mason struct btrfs_path *path, 1955b18c6685SChris Mason u32 new_size) 1956b18c6685SChris Mason { 1957b18c6685SChris Mason int ret = 0; 1958b18c6685SChris Mason int slot; 1959b18c6685SChris Mason int slot_orig; 19605f39d397SChris Mason struct extent_buffer *leaf; 19615f39d397SChris Mason struct btrfs_item *item; 1962b18c6685SChris Mason u32 nritems; 1963b18c6685SChris Mason unsigned int data_end; 1964b18c6685SChris Mason unsigned int old_data_start; 1965b18c6685SChris Mason unsigned int old_size; 1966b18c6685SChris Mason unsigned int size_diff; 1967b18c6685SChris Mason int i; 1968b18c6685SChris Mason 1969b18c6685SChris Mason slot_orig = path->slots[0]; 19705f39d397SChris Mason leaf = path->nodes[0]; 1971b18c6685SChris Mason 19725f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 1973b18c6685SChris Mason data_end = leaf_data_end(root, leaf); 1974b18c6685SChris Mason 1975b18c6685SChris Mason slot = path->slots[0]; 19765f39d397SChris Mason old_data_start = btrfs_item_offset_nr(leaf, slot); 19775f39d397SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 1978b18c6685SChris Mason BUG_ON(old_size <= new_size); 1979b18c6685SChris Mason size_diff = old_size - new_size; 1980b18c6685SChris Mason 1981b18c6685SChris Mason BUG_ON(slot < 0); 1982b18c6685SChris Mason BUG_ON(slot >= nritems); 1983b18c6685SChris Mason 1984b18c6685SChris Mason /* 1985b18c6685SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 1986b18c6685SChris Mason */ 1987b18c6685SChris Mason /* first correct the data pointers */ 1988b18c6685SChris Mason for (i = slot; i < nritems; i++) { 19895f39d397SChris Mason u32 ioff; 19905f39d397SChris Mason item = btrfs_item_nr(leaf, i); 1991*db94535dSChris Mason 1992*db94535dSChris Mason if (!leaf->map_token) { 1993*db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 1994*db94535dSChris Mason sizeof(struct btrfs_item), 1995*db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 1996*db94535dSChris Mason &leaf->map_start, &leaf->map_len, 1997*db94535dSChris Mason KM_USER1); 1998*db94535dSChris Mason } 1999*db94535dSChris Mason 20005f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 20015f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + size_diff); 2002b18c6685SChris Mason } 2003*db94535dSChris Mason 2004*db94535dSChris Mason if (leaf->map_token) { 2005*db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2006*db94535dSChris Mason leaf->map_token = NULL; 2007*db94535dSChris Mason } 2008*db94535dSChris Mason 2009b18c6685SChris Mason /* shift the data */ 20105f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2011b18c6685SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 2012b18c6685SChris Mason data_end, old_data_start + new_size - data_end); 20135f39d397SChris Mason 20145f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 20155f39d397SChris Mason btrfs_set_item_size(leaf, item, new_size); 20165f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2017b18c6685SChris Mason 2018b18c6685SChris Mason ret = 0; 20195f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 20205f39d397SChris Mason btrfs_print_leaf(root, leaf); 2021b18c6685SChris Mason BUG(); 20225f39d397SChris Mason } 2023b18c6685SChris Mason return ret; 2024b18c6685SChris Mason } 2025b18c6685SChris Mason 20265f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans, 20275f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 20285f39d397SChris Mason u32 data_size) 20296567e837SChris Mason { 20306567e837SChris Mason int ret = 0; 20316567e837SChris Mason int slot; 20326567e837SChris Mason int slot_orig; 20335f39d397SChris Mason struct extent_buffer *leaf; 20345f39d397SChris Mason struct btrfs_item *item; 20356567e837SChris Mason u32 nritems; 20366567e837SChris Mason unsigned int data_end; 20376567e837SChris Mason unsigned int old_data; 20386567e837SChris Mason unsigned int old_size; 20396567e837SChris Mason int i; 20406567e837SChris Mason 20416567e837SChris Mason slot_orig = path->slots[0]; 20425f39d397SChris Mason leaf = path->nodes[0]; 20436567e837SChris Mason 20445f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 20456567e837SChris Mason data_end = leaf_data_end(root, leaf); 20466567e837SChris Mason 20475f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < data_size) { 20485f39d397SChris Mason btrfs_print_leaf(root, leaf); 20496567e837SChris Mason BUG(); 20505f39d397SChris Mason } 20516567e837SChris Mason slot = path->slots[0]; 20525f39d397SChris Mason old_data = btrfs_item_end_nr(leaf, slot); 20536567e837SChris Mason 20546567e837SChris Mason BUG_ON(slot < 0); 20556567e837SChris Mason BUG_ON(slot >= nritems); 20566567e837SChris Mason 20576567e837SChris Mason /* 20586567e837SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 20596567e837SChris Mason */ 20606567e837SChris Mason /* first correct the data pointers */ 20616567e837SChris Mason for (i = slot; i < nritems; i++) { 20625f39d397SChris Mason u32 ioff; 20635f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2064*db94535dSChris Mason 2065*db94535dSChris Mason if (!leaf->map_token) { 2066*db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2067*db94535dSChris Mason sizeof(struct btrfs_item), 2068*db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2069*db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2070*db94535dSChris Mason KM_USER1); 2071*db94535dSChris Mason } 20725f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 20735f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff - data_size); 20746567e837SChris Mason } 20755f39d397SChris Mason 2076*db94535dSChris Mason if (leaf->map_token) { 2077*db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2078*db94535dSChris Mason leaf->map_token = NULL; 2079*db94535dSChris Mason } 2080*db94535dSChris Mason 20816567e837SChris Mason /* shift the data */ 20825f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 20836567e837SChris Mason data_end - data_size, btrfs_leaf_data(leaf) + 20846567e837SChris Mason data_end, old_data - data_end); 20855f39d397SChris Mason 20866567e837SChris Mason data_end = old_data; 20875f39d397SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 20885f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 20895f39d397SChris Mason btrfs_set_item_size(leaf, item, old_size + data_size); 20905f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 20916567e837SChris Mason 20926567e837SChris Mason ret = 0; 20935f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 20945f39d397SChris Mason btrfs_print_leaf(root, leaf); 20956567e837SChris Mason BUG(); 20965f39d397SChris Mason } 20976567e837SChris Mason return ret; 20986567e837SChris Mason } 20996567e837SChris Mason 210074123bd7SChris Mason /* 210174123bd7SChris Mason * Given a key and some data, insert an item into the tree. 210274123bd7SChris Mason * This does all the path init required, making room in the tree if needed. 210374123bd7SChris Mason */ 21045f39d397SChris Mason int btrfs_insert_empty_item(struct btrfs_trans_handle *trans, 21055f39d397SChris Mason struct btrfs_root *root, 21065f39d397SChris Mason struct btrfs_path *path, 21075f39d397SChris Mason struct btrfs_key *cpu_key, u32 data_size) 2108be0e5c09SChris Mason { 21095f39d397SChris Mason struct extent_buffer *leaf; 21105f39d397SChris Mason struct btrfs_item *item; 2111aa5d6bedSChris Mason int ret = 0; 2112be0e5c09SChris Mason int slot; 2113eb60ceacSChris Mason int slot_orig; 21147518a238SChris Mason u32 nritems; 2115be0e5c09SChris Mason unsigned int data_end; 2116e2fa7227SChris Mason struct btrfs_disk_key disk_key; 2117e2fa7227SChris Mason 2118e2fa7227SChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key); 2119be0e5c09SChris Mason 212074123bd7SChris Mason /* create a root if there isn't one */ 21215c680ed6SChris Mason if (!root->node) 2122cfaa7295SChris Mason BUG(); 21235f39d397SChris Mason 2124e089f05cSChris Mason ret = btrfs_search_slot(trans, root, cpu_key, path, data_size, 1); 2125eb60ceacSChris Mason if (ret == 0) { 2126f0930a37SChris Mason return -EEXIST; 2127aa5d6bedSChris Mason } 2128ed2ff2cbSChris Mason if (ret < 0) 2129ed2ff2cbSChris Mason goto out; 2130be0e5c09SChris Mason 213162e2749eSChris Mason slot_orig = path->slots[0]; 21325f39d397SChris Mason leaf = path->nodes[0]; 213374123bd7SChris Mason 21345f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2135123abc88SChris Mason data_end = leaf_data_end(root, leaf); 2136eb60ceacSChris Mason 2137123abc88SChris Mason if (btrfs_leaf_free_space(root, leaf) < 2138d4dbff95SChris Mason sizeof(struct btrfs_item) + data_size) { 2139be0e5c09SChris Mason BUG(); 2140d4dbff95SChris Mason } 21415f39d397SChris Mason 214262e2749eSChris Mason slot = path->slots[0]; 2143eb60ceacSChris Mason BUG_ON(slot < 0); 21445f39d397SChris Mason 2145be0e5c09SChris Mason if (slot != nritems) { 2146be0e5c09SChris Mason int i; 21475f39d397SChris Mason unsigned int old_data = btrfs_item_end_nr(leaf, slot); 2148be0e5c09SChris Mason 21495f39d397SChris Mason if (old_data < data_end) { 21505f39d397SChris Mason btrfs_print_leaf(root, leaf); 21515f39d397SChris Mason printk("slot %d old_data %d data_end %d\n", 21525f39d397SChris Mason slot, old_data, data_end); 21535f39d397SChris Mason BUG_ON(1); 21545f39d397SChris Mason } 2155be0e5c09SChris Mason /* 2156be0e5c09SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2157be0e5c09SChris Mason */ 2158be0e5c09SChris Mason /* first correct the data pointers */ 2159*db94535dSChris Mason WARN_ON(leaf->map_token); 21600783fcfcSChris Mason for (i = slot; i < nritems; i++) { 21615f39d397SChris Mason u32 ioff; 2162*db94535dSChris Mason 21635f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2164*db94535dSChris Mason if (!leaf->map_token) { 2165*db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2166*db94535dSChris Mason sizeof(struct btrfs_item), 2167*db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2168*db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2169*db94535dSChris Mason KM_USER1); 2170*db94535dSChris Mason } 2171*db94535dSChris Mason 21725f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 21735f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff - data_size); 21740783fcfcSChris Mason } 2175*db94535dSChris Mason if (leaf->map_token) { 2176*db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2177*db94535dSChris Mason leaf->map_token = NULL; 2178*db94535dSChris Mason } 2179be0e5c09SChris Mason 2180be0e5c09SChris Mason /* shift the items */ 21815f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1), 21825f39d397SChris Mason btrfs_item_nr_offset(slot), 21830783fcfcSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 2184be0e5c09SChris Mason 2185be0e5c09SChris Mason /* shift the data */ 21865f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2187d6025579SChris Mason data_end - data_size, btrfs_leaf_data(leaf) + 2188be0e5c09SChris Mason data_end, old_data - data_end); 2189be0e5c09SChris Mason data_end = old_data; 2190be0e5c09SChris Mason } 21915f39d397SChris Mason 219262e2749eSChris Mason /* setup the item for the new data */ 21935f39d397SChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 21945f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 21955f39d397SChris Mason btrfs_set_item_offset(leaf, item, data_end - data_size); 21965f39d397SChris Mason btrfs_set_item_size(leaf, item, data_size); 21975f39d397SChris Mason btrfs_set_header_nritems(leaf, nritems + 1); 21985f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2199aa5d6bedSChris Mason 2200aa5d6bedSChris Mason ret = 0; 22018e19f2cdSChris Mason if (slot == 0) 2202e089f05cSChris Mason ret = fixup_low_keys(trans, root, path, &disk_key, 1); 2203aa5d6bedSChris Mason 22045f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 22055f39d397SChris Mason btrfs_print_leaf(root, leaf); 2206be0e5c09SChris Mason BUG(); 22075f39d397SChris Mason } 2208ed2ff2cbSChris Mason out: 220962e2749eSChris Mason return ret; 221062e2749eSChris Mason } 221162e2749eSChris Mason 221262e2749eSChris Mason /* 221362e2749eSChris Mason * Given a key and some data, insert an item into the tree. 221462e2749eSChris Mason * This does all the path init required, making room in the tree if needed. 221562e2749eSChris Mason */ 2216e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root 2217e089f05cSChris Mason *root, struct btrfs_key *cpu_key, void *data, u32 2218e089f05cSChris Mason data_size) 221962e2749eSChris Mason { 222062e2749eSChris Mason int ret = 0; 22212c90e5d6SChris Mason struct btrfs_path *path; 22225f39d397SChris Mason struct extent_buffer *leaf; 22235f39d397SChris Mason unsigned long ptr; 222462e2749eSChris Mason 22252c90e5d6SChris Mason path = btrfs_alloc_path(); 22262c90e5d6SChris Mason BUG_ON(!path); 22272c90e5d6SChris Mason ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size); 222862e2749eSChris Mason if (!ret) { 22295f39d397SChris Mason leaf = path->nodes[0]; 22305f39d397SChris Mason ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); 22315f39d397SChris Mason write_extent_buffer(leaf, data, ptr, data_size); 22325f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 223362e2749eSChris Mason } 22342c90e5d6SChris Mason btrfs_free_path(path); 2235aa5d6bedSChris Mason return ret; 2236be0e5c09SChris Mason } 2237be0e5c09SChris Mason 223874123bd7SChris Mason /* 22395de08d7dSChris Mason * delete the pointer from a given node. 224074123bd7SChris Mason * 224174123bd7SChris Mason * If the delete empties a node, the node is removed from the tree, 224274123bd7SChris Mason * continuing all the way the root if required. The root is converted into 224374123bd7SChris Mason * a leaf if all the nodes are emptied. 224474123bd7SChris Mason */ 2245e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 2246e089f05cSChris Mason struct btrfs_path *path, int level, int slot) 2247be0e5c09SChris Mason { 22485f39d397SChris Mason struct extent_buffer *parent = path->nodes[level]; 22497518a238SChris Mason u32 nritems; 2250aa5d6bedSChris Mason int ret = 0; 2251bb803951SChris Mason int wret; 2252be0e5c09SChris Mason 22535f39d397SChris Mason nritems = btrfs_header_nritems(parent); 2254be0e5c09SChris Mason if (slot != nritems -1) { 22555f39d397SChris Mason memmove_extent_buffer(parent, 22565f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 22575f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 2258d6025579SChris Mason sizeof(struct btrfs_key_ptr) * 2259d6025579SChris Mason (nritems - slot - 1)); 2260be0e5c09SChris Mason } 22617518a238SChris Mason nritems--; 22625f39d397SChris Mason btrfs_set_header_nritems(parent, nritems); 22637518a238SChris Mason if (nritems == 0 && parent == root->node) { 22645f39d397SChris Mason BUG_ON(btrfs_header_level(root->node) != 1); 2265eb60ceacSChris Mason /* just turn the root into a leaf and break */ 22665f39d397SChris Mason btrfs_set_header_level(root->node, 0); 2267bb803951SChris Mason } else if (slot == 0) { 22685f39d397SChris Mason struct btrfs_disk_key disk_key; 22695f39d397SChris Mason 22705f39d397SChris Mason btrfs_node_key(parent, &disk_key, 0); 22715f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, level + 1); 22720f70abe2SChris Mason if (wret) 22730f70abe2SChris Mason ret = wret; 2274be0e5c09SChris Mason } 2275d6025579SChris Mason btrfs_mark_buffer_dirty(parent); 2276aa5d6bedSChris Mason return ret; 2277be0e5c09SChris Mason } 2278be0e5c09SChris Mason 227974123bd7SChris Mason /* 228074123bd7SChris Mason * delete the item at the leaf level in path. If that empties 228174123bd7SChris Mason * the leaf, remove it from the tree 228274123bd7SChris Mason */ 2283e089f05cSChris Mason int btrfs_del_item(struct btrfs_trans_handle *trans, struct btrfs_root *root, 2284e089f05cSChris Mason struct btrfs_path *path) 2285be0e5c09SChris Mason { 2286be0e5c09SChris Mason int slot; 22875f39d397SChris Mason struct extent_buffer *leaf; 22885f39d397SChris Mason struct btrfs_item *item; 2289be0e5c09SChris Mason int doff; 2290be0e5c09SChris Mason int dsize; 2291aa5d6bedSChris Mason int ret = 0; 2292aa5d6bedSChris Mason int wret; 22937518a238SChris Mason u32 nritems; 2294be0e5c09SChris Mason 22955f39d397SChris Mason leaf = path->nodes[0]; 22964920c9acSChris Mason slot = path->slots[0]; 22975f39d397SChris Mason doff = btrfs_item_offset_nr(leaf, slot); 22985f39d397SChris Mason dsize = btrfs_item_size_nr(leaf, slot); 22995f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2300be0e5c09SChris Mason 23017518a238SChris Mason if (slot != nritems - 1) { 2302be0e5c09SChris Mason int i; 2303123abc88SChris Mason int data_end = leaf_data_end(root, leaf); 23045f39d397SChris Mason 23055f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2306d6025579SChris Mason data_end + dsize, 2307123abc88SChris Mason btrfs_leaf_data(leaf) + data_end, 2308be0e5c09SChris Mason doff - data_end); 23095f39d397SChris Mason 23100783fcfcSChris Mason for (i = slot + 1; i < nritems; i++) { 23115f39d397SChris Mason u32 ioff; 2312*db94535dSChris Mason 23135f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2314*db94535dSChris Mason if (!leaf->map_token) { 2315*db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2316*db94535dSChris Mason sizeof(struct btrfs_item), 2317*db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2318*db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2319*db94535dSChris Mason KM_USER1); 2320*db94535dSChris Mason } 23215f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 23225f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + dsize); 23230783fcfcSChris Mason } 2324*db94535dSChris Mason 2325*db94535dSChris Mason if (leaf->map_token) { 2326*db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2327*db94535dSChris Mason leaf->map_token = NULL; 2328*db94535dSChris Mason } 2329*db94535dSChris Mason 23305f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot), 23315f39d397SChris Mason btrfs_item_nr_offset(slot + 1), 23320783fcfcSChris Mason sizeof(struct btrfs_item) * 23337518a238SChris Mason (nritems - slot - 1)); 2334be0e5c09SChris Mason } 23355f39d397SChris Mason btrfs_set_header_nritems(leaf, nritems - 1); 23367518a238SChris Mason nritems--; 23375f39d397SChris Mason 233874123bd7SChris Mason /* delete the leaf if we've emptied it */ 23397518a238SChris Mason if (nritems == 0) { 23405f39d397SChris Mason if (leaf == root->node) { 23415f39d397SChris Mason btrfs_set_header_level(leaf, 0); 23429a8dd150SChris Mason } else { 23435f39d397SChris Mason clean_tree_block(trans, root, leaf); 23445f39d397SChris Mason wait_on_tree_block_writeback(root, leaf); 2345e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, path->slots[1]); 2346aa5d6bedSChris Mason if (wret) 2347aa5d6bedSChris Mason ret = wret; 2348e089f05cSChris Mason wret = btrfs_free_extent(trans, root, 2349*db94535dSChris Mason leaf->start, leaf->len, 1); 23500f70abe2SChris Mason if (wret) 23510f70abe2SChris Mason ret = wret; 23529a8dd150SChris Mason } 2353be0e5c09SChris Mason } else { 23547518a238SChris Mason int used = leaf_space_used(leaf, 0, nritems); 2355aa5d6bedSChris Mason if (slot == 0) { 23565f39d397SChris Mason struct btrfs_disk_key disk_key; 23575f39d397SChris Mason 23585f39d397SChris Mason btrfs_item_key(leaf, &disk_key, 0); 2359e089f05cSChris Mason wret = fixup_low_keys(trans, root, path, 23605f39d397SChris Mason &disk_key, 1); 2361aa5d6bedSChris Mason if (wret) 2362aa5d6bedSChris Mason ret = wret; 2363aa5d6bedSChris Mason } 2364aa5d6bedSChris Mason 236574123bd7SChris Mason /* delete the leaf if it is mostly empty */ 2366123abc88SChris Mason if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) { 2367be0e5c09SChris Mason /* push_leaf_left fixes the path. 2368be0e5c09SChris Mason * make sure the path still points to our leaf 2369be0e5c09SChris Mason * for possible call to del_ptr below 2370be0e5c09SChris Mason */ 23714920c9acSChris Mason slot = path->slots[1]; 23725f39d397SChris Mason extent_buffer_get(leaf); 23735f39d397SChris Mason 2374e089f05cSChris Mason wret = push_leaf_left(trans, root, path, 1); 237554aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2376aa5d6bedSChris Mason ret = wret; 23775f39d397SChris Mason 23785f39d397SChris Mason if (path->nodes[0] == leaf && 23795f39d397SChris Mason btrfs_header_nritems(leaf)) { 2380e089f05cSChris Mason wret = push_leaf_right(trans, root, path, 1); 238154aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2382aa5d6bedSChris Mason ret = wret; 2383aa5d6bedSChris Mason } 23845f39d397SChris Mason 23855f39d397SChris Mason if (btrfs_header_nritems(leaf) == 0) { 2386*db94535dSChris Mason u64 bytenr = leaf->start; 2387*db94535dSChris Mason u32 blocksize = leaf->len; 23885f39d397SChris Mason 23895f39d397SChris Mason clean_tree_block(trans, root, leaf); 23905f39d397SChris Mason wait_on_tree_block_writeback(root, leaf); 23915f39d397SChris Mason 2392e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, slot); 2393aa5d6bedSChris Mason if (wret) 2394aa5d6bedSChris Mason ret = wret; 23955f39d397SChris Mason 23965f39d397SChris Mason free_extent_buffer(leaf); 2397*db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 2398*db94535dSChris Mason blocksize, 1); 23990f70abe2SChris Mason if (wret) 24000f70abe2SChris Mason ret = wret; 24015de08d7dSChris Mason } else { 24025f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 24035f39d397SChris Mason free_extent_buffer(leaf); 2404be0e5c09SChris Mason } 2405d5719762SChris Mason } else { 24065f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2407be0e5c09SChris Mason } 24089a8dd150SChris Mason } 2409aa5d6bedSChris Mason return ret; 24109a8dd150SChris Mason } 24119a8dd150SChris Mason 241297571fd0SChris Mason /* 241397571fd0SChris Mason * walk up the tree as far as required to find the next leaf. 24140f70abe2SChris Mason * returns 0 if it found something or 1 if there are no greater leaves. 24150f70abe2SChris Mason * returns < 0 on io errors. 241697571fd0SChris Mason */ 2417234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path) 2418d97e63b6SChris Mason { 2419d97e63b6SChris Mason int slot; 2420d97e63b6SChris Mason int level = 1; 2421*db94535dSChris Mason u64 bytenr; 24225f39d397SChris Mason struct extent_buffer *c; 24235f39d397SChris Mason struct extent_buffer *next = NULL; 2424d97e63b6SChris Mason 2425234b63a0SChris Mason while(level < BTRFS_MAX_LEVEL) { 2426d97e63b6SChris Mason if (!path->nodes[level]) 24270f70abe2SChris Mason return 1; 24285f39d397SChris Mason 2429d97e63b6SChris Mason slot = path->slots[level] + 1; 2430d97e63b6SChris Mason c = path->nodes[level]; 24315f39d397SChris Mason if (slot >= btrfs_header_nritems(c)) { 2432d97e63b6SChris Mason level++; 2433d97e63b6SChris Mason continue; 2434d97e63b6SChris Mason } 24355f39d397SChris Mason 2436*db94535dSChris Mason bytenr = btrfs_node_blockptr(c, slot); 2437cfaa7295SChris Mason if (next) 24385f39d397SChris Mason free_extent_buffer(next); 24395f39d397SChris Mason 24406702ed49SChris Mason if (path->reada) 24416702ed49SChris Mason reada_for_search(root, path, level, slot); 24425f39d397SChris Mason 2443*db94535dSChris Mason next = read_tree_block(root, bytenr, 2444*db94535dSChris Mason btrfs_level_size(root, level -1)); 2445d97e63b6SChris Mason break; 2446d97e63b6SChris Mason } 2447d97e63b6SChris Mason path->slots[level] = slot; 2448d97e63b6SChris Mason while(1) { 2449d97e63b6SChris Mason level--; 2450d97e63b6SChris Mason c = path->nodes[level]; 24515f39d397SChris Mason free_extent_buffer(c); 2452d97e63b6SChris Mason path->nodes[level] = next; 2453d97e63b6SChris Mason path->slots[level] = 0; 2454d97e63b6SChris Mason if (!level) 2455d97e63b6SChris Mason break; 24566702ed49SChris Mason if (path->reada) 245732020611SYan reada_for_search(root, path, level, 0); 2458*db94535dSChris Mason next = read_tree_block(root, btrfs_node_blockptr(next, 0), 2459*db94535dSChris Mason btrfs_level_size(root, level - 1)); 2460d97e63b6SChris Mason } 2461d97e63b6SChris Mason return 0; 2462d97e63b6SChris Mason } 2463