Btrfs: Use an array of pages in the extent buffers to reduce the cost of find_get_page
[GitHub/mt8127/android_kernel_alcatel_ttab.git] / fs / btrfs / ctree.c
CommitLineData
6cbd5570
CM
1/*
2 * Copyright (C) 2007 Oracle. All rights reserved.
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public
6 * License v2 as published by the Free Software Foundation.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 * General Public License for more details.
12 *
13 * You should have received a copy of the GNU General Public
14 * License along with this program; if not, write to the
15 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
16 * Boston, MA 021110-1307, USA.
17 */
18
5f39d397 19#include <linux/highmem.h>
eb60ceac
CM
20#include "ctree.h"
21#include "disk-io.h"
7f5c1516 22#include "transaction.h"
5f39d397 23#include "print-tree.h"
9a8dd150 24
e089f05c
CM
25static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
26 *root, struct btrfs_path *path, int level);
27static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
d4dbff95
CM
28 *root, struct btrfs_key *ins_key,
29 struct btrfs_path *path, int data_size);
5f39d397
CM
30static int push_node_left(struct btrfs_trans_handle *trans,
31 struct btrfs_root *root, struct extent_buffer *dst,
32 struct extent_buffer *src);
33static int balance_node_right(struct btrfs_trans_handle *trans,
34 struct btrfs_root *root,
35 struct extent_buffer *dst_buf,
36 struct extent_buffer *src_buf);
e089f05c
CM
37static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
38 struct btrfs_path *path, int level, int slot);
d97e63b6 39
df24a2b9 40inline void btrfs_init_path(struct btrfs_path *p)
2c90e5d6 41{
df24a2b9 42 memset(p, 0, sizeof(*p));
2c90e5d6
CM
43}
44
df24a2b9 45struct btrfs_path *btrfs_alloc_path(void)
2c90e5d6 46{
df24a2b9
CM
47 struct btrfs_path *path;
48 path = kmem_cache_alloc(btrfs_path_cachep, GFP_NOFS);
2cc58cf2 49 if (path) {
df24a2b9 50 btrfs_init_path(path);
2cc58cf2
CM
51 path->reada = 1;
52 }
df24a2b9 53 return path;
2c90e5d6
CM
54}
55
df24a2b9 56void btrfs_free_path(struct btrfs_path *p)
be0e5c09 57{
df24a2b9
CM
58 btrfs_release_path(NULL, p);
59 kmem_cache_free(btrfs_path_cachep, p);
be0e5c09
CM
60}
61
234b63a0 62void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p)
eb60ceac
CM
63{
64 int i;
234b63a0 65 for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
eb60ceac
CM
66 if (!p->nodes[i])
67 break;
5f39d397 68 free_extent_buffer(p->nodes[i]);
eb60ceac 69 }
aa5d6bed 70 memset(p, 0, sizeof(*p));
eb60ceac
CM
71}
72
5f39d397
CM
73static int __btrfs_cow_block(struct btrfs_trans_handle *trans,
74 struct btrfs_root *root,
75 struct extent_buffer *buf,
76 struct extent_buffer *parent, int parent_slot,
77 struct extent_buffer **cow_ret,
78 u64 search_start, u64 empty_size)
02217ed2 79{
5f39d397 80 struct extent_buffer *cow;
6702ed49
CM
81 int ret = 0;
82 int different_trans = 0;
02217ed2 83
6702ed49 84 WARN_ON(root->ref_cows && trans->transid != root->last_trans);
5f39d397 85
db94535d
CM
86 cow = btrfs_alloc_free_block(trans, root, buf->len,
87 search_start, empty_size);
54aa1f4d
CM
88 if (IS_ERR(cow))
89 return PTR_ERR(cow);
6702ed49 90
f510cfec 91 cow->alloc_addr = (unsigned long)__builtin_return_address(0);
6702ed49 92
5f39d397 93 copy_extent_buffer(cow, buf, 0, 0, cow->len);
db94535d 94 btrfs_set_header_bytenr(cow, cow->start);
5f39d397
CM
95 btrfs_set_header_generation(cow, trans->transid);
96 btrfs_set_header_owner(cow, root->root_key.objectid);
6702ed49 97
5f39d397
CM
98 WARN_ON(btrfs_header_generation(buf) > trans->transid);
99 if (btrfs_header_generation(buf) != trans->transid) {
6702ed49
CM
100 different_trans = 1;
101 ret = btrfs_inc_ref(trans, root, buf);
102 if (ret)
103 return ret;
104 } else {
6702ed49
CM
105 clean_tree_block(trans, root, buf);
106 }
107
02217ed2
CM
108 if (buf == root->node) {
109 root->node = cow;
5f39d397 110 extent_buffer_get(cow);
2c90e5d6 111 if (buf != root->commit_root) {
db94535d
CM
112 btrfs_free_extent(trans, root, buf->start,
113 buf->len, 1);
2c90e5d6 114 }
5f39d397 115 free_extent_buffer(buf);
02217ed2 116 } else {
5f39d397 117 btrfs_set_node_blockptr(parent, parent_slot,
db94535d 118 cow->start);
d6025579 119 btrfs_mark_buffer_dirty(parent);
5f39d397 120 WARN_ON(btrfs_header_generation(parent) != trans->transid);
db94535d 121 btrfs_free_extent(trans, root, buf->start, buf->len, 1);
02217ed2 122 }
5f39d397 123 free_extent_buffer(buf);
ccd467d6 124 btrfs_mark_buffer_dirty(cow);
2c90e5d6 125 *cow_ret = cow;
02217ed2
CM
126 return 0;
127}
128
5f39d397
CM
129int btrfs_cow_block(struct btrfs_trans_handle *trans,
130 struct btrfs_root *root, struct extent_buffer *buf,
131 struct extent_buffer *parent, int parent_slot,
132 struct extent_buffer **cow_ret)
6702ed49
CM
133{
134 u64 search_start;
f510cfec 135 int ret;
6702ed49
CM
136 if (trans->transaction != root->fs_info->running_transaction) {
137 printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
138 root->fs_info->running_transaction->transid);
139 WARN_ON(1);
140 }
141 if (trans->transid != root->fs_info->generation) {
142 printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
143 root->fs_info->generation);
144 WARN_ON(1);
145 }
5f39d397 146 if (btrfs_header_generation(buf) == trans->transid) {
6702ed49
CM
147 *cow_ret = buf;
148 return 0;
149 }
150
db94535d 151 search_start = buf->start & ~((u64)BTRFS_BLOCK_GROUP_SIZE - 1);
f510cfec 152 ret = __btrfs_cow_block(trans, root, buf, parent,
6702ed49 153 parent_slot, cow_ret, search_start, 0);
f510cfec
CM
154 (*cow_ret)->alloc_addr = (unsigned long)__builtin_return_address(0);
155 return ret;
6702ed49
CM
156}
157
db94535d 158#if 0
6702ed49
CM
159static int close_blocks(u64 blocknr, u64 other)
160{
161 if (blocknr < other && other - blocknr < 8)
162 return 1;
163 if (blocknr > other && blocknr - other < 8)
164 return 1;
165 return 0;
166}
167
5f39d397 168static int should_defrag_leaf(struct extent_buffer *eb)
2cc58cf2 169{
5f39d397
CM
170 return 0;
171 struct btrfs_leaf *leaf = btrfs_buffer_leaf(eb);
2cc58cf2
CM
172 struct btrfs_disk_key *key;
173 u32 nritems;
174
175 if (buffer_defrag(bh))
176 return 1;
177
178 nritems = btrfs_header_nritems(&leaf->header);
179 if (nritems == 0)
180 return 0;
181
182 key = &leaf->items[0].key;
183 if (btrfs_disk_key_type(key) == BTRFS_DIR_ITEM_KEY)
184 return 1;
185
186 key = &leaf->items[nritems-1].key;
187 if (btrfs_disk_key_type(key) == BTRFS_DIR_ITEM_KEY)
188 return 1;
189 if (nritems > 4) {
190 key = &leaf->items[nritems/2].key;
191 if (btrfs_disk_key_type(key) == BTRFS_DIR_ITEM_KEY)
192 return 1;
193 }
194 return 0;
195}
5f39d397 196#endif
2cc58cf2 197
6702ed49 198int btrfs_realloc_node(struct btrfs_trans_handle *trans,
5f39d397 199 struct btrfs_root *root, struct extent_buffer *parent,
e9d0b13b 200 int cache_only, u64 *last_ret)
6702ed49 201{
5f39d397
CM
202 return 0;
203#if 0
6702ed49 204 struct btrfs_node *parent_node;
5f39d397
CM
205 struct extent_buffer *cur_eb;
206 struct extent_buffer *tmp_eb;
6702ed49 207 u64 blocknr;
e9d0b13b
CM
208 u64 search_start = *last_ret;
209 u64 last_block = 0;
6702ed49
CM
210 u64 other;
211 u32 parent_nritems;
212 int start_slot;
213 int end_slot;
214 int i;
215 int err = 0;
f2183bde 216 int parent_level;
6702ed49
CM
217
218 if (trans->transaction != root->fs_info->running_transaction) {
219 printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
220 root->fs_info->running_transaction->transid);
221 WARN_ON(1);
222 }
223 if (trans->transid != root->fs_info->generation) {
224 printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
225 root->fs_info->generation);
226 WARN_ON(1);
227 }
86479a04
CM
228 if (buffer_defrag_done(parent))
229 return 0;
230
6702ed49
CM
231 parent_node = btrfs_buffer_node(parent);
232 parent_nritems = btrfs_header_nritems(&parent_node->header);
f2183bde 233 parent_level = btrfs_header_level(&parent_node->header);
6702ed49
CM
234
235 start_slot = 0;
236 end_slot = parent_nritems;
237
238 if (parent_nritems == 1)
239 return 0;
240
241 for (i = start_slot; i < end_slot; i++) {
242 int close = 1;
243 blocknr = btrfs_node_blockptr(parent_node, i);
e9d0b13b
CM
244 if (last_block == 0)
245 last_block = blocknr;
6702ed49
CM
246 if (i > 0) {
247 other = btrfs_node_blockptr(parent_node, i - 1);
248 close = close_blocks(blocknr, other);
249 }
250 if (close && i < end_slot - 1) {
251 other = btrfs_node_blockptr(parent_node, i + 1);
252 close = close_blocks(blocknr, other);
253 }
e9d0b13b
CM
254 if (close) {
255 last_block = blocknr;
6702ed49 256 continue;
e9d0b13b 257 }
6702ed49
CM
258
259 cur_bh = btrfs_find_tree_block(root, blocknr);
260 if (!cur_bh || !buffer_uptodate(cur_bh) ||
2cc58cf2
CM
261 buffer_locked(cur_bh) ||
262 (parent_level != 1 && !buffer_defrag(cur_bh)) ||
263 (parent_level == 1 && !should_defrag_leaf(cur_bh))) {
6702ed49
CM
264 if (cache_only) {
265 brelse(cur_bh);
266 continue;
267 }
f2183bde
CM
268 if (!cur_bh || !buffer_uptodate(cur_bh) ||
269 buffer_locked(cur_bh)) {
270 brelse(cur_bh);
271 cur_bh = read_tree_block(root, blocknr);
272 }
6702ed49 273 }
e9d0b13b
CM
274 if (search_start == 0)
275 search_start = last_block & ~((u64)65535);
276
6702ed49
CM
277 err = __btrfs_cow_block(trans, root, cur_bh, parent, i,
278 &tmp_bh, search_start,
279 min(8, end_slot - i));
252c38f0
Y
280 if (err) {
281 brelse(cur_bh);
6702ed49 282 break;
252c38f0 283 }
6702ed49 284 search_start = bh_blocknr(tmp_bh);
f2183bde
CM
285 *last_ret = search_start;
286 if (parent_level == 1)
287 clear_buffer_defrag(tmp_bh);
86479a04 288 set_buffer_defrag_done(tmp_bh);
6702ed49
CM
289 brelse(tmp_bh);
290 }
291 return err;
5f39d397 292#endif
6702ed49
CM
293}
294
74123bd7
CM
295/*
296 * The leaf data grows from end-to-front in the node.
297 * this returns the address of the start of the last item,
298 * which is the stop of the leaf data stack
299 */
123abc88 300static inline unsigned int leaf_data_end(struct btrfs_root *root,
5f39d397 301 struct extent_buffer *leaf)
be0e5c09 302{
5f39d397 303 u32 nr = btrfs_header_nritems(leaf);
be0e5c09 304 if (nr == 0)
123abc88 305 return BTRFS_LEAF_DATA_SIZE(root);
5f39d397 306 return btrfs_item_offset_nr(leaf, nr - 1);
be0e5c09
CM
307}
308
74123bd7
CM
309/*
310 * compare two keys in a memcmp fashion
311 */
9aca1d51 312static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
be0e5c09 313{
e2fa7227
CM
314 struct btrfs_key k1;
315
316 btrfs_disk_key_to_cpu(&k1, disk);
317
318 if (k1.objectid > k2->objectid)
be0e5c09 319 return 1;
e2fa7227 320 if (k1.objectid < k2->objectid)
be0e5c09 321 return -1;
5f39d397 322 if (k1.type > k2->type)
f254e52c 323 return 1;
5f39d397 324 if (k1.type < k2->type)
f254e52c 325 return -1;
70b2befd
CM
326 if (k1.offset > k2->offset)
327 return 1;
328 if (k1.offset < k2->offset)
329 return -1;
be0e5c09
CM
330 return 0;
331}
74123bd7 332
123abc88
CM
333static int check_node(struct btrfs_root *root, struct btrfs_path *path,
334 int level)
aa5d6bed 335{
5f39d397
CM
336 struct extent_buffer *parent = NULL;
337 struct extent_buffer *node = path->nodes[level];
338 struct btrfs_disk_key parent_key;
339 struct btrfs_disk_key node_key;
aa5d6bed 340 int parent_slot;
8d7be552
CM
341 int slot;
342 struct btrfs_key cpukey;
5f39d397 343 u32 nritems = btrfs_header_nritems(node);
aa5d6bed
CM
344
345 if (path->nodes[level + 1])
5f39d397 346 parent = path->nodes[level + 1];
a1f39630 347
8d7be552 348 slot = path->slots[level];
7518a238
CM
349 BUG_ON(nritems == 0);
350 if (parent) {
a1f39630 351 parent_slot = path->slots[level + 1];
5f39d397
CM
352 btrfs_node_key(parent, &parent_key, parent_slot);
353 btrfs_node_key(node, &node_key, 0);
354 BUG_ON(memcmp(&parent_key, &node_key,
e2fa7227 355 sizeof(struct btrfs_disk_key)));
1d4f8a0c 356 BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
db94535d 357 btrfs_header_bytenr(node));
aa5d6bed 358 }
123abc88 359 BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
8d7be552 360 if (slot != 0) {
5f39d397
CM
361 btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
362 btrfs_node_key(node, &node_key, slot);
363 BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
8d7be552
CM
364 }
365 if (slot < nritems - 1) {
5f39d397
CM
366 btrfs_node_key_to_cpu(node, &cpukey, slot + 1);
367 btrfs_node_key(node, &node_key, slot);
368 BUG_ON(comp_keys(&node_key, &cpukey) >= 0);
aa5d6bed
CM
369 }
370 return 0;
371}
372
123abc88
CM
373static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
374 int level)
aa5d6bed 375{
5f39d397
CM
376 struct extent_buffer *leaf = path->nodes[level];
377 struct extent_buffer *parent = NULL;
aa5d6bed 378 int parent_slot;
8d7be552 379 struct btrfs_key cpukey;
5f39d397
CM
380 struct btrfs_disk_key parent_key;
381 struct btrfs_disk_key leaf_key;
382 int slot = path->slots[0];
8d7be552 383
5f39d397 384 u32 nritems = btrfs_header_nritems(leaf);
aa5d6bed
CM
385
386 if (path->nodes[level + 1])
5f39d397 387 parent = path->nodes[level + 1];
7518a238
CM
388
389 if (nritems == 0)
390 return 0;
391
392 if (parent) {
a1f39630 393 parent_slot = path->slots[level + 1];
5f39d397
CM
394 btrfs_node_key(parent, &parent_key, parent_slot);
395 btrfs_item_key(leaf, &leaf_key, 0);
6702ed49 396
5f39d397 397 BUG_ON(memcmp(&parent_key, &leaf_key,
e2fa7227 398 sizeof(struct btrfs_disk_key)));
1d4f8a0c 399 BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
db94535d 400 btrfs_header_bytenr(leaf));
5f39d397
CM
401 }
402#if 0
403 for (i = 0; nritems > 1 && i < nritems - 2; i++) {
404 btrfs_item_key_to_cpu(leaf, &cpukey, i + 1);
405 btrfs_item_key(leaf, &leaf_key, i);
406 if (comp_keys(&leaf_key, &cpukey) >= 0) {
407 btrfs_print_leaf(root, leaf);
408 printk("slot %d offset bad key\n", i);
409 BUG_ON(1);
410 }
411 if (btrfs_item_offset_nr(leaf, i) !=
412 btrfs_item_end_nr(leaf, i + 1)) {
413 btrfs_print_leaf(root, leaf);
414 printk("slot %d offset bad\n", i);
415 BUG_ON(1);
416 }
417 if (i == 0) {
418 if (btrfs_item_offset_nr(leaf, i) +
419 btrfs_item_size_nr(leaf, i) !=
420 BTRFS_LEAF_DATA_SIZE(root)) {
421 btrfs_print_leaf(root, leaf);
422 printk("slot %d first offset bad\n", i);
423 BUG_ON(1);
424 }
425 }
aa5d6bed 426 }
5f39d397
CM
427 if (nritems > 0) {
428 if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) {
429 btrfs_print_leaf(root, leaf);
430 printk("slot %d bad size \n", nritems - 1);
431 BUG_ON(1);
432 }
433 }
434#endif
435 if (slot != 0 && slot < nritems - 1) {
436 btrfs_item_key(leaf, &leaf_key, slot);
437 btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1);
438 if (comp_keys(&leaf_key, &cpukey) <= 0) {
439 btrfs_print_leaf(root, leaf);
440 printk("slot %d offset bad key\n", slot);
441 BUG_ON(1);
442 }
443 if (btrfs_item_offset_nr(leaf, slot - 1) !=
444 btrfs_item_end_nr(leaf, slot)) {
445 btrfs_print_leaf(root, leaf);
446 printk("slot %d offset bad\n", slot);
447 BUG_ON(1);
448 }
8d7be552
CM
449 }
450 if (slot < nritems - 1) {
5f39d397
CM
451 btrfs_item_key(leaf, &leaf_key, slot);
452 btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1);
453 BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0);
454 if (btrfs_item_offset_nr(leaf, slot) !=
455 btrfs_item_end_nr(leaf, slot + 1)) {
456 btrfs_print_leaf(root, leaf);
457 printk("slot %d offset bad\n", slot);
458 BUG_ON(1);
459 }
aa5d6bed 460 }
5f39d397
CM
461 BUG_ON(btrfs_item_offset_nr(leaf, 0) +
462 btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
aa5d6bed
CM
463 return 0;
464}
465
123abc88
CM
466static int check_block(struct btrfs_root *root, struct btrfs_path *path,
467 int level)
aa5d6bed 468{
db94535d 469#if 0
5f39d397 470 struct extent_buffer *buf = path->nodes[level];
5f39d397 471
479965d6
CM
472 if (memcmp_extent_buffer(buf, root->fs_info->fsid,
473 (unsigned long)btrfs_header_fsid(buf),
474 BTRFS_FSID_SIZE)) {
5f39d397 475 printk("warning bad block %Lu\n", buf->start);
db94535d 476 return 1;
5f39d397 477 }
db94535d 478#endif
aa5d6bed 479 if (level == 0)
123abc88
CM
480 return check_leaf(root, path, level);
481 return check_node(root, path, level);
aa5d6bed
CM
482}
483
74123bd7 484/*
5f39d397
CM
485 * search for key in the extent_buffer. The items start at offset p,
486 * and they are item_size apart. There are 'max' items in p.
487 *
74123bd7
CM
488 * the slot in the array is returned via slot, and it points to
489 * the place where you would insert key if it is not found in
490 * the array.
491 *
492 * slot may point to max if the key is bigger than all of the keys
493 */
5f39d397
CM
494static int generic_bin_search(struct extent_buffer *eb, unsigned long p,
495 int item_size, struct btrfs_key *key,
496 int max, int *slot)
be0e5c09
CM
497{
498 int low = 0;
499 int high = max;
500 int mid;
501 int ret;
479965d6 502 struct btrfs_disk_key *tmp = NULL;
5f39d397
CM
503 struct btrfs_disk_key unaligned;
504 unsigned long offset;
505 char *map_token = NULL;
506 char *kaddr = NULL;
507 unsigned long map_start = 0;
508 unsigned long map_len = 0;
479965d6 509 int err;
be0e5c09
CM
510
511 while(low < high) {
512 mid = (low + high) / 2;
5f39d397
CM
513 offset = p + mid * item_size;
514
515 if (!map_token || offset < map_start ||
516 (offset + sizeof(struct btrfs_disk_key)) >
517 map_start + map_len) {
479965d6 518 if (map_token) {
5f39d397 519 unmap_extent_buffer(eb, map_token, KM_USER0);
479965d6
CM
520 map_token = NULL;
521 }
522 err = map_extent_buffer(eb, offset,
523 sizeof(struct btrfs_disk_key),
524 &map_token, &kaddr,
525 &map_start, &map_len, KM_USER0);
526
527 if (!err) {
528 tmp = (struct btrfs_disk_key *)(kaddr + offset -
529 map_start);
530 } else {
531 read_extent_buffer(eb, &unaligned,
532 offset, sizeof(unaligned));
533 tmp = &unaligned;
534 }
5f39d397 535
5f39d397
CM
536 } else {
537 tmp = (struct btrfs_disk_key *)(kaddr + offset -
538 map_start);
539 }
be0e5c09
CM
540 ret = comp_keys(tmp, key);
541
542 if (ret < 0)
543 low = mid + 1;
544 else if (ret > 0)
545 high = mid;
546 else {
547 *slot = mid;
479965d6
CM
548 if (map_token)
549 unmap_extent_buffer(eb, map_token, KM_USER0);
be0e5c09
CM
550 return 0;
551 }
552 }
553 *slot = low;
5f39d397
CM
554 if (map_token)
555 unmap_extent_buffer(eb, map_token, KM_USER0);
be0e5c09
CM
556 return 1;
557}
558
97571fd0
CM
559/*
560 * simple bin_search frontend that does the right thing for
561 * leaves vs nodes
562 */
5f39d397
CM
563static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
564 int level, int *slot)
be0e5c09 565{
5f39d397
CM
566 if (level == 0) {
567 return generic_bin_search(eb,
568 offsetof(struct btrfs_leaf, items),
0783fcfc 569 sizeof(struct btrfs_item),
5f39d397 570 key, btrfs_header_nritems(eb),
7518a238 571 slot);
be0e5c09 572 } else {
5f39d397
CM
573 return generic_bin_search(eb,
574 offsetof(struct btrfs_node, ptrs),
123abc88 575 sizeof(struct btrfs_key_ptr),
5f39d397 576 key, btrfs_header_nritems(eb),
7518a238 577 slot);
be0e5c09
CM
578 }
579 return -1;
580}
581
5f39d397
CM
582static struct extent_buffer *read_node_slot(struct btrfs_root *root,
583 struct extent_buffer *parent, int slot)
bb803951 584{
bb803951
CM
585 if (slot < 0)
586 return NULL;
5f39d397 587 if (slot >= btrfs_header_nritems(parent))
bb803951 588 return NULL;
db94535d
CM
589 return read_tree_block(root, btrfs_node_blockptr(parent, slot),
590 btrfs_level_size(root, btrfs_header_level(parent) - 1));
bb803951
CM
591}
592
e089f05c
CM
593static int balance_level(struct btrfs_trans_handle *trans, struct btrfs_root
594 *root, struct btrfs_path *path, int level)
bb803951 595{
5f39d397
CM
596 struct extent_buffer *right = NULL;
597 struct extent_buffer *mid;
598 struct extent_buffer *left = NULL;
599 struct extent_buffer *parent = NULL;
bb803951
CM
600 int ret = 0;
601 int wret;
602 int pslot;
bb803951 603 int orig_slot = path->slots[level];
54aa1f4d 604 int err_on_enospc = 0;
79f95c82 605 u64 orig_ptr;
bb803951
CM
606
607 if (level == 0)
608 return 0;
609
5f39d397 610 mid = path->nodes[level];
1d4f8a0c 611 orig_ptr = btrfs_node_blockptr(mid, orig_slot);
79f95c82 612
234b63a0 613 if (level < BTRFS_MAX_LEVEL - 1)
5f39d397 614 parent = path->nodes[level + 1];
bb803951
CM
615 pslot = path->slots[level + 1];
616
40689478
CM
617 /*
618 * deal with the case where there is only one pointer in the root
619 * by promoting the node below to a root
620 */
5f39d397
CM
621 if (!parent) {
622 struct extent_buffer *child;
bb803951 623
5f39d397 624 if (btrfs_header_nritems(mid) != 1)
bb803951
CM
625 return 0;
626
627 /* promote the child to a root */
5f39d397 628 child = read_node_slot(root, mid, 0);
bb803951
CM
629 BUG_ON(!child);
630 root->node = child;
631 path->nodes[level] = NULL;
5f39d397
CM
632 clean_tree_block(trans, root, mid);
633 wait_on_tree_block_writeback(root, mid);
bb803951 634 /* once for the path */
5f39d397 635 free_extent_buffer(mid);
db94535d 636 ret = btrfs_free_extent(trans, root, mid->start, mid->len, 1);
bb803951 637 /* once for the root ptr */
5f39d397 638 free_extent_buffer(mid);
db94535d 639 return ret;
bb803951 640 }
5f39d397 641 if (btrfs_header_nritems(mid) >
123abc88 642 BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
bb803951
CM
643 return 0;
644
5f39d397 645 if (btrfs_header_nritems(mid) < 2)
54aa1f4d
CM
646 err_on_enospc = 1;
647
5f39d397
CM
648 left = read_node_slot(root, parent, pslot - 1);
649 if (left) {
650 wret = btrfs_cow_block(trans, root, left,
651 parent, pslot - 1, &left);
54aa1f4d
CM
652 if (wret) {
653 ret = wret;
654 goto enospc;
655 }
2cc58cf2 656 }
5f39d397
CM
657 right = read_node_slot(root, parent, pslot + 1);
658 if (right) {
659 wret = btrfs_cow_block(trans, root, right,
660 parent, pslot + 1, &right);
2cc58cf2
CM
661 if (wret) {
662 ret = wret;
663 goto enospc;
664 }
665 }
666
667 /* first, try to make some room in the middle buffer */
5f39d397
CM
668 if (left) {
669 orig_slot += btrfs_header_nritems(left);
670 wret = push_node_left(trans, root, left, mid);
79f95c82
CM
671 if (wret < 0)
672 ret = wret;
5f39d397 673 if (btrfs_header_nritems(mid) < 2)
54aa1f4d 674 err_on_enospc = 1;
bb803951 675 }
79f95c82
CM
676
677 /*
678 * then try to empty the right most buffer into the middle
679 */
5f39d397
CM
680 if (right) {
681 wret = push_node_left(trans, root, mid, right);
54aa1f4d 682 if (wret < 0 && wret != -ENOSPC)
79f95c82 683 ret = wret;
5f39d397 684 if (btrfs_header_nritems(right) == 0) {
db94535d
CM
685 u64 bytenr = right->start;
686 u32 blocksize = right->len;
687
5f39d397
CM
688 clean_tree_block(trans, root, right);
689 wait_on_tree_block_writeback(root, right);
690 free_extent_buffer(right);
bb803951 691 right = NULL;
e089f05c
CM
692 wret = del_ptr(trans, root, path, level + 1, pslot +
693 1);
bb803951
CM
694 if (wret)
695 ret = wret;
db94535d
CM
696 wret = btrfs_free_extent(trans, root, bytenr,
697 blocksize, 1);
bb803951
CM
698 if (wret)
699 ret = wret;
700 } else {
5f39d397
CM
701 struct btrfs_disk_key right_key;
702 btrfs_node_key(right, &right_key, 0);
703 btrfs_set_node_key(parent, &right_key, pslot + 1);
704 btrfs_mark_buffer_dirty(parent);
bb803951
CM
705 }
706 }
5f39d397 707 if (btrfs_header_nritems(mid) == 1) {
79f95c82
CM
708 /*
709 * we're not allowed to leave a node with one item in the
710 * tree during a delete. A deletion from lower in the tree
711 * could try to delete the only pointer in this node.
712 * So, pull some keys from the left.
713 * There has to be a left pointer at this point because
714 * otherwise we would have pulled some pointers from the
715 * right
716 */
5f39d397
CM
717 BUG_ON(!left);
718 wret = balance_node_right(trans, root, mid, left);
54aa1f4d 719 if (wret < 0) {
79f95c82 720 ret = wret;
54aa1f4d
CM
721 goto enospc;
722 }
79f95c82
CM
723 BUG_ON(wret == 1);
724 }
5f39d397 725 if (btrfs_header_nritems(mid) == 0) {
79f95c82 726 /* we've managed to empty the middle node, drop it */
db94535d
CM
727 u64 bytenr = mid->start;
728 u32 blocksize = mid->len;
5f39d397
CM
729 clean_tree_block(trans, root, mid);
730 wait_on_tree_block_writeback(root, mid);
731 free_extent_buffer(mid);
bb803951 732 mid = NULL;
e089f05c 733 wret = del_ptr(trans, root, path, level + 1, pslot);
bb803951
CM
734 if (wret)
735 ret = wret;
db94535d 736 wret = btrfs_free_extent(trans, root, bytenr, blocksize, 1);
bb803951
CM
737 if (wret)
738 ret = wret;
79f95c82
CM
739 } else {
740 /* update the parent key to reflect our changes */
5f39d397
CM
741 struct btrfs_disk_key mid_key;
742 btrfs_node_key(mid, &mid_key, 0);
743 btrfs_set_node_key(parent, &mid_key, pslot);
744 btrfs_mark_buffer_dirty(parent);
79f95c82 745 }
bb803951 746
79f95c82 747 /* update the path */
5f39d397
CM
748 if (left) {
749 if (btrfs_header_nritems(left) > orig_slot) {
750 extent_buffer_get(left);
751 path->nodes[level] = left;
bb803951
CM
752 path->slots[level + 1] -= 1;
753 path->slots[level] = orig_slot;
5f39d397
CM
754 if (mid)
755 free_extent_buffer(mid);
bb803951 756 } else {
5f39d397 757 orig_slot -= btrfs_header_nritems(left);
bb803951
CM
758 path->slots[level] = orig_slot;
759 }
760 }
79f95c82 761 /* double check we haven't messed things up */
123abc88 762 check_block(root, path, level);
e20d96d6 763 if (orig_ptr !=
5f39d397 764 btrfs_node_blockptr(path->nodes[level], path->slots[level]))
79f95c82 765 BUG();
54aa1f4d 766enospc:
5f39d397
CM
767 if (right)
768 free_extent_buffer(right);
769 if (left)
770 free_extent_buffer(left);
bb803951
CM
771 return ret;
772}
773
e66f709b
CM
774/* returns zero if the push worked, non-zero otherwise */
775static int push_nodes_for_insert(struct btrfs_trans_handle *trans,
776 struct btrfs_root *root,
777 struct btrfs_path *path, int level)
778{
5f39d397
CM
779 struct extent_buffer *right = NULL;
780 struct extent_buffer *mid;
781 struct extent_buffer *left = NULL;
782 struct extent_buffer *parent = NULL;
e66f709b
CM
783 int ret = 0;
784 int wret;
785 int pslot;
786 int orig_slot = path->slots[level];
787 u64 orig_ptr;
788
789 if (level == 0)
790 return 1;
791
5f39d397 792 mid = path->nodes[level];
e66f709b
CM
793 orig_ptr = btrfs_node_blockptr(mid, orig_slot);
794
795 if (level < BTRFS_MAX_LEVEL - 1)
5f39d397 796 parent = path->nodes[level + 1];
e66f709b
CM
797 pslot = path->slots[level + 1];
798
5f39d397 799 if (!parent)
e66f709b 800 return 1;
e66f709b 801
5f39d397 802 left = read_node_slot(root, parent, pslot - 1);
e66f709b
CM
803
804 /* first, try to make some room in the middle buffer */
5f39d397 805 if (left) {
e66f709b 806 u32 left_nr;
5f39d397 807 left_nr = btrfs_header_nritems(left);
33ade1f8
CM
808 if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
809 wret = 1;
810 } else {
5f39d397
CM
811 ret = btrfs_cow_block(trans, root, left, parent,
812 pslot - 1, &left);
54aa1f4d
CM
813 if (ret)
814 wret = 1;
815 else {
54aa1f4d 816 wret = push_node_left(trans, root,
5f39d397 817 left, mid);
54aa1f4d 818 }
33ade1f8 819 }
e66f709b
CM
820 if (wret < 0)
821 ret = wret;
822 if (wret == 0) {
5f39d397 823 struct btrfs_disk_key disk_key;
e66f709b 824 orig_slot += left_nr;
5f39d397
CM
825 btrfs_node_key(mid, &disk_key, 0);
826 btrfs_set_node_key(parent, &disk_key, pslot);
827 btrfs_mark_buffer_dirty(parent);
828 if (btrfs_header_nritems(left) > orig_slot) {
829 path->nodes[level] = left;
e66f709b
CM
830 path->slots[level + 1] -= 1;
831 path->slots[level] = orig_slot;
5f39d397 832 free_extent_buffer(mid);
e66f709b
CM
833 } else {
834 orig_slot -=
5f39d397 835 btrfs_header_nritems(left);
e66f709b 836 path->slots[level] = orig_slot;
5f39d397 837 free_extent_buffer(left);
e66f709b 838 }
e66f709b
CM
839 return 0;
840 }
5f39d397 841 free_extent_buffer(left);
e66f709b 842 }
5f39d397 843 right= read_node_slot(root, parent, pslot + 1);
e66f709b
CM
844
845 /*
846 * then try to empty the right most buffer into the middle
847 */
5f39d397 848 if (right) {
33ade1f8 849 u32 right_nr;
5f39d397 850 right_nr = btrfs_header_nritems(right);
33ade1f8
CM
851 if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
852 wret = 1;
853 } else {
5f39d397
CM
854 ret = btrfs_cow_block(trans, root, right,
855 parent, pslot + 1,
856 &right);
54aa1f4d
CM
857 if (ret)
858 wret = 1;
859 else {
54aa1f4d 860 wret = balance_node_right(trans, root,
5f39d397 861 right, mid);
54aa1f4d 862 }
33ade1f8 863 }
e66f709b
CM
864 if (wret < 0)
865 ret = wret;
866 if (wret == 0) {
5f39d397
CM
867 struct btrfs_disk_key disk_key;
868
869 btrfs_node_key(right, &disk_key, 0);
870 btrfs_set_node_key(parent, &disk_key, pslot + 1);
871 btrfs_mark_buffer_dirty(parent);
872
873 if (btrfs_header_nritems(mid) <= orig_slot) {
874 path->nodes[level] = right;
e66f709b
CM
875 path->slots[level + 1] += 1;
876 path->slots[level] = orig_slot -
5f39d397
CM
877 btrfs_header_nritems(mid);
878 free_extent_buffer(mid);
e66f709b 879 } else {
5f39d397 880 free_extent_buffer(right);
e66f709b 881 }
e66f709b
CM
882 return 0;
883 }
5f39d397 884 free_extent_buffer(right);
e66f709b 885 }
e66f709b
CM
886 return 1;
887}
888
3c69faec
CM
889/*
890 * readahead one full node of leaves
891 */
892static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path,
6702ed49 893 int level, int slot)
3c69faec 894{
db94535d
CM
895 return;
896#if 0
5f39d397 897 struct extent_buffer *node;
3c69faec
CM
898 int i;
899 u32 nritems;
db94535d 900 u64 bytenr;
3c69faec
CM
901 u64 search;
902 u64 cluster_start;
903 int ret;
904 int nread = 0;
905 int direction = path->reada;
db94535d 906 int level;
3c69faec
CM
907 struct radix_tree_root found;
908 unsigned long gang[8];
5f39d397 909 struct extent_buffer *eb;
3c69faec 910
db94535d 911
6702ed49
CM
912 if (level == 0)
913 return;
914
915 if (!path->nodes[level])
3c69faec
CM
916 return;
917
5f39d397 918 node = path->nodes[level];
3c69faec 919 search = btrfs_node_blockptr(node, slot);
5f39d397
CM
920 eb = btrfs_find_tree_block(root, search);
921 if (eb) {
922 free_extent_buffer(eb);
3c69faec
CM
923 return;
924 }
925
926 init_bit_radix(&found);
5f39d397 927 nritems = btrfs_header_nritems(node);
db94535d 928 level = btrfs_header_level(node) - 1;
3c69faec 929 for (i = slot; i < nritems; i++) {
db94535d 930 bytenr = btrfs_node_blockptr(node, i);
3c69faec
CM
931 set_radix_bit(&found, blocknr);
932 }
933 if (direction > 0) {
934 cluster_start = search - 4;
935 if (cluster_start > search)
936 cluster_start = 0;
937 } else
938 cluster_start = search + 4;
939 while(1) {
940 ret = find_first_radix_bit(&found, gang, 0, ARRAY_SIZE(gang));
941 if (!ret)
942 break;
943 for (i = 0; i < ret; i++) {
944 blocknr = gang[i];
945 clear_radix_bit(&found, blocknr);
2cc58cf2 946 if (path->reada == 1 && nread > 16)
3c69faec 947 continue;
f2183bde 948 if (close_blocks(cluster_start, blocknr)) {
3c69faec
CM
949 readahead_tree_block(root, blocknr);
950 nread++;
3c69faec 951 cluster_start = blocknr;
3c69faec
CM
952 }
953 }
954 }
db94535d 955#endif
3c69faec 956}
74123bd7
CM
957/*
958 * look for key in the tree. path is filled in with nodes along the way
959 * if key is found, we return zero and you can find the item in the leaf
960 * level of the path (level 0)
961 *
962 * If the key isn't found, the path points to the slot where it should
aa5d6bed
CM
963 * be inserted, and 1 is returned. If there are other errors during the
964 * search a negative error number is returned.
97571fd0
CM
965 *
966 * if ins_len > 0, nodes and leaves will be split as we walk down the
967 * tree. if ins_len < 0, nodes will be merged as we walk down the tree (if
968 * possible)
74123bd7 969 */
e089f05c
CM
970int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root
971 *root, struct btrfs_key *key, struct btrfs_path *p, int
972 ins_len, int cow)
be0e5c09 973{
5f39d397 974 struct extent_buffer *b;
db94535d 975 u64 bytenr;
be0e5c09
CM
976 int slot;
977 int ret;
978 int level;
3c69faec 979 int should_reada = p->reada;
9f3a7427
CM
980 u8 lowest_level = 0;
981
6702ed49
CM
982 lowest_level = p->lowest_level;
983 WARN_ON(lowest_level && ins_len);
22b0ebda
CM
984 WARN_ON(p->nodes[0] != NULL);
985 WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex));
bb803951
CM
986again:
987 b = root->node;
5f39d397 988 extent_buffer_get(b);
eb60ceac 989 while (b) {
5f39d397 990 level = btrfs_header_level(b);
02217ed2
CM
991 if (cow) {
992 int wret;
e20d96d6
CM
993 wret = btrfs_cow_block(trans, root, b,
994 p->nodes[level + 1],
995 p->slots[level + 1],
252c38f0 996 &b);
54aa1f4d 997 if (wret) {
5f39d397 998 free_extent_buffer(b);
54aa1f4d
CM
999 return wret;
1000 }
02217ed2
CM
1001 }
1002 BUG_ON(!cow && ins_len);
5f39d397 1003 if (level != btrfs_header_level(b))
2c90e5d6 1004 WARN_ON(1);
5f39d397 1005 level = btrfs_header_level(b);
eb60ceac 1006 p->nodes[level] = b;
123abc88 1007 ret = check_block(root, p, level);
aa5d6bed
CM
1008 if (ret)
1009 return -1;
5f39d397
CM
1010 ret = bin_search(b, key, level, &slot);
1011 if (level != 0) {
be0e5c09
CM
1012 if (ret && slot > 0)
1013 slot -= 1;
1014 p->slots[level] = slot;
5f39d397 1015 if (ins_len > 0 && btrfs_header_nritems(b) >=
d4dbff95 1016 BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
e089f05c 1017 int sret = split_node(trans, root, p, level);
5c680ed6
CM
1018 BUG_ON(sret > 0);
1019 if (sret)
1020 return sret;
1021 b = p->nodes[level];
5c680ed6 1022 slot = p->slots[level];
bb803951 1023 } else if (ins_len < 0) {
e089f05c
CM
1024 int sret = balance_level(trans, root, p,
1025 level);
bb803951
CM
1026 if (sret)
1027 return sret;
1028 b = p->nodes[level];
f510cfec
CM
1029 if (!b) {
1030 btrfs_release_path(NULL, p);
bb803951 1031 goto again;
f510cfec 1032 }
bb803951 1033 slot = p->slots[level];
5f39d397 1034 BUG_ON(btrfs_header_nritems(b) == 1);
5c680ed6 1035 }
9f3a7427
CM
1036 /* this is only true while dropping a snapshot */
1037 if (level == lowest_level)
1038 break;
db94535d 1039 bytenr = btrfs_node_blockptr(b, slot);
6702ed49
CM
1040 if (should_reada)
1041 reada_for_search(root, p, level, slot);
db94535d
CM
1042 b = read_tree_block(root, bytenr,
1043 btrfs_level_size(root, level - 1));
be0e5c09
CM
1044 } else {
1045 p->slots[level] = slot;
5f39d397 1046 if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
0783fcfc 1047 sizeof(struct btrfs_item) + ins_len) {
d4dbff95
CM
1048 int sret = split_leaf(trans, root, key,
1049 p, ins_len);
5c680ed6
CM
1050 BUG_ON(sret > 0);
1051 if (sret)
1052 return sret;
1053 }
be0e5c09
CM
1054 return ret;
1055 }
1056 }
aa5d6bed 1057 return 1;
be0e5c09
CM
1058}
1059
74123bd7
CM
1060/*
1061 * adjust the pointers going up the tree, starting at level
1062 * making sure the right key of each node is points to 'key'.
1063 * This is used after shifting pointers to the left, so it stops
1064 * fixing up pointers when a given leaf/node is not in slot 0 of the
1065 * higher levels
aa5d6bed
CM
1066 *
1067 * If this fails to write a tree block, it returns -1, but continues
1068 * fixing up the blocks in ram so the tree is consistent.
74123bd7 1069 */
5f39d397
CM
1070static int fixup_low_keys(struct btrfs_trans_handle *trans,
1071 struct btrfs_root *root, struct btrfs_path *path,
1072 struct btrfs_disk_key *key, int level)
be0e5c09
CM
1073{
1074 int i;
aa5d6bed 1075 int ret = 0;
5f39d397
CM
1076 struct extent_buffer *t;
1077
234b63a0 1078 for (i = level; i < BTRFS_MAX_LEVEL; i++) {
be0e5c09 1079 int tslot = path->slots[i];
eb60ceac 1080 if (!path->nodes[i])
be0e5c09 1081 break;
5f39d397
CM
1082 t = path->nodes[i];
1083 btrfs_set_node_key(t, key, tslot);
d6025579 1084 btrfs_mark_buffer_dirty(path->nodes[i]);
be0e5c09
CM
1085 if (tslot != 0)
1086 break;
1087 }
aa5d6bed 1088 return ret;
be0e5c09
CM
1089}
1090
74123bd7
CM
1091/*
1092 * try to push data from one node into the next node left in the
79f95c82 1093 * tree.
aa5d6bed
CM
1094 *
1095 * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
1096 * error, and > 0 if there was no room in the left hand block.
74123bd7 1097 */
e089f05c 1098static int push_node_left(struct btrfs_trans_handle *trans, struct btrfs_root
5f39d397
CM
1099 *root, struct extent_buffer *dst,
1100 struct extent_buffer *src)
be0e5c09 1101{
be0e5c09 1102 int push_items = 0;
bb803951
CM
1103 int src_nritems;
1104 int dst_nritems;
aa5d6bed 1105 int ret = 0;
be0e5c09 1106
5f39d397
CM
1107 src_nritems = btrfs_header_nritems(src);
1108 dst_nritems = btrfs_header_nritems(dst);
123abc88 1109 push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
54aa1f4d 1110
eb60ceac 1111 if (push_items <= 0) {
be0e5c09 1112 return 1;
eb60ceac 1113 }
be0e5c09 1114
bb803951 1115 if (src_nritems < push_items)
79f95c82
CM
1116 push_items = src_nritems;
1117
5f39d397
CM
1118 copy_extent_buffer(dst, src,
1119 btrfs_node_key_ptr_offset(dst_nritems),
1120 btrfs_node_key_ptr_offset(0),
1121 push_items * sizeof(struct btrfs_key_ptr));
1122
bb803951 1123 if (push_items < src_nritems) {
5f39d397
CM
1124 memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
1125 btrfs_node_key_ptr_offset(push_items),
1126 (src_nritems - push_items) *
1127 sizeof(struct btrfs_key_ptr));
1128 }
1129 btrfs_set_header_nritems(src, src_nritems - push_items);
1130 btrfs_set_header_nritems(dst, dst_nritems + push_items);
1131 btrfs_mark_buffer_dirty(src);
1132 btrfs_mark_buffer_dirty(dst);
79f95c82
CM
1133 return ret;
1134}
1135
1136/*
1137 * try to push data from one node into the next node right in the
1138 * tree.
1139 *
1140 * returns 0 if some ptrs were pushed, < 0 if there was some horrible
1141 * error, and > 0 if there was no room in the right hand block.
1142 *
1143 * this will only push up to 1/2 the contents of the left node over
1144 */
5f39d397
CM
1145static int balance_node_right(struct btrfs_trans_handle *trans,
1146 struct btrfs_root *root,
1147 struct extent_buffer *dst,
1148 struct extent_buffer *src)
79f95c82 1149{
79f95c82
CM
1150 int push_items = 0;
1151 int max_push;
1152 int src_nritems;
1153 int dst_nritems;
1154 int ret = 0;
79f95c82 1155
5f39d397
CM
1156 src_nritems = btrfs_header_nritems(src);
1157 dst_nritems = btrfs_header_nritems(dst);
123abc88 1158 push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
5f39d397 1159 if (push_items <= 0)
79f95c82 1160 return 1;
79f95c82
CM
1161
1162 max_push = src_nritems / 2 + 1;
1163 /* don't try to empty the node */
252c38f0 1164 if (max_push >= src_nritems)
79f95c82 1165 return 1;
252c38f0 1166
79f95c82
CM
1167 if (max_push < push_items)
1168 push_items = max_push;
1169
5f39d397
CM
1170 memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
1171 btrfs_node_key_ptr_offset(0),
1172 (dst_nritems) *
1173 sizeof(struct btrfs_key_ptr));
d6025579 1174
5f39d397
CM
1175 copy_extent_buffer(dst, src,
1176 btrfs_node_key_ptr_offset(0),
1177 btrfs_node_key_ptr_offset(src_nritems - push_items),
1178 push_items * sizeof(struct btrfs_key_ptr));
79f95c82 1179
5f39d397
CM
1180 btrfs_set_header_nritems(src, src_nritems - push_items);
1181 btrfs_set_header_nritems(dst, dst_nritems + push_items);
79f95c82 1182
5f39d397
CM
1183 btrfs_mark_buffer_dirty(src);
1184 btrfs_mark_buffer_dirty(dst);
aa5d6bed 1185 return ret;
be0e5c09
CM
1186}
1187
97571fd0
CM
1188/*
1189 * helper function to insert a new root level in the tree.
1190 * A new node is allocated, and a single item is inserted to
1191 * point to the existing root
aa5d6bed
CM
1192 *
1193 * returns zero on success or < 0 on failure.
97571fd0 1194 */
5f39d397
CM
1195static int insert_new_root(struct btrfs_trans_handle *trans,
1196 struct btrfs_root *root,
1197 struct btrfs_path *path, int level)
5c680ed6 1198{
5f39d397
CM
1199 struct extent_buffer *lower;
1200 struct extent_buffer *c;
1201 struct btrfs_disk_key lower_key;
5c680ed6
CM
1202
1203 BUG_ON(path->nodes[level]);
1204 BUG_ON(path->nodes[level-1] != root->node);
1205
db94535d
CM
1206 c = btrfs_alloc_free_block(trans, root, root->nodesize,
1207 root->node->start, 0);
5f39d397
CM
1208 if (IS_ERR(c))
1209 return PTR_ERR(c);
1210 memset_extent_buffer(c, 0, 0, root->nodesize);
1211 btrfs_set_header_nritems(c, 1);
1212 btrfs_set_header_level(c, level);
db94535d 1213 btrfs_set_header_bytenr(c, c->start);
5f39d397
CM
1214 btrfs_set_header_generation(c, trans->transid);
1215 btrfs_set_header_owner(c, root->root_key.objectid);
1216 lower = path->nodes[level-1];
1217
1218 write_extent_buffer(c, root->fs_info->fsid,
1219 (unsigned long)btrfs_header_fsid(c),
1220 BTRFS_FSID_SIZE);
1221 if (level == 1)
1222 btrfs_item_key(lower, &lower_key, 0);
5c680ed6 1223 else
5f39d397
CM
1224 btrfs_node_key(lower, &lower_key, 0);
1225 btrfs_set_node_key(c, &lower_key, 0);
db94535d 1226 btrfs_set_node_blockptr(c, 0, lower->start);
d5719762 1227
5f39d397 1228 btrfs_mark_buffer_dirty(c);
d5719762 1229
5c680ed6 1230 /* the super has an extra ref to root->node */
5f39d397
CM
1231 free_extent_buffer(root->node);
1232 root->node = c;
1233 extent_buffer_get(c);
1234 path->nodes[level] = c;
5c680ed6
CM
1235 path->slots[level] = 0;
1236 return 0;
1237}
1238
74123bd7
CM
1239/*
1240 * worker function to insert a single pointer in a node.
1241 * the node should have enough room for the pointer already
97571fd0 1242 *
74123bd7
CM
1243 * slot and level indicate where you want the key to go, and
1244 * blocknr is the block the key points to.
aa5d6bed
CM
1245 *
1246 * returns zero on success and < 0 on any error
74123bd7 1247 */
e089f05c
CM
1248static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
1249 *root, struct btrfs_path *path, struct btrfs_disk_key
db94535d 1250 *key, u64 bytenr, int slot, int level)
74123bd7 1251{
5f39d397 1252 struct extent_buffer *lower;
74123bd7 1253 int nritems;
5c680ed6
CM
1254
1255 BUG_ON(!path->nodes[level]);
5f39d397
CM
1256 lower = path->nodes[level];
1257 nritems = btrfs_header_nritems(lower);
74123bd7
CM
1258 if (slot > nritems)
1259 BUG();
123abc88 1260 if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
74123bd7
CM
1261 BUG();
1262 if (slot != nritems) {
5f39d397
CM
1263 memmove_extent_buffer(lower,
1264 btrfs_node_key_ptr_offset(slot + 1),
1265 btrfs_node_key_ptr_offset(slot),
d6025579 1266 (nritems - slot) * sizeof(struct btrfs_key_ptr));
74123bd7 1267 }
5f39d397 1268 btrfs_set_node_key(lower, key, slot);
db94535d 1269 btrfs_set_node_blockptr(lower, slot, bytenr);
5f39d397
CM
1270 btrfs_set_header_nritems(lower, nritems + 1);
1271 btrfs_mark_buffer_dirty(lower);
74123bd7
CM
1272 return 0;
1273}
1274
97571fd0
CM
1275/*
1276 * split the node at the specified level in path in two.
1277 * The path is corrected to point to the appropriate node after the split
1278 *
1279 * Before splitting this tries to make some room in the node by pushing
1280 * left and right, if either one works, it returns right away.
aa5d6bed
CM
1281 *
1282 * returns 0 on success and < 0 on failure
97571fd0 1283 */
e089f05c
CM
1284static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
1285 *root, struct btrfs_path *path, int level)
be0e5c09 1286{
5f39d397
CM
1287 struct extent_buffer *c;
1288 struct extent_buffer *split;
1289 struct btrfs_disk_key disk_key;
be0e5c09 1290 int mid;
5c680ed6 1291 int ret;
aa5d6bed 1292 int wret;
7518a238 1293 u32 c_nritems;
eb60ceac 1294
5f39d397
CM
1295 c = path->nodes[level];
1296 if (c == root->node) {
5c680ed6 1297 /* trying to split the root, lets make a new one */
e089f05c 1298 ret = insert_new_root(trans, root, path, level + 1);
5c680ed6
CM
1299 if (ret)
1300 return ret;
e66f709b
CM
1301 } else {
1302 ret = push_nodes_for_insert(trans, root, path, level);
5f39d397
CM
1303 c = path->nodes[level];
1304 if (!ret && btrfs_header_nritems(c) <
e66f709b
CM
1305 BTRFS_NODEPTRS_PER_BLOCK(root) - 1)
1306 return 0;
54aa1f4d
CM
1307 if (ret < 0)
1308 return ret;
be0e5c09 1309 }
e66f709b 1310
5f39d397 1311 c_nritems = btrfs_header_nritems(c);
db94535d
CM
1312 split = btrfs_alloc_free_block(trans, root, root->nodesize,
1313 c->start, 0);
5f39d397
CM
1314 if (IS_ERR(split))
1315 return PTR_ERR(split);
1316
1317 btrfs_set_header_flags(split, btrfs_header_flags(c));
1318 btrfs_set_header_level(split, btrfs_header_level(c));
db94535d 1319 btrfs_set_header_bytenr(split, split->start);
5f39d397
CM
1320 btrfs_set_header_generation(split, trans->transid);
1321 btrfs_set_header_owner(split, root->root_key.objectid);
1322 write_extent_buffer(split, root->fs_info->fsid,
1323 (unsigned long)btrfs_header_fsid(split),
1324 BTRFS_FSID_SIZE);
54aa1f4d 1325
7518a238 1326 mid = (c_nritems + 1) / 2;
5f39d397
CM
1327
1328 copy_extent_buffer(split, c,
1329 btrfs_node_key_ptr_offset(0),
1330 btrfs_node_key_ptr_offset(mid),
1331 (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
1332 btrfs_set_header_nritems(split, c_nritems - mid);
1333 btrfs_set_header_nritems(c, mid);
aa5d6bed
CM
1334 ret = 0;
1335
5f39d397
CM
1336 btrfs_mark_buffer_dirty(c);
1337 btrfs_mark_buffer_dirty(split);
1338
1339 btrfs_node_key(split, &disk_key, 0);
db94535d 1340 wret = insert_ptr(trans, root, path, &disk_key, split->start,
5f39d397 1341 path->slots[level + 1] + 1,
123abc88 1342 level + 1);
aa5d6bed
CM
1343 if (wret)
1344 ret = wret;
1345
5de08d7d 1346 if (path->slots[level] >= mid) {
5c680ed6 1347 path->slots[level] -= mid;
5f39d397
CM
1348 free_extent_buffer(c);
1349 path->nodes[level] = split;
5c680ed6
CM
1350 path->slots[level + 1] += 1;
1351 } else {
5f39d397 1352 free_extent_buffer(split);
be0e5c09 1353 }
aa5d6bed 1354 return ret;
be0e5c09
CM
1355}
1356
74123bd7
CM
1357/*
1358 * how many bytes are required to store the items in a leaf. start
1359 * and nr indicate which items in the leaf to check. This totals up the
1360 * space used both by the item structs and the item data
1361 */
5f39d397 1362static int leaf_space_used(struct extent_buffer *l, int start, int nr)
be0e5c09
CM
1363{
1364 int data_len;
5f39d397 1365 int nritems = btrfs_header_nritems(l);
d4dbff95 1366 int end = min(nritems, start + nr) - 1;
be0e5c09
CM
1367
1368 if (!nr)
1369 return 0;
5f39d397
CM
1370 data_len = btrfs_item_end_nr(l, start);
1371 data_len = data_len - btrfs_item_offset_nr(l, end);
0783fcfc 1372 data_len += sizeof(struct btrfs_item) * nr;
d4dbff95 1373 WARN_ON(data_len < 0);
be0e5c09
CM
1374 return data_len;
1375}
1376
d4dbff95
CM
1377/*
1378 * The space between the end of the leaf items and
1379 * the start of the leaf data. IOW, how much room
1380 * the leaf has left for both items and data
1381 */
5f39d397 1382int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
d4dbff95 1383{
5f39d397
CM
1384 int nritems = btrfs_header_nritems(leaf);
1385 int ret;
1386 ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
1387 if (ret < 0) {
1388 printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
1389 ret, BTRFS_LEAF_DATA_SIZE(root),
1390 leaf_space_used(leaf, 0, nritems), nritems);
1391 }
1392 return ret;
d4dbff95
CM
1393}
1394
00ec4c51
CM
1395/*
1396 * push some data in the path leaf to the right, trying to free up at
1397 * least data_size bytes. returns zero if the push worked, nonzero otherwise
aa5d6bed
CM
1398 *
1399 * returns 1 if the push failed because the other node didn't have enough
1400 * room, 0 if everything worked out and < 0 if there were major errors.
00ec4c51 1401 */
e089f05c
CM
1402static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
1403 *root, struct btrfs_path *path, int data_size)
00ec4c51 1404{
5f39d397
CM
1405 struct extent_buffer *left = path->nodes[0];
1406 struct extent_buffer *right;
1407 struct extent_buffer *upper;
1408 struct btrfs_disk_key disk_key;
00ec4c51
CM
1409 int slot;
1410 int i;
1411 int free_space;
1412 int push_space = 0;
1413 int push_items = 0;
0783fcfc 1414 struct btrfs_item *item;
7518a238
CM
1415 u32 left_nritems;
1416 u32 right_nritems;
5f39d397 1417 u32 data_end;
db94535d 1418 u32 this_item_size;
54aa1f4d 1419 int ret;
00ec4c51
CM
1420
1421 slot = path->slots[1];
1422 if (!path->nodes[1]) {
1423 return 1;
1424 }
1425 upper = path->nodes[1];
5f39d397 1426 if (slot >= btrfs_header_nritems(upper) - 1)
00ec4c51 1427 return 1;
5f39d397 1428
db94535d
CM
1429 right = read_tree_block(root, btrfs_node_blockptr(upper, slot + 1),
1430 root->leafsize);
123abc88 1431 free_space = btrfs_leaf_free_space(root, right);
0783fcfc 1432 if (free_space < data_size + sizeof(struct btrfs_item)) {
5f39d397 1433 free_extent_buffer(right);
00ec4c51
CM
1434 return 1;
1435 }
5f39d397 1436
02217ed2 1437 /* cow and double check */
5f39d397
CM
1438 ret = btrfs_cow_block(trans, root, right, upper,
1439 slot + 1, &right);
54aa1f4d 1440 if (ret) {
5f39d397 1441 free_extent_buffer(right);
54aa1f4d
CM
1442 return 1;
1443 }
123abc88 1444 free_space = btrfs_leaf_free_space(root, right);
0783fcfc 1445 if (free_space < data_size + sizeof(struct btrfs_item)) {
5f39d397 1446 free_extent_buffer(right);
02217ed2
CM
1447 return 1;
1448 }
1449
5f39d397 1450 left_nritems = btrfs_header_nritems(left);
a429e513 1451 if (left_nritems == 0) {
5f39d397 1452 free_extent_buffer(right);
a429e513
CM
1453 return 1;
1454 }
5f39d397 1455
a429e513 1456 for (i = left_nritems - 1; i >= 1; i--) {
5f39d397 1457 item = btrfs_item_nr(left, i);
db94535d 1458
00ec4c51
CM
1459 if (path->slots[0] == i)
1460 push_space += data_size + sizeof(*item);
db94535d
CM
1461
1462 if (!left->map_token) {
1463 map_extent_buffer(left, (unsigned long)item,
1464 sizeof(struct btrfs_item),
1465 &left->map_token, &left->kaddr,
1466 &left->map_start, &left->map_len,
1467 KM_USER1);
1468 }
1469
1470 this_item_size = btrfs_item_size(left, item);
1471 if (this_item_size + sizeof(*item) + push_space > free_space)
00ec4c51
CM
1472 break;
1473 push_items++;
db94535d
CM
1474 push_space += this_item_size + sizeof(*item);
1475 }
1476 if (left->map_token) {
1477 unmap_extent_buffer(left, left->map_token, KM_USER1);
1478 left->map_token = NULL;
00ec4c51 1479 }
5f39d397 1480
00ec4c51 1481 if (push_items == 0) {
5f39d397 1482 free_extent_buffer(right);
00ec4c51
CM
1483 return 1;
1484 }
5f39d397 1485
a429e513
CM
1486 if (push_items == left_nritems)
1487 WARN_ON(1);
5f39d397 1488
00ec4c51 1489 /* push left to right */
5f39d397
CM
1490 right_nritems = btrfs_header_nritems(right);
1491 push_space = btrfs_item_end_nr(left, left_nritems - push_items);
123abc88 1492 push_space -= leaf_data_end(root, left);
5f39d397 1493
00ec4c51 1494 /* make room in the right data area */
5f39d397
CM
1495 data_end = leaf_data_end(root, right);
1496 memmove_extent_buffer(right,
1497 btrfs_leaf_data(right) + data_end - push_space,
1498 btrfs_leaf_data(right) + data_end,
1499 BTRFS_LEAF_DATA_SIZE(root) - data_end);
1500
00ec4c51 1501 /* copy from the left data area */
5f39d397 1502 copy_extent_buffer(right, left, btrfs_leaf_data(right) +
d6025579
CM
1503 BTRFS_LEAF_DATA_SIZE(root) - push_space,
1504 btrfs_leaf_data(left) + leaf_data_end(root, left),
1505 push_space);
5f39d397
CM
1506
1507 memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
1508 btrfs_item_nr_offset(0),
1509 right_nritems * sizeof(struct btrfs_item));
1510
00ec4c51 1511 /* copy the items from left to right */
5f39d397
CM
1512 copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
1513 btrfs_item_nr_offset(left_nritems - push_items),
1514 push_items * sizeof(struct btrfs_item));
00ec4c51
CM
1515
1516 /* update the item pointers */
7518a238 1517 right_nritems += push_items;
5f39d397 1518 btrfs_set_header_nritems(right, right_nritems);
123abc88 1519 push_space = BTRFS_LEAF_DATA_SIZE(root);
db94535d 1520
7518a238 1521 for (i = 0; i < right_nritems; i++) {
5f39d397 1522 item = btrfs_item_nr(right, i);
db94535d
CM
1523 if (!right->map_token) {
1524 map_extent_buffer(right, (unsigned long)item,
1525 sizeof(struct btrfs_item),
1526 &right->map_token, &right->kaddr,
1527 &right->map_start, &right->map_len,
1528 KM_USER1);
1529 }
1530 push_space -= btrfs_item_size(right, item);
1531 btrfs_set_item_offset(right, item, push_space);
1532 }
1533
1534 if (right->map_token) {
1535 unmap_extent_buffer(right, right->map_token, KM_USER1);
1536 right->map_token = NULL;
00ec4c51 1537 }
7518a238 1538 left_nritems -= push_items;
5f39d397 1539 btrfs_set_header_nritems(left, left_nritems);
00ec4c51 1540
5f39d397
CM
1541 btrfs_mark_buffer_dirty(left);
1542 btrfs_mark_buffer_dirty(right);
a429e513 1543
5f39d397
CM
1544 btrfs_item_key(right, &disk_key, 0);
1545 btrfs_set_node_key(upper, &disk_key, slot + 1);
d6025579 1546 btrfs_mark_buffer_dirty(upper);
02217ed2 1547
00ec4c51 1548 /* then fixup the leaf pointer in the path */
7518a238
CM
1549 if (path->slots[0] >= left_nritems) {
1550 path->slots[0] -= left_nritems;
5f39d397
CM
1551 free_extent_buffer(path->nodes[0]);
1552 path->nodes[0] = right;
00ec4c51
CM
1553 path->slots[1] += 1;
1554 } else {
5f39d397 1555 free_extent_buffer(right);
00ec4c51
CM
1556 }
1557 return 0;
1558}
74123bd7
CM
1559/*
1560 * push some data in the path leaf to the left, trying to free up at
1561 * least data_size bytes. returns zero if the push worked, nonzero otherwise
1562 */
e089f05c
CM
1563static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
1564 *root, struct btrfs_path *path, int data_size)
be0e5c09 1565{
5f39d397
CM
1566 struct btrfs_disk_key disk_key;
1567 struct extent_buffer *right = path->nodes[0];
1568 struct extent_buffer *left;
be0e5c09
CM
1569 int slot;
1570 int i;
1571 int free_space;
1572 int push_space = 0;
1573 int push_items = 0;
0783fcfc 1574 struct btrfs_item *item;
7518a238 1575 u32 old_left_nritems;
5f39d397 1576 u32 right_nritems;
aa5d6bed
CM
1577 int ret = 0;
1578 int wret;
db94535d
CM
1579 u32 this_item_size;
1580 u32 old_left_item_size;
be0e5c09
CM
1581
1582 slot = path->slots[1];
5f39d397 1583 if (slot == 0)
be0e5c09 1584 return 1;
5f39d397 1585 if (!path->nodes[1])
be0e5c09 1586 return 1;
5f39d397
CM
1587
1588 left = read_tree_block(root, btrfs_node_blockptr(path->nodes[1],
db94535d 1589 slot - 1), root->leafsize);
123abc88 1590 free_space = btrfs_leaf_free_space(root, left);
0783fcfc 1591 if (free_space < data_size + sizeof(struct btrfs_item)) {
5f39d397 1592 free_extent_buffer(left);
be0e5c09
CM
1593 return 1;
1594 }
02217ed2
CM
1595
1596 /* cow and double check */
5f39d397
CM
1597 ret = btrfs_cow_block(trans, root, left,
1598 path->nodes[1], slot - 1, &left);
54aa1f4d
CM
1599 if (ret) {
1600 /* we hit -ENOSPC, but it isn't fatal here */
5f39d397 1601 free_extent_buffer(left);
54aa1f4d
CM
1602 return 1;
1603 }
123abc88 1604 free_space = btrfs_leaf_free_space(root, left);
0783fcfc 1605 if (free_space < data_size + sizeof(struct btrfs_item)) {
5f39d397 1606 free_extent_buffer(left);
02217ed2
CM
1607 return 1;
1608 }
1609
5f39d397
CM
1610 right_nritems = btrfs_header_nritems(right);
1611 if (right_nritems == 0) {
1612 free_extent_buffer(left);
a429e513
CM
1613 return 1;
1614 }
1615
5f39d397
CM
1616 for (i = 0; i < right_nritems - 1; i++) {
1617 item = btrfs_item_nr(right, i);
db94535d
CM
1618 if (!right->map_token) {
1619 map_extent_buffer(right, (unsigned long)item,
1620 sizeof(struct btrfs_item),
1621 &right->map_token, &right->kaddr,
1622 &right->map_start, &right->map_len,
1623 KM_USER1);
1624 }
1625
be0e5c09
CM
1626 if (path->slots[0] == i)
1627 push_space += data_size + sizeof(*item);
db94535d
CM
1628
1629 this_item_size = btrfs_item_size(right, item);
1630 if (this_item_size + sizeof(*item) + push_space > free_space)
be0e5c09 1631 break;
db94535d 1632
be0e5c09 1633 push_items++;
db94535d
CM
1634 push_space += this_item_size + sizeof(*item);
1635 }
1636
1637 if (right->map_token) {
1638 unmap_extent_buffer(right, right->map_token, KM_USER1);
1639 right->map_token = NULL;
be0e5c09 1640 }
db94535d 1641
be0e5c09 1642 if (push_items == 0) {
5f39d397 1643 free_extent_buffer(left);
be0e5c09
CM
1644 return 1;
1645 }
5f39d397 1646 if (push_items == btrfs_header_nritems(right))
a429e513 1647 WARN_ON(1);
5f39d397 1648
be0e5c09 1649 /* push data from right to left */
5f39d397
CM
1650 copy_extent_buffer(left, right,
1651 btrfs_item_nr_offset(btrfs_header_nritems(left)),
1652 btrfs_item_nr_offset(0),
1653 push_items * sizeof(struct btrfs_item));
1654
123abc88 1655 push_space = BTRFS_LEAF_DATA_SIZE(root) -
5f39d397
CM
1656 btrfs_item_offset_nr(right, push_items -1);
1657
1658 copy_extent_buffer(left, right, btrfs_leaf_data(left) +
d6025579
CM
1659 leaf_data_end(root, left) - push_space,
1660 btrfs_leaf_data(right) +
5f39d397 1661 btrfs_item_offset_nr(right, push_items - 1),
d6025579 1662 push_space);
5f39d397 1663 old_left_nritems = btrfs_header_nritems(left);
eb60ceac
CM
1664 BUG_ON(old_left_nritems < 0);
1665
db94535d 1666 old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
0783fcfc 1667 for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
5f39d397 1668 u32 ioff;
db94535d 1669
5f39d397 1670 item = btrfs_item_nr(left, i);
db94535d
CM
1671 if (!left->map_token) {
1672 map_extent_buffer(left, (unsigned long)item,
1673 sizeof(struct btrfs_item),
1674 &left->map_token, &left->kaddr,
1675 &left->map_start, &left->map_len,
1676 KM_USER1);
1677 }
1678
5f39d397
CM
1679 ioff = btrfs_item_offset(left, item);
1680 btrfs_set_item_offset(left, item,
db94535d 1681 ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
be0e5c09 1682 }
5f39d397 1683 btrfs_set_header_nritems(left, old_left_nritems + push_items);
db94535d
CM
1684 if (left->map_token) {
1685 unmap_extent_buffer(left, left->map_token, KM_USER1);
1686 left->map_token = NULL;
1687 }
be0e5c09
CM
1688
1689 /* fixup right node */
5f39d397
CM
1690 push_space = btrfs_item_offset_nr(right, push_items - 1) -
1691 leaf_data_end(root, right);
1692 memmove_extent_buffer(right, btrfs_leaf_data(right) +
1693 BTRFS_LEAF_DATA_SIZE(root) - push_space,
1694 btrfs_leaf_data(right) +
1695 leaf_data_end(root, right), push_space);
1696
1697 memmove_extent_buffer(right, btrfs_item_nr_offset(0),
1698 btrfs_item_nr_offset(push_items),
1699 (btrfs_header_nritems(right) - push_items) *
1700 sizeof(struct btrfs_item));
1701
1702 right_nritems = btrfs_header_nritems(right) - push_items;
1703 btrfs_set_header_nritems(right, right_nritems);
123abc88 1704 push_space = BTRFS_LEAF_DATA_SIZE(root);
eb60ceac 1705
5f39d397
CM
1706 for (i = 0; i < right_nritems; i++) {
1707 item = btrfs_item_nr(right, i);
db94535d
CM
1708
1709 if (!right->map_token) {
1710 map_extent_buffer(right, (unsigned long)item,
1711 sizeof(struct btrfs_item),
1712 &right->map_token, &right->kaddr,
1713 &right->map_start, &right->map_len,
1714 KM_USER1);
1715 }
1716
1717 push_space = push_space - btrfs_item_size(right, item);
1718 btrfs_set_item_offset(right, item, push_space);
1719 }
1720 if (right->map_token) {
1721 unmap_extent_buffer(right, right->map_token, KM_USER1);
1722 right->map_token = NULL;
be0e5c09 1723 }
eb60ceac 1724
5f39d397
CM
1725 btrfs_mark_buffer_dirty(left);
1726 btrfs_mark_buffer_dirty(right);
098f59c2 1727
5f39d397
CM
1728 btrfs_item_key(right, &disk_key, 0);
1729 wret = fixup_low_keys(trans, root, path, &disk_key, 1);
aa5d6bed
CM
1730 if (wret)
1731 ret = wret;
be0e5c09
CM
1732
1733 /* then fixup the leaf pointer in the path */
1734 if (path->slots[0] < push_items) {
1735 path->slots[0] += old_left_nritems;
5f39d397
CM
1736 free_extent_buffer(path->nodes[0]);
1737 path->nodes[0] = left;
be0e5c09
CM
1738 path->slots[1] -= 1;
1739 } else {
5f39d397 1740 free_extent_buffer(left);
be0e5c09
CM
1741 path->slots[0] -= push_items;
1742 }
eb60ceac 1743 BUG_ON(path->slots[0] < 0);
aa5d6bed 1744 return ret;
be0e5c09
CM
1745}
1746
74123bd7
CM
1747/*
1748 * split the path's leaf in two, making sure there is at least data_size
1749 * available for the resulting leaf level of the path.
aa5d6bed
CM
1750 *
1751 * returns 0 if all went well and < 0 on failure.
74123bd7 1752 */
e089f05c 1753static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
d4dbff95
CM
1754 *root, struct btrfs_key *ins_key,
1755 struct btrfs_path *path, int data_size)
be0e5c09 1756{
5f39d397 1757 struct extent_buffer *l;
7518a238 1758 u32 nritems;
eb60ceac
CM
1759 int mid;
1760 int slot;
5f39d397 1761 struct extent_buffer *right;
0783fcfc 1762 int space_needed = data_size + sizeof(struct btrfs_item);
be0e5c09
CM
1763 int data_copy_size;
1764 int rt_data_off;
1765 int i;
d4dbff95 1766 int ret = 0;
aa5d6bed 1767 int wret;
d4dbff95
CM
1768 int double_split = 0;
1769 struct btrfs_disk_key disk_key;
aa5d6bed 1770
40689478 1771 /* first try to make some room by pushing left and right */
e089f05c 1772 wret = push_leaf_left(trans, root, path, data_size);
eaee50e8
CM
1773 if (wret < 0)
1774 return wret;
1775 if (wret) {
e089f05c 1776 wret = push_leaf_right(trans, root, path, data_size);
eaee50e8
CM
1777 if (wret < 0)
1778 return wret;
1779 }
5f39d397 1780 l = path->nodes[0];
aa5d6bed
CM
1781
1782 /* did the pushes work? */
123abc88
CM
1783 if (btrfs_leaf_free_space(root, l) >=
1784 sizeof(struct btrfs_item) + data_size)
aa5d6bed
CM
1785 return 0;
1786
5c680ed6 1787 if (!path->nodes[1]) {
e089f05c 1788 ret = insert_new_root(trans, root, path, 1);
5c680ed6
CM
1789 if (ret)
1790 return ret;
1791 }
eb60ceac 1792 slot = path->slots[0];
5f39d397 1793 nritems = btrfs_header_nritems(l);
eb60ceac 1794 mid = (nritems + 1)/ 2;
54aa1f4d 1795
db94535d
CM
1796 right = btrfs_alloc_free_block(trans, root, root->leafsize,
1797 l->start, 0);
5f39d397
CM
1798 if (IS_ERR(right))
1799 return PTR_ERR(right);
1800
1801 memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
db94535d 1802 btrfs_set_header_bytenr(right, right->start);
5f39d397
CM
1803 btrfs_set_header_generation(right, trans->transid);
1804 btrfs_set_header_owner(right, root->root_key.objectid);
1805 btrfs_set_header_level(right, 0);
1806 write_extent_buffer(right, root->fs_info->fsid,
1807 (unsigned long)btrfs_header_fsid(right),
1808 BTRFS_FSID_SIZE);
1809
d4dbff95
CM
1810 if (mid <= slot) {
1811 if (nritems == 1 ||
1812 leaf_space_used(l, mid, nritems - mid) + space_needed >
1813 BTRFS_LEAF_DATA_SIZE(root)) {
1814 if (slot >= nritems) {
1815 btrfs_cpu_key_to_disk(&disk_key, ins_key);
5f39d397 1816 btrfs_set_header_nritems(right, 0);
d4dbff95 1817 wret = insert_ptr(trans, root, path,
db94535d 1818 &disk_key, right->start,
d4dbff95
CM
1819 path->slots[1] + 1, 1);
1820 if (wret)
1821 ret = wret;
5f39d397
CM
1822 free_extent_buffer(path->nodes[0]);
1823 path->nodes[0] = right;
d4dbff95
CM
1824 path->slots[0] = 0;
1825 path->slots[1] += 1;
1826 return ret;
1827 }
1828 mid = slot;
1829 double_split = 1;
1830 }
1831 } else {
1832 if (leaf_space_used(l, 0, mid + 1) + space_needed >
1833 BTRFS_LEAF_DATA_SIZE(root)) {
1834 if (slot == 0) {
1835 btrfs_cpu_key_to_disk(&disk_key, ins_key);
5f39d397 1836 btrfs_set_header_nritems(right, 0);
d4dbff95
CM
1837 wret = insert_ptr(trans, root, path,
1838 &disk_key,
db94535d 1839 right->start,
098f59c2 1840 path->slots[1], 1);
d4dbff95
CM
1841 if (wret)
1842 ret = wret;
5f39d397
CM
1843 free_extent_buffer(path->nodes[0]);
1844 path->nodes[0] = right;
d4dbff95 1845 path->slots[0] = 0;
a429e513
CM
1846 if (path->slots[1] == 0) {
1847 wret = fixup_low_keys(trans, root,
1848 path, &disk_key, 1);
1849 if (wret)
1850 ret = wret;
1851 }
d4dbff95
CM
1852 return ret;
1853 }
1854 mid = slot;
1855 double_split = 1;
1856 }
1857 }
5f39d397
CM
1858 nritems = nritems - mid;
1859 btrfs_set_header_nritems(right, nritems);
1860 data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);
1861
1862 copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
1863 btrfs_item_nr_offset(mid),
1864 nritems * sizeof(struct btrfs_item));
1865
1866 copy_extent_buffer(right, l,
d6025579
CM
1867 btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
1868 data_copy_size, btrfs_leaf_data(l) +
1869 leaf_data_end(root, l), data_copy_size);
5f39d397 1870
123abc88 1871 rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
5f39d397 1872 btrfs_item_end_nr(l, mid);
74123bd7 1873
5f39d397
CM
1874 for (i = 0; i < nritems; i++) {
1875 struct btrfs_item *item = btrfs_item_nr(right, i);
db94535d
CM
1876 u32 ioff;
1877
1878 if (!right->map_token) {
1879 map_extent_buffer(right, (unsigned long)item,
1880 sizeof(struct btrfs_item),
1881 &right->map_token, &right->kaddr,
1882 &right->map_start, &right->map_len,
1883 KM_USER1);
1884 }
1885
1886 ioff = btrfs_item_offset(right, item);
5f39d397 1887 btrfs_set_item_offset(right, item, ioff + rt_data_off);
0783fcfc 1888 }
74123bd7 1889
db94535d
CM
1890 if (right->map_token) {
1891 unmap_extent_buffer(right, right->map_token, KM_USER1);
1892 right->map_token = NULL;
1893 }
1894
5f39d397 1895 btrfs_set_header_nritems(l, mid);
aa5d6bed 1896 ret = 0;
5f39d397 1897 btrfs_item_key(right, &disk_key, 0);
db94535d
CM
1898 wret = insert_ptr(trans, root, path, &disk_key, right->start,
1899 path->slots[1] + 1, 1);
aa5d6bed
CM
1900 if (wret)
1901 ret = wret;
5f39d397
CM
1902
1903 btrfs_mark_buffer_dirty(right);
1904 btrfs_mark_buffer_dirty(l);
eb60ceac 1905 BUG_ON(path->slots[0] != slot);
5f39d397 1906
be0e5c09 1907 if (mid <= slot) {
5f39d397
CM
1908 free_extent_buffer(path->nodes[0]);
1909 path->nodes[0] = right;
be0e5c09
CM
1910 path->slots[0] -= mid;
1911 path->slots[1] += 1;
eb60ceac 1912 } else
5f39d397
CM
1913 free_extent_buffer(right);
1914
eb60ceac 1915 BUG_ON(path->slots[0] < 0);
d4dbff95
CM
1916
1917 if (!double_split)
1918 return ret;
5f39d397 1919
db94535d
CM
1920 right = btrfs_alloc_free_block(trans, root, root->leafsize,
1921 l->start, 0);
5f39d397
CM
1922 if (IS_ERR(right))
1923 return PTR_ERR(right);
1924
1925 memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
db94535d 1926 btrfs_set_header_bytenr(right, right->start);
5f39d397
CM
1927 btrfs_set_header_generation(right, trans->transid);
1928 btrfs_set_header_owner(right, root->root_key.objectid);
1929 btrfs_set_header_level(right, 0);
1930 write_extent_buffer(right, root->fs_info->fsid,
1931 (unsigned long)btrfs_header_fsid(right),
1932 BTRFS_FSID_SIZE);
1933
d4dbff95 1934 btrfs_cpu_key_to_disk(&disk_key, ins_key);
5f39d397 1935 btrfs_set_header_nritems(right, 0);
d4dbff95 1936 wret = insert_ptr(trans, root, path,
db94535d 1937 &disk_key, right->start,
d4dbff95
CM
1938 path->slots[1], 1);
1939 if (wret)
1940 ret = wret;
a429e513
CM
1941 if (path->slots[1] == 0) {
1942 wret = fixup_low_keys(trans, root, path, &disk_key, 1);
1943 if (wret)
1944 ret = wret;
1945 }
5f39d397
CM
1946 free_extent_buffer(path->nodes[0]);
1947 path->nodes[0] = right;
d4dbff95 1948 path->slots[0] = 0;
be0e5c09
CM
1949 return ret;
1950}
1951
b18c6685
CM
1952int btrfs_truncate_item(struct btrfs_trans_handle *trans,
1953 struct btrfs_root *root,
1954 struct btrfs_path *path,
1955 u32 new_size)
1956{
1957 int ret = 0;
1958 int slot;
1959 int slot_orig;
5f39d397
CM
1960 struct extent_buffer *leaf;
1961 struct btrfs_item *item;
b18c6685
CM
1962 u32 nritems;
1963 unsigned int data_end;
1964 unsigned int old_data_start;
1965 unsigned int old_size;
1966 unsigned int size_diff;
1967 int i;
1968
1969 slot_orig = path->slots[0];
5f39d397 1970 leaf = path->nodes[0];
b18c6685 1971
5f39d397 1972 nritems = btrfs_header_nritems(leaf);
b18c6685
CM
1973 data_end = leaf_data_end(root, leaf);
1974
1975 slot = path->slots[0];
5f39d397
CM
1976 old_data_start = btrfs_item_offset_nr(leaf, slot);
1977 old_size = btrfs_item_size_nr(leaf, slot);
b18c6685
CM
1978 BUG_ON(old_size <= new_size);
1979 size_diff = old_size - new_size;
1980
1981 BUG_ON(slot < 0);
1982 BUG_ON(slot >= nritems);
1983
1984 /*
1985 * item0..itemN ... dataN.offset..dataN.size .. data0.size
1986 */
1987 /* first correct the data pointers */
1988 for (i = slot; i < nritems; i++) {
5f39d397
CM
1989 u32 ioff;
1990 item = btrfs_item_nr(leaf, i);
db94535d
CM
1991
1992 if (!leaf->map_token) {
1993 map_extent_buffer(leaf, (unsigned long)item,
1994 sizeof(struct btrfs_item),
1995 &leaf->map_token, &leaf->kaddr,
1996 &leaf->map_start, &leaf->map_len,
1997 KM_USER1);
1998 }
1999
5f39d397
CM
2000 ioff = btrfs_item_offset(leaf, item);
2001 btrfs_set_item_offset(leaf, item, ioff + size_diff);
b18c6685 2002 }
db94535d
CM
2003
2004 if (leaf->map_token) {
2005 unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2006 leaf->map_token = NULL;
2007 }
2008
b18c6685 2009 /* shift the data */
5f39d397 2010 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
b18c6685
CM
2011 data_end + size_diff, btrfs_leaf_data(leaf) +
2012 data_end, old_data_start + new_size - data_end);
5f39d397
CM
2013
2014 item = btrfs_item_nr(leaf, slot);
2015 btrfs_set_item_size(leaf, item, new_size);
2016 btrfs_mark_buffer_dirty(leaf);
b18c6685
CM
2017
2018 ret = 0;
5f39d397
CM
2019 if (btrfs_leaf_free_space(root, leaf) < 0) {
2020 btrfs_print_leaf(root, leaf);
b18c6685 2021 BUG();
5f39d397 2022 }
b18c6685
CM
2023 return ret;
2024}
2025
5f39d397
CM
2026int btrfs_extend_item(struct btrfs_trans_handle *trans,
2027 struct btrfs_root *root, struct btrfs_path *path,
2028 u32 data_size)
6567e837
CM
2029{
2030 int ret = 0;
2031 int slot;
2032 int slot_orig;
5f39d397
CM
2033 struct extent_buffer *leaf;
2034 struct btrfs_item *item;
6567e837
CM
2035 u32 nritems;
2036 unsigned int data_end;
2037 unsigned int old_data;
2038 unsigned int old_size;
2039 int i;
2040
2041 slot_orig = path->slots[0];
5f39d397 2042 leaf = path->nodes[0];
6567e837 2043
5f39d397 2044 nritems = btrfs_header_nritems(leaf);
6567e837
CM
2045 data_end = leaf_data_end(root, leaf);
2046
5f39d397
CM
2047 if (btrfs_leaf_free_space(root, leaf) < data_size) {
2048 btrfs_print_leaf(root, leaf);
6567e837 2049 BUG();
5f39d397 2050 }
6567e837 2051 slot = path->slots[0];
5f39d397 2052 old_data = btrfs_item_end_nr(leaf, slot);
6567e837
CM
2053
2054 BUG_ON(slot < 0);
2055 BUG_ON(slot >= nritems);
2056
2057 /*
2058 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2059 */
2060 /* first correct the data pointers */
2061 for (i = slot; i < nritems; i++) {
5f39d397
CM
2062 u32 ioff;
2063 item = btrfs_item_nr(leaf, i);
db94535d
CM
2064
2065 if (!leaf->map_token) {
2066 map_extent_buffer(leaf, (unsigned long)item,
2067 sizeof(struct btrfs_item),
2068 &leaf->map_token, &leaf->kaddr,
2069 &leaf->map_start, &leaf->map_len,
2070 KM_USER1);
2071 }
5f39d397
CM
2072 ioff = btrfs_item_offset(leaf, item);
2073 btrfs_set_item_offset(leaf, item, ioff - data_size);
6567e837 2074 }
5f39d397 2075
db94535d
CM
2076 if (leaf->map_token) {
2077 unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2078 leaf->map_token = NULL;
2079 }
2080
6567e837 2081 /* shift the data */
5f39d397 2082 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
6567e837
CM
2083 data_end - data_size, btrfs_leaf_data(leaf) +
2084 data_end, old_data - data_end);
5f39d397 2085
6567e837 2086 data_end = old_data;
5f39d397
CM
2087 old_size = btrfs_item_size_nr(leaf, slot);
2088 item = btrfs_item_nr(leaf, slot);
2089 btrfs_set_item_size(leaf, item, old_size + data_size);
2090 btrfs_mark_buffer_dirty(leaf);
6567e837
CM
2091
2092 ret = 0;
5f39d397
CM
2093 if (btrfs_leaf_free_space(root, leaf) < 0) {
2094 btrfs_print_leaf(root, leaf);
6567e837 2095 BUG();
5f39d397 2096 }
6567e837
CM
2097 return ret;
2098}
2099
74123bd7
CM
2100/*
2101 * Given a key and some data, insert an item into the tree.
2102 * This does all the path init required, making room in the tree if needed.
2103 */
5f39d397
CM
2104int btrfs_insert_empty_item(struct btrfs_trans_handle *trans,
2105 struct btrfs_root *root,
2106 struct btrfs_path *path,
2107 struct btrfs_key *cpu_key, u32 data_size)
be0e5c09 2108{
5f39d397
CM
2109 struct extent_buffer *leaf;
2110 struct btrfs_item *item;
aa5d6bed 2111 int ret = 0;
be0e5c09 2112 int slot;
eb60ceac 2113 int slot_orig;
7518a238 2114 u32 nritems;
be0e5c09 2115 unsigned int data_end;
e2fa7227
CM
2116 struct btrfs_disk_key disk_key;
2117
2118 btrfs_cpu_key_to_disk(&disk_key, cpu_key);
be0e5c09 2119
74123bd7 2120 /* create a root if there isn't one */
5c680ed6 2121 if (!root->node)
cfaa7295 2122 BUG();
5f39d397 2123
e089f05c 2124 ret = btrfs_search_slot(trans, root, cpu_key, path, data_size, 1);
eb60ceac 2125 if (ret == 0) {
f0930a37 2126 return -EEXIST;
aa5d6bed 2127 }
ed2ff2cb
CM
2128 if (ret < 0)
2129 goto out;
be0e5c09 2130
62e2749e 2131 slot_orig = path->slots[0];
5f39d397 2132 leaf = path->nodes[0];
74123bd7 2133
5f39d397 2134 nritems = btrfs_header_nritems(leaf);
123abc88 2135 data_end = leaf_data_end(root, leaf);
eb60ceac 2136
123abc88 2137 if (btrfs_leaf_free_space(root, leaf) <
d4dbff95 2138 sizeof(struct btrfs_item) + data_size) {
be0e5c09 2139 BUG();
d4dbff95 2140 }
5f39d397 2141
62e2749e 2142 slot = path->slots[0];
eb60ceac 2143 BUG_ON(slot < 0);
5f39d397 2144
be0e5c09
CM
2145 if (slot != nritems) {
2146 int i;
5f39d397 2147 unsigned int old_data = btrfs_item_end_nr(leaf, slot);
be0e5c09 2148
5f39d397
CM
2149 if (old_data < data_end) {
2150 btrfs_print_leaf(root, leaf);
2151 printk("slot %d old_data %d data_end %d\n",
2152 slot, old_data, data_end);
2153 BUG_ON(1);
2154 }
be0e5c09
CM
2155 /*
2156 * item0..itemN ... dataN.offset..dataN.size .. data0.size
2157 */
2158 /* first correct the data pointers */
db94535d 2159 WARN_ON(leaf->map_token);
0783fcfc 2160 for (i = slot; i < nritems; i++) {
5f39d397 2161 u32 ioff;
db94535d 2162
5f39d397 2163 item = btrfs_item_nr(leaf, i);
db94535d
CM
2164 if (!leaf->map_token) {
2165 map_extent_buffer(leaf, (unsigned long)item,
2166 sizeof(struct btrfs_item),
2167 &leaf->map_token, &leaf->kaddr,
2168 &leaf->map_start, &leaf->map_len,
2169 KM_USER1);
2170 }
2171
5f39d397
CM
2172 ioff = btrfs_item_offset(leaf, item);
2173 btrfs_set_item_offset(leaf, item, ioff - data_size);
0783fcfc 2174 }
db94535d
CM
2175 if (leaf->map_token) {
2176 unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2177 leaf->map_token = NULL;
2178 }
be0e5c09
CM
2179
2180 /* shift the items */
5f39d397
CM
2181 memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1),
2182 btrfs_item_nr_offset(slot),
d6025579 2183 (nritems - slot) * sizeof(struct btrfs_item));
be0e5c09
CM
2184
2185 /* shift the data */
5f39d397 2186 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
d6025579
CM
2187 data_end - data_size, btrfs_leaf_data(leaf) +
2188 data_end, old_data - data_end);
be0e5c09
CM
2189 data_end = old_data;
2190 }
5f39d397 2191
62e2749e 2192 /* setup the item for the new data */
5f39d397
CM
2193 btrfs_set_item_key(leaf, &disk_key, slot);
2194 item = btrfs_item_nr(leaf, slot);
2195 btrfs_set_item_offset(leaf, item, data_end - data_size);
2196 btrfs_set_item_size(leaf, item, data_size);
2197 btrfs_set_header_nritems(leaf, nritems + 1);
2198 btrfs_mark_buffer_dirty(leaf);
aa5d6bed
CM
2199
2200 ret = 0;
8e19f2cd 2201 if (slot == 0)
e089f05c 2202 ret = fixup_low_keys(trans, root, path, &disk_key, 1);
aa5d6bed 2203
5f39d397
CM
2204 if (btrfs_leaf_free_space(root, leaf) < 0) {
2205 btrfs_print_leaf(root, leaf);
be0e5c09 2206 BUG();
5f39d397 2207 }
ed2ff2cb 2208out:
62e2749e
CM
2209 return ret;
2210}
2211
2212/*
2213 * Given a key and some data, insert an item into the tree.
2214 * This does all the path init required, making room in the tree if needed.
2215 */
e089f05c
CM
2216int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
2217 *root, struct btrfs_key *cpu_key, void *data, u32
2218 data_size)
62e2749e
CM
2219{
2220 int ret = 0;
2c90e5d6 2221 struct btrfs_path *path;
5f39d397
CM
2222 struct extent_buffer *leaf;
2223 unsigned long ptr;
62e2749e 2224
2c90e5d6
CM
2225 path = btrfs_alloc_path();
2226 BUG_ON(!path);
2c90e5d6 2227 ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
62e2749e 2228 if (!ret) {
5f39d397
CM
2229 leaf = path->nodes[0];
2230 ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
2231 write_extent_buffer(leaf, data, ptr, data_size);
2232 btrfs_mark_buffer_dirty(leaf);
62e2749e 2233 }
2c90e5d6 2234 btrfs_free_path(path);
aa5d6bed 2235 return ret;
be0e5c09
CM
2236}
2237
74123bd7 2238/*
5de08d7d 2239 * delete the pointer from a given node.
74123bd7
CM
2240 *
2241 * If the delete empties a node, the node is removed from the tree,
2242 * continuing all the way the root if required. The root is converted into
2243 * a leaf if all the nodes are emptied.
2244 */
e089f05c
CM
2245static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2246 struct btrfs_path *path, int level, int slot)
be0e5c09 2247{
5f39d397 2248 struct extent_buffer *parent = path->nodes[level];
7518a238 2249 u32 nritems;
aa5d6bed 2250 int ret = 0;
bb803951 2251 int wret;
be0e5c09 2252
5f39d397 2253 nritems = btrfs_header_nritems(parent);
bb803951 2254 if (slot != nritems -1) {
5f39d397
CM
2255 memmove_extent_buffer(parent,
2256 btrfs_node_key_ptr_offset(slot),
2257 btrfs_node_key_ptr_offset(slot + 1),
d6025579
CM
2258 sizeof(struct btrfs_key_ptr) *
2259 (nritems - slot - 1));
bb803951 2260 }
7518a238 2261 nritems--;
5f39d397 2262 btrfs_set_header_nritems(parent, nritems);
7518a238 2263 if (nritems == 0 && parent == root->node) {
5f39d397 2264 BUG_ON(btrfs_header_level(root->node) != 1);
bb803951 2265 /* just turn the root into a leaf and break */
5f39d397 2266 btrfs_set_header_level(root->node, 0);
bb803951 2267 } else if (slot == 0) {
5f39d397
CM
2268 struct btrfs_disk_key disk_key;
2269
2270 btrfs_node_key(parent, &disk_key, 0);
2271 wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
0f70abe2
CM
2272 if (wret)
2273 ret = wret;
be0e5c09 2274 }
d6025579 2275 btrfs_mark_buffer_dirty(parent);
aa5d6bed 2276 return ret;
be0e5c09
CM
2277}
2278
74123bd7
CM
2279/*
2280 * delete the item at the leaf level in path. If that empties
2281 * the leaf, remove it from the tree
2282 */
e089f05c
CM
2283int btrfs_del_item(struct btrfs_trans_handle *trans, struct btrfs_root *root,
2284 struct btrfs_path *path)
be0e5c09 2285{
be0e5c09 2286 int slot;
5f39d397
CM
2287 struct extent_buffer *leaf;
2288 struct btrfs_item *item;
be0e5c09
CM
2289 int doff;
2290 int dsize;
aa5d6bed
CM
2291 int ret = 0;
2292 int wret;
7518a238 2293 u32 nritems;
be0e5c09 2294
5f39d397 2295 leaf = path->nodes[0];
4920c9ac 2296 slot = path->slots[0];
5f39d397
CM
2297 doff = btrfs_item_offset_nr(leaf, slot);
2298 dsize = btrfs_item_size_nr(leaf, slot);
2299 nritems = btrfs_header_nritems(leaf);
be0e5c09 2300
7518a238 2301 if (slot != nritems - 1) {
be0e5c09 2302 int i;
123abc88 2303 int data_end = leaf_data_end(root, leaf);
5f39d397
CM
2304
2305 memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
d6025579
CM
2306 data_end + dsize,
2307 btrfs_leaf_data(leaf) + data_end,
2308 doff - data_end);
5f39d397 2309
0783fcfc 2310 for (i = slot + 1; i < nritems; i++) {
5f39d397 2311 u32 ioff;
db94535d 2312
5f39d397 2313 item = btrfs_item_nr(leaf, i);
db94535d
CM
2314 if (!leaf->map_token) {
2315 map_extent_buffer(leaf, (unsigned long)item,
2316 sizeof(struct btrfs_item),
2317 &leaf->map_token, &leaf->kaddr,
2318 &leaf->map_start, &leaf->map_len,
2319 KM_USER1);
2320 }
5f39d397
CM
2321 ioff = btrfs_item_offset(leaf, item);
2322 btrfs_set_item_offset(leaf, item, ioff + dsize);
0783fcfc 2323 }
db94535d
CM
2324
2325 if (leaf->map_token) {
2326 unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
2327 leaf->map_token = NULL;
2328 }
2329
5f39d397
CM
2330 memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
2331 btrfs_item_nr_offset(slot + 1),
d6025579
CM
2332 sizeof(struct btrfs_item) *
2333 (nritems - slot - 1));
be0e5c09 2334 }
5f39d397 2335 btrfs_set_header_nritems(leaf, nritems - 1);
7518a238 2336 nritems--;
5f39d397 2337
74123bd7 2338 /* delete the leaf if we've emptied it */
7518a238 2339 if (nritems == 0) {
5f39d397
CM
2340 if (leaf == root->node) {
2341 btrfs_set_header_level(leaf, 0);
9a8dd150 2342 } else {
5f39d397
CM
2343 clean_tree_block(trans, root, leaf);
2344 wait_on_tree_block_writeback(root, leaf);
e089f05c 2345 wret = del_ptr(trans, root, path, 1, path->slots[1]);
aa5d6bed
CM
2346 if (wret)
2347 ret = wret;
e089f05c 2348 wret = btrfs_free_extent(trans, root,
db94535d 2349 leaf->start, leaf->len, 1);
0f70abe2
CM
2350 if (wret)
2351 ret = wret;
9a8dd150 2352 }
be0e5c09 2353 } else {
7518a238 2354 int used = leaf_space_used(leaf, 0, nritems);
aa5d6bed 2355 if (slot == 0) {
5f39d397
CM
2356 struct btrfs_disk_key disk_key;
2357
2358 btrfs_item_key(leaf, &disk_key, 0);
e089f05c 2359 wret = fixup_low_keys(trans, root, path,
5f39d397 2360 &disk_key, 1);
aa5d6bed
CM
2361 if (wret)
2362 ret = wret;
2363 }
aa5d6bed 2364
74123bd7 2365 /* delete the leaf if it is mostly empty */
123abc88 2366 if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) {
be0e5c09
CM
2367 /* push_leaf_left fixes the path.
2368 * make sure the path still points to our leaf
2369 * for possible call to del_ptr below
2370 */
4920c9ac 2371 slot = path->slots[1];
5f39d397
CM
2372 extent_buffer_get(leaf);
2373
e089f05c 2374 wret = push_leaf_left(trans, root, path, 1);
54aa1f4d 2375 if (wret < 0 && wret != -ENOSPC)
aa5d6bed 2376 ret = wret;
5f39d397
CM
2377
2378 if (path->nodes[0] == leaf &&
2379 btrfs_header_nritems(leaf)) {
e089f05c 2380 wret = push_leaf_right(trans, root, path, 1);
54aa1f4d 2381 if (wret < 0 && wret != -ENOSPC)
aa5d6bed
CM
2382 ret = wret;
2383 }
5f39d397
CM
2384
2385 if (btrfs_header_nritems(leaf) == 0) {
db94535d
CM
2386 u64 bytenr = leaf->start;
2387 u32 blocksize = leaf->len;
5f39d397
CM
2388
2389 clean_tree_block(trans, root, leaf);
2390 wait_on_tree_block_writeback(root, leaf);
2391
e089f05c 2392 wret = del_ptr(trans, root, path, 1, slot);
aa5d6bed
CM
2393 if (wret)
2394 ret = wret;
5f39d397
CM
2395
2396 free_extent_buffer(leaf);
db94535d
CM
2397 wret = btrfs_free_extent(trans, root, bytenr,
2398 blocksize, 1);
0f70abe2
CM
2399 if (wret)
2400 ret = wret;
5de08d7d 2401 } else {
5f39d397
CM
2402 btrfs_mark_buffer_dirty(leaf);
2403 free_extent_buffer(leaf);
be0e5c09 2404 }
d5719762 2405 } else {
5f39d397 2406 btrfs_mark_buffer_dirty(leaf);
be0e5c09
CM
2407 }
2408 }
aa5d6bed 2409 return ret;
be0e5c09
CM
2410}
2411
97571fd0
CM
2412/*
2413 * walk up the tree as far as required to find the next leaf.
0f70abe2
CM
2414 * returns 0 if it found something or 1 if there are no greater leaves.
2415 * returns < 0 on io errors.
97571fd0 2416 */
234b63a0 2417int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
d97e63b6
CM
2418{
2419 int slot;
2420 int level = 1;
db94535d 2421 u64 bytenr;
5f39d397
CM
2422 struct extent_buffer *c;
2423 struct extent_buffer *next = NULL;
d97e63b6 2424
234b63a0 2425 while(level < BTRFS_MAX_LEVEL) {
d97e63b6 2426 if (!path->nodes[level])
0f70abe2 2427 return 1;
5f39d397 2428
d97e63b6
CM
2429 slot = path->slots[level] + 1;
2430 c = path->nodes[level];
5f39d397 2431 if (slot >= btrfs_header_nritems(c)) {
d97e63b6
CM
2432 level++;
2433 continue;
2434 }
5f39d397 2435
db94535d 2436 bytenr = btrfs_node_blockptr(c, slot);
cfaa7295 2437 if (next)
5f39d397
CM
2438 free_extent_buffer(next);
2439
6702ed49
CM
2440 if (path->reada)
2441 reada_for_search(root, path, level, slot);
5f39d397 2442
db94535d
CM
2443 next = read_tree_block(root, bytenr,
2444 btrfs_level_size(root, level -1));
d97e63b6
CM
2445 break;
2446 }
2447 path->slots[level] = slot;
2448 while(1) {
2449 level--;
2450 c = path->nodes[level];
5f39d397 2451 free_extent_buffer(c);
d97e63b6
CM
2452 path->nodes[level] = next;
2453 path->slots[level] = 0;
2454 if (!level)
2455 break;
6702ed49 2456 if (path->reada)
32020611 2457 reada_for_search(root, path, level, 0);
db94535d
CM
2458 next = read_tree_block(root, btrfs_node_blockptr(next, 0),
2459 btrfs_level_size(root, level - 1));
d97e63b6
CM
2460 }
2461 return 0;
2462}