12e635a27SChris Mason #include <linux/module.h> 2eb60ceacSChris Mason #include "ctree.h" 3eb60ceacSChris Mason #include "disk-io.h" 47f5c1516SChris Mason #include "transaction.h" 59a8dd150SChris Mason 6e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 7e089f05cSChris Mason *root, struct btrfs_path *path, int level); 8e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 9e089f05cSChris Mason *root, struct btrfs_path *path, int data_size); 10e089f05cSChris Mason static int push_node_left(struct btrfs_trans_handle *trans, struct btrfs_root 11e20d96d6SChris Mason *root, struct buffer_head *dst, struct buffer_head 12e089f05cSChris Mason *src); 13e089f05cSChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, struct 14e20d96d6SChris Mason btrfs_root *root, struct buffer_head *dst_buf, 15e20d96d6SChris Mason struct buffer_head *src_buf); 16e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 17e089f05cSChris Mason struct btrfs_path *path, int level, int slot); 18d97e63b6SChris Mason 19234b63a0SChris Mason inline void btrfs_init_path(struct btrfs_path *p) 20be0e5c09SChris Mason { 21be0e5c09SChris Mason memset(p, 0, sizeof(*p)); 22be0e5c09SChris Mason } 23be0e5c09SChris Mason 24234b63a0SChris Mason void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p) 25eb60ceacSChris Mason { 26eb60ceacSChris Mason int i; 27234b63a0SChris Mason for (i = 0; i < BTRFS_MAX_LEVEL; i++) { 28eb60ceacSChris Mason if (!p->nodes[i]) 29eb60ceacSChris Mason break; 30234b63a0SChris Mason btrfs_block_release(root, p->nodes[i]); 31eb60ceacSChris Mason } 32aa5d6bedSChris Mason memset(p, 0, sizeof(*p)); 33eb60ceacSChris Mason } 34eb60ceacSChris Mason 35e089f05cSChris Mason static int btrfs_cow_block(struct btrfs_trans_handle *trans, struct btrfs_root 36e20d96d6SChris Mason *root, struct buffer_head *buf, struct buffer_head 37e20d96d6SChris Mason *parent, int parent_slot, struct buffer_head 38e089f05cSChris Mason **cow_ret) 3902217ed2SChris Mason { 40e20d96d6SChris Mason struct buffer_head *cow; 41e20d96d6SChris Mason struct btrfs_node *cow_node; 4202217ed2SChris Mason 437f5c1516SChris Mason if (btrfs_header_generation(btrfs_buffer_header(buf)) == 447f5c1516SChris Mason trans->transid) { 4502217ed2SChris Mason *cow_ret = buf; 4602217ed2SChris Mason return 0; 4702217ed2SChris Mason } 48e089f05cSChris Mason cow = btrfs_alloc_free_block(trans, root); 49e20d96d6SChris Mason cow_node = btrfs_buffer_node(cow); 50e20d96d6SChris Mason memcpy(cow_node, btrfs_buffer_node(buf), root->blocksize); 51e20d96d6SChris Mason btrfs_set_header_blocknr(&cow_node->header, cow->b_blocknr); 527f5c1516SChris Mason btrfs_set_header_generation(&cow_node->header, trans->transid); 5302217ed2SChris Mason *cow_ret = cow; 54d5719762SChris Mason mark_buffer_dirty(cow); 55e089f05cSChris Mason btrfs_inc_ref(trans, root, buf); 5602217ed2SChris Mason if (buf == root->node) { 5702217ed2SChris Mason root->node = cow; 58e20d96d6SChris Mason get_bh(cow); 59a28ec197SChris Mason if (buf != root->commit_root) 60e20d96d6SChris Mason btrfs_free_extent(trans, root, buf->b_blocknr, 1, 1); 61234b63a0SChris Mason btrfs_block_release(root, buf); 6202217ed2SChris Mason } else { 63e20d96d6SChris Mason btrfs_set_node_blockptr(btrfs_buffer_node(parent), parent_slot, 64e20d96d6SChris Mason cow->b_blocknr); 65d5719762SChris Mason mark_buffer_dirty(parent); 66e20d96d6SChris Mason btrfs_free_extent(trans, root, buf->b_blocknr, 1, 1); 6702217ed2SChris Mason } 68234b63a0SChris Mason btrfs_block_release(root, buf); 6902217ed2SChris Mason return 0; 7002217ed2SChris Mason } 7102217ed2SChris Mason 7274123bd7SChris Mason /* 7374123bd7SChris Mason * The leaf data grows from end-to-front in the node. 7474123bd7SChris Mason * this returns the address of the start of the last item, 7574123bd7SChris Mason * which is the stop of the leaf data stack 7674123bd7SChris Mason */ 77123abc88SChris Mason static inline unsigned int leaf_data_end(struct btrfs_root *root, 78123abc88SChris Mason struct btrfs_leaf *leaf) 79be0e5c09SChris Mason { 807518a238SChris Mason u32 nr = btrfs_header_nritems(&leaf->header); 81be0e5c09SChris Mason if (nr == 0) 82123abc88SChris Mason return BTRFS_LEAF_DATA_SIZE(root); 830783fcfcSChris Mason return btrfs_item_offset(leaf->items + nr - 1); 84be0e5c09SChris Mason } 85be0e5c09SChris Mason 8674123bd7SChris Mason /* 8774123bd7SChris Mason * The space between the end of the leaf items and 8874123bd7SChris Mason * the start of the leaf data. IOW, how much room 8974123bd7SChris Mason * the leaf has left for both items and data 9074123bd7SChris Mason */ 91123abc88SChris Mason int btrfs_leaf_free_space(struct btrfs_root *root, struct btrfs_leaf *leaf) 92be0e5c09SChris Mason { 93123abc88SChris Mason int data_end = leaf_data_end(root, leaf); 947518a238SChris Mason int nritems = btrfs_header_nritems(&leaf->header); 95be0e5c09SChris Mason char *items_end = (char *)(leaf->items + nritems + 1); 96123abc88SChris Mason return (char *)(btrfs_leaf_data(leaf) + data_end) - (char *)items_end; 97be0e5c09SChris Mason } 98be0e5c09SChris Mason 9974123bd7SChris Mason /* 10074123bd7SChris Mason * compare two keys in a memcmp fashion 10174123bd7SChris Mason */ 1029aca1d51SChris Mason static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2) 103be0e5c09SChris Mason { 104e2fa7227SChris Mason struct btrfs_key k1; 105e2fa7227SChris Mason 106e2fa7227SChris Mason btrfs_disk_key_to_cpu(&k1, disk); 107e2fa7227SChris Mason 108e2fa7227SChris Mason if (k1.objectid > k2->objectid) 109be0e5c09SChris Mason return 1; 110e2fa7227SChris Mason if (k1.objectid < k2->objectid) 111be0e5c09SChris Mason return -1; 112a8a2ee0cSChris Mason if (k1.offset > k2->offset) 113a8a2ee0cSChris Mason return 1; 114a8a2ee0cSChris Mason if (k1.offset < k2->offset) 115a8a2ee0cSChris Mason return -1; 116f254e52cSChris Mason if (k1.flags > k2->flags) 117f254e52cSChris Mason return 1; 118f254e52cSChris Mason if (k1.flags < k2->flags) 119f254e52cSChris Mason return -1; 120be0e5c09SChris Mason return 0; 121be0e5c09SChris Mason } 12274123bd7SChris Mason 123123abc88SChris Mason static int check_node(struct btrfs_root *root, struct btrfs_path *path, 124123abc88SChris Mason int level) 125aa5d6bedSChris Mason { 126aa5d6bedSChris Mason int i; 127234b63a0SChris Mason struct btrfs_node *parent = NULL; 128e20d96d6SChris Mason struct btrfs_node *node = btrfs_buffer_node(path->nodes[level]); 129aa5d6bedSChris Mason int parent_slot; 1307518a238SChris Mason u32 nritems = btrfs_header_nritems(&node->header); 131aa5d6bedSChris Mason 132aa5d6bedSChris Mason if (path->nodes[level + 1]) 133e20d96d6SChris Mason parent = btrfs_buffer_node(path->nodes[level + 1]); 134aa5d6bedSChris Mason parent_slot = path->slots[level + 1]; 1357518a238SChris Mason BUG_ON(nritems == 0); 1367518a238SChris Mason if (parent) { 137e2fa7227SChris Mason struct btrfs_disk_key *parent_key; 138123abc88SChris Mason parent_key = &parent->ptrs[parent_slot].key; 139123abc88SChris Mason BUG_ON(memcmp(parent_key, &node->ptrs[0].key, 140e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 1411d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 1427518a238SChris Mason btrfs_header_blocknr(&node->header)); 143aa5d6bedSChris Mason } 144123abc88SChris Mason BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root)); 1457518a238SChris Mason for (i = 0; nritems > 1 && i < nritems - 2; i++) { 146e2fa7227SChris Mason struct btrfs_key cpukey; 147123abc88SChris Mason btrfs_disk_key_to_cpu(&cpukey, &node->ptrs[i + 1].key); 148123abc88SChris Mason BUG_ON(comp_keys(&node->ptrs[i].key, &cpukey) >= 0); 149aa5d6bedSChris Mason } 150aa5d6bedSChris Mason return 0; 151aa5d6bedSChris Mason } 152aa5d6bedSChris Mason 153123abc88SChris Mason static int check_leaf(struct btrfs_root *root, struct btrfs_path *path, 154123abc88SChris Mason int level) 155aa5d6bedSChris Mason { 156aa5d6bedSChris Mason int i; 157e20d96d6SChris Mason struct btrfs_leaf *leaf = btrfs_buffer_leaf(path->nodes[level]); 158234b63a0SChris Mason struct btrfs_node *parent = NULL; 159aa5d6bedSChris Mason int parent_slot; 1607518a238SChris Mason u32 nritems = btrfs_header_nritems(&leaf->header); 161aa5d6bedSChris Mason 162aa5d6bedSChris Mason if (path->nodes[level + 1]) 163e20d96d6SChris Mason parent = btrfs_buffer_node(path->nodes[level + 1]); 164aa5d6bedSChris Mason parent_slot = path->slots[level + 1]; 165123abc88SChris Mason BUG_ON(btrfs_leaf_free_space(root, leaf) < 0); 1667518a238SChris Mason 1677518a238SChris Mason if (nritems == 0) 1687518a238SChris Mason return 0; 1697518a238SChris Mason 1707518a238SChris Mason if (parent) { 171e2fa7227SChris Mason struct btrfs_disk_key *parent_key; 172123abc88SChris Mason parent_key = &parent->ptrs[parent_slot].key; 173aa5d6bedSChris Mason BUG_ON(memcmp(parent_key, &leaf->items[0].key, 174e2fa7227SChris Mason sizeof(struct btrfs_disk_key))); 1751d4f8a0cSChris Mason BUG_ON(btrfs_node_blockptr(parent, parent_slot) != 1767518a238SChris Mason btrfs_header_blocknr(&leaf->header)); 177aa5d6bedSChris Mason } 1787518a238SChris Mason for (i = 0; nritems > 1 && i < nritems - 2; i++) { 179e2fa7227SChris Mason struct btrfs_key cpukey; 180e2fa7227SChris Mason btrfs_disk_key_to_cpu(&cpukey, &leaf->items[i + 1].key); 181aa5d6bedSChris Mason BUG_ON(comp_keys(&leaf->items[i].key, 182e2fa7227SChris Mason &cpukey) >= 0); 1830783fcfcSChris Mason BUG_ON(btrfs_item_offset(leaf->items + i) != 1840783fcfcSChris Mason btrfs_item_end(leaf->items + i + 1)); 185aa5d6bedSChris Mason if (i == 0) { 1860783fcfcSChris Mason BUG_ON(btrfs_item_offset(leaf->items + i) + 1870783fcfcSChris Mason btrfs_item_size(leaf->items + i) != 188123abc88SChris Mason BTRFS_LEAF_DATA_SIZE(root)); 189aa5d6bedSChris Mason } 190aa5d6bedSChris Mason } 191aa5d6bedSChris Mason return 0; 192aa5d6bedSChris Mason } 193aa5d6bedSChris Mason 194123abc88SChris Mason static int check_block(struct btrfs_root *root, struct btrfs_path *path, 195123abc88SChris Mason int level) 196aa5d6bedSChris Mason { 197aa5d6bedSChris Mason if (level == 0) 198123abc88SChris Mason return check_leaf(root, path, level); 199123abc88SChris Mason return check_node(root, path, level); 200aa5d6bedSChris Mason } 201aa5d6bedSChris Mason 20274123bd7SChris Mason /* 20374123bd7SChris Mason * search for key in the array p. items p are item_size apart 20474123bd7SChris Mason * and there are 'max' items in p 20574123bd7SChris Mason * the slot in the array is returned via slot, and it points to 20674123bd7SChris Mason * the place where you would insert key if it is not found in 20774123bd7SChris Mason * the array. 20874123bd7SChris Mason * 20974123bd7SChris Mason * slot may point to max if the key is bigger than all of the keys 21074123bd7SChris Mason */ 2119aca1d51SChris Mason static int generic_bin_search(char *p, int item_size, struct btrfs_key *key, 212be0e5c09SChris Mason int max, int *slot) 213be0e5c09SChris Mason { 214be0e5c09SChris Mason int low = 0; 215be0e5c09SChris Mason int high = max; 216be0e5c09SChris Mason int mid; 217be0e5c09SChris Mason int ret; 218e2fa7227SChris Mason struct btrfs_disk_key *tmp; 219be0e5c09SChris Mason 220be0e5c09SChris Mason while(low < high) { 221be0e5c09SChris Mason mid = (low + high) / 2; 222e2fa7227SChris Mason tmp = (struct btrfs_disk_key *)(p + mid * item_size); 223be0e5c09SChris Mason ret = comp_keys(tmp, key); 224be0e5c09SChris Mason 225be0e5c09SChris Mason if (ret < 0) 226be0e5c09SChris Mason low = mid + 1; 227be0e5c09SChris Mason else if (ret > 0) 228be0e5c09SChris Mason high = mid; 229be0e5c09SChris Mason else { 230be0e5c09SChris Mason *slot = mid; 231be0e5c09SChris Mason return 0; 232be0e5c09SChris Mason } 233be0e5c09SChris Mason } 234be0e5c09SChris Mason *slot = low; 235be0e5c09SChris Mason return 1; 236be0e5c09SChris Mason } 237be0e5c09SChris Mason 23897571fd0SChris Mason /* 23997571fd0SChris Mason * simple bin_search frontend that does the right thing for 24097571fd0SChris Mason * leaves vs nodes 24197571fd0SChris Mason */ 2429aca1d51SChris Mason static int bin_search(struct btrfs_node *c, struct btrfs_key *key, int *slot) 243be0e5c09SChris Mason { 2447518a238SChris Mason if (btrfs_is_leaf(c)) { 245234b63a0SChris Mason struct btrfs_leaf *l = (struct btrfs_leaf *)c; 2460783fcfcSChris Mason return generic_bin_search((void *)l->items, 2470783fcfcSChris Mason sizeof(struct btrfs_item), 2487518a238SChris Mason key, btrfs_header_nritems(&c->header), 2497518a238SChris Mason slot); 250be0e5c09SChris Mason } else { 251123abc88SChris Mason return generic_bin_search((void *)c->ptrs, 252123abc88SChris Mason sizeof(struct btrfs_key_ptr), 2537518a238SChris Mason key, btrfs_header_nritems(&c->header), 2547518a238SChris Mason slot); 255be0e5c09SChris Mason } 256be0e5c09SChris Mason return -1; 257be0e5c09SChris Mason } 258be0e5c09SChris Mason 259e20d96d6SChris Mason static struct buffer_head *read_node_slot(struct btrfs_root *root, 260e20d96d6SChris Mason struct buffer_head *parent_buf, 261bb803951SChris Mason int slot) 262bb803951SChris Mason { 263e20d96d6SChris Mason struct btrfs_node *node = btrfs_buffer_node(parent_buf); 264bb803951SChris Mason if (slot < 0) 265bb803951SChris Mason return NULL; 2667518a238SChris Mason if (slot >= btrfs_header_nritems(&node->header)) 267bb803951SChris Mason return NULL; 2681d4f8a0cSChris Mason return read_tree_block(root, btrfs_node_blockptr(node, slot)); 269bb803951SChris Mason } 270bb803951SChris Mason 271e089f05cSChris Mason static int balance_level(struct btrfs_trans_handle *trans, struct btrfs_root 272e089f05cSChris Mason *root, struct btrfs_path *path, int level) 273bb803951SChris Mason { 274e20d96d6SChris Mason struct buffer_head *right_buf; 275e20d96d6SChris Mason struct buffer_head *mid_buf; 276e20d96d6SChris Mason struct buffer_head *left_buf; 277e20d96d6SChris Mason struct buffer_head *parent_buf = NULL; 278234b63a0SChris Mason struct btrfs_node *right = NULL; 279234b63a0SChris Mason struct btrfs_node *mid; 280234b63a0SChris Mason struct btrfs_node *left = NULL; 281234b63a0SChris Mason struct btrfs_node *parent = NULL; 282bb803951SChris Mason int ret = 0; 283bb803951SChris Mason int wret; 284bb803951SChris Mason int pslot; 285bb803951SChris Mason int orig_slot = path->slots[level]; 28679f95c82SChris Mason u64 orig_ptr; 287bb803951SChris Mason 288bb803951SChris Mason if (level == 0) 289bb803951SChris Mason return 0; 290bb803951SChris Mason 291bb803951SChris Mason mid_buf = path->nodes[level]; 292e20d96d6SChris Mason mid = btrfs_buffer_node(mid_buf); 2931d4f8a0cSChris Mason orig_ptr = btrfs_node_blockptr(mid, orig_slot); 29479f95c82SChris Mason 295234b63a0SChris Mason if (level < BTRFS_MAX_LEVEL - 1) 296bb803951SChris Mason parent_buf = path->nodes[level + 1]; 297bb803951SChris Mason pslot = path->slots[level + 1]; 298bb803951SChris Mason 29940689478SChris Mason /* 30040689478SChris Mason * deal with the case where there is only one pointer in the root 30140689478SChris Mason * by promoting the node below to a root 30240689478SChris Mason */ 303bb803951SChris Mason if (!parent_buf) { 304e20d96d6SChris Mason struct buffer_head *child; 305e20d96d6SChris Mason u64 blocknr = mid_buf->b_blocknr; 306bb803951SChris Mason 3077518a238SChris Mason if (btrfs_header_nritems(&mid->header) != 1) 308bb803951SChris Mason return 0; 309bb803951SChris Mason 310bb803951SChris Mason /* promote the child to a root */ 311bb803951SChris Mason child = read_node_slot(root, mid_buf, 0); 312bb803951SChris Mason BUG_ON(!child); 313bb803951SChris Mason root->node = child; 314bb803951SChris Mason path->nodes[level] = NULL; 315bb803951SChris Mason /* once for the path */ 316234b63a0SChris Mason btrfs_block_release(root, mid_buf); 317bb803951SChris Mason /* once for the root ptr */ 318234b63a0SChris Mason btrfs_block_release(root, mid_buf); 319e089f05cSChris Mason clean_tree_block(trans, root, mid_buf); 320e089f05cSChris Mason return btrfs_free_extent(trans, root, blocknr, 1, 1); 321bb803951SChris Mason } 322e20d96d6SChris Mason parent = btrfs_buffer_node(parent_buf); 323bb803951SChris Mason 324123abc88SChris Mason if (btrfs_header_nritems(&mid->header) > 325123abc88SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root) / 4) 326bb803951SChris Mason return 0; 327bb803951SChris Mason 328bb803951SChris Mason left_buf = read_node_slot(root, parent_buf, pslot - 1); 329bb803951SChris Mason right_buf = read_node_slot(root, parent_buf, pslot + 1); 33079f95c82SChris Mason 33179f95c82SChris Mason /* first, try to make some room in the middle buffer */ 332bb803951SChris Mason if (left_buf) { 333e089f05cSChris Mason btrfs_cow_block(trans, root, left_buf, parent_buf, pslot - 1, 334e089f05cSChris Mason &left_buf); 335e20d96d6SChris Mason left = btrfs_buffer_node(left_buf); 3367518a238SChris Mason orig_slot += btrfs_header_nritems(&left->header); 337e089f05cSChris Mason wret = push_node_left(trans, root, left_buf, mid_buf); 33879f95c82SChris Mason if (wret < 0) 33979f95c82SChris Mason ret = wret; 340bb803951SChris Mason } 34179f95c82SChris Mason 34279f95c82SChris Mason /* 34379f95c82SChris Mason * then try to empty the right most buffer into the middle 34479f95c82SChris Mason */ 345bb803951SChris Mason if (right_buf) { 346e089f05cSChris Mason btrfs_cow_block(trans, root, right_buf, parent_buf, pslot + 1, 347e089f05cSChris Mason &right_buf); 348e20d96d6SChris Mason right = btrfs_buffer_node(right_buf); 349e089f05cSChris Mason wret = push_node_left(trans, root, mid_buf, right_buf); 35079f95c82SChris Mason if (wret < 0) 35179f95c82SChris Mason ret = wret; 3527518a238SChris Mason if (btrfs_header_nritems(&right->header) == 0) { 353e20d96d6SChris Mason u64 blocknr = right_buf->b_blocknr; 354234b63a0SChris Mason btrfs_block_release(root, right_buf); 355e089f05cSChris Mason clean_tree_block(trans, root, right_buf); 356bb803951SChris Mason right_buf = NULL; 357bb803951SChris Mason right = NULL; 358e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot + 359e089f05cSChris Mason 1); 360bb803951SChris Mason if (wret) 361bb803951SChris Mason ret = wret; 362e089f05cSChris Mason wret = btrfs_free_extent(trans, root, blocknr, 1, 1); 363bb803951SChris Mason if (wret) 364bb803951SChris Mason ret = wret; 365bb803951SChris Mason } else { 366123abc88SChris Mason memcpy(&parent->ptrs[pslot + 1].key, 367123abc88SChris Mason &right->ptrs[0].key, 368e2fa7227SChris Mason sizeof(struct btrfs_disk_key)); 369d5719762SChris Mason mark_buffer_dirty(parent_buf); 370bb803951SChris Mason } 371bb803951SChris Mason } 3727518a238SChris Mason if (btrfs_header_nritems(&mid->header) == 1) { 37379f95c82SChris Mason /* 37479f95c82SChris Mason * we're not allowed to leave a node with one item in the 37579f95c82SChris Mason * tree during a delete. A deletion from lower in the tree 37679f95c82SChris Mason * could try to delete the only pointer in this node. 37779f95c82SChris Mason * So, pull some keys from the left. 37879f95c82SChris Mason * There has to be a left pointer at this point because 37979f95c82SChris Mason * otherwise we would have pulled some pointers from the 38079f95c82SChris Mason * right 38179f95c82SChris Mason */ 38279f95c82SChris Mason BUG_ON(!left_buf); 383e089f05cSChris Mason wret = balance_node_right(trans, root, mid_buf, left_buf); 38479f95c82SChris Mason if (wret < 0) 38579f95c82SChris Mason ret = wret; 38679f95c82SChris Mason BUG_ON(wret == 1); 38779f95c82SChris Mason } 3887518a238SChris Mason if (btrfs_header_nritems(&mid->header) == 0) { 38979f95c82SChris Mason /* we've managed to empty the middle node, drop it */ 390e20d96d6SChris Mason u64 blocknr = mid_buf->b_blocknr; 391234b63a0SChris Mason btrfs_block_release(root, mid_buf); 392e089f05cSChris Mason clean_tree_block(trans, root, mid_buf); 393bb803951SChris Mason mid_buf = NULL; 394bb803951SChris Mason mid = NULL; 395e089f05cSChris Mason wret = del_ptr(trans, root, path, level + 1, pslot); 396bb803951SChris Mason if (wret) 397bb803951SChris Mason ret = wret; 398e089f05cSChris Mason wret = btrfs_free_extent(trans, root, blocknr, 1, 1); 399bb803951SChris Mason if (wret) 400bb803951SChris Mason ret = wret; 40179f95c82SChris Mason } else { 40279f95c82SChris Mason /* update the parent key to reflect our changes */ 403123abc88SChris Mason memcpy(&parent->ptrs[pslot].key, &mid->ptrs[0].key, 404e2fa7227SChris Mason sizeof(struct btrfs_disk_key)); 405d5719762SChris Mason mark_buffer_dirty(parent_buf); 40679f95c82SChris Mason } 407bb803951SChris Mason 40879f95c82SChris Mason /* update the path */ 409bb803951SChris Mason if (left_buf) { 4107518a238SChris Mason if (btrfs_header_nritems(&left->header) > orig_slot) { 411e20d96d6SChris Mason get_bh(left_buf); 412bb803951SChris Mason path->nodes[level] = left_buf; 413bb803951SChris Mason path->slots[level + 1] -= 1; 414bb803951SChris Mason path->slots[level] = orig_slot; 415bb803951SChris Mason if (mid_buf) 416234b63a0SChris Mason btrfs_block_release(root, mid_buf); 417bb803951SChris Mason } else { 4187518a238SChris Mason orig_slot -= btrfs_header_nritems(&left->header); 419bb803951SChris Mason path->slots[level] = orig_slot; 420bb803951SChris Mason } 421bb803951SChris Mason } 42279f95c82SChris Mason /* double check we haven't messed things up */ 423123abc88SChris Mason check_block(root, path, level); 424e20d96d6SChris Mason if (orig_ptr != 425e20d96d6SChris Mason btrfs_node_blockptr(btrfs_buffer_node(path->nodes[level]), 4261d4f8a0cSChris Mason path->slots[level])) 42779f95c82SChris Mason BUG(); 428bb803951SChris Mason 429bb803951SChris Mason if (right_buf) 430234b63a0SChris Mason btrfs_block_release(root, right_buf); 431bb803951SChris Mason if (left_buf) 432234b63a0SChris Mason btrfs_block_release(root, left_buf); 433bb803951SChris Mason return ret; 434bb803951SChris Mason } 435bb803951SChris Mason 43674123bd7SChris Mason /* 43774123bd7SChris Mason * look for key in the tree. path is filled in with nodes along the way 43874123bd7SChris Mason * if key is found, we return zero and you can find the item in the leaf 43974123bd7SChris Mason * level of the path (level 0) 44074123bd7SChris Mason * 44174123bd7SChris Mason * If the key isn't found, the path points to the slot where it should 442aa5d6bedSChris Mason * be inserted, and 1 is returned. If there are other errors during the 443aa5d6bedSChris Mason * search a negative error number is returned. 44497571fd0SChris Mason * 44597571fd0SChris Mason * if ins_len > 0, nodes and leaves will be split as we walk down the 44697571fd0SChris Mason * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if 44797571fd0SChris Mason * possible) 44874123bd7SChris Mason */ 449e089f05cSChris Mason int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root 450e089f05cSChris Mason *root, struct btrfs_key *key, struct btrfs_path *p, int 451e089f05cSChris Mason ins_len, int cow) 452be0e5c09SChris Mason { 453e20d96d6SChris Mason struct buffer_head *b; 454e20d96d6SChris Mason struct buffer_head *cow_buf; 455234b63a0SChris Mason struct btrfs_node *c; 456be0e5c09SChris Mason int slot; 457be0e5c09SChris Mason int ret; 458be0e5c09SChris Mason int level; 4595c680ed6SChris Mason 46022b0ebdaSChris Mason WARN_ON(p->nodes[0] != NULL); 46122b0ebdaSChris Mason WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex)); 462bb803951SChris Mason again: 463bb803951SChris Mason b = root->node; 464e20d96d6SChris Mason get_bh(b); 465eb60ceacSChris Mason while (b) { 466e20d96d6SChris Mason c = btrfs_buffer_node(b); 467e20d96d6SChris Mason level = btrfs_header_level(&c->header); 46802217ed2SChris Mason if (cow) { 46902217ed2SChris Mason int wret; 470e20d96d6SChris Mason wret = btrfs_cow_block(trans, root, b, 471e20d96d6SChris Mason p->nodes[level + 1], 472e20d96d6SChris Mason p->slots[level + 1], 473e089f05cSChris Mason &cow_buf); 47402217ed2SChris Mason b = cow_buf; 47502217ed2SChris Mason } 47602217ed2SChris Mason BUG_ON(!cow && ins_len); 477e20d96d6SChris Mason c = btrfs_buffer_node(b); 478eb60ceacSChris Mason p->nodes[level] = b; 479123abc88SChris Mason ret = check_block(root, p, level); 480aa5d6bedSChris Mason if (ret) 481aa5d6bedSChris Mason return -1; 482be0e5c09SChris Mason ret = bin_search(c, key, &slot); 4837518a238SChris Mason if (!btrfs_is_leaf(c)) { 484be0e5c09SChris Mason if (ret && slot > 0) 485be0e5c09SChris Mason slot -= 1; 486be0e5c09SChris Mason p->slots[level] = slot; 4877518a238SChris Mason if (ins_len > 0 && btrfs_header_nritems(&c->header) == 488123abc88SChris Mason BTRFS_NODEPTRS_PER_BLOCK(root)) { 489e089f05cSChris Mason int sret = split_node(trans, root, p, level); 4905c680ed6SChris Mason BUG_ON(sret > 0); 4915c680ed6SChris Mason if (sret) 4925c680ed6SChris Mason return sret; 4935c680ed6SChris Mason b = p->nodes[level]; 494e20d96d6SChris Mason c = btrfs_buffer_node(b); 4955c680ed6SChris Mason slot = p->slots[level]; 496bb803951SChris Mason } else if (ins_len < 0) { 497e089f05cSChris Mason int sret = balance_level(trans, root, p, 498e089f05cSChris Mason level); 499bb803951SChris Mason if (sret) 500bb803951SChris Mason return sret; 501bb803951SChris Mason b = p->nodes[level]; 502bb803951SChris Mason if (!b) 503bb803951SChris Mason goto again; 504e20d96d6SChris Mason c = btrfs_buffer_node(b); 505bb803951SChris Mason slot = p->slots[level]; 5067518a238SChris Mason BUG_ON(btrfs_header_nritems(&c->header) == 1); 5075c680ed6SChris Mason } 5081d4f8a0cSChris Mason b = read_tree_block(root, btrfs_node_blockptr(c, slot)); 509be0e5c09SChris Mason } else { 510234b63a0SChris Mason struct btrfs_leaf *l = (struct btrfs_leaf *)c; 511be0e5c09SChris Mason p->slots[level] = slot; 512123abc88SChris Mason if (ins_len > 0 && btrfs_leaf_free_space(root, l) < 5130783fcfcSChris Mason sizeof(struct btrfs_item) + ins_len) { 514e089f05cSChris Mason int sret = split_leaf(trans, root, p, ins_len); 5155c680ed6SChris Mason BUG_ON(sret > 0); 5165c680ed6SChris Mason if (sret) 5175c680ed6SChris Mason return sret; 5185c680ed6SChris Mason } 519be0e5c09SChris Mason return ret; 520be0e5c09SChris Mason } 521be0e5c09SChris Mason } 522aa5d6bedSChris Mason return 1; 523be0e5c09SChris Mason } 524be0e5c09SChris Mason 52574123bd7SChris Mason /* 52674123bd7SChris Mason * adjust the pointers going up the tree, starting at level 52774123bd7SChris Mason * making sure the right key of each node is points to 'key'. 52874123bd7SChris Mason * This is used after shifting pointers to the left, so it stops 52974123bd7SChris Mason * fixing up pointers when a given leaf/node is not in slot 0 of the 53074123bd7SChris Mason * higher levels 531aa5d6bedSChris Mason * 532aa5d6bedSChris Mason * If this fails to write a tree block, it returns -1, but continues 533aa5d6bedSChris Mason * fixing up the blocks in ram so the tree is consistent. 53474123bd7SChris Mason */ 535e089f05cSChris Mason static int fixup_low_keys(struct btrfs_trans_handle *trans, struct btrfs_root 536e089f05cSChris Mason *root, struct btrfs_path *path, struct btrfs_disk_key 537e089f05cSChris Mason *key, int level) 538be0e5c09SChris Mason { 539be0e5c09SChris Mason int i; 540aa5d6bedSChris Mason int ret = 0; 541234b63a0SChris Mason for (i = level; i < BTRFS_MAX_LEVEL; i++) { 542234b63a0SChris Mason struct btrfs_node *t; 543be0e5c09SChris Mason int tslot = path->slots[i]; 544eb60ceacSChris Mason if (!path->nodes[i]) 545be0e5c09SChris Mason break; 546e20d96d6SChris Mason t = btrfs_buffer_node(path->nodes[i]); 547123abc88SChris Mason memcpy(&t->ptrs[tslot].key, key, sizeof(*key)); 548d5719762SChris Mason mark_buffer_dirty(path->nodes[i]); 549be0e5c09SChris Mason if (tslot != 0) 550be0e5c09SChris Mason break; 551be0e5c09SChris Mason } 552aa5d6bedSChris Mason return ret; 553be0e5c09SChris Mason } 554be0e5c09SChris Mason 55574123bd7SChris Mason /* 55674123bd7SChris Mason * try to push data from one node into the next node left in the 55779f95c82SChris Mason * tree. 558aa5d6bedSChris Mason * 559aa5d6bedSChris Mason * returns 0 if some ptrs were pushed left, < 0 if there was some horrible 560aa5d6bedSChris Mason * error, and > 0 if there was no room in the left hand block. 56174123bd7SChris Mason */ 562e089f05cSChris Mason static int push_node_left(struct btrfs_trans_handle *trans, struct btrfs_root 563e20d96d6SChris Mason *root, struct buffer_head *dst_buf, struct 564e20d96d6SChris Mason buffer_head *src_buf) 565be0e5c09SChris Mason { 566e20d96d6SChris Mason struct btrfs_node *src = btrfs_buffer_node(src_buf); 567e20d96d6SChris Mason struct btrfs_node *dst = btrfs_buffer_node(dst_buf); 568be0e5c09SChris Mason int push_items = 0; 569bb803951SChris Mason int src_nritems; 570bb803951SChris Mason int dst_nritems; 571aa5d6bedSChris Mason int ret = 0; 572be0e5c09SChris Mason 5737518a238SChris Mason src_nritems = btrfs_header_nritems(&src->header); 5747518a238SChris Mason dst_nritems = btrfs_header_nritems(&dst->header); 575123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 576eb60ceacSChris Mason if (push_items <= 0) { 577be0e5c09SChris Mason return 1; 578eb60ceacSChris Mason } 579be0e5c09SChris Mason 580be0e5c09SChris Mason if (src_nritems < push_items) 581be0e5c09SChris Mason push_items = src_nritems; 58279f95c82SChris Mason 583123abc88SChris Mason memcpy(dst->ptrs + dst_nritems, src->ptrs, 584123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 585bb803951SChris Mason if (push_items < src_nritems) { 586123abc88SChris Mason memmove(src->ptrs, src->ptrs + push_items, 587e2fa7227SChris Mason (src_nritems - push_items) * 588123abc88SChris Mason sizeof(struct btrfs_key_ptr)); 589bb803951SChris Mason } 5907518a238SChris Mason btrfs_set_header_nritems(&src->header, src_nritems - push_items); 5917518a238SChris Mason btrfs_set_header_nritems(&dst->header, dst_nritems + push_items); 592d5719762SChris Mason mark_buffer_dirty(src_buf); 593d5719762SChris Mason mark_buffer_dirty(dst_buf); 594bb803951SChris Mason return ret; 595be0e5c09SChris Mason } 596be0e5c09SChris Mason 59797571fd0SChris Mason /* 59879f95c82SChris Mason * try to push data from one node into the next node right in the 59979f95c82SChris Mason * tree. 60079f95c82SChris Mason * 60179f95c82SChris Mason * returns 0 if some ptrs were pushed, < 0 if there was some horrible 60279f95c82SChris Mason * error, and > 0 if there was no room in the right hand block. 60379f95c82SChris Mason * 60479f95c82SChris Mason * this will only push up to 1/2 the contents of the left node over 60579f95c82SChris Mason */ 606e089f05cSChris Mason static int balance_node_right(struct btrfs_trans_handle *trans, struct 607e20d96d6SChris Mason btrfs_root *root, struct buffer_head *dst_buf, 608e20d96d6SChris Mason struct buffer_head *src_buf) 60979f95c82SChris Mason { 610e20d96d6SChris Mason struct btrfs_node *src = btrfs_buffer_node(src_buf); 611e20d96d6SChris Mason struct btrfs_node *dst = btrfs_buffer_node(dst_buf); 61279f95c82SChris Mason int push_items = 0; 61379f95c82SChris Mason int max_push; 61479f95c82SChris Mason int src_nritems; 61579f95c82SChris Mason int dst_nritems; 61679f95c82SChris Mason int ret = 0; 61779f95c82SChris Mason 6187518a238SChris Mason src_nritems = btrfs_header_nritems(&src->header); 6197518a238SChris Mason dst_nritems = btrfs_header_nritems(&dst->header); 620123abc88SChris Mason push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems; 62179f95c82SChris Mason if (push_items <= 0) { 62279f95c82SChris Mason return 1; 62379f95c82SChris Mason } 62479f95c82SChris Mason 62579f95c82SChris Mason max_push = src_nritems / 2 + 1; 62679f95c82SChris Mason /* don't try to empty the node */ 62779f95c82SChris Mason if (max_push > src_nritems) 62879f95c82SChris Mason return 1; 62979f95c82SChris Mason if (max_push < push_items) 63079f95c82SChris Mason push_items = max_push; 63179f95c82SChris Mason 632123abc88SChris Mason memmove(dst->ptrs + push_items, dst->ptrs, 633123abc88SChris Mason dst_nritems * sizeof(struct btrfs_key_ptr)); 634123abc88SChris Mason memcpy(dst->ptrs, src->ptrs + src_nritems - push_items, 635123abc88SChris Mason push_items * sizeof(struct btrfs_key_ptr)); 63679f95c82SChris Mason 6377518a238SChris Mason btrfs_set_header_nritems(&src->header, src_nritems - push_items); 6387518a238SChris Mason btrfs_set_header_nritems(&dst->header, dst_nritems + push_items); 63979f95c82SChris Mason 640d5719762SChris Mason mark_buffer_dirty(src_buf); 641d5719762SChris Mason mark_buffer_dirty(dst_buf); 64279f95c82SChris Mason return ret; 64379f95c82SChris Mason } 64479f95c82SChris Mason 64579f95c82SChris Mason /* 64697571fd0SChris Mason * helper function to insert a new root level in the tree. 64797571fd0SChris Mason * A new node is allocated, and a single item is inserted to 64897571fd0SChris Mason * point to the existing root 649aa5d6bedSChris Mason * 650aa5d6bedSChris Mason * returns zero on success or < 0 on failure. 65197571fd0SChris Mason */ 652e089f05cSChris Mason static int insert_new_root(struct btrfs_trans_handle *trans, struct btrfs_root 653e089f05cSChris Mason *root, struct btrfs_path *path, int level) 65474123bd7SChris Mason { 655e20d96d6SChris Mason struct buffer_head *t; 656234b63a0SChris Mason struct btrfs_node *lower; 657234b63a0SChris Mason struct btrfs_node *c; 658e2fa7227SChris Mason struct btrfs_disk_key *lower_key; 6595c680ed6SChris Mason 6605c680ed6SChris Mason BUG_ON(path->nodes[level]); 6615c680ed6SChris Mason BUG_ON(path->nodes[level-1] != root->node); 6625c680ed6SChris Mason 663e089f05cSChris Mason t = btrfs_alloc_free_block(trans, root); 664e20d96d6SChris Mason c = btrfs_buffer_node(t); 665123abc88SChris Mason memset(c, 0, root->blocksize); 6667518a238SChris Mason btrfs_set_header_nritems(&c->header, 1); 6677518a238SChris Mason btrfs_set_header_level(&c->header, level); 668e20d96d6SChris Mason btrfs_set_header_blocknr(&c->header, t->b_blocknr); 6697f5c1516SChris Mason btrfs_set_header_generation(&c->header, trans->transid); 6707518a238SChris Mason btrfs_set_header_parentid(&c->header, 671e20d96d6SChris Mason btrfs_header_parentid(btrfs_buffer_header(root->node))); 672e20d96d6SChris Mason lower = btrfs_buffer_node(path->nodes[level-1]); 6737518a238SChris Mason if (btrfs_is_leaf(lower)) 674234b63a0SChris Mason lower_key = &((struct btrfs_leaf *)lower)->items[0].key; 67574123bd7SChris Mason else 676123abc88SChris Mason lower_key = &lower->ptrs[0].key; 677123abc88SChris Mason memcpy(&c->ptrs[0].key, lower_key, sizeof(struct btrfs_disk_key)); 678e20d96d6SChris Mason btrfs_set_node_blockptr(c, 0, path->nodes[level - 1]->b_blocknr); 679d5719762SChris Mason 680d5719762SChris Mason mark_buffer_dirty(t); 681d5719762SChris Mason 682cfaa7295SChris Mason /* the super has an extra ref to root->node */ 683234b63a0SChris Mason btrfs_block_release(root, root->node); 68474123bd7SChris Mason root->node = t; 685e20d96d6SChris Mason get_bh(t); 68674123bd7SChris Mason path->nodes[level] = t; 68774123bd7SChris Mason path->slots[level] = 0; 68874123bd7SChris Mason return 0; 68974123bd7SChris Mason } 6905c680ed6SChris Mason 6915c680ed6SChris Mason /* 6925c680ed6SChris Mason * worker function to insert a single pointer in a node. 6935c680ed6SChris Mason * the node should have enough room for the pointer already 69497571fd0SChris Mason * 6955c680ed6SChris Mason * slot and level indicate where you want the key to go, and 6965c680ed6SChris Mason * blocknr is the block the key points to. 697aa5d6bedSChris Mason * 698aa5d6bedSChris Mason * returns zero on success and < 0 on any error 6995c680ed6SChris Mason */ 700e089f05cSChris Mason static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root 701e089f05cSChris Mason *root, struct btrfs_path *path, struct btrfs_disk_key 702e089f05cSChris Mason *key, u64 blocknr, int slot, int level) 7035c680ed6SChris Mason { 704234b63a0SChris Mason struct btrfs_node *lower; 7055c680ed6SChris Mason int nritems; 7065c680ed6SChris Mason 7075c680ed6SChris Mason BUG_ON(!path->nodes[level]); 708e20d96d6SChris Mason lower = btrfs_buffer_node(path->nodes[level]); 7097518a238SChris Mason nritems = btrfs_header_nritems(&lower->header); 71074123bd7SChris Mason if (slot > nritems) 71174123bd7SChris Mason BUG(); 712123abc88SChris Mason if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root)) 71374123bd7SChris Mason BUG(); 71474123bd7SChris Mason if (slot != nritems) { 715123abc88SChris Mason memmove(lower->ptrs + slot + 1, lower->ptrs + slot, 716123abc88SChris Mason (nritems - slot) * sizeof(struct btrfs_key_ptr)); 71774123bd7SChris Mason } 718123abc88SChris Mason memcpy(&lower->ptrs[slot].key, key, sizeof(struct btrfs_disk_key)); 7191d4f8a0cSChris Mason btrfs_set_node_blockptr(lower, slot, blocknr); 7207518a238SChris Mason btrfs_set_header_nritems(&lower->header, nritems + 1); 721d5719762SChris Mason mark_buffer_dirty(path->nodes[level]); 72274123bd7SChris Mason return 0; 72374123bd7SChris Mason } 72474123bd7SChris Mason 72597571fd0SChris Mason /* 72697571fd0SChris Mason * split the node at the specified level in path in two. 72797571fd0SChris Mason * The path is corrected to point to the appropriate node after the split 72897571fd0SChris Mason * 72997571fd0SChris Mason * Before splitting this tries to make some room in the node by pushing 73097571fd0SChris Mason * left and right, if either one works, it returns right away. 731aa5d6bedSChris Mason * 732aa5d6bedSChris Mason * returns 0 on success and < 0 on failure 73397571fd0SChris Mason */ 734e089f05cSChris Mason static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root 735e089f05cSChris Mason *root, struct btrfs_path *path, int level) 736be0e5c09SChris Mason { 737e20d96d6SChris Mason struct buffer_head *t; 738234b63a0SChris Mason struct btrfs_node *c; 739e20d96d6SChris Mason struct buffer_head *split_buffer; 740234b63a0SChris Mason struct btrfs_node *split; 741be0e5c09SChris Mason int mid; 7425c680ed6SChris Mason int ret; 743aa5d6bedSChris Mason int wret; 7447518a238SChris Mason u32 c_nritems; 745be0e5c09SChris Mason 7465c680ed6SChris Mason t = path->nodes[level]; 747e20d96d6SChris Mason c = btrfs_buffer_node(t); 7485c680ed6SChris Mason if (t == root->node) { 7495c680ed6SChris Mason /* trying to split the root, lets make a new one */ 750e089f05cSChris Mason ret = insert_new_root(trans, root, path, level + 1); 7515c680ed6SChris Mason if (ret) 7525c680ed6SChris Mason return ret; 7535c680ed6SChris Mason } 7547518a238SChris Mason c_nritems = btrfs_header_nritems(&c->header); 755e089f05cSChris Mason split_buffer = btrfs_alloc_free_block(trans, root); 756e20d96d6SChris Mason split = btrfs_buffer_node(split_buffer); 7577518a238SChris Mason btrfs_set_header_flags(&split->header, btrfs_header_flags(&c->header)); 7589a6f11edSChris Mason btrfs_set_header_level(&split->header, btrfs_header_level(&c->header)); 759e20d96d6SChris Mason btrfs_set_header_blocknr(&split->header, split_buffer->b_blocknr); 7607f5c1516SChris Mason btrfs_set_header_generation(&split->header, trans->transid); 7617518a238SChris Mason btrfs_set_header_parentid(&split->header, 762e20d96d6SChris Mason btrfs_header_parentid(btrfs_buffer_header(root->node))); 7637518a238SChris Mason mid = (c_nritems + 1) / 2; 764123abc88SChris Mason memcpy(split->ptrs, c->ptrs + mid, 765123abc88SChris Mason (c_nritems - mid) * sizeof(struct btrfs_key_ptr)); 7667518a238SChris Mason btrfs_set_header_nritems(&split->header, c_nritems - mid); 7677518a238SChris Mason btrfs_set_header_nritems(&c->header, mid); 768aa5d6bedSChris Mason ret = 0; 769aa5d6bedSChris Mason 770d5719762SChris Mason mark_buffer_dirty(t); 771d5719762SChris Mason mark_buffer_dirty(split_buffer); 772e089f05cSChris Mason wret = insert_ptr(trans, root, path, &split->ptrs[0].key, 773e20d96d6SChris Mason split_buffer->b_blocknr, path->slots[level + 1] + 1, 774123abc88SChris Mason level + 1); 775aa5d6bedSChris Mason if (wret) 776aa5d6bedSChris Mason ret = wret; 777aa5d6bedSChris Mason 7785de08d7dSChris Mason if (path->slots[level] >= mid) { 7795c680ed6SChris Mason path->slots[level] -= mid; 780234b63a0SChris Mason btrfs_block_release(root, t); 7815c680ed6SChris Mason path->nodes[level] = split_buffer; 7825c680ed6SChris Mason path->slots[level + 1] += 1; 783eb60ceacSChris Mason } else { 784234b63a0SChris Mason btrfs_block_release(root, split_buffer); 785be0e5c09SChris Mason } 786aa5d6bedSChris Mason return ret; 787be0e5c09SChris Mason } 788be0e5c09SChris Mason 78974123bd7SChris Mason /* 79074123bd7SChris Mason * how many bytes are required to store the items in a leaf. start 79174123bd7SChris Mason * and nr indicate which items in the leaf to check. This totals up the 79274123bd7SChris Mason * space used both by the item structs and the item data 79374123bd7SChris Mason */ 794234b63a0SChris Mason static int leaf_space_used(struct btrfs_leaf *l, int start, int nr) 795be0e5c09SChris Mason { 796be0e5c09SChris Mason int data_len; 797be0e5c09SChris Mason int end = start + nr - 1; 798be0e5c09SChris Mason 799be0e5c09SChris Mason if (!nr) 800be0e5c09SChris Mason return 0; 8010783fcfcSChris Mason data_len = btrfs_item_end(l->items + start); 8020783fcfcSChris Mason data_len = data_len - btrfs_item_offset(l->items + end); 8030783fcfcSChris Mason data_len += sizeof(struct btrfs_item) * nr; 804be0e5c09SChris Mason return data_len; 805be0e5c09SChris Mason } 806be0e5c09SChris Mason 80774123bd7SChris Mason /* 80800ec4c51SChris Mason * push some data in the path leaf to the right, trying to free up at 80900ec4c51SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 810aa5d6bedSChris Mason * 811aa5d6bedSChris Mason * returns 1 if the push failed because the other node didn't have enough 812aa5d6bedSChris Mason * room, 0 if everything worked out and < 0 if there were major errors. 81300ec4c51SChris Mason */ 814e089f05cSChris Mason static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root 815e089f05cSChris Mason *root, struct btrfs_path *path, int data_size) 81600ec4c51SChris Mason { 817e20d96d6SChris Mason struct buffer_head *left_buf = path->nodes[0]; 818e20d96d6SChris Mason struct btrfs_leaf *left = btrfs_buffer_leaf(left_buf); 819234b63a0SChris Mason struct btrfs_leaf *right; 820e20d96d6SChris Mason struct buffer_head *right_buf; 821e20d96d6SChris Mason struct buffer_head *upper; 822e20d96d6SChris Mason struct btrfs_node *upper_node; 82300ec4c51SChris Mason int slot; 82400ec4c51SChris Mason int i; 82500ec4c51SChris Mason int free_space; 82600ec4c51SChris Mason int push_space = 0; 82700ec4c51SChris Mason int push_items = 0; 8280783fcfcSChris Mason struct btrfs_item *item; 8297518a238SChris Mason u32 left_nritems; 8307518a238SChris Mason u32 right_nritems; 83100ec4c51SChris Mason 83200ec4c51SChris Mason slot = path->slots[1]; 83300ec4c51SChris Mason if (!path->nodes[1]) { 83400ec4c51SChris Mason return 1; 83500ec4c51SChris Mason } 83600ec4c51SChris Mason upper = path->nodes[1]; 837e20d96d6SChris Mason upper_node = btrfs_buffer_node(upper); 838e20d96d6SChris Mason if (slot >= btrfs_header_nritems(&upper_node->header) - 1) { 83900ec4c51SChris Mason return 1; 84000ec4c51SChris Mason } 841e20d96d6SChris Mason right_buf = read_tree_block(root, 842e20d96d6SChris Mason btrfs_node_blockptr(btrfs_buffer_node(upper), slot + 1)); 843e20d96d6SChris Mason right = btrfs_buffer_leaf(right_buf); 844123abc88SChris Mason free_space = btrfs_leaf_free_space(root, right); 8450783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 846234b63a0SChris Mason btrfs_block_release(root, right_buf); 84700ec4c51SChris Mason return 1; 84800ec4c51SChris Mason } 84902217ed2SChris Mason /* cow and double check */ 850e089f05cSChris Mason btrfs_cow_block(trans, root, right_buf, upper, slot + 1, &right_buf); 851e20d96d6SChris Mason right = btrfs_buffer_leaf(right_buf); 852123abc88SChris Mason free_space = btrfs_leaf_free_space(root, right); 8530783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 854234b63a0SChris Mason btrfs_block_release(root, right_buf); 85502217ed2SChris Mason return 1; 85602217ed2SChris Mason } 85702217ed2SChris Mason 8587518a238SChris Mason left_nritems = btrfs_header_nritems(&left->header); 8597518a238SChris Mason for (i = left_nritems - 1; i >= 0; i--) { 86000ec4c51SChris Mason item = left->items + i; 86100ec4c51SChris Mason if (path->slots[0] == i) 86200ec4c51SChris Mason push_space += data_size + sizeof(*item); 8630783fcfcSChris Mason if (btrfs_item_size(item) + sizeof(*item) + push_space > 8640783fcfcSChris Mason free_space) 86500ec4c51SChris Mason break; 86600ec4c51SChris Mason push_items++; 8670783fcfcSChris Mason push_space += btrfs_item_size(item) + sizeof(*item); 86800ec4c51SChris Mason } 86900ec4c51SChris Mason if (push_items == 0) { 870234b63a0SChris Mason btrfs_block_release(root, right_buf); 87100ec4c51SChris Mason return 1; 87200ec4c51SChris Mason } 8737518a238SChris Mason right_nritems = btrfs_header_nritems(&right->header); 87400ec4c51SChris Mason /* push left to right */ 8750783fcfcSChris Mason push_space = btrfs_item_end(left->items + left_nritems - push_items); 876123abc88SChris Mason push_space -= leaf_data_end(root, left); 87700ec4c51SChris Mason /* make room in the right data area */ 878123abc88SChris Mason memmove(btrfs_leaf_data(right) + leaf_data_end(root, right) - 879123abc88SChris Mason push_space, btrfs_leaf_data(right) + leaf_data_end(root, right), 880123abc88SChris Mason BTRFS_LEAF_DATA_SIZE(root) - leaf_data_end(root, right)); 88100ec4c51SChris Mason /* copy from the left data area */ 882123abc88SChris Mason memcpy(btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - push_space, 883123abc88SChris Mason btrfs_leaf_data(left) + leaf_data_end(root, left), push_space); 88400ec4c51SChris Mason memmove(right->items + push_items, right->items, 8850783fcfcSChris Mason right_nritems * sizeof(struct btrfs_item)); 88600ec4c51SChris Mason /* copy the items from left to right */ 8877518a238SChris Mason memcpy(right->items, left->items + left_nritems - push_items, 8880783fcfcSChris Mason push_items * sizeof(struct btrfs_item)); 88900ec4c51SChris Mason 89000ec4c51SChris Mason /* update the item pointers */ 8917518a238SChris Mason right_nritems += push_items; 8927518a238SChris Mason btrfs_set_header_nritems(&right->header, right_nritems); 893123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 8947518a238SChris Mason for (i = 0; i < right_nritems; i++) { 8950783fcfcSChris Mason btrfs_set_item_offset(right->items + i, push_space - 8960783fcfcSChris Mason btrfs_item_size(right->items + i)); 8970783fcfcSChris Mason push_space = btrfs_item_offset(right->items + i); 89800ec4c51SChris Mason } 8997518a238SChris Mason left_nritems -= push_items; 9007518a238SChris Mason btrfs_set_header_nritems(&left->header, left_nritems); 90100ec4c51SChris Mason 902d5719762SChris Mason mark_buffer_dirty(left_buf); 903d5719762SChris Mason mark_buffer_dirty(right_buf); 904e20d96d6SChris Mason memcpy(&upper_node->ptrs[slot + 1].key, 905e2fa7227SChris Mason &right->items[0].key, sizeof(struct btrfs_disk_key)); 906d5719762SChris Mason mark_buffer_dirty(upper); 90702217ed2SChris Mason 90800ec4c51SChris Mason /* then fixup the leaf pointer in the path */ 9097518a238SChris Mason if (path->slots[0] >= left_nritems) { 9107518a238SChris Mason path->slots[0] -= left_nritems; 911234b63a0SChris Mason btrfs_block_release(root, path->nodes[0]); 91200ec4c51SChris Mason path->nodes[0] = right_buf; 91300ec4c51SChris Mason path->slots[1] += 1; 91400ec4c51SChris Mason } else { 915234b63a0SChris Mason btrfs_block_release(root, right_buf); 91600ec4c51SChris Mason } 91700ec4c51SChris Mason return 0; 91800ec4c51SChris Mason } 91900ec4c51SChris Mason /* 92074123bd7SChris Mason * push some data in the path leaf to the left, trying to free up at 92174123bd7SChris Mason * least data_size bytes. returns zero if the push worked, nonzero otherwise 92274123bd7SChris Mason */ 923e089f05cSChris Mason static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root 924e089f05cSChris Mason *root, struct btrfs_path *path, int data_size) 925be0e5c09SChris Mason { 926e20d96d6SChris Mason struct buffer_head *right_buf = path->nodes[0]; 927e20d96d6SChris Mason struct btrfs_leaf *right = btrfs_buffer_leaf(right_buf); 928e20d96d6SChris Mason struct buffer_head *t; 929234b63a0SChris Mason struct btrfs_leaf *left; 930be0e5c09SChris Mason int slot; 931be0e5c09SChris Mason int i; 932be0e5c09SChris Mason int free_space; 933be0e5c09SChris Mason int push_space = 0; 934be0e5c09SChris Mason int push_items = 0; 9350783fcfcSChris Mason struct btrfs_item *item; 9367518a238SChris Mason u32 old_left_nritems; 937aa5d6bedSChris Mason int ret = 0; 938aa5d6bedSChris Mason int wret; 939be0e5c09SChris Mason 940be0e5c09SChris Mason slot = path->slots[1]; 941be0e5c09SChris Mason if (slot == 0) { 942be0e5c09SChris Mason return 1; 943be0e5c09SChris Mason } 944be0e5c09SChris Mason if (!path->nodes[1]) { 945be0e5c09SChris Mason return 1; 946be0e5c09SChris Mason } 947e20d96d6SChris Mason t = read_tree_block(root, 948e20d96d6SChris Mason btrfs_node_blockptr(btrfs_buffer_node(path->nodes[1]), slot - 1)); 949e20d96d6SChris Mason left = btrfs_buffer_leaf(t); 950123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 9510783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 952234b63a0SChris Mason btrfs_block_release(root, t); 953be0e5c09SChris Mason return 1; 954be0e5c09SChris Mason } 95502217ed2SChris Mason 95602217ed2SChris Mason /* cow and double check */ 957e089f05cSChris Mason btrfs_cow_block(trans, root, t, path->nodes[1], slot - 1, &t); 958e20d96d6SChris Mason left = btrfs_buffer_leaf(t); 959123abc88SChris Mason free_space = btrfs_leaf_free_space(root, left); 9600783fcfcSChris Mason if (free_space < data_size + sizeof(struct btrfs_item)) { 961234b63a0SChris Mason btrfs_block_release(root, t); 96202217ed2SChris Mason return 1; 96302217ed2SChris Mason } 96402217ed2SChris Mason 9657518a238SChris Mason for (i = 0; i < btrfs_header_nritems(&right->header); i++) { 966be0e5c09SChris Mason item = right->items + i; 967be0e5c09SChris Mason if (path->slots[0] == i) 968be0e5c09SChris Mason push_space += data_size + sizeof(*item); 9690783fcfcSChris Mason if (btrfs_item_size(item) + sizeof(*item) + push_space > 9700783fcfcSChris Mason free_space) 971be0e5c09SChris Mason break; 972be0e5c09SChris Mason push_items++; 9730783fcfcSChris Mason push_space += btrfs_item_size(item) + sizeof(*item); 974be0e5c09SChris Mason } 975be0e5c09SChris Mason if (push_items == 0) { 976234b63a0SChris Mason btrfs_block_release(root, t); 977be0e5c09SChris Mason return 1; 978be0e5c09SChris Mason } 979be0e5c09SChris Mason /* push data from right to left */ 9807518a238SChris Mason memcpy(left->items + btrfs_header_nritems(&left->header), 9810783fcfcSChris Mason right->items, push_items * sizeof(struct btrfs_item)); 982123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root) - 9830783fcfcSChris Mason btrfs_item_offset(right->items + push_items -1); 984123abc88SChris Mason memcpy(btrfs_leaf_data(left) + leaf_data_end(root, left) - push_space, 985123abc88SChris Mason btrfs_leaf_data(right) + 986123abc88SChris Mason btrfs_item_offset(right->items + push_items - 1), 987be0e5c09SChris Mason push_space); 9887518a238SChris Mason old_left_nritems = btrfs_header_nritems(&left->header); 989eb60ceacSChris Mason BUG_ON(old_left_nritems < 0); 990eb60ceacSChris Mason 991be0e5c09SChris Mason for (i = old_left_nritems; i < old_left_nritems + push_items; i++) { 992123abc88SChris Mason u32 ioff = btrfs_item_offset(left->items + i); 993123abc88SChris Mason btrfs_set_item_offset(left->items + i, ioff - 994123abc88SChris Mason (BTRFS_LEAF_DATA_SIZE(root) - 9950783fcfcSChris Mason btrfs_item_offset(left->items + 9960783fcfcSChris Mason old_left_nritems - 1))); 997be0e5c09SChris Mason } 9987518a238SChris Mason btrfs_set_header_nritems(&left->header, old_left_nritems + push_items); 999be0e5c09SChris Mason 1000be0e5c09SChris Mason /* fixup right node */ 10010783fcfcSChris Mason push_space = btrfs_item_offset(right->items + push_items - 1) - 1002123abc88SChris Mason leaf_data_end(root, right); 1003123abc88SChris Mason memmove(btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - 1004123abc88SChris Mason push_space, btrfs_leaf_data(right) + 1005123abc88SChris Mason leaf_data_end(root, right), push_space); 1006be0e5c09SChris Mason memmove(right->items, right->items + push_items, 10077518a238SChris Mason (btrfs_header_nritems(&right->header) - push_items) * 10080783fcfcSChris Mason sizeof(struct btrfs_item)); 10097518a238SChris Mason btrfs_set_header_nritems(&right->header, 10107518a238SChris Mason btrfs_header_nritems(&right->header) - 10117518a238SChris Mason push_items); 1012123abc88SChris Mason push_space = BTRFS_LEAF_DATA_SIZE(root); 1013eb60ceacSChris Mason 10147518a238SChris Mason for (i = 0; i < btrfs_header_nritems(&right->header); i++) { 10150783fcfcSChris Mason btrfs_set_item_offset(right->items + i, push_space - 10160783fcfcSChris Mason btrfs_item_size(right->items + i)); 10170783fcfcSChris Mason push_space = btrfs_item_offset(right->items + i); 1018be0e5c09SChris Mason } 1019eb60ceacSChris Mason 1020d5719762SChris Mason mark_buffer_dirty(t); 1021d5719762SChris Mason mark_buffer_dirty(right_buf); 1022eb60ceacSChris Mason 1023e089f05cSChris Mason wret = fixup_low_keys(trans, root, path, &right->items[0].key, 1); 1024aa5d6bedSChris Mason if (wret) 1025aa5d6bedSChris Mason ret = wret; 1026be0e5c09SChris Mason 1027be0e5c09SChris Mason /* then fixup the leaf pointer in the path */ 1028be0e5c09SChris Mason if (path->slots[0] < push_items) { 1029be0e5c09SChris Mason path->slots[0] += old_left_nritems; 1030234b63a0SChris Mason btrfs_block_release(root, path->nodes[0]); 1031eb60ceacSChris Mason path->nodes[0] = t; 1032be0e5c09SChris Mason path->slots[1] -= 1; 1033be0e5c09SChris Mason } else { 1034234b63a0SChris Mason btrfs_block_release(root, t); 1035be0e5c09SChris Mason path->slots[0] -= push_items; 1036be0e5c09SChris Mason } 1037eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 1038aa5d6bedSChris Mason return ret; 1039be0e5c09SChris Mason } 1040be0e5c09SChris Mason 104174123bd7SChris Mason /* 104274123bd7SChris Mason * split the path's leaf in two, making sure there is at least data_size 104374123bd7SChris Mason * available for the resulting leaf level of the path. 1044aa5d6bedSChris Mason * 1045aa5d6bedSChris Mason * returns 0 if all went well and < 0 on failure. 104674123bd7SChris Mason */ 1047e089f05cSChris Mason static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root 1048e089f05cSChris Mason *root, struct btrfs_path *path, int data_size) 1049be0e5c09SChris Mason { 1050e20d96d6SChris Mason struct buffer_head *l_buf; 1051234b63a0SChris Mason struct btrfs_leaf *l; 10527518a238SChris Mason u32 nritems; 1053eb60ceacSChris Mason int mid; 1054eb60ceacSChris Mason int slot; 1055234b63a0SChris Mason struct btrfs_leaf *right; 1056e20d96d6SChris Mason struct buffer_head *right_buffer; 10570783fcfcSChris Mason int space_needed = data_size + sizeof(struct btrfs_item); 1058be0e5c09SChris Mason int data_copy_size; 1059be0e5c09SChris Mason int rt_data_off; 1060be0e5c09SChris Mason int i; 1061be0e5c09SChris Mason int ret; 1062aa5d6bedSChris Mason int wret; 1063be0e5c09SChris Mason 106440689478SChris Mason /* first try to make some room by pushing left and right */ 1065e089f05cSChris Mason wret = push_leaf_left(trans, root, path, data_size); 1066eaee50e8SChris Mason if (wret < 0) 1067eaee50e8SChris Mason return wret; 1068eaee50e8SChris Mason if (wret) { 1069e089f05cSChris Mason wret = push_leaf_right(trans, root, path, data_size); 1070eaee50e8SChris Mason if (wret < 0) 1071eaee50e8SChris Mason return wret; 1072eaee50e8SChris Mason } 1073eb60ceacSChris Mason l_buf = path->nodes[0]; 1074e20d96d6SChris Mason l = btrfs_buffer_leaf(l_buf); 1075aa5d6bedSChris Mason 1076aa5d6bedSChris Mason /* did the pushes work? */ 1077123abc88SChris Mason if (btrfs_leaf_free_space(root, l) >= 1078123abc88SChris Mason sizeof(struct btrfs_item) + data_size) 1079be0e5c09SChris Mason return 0; 1080aa5d6bedSChris Mason 10815c680ed6SChris Mason if (!path->nodes[1]) { 1082e089f05cSChris Mason ret = insert_new_root(trans, root, path, 1); 10835c680ed6SChris Mason if (ret) 10845c680ed6SChris Mason return ret; 10855c680ed6SChris Mason } 1086eb60ceacSChris Mason slot = path->slots[0]; 10877518a238SChris Mason nritems = btrfs_header_nritems(&l->header); 1088eb60ceacSChris Mason mid = (nritems + 1)/ 2; 1089e089f05cSChris Mason right_buffer = btrfs_alloc_free_block(trans, root); 1090eb60ceacSChris Mason BUG_ON(!right_buffer); 1091eb60ceacSChris Mason BUG_ON(mid == nritems); 1092e20d96d6SChris Mason right = btrfs_buffer_leaf(right_buffer); 1093123abc88SChris Mason memset(&right->header, 0, sizeof(right->header)); 1094be0e5c09SChris Mason if (mid <= slot) { 109597571fd0SChris Mason /* FIXME, just alloc a new leaf here */ 1096be0e5c09SChris Mason if (leaf_space_used(l, mid, nritems - mid) + space_needed > 1097123abc88SChris Mason BTRFS_LEAF_DATA_SIZE(root)) 1098be0e5c09SChris Mason BUG(); 1099be0e5c09SChris Mason } else { 110097571fd0SChris Mason /* FIXME, just alloc a new leaf here */ 1101be0e5c09SChris Mason if (leaf_space_used(l, 0, mid + 1) + space_needed > 1102123abc88SChris Mason BTRFS_LEAF_DATA_SIZE(root)) 1103be0e5c09SChris Mason BUG(); 1104be0e5c09SChris Mason } 11057518a238SChris Mason btrfs_set_header_nritems(&right->header, nritems - mid); 1106e20d96d6SChris Mason btrfs_set_header_blocknr(&right->header, right_buffer->b_blocknr); 11077f5c1516SChris Mason btrfs_set_header_generation(&right->header, trans->transid); 11087518a238SChris Mason btrfs_set_header_level(&right->header, 0); 11097518a238SChris Mason btrfs_set_header_parentid(&right->header, 1110e20d96d6SChris Mason btrfs_header_parentid(btrfs_buffer_header(root->node))); 1111123abc88SChris Mason data_copy_size = btrfs_item_end(l->items + mid) - 1112123abc88SChris Mason leaf_data_end(root, l); 1113be0e5c09SChris Mason memcpy(right->items, l->items + mid, 11140783fcfcSChris Mason (nritems - mid) * sizeof(struct btrfs_item)); 1115123abc88SChris Mason memcpy(btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) - 1116123abc88SChris Mason data_copy_size, btrfs_leaf_data(l) + 1117123abc88SChris Mason leaf_data_end(root, l), data_copy_size); 1118123abc88SChris Mason rt_data_off = BTRFS_LEAF_DATA_SIZE(root) - 1119123abc88SChris Mason btrfs_item_end(l->items + mid); 112074123bd7SChris Mason 11210783fcfcSChris Mason for (i = 0; i < btrfs_header_nritems(&right->header); i++) { 1122123abc88SChris Mason u32 ioff = btrfs_item_offset(right->items + i); 11230783fcfcSChris Mason btrfs_set_item_offset(right->items + i, ioff + rt_data_off); 11240783fcfcSChris Mason } 112574123bd7SChris Mason 11267518a238SChris Mason btrfs_set_header_nritems(&l->header, mid); 1127aa5d6bedSChris Mason ret = 0; 1128e089f05cSChris Mason wret = insert_ptr(trans, root, path, &right->items[0].key, 1129e20d96d6SChris Mason right_buffer->b_blocknr, path->slots[1] + 1, 1); 1130aa5d6bedSChris Mason if (wret) 1131aa5d6bedSChris Mason ret = wret; 1132d5719762SChris Mason mark_buffer_dirty(right_buffer); 1133d5719762SChris Mason mark_buffer_dirty(l_buf); 1134eb60ceacSChris Mason BUG_ON(path->slots[0] != slot); 1135be0e5c09SChris Mason if (mid <= slot) { 1136234b63a0SChris Mason btrfs_block_release(root, path->nodes[0]); 1137eb60ceacSChris Mason path->nodes[0] = right_buffer; 1138be0e5c09SChris Mason path->slots[0] -= mid; 1139be0e5c09SChris Mason path->slots[1] += 1; 1140eb60ceacSChris Mason } else 1141234b63a0SChris Mason btrfs_block_release(root, right_buffer); 1142eb60ceacSChris Mason BUG_ON(path->slots[0] < 0); 1143be0e5c09SChris Mason return ret; 1144be0e5c09SChris Mason } 1145be0e5c09SChris Mason 114674123bd7SChris Mason /* 114774123bd7SChris Mason * Given a key and some data, insert an item into the tree. 114874123bd7SChris Mason * This does all the path init required, making room in the tree if needed. 114974123bd7SChris Mason */ 1150e089f05cSChris Mason int btrfs_insert_empty_item(struct btrfs_trans_handle *trans, struct btrfs_root 1151e089f05cSChris Mason *root, struct btrfs_path *path, struct btrfs_key 1152e089f05cSChris Mason *cpu_key, u32 data_size) 1153be0e5c09SChris Mason { 1154aa5d6bedSChris Mason int ret = 0; 1155be0e5c09SChris Mason int slot; 1156eb60ceacSChris Mason int slot_orig; 1157234b63a0SChris Mason struct btrfs_leaf *leaf; 1158e20d96d6SChris Mason struct buffer_head *leaf_buf; 11597518a238SChris Mason u32 nritems; 1160be0e5c09SChris Mason unsigned int data_end; 1161e2fa7227SChris Mason struct btrfs_disk_key disk_key; 1162e2fa7227SChris Mason 1163e2fa7227SChris Mason btrfs_cpu_key_to_disk(&disk_key, cpu_key); 1164be0e5c09SChris Mason 116574123bd7SChris Mason /* create a root if there isn't one */ 11665c680ed6SChris Mason if (!root->node) 1167cfaa7295SChris Mason BUG(); 1168e089f05cSChris Mason ret = btrfs_search_slot(trans, root, cpu_key, path, data_size, 1); 1169eb60ceacSChris Mason if (ret == 0) { 1170f0930a37SChris Mason return -EEXIST; 1171aa5d6bedSChris Mason } 1172ed2ff2cbSChris Mason if (ret < 0) 1173ed2ff2cbSChris Mason goto out; 1174be0e5c09SChris Mason 117562e2749eSChris Mason slot_orig = path->slots[0]; 117662e2749eSChris Mason leaf_buf = path->nodes[0]; 1177e20d96d6SChris Mason leaf = btrfs_buffer_leaf(leaf_buf); 117874123bd7SChris Mason 11797518a238SChris Mason nritems = btrfs_header_nritems(&leaf->header); 1180123abc88SChris Mason data_end = leaf_data_end(root, leaf); 1181eb60ceacSChris Mason 1182123abc88SChris Mason if (btrfs_leaf_free_space(root, leaf) < 1183234b63a0SChris Mason sizeof(struct btrfs_item) + data_size) 1184be0e5c09SChris Mason BUG(); 1185be0e5c09SChris Mason 118662e2749eSChris Mason slot = path->slots[0]; 1187eb60ceacSChris Mason BUG_ON(slot < 0); 1188be0e5c09SChris Mason if (slot != nritems) { 1189be0e5c09SChris Mason int i; 11900783fcfcSChris Mason unsigned int old_data = btrfs_item_end(leaf->items + slot); 1191be0e5c09SChris Mason 1192be0e5c09SChris Mason /* 1193be0e5c09SChris Mason * item0..itemN ... dataN.offset..dataN.size .. data0.size 1194be0e5c09SChris Mason */ 1195be0e5c09SChris Mason /* first correct the data pointers */ 11960783fcfcSChris Mason for (i = slot; i < nritems; i++) { 1197123abc88SChris Mason u32 ioff = btrfs_item_offset(leaf->items + i); 11980783fcfcSChris Mason btrfs_set_item_offset(leaf->items + i, 11990783fcfcSChris Mason ioff - data_size); 12000783fcfcSChris Mason } 1201be0e5c09SChris Mason 1202be0e5c09SChris Mason /* shift the items */ 1203be0e5c09SChris Mason memmove(leaf->items + slot + 1, leaf->items + slot, 12040783fcfcSChris Mason (nritems - slot) * sizeof(struct btrfs_item)); 1205be0e5c09SChris Mason 1206be0e5c09SChris Mason /* shift the data */ 1207123abc88SChris Mason memmove(btrfs_leaf_data(leaf) + data_end - data_size, 1208123abc88SChris Mason btrfs_leaf_data(leaf) + 1209be0e5c09SChris Mason data_end, old_data - data_end); 1210be0e5c09SChris Mason data_end = old_data; 1211be0e5c09SChris Mason } 121262e2749eSChris Mason /* setup the item for the new data */ 1213e2fa7227SChris Mason memcpy(&leaf->items[slot].key, &disk_key, 1214e2fa7227SChris Mason sizeof(struct btrfs_disk_key)); 12150783fcfcSChris Mason btrfs_set_item_offset(leaf->items + slot, data_end - data_size); 12160783fcfcSChris Mason btrfs_set_item_size(leaf->items + slot, data_size); 12177518a238SChris Mason btrfs_set_header_nritems(&leaf->header, nritems + 1); 1218d5719762SChris Mason mark_buffer_dirty(leaf_buf); 1219aa5d6bedSChris Mason 1220aa5d6bedSChris Mason ret = 0; 12218e19f2cdSChris Mason if (slot == 0) 1222e089f05cSChris Mason ret = fixup_low_keys(trans, root, path, &disk_key, 1); 1223aa5d6bedSChris Mason 1224123abc88SChris Mason if (btrfs_leaf_free_space(root, leaf) < 0) 1225be0e5c09SChris Mason BUG(); 122662e2749eSChris Mason check_leaf(root, path, 0); 1227ed2ff2cbSChris Mason out: 122862e2749eSChris Mason return ret; 122962e2749eSChris Mason } 123062e2749eSChris Mason 123162e2749eSChris Mason /* 123262e2749eSChris Mason * Given a key and some data, insert an item into the tree. 123362e2749eSChris Mason * This does all the path init required, making room in the tree if needed. 123462e2749eSChris Mason */ 1235e089f05cSChris Mason int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root 1236e089f05cSChris Mason *root, struct btrfs_key *cpu_key, void *data, u32 1237e089f05cSChris Mason data_size) 123862e2749eSChris Mason { 123962e2749eSChris Mason int ret = 0; 124062e2749eSChris Mason struct btrfs_path path; 124162e2749eSChris Mason u8 *ptr; 124262e2749eSChris Mason 124362e2749eSChris Mason btrfs_init_path(&path); 1244e089f05cSChris Mason ret = btrfs_insert_empty_item(trans, root, &path, cpu_key, data_size); 124562e2749eSChris Mason if (!ret) { 1246e20d96d6SChris Mason ptr = btrfs_item_ptr(btrfs_buffer_leaf(path.nodes[0]), 1247e20d96d6SChris Mason path.slots[0], u8); 124862e2749eSChris Mason memcpy(ptr, data, data_size); 1249d5719762SChris Mason mark_buffer_dirty(path.nodes[0]); 125062e2749eSChris Mason } 1251234b63a0SChris Mason btrfs_release_path(root, &path); 1252aa5d6bedSChris Mason return ret; 1253be0e5c09SChris Mason } 1254be0e5c09SChris Mason 125574123bd7SChris Mason /* 12565de08d7dSChris Mason * delete the pointer from a given node. 125774123bd7SChris Mason * 125874123bd7SChris Mason * If the delete empties a node, the node is removed from the tree, 125974123bd7SChris Mason * continuing all the way the root if required. The root is converted into 126074123bd7SChris Mason * a leaf if all the nodes are emptied. 126174123bd7SChris Mason */ 1262e089f05cSChris Mason static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root, 1263e089f05cSChris Mason struct btrfs_path *path, int level, int slot) 1264be0e5c09SChris Mason { 1265234b63a0SChris Mason struct btrfs_node *node; 1266e20d96d6SChris Mason struct buffer_head *parent = path->nodes[level]; 12677518a238SChris Mason u32 nritems; 1268aa5d6bedSChris Mason int ret = 0; 1269bb803951SChris Mason int wret; 1270be0e5c09SChris Mason 1271e20d96d6SChris Mason node = btrfs_buffer_node(parent); 12727518a238SChris Mason nritems = btrfs_header_nritems(&node->header); 1273be0e5c09SChris Mason if (slot != nritems -1) { 1274123abc88SChris Mason memmove(node->ptrs + slot, node->ptrs + slot + 1, 1275123abc88SChris Mason sizeof(struct btrfs_key_ptr) * (nritems - slot - 1)); 1276be0e5c09SChris Mason } 12777518a238SChris Mason nritems--; 12787518a238SChris Mason btrfs_set_header_nritems(&node->header, nritems); 12797518a238SChris Mason if (nritems == 0 && parent == root->node) { 1280e20d96d6SChris Mason struct btrfs_header *header = btrfs_buffer_header(root->node); 1281e20d96d6SChris Mason BUG_ON(btrfs_header_level(header) != 1); 1282eb60ceacSChris Mason /* just turn the root into a leaf and break */ 1283e20d96d6SChris Mason btrfs_set_header_level(header, 0); 1284bb803951SChris Mason } else if (slot == 0) { 1285e089f05cSChris Mason wret = fixup_low_keys(trans, root, path, &node->ptrs[0].key, 1286123abc88SChris Mason level + 1); 12870f70abe2SChris Mason if (wret) 12880f70abe2SChris Mason ret = wret; 1289be0e5c09SChris Mason } 1290d5719762SChris Mason mark_buffer_dirty(parent); 1291aa5d6bedSChris Mason return ret; 1292be0e5c09SChris Mason } 1293be0e5c09SChris Mason 129474123bd7SChris Mason /* 129574123bd7SChris Mason * delete the item at the leaf level in path. If that empties 129674123bd7SChris Mason * the leaf, remove it from the tree 129774123bd7SChris Mason */ 1298e089f05cSChris Mason int btrfs_del_item(struct btrfs_trans_handle *trans, struct btrfs_root *root, 1299e089f05cSChris Mason struct btrfs_path *path) 1300be0e5c09SChris Mason { 1301be0e5c09SChris Mason int slot; 1302234b63a0SChris Mason struct btrfs_leaf *leaf; 1303e20d96d6SChris Mason struct buffer_head *leaf_buf; 1304be0e5c09SChris Mason int doff; 1305be0e5c09SChris Mason int dsize; 1306aa5d6bedSChris Mason int ret = 0; 1307aa5d6bedSChris Mason int wret; 13087518a238SChris Mason u32 nritems; 1309be0e5c09SChris Mason 1310eb60ceacSChris Mason leaf_buf = path->nodes[0]; 1311e20d96d6SChris Mason leaf = btrfs_buffer_leaf(leaf_buf); 13124920c9acSChris Mason slot = path->slots[0]; 13130783fcfcSChris Mason doff = btrfs_item_offset(leaf->items + slot); 13140783fcfcSChris Mason dsize = btrfs_item_size(leaf->items + slot); 13157518a238SChris Mason nritems = btrfs_header_nritems(&leaf->header); 1316be0e5c09SChris Mason 13177518a238SChris Mason if (slot != nritems - 1) { 1318be0e5c09SChris Mason int i; 1319123abc88SChris Mason int data_end = leaf_data_end(root, leaf); 1320123abc88SChris Mason memmove(btrfs_leaf_data(leaf) + data_end + dsize, 1321123abc88SChris Mason btrfs_leaf_data(leaf) + data_end, 1322be0e5c09SChris Mason doff - data_end); 13230783fcfcSChris Mason for (i = slot + 1; i < nritems; i++) { 1324123abc88SChris Mason u32 ioff = btrfs_item_offset(leaf->items + i); 13250783fcfcSChris Mason btrfs_set_item_offset(leaf->items + i, ioff + dsize); 13260783fcfcSChris Mason } 1327be0e5c09SChris Mason memmove(leaf->items + slot, leaf->items + slot + 1, 13280783fcfcSChris Mason sizeof(struct btrfs_item) * 13297518a238SChris Mason (nritems - slot - 1)); 1330be0e5c09SChris Mason } 13317518a238SChris Mason btrfs_set_header_nritems(&leaf->header, nritems - 1); 13327518a238SChris Mason nritems--; 133374123bd7SChris Mason /* delete the leaf if we've emptied it */ 13347518a238SChris Mason if (nritems == 0) { 1335eb60ceacSChris Mason if (leaf_buf == root->node) { 13367518a238SChris Mason btrfs_set_header_level(&leaf->header, 0); 13379a8dd150SChris Mason } else { 1338e089f05cSChris Mason clean_tree_block(trans, root, leaf_buf); 1339e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, path->slots[1]); 1340aa5d6bedSChris Mason if (wret) 1341aa5d6bedSChris Mason ret = wret; 1342e089f05cSChris Mason wret = btrfs_free_extent(trans, root, 1343e20d96d6SChris Mason leaf_buf->b_blocknr, 1, 1); 13440f70abe2SChris Mason if (wret) 13450f70abe2SChris Mason ret = wret; 13469a8dd150SChris Mason } 1347be0e5c09SChris Mason } else { 13487518a238SChris Mason int used = leaf_space_used(leaf, 0, nritems); 1349aa5d6bedSChris Mason if (slot == 0) { 1350e089f05cSChris Mason wret = fixup_low_keys(trans, root, path, 1351aa5d6bedSChris Mason &leaf->items[0].key, 1); 1352aa5d6bedSChris Mason if (wret) 1353aa5d6bedSChris Mason ret = wret; 1354aa5d6bedSChris Mason } 1355aa5d6bedSChris Mason 135674123bd7SChris Mason /* delete the leaf if it is mostly empty */ 1357123abc88SChris Mason if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) { 1358be0e5c09SChris Mason /* push_leaf_left fixes the path. 1359be0e5c09SChris Mason * make sure the path still points to our leaf 1360be0e5c09SChris Mason * for possible call to del_ptr below 1361be0e5c09SChris Mason */ 13624920c9acSChris Mason slot = path->slots[1]; 1363e20d96d6SChris Mason get_bh(leaf_buf); 1364e089f05cSChris Mason wret = push_leaf_left(trans, root, path, 1); 1365aa5d6bedSChris Mason if (wret < 0) 1366aa5d6bedSChris Mason ret = wret; 1367f0930a37SChris Mason if (path->nodes[0] == leaf_buf && 13687518a238SChris Mason btrfs_header_nritems(&leaf->header)) { 1369e089f05cSChris Mason wret = push_leaf_right(trans, root, path, 1); 1370aa5d6bedSChris Mason if (wret < 0) 1371aa5d6bedSChris Mason ret = wret; 1372aa5d6bedSChris Mason } 13737518a238SChris Mason if (btrfs_header_nritems(&leaf->header) == 0) { 1374e20d96d6SChris Mason u64 blocknr = leaf_buf->b_blocknr; 1375e089f05cSChris Mason clean_tree_block(trans, root, leaf_buf); 1376e089f05cSChris Mason wret = del_ptr(trans, root, path, 1, slot); 1377aa5d6bedSChris Mason if (wret) 1378aa5d6bedSChris Mason ret = wret; 1379234b63a0SChris Mason btrfs_block_release(root, leaf_buf); 1380e089f05cSChris Mason wret = btrfs_free_extent(trans, root, blocknr, 1381e089f05cSChris Mason 1, 1); 13820f70abe2SChris Mason if (wret) 13830f70abe2SChris Mason ret = wret; 13845de08d7dSChris Mason } else { 1385d5719762SChris Mason mark_buffer_dirty(leaf_buf); 1386234b63a0SChris Mason btrfs_block_release(root, leaf_buf); 1387be0e5c09SChris Mason } 1388d5719762SChris Mason } else { 1389d5719762SChris Mason mark_buffer_dirty(leaf_buf); 1390be0e5c09SChris Mason } 13919a8dd150SChris Mason } 1392aa5d6bedSChris Mason return ret; 13939a8dd150SChris Mason } 13949a8dd150SChris Mason 139597571fd0SChris Mason /* 139697571fd0SChris Mason * walk up the tree as far as required to find the next leaf. 13970f70abe2SChris Mason * returns 0 if it found something or 1 if there are no greater leaves. 13980f70abe2SChris Mason * returns < 0 on io errors. 139997571fd0SChris Mason */ 1400234b63a0SChris Mason int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path) 1401d97e63b6SChris Mason { 1402d97e63b6SChris Mason int slot; 1403d97e63b6SChris Mason int level = 1; 1404d97e63b6SChris Mason u64 blocknr; 1405e20d96d6SChris Mason struct buffer_head *c; 1406e20d96d6SChris Mason struct btrfs_node *c_node; 1407e20d96d6SChris Mason struct buffer_head *next = NULL; 1408d97e63b6SChris Mason 1409234b63a0SChris Mason while(level < BTRFS_MAX_LEVEL) { 1410d97e63b6SChris Mason if (!path->nodes[level]) 14110f70abe2SChris Mason return 1; 1412d97e63b6SChris Mason slot = path->slots[level] + 1; 1413d97e63b6SChris Mason c = path->nodes[level]; 1414e20d96d6SChris Mason c_node = btrfs_buffer_node(c); 1415e20d96d6SChris Mason if (slot >= btrfs_header_nritems(&c_node->header)) { 1416d97e63b6SChris Mason level++; 1417d97e63b6SChris Mason continue; 1418d97e63b6SChris Mason } 1419e20d96d6SChris Mason blocknr = btrfs_node_blockptr(c_node, slot); 1420cfaa7295SChris Mason if (next) 1421234b63a0SChris Mason btrfs_block_release(root, next); 1422d97e63b6SChris Mason next = read_tree_block(root, blocknr); 1423d97e63b6SChris Mason break; 1424d97e63b6SChris Mason } 1425d97e63b6SChris Mason path->slots[level] = slot; 1426d97e63b6SChris Mason while(1) { 1427d97e63b6SChris Mason level--; 1428d97e63b6SChris Mason c = path->nodes[level]; 1429234b63a0SChris Mason btrfs_block_release(root, c); 1430d97e63b6SChris Mason path->nodes[level] = next; 1431d97e63b6SChris Mason path->slots[level] = 0; 1432d97e63b6SChris Mason if (!level) 1433d97e63b6SChris Mason break; 14341d4f8a0cSChris Mason next = read_tree_block(root, 1435e20d96d6SChris Mason btrfs_node_blockptr(btrfs_buffer_node(next), 0)); 1436d97e63b6SChris Mason } 1437d97e63b6SChris Mason return 0; 1438d97e63b6SChris Mason } 1439