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