16cbd5570SChris Mason /* 26cbd5570SChris Mason * Copyright (C) 2007 Oracle. All rights reserved. 36cbd5570SChris Mason * 46cbd5570SChris Mason * This program is free software; you can redistribute it and/or 56cbd5570SChris Mason * modify it under the terms of the GNU General Public 66cbd5570SChris Mason * License v2 as published by the Free Software Foundation. 76cbd5570SChris Mason * 86cbd5570SChris Mason * This program is distributed in the hope that it will be useful, 96cbd5570SChris Mason * but WITHOUT ANY WARRANTY; without even the implied warranty of 106cbd5570SChris Mason * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 116cbd5570SChris Mason * General Public License for more details. 126cbd5570SChris Mason * 136cbd5570SChris Mason * You should have received a copy of the GNU General Public 146cbd5570SChris Mason * License along with this program; if not, write to the 156cbd5570SChris Mason * Free Software Foundation, Inc., 59 Temple Place - Suite 330, 166cbd5570SChris Mason * Boston, MA 021110-1307, USA. 176cbd5570SChris Mason */ 186cbd5570SChris Mason 19a6b6e75eSChris Mason #include <linux/sched.h> 20eb60ceacSChris Mason #include "ctree.h" 21eb60ceacSChris Mason #include "disk-io.h" 227f5c1516SChris Mason #include "transaction.h" 235f39d397SChris Mason #include "print-tree.h" 249a8dd150SChris Mason 25e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 26e089f05cSChris Mason *root, struct btrfs_path *path, int level); 27e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 28d4dbff95SChris Mason *root, struct btrfs_key *ins_key, 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 86db94535dSChris Mason cow = btrfs_alloc_free_block(trans, root, buf->len, 87db94535dSChris Mason search_start, empty_size); 886702ed49SChris Mason if (IS_ERR(cow)) 896702ed49SChris Mason return PTR_ERR(cow); 906702ed49SChris Mason 915f39d397SChris Mason copy_extent_buffer(cow, buf, 0, 0, cow->len); 92db94535dSChris Mason btrfs_set_header_bytenr(cow, cow->start); 935f39d397SChris Mason btrfs_set_header_generation(cow, trans->transid); 945f39d397SChris Mason btrfs_set_header_owner(cow, root->root_key.objectid); 956702ed49SChris Mason 965f39d397SChris Mason WARN_ON(btrfs_header_generation(buf) > trans->transid); 975f39d397SChris Mason if (btrfs_header_generation(buf) != trans->transid) { 986702ed49SChris Mason different_trans = 1; 996702ed49SChris Mason ret = btrfs_inc_ref(trans, root, buf); 1006702ed49SChris Mason if (ret) 1016702ed49SChris Mason return ret; 1026702ed49SChris Mason } else { 1036702ed49SChris Mason clean_tree_block(trans, root, buf); 1046702ed49SChris Mason } 1056702ed49SChris Mason 1066702ed49SChris Mason if (buf == root->node) { 1076702ed49SChris Mason root->node = cow; 1085f39d397SChris Mason extent_buffer_get(cow); 1096702ed49SChris Mason if (buf != root->commit_root) { 110db94535dSChris Mason btrfs_free_extent(trans, root, buf->start, 111db94535dSChris Mason buf->len, 1); 1126702ed49SChris Mason } 1135f39d397SChris Mason free_extent_buffer(buf); 1146702ed49SChris Mason } else { 1155f39d397SChris Mason btrfs_set_node_blockptr(parent, parent_slot, 116db94535dSChris Mason cow->start); 1176702ed49SChris Mason btrfs_mark_buffer_dirty(parent); 1185f39d397SChris Mason WARN_ON(btrfs_header_generation(parent) != trans->transid); 119db94535dSChris Mason btrfs_free_extent(trans, root, buf->start, buf->len, 1); 1206702ed49SChris Mason } 1215f39d397SChris Mason free_extent_buffer(buf); 1226702ed49SChris Mason btrfs_mark_buffer_dirty(cow); 1236702ed49SChris Mason *cow_ret = cow; 1246702ed49SChris Mason return 0; 1256702ed49SChris Mason } 1266702ed49SChris Mason 1275f39d397SChris Mason int btrfs_cow_block(struct btrfs_trans_handle *trans, 1285f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *buf, 1295f39d397SChris Mason struct extent_buffer *parent, int parent_slot, 1305f39d397SChris Mason struct extent_buffer **cow_ret) 13102217ed2SChris Mason { 1326702ed49SChris Mason u64 search_start; 133f510cfecSChris Mason int ret; 134ccd467d6SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 135ccd467d6SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 136ccd467d6SChris Mason root->fs_info->running_transaction->transid); 137ccd467d6SChris Mason WARN_ON(1); 138ccd467d6SChris Mason } 139ccd467d6SChris Mason if (trans->transid != root->fs_info->generation) { 140ccd467d6SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 141ccd467d6SChris Mason root->fs_info->generation); 142ccd467d6SChris Mason WARN_ON(1); 143ccd467d6SChris Mason } 1445f39d397SChris Mason if (btrfs_header_generation(buf) == trans->transid) { 14502217ed2SChris Mason *cow_ret = buf; 14602217ed2SChris Mason return 0; 14702217ed2SChris Mason } 1486702ed49SChris Mason 149db94535dSChris Mason search_start = buf->start & ~((u64)BTRFS_BLOCK_GROUP_SIZE - 1); 150f510cfecSChris Mason ret = __btrfs_cow_block(trans, root, buf, parent, 1516702ed49SChris Mason parent_slot, cow_ret, search_start, 0); 152f510cfecSChris Mason return ret; 1532c90e5d6SChris Mason } 1546702ed49SChris Mason 1556b80053dSChris Mason static int close_blocks(u64 blocknr, u64 other, u32 blocksize) 1566702ed49SChris Mason { 1576b80053dSChris Mason if (blocknr < other && other - (blocknr + blocksize) < 32768) 1586702ed49SChris Mason return 1; 1596b80053dSChris Mason if (blocknr > other && blocknr - (other + blocksize) < 32768) 1606702ed49SChris Mason return 1; 16102217ed2SChris Mason return 0; 16202217ed2SChris Mason } 16302217ed2SChris Mason 1646b80053dSChris Mason static int should_defrag_leaf(struct extent_buffer *leaf) 1652cc58cf2SChris Mason { 1666b80053dSChris Mason struct btrfs_key key; 1672cc58cf2SChris Mason u32 nritems; 1682cc58cf2SChris Mason 1696b80053dSChris Mason if (btrfs_buffer_defrag(leaf)) 1702cc58cf2SChris Mason return 1; 1712cc58cf2SChris Mason 1726b80053dSChris Mason nritems = btrfs_header_nritems(leaf); 1732cc58cf2SChris Mason if (nritems == 0) 1742cc58cf2SChris Mason return 0; 1752cc58cf2SChris Mason 1766b80053dSChris Mason btrfs_item_key_to_cpu(leaf, &key, 0); 1776b80053dSChris Mason if (key.type == BTRFS_DIR_ITEM_KEY) 1782cc58cf2SChris Mason return 1; 1792cc58cf2SChris Mason 1806b80053dSChris Mason 1816b80053dSChris Mason btrfs_item_key_to_cpu(leaf, &key, nritems - 1); 1826b80053dSChris Mason if (key.type == BTRFS_DIR_ITEM_KEY) 1832cc58cf2SChris Mason return 1; 1842cc58cf2SChris Mason if (nritems > 4) { 1856b80053dSChris Mason btrfs_item_key_to_cpu(leaf, &key, nritems / 2); 1866b80053dSChris Mason if (key.type == BTRFS_DIR_ITEM_KEY) 1872cc58cf2SChris Mason return 1; 1882cc58cf2SChris Mason } 1892cc58cf2SChris Mason return 0; 1902cc58cf2SChris Mason } 1912cc58cf2SChris Mason 1926702ed49SChris Mason int btrfs_realloc_node(struct btrfs_trans_handle *trans, 1935f39d397SChris Mason struct btrfs_root *root, struct extent_buffer *parent, 194a6b6e75eSChris Mason int start_slot, int cache_only, u64 *last_ret, 195a6b6e75eSChris Mason struct btrfs_key *progress) 1966702ed49SChris Mason { 1976b80053dSChris Mason struct extent_buffer *cur; 1986b80053dSChris Mason struct extent_buffer *tmp; 1996702ed49SChris Mason u64 blocknr; 200e9d0b13bSChris Mason u64 search_start = *last_ret; 201e9d0b13bSChris Mason u64 last_block = 0; 2026702ed49SChris Mason u64 other; 2036702ed49SChris Mason u32 parent_nritems; 2046702ed49SChris Mason int end_slot; 2056702ed49SChris Mason int i; 2066702ed49SChris Mason int err = 0; 207f2183bdeSChris Mason int parent_level; 2086b80053dSChris Mason int uptodate; 2096b80053dSChris Mason u32 blocksize; 2106702ed49SChris Mason 2116702ed49SChris Mason if (trans->transaction != root->fs_info->running_transaction) { 2126702ed49SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 2136702ed49SChris Mason root->fs_info->running_transaction->transid); 2146702ed49SChris Mason WARN_ON(1); 2156702ed49SChris Mason } 2166702ed49SChris Mason if (trans->transid != root->fs_info->generation) { 2176702ed49SChris Mason printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid, 2186702ed49SChris Mason root->fs_info->generation); 2196702ed49SChris Mason WARN_ON(1); 2206702ed49SChris Mason } 2210f1ebbd1SChris Mason parent_level = btrfs_header_level(parent); 22286479a04SChris Mason 2236b80053dSChris Mason parent_nritems = btrfs_header_nritems(parent); 2246b80053dSChris Mason blocksize = btrfs_level_size(root, parent_level - 1); 2256702ed49SChris Mason end_slot = parent_nritems; 2266702ed49SChris Mason 2276702ed49SChris Mason if (parent_nritems == 1) 2286702ed49SChris Mason return 0; 2296702ed49SChris Mason 230a6b6e75eSChris Mason if (root != root->fs_info->extent_root) { 231a6b6e75eSChris Mason struct btrfs_key first_key; 232a6b6e75eSChris Mason struct btrfs_key last_key; 233a6b6e75eSChris Mason 234a6b6e75eSChris Mason btrfs_node_key_to_cpu(parent, &first_key, 0); 235a6b6e75eSChris Mason btrfs_node_key_to_cpu(parent, &last_key, parent_nritems - 1); 236a6b6e75eSChris Mason if (first_key.objectid != last_key.objectid) 237a6b6e75eSChris Mason return 0; 238a6b6e75eSChris Mason } 239a6b6e75eSChris Mason 2406702ed49SChris Mason for (i = start_slot; i < end_slot; i++) { 2416702ed49SChris Mason int close = 1; 242a6b6e75eSChris Mason 2436b80053dSChris Mason blocknr = btrfs_node_blockptr(parent, i); 244e9d0b13bSChris Mason if (last_block == 0) 245e9d0b13bSChris Mason last_block = blocknr; 2466702ed49SChris Mason if (i > 0) { 2476b80053dSChris Mason other = btrfs_node_blockptr(parent, i - 1); 2486b80053dSChris Mason close = close_blocks(blocknr, other, blocksize); 2496702ed49SChris Mason } 2506702ed49SChris Mason if (close && i < end_slot - 1) { 2516b80053dSChris Mason other = btrfs_node_blockptr(parent, i + 1); 2526b80053dSChris Mason close = close_blocks(blocknr, other, blocksize); 2536702ed49SChris Mason } 254e9d0b13bSChris Mason if (close) { 255e9d0b13bSChris Mason last_block = blocknr; 2566702ed49SChris Mason continue; 257e9d0b13bSChris Mason } 2586702ed49SChris Mason 2596b80053dSChris Mason cur = btrfs_find_tree_block(root, blocknr, blocksize); 2606b80053dSChris Mason if (cur) 2616b80053dSChris Mason uptodate = btrfs_buffer_uptodate(cur); 2626b80053dSChris Mason else 2636b80053dSChris Mason uptodate = 0; 2646b80053dSChris Mason if (!cur || !uptodate || 2656b80053dSChris Mason (parent_level != 1 && !btrfs_buffer_defrag(cur)) || 2666b80053dSChris Mason (parent_level == 1 && !should_defrag_leaf(cur))) { 2676702ed49SChris Mason if (cache_only) { 2686b80053dSChris Mason free_extent_buffer(cur); 2696702ed49SChris Mason continue; 2706702ed49SChris Mason } 2716b80053dSChris Mason if (!cur) { 2726b80053dSChris Mason cur = read_tree_block(root, blocknr, 2736b80053dSChris Mason blocksize); 2746b80053dSChris Mason } else if (!uptodate) { 2756b80053dSChris Mason btrfs_read_buffer(cur); 2766702ed49SChris Mason } 277f2183bdeSChris Mason } 278e9d0b13bSChris Mason if (search_start == 0) 2796b80053dSChris Mason search_start = last_block; 280e9d0b13bSChris Mason 2816b80053dSChris Mason err = __btrfs_cow_block(trans, root, cur, parent, i, 2826b80053dSChris Mason &tmp, search_start, 2836b80053dSChris Mason min(16 * blocksize, 2846b80053dSChris Mason (end_slot - i) * blocksize)); 285252c38f0SYan if (err) { 2866b80053dSChris Mason free_extent_buffer(cur); 2876702ed49SChris Mason break; 288252c38f0SYan } 2896b80053dSChris Mason search_start = tmp->start; 290f2183bdeSChris Mason *last_ret = search_start; 291f2183bdeSChris Mason if (parent_level == 1) 2926b80053dSChris Mason btrfs_clear_buffer_defrag(tmp); 2936b80053dSChris Mason free_extent_buffer(tmp); 2946702ed49SChris Mason } 2956702ed49SChris Mason return err; 2966702ed49SChris Mason } 2976702ed49SChris Mason 29874123bd7SChris Mason /* 29974123bd7SChris Mason * The leaf data grows from end-to-front in the node. 30074123bd7SChris Mason * this returns the address of the start of the last item, 30174123bd7SChris Mason * which is the stop of the leaf data stack 30274123bd7SChris Mason */ 303123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root, 3045f39d397SChris Mason struct extent_buffer *leaf) 305be0e5c09SChris Mason { 3065f39d397SChris Mason u32 nr = btrfs_header_nritems(leaf); 307be0e5c09SChris Mason if (nr == 0) 308123abc88SChris Mason return BTRFS_LEAF_DATA_SIZE(root); 3095f39d397SChris Mason return btrfs_item_offset_nr(leaf, nr - 1); 310be0e5c09SChris Mason } 311be0e5c09SChris Mason 31274123bd7SChris Mason /* 31374123bd7SChris Mason * compare two keys in a memcmp fashion 31474123bd7SChris Mason */ 3159aca1d51SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2) 316be0e5c09SChris Mason { 317e2fa7227SChris Mason struct btrfs_key k1; 318e2fa7227SChris Mason 319e2fa7227SChris Mason btrfs_disk_key_to_cpu(&k1, disk); 320e2fa7227SChris Mason 321e2fa7227SChris Mason if (k1.objectid > k2->objectid) 322be0e5c09SChris Mason return 1; 323e2fa7227SChris Mason if (k1.objectid < k2->objectid) 324be0e5c09SChris Mason return -1; 3255f39d397SChris Mason if (k1.type > k2->type) 326f254e52cSChris Mason return 1; 3275f39d397SChris Mason if (k1.type < k2->type) 328f254e52cSChris Mason return -1; 32970b2befdSChris Mason if (k1.offset > k2->offset) 33070b2befdSChris Mason return 1; 33170b2befdSChris Mason if (k1.offset < k2->offset) 33270b2befdSChris Mason return -1; 333be0e5c09SChris Mason return 0; 334be0e5c09SChris Mason } 33574123bd7SChris Mason 336123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path, 337123abc88SChris Mason int level) 338aa5d6bedSChris Mason { 3395f39d397SChris Mason struct extent_buffer *parent = NULL; 3405f39d397SChris Mason struct extent_buffer *node = path->nodes[level]; 3415f39d397SChris Mason struct btrfs_disk_key parent_key; 3425f39d397SChris Mason struct btrfs_disk_key node_key; 343aa5d6bedSChris Mason int parent_slot; 3448d7be552SChris Mason int slot; 3458d7be552SChris Mason struct btrfs_key cpukey; 3465f39d397SChris Mason u32 nritems = btrfs_header_nritems(node); 347aa5d6bedSChris Mason 348aa5d6bedSChris Mason if (path->nodes[level + 1]) 3495f39d397SChris Mason parent = path->nodes[level + 1]; 350a1f39630SAneesh 3518d7be552SChris Mason slot = path->slots[level]; 3527518a238SChris Mason BUG_ON(nritems == 0); 3537518a238SChris Mason if (parent) { 354a1f39630SAneesh parent_slot = path->slots[level + 1]; 3555f39d397SChris Mason btrfs_node_key(parent, &parent_key, parent_slot); 3565f39d397SChris Mason btrfs_node_key(node, &node_key, 0); 3575f39d397SChris Mason BUG_ON(memcmp(&parent_key, &node_key, 358e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 3591d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 360db94535dSChris Mason btrfs_header_bytenr(node)); 361aa5d6bedSChris Mason } 362123abc88SChris Mason BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root)); 3638d7be552SChris Mason if (slot != 0) { 3645f39d397SChris Mason btrfs_node_key_to_cpu(node, &cpukey, slot - 1); 3655f39d397SChris Mason btrfs_node_key(node, &node_key, slot); 3665f39d397SChris Mason BUG_ON(comp_keys(&node_key, &cpukey) <= 0); 3678d7be552SChris Mason } 3688d7be552SChris Mason if (slot < nritems - 1) { 3695f39d397SChris Mason btrfs_node_key_to_cpu(node, &cpukey, slot + 1); 3705f39d397SChris Mason btrfs_node_key(node, &node_key, slot); 3715f39d397SChris Mason BUG_ON(comp_keys(&node_key, &cpukey) >= 0); 372aa5d6bedSChris Mason } 373aa5d6bedSChris Mason return 0; 374aa5d6bedSChris Mason } 375aa5d6bedSChris Mason 376123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path, 377123abc88SChris Mason int level) 378aa5d6bedSChris Mason { 3795f39d397SChris Mason struct extent_buffer *leaf = path->nodes[level]; 3805f39d397SChris Mason struct extent_buffer *parent = NULL; 381aa5d6bedSChris Mason int parent_slot; 3828d7be552SChris Mason struct btrfs_key cpukey; 3835f39d397SChris Mason struct btrfs_disk_key parent_key; 3845f39d397SChris Mason struct btrfs_disk_key leaf_key; 3855f39d397SChris Mason int slot = path->slots[0]; 3868d7be552SChris Mason 3875f39d397SChris Mason u32 nritems = btrfs_header_nritems(leaf); 388aa5d6bedSChris Mason 389aa5d6bedSChris Mason if (path->nodes[level + 1]) 3905f39d397SChris Mason parent = path->nodes[level + 1]; 3917518a238SChris Mason 3927518a238SChris Mason if (nritems == 0) 3937518a238SChris Mason return 0; 3947518a238SChris Mason 3957518a238SChris Mason if (parent) { 396a1f39630SAneesh parent_slot = path->slots[level + 1]; 3975f39d397SChris Mason btrfs_node_key(parent, &parent_key, parent_slot); 3985f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, 0); 3996702ed49SChris Mason 4005f39d397SChris Mason BUG_ON(memcmp(&parent_key, &leaf_key, 401e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 4021d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 403db94535dSChris Mason btrfs_header_bytenr(leaf)); 404aa5d6bedSChris Mason } 4055f39d397SChris Mason #if 0 4065f39d397SChris Mason for (i = 0; nritems > 1 && i < nritems - 2; i++) { 4075f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, i + 1); 4085f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, i); 4095f39d397SChris Mason if (comp_keys(&leaf_key, &cpukey) >= 0) { 4105f39d397SChris Mason btrfs_print_leaf(root, leaf); 4115f39d397SChris Mason printk("slot %d offset bad key\n", i); 4125f39d397SChris Mason BUG_ON(1); 4135f39d397SChris Mason } 4145f39d397SChris Mason if (btrfs_item_offset_nr(leaf, i) != 4155f39d397SChris Mason btrfs_item_end_nr(leaf, i + 1)) { 4165f39d397SChris Mason btrfs_print_leaf(root, leaf); 4175f39d397SChris Mason printk("slot %d offset bad\n", i); 4185f39d397SChris Mason BUG_ON(1); 4195f39d397SChris Mason } 4205f39d397SChris Mason if (i == 0) { 4215f39d397SChris Mason if (btrfs_item_offset_nr(leaf, i) + 4225f39d397SChris Mason btrfs_item_size_nr(leaf, i) != 4235f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 4245f39d397SChris Mason btrfs_print_leaf(root, leaf); 4255f39d397SChris Mason printk("slot %d first offset bad\n", i); 4265f39d397SChris Mason BUG_ON(1); 4275f39d397SChris Mason } 4285f39d397SChris Mason } 4295f39d397SChris Mason } 4305f39d397SChris Mason if (nritems > 0) { 4315f39d397SChris Mason if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) { 4325f39d397SChris Mason btrfs_print_leaf(root, leaf); 4335f39d397SChris Mason printk("slot %d bad size \n", nritems - 1); 4345f39d397SChris Mason BUG_ON(1); 4355f39d397SChris Mason } 4365f39d397SChris Mason } 4375f39d397SChris Mason #endif 4385f39d397SChris Mason if (slot != 0 && slot < nritems - 1) { 4395f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, slot); 4405f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1); 4415f39d397SChris Mason if (comp_keys(&leaf_key, &cpukey) <= 0) { 4425f39d397SChris Mason btrfs_print_leaf(root, leaf); 4435f39d397SChris Mason printk("slot %d offset bad key\n", slot); 4445f39d397SChris Mason BUG_ON(1); 4455f39d397SChris Mason } 4465f39d397SChris Mason if (btrfs_item_offset_nr(leaf, slot - 1) != 4475f39d397SChris Mason btrfs_item_end_nr(leaf, slot)) { 4485f39d397SChris Mason btrfs_print_leaf(root, leaf); 4495f39d397SChris Mason printk("slot %d offset bad\n", slot); 4505f39d397SChris Mason BUG_ON(1); 4515f39d397SChris Mason } 452aa5d6bedSChris Mason } 4538d7be552SChris Mason if (slot < nritems - 1) { 4545f39d397SChris Mason btrfs_item_key(leaf, &leaf_key, slot); 4555f39d397SChris Mason btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1); 4565f39d397SChris Mason BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0); 4575f39d397SChris Mason if (btrfs_item_offset_nr(leaf, slot) != 4585f39d397SChris Mason btrfs_item_end_nr(leaf, slot + 1)) { 4595f39d397SChris Mason btrfs_print_leaf(root, leaf); 4605f39d397SChris Mason printk("slot %d offset bad\n", slot); 4615f39d397SChris Mason BUG_ON(1); 462aa5d6bedSChris Mason } 4635f39d397SChris Mason } 4645f39d397SChris Mason BUG_ON(btrfs_item_offset_nr(leaf, 0) + 4655f39d397SChris Mason btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root)); 466aa5d6bedSChris Mason return 0; 467aa5d6bedSChris Mason } 468aa5d6bedSChris Mason 469123abc88SChris Mason static int check_block(struct btrfs_root *root, struct btrfs_path *path, 470123abc88SChris Mason int level) 471aa5d6bedSChris Mason { 472810191ffSChris Mason return 0; 473db94535dSChris Mason #if 0 4745f39d397SChris Mason struct extent_buffer *buf = path->nodes[level]; 4755f39d397SChris Mason 476479965d6SChris Mason if (memcmp_extent_buffer(buf, root->fs_info->fsid, 477479965d6SChris Mason (unsigned long)btrfs_header_fsid(buf), 478479965d6SChris Mason BTRFS_FSID_SIZE)) { 4795f39d397SChris Mason printk("warning bad block %Lu\n", buf->start); 480db94535dSChris Mason return 1; 4815f39d397SChris Mason } 482db94535dSChris Mason #endif 483aa5d6bedSChris Mason if (level == 0) 484123abc88SChris Mason return check_leaf(root, path, level); 485123abc88SChris Mason return check_node(root, path, level); 486aa5d6bedSChris Mason } 487aa5d6bedSChris Mason 48874123bd7SChris Mason /* 4895f39d397SChris Mason * search for key in the extent_buffer. The items start at offset p, 4905f39d397SChris Mason * and they are item_size apart. There are 'max' items in p. 4915f39d397SChris Mason * 49274123bd7SChris Mason * the slot in the array is returned via slot, and it points to 49374123bd7SChris Mason * the place where you would insert key if it is not found in 49474123bd7SChris Mason * the array. 49574123bd7SChris Mason * 49674123bd7SChris Mason * slot may point to max if the key is bigger than all of the keys 49774123bd7SChris Mason */ 4985f39d397SChris Mason static int generic_bin_search(struct extent_buffer *eb, unsigned long p, 4995f39d397SChris Mason int item_size, struct btrfs_key *key, 500be0e5c09SChris Mason int max, int *slot) 501be0e5c09SChris Mason { 502be0e5c09SChris Mason int low = 0; 503be0e5c09SChris Mason int high = max; 504be0e5c09SChris Mason int mid; 505be0e5c09SChris Mason int ret; 506479965d6SChris Mason struct btrfs_disk_key *tmp = NULL; 5075f39d397SChris Mason struct btrfs_disk_key unaligned; 5085f39d397SChris Mason unsigned long offset; 5095f39d397SChris Mason char *map_token = NULL; 5105f39d397SChris Mason char *kaddr = NULL; 5115f39d397SChris Mason unsigned long map_start = 0; 5125f39d397SChris Mason unsigned long map_len = 0; 513479965d6SChris Mason int err; 514be0e5c09SChris Mason 515be0e5c09SChris Mason while(low < high) { 516be0e5c09SChris Mason mid = (low + high) / 2; 5175f39d397SChris Mason offset = p + mid * item_size; 5185f39d397SChris Mason 5195f39d397SChris Mason if (!map_token || offset < map_start || 5205f39d397SChris Mason (offset + sizeof(struct btrfs_disk_key)) > 5215f39d397SChris Mason map_start + map_len) { 522479965d6SChris Mason if (map_token) { 5235f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 524479965d6SChris Mason map_token = NULL; 525479965d6SChris Mason } 526479965d6SChris Mason err = map_extent_buffer(eb, offset, 527479965d6SChris Mason sizeof(struct btrfs_disk_key), 528479965d6SChris Mason &map_token, &kaddr, 5295f39d397SChris Mason &map_start, &map_len, KM_USER0); 5305f39d397SChris Mason 531479965d6SChris Mason if (!err) { 532479965d6SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 533479965d6SChris Mason map_start); 534479965d6SChris Mason } else { 5355f39d397SChris Mason read_extent_buffer(eb, &unaligned, 5365f39d397SChris Mason offset, sizeof(unaligned)); 5375f39d397SChris Mason tmp = &unaligned; 538479965d6SChris Mason } 539479965d6SChris Mason 5405f39d397SChris Mason } else { 5415f39d397SChris Mason tmp = (struct btrfs_disk_key *)(kaddr + offset - 5425f39d397SChris Mason map_start); 5435f39d397SChris Mason } 544be0e5c09SChris Mason ret = comp_keys(tmp, key); 545be0e5c09SChris Mason 546be0e5c09SChris Mason if (ret < 0) 547be0e5c09SChris Mason low = mid + 1; 548be0e5c09SChris Mason else if (ret > 0) 549be0e5c09SChris Mason high = mid; 550be0e5c09SChris Mason else { 551be0e5c09SChris Mason *slot = mid; 552479965d6SChris Mason if (map_token) 5535f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 554be0e5c09SChris Mason return 0; 555be0e5c09SChris Mason } 556be0e5c09SChris Mason } 557be0e5c09SChris Mason *slot = low; 5585f39d397SChris Mason if (map_token) 5595f39d397SChris Mason unmap_extent_buffer(eb, map_token, KM_USER0); 560be0e5c09SChris Mason return 1; 561be0e5c09SChris Mason } 562be0e5c09SChris Mason 56397571fd0SChris Mason /* 56497571fd0SChris Mason * simple bin_search frontend that does the right thing for 56597571fd0SChris Mason * leaves vs nodes 56697571fd0SChris Mason */ 5675f39d397SChris Mason static int bin_search(struct extent_buffer *eb, struct btrfs_key *key, 5685f39d397SChris Mason int level, int *slot) 569be0e5c09SChris Mason { 5705f39d397SChris Mason if (level == 0) { 5715f39d397SChris Mason return generic_bin_search(eb, 5725f39d397SChris Mason offsetof(struct btrfs_leaf, items), 5730783fcfcSChris Mason sizeof(struct btrfs_item), 5745f39d397SChris Mason key, btrfs_header_nritems(eb), 5757518a238SChris Mason slot); 576be0e5c09SChris Mason } else { 5775f39d397SChris Mason return generic_bin_search(eb, 5785f39d397SChris Mason offsetof(struct btrfs_node, ptrs), 579123abc88SChris Mason sizeof(struct btrfs_key_ptr), 5805f39d397SChris Mason key, btrfs_header_nritems(eb), 5817518a238SChris Mason slot); 582be0e5c09SChris Mason } 583be0e5c09SChris Mason return -1; 584be0e5c09SChris Mason } 585be0e5c09SChris Mason 5865f39d397SChris Mason static struct extent_buffer *read_node_slot(struct btrfs_root *root, 5875f39d397SChris Mason struct extent_buffer *parent, int slot) 588bb803951SChris Mason { 589bb803951SChris Mason if (slot < 0) 590bb803951SChris Mason return NULL; 5915f39d397SChris Mason if (slot >= btrfs_header_nritems(parent)) 592bb803951SChris Mason return NULL; 593db94535dSChris Mason return read_tree_block(root, btrfs_node_blockptr(parent, slot), 594db94535dSChris Mason btrfs_level_size(root, btrfs_header_level(parent) - 1)); 595bb803951SChris Mason } 596bb803951SChris Mason 597e089f05cSChris Mason static int balance_level(struct btrfs_trans_handle *trans, struct btrfs_root 598e089f05cSChris Mason *root, struct btrfs_path *path, int level) 599bb803951SChris Mason { 6005f39d397SChris Mason struct extent_buffer *right = NULL; 6015f39d397SChris Mason struct extent_buffer *mid; 6025f39d397SChris Mason struct extent_buffer *left = NULL; 6035f39d397SChris Mason struct extent_buffer *parent = NULL; 604bb803951SChris Mason int ret = 0; 605bb803951SChris Mason int wret; 606bb803951SChris Mason int pslot; 607bb803951SChris Mason int orig_slot = path->slots[level]; 60854aa1f4dSChris Mason int err_on_enospc = 0; 60979f95c82SChris Mason u64 orig_ptr; 610bb803951SChris Mason 611bb803951SChris Mason if (level == 0) 612bb803951SChris Mason return 0; 613bb803951SChris Mason 6145f39d397SChris Mason mid = path->nodes[level]; 6151d4f8a0cSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 61679f95c82SChris Mason 617234b63a0SChris Mason if (level < BTRFS_MAX_LEVEL - 1) 6185f39d397SChris Mason parent = path->nodes[level + 1]; 619bb803951SChris Mason pslot = path->slots[level + 1]; 620bb803951SChris Mason 62140689478SChris Mason /* 62240689478SChris Mason * deal with the case where there is only one pointer in the root 62340689478SChris Mason * by promoting the node below to a root 62440689478SChris Mason */ 6255f39d397SChris Mason if (!parent) { 6265f39d397SChris Mason struct extent_buffer *child; 627bb803951SChris Mason 6285f39d397SChris Mason if (btrfs_header_nritems(mid) != 1) 629bb803951SChris Mason return 0; 630bb803951SChris Mason 631bb803951SChris Mason /* promote the child to a root */ 6325f39d397SChris Mason child = read_node_slot(root, mid, 0); 633bb803951SChris Mason BUG_ON(!child); 634bb803951SChris Mason root->node = child; 635bb803951SChris Mason path->nodes[level] = NULL; 6365f39d397SChris Mason clean_tree_block(trans, root, mid); 6375f39d397SChris Mason wait_on_tree_block_writeback(root, mid); 638bb803951SChris Mason /* once for the path */ 6395f39d397SChris Mason free_extent_buffer(mid); 640db94535dSChris Mason ret = btrfs_free_extent(trans, root, mid->start, mid->len, 1); 641bb803951SChris Mason /* once for the root ptr */ 6425f39d397SChris Mason free_extent_buffer(mid); 643db94535dSChris Mason return ret; 644bb803951SChris Mason } 6455f39d397SChris Mason if (btrfs_header_nritems(mid) > 646123abc88SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) / 4) 647bb803951SChris Mason return 0; 648bb803951SChris Mason 6495f39d397SChris Mason if (btrfs_header_nritems(mid) < 2) 65054aa1f4dSChris Mason err_on_enospc = 1; 65154aa1f4dSChris Mason 6525f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 6535f39d397SChris Mason if (left) { 6545f39d397SChris Mason wret = btrfs_cow_block(trans, root, left, 6555f39d397SChris Mason parent, pslot - 1, &left); 65654aa1f4dSChris Mason if (wret) { 65754aa1f4dSChris Mason ret = wret; 65854aa1f4dSChris Mason goto enospc; 65954aa1f4dSChris Mason } 6602cc58cf2SChris Mason } 6615f39d397SChris Mason right = read_node_slot(root, parent, pslot + 1); 6625f39d397SChris Mason if (right) { 6635f39d397SChris Mason wret = btrfs_cow_block(trans, root, right, 6645f39d397SChris Mason parent, pslot + 1, &right); 6652cc58cf2SChris Mason if (wret) { 6662cc58cf2SChris Mason ret = wret; 6672cc58cf2SChris Mason goto enospc; 6682cc58cf2SChris Mason } 6692cc58cf2SChris Mason } 6702cc58cf2SChris Mason 6712cc58cf2SChris Mason /* first, try to make some room in the middle buffer */ 6725f39d397SChris Mason if (left) { 6735f39d397SChris Mason orig_slot += btrfs_header_nritems(left); 6745f39d397SChris Mason wret = push_node_left(trans, root, left, mid); 67579f95c82SChris Mason if (wret < 0) 67679f95c82SChris Mason ret = wret; 6775f39d397SChris Mason if (btrfs_header_nritems(mid) < 2) 67854aa1f4dSChris Mason err_on_enospc = 1; 679bb803951SChris Mason } 68079f95c82SChris Mason 68179f95c82SChris Mason /* 68279f95c82SChris Mason * then try to empty the right most buffer into the middle 68379f95c82SChris Mason */ 6845f39d397SChris Mason if (right) { 6855f39d397SChris Mason wret = push_node_left(trans, root, mid, right); 68654aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 68779f95c82SChris Mason ret = wret; 6885f39d397SChris Mason if (btrfs_header_nritems(right) == 0) { 689db94535dSChris Mason u64 bytenr = right->start; 690db94535dSChris Mason u32 blocksize = right->len; 691db94535dSChris Mason 6925f39d397SChris Mason clean_tree_block(trans, root, right); 6935f39d397SChris Mason wait_on_tree_block_writeback(root, right); 6945f39d397SChris Mason free_extent_buffer(right); 695bb803951SChris Mason right = NULL; 696e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot + 697e089f05cSChris Mason 1); 698bb803951SChris Mason if (wret) 699bb803951SChris Mason ret = wret; 700db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 701db94535dSChris Mason blocksize, 1); 702bb803951SChris Mason if (wret) 703bb803951SChris Mason ret = wret; 704bb803951SChris Mason } else { 7055f39d397SChris Mason struct btrfs_disk_key right_key; 7065f39d397SChris Mason btrfs_node_key(right, &right_key, 0); 7075f39d397SChris Mason btrfs_set_node_key(parent, &right_key, pslot + 1); 7085f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 709bb803951SChris Mason } 710bb803951SChris Mason } 7115f39d397SChris Mason if (btrfs_header_nritems(mid) == 1) { 71279f95c82SChris Mason /* 71379f95c82SChris Mason * we're not allowed to leave a node with one item in the 71479f95c82SChris Mason * tree during a delete. A deletion from lower in the tree 71579f95c82SChris Mason * could try to delete the only pointer in this node. 71679f95c82SChris Mason * So, pull some keys from the left. 71779f95c82SChris Mason * There has to be a left pointer at this point because 71879f95c82SChris Mason * otherwise we would have pulled some pointers from the 71979f95c82SChris Mason * right 72079f95c82SChris Mason */ 7215f39d397SChris Mason BUG_ON(!left); 7225f39d397SChris Mason wret = balance_node_right(trans, root, mid, left); 72354aa1f4dSChris Mason if (wret < 0) { 72479f95c82SChris Mason ret = wret; 72554aa1f4dSChris Mason goto enospc; 72654aa1f4dSChris Mason } 72779f95c82SChris Mason BUG_ON(wret == 1); 72879f95c82SChris Mason } 7295f39d397SChris Mason if (btrfs_header_nritems(mid) == 0) { 73079f95c82SChris Mason /* we've managed to empty the middle node, drop it */ 731db94535dSChris Mason u64 bytenr = mid->start; 732db94535dSChris Mason u32 blocksize = mid->len; 7335f39d397SChris Mason clean_tree_block(trans, root, mid); 7345f39d397SChris Mason wait_on_tree_block_writeback(root, mid); 7355f39d397SChris Mason free_extent_buffer(mid); 736bb803951SChris Mason mid = NULL; 737e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot); 738bb803951SChris Mason if (wret) 739bb803951SChris Mason ret = wret; 740db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, blocksize, 1); 741bb803951SChris Mason if (wret) 742bb803951SChris Mason ret = wret; 74379f95c82SChris Mason } else { 74479f95c82SChris Mason /* update the parent key to reflect our changes */ 7455f39d397SChris Mason struct btrfs_disk_key mid_key; 7465f39d397SChris Mason btrfs_node_key(mid, &mid_key, 0); 7475f39d397SChris Mason btrfs_set_node_key(parent, &mid_key, pslot); 7485f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 74979f95c82SChris Mason } 750bb803951SChris Mason 75179f95c82SChris Mason /* update the path */ 7525f39d397SChris Mason if (left) { 7535f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 7545f39d397SChris Mason extent_buffer_get(left); 7555f39d397SChris Mason path->nodes[level] = left; 756bb803951SChris Mason path->slots[level + 1] -= 1; 757bb803951SChris Mason path->slots[level] = orig_slot; 7585f39d397SChris Mason if (mid) 7595f39d397SChris Mason free_extent_buffer(mid); 760bb803951SChris Mason } else { 7615f39d397SChris Mason orig_slot -= btrfs_header_nritems(left); 762bb803951SChris Mason path->slots[level] = orig_slot; 763bb803951SChris Mason } 764bb803951SChris Mason } 76579f95c82SChris Mason /* double check we haven't messed things up */ 766123abc88SChris Mason check_block(root, path, level); 767e20d96d6SChris Mason if (orig_ptr != 7685f39d397SChris Mason btrfs_node_blockptr(path->nodes[level], path->slots[level])) 76979f95c82SChris Mason BUG(); 77054aa1f4dSChris Mason enospc: 7715f39d397SChris Mason if (right) 7725f39d397SChris Mason free_extent_buffer(right); 7735f39d397SChris Mason if (left) 7745f39d397SChris Mason free_extent_buffer(left); 775bb803951SChris Mason return ret; 776bb803951SChris Mason } 777bb803951SChris Mason 778e66f709bSChris Mason /* returns zero if the push worked, non-zero otherwise */ 779e66f709bSChris Mason static int push_nodes_for_insert(struct btrfs_trans_handle *trans, 780e66f709bSChris Mason struct btrfs_root *root, 781e66f709bSChris Mason struct btrfs_path *path, int level) 782e66f709bSChris Mason { 7835f39d397SChris Mason struct extent_buffer *right = NULL; 7845f39d397SChris Mason struct extent_buffer *mid; 7855f39d397SChris Mason struct extent_buffer *left = NULL; 7865f39d397SChris Mason struct extent_buffer *parent = NULL; 787e66f709bSChris Mason int ret = 0; 788e66f709bSChris Mason int wret; 789e66f709bSChris Mason int pslot; 790e66f709bSChris Mason int orig_slot = path->slots[level]; 791e66f709bSChris Mason u64 orig_ptr; 792e66f709bSChris Mason 793e66f709bSChris Mason if (level == 0) 794e66f709bSChris Mason return 1; 795e66f709bSChris Mason 7965f39d397SChris Mason mid = path->nodes[level]; 797e66f709bSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 798e66f709bSChris Mason 799e66f709bSChris Mason if (level < BTRFS_MAX_LEVEL - 1) 8005f39d397SChris Mason parent = path->nodes[level + 1]; 801e66f709bSChris Mason pslot = path->slots[level + 1]; 802e66f709bSChris Mason 8035f39d397SChris Mason if (!parent) 804e66f709bSChris Mason return 1; 805e66f709bSChris Mason 8065f39d397SChris Mason left = read_node_slot(root, parent, pslot - 1); 807e66f709bSChris Mason 808e66f709bSChris Mason /* first, try to make some room in the middle buffer */ 8095f39d397SChris Mason if (left) { 810e66f709bSChris Mason u32 left_nr; 8115f39d397SChris Mason left_nr = btrfs_header_nritems(left); 81233ade1f8SChris Mason if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 81333ade1f8SChris Mason wret = 1; 81433ade1f8SChris Mason } else { 8155f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, parent, 8165f39d397SChris Mason pslot - 1, &left); 81754aa1f4dSChris Mason if (ret) 81854aa1f4dSChris Mason wret = 1; 81954aa1f4dSChris Mason else { 82054aa1f4dSChris Mason wret = push_node_left(trans, root, 8215f39d397SChris Mason left, mid); 82254aa1f4dSChris Mason } 82333ade1f8SChris Mason } 824e66f709bSChris Mason if (wret < 0) 825e66f709bSChris Mason ret = wret; 826e66f709bSChris Mason if (wret == 0) { 8275f39d397SChris Mason struct btrfs_disk_key disk_key; 828e66f709bSChris Mason orig_slot += left_nr; 8295f39d397SChris Mason btrfs_node_key(mid, &disk_key, 0); 8305f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot); 8315f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 8325f39d397SChris Mason if (btrfs_header_nritems(left) > orig_slot) { 8335f39d397SChris Mason path->nodes[level] = left; 834e66f709bSChris Mason path->slots[level + 1] -= 1; 835e66f709bSChris Mason path->slots[level] = orig_slot; 8365f39d397SChris Mason free_extent_buffer(mid); 837e66f709bSChris Mason } else { 838e66f709bSChris Mason orig_slot -= 8395f39d397SChris Mason btrfs_header_nritems(left); 840e66f709bSChris Mason path->slots[level] = orig_slot; 8415f39d397SChris Mason free_extent_buffer(left); 842e66f709bSChris Mason } 843e66f709bSChris Mason return 0; 844e66f709bSChris Mason } 8455f39d397SChris Mason free_extent_buffer(left); 846e66f709bSChris Mason } 8475f39d397SChris Mason right= read_node_slot(root, parent, pslot + 1); 848e66f709bSChris Mason 849e66f709bSChris Mason /* 850e66f709bSChris Mason * then try to empty the right most buffer into the middle 851e66f709bSChris Mason */ 8525f39d397SChris Mason if (right) { 85333ade1f8SChris Mason u32 right_nr; 8545f39d397SChris Mason right_nr = btrfs_header_nritems(right); 85533ade1f8SChris Mason if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 85633ade1f8SChris Mason wret = 1; 85733ade1f8SChris Mason } else { 8585f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, 8595f39d397SChris Mason parent, pslot + 1, 8605f39d397SChris Mason &right); 86154aa1f4dSChris Mason if (ret) 86254aa1f4dSChris Mason wret = 1; 86354aa1f4dSChris Mason else { 86433ade1f8SChris Mason wret = balance_node_right(trans, root, 8655f39d397SChris Mason right, mid); 86633ade1f8SChris Mason } 86754aa1f4dSChris Mason } 868e66f709bSChris Mason if (wret < 0) 869e66f709bSChris Mason ret = wret; 870e66f709bSChris Mason if (wret == 0) { 8715f39d397SChris Mason struct btrfs_disk_key disk_key; 8725f39d397SChris Mason 8735f39d397SChris Mason btrfs_node_key(right, &disk_key, 0); 8745f39d397SChris Mason btrfs_set_node_key(parent, &disk_key, pslot + 1); 8755f39d397SChris Mason btrfs_mark_buffer_dirty(parent); 8765f39d397SChris Mason 8775f39d397SChris Mason if (btrfs_header_nritems(mid) <= orig_slot) { 8785f39d397SChris Mason path->nodes[level] = right; 879e66f709bSChris Mason path->slots[level + 1] += 1; 880e66f709bSChris Mason path->slots[level] = orig_slot - 8815f39d397SChris Mason btrfs_header_nritems(mid); 8825f39d397SChris Mason free_extent_buffer(mid); 883e66f709bSChris Mason } else { 8845f39d397SChris Mason free_extent_buffer(right); 885e66f709bSChris Mason } 886e66f709bSChris Mason return 0; 887e66f709bSChris Mason } 8885f39d397SChris Mason free_extent_buffer(right); 889e66f709bSChris Mason } 890e66f709bSChris Mason return 1; 891e66f709bSChris Mason } 892e66f709bSChris Mason 89374123bd7SChris Mason /* 8943c69faecSChris Mason * readahead one full node of leaves 8953c69faecSChris Mason */ 8963c69faecSChris Mason static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path, 8976702ed49SChris Mason int level, int slot) 8983c69faecSChris Mason { 8995f39d397SChris Mason struct extent_buffer *node; 9003c69faecSChris Mason u32 nritems; 9013c69faecSChris Mason u64 search; 9026b80053dSChris Mason u64 lowest_read; 9036b80053dSChris Mason u64 highest_read; 9046b80053dSChris Mason u64 nread = 0; 9053c69faecSChris Mason int direction = path->reada; 9065f39d397SChris Mason struct extent_buffer *eb; 9076b80053dSChris Mason u32 nr; 9086b80053dSChris Mason u32 blocksize; 9096b80053dSChris Mason u32 nscan = 0; 910db94535dSChris Mason 911a6b6e75eSChris Mason if (level != 1) 9123c69faecSChris Mason return; 9133c69faecSChris Mason 9146702ed49SChris Mason if (!path->nodes[level]) 9156702ed49SChris Mason return; 9166702ed49SChris Mason 9175f39d397SChris Mason node = path->nodes[level]; 9183c69faecSChris Mason search = btrfs_node_blockptr(node, slot); 9196b80053dSChris Mason blocksize = btrfs_level_size(root, level - 1); 9206b80053dSChris Mason eb = btrfs_find_tree_block(root, search, blocksize); 9215f39d397SChris Mason if (eb) { 9225f39d397SChris Mason free_extent_buffer(eb); 9233c69faecSChris Mason return; 9243c69faecSChris Mason } 9253c69faecSChris Mason 9266b80053dSChris Mason highest_read = search; 9276b80053dSChris Mason lowest_read = search; 9286b80053dSChris Mason 9295f39d397SChris Mason nritems = btrfs_header_nritems(node); 9306b80053dSChris Mason nr = slot; 9313c69faecSChris Mason while(1) { 9326b80053dSChris Mason if (direction < 0) { 9336b80053dSChris Mason if (nr == 0) 9343c69faecSChris Mason break; 9356b80053dSChris Mason nr--; 9366b80053dSChris Mason } else if (direction > 0) { 9376b80053dSChris Mason nr++; 9386b80053dSChris Mason if (nr >= nritems) 9396b80053dSChris Mason break; 9403c69faecSChris Mason } 9416b80053dSChris Mason search = btrfs_node_blockptr(node, nr); 9426b80053dSChris Mason if ((search >= lowest_read && search <= highest_read) || 9436b80053dSChris Mason (search < lowest_read && lowest_read - search <= 32768) || 9446b80053dSChris Mason (search > highest_read && search - highest_read <= 32768)) { 9456b80053dSChris Mason readahead_tree_block(root, search, blocksize); 9466b80053dSChris Mason nread += blocksize; 9473c69faecSChris Mason } 9486b80053dSChris Mason nscan++; 9496b80053dSChris Mason if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32)) 9506b80053dSChris Mason break; 9516b80053dSChris Mason if(nread > (1024 * 1024) || nscan > 128) 9526b80053dSChris Mason break; 9536b80053dSChris Mason 9546b80053dSChris Mason if (search < lowest_read) 9556b80053dSChris Mason lowest_read = search; 9566b80053dSChris Mason if (search > highest_read) 9576b80053dSChris Mason highest_read = search; 9583c69faecSChris Mason } 9593c69faecSChris Mason } 9603c69faecSChris Mason /* 96174123bd7SChris Mason * look for key in the tree. path is filled in with nodes along the way 96274123bd7SChris Mason * if key is found, we return zero and you can find the item in the leaf 96374123bd7SChris Mason * level of the path (level 0) 96474123bd7SChris Mason * 96574123bd7SChris Mason * If the key isn't found, the path points to the slot where it should 966aa5d6bedSChris Mason * be inserted, and 1 is returned. If there are other errors during the 967aa5d6bedSChris Mason * search a negative error number is returned. 96897571fd0SChris Mason * 96997571fd0SChris Mason * if ins_len > 0, nodes and leaves will be split as we walk down the 97097571fd0SChris Mason * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if 97197571fd0SChris Mason * possible) 97274123bd7SChris Mason */ 973e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root 974e089f05cSChris Mason *root, struct btrfs_key *key, struct btrfs_path *p, int 975e089f05cSChris Mason ins_len, int cow) 976be0e5c09SChris Mason { 9775f39d397SChris Mason struct extent_buffer *b; 978db94535dSChris Mason u64 bytenr; 979be0e5c09SChris Mason int slot; 980be0e5c09SChris Mason int ret; 981be0e5c09SChris Mason int level; 9823c69faecSChris Mason int should_reada = p->reada; 9839f3a7427SChris Mason u8 lowest_level = 0; 9849f3a7427SChris Mason 9856702ed49SChris Mason lowest_level = p->lowest_level; 9866702ed49SChris Mason WARN_ON(lowest_level && ins_len); 98722b0ebdaSChris Mason WARN_ON(p->nodes[0] != NULL); 98822b0ebdaSChris Mason WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex)); 989bb803951SChris Mason again: 990bb803951SChris Mason b = root->node; 9915f39d397SChris Mason extent_buffer_get(b); 992eb60ceacSChris Mason while (b) { 9935f39d397SChris Mason level = btrfs_header_level(b); 99402217ed2SChris Mason if (cow) { 99502217ed2SChris Mason int wret; 996e20d96d6SChris Mason wret = btrfs_cow_block(trans, root, b, 997e20d96d6SChris Mason p->nodes[level + 1], 998e20d96d6SChris Mason p->slots[level + 1], 999252c38f0SYan &b); 100054aa1f4dSChris Mason if (wret) { 10015f39d397SChris Mason free_extent_buffer(b); 100254aa1f4dSChris Mason return wret; 100354aa1f4dSChris Mason } 100402217ed2SChris Mason } 100502217ed2SChris Mason BUG_ON(!cow && ins_len); 10065f39d397SChris Mason if (level != btrfs_header_level(b)) 10072c90e5d6SChris Mason WARN_ON(1); 10085f39d397SChris Mason level = btrfs_header_level(b); 1009eb60ceacSChris Mason p->nodes[level] = b; 1010123abc88SChris Mason ret = check_block(root, p, level); 1011aa5d6bedSChris Mason if (ret) 1012aa5d6bedSChris Mason return -1; 10135f39d397SChris Mason ret = bin_search(b, key, level, &slot); 10145f39d397SChris Mason if (level != 0) { 1015be0e5c09SChris Mason if (ret && slot > 0) 1016be0e5c09SChris Mason slot -= 1; 1017be0e5c09SChris Mason p->slots[level] = slot; 10185f39d397SChris Mason if (ins_len > 0 && btrfs_header_nritems(b) >= 1019d4dbff95SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 1) { 1020e089f05cSChris Mason int sret = split_node(trans, root, p, level); 10215c680ed6SChris Mason BUG_ON(sret > 0); 10225c680ed6SChris Mason if (sret) 10235c680ed6SChris Mason return sret; 10245c680ed6SChris Mason b = p->nodes[level]; 10255c680ed6SChris Mason slot = p->slots[level]; 1026bb803951SChris Mason } else if (ins_len < 0) { 1027e089f05cSChris Mason int sret = balance_level(trans, root, p, 1028e089f05cSChris Mason level); 1029bb803951SChris Mason if (sret) 1030bb803951SChris Mason return sret; 1031bb803951SChris Mason b = p->nodes[level]; 1032f510cfecSChris Mason if (!b) { 1033f510cfecSChris Mason btrfs_release_path(NULL, p); 1034bb803951SChris Mason goto again; 1035f510cfecSChris Mason } 1036bb803951SChris Mason slot = p->slots[level]; 10375f39d397SChris Mason BUG_ON(btrfs_header_nritems(b) == 1); 10385c680ed6SChris Mason } 10399f3a7427SChris Mason /* this is only true while dropping a snapshot */ 10409f3a7427SChris Mason if (level == lowest_level) 10419f3a7427SChris Mason break; 1042db94535dSChris Mason bytenr = btrfs_node_blockptr(b, slot); 10436702ed49SChris Mason if (should_reada) 10446702ed49SChris Mason reada_for_search(root, p, level, slot); 1045db94535dSChris Mason b = read_tree_block(root, bytenr, 1046db94535dSChris Mason btrfs_level_size(root, level - 1)); 1047be0e5c09SChris Mason } else { 1048be0e5c09SChris Mason p->slots[level] = slot; 10495f39d397SChris Mason if (ins_len > 0 && btrfs_leaf_free_space(root, b) < 10500783fcfcSChris Mason sizeof(struct btrfs_item) + ins_len) { 1051d4dbff95SChris Mason int sret = split_leaf(trans, root, key, 1052d4dbff95SChris Mason p, ins_len); 10535c680ed6SChris Mason BUG_ON(sret > 0); 10545c680ed6SChris Mason if (sret) 10555c680ed6SChris Mason return sret; 10565c680ed6SChris Mason } 1057be0e5c09SChris Mason return ret; 1058be0e5c09SChris Mason } 1059be0e5c09SChris Mason } 1060aa5d6bedSChris Mason return 1; 1061be0e5c09SChris Mason } 1062be0e5c09SChris Mason 106374123bd7SChris Mason /* 106474123bd7SChris Mason * adjust the pointers going up the tree, starting at level 106574123bd7SChris Mason * making sure the right key of each node is points to 'key'. 106674123bd7SChris Mason * This is used after shifting pointers to the left, so it stops 106774123bd7SChris Mason * fixing up pointers when a given leaf/node is not in slot 0 of the 106874123bd7SChris Mason * higher levels 1069aa5d6bedSChris Mason * 1070aa5d6bedSChris Mason * If this fails to write a tree block, it returns -1, but continues 1071aa5d6bedSChris Mason * fixing up the blocks in ram so the tree is consistent. 107274123bd7SChris Mason */ 10735f39d397SChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans, 10745f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 10755f39d397SChris Mason struct btrfs_disk_key *key, int level) 1076be0e5c09SChris Mason { 1077be0e5c09SChris Mason int i; 1078aa5d6bedSChris Mason int ret = 0; 10795f39d397SChris Mason struct extent_buffer *t; 10805f39d397SChris Mason 1081234b63a0SChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 1082be0e5c09SChris Mason int tslot = path->slots[i]; 1083eb60ceacSChris Mason if (!path->nodes[i]) 1084be0e5c09SChris Mason break; 10855f39d397SChris Mason t = path->nodes[i]; 10865f39d397SChris Mason btrfs_set_node_key(t, key, tslot); 1087d6025579SChris Mason btrfs_mark_buffer_dirty(path->nodes[i]); 1088be0e5c09SChris Mason if (tslot != 0) 1089be0e5c09SChris Mason break; 1090be0e5c09SChris Mason } 1091aa5d6bedSChris Mason return ret; 1092be0e5c09SChris Mason } 1093be0e5c09SChris Mason 109474123bd7SChris Mason /* 109574123bd7SChris Mason * try to push data from one node into the next node left in the 109679f95c82SChris Mason * tree. 1097aa5d6bedSChris Mason * 1098aa5d6bedSChris Mason * returns 0 if some ptrs were pushed left, < 0 if there was some horrible 1099aa5d6bedSChris Mason * error, and > 0 if there was no room in the left hand block. 110074123bd7SChris Mason */ 1101e089f05cSChris Mason static int push_node_left(struct btrfs_trans_handle *trans, struct btrfs_root 11025f39d397SChris Mason *root, struct extent_buffer *dst, 11035f39d397SChris Mason struct extent_buffer *src) 1104be0e5c09SChris Mason { 1105be0e5c09SChris Mason int push_items = 0; 1106bb803951SChris Mason int src_nritems; 1107bb803951SChris Mason int dst_nritems; 1108aa5d6bedSChris Mason int ret = 0; 1109be0e5c09SChris Mason 11105f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 11115f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1112123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 111354aa1f4dSChris Mason 1114eb60ceacSChris Mason if (push_items <= 0) { 1115be0e5c09SChris Mason return 1; 1116eb60ceacSChris Mason } 1117be0e5c09SChris Mason 1118be0e5c09SChris Mason if (src_nritems < push_items) 1119be0e5c09SChris Mason push_items = src_nritems; 112079f95c82SChris Mason 11215f39d397SChris Mason copy_extent_buffer(dst, src, 11225f39d397SChris Mason btrfs_node_key_ptr_offset(dst_nritems), 11235f39d397SChris Mason btrfs_node_key_ptr_offset(0), 1124123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 11255f39d397SChris Mason 1126bb803951SChris Mason if (push_items < src_nritems) { 11275f39d397SChris Mason memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0), 11285f39d397SChris Mason btrfs_node_key_ptr_offset(push_items), 1129e2fa7227SChris Mason (src_nritems - push_items) * 1130123abc88SChris Mason sizeof(struct btrfs_key_ptr)); 1131bb803951SChris Mason } 11325f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 11335f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 11345f39d397SChris Mason btrfs_mark_buffer_dirty(src); 11355f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 1136bb803951SChris Mason return ret; 1137be0e5c09SChris Mason } 1138be0e5c09SChris Mason 113997571fd0SChris Mason /* 114079f95c82SChris Mason * try to push data from one node into the next node right in the 114179f95c82SChris Mason * tree. 114279f95c82SChris Mason * 114379f95c82SChris Mason * returns 0 if some ptrs were pushed, < 0 if there was some horrible 114479f95c82SChris Mason * error, and > 0 if there was no room in the right hand block. 114579f95c82SChris Mason * 114679f95c82SChris Mason * this will only push up to 1/2 the contents of the left node over 114779f95c82SChris Mason */ 11485f39d397SChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, 11495f39d397SChris Mason struct btrfs_root *root, 11505f39d397SChris Mason struct extent_buffer *dst, 11515f39d397SChris Mason struct extent_buffer *src) 115279f95c82SChris Mason { 115379f95c82SChris Mason int push_items = 0; 115479f95c82SChris Mason int max_push; 115579f95c82SChris Mason int src_nritems; 115679f95c82SChris Mason int dst_nritems; 115779f95c82SChris Mason int ret = 0; 115879f95c82SChris Mason 11595f39d397SChris Mason src_nritems = btrfs_header_nritems(src); 11605f39d397SChris Mason dst_nritems = btrfs_header_nritems(dst); 1161123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 11625f39d397SChris Mason if (push_items <= 0) 116379f95c82SChris Mason return 1; 116479f95c82SChris Mason 116579f95c82SChris Mason max_push = src_nritems / 2 + 1; 116679f95c82SChris Mason /* don't try to empty the node */ 1167252c38f0SYan if (max_push >= src_nritems) 116879f95c82SChris Mason return 1; 1169252c38f0SYan 117079f95c82SChris Mason if (max_push < push_items) 117179f95c82SChris Mason push_items = max_push; 117279f95c82SChris Mason 11735f39d397SChris Mason memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items), 11745f39d397SChris Mason btrfs_node_key_ptr_offset(0), 11755f39d397SChris Mason (dst_nritems) * 11765f39d397SChris Mason sizeof(struct btrfs_key_ptr)); 1177d6025579SChris Mason 11785f39d397SChris Mason copy_extent_buffer(dst, src, 11795f39d397SChris Mason btrfs_node_key_ptr_offset(0), 11805f39d397SChris Mason btrfs_node_key_ptr_offset(src_nritems - push_items), 1181123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 118279f95c82SChris Mason 11835f39d397SChris Mason btrfs_set_header_nritems(src, src_nritems - push_items); 11845f39d397SChris Mason btrfs_set_header_nritems(dst, dst_nritems + push_items); 118579f95c82SChris Mason 11865f39d397SChris Mason btrfs_mark_buffer_dirty(src); 11875f39d397SChris Mason btrfs_mark_buffer_dirty(dst); 118879f95c82SChris Mason return ret; 118979f95c82SChris Mason } 119079f95c82SChris Mason 119179f95c82SChris Mason /* 119297571fd0SChris Mason * helper function to insert a new root level in the tree. 119397571fd0SChris Mason * A new node is allocated, and a single item is inserted to 119497571fd0SChris Mason * point to the existing root 1195aa5d6bedSChris Mason * 1196aa5d6bedSChris Mason * returns zero on success or < 0 on failure. 119797571fd0SChris Mason */ 11985f39d397SChris Mason static int insert_new_root(struct btrfs_trans_handle *trans, 11995f39d397SChris Mason struct btrfs_root *root, 12005f39d397SChris Mason struct btrfs_path *path, int level) 120174123bd7SChris Mason { 12025f39d397SChris Mason struct extent_buffer *lower; 12035f39d397SChris Mason struct extent_buffer *c; 12045f39d397SChris Mason struct btrfs_disk_key lower_key; 12055c680ed6SChris Mason 12065c680ed6SChris Mason BUG_ON(path->nodes[level]); 12075c680ed6SChris Mason BUG_ON(path->nodes[level-1] != root->node); 12085c680ed6SChris Mason 1209db94535dSChris Mason c = btrfs_alloc_free_block(trans, root, root->nodesize, 1210db94535dSChris Mason root->node->start, 0); 12115f39d397SChris Mason if (IS_ERR(c)) 12125f39d397SChris Mason return PTR_ERR(c); 12135f39d397SChris Mason memset_extent_buffer(c, 0, 0, root->nodesize); 12145f39d397SChris Mason btrfs_set_header_nritems(c, 1); 12155f39d397SChris Mason btrfs_set_header_level(c, level); 1216db94535dSChris Mason btrfs_set_header_bytenr(c, c->start); 12175f39d397SChris Mason btrfs_set_header_generation(c, trans->transid); 12185f39d397SChris Mason btrfs_set_header_owner(c, root->root_key.objectid); 12195f39d397SChris Mason lower = path->nodes[level-1]; 1220d5719762SChris Mason 12215f39d397SChris Mason write_extent_buffer(c, root->fs_info->fsid, 12225f39d397SChris Mason (unsigned long)btrfs_header_fsid(c), 12235f39d397SChris Mason BTRFS_FSID_SIZE); 12245f39d397SChris Mason if (level == 1) 12255f39d397SChris Mason btrfs_item_key(lower, &lower_key, 0); 12265f39d397SChris Mason else 12275f39d397SChris Mason btrfs_node_key(lower, &lower_key, 0); 12285f39d397SChris Mason btrfs_set_node_key(c, &lower_key, 0); 1229db94535dSChris Mason btrfs_set_node_blockptr(c, 0, lower->start); 12305f39d397SChris Mason 12315f39d397SChris Mason btrfs_mark_buffer_dirty(c); 1232d5719762SChris Mason 1233cfaa7295SChris Mason /* the super has an extra ref to root->node */ 12345f39d397SChris Mason free_extent_buffer(root->node); 12355f39d397SChris Mason root->node = c; 12365f39d397SChris Mason extent_buffer_get(c); 12375f39d397SChris Mason path->nodes[level] = c; 123874123bd7SChris Mason path->slots[level] = 0; 123974123bd7SChris Mason return 0; 124074123bd7SChris Mason } 12415c680ed6SChris Mason 12425c680ed6SChris Mason /* 12435c680ed6SChris Mason * worker function to insert a single pointer in a node. 12445c680ed6SChris Mason * the node should have enough room for the pointer already 124597571fd0SChris Mason * 12465c680ed6SChris Mason * slot and level indicate where you want the key to go, and 12475c680ed6SChris Mason * blocknr is the block the key points to. 1248aa5d6bedSChris Mason * 1249aa5d6bedSChris Mason * returns zero on success and < 0 on any error 12505c680ed6SChris Mason */ 1251e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root 1252e089f05cSChris Mason *root, struct btrfs_path *path, struct btrfs_disk_key 1253db94535dSChris Mason *key, u64 bytenr, int slot, int level) 12545c680ed6SChris Mason { 12555f39d397SChris Mason struct extent_buffer *lower; 12565c680ed6SChris Mason int nritems; 12575c680ed6SChris Mason 12585c680ed6SChris Mason BUG_ON(!path->nodes[level]); 12595f39d397SChris Mason lower = path->nodes[level]; 12605f39d397SChris Mason nritems = btrfs_header_nritems(lower); 126174123bd7SChris Mason if (slot > nritems) 126274123bd7SChris Mason BUG(); 1263123abc88SChris Mason if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root)) 126474123bd7SChris Mason BUG(); 126574123bd7SChris Mason if (slot != nritems) { 12665f39d397SChris Mason memmove_extent_buffer(lower, 12675f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 12685f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 1269123abc88SChris Mason (nritems - slot) * sizeof(struct btrfs_key_ptr)); 127074123bd7SChris Mason } 12715f39d397SChris Mason btrfs_set_node_key(lower, key, slot); 1272db94535dSChris Mason btrfs_set_node_blockptr(lower, slot, bytenr); 12735f39d397SChris Mason btrfs_set_header_nritems(lower, nritems + 1); 12745f39d397SChris Mason btrfs_mark_buffer_dirty(lower); 127574123bd7SChris Mason return 0; 127674123bd7SChris Mason } 127774123bd7SChris Mason 127897571fd0SChris Mason /* 127997571fd0SChris Mason * split the node at the specified level in path in two. 128097571fd0SChris Mason * The path is corrected to point to the appropriate node after the split 128197571fd0SChris Mason * 128297571fd0SChris Mason * Before splitting this tries to make some room in the node by pushing 128397571fd0SChris Mason * left and right, if either one works, it returns right away. 1284aa5d6bedSChris Mason * 1285aa5d6bedSChris Mason * returns 0 on success and < 0 on failure 128697571fd0SChris Mason */ 1287e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 1288e089f05cSChris Mason *root, struct btrfs_path *path, int level) 1289be0e5c09SChris Mason { 12905f39d397SChris Mason struct extent_buffer *c; 12915f39d397SChris Mason struct extent_buffer *split; 12925f39d397SChris Mason struct btrfs_disk_key disk_key; 1293be0e5c09SChris Mason int mid; 12945c680ed6SChris Mason int ret; 1295aa5d6bedSChris Mason int wret; 12967518a238SChris Mason u32 c_nritems; 1297be0e5c09SChris Mason 12985f39d397SChris Mason c = path->nodes[level]; 12995f39d397SChris Mason if (c == root->node) { 13005c680ed6SChris Mason /* trying to split the root, lets make a new one */ 1301e089f05cSChris Mason ret = insert_new_root(trans, root, path, level + 1); 13025c680ed6SChris Mason if (ret) 13035c680ed6SChris Mason return ret; 1304e66f709bSChris Mason } else { 1305e66f709bSChris Mason ret = push_nodes_for_insert(trans, root, path, level); 13065f39d397SChris Mason c = path->nodes[level]; 13075f39d397SChris Mason if (!ret && btrfs_header_nritems(c) < 1308e66f709bSChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) - 1) 1309e66f709bSChris Mason return 0; 131054aa1f4dSChris Mason if (ret < 0) 131154aa1f4dSChris Mason return ret; 13125c680ed6SChris Mason } 1313e66f709bSChris Mason 13145f39d397SChris Mason c_nritems = btrfs_header_nritems(c); 1315db94535dSChris Mason split = btrfs_alloc_free_block(trans, root, root->nodesize, 1316db94535dSChris Mason c->start, 0); 13175f39d397SChris Mason if (IS_ERR(split)) 13185f39d397SChris Mason return PTR_ERR(split); 131954aa1f4dSChris Mason 13205f39d397SChris Mason btrfs_set_header_flags(split, btrfs_header_flags(c)); 13215f39d397SChris Mason btrfs_set_header_level(split, btrfs_header_level(c)); 1322db94535dSChris Mason btrfs_set_header_bytenr(split, split->start); 13235f39d397SChris Mason btrfs_set_header_generation(split, trans->transid); 13245f39d397SChris Mason btrfs_set_header_owner(split, root->root_key.objectid); 13255f39d397SChris Mason write_extent_buffer(split, root->fs_info->fsid, 13265f39d397SChris Mason (unsigned long)btrfs_header_fsid(split), 13275f39d397SChris Mason BTRFS_FSID_SIZE); 13285f39d397SChris Mason 13297518a238SChris Mason mid = (c_nritems + 1) / 2; 13305f39d397SChris Mason 13315f39d397SChris Mason copy_extent_buffer(split, c, 13325f39d397SChris Mason btrfs_node_key_ptr_offset(0), 13335f39d397SChris Mason btrfs_node_key_ptr_offset(mid), 1334123abc88SChris Mason (c_nritems - mid) * sizeof(struct btrfs_key_ptr)); 13355f39d397SChris Mason btrfs_set_header_nritems(split, c_nritems - mid); 13365f39d397SChris Mason btrfs_set_header_nritems(c, mid); 1337aa5d6bedSChris Mason ret = 0; 1338aa5d6bedSChris Mason 13395f39d397SChris Mason btrfs_mark_buffer_dirty(c); 13405f39d397SChris Mason btrfs_mark_buffer_dirty(split); 13415f39d397SChris Mason 13425f39d397SChris Mason btrfs_node_key(split, &disk_key, 0); 1343db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, split->start, 13445f39d397SChris Mason path->slots[level + 1] + 1, 1345123abc88SChris Mason level + 1); 1346aa5d6bedSChris Mason if (wret) 1347aa5d6bedSChris Mason ret = wret; 1348aa5d6bedSChris Mason 13495de08d7dSChris Mason if (path->slots[level] >= mid) { 13505c680ed6SChris Mason path->slots[level] -= mid; 13515f39d397SChris Mason free_extent_buffer(c); 13525f39d397SChris Mason path->nodes[level] = split; 13535c680ed6SChris Mason path->slots[level + 1] += 1; 1354eb60ceacSChris Mason } else { 13555f39d397SChris Mason free_extent_buffer(split); 1356be0e5c09SChris Mason } 1357aa5d6bedSChris Mason return ret; 1358be0e5c09SChris Mason } 1359be0e5c09SChris Mason 136074123bd7SChris Mason /* 136174123bd7SChris Mason * how many bytes are required to store the items in a leaf. start 136274123bd7SChris Mason * and nr indicate which items in the leaf to check. This totals up the 136374123bd7SChris Mason * space used both by the item structs and the item data 136474123bd7SChris Mason */ 13655f39d397SChris Mason static int leaf_space_used(struct extent_buffer *l, int start, int nr) 1366be0e5c09SChris Mason { 1367be0e5c09SChris Mason int data_len; 13685f39d397SChris Mason int nritems = btrfs_header_nritems(l); 1369d4dbff95SChris Mason int end = min(nritems, start + nr) - 1; 1370be0e5c09SChris Mason 1371be0e5c09SChris Mason if (!nr) 1372be0e5c09SChris Mason return 0; 13735f39d397SChris Mason data_len = btrfs_item_end_nr(l, start); 13745f39d397SChris Mason data_len = data_len - btrfs_item_offset_nr(l, end); 13750783fcfcSChris Mason data_len += sizeof(struct btrfs_item) * nr; 1376d4dbff95SChris Mason WARN_ON(data_len < 0); 1377be0e5c09SChris Mason return data_len; 1378be0e5c09SChris Mason } 1379be0e5c09SChris Mason 138074123bd7SChris Mason /* 1381d4dbff95SChris Mason * The space between the end of the leaf items and 1382d4dbff95SChris Mason * the start of the leaf data. IOW, how much room 1383d4dbff95SChris Mason * the leaf has left for both items and data 1384d4dbff95SChris Mason */ 13855f39d397SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf) 1386d4dbff95SChris Mason { 13875f39d397SChris Mason int nritems = btrfs_header_nritems(leaf); 13885f39d397SChris Mason int ret; 13895f39d397SChris Mason ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems); 13905f39d397SChris Mason if (ret < 0) { 13915f39d397SChris Mason printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n", 1392ae2f5411SJens Axboe ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root), 13935f39d397SChris Mason leaf_space_used(leaf, 0, nritems), nritems); 13945f39d397SChris Mason } 13955f39d397SChris Mason return ret; 1396d4dbff95SChris Mason } 1397d4dbff95SChris Mason 1398d4dbff95SChris Mason /* 139900ec4c51SChris Mason * push some data in the path leaf to the right, trying to free up at 140000ec4c51SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 1401aa5d6bedSChris Mason * 1402aa5d6bedSChris Mason * returns 1 if the push failed because the other node didn't have enough 1403aa5d6bedSChris Mason * room, 0 if everything worked out and < 0 if there were major errors. 140400ec4c51SChris Mason */ 1405e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root 1406e089f05cSChris Mason *root, struct btrfs_path *path, int data_size) 140700ec4c51SChris Mason { 14085f39d397SChris Mason struct extent_buffer *left = path->nodes[0]; 14095f39d397SChris Mason struct extent_buffer *right; 14105f39d397SChris Mason struct extent_buffer *upper; 14115f39d397SChris Mason struct btrfs_disk_key disk_key; 141200ec4c51SChris Mason int slot; 141300ec4c51SChris Mason int i; 141400ec4c51SChris Mason int free_space; 141500ec4c51SChris Mason int push_space = 0; 141600ec4c51SChris Mason int push_items = 0; 14170783fcfcSChris Mason struct btrfs_item *item; 14187518a238SChris Mason u32 left_nritems; 14197518a238SChris Mason u32 right_nritems; 14205f39d397SChris Mason u32 data_end; 1421db94535dSChris Mason u32 this_item_size; 142254aa1f4dSChris Mason int ret; 142300ec4c51SChris Mason 142400ec4c51SChris Mason slot = path->slots[1]; 142500ec4c51SChris Mason if (!path->nodes[1]) { 142600ec4c51SChris Mason return 1; 142700ec4c51SChris Mason } 142800ec4c51SChris Mason upper = path->nodes[1]; 14295f39d397SChris Mason if (slot >= btrfs_header_nritems(upper) - 1) 143000ec4c51SChris Mason return 1; 14315f39d397SChris Mason 1432db94535dSChris Mason right = read_tree_block(root, btrfs_node_blockptr(upper, slot + 1), 1433db94535dSChris Mason root->leafsize); 1434123abc88SChris Mason free_space = btrfs_leaf_free_space(root, right); 14350783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 14365f39d397SChris Mason free_extent_buffer(right); 143702217ed2SChris Mason return 1; 143802217ed2SChris Mason } 143902217ed2SChris Mason 14405f39d397SChris Mason /* cow and double check */ 14415f39d397SChris Mason ret = btrfs_cow_block(trans, root, right, upper, 14425f39d397SChris Mason slot + 1, &right); 14435f39d397SChris Mason if (ret) { 14445f39d397SChris Mason free_extent_buffer(right); 1445a429e513SChris Mason return 1; 1446a429e513SChris Mason } 14475f39d397SChris Mason free_space = btrfs_leaf_free_space(root, right); 14485f39d397SChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 14495f39d397SChris Mason free_extent_buffer(right); 14505f39d397SChris Mason return 1; 14515f39d397SChris Mason } 14525f39d397SChris Mason 14535f39d397SChris Mason left_nritems = btrfs_header_nritems(left); 14545f39d397SChris Mason if (left_nritems == 0) { 14555f39d397SChris Mason free_extent_buffer(right); 14565f39d397SChris Mason return 1; 14575f39d397SChris Mason } 14585f39d397SChris Mason 1459a429e513SChris Mason for (i = left_nritems - 1; i >= 1; i--) { 14605f39d397SChris Mason item = btrfs_item_nr(left, i); 1461db94535dSChris Mason 146200ec4c51SChris Mason if (path->slots[0] == i) 146300ec4c51SChris Mason push_space += data_size + sizeof(*item); 1464db94535dSChris Mason 1465db94535dSChris Mason if (!left->map_token) { 1466db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 1467db94535dSChris Mason sizeof(struct btrfs_item), 1468db94535dSChris Mason &left->map_token, &left->kaddr, 1469db94535dSChris Mason &left->map_start, &left->map_len, 1470db94535dSChris Mason KM_USER1); 1471db94535dSChris Mason } 1472db94535dSChris Mason 1473db94535dSChris Mason this_item_size = btrfs_item_size(left, item); 1474db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 147500ec4c51SChris Mason break; 147600ec4c51SChris Mason push_items++; 1477db94535dSChris Mason push_space += this_item_size + sizeof(*item); 1478db94535dSChris Mason } 1479db94535dSChris Mason if (left->map_token) { 1480db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 1481db94535dSChris Mason left->map_token = NULL; 148200ec4c51SChris Mason } 14835f39d397SChris Mason 148400ec4c51SChris Mason if (push_items == 0) { 14855f39d397SChris Mason free_extent_buffer(right); 148600ec4c51SChris Mason return 1; 148700ec4c51SChris Mason } 14885f39d397SChris Mason 1489a429e513SChris Mason if (push_items == left_nritems) 1490a429e513SChris Mason WARN_ON(1); 14915f39d397SChris Mason 149200ec4c51SChris Mason /* push left to right */ 14935f39d397SChris Mason right_nritems = btrfs_header_nritems(right); 14945f39d397SChris Mason push_space = btrfs_item_end_nr(left, left_nritems - push_items); 1495123abc88SChris Mason push_space -= leaf_data_end(root, left); 14965f39d397SChris Mason 149700ec4c51SChris Mason /* make room in the right data area */ 14985f39d397SChris Mason data_end = leaf_data_end(root, right); 14995f39d397SChris Mason memmove_extent_buffer(right, 15005f39d397SChris Mason btrfs_leaf_data(right) + data_end - push_space, 15015f39d397SChris Mason btrfs_leaf_data(right) + data_end, 15025f39d397SChris Mason BTRFS_LEAF_DATA_SIZE(root) - data_end); 15035f39d397SChris Mason 150400ec4c51SChris Mason /* copy from the left data area */ 15055f39d397SChris Mason copy_extent_buffer(right, left, btrfs_leaf_data(right) + 1506d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 1507d6025579SChris Mason btrfs_leaf_data(left) + leaf_data_end(root, left), 1508d6025579SChris Mason push_space); 15095f39d397SChris Mason 15105f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(push_items), 15115f39d397SChris Mason btrfs_item_nr_offset(0), 15120783fcfcSChris Mason right_nritems * sizeof(struct btrfs_item)); 15135f39d397SChris Mason 151400ec4c51SChris Mason /* copy the items from left to right */ 15155f39d397SChris Mason copy_extent_buffer(right, left, btrfs_item_nr_offset(0), 15165f39d397SChris Mason btrfs_item_nr_offset(left_nritems - push_items), 15170783fcfcSChris Mason push_items * sizeof(struct btrfs_item)); 151800ec4c51SChris Mason 151900ec4c51SChris Mason /* update the item pointers */ 15207518a238SChris Mason right_nritems += push_items; 15215f39d397SChris Mason btrfs_set_header_nritems(right, right_nritems); 1522123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 1523db94535dSChris Mason 15247518a238SChris Mason for (i = 0; i < right_nritems; i++) { 15255f39d397SChris Mason item = btrfs_item_nr(right, i); 1526db94535dSChris Mason if (!right->map_token) { 1527db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1528db94535dSChris Mason sizeof(struct btrfs_item), 1529db94535dSChris Mason &right->map_token, &right->kaddr, 1530db94535dSChris Mason &right->map_start, &right->map_len, 1531db94535dSChris Mason KM_USER1); 1532db94535dSChris Mason } 1533db94535dSChris Mason push_space -= btrfs_item_size(right, item); 1534db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 1535db94535dSChris Mason } 1536db94535dSChris Mason 1537db94535dSChris Mason if (right->map_token) { 1538db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1539db94535dSChris Mason right->map_token = NULL; 154000ec4c51SChris Mason } 15417518a238SChris Mason left_nritems -= push_items; 15425f39d397SChris Mason btrfs_set_header_nritems(left, left_nritems); 154300ec4c51SChris Mason 15445f39d397SChris Mason btrfs_mark_buffer_dirty(left); 15455f39d397SChris Mason btrfs_mark_buffer_dirty(right); 1546a429e513SChris Mason 15475f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 15485f39d397SChris Mason btrfs_set_node_key(upper, &disk_key, slot + 1); 1549d6025579SChris Mason btrfs_mark_buffer_dirty(upper); 155002217ed2SChris Mason 155100ec4c51SChris Mason /* then fixup the leaf pointer in the path */ 15527518a238SChris Mason if (path->slots[0] >= left_nritems) { 15537518a238SChris Mason path->slots[0] -= left_nritems; 15545f39d397SChris Mason free_extent_buffer(path->nodes[0]); 15555f39d397SChris Mason path->nodes[0] = right; 155600ec4c51SChris Mason path->slots[1] += 1; 155700ec4c51SChris Mason } else { 15585f39d397SChris Mason free_extent_buffer(right); 155900ec4c51SChris Mason } 156000ec4c51SChris Mason return 0; 156100ec4c51SChris Mason } 156200ec4c51SChris Mason /* 156374123bd7SChris Mason * push some data in the path leaf to the left, trying to free up at 156474123bd7SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 156574123bd7SChris Mason */ 1566e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root 1567e089f05cSChris Mason *root, struct btrfs_path *path, int data_size) 1568be0e5c09SChris Mason { 15695f39d397SChris Mason struct btrfs_disk_key disk_key; 15705f39d397SChris Mason struct extent_buffer *right = path->nodes[0]; 15715f39d397SChris Mason struct extent_buffer *left; 1572be0e5c09SChris Mason int slot; 1573be0e5c09SChris Mason int i; 1574be0e5c09SChris Mason int free_space; 1575be0e5c09SChris Mason int push_space = 0; 1576be0e5c09SChris Mason int push_items = 0; 15770783fcfcSChris Mason struct btrfs_item *item; 15787518a238SChris Mason u32 old_left_nritems; 15795f39d397SChris Mason u32 right_nritems; 1580aa5d6bedSChris Mason int ret = 0; 1581aa5d6bedSChris Mason int wret; 1582db94535dSChris Mason u32 this_item_size; 1583db94535dSChris Mason u32 old_left_item_size; 1584be0e5c09SChris Mason 1585be0e5c09SChris Mason slot = path->slots[1]; 15865f39d397SChris Mason if (slot == 0) 1587be0e5c09SChris Mason return 1; 15885f39d397SChris Mason if (!path->nodes[1]) 1589be0e5c09SChris Mason return 1; 15905f39d397SChris Mason 1591*3685f791SChris Mason right_nritems = btrfs_header_nritems(right); 1592*3685f791SChris Mason if (right_nritems == 0) { 1593*3685f791SChris Mason return 1; 1594*3685f791SChris Mason } 1595*3685f791SChris Mason 15965f39d397SChris Mason left = read_tree_block(root, btrfs_node_blockptr(path->nodes[1], 1597db94535dSChris Mason slot - 1), root->leafsize); 1598123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 15990783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 16005f39d397SChris Mason free_extent_buffer(left); 1601be0e5c09SChris Mason return 1; 1602be0e5c09SChris Mason } 160302217ed2SChris Mason 160402217ed2SChris Mason /* cow and double check */ 16055f39d397SChris Mason ret = btrfs_cow_block(trans, root, left, 16065f39d397SChris Mason path->nodes[1], slot - 1, &left); 160754aa1f4dSChris Mason if (ret) { 160854aa1f4dSChris Mason /* we hit -ENOSPC, but it isn't fatal here */ 16095f39d397SChris Mason free_extent_buffer(left); 161054aa1f4dSChris Mason return 1; 161154aa1f4dSChris Mason } 1612*3685f791SChris Mason 1613123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 16140783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 16155f39d397SChris Mason free_extent_buffer(left); 161602217ed2SChris Mason return 1; 161702217ed2SChris Mason } 161802217ed2SChris Mason 16195f39d397SChris Mason for (i = 0; i < right_nritems - 1; i++) { 16205f39d397SChris Mason item = btrfs_item_nr(right, i); 1621db94535dSChris Mason if (!right->map_token) { 1622db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1623db94535dSChris Mason sizeof(struct btrfs_item), 1624db94535dSChris Mason &right->map_token, &right->kaddr, 1625db94535dSChris Mason &right->map_start, &right->map_len, 1626db94535dSChris Mason KM_USER1); 1627db94535dSChris Mason } 1628db94535dSChris Mason 1629be0e5c09SChris Mason if (path->slots[0] == i) 1630be0e5c09SChris Mason push_space += data_size + sizeof(*item); 1631db94535dSChris Mason 1632db94535dSChris Mason this_item_size = btrfs_item_size(right, item); 1633db94535dSChris Mason if (this_item_size + sizeof(*item) + push_space > free_space) 1634be0e5c09SChris Mason break; 1635db94535dSChris Mason 1636be0e5c09SChris Mason push_items++; 1637db94535dSChris Mason push_space += this_item_size + sizeof(*item); 1638be0e5c09SChris Mason } 1639db94535dSChris Mason 1640db94535dSChris Mason if (right->map_token) { 1641db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1642db94535dSChris Mason right->map_token = NULL; 1643db94535dSChris Mason } 1644db94535dSChris Mason 1645be0e5c09SChris Mason if (push_items == 0) { 16465f39d397SChris Mason free_extent_buffer(left); 1647be0e5c09SChris Mason return 1; 1648be0e5c09SChris Mason } 16495f39d397SChris Mason if (push_items == btrfs_header_nritems(right)) 1650a429e513SChris Mason WARN_ON(1); 16515f39d397SChris Mason 1652be0e5c09SChris Mason /* push data from right to left */ 16535f39d397SChris Mason copy_extent_buffer(left, right, 16545f39d397SChris Mason btrfs_item_nr_offset(btrfs_header_nritems(left)), 16555f39d397SChris Mason btrfs_item_nr_offset(0), 16565f39d397SChris Mason push_items * sizeof(struct btrfs_item)); 16575f39d397SChris Mason 1658123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root) - 16595f39d397SChris Mason btrfs_item_offset_nr(right, push_items -1); 16605f39d397SChris Mason 16615f39d397SChris Mason copy_extent_buffer(left, right, btrfs_leaf_data(left) + 1662d6025579SChris Mason leaf_data_end(root, left) - push_space, 1663123abc88SChris Mason btrfs_leaf_data(right) + 16645f39d397SChris Mason btrfs_item_offset_nr(right, push_items - 1), 1665be0e5c09SChris Mason push_space); 16665f39d397SChris Mason old_left_nritems = btrfs_header_nritems(left); 1667eb60ceacSChris Mason BUG_ON(old_left_nritems < 0); 1668eb60ceacSChris Mason 1669db94535dSChris Mason old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1); 1670be0e5c09SChris Mason for (i = old_left_nritems; i < old_left_nritems + push_items; i++) { 16715f39d397SChris Mason u32 ioff; 1672db94535dSChris Mason 16735f39d397SChris Mason item = btrfs_item_nr(left, i); 1674db94535dSChris Mason if (!left->map_token) { 1675db94535dSChris Mason map_extent_buffer(left, (unsigned long)item, 1676db94535dSChris Mason sizeof(struct btrfs_item), 1677db94535dSChris Mason &left->map_token, &left->kaddr, 1678db94535dSChris Mason &left->map_start, &left->map_len, 1679db94535dSChris Mason KM_USER1); 1680db94535dSChris Mason } 1681db94535dSChris Mason 16825f39d397SChris Mason ioff = btrfs_item_offset(left, item); 16835f39d397SChris Mason btrfs_set_item_offset(left, item, 1684db94535dSChris Mason ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size)); 1685be0e5c09SChris Mason } 16865f39d397SChris Mason btrfs_set_header_nritems(left, old_left_nritems + push_items); 1687db94535dSChris Mason if (left->map_token) { 1688db94535dSChris Mason unmap_extent_buffer(left, left->map_token, KM_USER1); 1689db94535dSChris Mason left->map_token = NULL; 1690db94535dSChris Mason } 1691be0e5c09SChris Mason 1692be0e5c09SChris Mason /* fixup right node */ 16935f39d397SChris Mason push_space = btrfs_item_offset_nr(right, push_items - 1) - 1694123abc88SChris Mason leaf_data_end(root, right); 16955f39d397SChris Mason memmove_extent_buffer(right, btrfs_leaf_data(right) + 1696d6025579SChris Mason BTRFS_LEAF_DATA_SIZE(root) - push_space, 1697d6025579SChris Mason btrfs_leaf_data(right) + 1698123abc88SChris Mason leaf_data_end(root, right), push_space); 16995f39d397SChris Mason 17005f39d397SChris Mason memmove_extent_buffer(right, btrfs_item_nr_offset(0), 17015f39d397SChris Mason btrfs_item_nr_offset(push_items), 17025f39d397SChris Mason (btrfs_header_nritems(right) - push_items) * 17030783fcfcSChris Mason sizeof(struct btrfs_item)); 17045f39d397SChris Mason 17055f39d397SChris Mason right_nritems = btrfs_header_nritems(right) - push_items; 17065f39d397SChris Mason btrfs_set_header_nritems(right, right_nritems); 1707123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 1708eb60ceacSChris Mason 17095f39d397SChris Mason for (i = 0; i < right_nritems; i++) { 17105f39d397SChris Mason item = btrfs_item_nr(right, i); 1711db94535dSChris Mason 1712db94535dSChris Mason if (!right->map_token) { 1713db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1714db94535dSChris Mason sizeof(struct btrfs_item), 1715db94535dSChris Mason &right->map_token, &right->kaddr, 1716db94535dSChris Mason &right->map_start, &right->map_len, 1717db94535dSChris Mason KM_USER1); 1718db94535dSChris Mason } 1719db94535dSChris Mason 1720db94535dSChris Mason push_space = push_space - btrfs_item_size(right, item); 1721db94535dSChris Mason btrfs_set_item_offset(right, item, push_space); 1722db94535dSChris Mason } 1723db94535dSChris Mason if (right->map_token) { 1724db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1725db94535dSChris Mason right->map_token = NULL; 1726be0e5c09SChris Mason } 1727eb60ceacSChris Mason 17285f39d397SChris Mason btrfs_mark_buffer_dirty(left); 17295f39d397SChris Mason btrfs_mark_buffer_dirty(right); 1730098f59c2SChris Mason 17315f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 17325f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, 1); 1733aa5d6bedSChris Mason if (wret) 1734aa5d6bedSChris Mason ret = wret; 1735be0e5c09SChris Mason 1736be0e5c09SChris Mason /* then fixup the leaf pointer in the path */ 1737be0e5c09SChris Mason if (path->slots[0] < push_items) { 1738be0e5c09SChris Mason path->slots[0] += old_left_nritems; 17395f39d397SChris Mason free_extent_buffer(path->nodes[0]); 17405f39d397SChris Mason path->nodes[0] = left; 1741be0e5c09SChris Mason path->slots[1] -= 1; 1742be0e5c09SChris Mason } else { 17435f39d397SChris Mason free_extent_buffer(left); 1744be0e5c09SChris Mason path->slots[0] -= push_items; 1745be0e5c09SChris Mason } 1746eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 1747aa5d6bedSChris Mason return ret; 1748be0e5c09SChris Mason } 1749be0e5c09SChris Mason 175074123bd7SChris Mason /* 175174123bd7SChris Mason * split the path's leaf in two, making sure there is at least data_size 175274123bd7SChris Mason * available for the resulting leaf level of the path. 1753aa5d6bedSChris Mason * 1754aa5d6bedSChris Mason * returns 0 if all went well and < 0 on failure. 175574123bd7SChris Mason */ 1756e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 1757d4dbff95SChris Mason *root, struct btrfs_key *ins_key, 1758d4dbff95SChris Mason struct btrfs_path *path, int data_size) 1759be0e5c09SChris Mason { 17605f39d397SChris Mason struct extent_buffer *l; 17617518a238SChris Mason u32 nritems; 1762eb60ceacSChris Mason int mid; 1763eb60ceacSChris Mason int slot; 17645f39d397SChris Mason struct extent_buffer *right; 17650783fcfcSChris Mason int space_needed = data_size + sizeof(struct btrfs_item); 1766be0e5c09SChris Mason int data_copy_size; 1767be0e5c09SChris Mason int rt_data_off; 1768be0e5c09SChris Mason int i; 1769d4dbff95SChris Mason int ret = 0; 1770aa5d6bedSChris Mason int wret; 1771d4dbff95SChris Mason int double_split = 0; 1772d4dbff95SChris Mason struct btrfs_disk_key disk_key; 1773be0e5c09SChris Mason 177440689478SChris Mason /* first try to make some room by pushing left and right */ 1775*3685f791SChris Mason if (ins_key->type != BTRFS_DIR_ITEM_KEY) { 1776*3685f791SChris Mason wret = push_leaf_right(trans, root, path, data_size); 17773326d1b0SChris Mason if (wret < 0) { 1778eaee50e8SChris Mason return wret; 17793326d1b0SChris Mason } 1780eaee50e8SChris Mason if (wret) { 1781*3685f791SChris Mason wret = push_leaf_left(trans, root, path, data_size); 1782eaee50e8SChris Mason if (wret < 0) 1783eaee50e8SChris Mason return wret; 1784eaee50e8SChris Mason } 17855f39d397SChris Mason l = path->nodes[0]; 1786aa5d6bedSChris Mason 1787aa5d6bedSChris Mason /* did the pushes work? */ 1788123abc88SChris Mason if (btrfs_leaf_free_space(root, l) >= 17893326d1b0SChris Mason sizeof(struct btrfs_item) + data_size) { 1790be0e5c09SChris Mason return 0; 17913326d1b0SChris Mason } 1792*3685f791SChris Mason } else { 1793*3685f791SChris Mason l = path->nodes[0]; 1794*3685f791SChris Mason } 1795aa5d6bedSChris Mason 17965c680ed6SChris Mason if (!path->nodes[1]) { 1797e089f05cSChris Mason ret = insert_new_root(trans, root, path, 1); 17985c680ed6SChris Mason if (ret) 17995c680ed6SChris Mason return ret; 18005c680ed6SChris Mason } 1801eb60ceacSChris Mason slot = path->slots[0]; 18025f39d397SChris Mason nritems = btrfs_header_nritems(l); 1803eb60ceacSChris Mason mid = (nritems + 1)/ 2; 180454aa1f4dSChris Mason 1805db94535dSChris Mason right = btrfs_alloc_free_block(trans, root, root->leafsize, 1806db94535dSChris Mason l->start, 0); 18075f39d397SChris Mason if (IS_ERR(right)) 18085f39d397SChris Mason return PTR_ERR(right); 180954aa1f4dSChris Mason 18105f39d397SChris Mason memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header)); 1811db94535dSChris Mason btrfs_set_header_bytenr(right, right->start); 18125f39d397SChris Mason btrfs_set_header_generation(right, trans->transid); 18135f39d397SChris Mason btrfs_set_header_owner(right, root->root_key.objectid); 18145f39d397SChris Mason btrfs_set_header_level(right, 0); 18155f39d397SChris Mason write_extent_buffer(right, root->fs_info->fsid, 18165f39d397SChris Mason (unsigned long)btrfs_header_fsid(right), 18175f39d397SChris Mason BTRFS_FSID_SIZE); 18185f39d397SChris Mason 1819d4dbff95SChris Mason if (mid <= slot) { 1820d4dbff95SChris Mason if (nritems == 1 || 1821d4dbff95SChris Mason leaf_space_used(l, mid, nritems - mid) + space_needed > 1822d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 1823d4dbff95SChris Mason if (slot >= nritems) { 1824d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 18255f39d397SChris Mason btrfs_set_header_nritems(right, 0); 1826d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 1827db94535dSChris Mason &disk_key, right->start, 1828d4dbff95SChris Mason path->slots[1] + 1, 1); 1829d4dbff95SChris Mason if (wret) 1830d4dbff95SChris Mason ret = wret; 18315f39d397SChris Mason free_extent_buffer(path->nodes[0]); 18325f39d397SChris Mason path->nodes[0] = right; 1833d4dbff95SChris Mason path->slots[0] = 0; 1834d4dbff95SChris Mason path->slots[1] += 1; 1835d4dbff95SChris Mason return ret; 1836d4dbff95SChris Mason } 1837d4dbff95SChris Mason mid = slot; 18383326d1b0SChris Mason if (mid != nritems && 18393326d1b0SChris Mason leaf_space_used(l, mid, nritems - mid) + 18403326d1b0SChris Mason space_needed > BTRFS_LEAF_DATA_SIZE(root)) { 1841d4dbff95SChris Mason double_split = 1; 1842d4dbff95SChris Mason } 18433326d1b0SChris Mason } 1844d4dbff95SChris Mason } else { 1845d4dbff95SChris Mason if (leaf_space_used(l, 0, mid + 1) + space_needed > 1846d4dbff95SChris Mason BTRFS_LEAF_DATA_SIZE(root)) { 1847d4dbff95SChris Mason if (slot == 0) { 1848d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 18495f39d397SChris Mason btrfs_set_header_nritems(right, 0); 1850d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 1851d4dbff95SChris Mason &disk_key, 1852db94535dSChris Mason right->start, 1853098f59c2SChris Mason path->slots[1], 1); 1854d4dbff95SChris Mason if (wret) 1855d4dbff95SChris Mason ret = wret; 18565f39d397SChris Mason free_extent_buffer(path->nodes[0]); 18575f39d397SChris Mason path->nodes[0] = right; 1858d4dbff95SChris Mason path->slots[0] = 0; 1859a429e513SChris Mason if (path->slots[1] == 0) { 1860a429e513SChris Mason wret = fixup_low_keys(trans, root, 1861a429e513SChris Mason path, &disk_key, 1); 1862a429e513SChris Mason if (wret) 1863a429e513SChris Mason ret = wret; 1864a429e513SChris Mason } 1865d4dbff95SChris Mason return ret; 1866d4dbff95SChris Mason } 1867d4dbff95SChris Mason mid = slot; 1868d4dbff95SChris Mason double_split = 1; 1869d4dbff95SChris Mason } 1870d4dbff95SChris Mason } 18715f39d397SChris Mason nritems = nritems - mid; 18725f39d397SChris Mason btrfs_set_header_nritems(right, nritems); 18735f39d397SChris Mason data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l); 18745f39d397SChris Mason 18755f39d397SChris Mason copy_extent_buffer(right, l, btrfs_item_nr_offset(0), 18765f39d397SChris Mason btrfs_item_nr_offset(mid), 18775f39d397SChris Mason nritems * sizeof(struct btrfs_item)); 18785f39d397SChris Mason 18795f39d397SChris Mason copy_extent_buffer(right, l, 1880d6025579SChris Mason btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - 1881123abc88SChris Mason data_copy_size, btrfs_leaf_data(l) + 1882123abc88SChris Mason leaf_data_end(root, l), data_copy_size); 188374123bd7SChris Mason 18845f39d397SChris Mason rt_data_off = BTRFS_LEAF_DATA_SIZE(root) - 18855f39d397SChris Mason btrfs_item_end_nr(l, mid); 18865f39d397SChris Mason 18875f39d397SChris Mason for (i = 0; i < nritems; i++) { 18885f39d397SChris Mason struct btrfs_item *item = btrfs_item_nr(right, i); 1889db94535dSChris Mason u32 ioff; 1890db94535dSChris Mason 1891db94535dSChris Mason if (!right->map_token) { 1892db94535dSChris Mason map_extent_buffer(right, (unsigned long)item, 1893db94535dSChris Mason sizeof(struct btrfs_item), 1894db94535dSChris Mason &right->map_token, &right->kaddr, 1895db94535dSChris Mason &right->map_start, &right->map_len, 1896db94535dSChris Mason KM_USER1); 1897db94535dSChris Mason } 1898db94535dSChris Mason 1899db94535dSChris Mason ioff = btrfs_item_offset(right, item); 19005f39d397SChris Mason btrfs_set_item_offset(right, item, ioff + rt_data_off); 19010783fcfcSChris Mason } 190274123bd7SChris Mason 1903db94535dSChris Mason if (right->map_token) { 1904db94535dSChris Mason unmap_extent_buffer(right, right->map_token, KM_USER1); 1905db94535dSChris Mason right->map_token = NULL; 1906db94535dSChris Mason } 1907db94535dSChris Mason 19085f39d397SChris Mason btrfs_set_header_nritems(l, mid); 1909aa5d6bedSChris Mason ret = 0; 19105f39d397SChris Mason btrfs_item_key(right, &disk_key, 0); 1911db94535dSChris Mason wret = insert_ptr(trans, root, path, &disk_key, right->start, 1912db94535dSChris Mason path->slots[1] + 1, 1); 1913aa5d6bedSChris Mason if (wret) 1914aa5d6bedSChris Mason ret = wret; 19155f39d397SChris Mason 19165f39d397SChris Mason btrfs_mark_buffer_dirty(right); 19175f39d397SChris Mason btrfs_mark_buffer_dirty(l); 1918eb60ceacSChris Mason BUG_ON(path->slots[0] != slot); 19195f39d397SChris Mason 1920be0e5c09SChris Mason if (mid <= slot) { 19215f39d397SChris Mason free_extent_buffer(path->nodes[0]); 19225f39d397SChris Mason path->nodes[0] = right; 1923be0e5c09SChris Mason path->slots[0] -= mid; 1924be0e5c09SChris Mason path->slots[1] += 1; 1925eb60ceacSChris Mason } else 19265f39d397SChris Mason free_extent_buffer(right); 19275f39d397SChris Mason 1928eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 1929d4dbff95SChris Mason 19303326d1b0SChris Mason if (!double_split) { 1931d4dbff95SChris Mason return ret; 19323326d1b0SChris Mason } 193354aa1f4dSChris Mason 1934db94535dSChris Mason right = btrfs_alloc_free_block(trans, root, root->leafsize, 1935db94535dSChris Mason l->start, 0); 19365f39d397SChris Mason if (IS_ERR(right)) 19375f39d397SChris Mason return PTR_ERR(right); 19385f39d397SChris Mason 19395f39d397SChris Mason memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header)); 1940db94535dSChris Mason btrfs_set_header_bytenr(right, right->start); 19415f39d397SChris Mason btrfs_set_header_generation(right, trans->transid); 19425f39d397SChris Mason btrfs_set_header_owner(right, root->root_key.objectid); 19435f39d397SChris Mason btrfs_set_header_level(right, 0); 19445f39d397SChris Mason write_extent_buffer(right, root->fs_info->fsid, 19455f39d397SChris Mason (unsigned long)btrfs_header_fsid(right), 19465f39d397SChris Mason BTRFS_FSID_SIZE); 19475f39d397SChris Mason 1948d4dbff95SChris Mason btrfs_cpu_key_to_disk(&disk_key, ins_key); 19495f39d397SChris Mason btrfs_set_header_nritems(right, 0); 1950d4dbff95SChris Mason wret = insert_ptr(trans, root, path, 1951db94535dSChris Mason &disk_key, right->start, 1952d4dbff95SChris Mason path->slots[1], 1); 1953d4dbff95SChris Mason if (wret) 1954d4dbff95SChris Mason ret = wret; 1955a429e513SChris Mason if (path->slots[1] == 0) { 1956a429e513SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, 1); 1957a429e513SChris Mason if (wret) 1958a429e513SChris Mason ret = wret; 1959a429e513SChris Mason } 19605f39d397SChris Mason free_extent_buffer(path->nodes[0]); 19615f39d397SChris Mason path->nodes[0] = right; 1962d4dbff95SChris Mason path->slots[0] = 0; 1963be0e5c09SChris Mason return ret; 1964be0e5c09SChris Mason } 1965be0e5c09SChris Mason 1966b18c6685SChris Mason int btrfs_truncate_item(struct btrfs_trans_handle *trans, 1967b18c6685SChris Mason struct btrfs_root *root, 1968b18c6685SChris Mason struct btrfs_path *path, 1969b18c6685SChris Mason u32 new_size) 1970b18c6685SChris Mason { 1971b18c6685SChris Mason int ret = 0; 1972b18c6685SChris Mason int slot; 1973b18c6685SChris Mason int slot_orig; 19745f39d397SChris Mason struct extent_buffer *leaf; 19755f39d397SChris Mason struct btrfs_item *item; 1976b18c6685SChris Mason u32 nritems; 1977b18c6685SChris Mason unsigned int data_end; 1978b18c6685SChris Mason unsigned int old_data_start; 1979b18c6685SChris Mason unsigned int old_size; 1980b18c6685SChris Mason unsigned int size_diff; 1981b18c6685SChris Mason int i; 1982b18c6685SChris Mason 1983b18c6685SChris Mason slot_orig = path->slots[0]; 19845f39d397SChris Mason leaf = path->nodes[0]; 1985b18c6685SChris Mason 19865f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 1987b18c6685SChris Mason data_end = leaf_data_end(root, leaf); 1988b18c6685SChris Mason 1989b18c6685SChris Mason slot = path->slots[0]; 19905f39d397SChris Mason old_data_start = btrfs_item_offset_nr(leaf, slot); 19915f39d397SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 1992b18c6685SChris Mason BUG_ON(old_size <= new_size); 1993b18c6685SChris Mason size_diff = old_size - new_size; 1994b18c6685SChris Mason 1995b18c6685SChris Mason BUG_ON(slot < 0); 1996b18c6685SChris Mason BUG_ON(slot >= nritems); 1997b18c6685SChris Mason 1998b18c6685SChris Mason /* 1999b18c6685SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2000b18c6685SChris Mason */ 2001b18c6685SChris Mason /* first correct the data pointers */ 2002b18c6685SChris Mason for (i = slot; i < nritems; i++) { 20035f39d397SChris Mason u32 ioff; 20045f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2005db94535dSChris Mason 2006db94535dSChris Mason if (!leaf->map_token) { 2007db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2008db94535dSChris Mason sizeof(struct btrfs_item), 2009db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2010db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2011db94535dSChris Mason KM_USER1); 2012db94535dSChris Mason } 2013db94535dSChris Mason 20145f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 20155f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + size_diff); 2016b18c6685SChris Mason } 2017db94535dSChris Mason 2018db94535dSChris Mason if (leaf->map_token) { 2019db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2020db94535dSChris Mason leaf->map_token = NULL; 2021db94535dSChris Mason } 2022db94535dSChris Mason 2023b18c6685SChris Mason /* shift the data */ 20245f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2025b18c6685SChris Mason data_end + size_diff, btrfs_leaf_data(leaf) + 2026b18c6685SChris Mason data_end, old_data_start + new_size - data_end); 20275f39d397SChris Mason 20285f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 20295f39d397SChris Mason btrfs_set_item_size(leaf, item, new_size); 20305f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2031b18c6685SChris Mason 2032b18c6685SChris Mason ret = 0; 20335f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 20345f39d397SChris Mason btrfs_print_leaf(root, leaf); 2035b18c6685SChris Mason BUG(); 20365f39d397SChris Mason } 2037b18c6685SChris Mason return ret; 2038b18c6685SChris Mason } 2039b18c6685SChris Mason 20405f39d397SChris Mason int btrfs_extend_item(struct btrfs_trans_handle *trans, 20415f39d397SChris Mason struct btrfs_root *root, struct btrfs_path *path, 20425f39d397SChris Mason u32 data_size) 20436567e837SChris Mason { 20446567e837SChris Mason int ret = 0; 20456567e837SChris Mason int slot; 20466567e837SChris Mason int slot_orig; 20475f39d397SChris Mason struct extent_buffer *leaf; 20485f39d397SChris Mason struct btrfs_item *item; 20496567e837SChris Mason u32 nritems; 20506567e837SChris Mason unsigned int data_end; 20516567e837SChris Mason unsigned int old_data; 20526567e837SChris Mason unsigned int old_size; 20536567e837SChris Mason int i; 20546567e837SChris Mason 20556567e837SChris Mason slot_orig = path->slots[0]; 20565f39d397SChris Mason leaf = path->nodes[0]; 20576567e837SChris Mason 20585f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 20596567e837SChris Mason data_end = leaf_data_end(root, leaf); 20606567e837SChris Mason 20615f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < data_size) { 20625f39d397SChris Mason btrfs_print_leaf(root, leaf); 20636567e837SChris Mason BUG(); 20645f39d397SChris Mason } 20656567e837SChris Mason slot = path->slots[0]; 20665f39d397SChris Mason old_data = btrfs_item_end_nr(leaf, slot); 20676567e837SChris Mason 20686567e837SChris Mason BUG_ON(slot < 0); 20693326d1b0SChris Mason if (slot >= nritems) { 20703326d1b0SChris Mason btrfs_print_leaf(root, leaf); 20713326d1b0SChris Mason printk("slot %d too large, nritems %d\n", slot, nritems); 20723326d1b0SChris Mason BUG_ON(1); 20733326d1b0SChris Mason } 20746567e837SChris Mason 20756567e837SChris Mason /* 20766567e837SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 20776567e837SChris Mason */ 20786567e837SChris Mason /* first correct the data pointers */ 20796567e837SChris Mason for (i = slot; i < nritems; i++) { 20805f39d397SChris Mason u32 ioff; 20815f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2082db94535dSChris Mason 2083db94535dSChris Mason if (!leaf->map_token) { 2084db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2085db94535dSChris Mason sizeof(struct btrfs_item), 2086db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2087db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2088db94535dSChris Mason KM_USER1); 2089db94535dSChris Mason } 20905f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 20915f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff - data_size); 20926567e837SChris Mason } 20935f39d397SChris Mason 2094db94535dSChris Mason if (leaf->map_token) { 2095db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2096db94535dSChris Mason leaf->map_token = NULL; 2097db94535dSChris Mason } 2098db94535dSChris Mason 20996567e837SChris Mason /* shift the data */ 21005f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 21016567e837SChris Mason data_end - data_size, btrfs_leaf_data(leaf) + 21026567e837SChris Mason data_end, old_data - data_end); 21035f39d397SChris Mason 21046567e837SChris Mason data_end = old_data; 21055f39d397SChris Mason old_size = btrfs_item_size_nr(leaf, slot); 21065f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 21075f39d397SChris Mason btrfs_set_item_size(leaf, item, old_size + data_size); 21085f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 21096567e837SChris Mason 21106567e837SChris Mason ret = 0; 21115f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 21125f39d397SChris Mason btrfs_print_leaf(root, leaf); 21136567e837SChris Mason BUG(); 21145f39d397SChris Mason } 21156567e837SChris Mason return ret; 21166567e837SChris Mason } 21176567e837SChris Mason 211874123bd7SChris Mason /* 211974123bd7SChris Mason * Given a key and some data, insert an item into the tree. 212074123bd7SChris Mason * This does all the path init required, making room in the tree if needed. 212174123bd7SChris Mason */ 21225f39d397SChris Mason int btrfs_insert_empty_item(struct btrfs_trans_handle *trans, 21235f39d397SChris Mason struct btrfs_root *root, 21245f39d397SChris Mason struct btrfs_path *path, 21255f39d397SChris Mason struct btrfs_key *cpu_key, u32 data_size) 2126be0e5c09SChris Mason { 21275f39d397SChris Mason struct extent_buffer *leaf; 21285f39d397SChris Mason struct btrfs_item *item; 2129aa5d6bedSChris Mason int ret = 0; 2130be0e5c09SChris Mason int slot; 2131eb60ceacSChris Mason int slot_orig; 21327518a238SChris Mason u32 nritems; 2133be0e5c09SChris Mason unsigned int data_end; 2134e2fa7227SChris Mason struct btrfs_disk_key disk_key; 2135e2fa7227SChris Mason 2136e2fa7227SChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key); 2137be0e5c09SChris Mason 213874123bd7SChris Mason /* create a root if there isn't one */ 21395c680ed6SChris Mason if (!root->node) 2140cfaa7295SChris Mason BUG(); 21415f39d397SChris Mason 2142e089f05cSChris Mason ret = btrfs_search_slot(trans, root, cpu_key, path, data_size, 1); 2143eb60ceacSChris Mason if (ret == 0) { 2144f0930a37SChris Mason return -EEXIST; 2145aa5d6bedSChris Mason } 2146ed2ff2cbSChris Mason if (ret < 0) 2147ed2ff2cbSChris Mason goto out; 2148be0e5c09SChris Mason 214962e2749eSChris Mason slot_orig = path->slots[0]; 21505f39d397SChris Mason leaf = path->nodes[0]; 215174123bd7SChris Mason 21525f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2153123abc88SChris Mason data_end = leaf_data_end(root, leaf); 2154eb60ceacSChris Mason 2155123abc88SChris Mason if (btrfs_leaf_free_space(root, leaf) < 2156d4dbff95SChris Mason sizeof(struct btrfs_item) + data_size) { 21573326d1b0SChris Mason btrfs_print_leaf(root, leaf); 21583326d1b0SChris Mason printk("not enough freespace need %u have %d\n", 21593326d1b0SChris Mason data_size, btrfs_leaf_free_space(root, leaf)); 2160be0e5c09SChris Mason BUG(); 2161d4dbff95SChris Mason } 21625f39d397SChris Mason 216362e2749eSChris Mason slot = path->slots[0]; 2164eb60ceacSChris Mason BUG_ON(slot < 0); 21655f39d397SChris Mason 2166be0e5c09SChris Mason if (slot != nritems) { 2167be0e5c09SChris Mason int i; 21685f39d397SChris Mason unsigned int old_data = btrfs_item_end_nr(leaf, slot); 2169be0e5c09SChris Mason 21705f39d397SChris Mason if (old_data < data_end) { 21715f39d397SChris Mason btrfs_print_leaf(root, leaf); 21725f39d397SChris Mason printk("slot %d old_data %d data_end %d\n", 21735f39d397SChris Mason slot, old_data, data_end); 21745f39d397SChris Mason BUG_ON(1); 21755f39d397SChris Mason } 2176be0e5c09SChris Mason /* 2177be0e5c09SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 2178be0e5c09SChris Mason */ 2179be0e5c09SChris Mason /* first correct the data pointers */ 2180db94535dSChris Mason WARN_ON(leaf->map_token); 21810783fcfcSChris Mason for (i = slot; i < nritems; i++) { 21825f39d397SChris Mason u32 ioff; 2183db94535dSChris Mason 21845f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2185db94535dSChris Mason if (!leaf->map_token) { 2186db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2187db94535dSChris Mason sizeof(struct btrfs_item), 2188db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2189db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2190db94535dSChris Mason KM_USER1); 2191db94535dSChris Mason } 2192db94535dSChris Mason 21935f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 21945f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff - data_size); 21950783fcfcSChris Mason } 2196db94535dSChris Mason if (leaf->map_token) { 2197db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2198db94535dSChris Mason leaf->map_token = NULL; 2199db94535dSChris Mason } 2200be0e5c09SChris Mason 2201be0e5c09SChris Mason /* shift the items */ 22025f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1), 22035f39d397SChris Mason btrfs_item_nr_offset(slot), 22040783fcfcSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 2205be0e5c09SChris Mason 2206be0e5c09SChris Mason /* shift the data */ 22075f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2208d6025579SChris Mason data_end - data_size, btrfs_leaf_data(leaf) + 2209be0e5c09SChris Mason data_end, old_data - data_end); 2210be0e5c09SChris Mason data_end = old_data; 2211be0e5c09SChris Mason } 22125f39d397SChris Mason 221362e2749eSChris Mason /* setup the item for the new data */ 22145f39d397SChris Mason btrfs_set_item_key(leaf, &disk_key, slot); 22155f39d397SChris Mason item = btrfs_item_nr(leaf, slot); 22165f39d397SChris Mason btrfs_set_item_offset(leaf, item, data_end - data_size); 22175f39d397SChris Mason btrfs_set_item_size(leaf, item, data_size); 22185f39d397SChris Mason btrfs_set_header_nritems(leaf, nritems + 1); 22195f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2220aa5d6bedSChris Mason 2221aa5d6bedSChris Mason ret = 0; 22228e19f2cdSChris Mason if (slot == 0) 2223e089f05cSChris Mason ret = fixup_low_keys(trans, root, path, &disk_key, 1); 2224aa5d6bedSChris Mason 22255f39d397SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) { 22265f39d397SChris Mason btrfs_print_leaf(root, leaf); 2227be0e5c09SChris Mason BUG(); 22285f39d397SChris Mason } 2229ed2ff2cbSChris Mason out: 223062e2749eSChris Mason return ret; 223162e2749eSChris Mason } 223262e2749eSChris Mason 223362e2749eSChris Mason /* 223462e2749eSChris Mason * Given a key and some data, insert an item into the tree. 223562e2749eSChris Mason * This does all the path init required, making room in the tree if needed. 223662e2749eSChris Mason */ 2237e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root 2238e089f05cSChris Mason *root, struct btrfs_key *cpu_key, void *data, u32 2239e089f05cSChris Mason data_size) 224062e2749eSChris Mason { 224162e2749eSChris Mason int ret = 0; 22422c90e5d6SChris Mason struct btrfs_path *path; 22435f39d397SChris Mason struct extent_buffer *leaf; 22445f39d397SChris Mason unsigned long ptr; 224562e2749eSChris Mason 22462c90e5d6SChris Mason path = btrfs_alloc_path(); 22472c90e5d6SChris Mason BUG_ON(!path); 22482c90e5d6SChris Mason ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size); 224962e2749eSChris Mason if (!ret) { 22505f39d397SChris Mason leaf = path->nodes[0]; 22515f39d397SChris Mason ptr = btrfs_item_ptr_offset(leaf, path->slots[0]); 22525f39d397SChris Mason write_extent_buffer(leaf, data, ptr, data_size); 22535f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 225462e2749eSChris Mason } 22552c90e5d6SChris Mason btrfs_free_path(path); 2256aa5d6bedSChris Mason return ret; 2257be0e5c09SChris Mason } 2258be0e5c09SChris Mason 225974123bd7SChris Mason /* 22605de08d7dSChris Mason * delete the pointer from a given node. 226174123bd7SChris Mason * 226274123bd7SChris Mason * If the delete empties a node, the node is removed from the tree, 226374123bd7SChris Mason * continuing all the way the root if required. The root is converted into 226474123bd7SChris Mason * a leaf if all the nodes are emptied. 226574123bd7SChris Mason */ 2266e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 2267e089f05cSChris Mason struct btrfs_path *path, int level, int slot) 2268be0e5c09SChris Mason { 22695f39d397SChris Mason struct extent_buffer *parent = path->nodes[level]; 22707518a238SChris Mason u32 nritems; 2271aa5d6bedSChris Mason int ret = 0; 2272bb803951SChris Mason int wret; 2273be0e5c09SChris Mason 22745f39d397SChris Mason nritems = btrfs_header_nritems(parent); 2275be0e5c09SChris Mason if (slot != nritems -1) { 22765f39d397SChris Mason memmove_extent_buffer(parent, 22775f39d397SChris Mason btrfs_node_key_ptr_offset(slot), 22785f39d397SChris Mason btrfs_node_key_ptr_offset(slot + 1), 2279d6025579SChris Mason sizeof(struct btrfs_key_ptr) * 2280d6025579SChris Mason (nritems - slot - 1)); 2281be0e5c09SChris Mason } 22827518a238SChris Mason nritems--; 22835f39d397SChris Mason btrfs_set_header_nritems(parent, nritems); 22847518a238SChris Mason if (nritems == 0 && parent == root->node) { 22855f39d397SChris Mason BUG_ON(btrfs_header_level(root->node) != 1); 2286eb60ceacSChris Mason /* just turn the root into a leaf and break */ 22875f39d397SChris Mason btrfs_set_header_level(root->node, 0); 2288bb803951SChris Mason } else if (slot == 0) { 22895f39d397SChris Mason struct btrfs_disk_key disk_key; 22905f39d397SChris Mason 22915f39d397SChris Mason btrfs_node_key(parent, &disk_key, 0); 22925f39d397SChris Mason wret = fixup_low_keys(trans, root, path, &disk_key, level + 1); 22930f70abe2SChris Mason if (wret) 22940f70abe2SChris Mason ret = wret; 2295be0e5c09SChris Mason } 2296d6025579SChris Mason btrfs_mark_buffer_dirty(parent); 2297aa5d6bedSChris Mason return ret; 2298be0e5c09SChris Mason } 2299be0e5c09SChris Mason 230074123bd7SChris Mason /* 230174123bd7SChris Mason * delete the item at the leaf level in path. If that empties 230274123bd7SChris Mason * the leaf, remove it from the tree 230374123bd7SChris Mason */ 2304e089f05cSChris Mason int btrfs_del_item(struct btrfs_trans_handle *trans, struct btrfs_root *root, 2305e089f05cSChris Mason struct btrfs_path *path) 2306be0e5c09SChris Mason { 2307be0e5c09SChris Mason int slot; 23085f39d397SChris Mason struct extent_buffer *leaf; 23095f39d397SChris Mason struct btrfs_item *item; 2310be0e5c09SChris Mason int doff; 2311be0e5c09SChris Mason int dsize; 2312aa5d6bedSChris Mason int ret = 0; 2313aa5d6bedSChris Mason int wret; 23147518a238SChris Mason u32 nritems; 2315be0e5c09SChris Mason 23165f39d397SChris Mason leaf = path->nodes[0]; 23174920c9acSChris Mason slot = path->slots[0]; 23185f39d397SChris Mason doff = btrfs_item_offset_nr(leaf, slot); 23195f39d397SChris Mason dsize = btrfs_item_size_nr(leaf, slot); 23205f39d397SChris Mason nritems = btrfs_header_nritems(leaf); 2321be0e5c09SChris Mason 23227518a238SChris Mason if (slot != nritems - 1) { 2323be0e5c09SChris Mason int i; 2324123abc88SChris Mason int data_end = leaf_data_end(root, leaf); 23255f39d397SChris Mason 23265f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) + 2327d6025579SChris Mason data_end + dsize, 2328123abc88SChris Mason btrfs_leaf_data(leaf) + data_end, 2329be0e5c09SChris Mason doff - data_end); 23305f39d397SChris Mason 23310783fcfcSChris Mason for (i = slot + 1; i < nritems; i++) { 23325f39d397SChris Mason u32 ioff; 2333db94535dSChris Mason 23345f39d397SChris Mason item = btrfs_item_nr(leaf, i); 2335db94535dSChris Mason if (!leaf->map_token) { 2336db94535dSChris Mason map_extent_buffer(leaf, (unsigned long)item, 2337db94535dSChris Mason sizeof(struct btrfs_item), 2338db94535dSChris Mason &leaf->map_token, &leaf->kaddr, 2339db94535dSChris Mason &leaf->map_start, &leaf->map_len, 2340db94535dSChris Mason KM_USER1); 2341db94535dSChris Mason } 23425f39d397SChris Mason ioff = btrfs_item_offset(leaf, item); 23435f39d397SChris Mason btrfs_set_item_offset(leaf, item, ioff + dsize); 23440783fcfcSChris Mason } 2345db94535dSChris Mason 2346db94535dSChris Mason if (leaf->map_token) { 2347db94535dSChris Mason unmap_extent_buffer(leaf, leaf->map_token, KM_USER1); 2348db94535dSChris Mason leaf->map_token = NULL; 2349db94535dSChris Mason } 2350db94535dSChris Mason 23515f39d397SChris Mason memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot), 23525f39d397SChris Mason btrfs_item_nr_offset(slot + 1), 23530783fcfcSChris Mason sizeof(struct btrfs_item) * 23547518a238SChris Mason (nritems - slot - 1)); 2355be0e5c09SChris Mason } 23565f39d397SChris Mason btrfs_set_header_nritems(leaf, nritems - 1); 23577518a238SChris Mason nritems--; 23585f39d397SChris Mason 235974123bd7SChris Mason /* delete the leaf if we've emptied it */ 23607518a238SChris Mason if (nritems == 0) { 23615f39d397SChris Mason if (leaf == root->node) { 23625f39d397SChris Mason btrfs_set_header_level(leaf, 0); 23639a8dd150SChris Mason } else { 23645f39d397SChris Mason clean_tree_block(trans, root, leaf); 23655f39d397SChris Mason wait_on_tree_block_writeback(root, leaf); 2366e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, path->slots[1]); 2367aa5d6bedSChris Mason if (wret) 2368aa5d6bedSChris Mason ret = wret; 2369e089f05cSChris Mason wret = btrfs_free_extent(trans, root, 2370db94535dSChris Mason leaf->start, leaf->len, 1); 23710f70abe2SChris Mason if (wret) 23720f70abe2SChris Mason ret = wret; 23739a8dd150SChris Mason } 2374be0e5c09SChris Mason } else { 23757518a238SChris Mason int used = leaf_space_used(leaf, 0, nritems); 2376aa5d6bedSChris Mason if (slot == 0) { 23775f39d397SChris Mason struct btrfs_disk_key disk_key; 23785f39d397SChris Mason 23795f39d397SChris Mason btrfs_item_key(leaf, &disk_key, 0); 2380e089f05cSChris Mason wret = fixup_low_keys(trans, root, path, 23815f39d397SChris Mason &disk_key, 1); 2382aa5d6bedSChris Mason if (wret) 2383aa5d6bedSChris Mason ret = wret; 2384aa5d6bedSChris Mason } 2385aa5d6bedSChris Mason 238674123bd7SChris Mason /* delete the leaf if it is mostly empty */ 23877936ca38SChris Mason if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) { 2388be0e5c09SChris Mason /* push_leaf_left fixes the path. 2389be0e5c09SChris Mason * make sure the path still points to our leaf 2390be0e5c09SChris Mason * for possible call to del_ptr below 2391be0e5c09SChris Mason */ 23924920c9acSChris Mason slot = path->slots[1]; 23935f39d397SChris Mason extent_buffer_get(leaf); 23945f39d397SChris Mason 2395*3685f791SChris Mason wret = push_leaf_right(trans, root, path, 1); 239654aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2397aa5d6bedSChris Mason ret = wret; 23985f39d397SChris Mason 23995f39d397SChris Mason if (path->nodes[0] == leaf && 24005f39d397SChris Mason btrfs_header_nritems(leaf)) { 2401*3685f791SChris Mason wret = push_leaf_left(trans, root, path, 1); 240254aa1f4dSChris Mason if (wret < 0 && wret != -ENOSPC) 2403aa5d6bedSChris Mason ret = wret; 2404aa5d6bedSChris Mason } 24055f39d397SChris Mason 24065f39d397SChris Mason if (btrfs_header_nritems(leaf) == 0) { 2407db94535dSChris Mason u64 bytenr = leaf->start; 2408db94535dSChris Mason u32 blocksize = leaf->len; 24095f39d397SChris Mason 24105f39d397SChris Mason clean_tree_block(trans, root, leaf); 24115f39d397SChris Mason wait_on_tree_block_writeback(root, leaf); 24125f39d397SChris Mason 2413e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, slot); 2414aa5d6bedSChris Mason if (wret) 2415aa5d6bedSChris Mason ret = wret; 24165f39d397SChris Mason 24175f39d397SChris Mason free_extent_buffer(leaf); 2418db94535dSChris Mason wret = btrfs_free_extent(trans, root, bytenr, 2419db94535dSChris Mason blocksize, 1); 24200f70abe2SChris Mason if (wret) 24210f70abe2SChris Mason ret = wret; 24225de08d7dSChris Mason } else { 24235f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 24245f39d397SChris Mason free_extent_buffer(leaf); 2425be0e5c09SChris Mason } 2426d5719762SChris Mason } else { 24275f39d397SChris Mason btrfs_mark_buffer_dirty(leaf); 2428be0e5c09SChris Mason } 24299a8dd150SChris Mason } 2430aa5d6bedSChris Mason return ret; 24319a8dd150SChris Mason } 24329a8dd150SChris Mason 243397571fd0SChris Mason /* 243497571fd0SChris Mason * walk up the tree as far as required to find the next leaf. 24350f70abe2SChris Mason * returns 0 if it found something or 1 if there are no greater leaves. 24360f70abe2SChris Mason * returns < 0 on io errors. 243797571fd0SChris Mason */ 2438234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path) 2439d97e63b6SChris Mason { 2440d97e63b6SChris Mason int slot; 2441d97e63b6SChris Mason int level = 1; 2442db94535dSChris Mason u64 bytenr; 24435f39d397SChris Mason struct extent_buffer *c; 24445f39d397SChris Mason struct extent_buffer *next = NULL; 2445d97e63b6SChris Mason 2446234b63a0SChris Mason while(level < BTRFS_MAX_LEVEL) { 2447d97e63b6SChris Mason if (!path->nodes[level]) 24480f70abe2SChris Mason return 1; 24495f39d397SChris Mason 2450d97e63b6SChris Mason slot = path->slots[level] + 1; 2451d97e63b6SChris Mason c = path->nodes[level]; 24525f39d397SChris Mason if (slot >= btrfs_header_nritems(c)) { 2453d97e63b6SChris Mason level++; 2454d97e63b6SChris Mason continue; 2455d97e63b6SChris Mason } 24565f39d397SChris Mason 2457db94535dSChris Mason bytenr = btrfs_node_blockptr(c, slot); 2458cfaa7295SChris Mason if (next) 24595f39d397SChris Mason free_extent_buffer(next); 24605f39d397SChris Mason 24616702ed49SChris Mason if (path->reada) 24626702ed49SChris Mason reada_for_search(root, path, level, slot); 24635f39d397SChris Mason 2464db94535dSChris Mason next = read_tree_block(root, bytenr, 2465db94535dSChris Mason btrfs_level_size(root, level -1)); 2466d97e63b6SChris Mason break; 2467d97e63b6SChris Mason } 2468d97e63b6SChris Mason path->slots[level] = slot; 2469d97e63b6SChris Mason while(1) { 2470d97e63b6SChris Mason level--; 2471d97e63b6SChris Mason c = path->nodes[level]; 24725f39d397SChris Mason free_extent_buffer(c); 2473d97e63b6SChris Mason path->nodes[level] = next; 2474d97e63b6SChris Mason path->slots[level] = 0; 2475d97e63b6SChris Mason if (!level) 2476d97e63b6SChris Mason break; 24776702ed49SChris Mason if (path->reada) 247832020611SYan reada_for_search(root, path, level, 0); 2479db94535dSChris Mason next = read_tree_block(root, btrfs_node_blockptr(next, 0), 2480db94535dSChris Mason btrfs_level_size(root, level - 1)); 2481d97e63b6SChris Mason } 2482d97e63b6SChris Mason return 0; 2483d97e63b6SChris Mason } 2484