ctree.c 63.8 KB
Newer Older
C
Chris Mason 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
/*
 * Copyright (C) 2007 Oracle.  All rights reserved.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public
 * License v2 as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public
 * License along with this program; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 021110-1307, USA.
 */

19
#include <linux/sched.h>
20 21
#include "ctree.h"
#include "disk-io.h"
22
#include "transaction.h"
23
#include "print-tree.h"
24

25 26 27
static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_path *path, int level);
static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
28
		      *root, struct btrfs_key *ins_key,
29
		      struct btrfs_path *path, int data_size, int extend);
30 31 32 33 34 35 36
static int push_node_left(struct btrfs_trans_handle *trans,
			  struct btrfs_root *root, struct extent_buffer *dst,
			  struct extent_buffer *src);
static int balance_node_right(struct btrfs_trans_handle *trans,
			      struct btrfs_root *root,
			      struct extent_buffer *dst_buf,
			      struct extent_buffer *src_buf);
37 38
static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
		   struct btrfs_path *path, int level, int slot);
39

C
Chris Mason 已提交
40
inline void btrfs_init_path(struct btrfs_path *p)
C
Chris Mason 已提交
41
{
C
Chris Mason 已提交
42
	memset(p, 0, sizeof(*p));
C
Chris Mason 已提交
43 44
}

C
Chris Mason 已提交
45
struct btrfs_path *btrfs_alloc_path(void)
C
Chris Mason 已提交
46
{
C
Chris Mason 已提交
47 48
	struct btrfs_path *path;
	path = kmem_cache_alloc(btrfs_path_cachep, GFP_NOFS);
49
	if (path) {
C
Chris Mason 已提交
50
		btrfs_init_path(path);
51 52
		path->reada = 1;
	}
C
Chris Mason 已提交
53
	return path;
C
Chris Mason 已提交
54 55
}

C
Chris Mason 已提交
56
void btrfs_free_path(struct btrfs_path *p)
57
{
C
Chris Mason 已提交
58 59
	btrfs_release_path(NULL, p);
	kmem_cache_free(btrfs_path_cachep, p);
60 61
}

C
Chris Mason 已提交
62
void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p)
63 64
{
	int i;
C
Chris Mason 已提交
65
	for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
66 67
		if (!p->nodes[i])
			break;
68
		free_extent_buffer(p->nodes[i]);
69
	}
C
Chris Mason 已提交
70
	memset(p, 0, sizeof(*p));
71 72
}

73 74 75 76 77 78
static int __btrfs_cow_block(struct btrfs_trans_handle *trans,
			     struct btrfs_root *root,
			     struct extent_buffer *buf,
			     struct extent_buffer *parent, int parent_slot,
			     struct extent_buffer **cow_ret,
			     u64 search_start, u64 empty_size)
C
Chris Mason 已提交
79
{
80
	struct extent_buffer *cow;
81 82
	int ret = 0;
	int different_trans = 0;
C
Chris Mason 已提交
83

84
	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
85

86 87
	cow = btrfs_alloc_free_block(trans, root, buf->len,
				     search_start, empty_size);
88 89
	if (IS_ERR(cow))
		return PTR_ERR(cow);
90

91
	copy_extent_buffer(cow, buf, 0, 0, cow->len);
92
	btrfs_set_header_bytenr(cow, cow->start);
93 94
	btrfs_set_header_generation(cow, trans->transid);
	btrfs_set_header_owner(cow, root->root_key.objectid);
95

96 97
	WARN_ON(btrfs_header_generation(buf) > trans->transid);
	if (btrfs_header_generation(buf) != trans->transid) {
98 99 100 101 102 103 104 105
		different_trans = 1;
		ret = btrfs_inc_ref(trans, root, buf);
		if (ret)
			return ret;
	} else {
		clean_tree_block(trans, root, buf);
	}

C
Chris Mason 已提交
106 107
	if (buf == root->node) {
		root->node = cow;
108
		extent_buffer_get(cow);
C
Chris Mason 已提交
109
		if (buf != root->commit_root) {
110 111
			btrfs_free_extent(trans, root, buf->start,
					  buf->len, 1);
C
Chris Mason 已提交
112
		}
113
		free_extent_buffer(buf);
C
Chris Mason 已提交
114
	} else {
115
		btrfs_set_node_blockptr(parent, parent_slot,
116
					cow->start);
C
Chris Mason 已提交
117
		btrfs_mark_buffer_dirty(parent);
118
		WARN_ON(btrfs_header_generation(parent) != trans->transid);
119
		btrfs_free_extent(trans, root, buf->start, buf->len, 1);
C
Chris Mason 已提交
120
	}
121
	free_extent_buffer(buf);
C
Chris Mason 已提交
122
	btrfs_mark_buffer_dirty(cow);
C
Chris Mason 已提交
123
	*cow_ret = cow;
C
Chris Mason 已提交
124 125 126
	return 0;
}

127 128 129 130
int btrfs_cow_block(struct btrfs_trans_handle *trans,
		    struct btrfs_root *root, struct extent_buffer *buf,
		    struct extent_buffer *parent, int parent_slot,
		    struct extent_buffer **cow_ret)
131 132
{
	u64 search_start;
133
	int ret;
134 135 136 137 138 139 140 141 142 143
	if (trans->transaction != root->fs_info->running_transaction) {
		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
		       root->fs_info->running_transaction->transid);
		WARN_ON(1);
	}
	if (trans->transid != root->fs_info->generation) {
		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
		       root->fs_info->generation);
		WARN_ON(1);
	}
144
	if (btrfs_header_generation(buf) == trans->transid) {
145 146 147 148
		*cow_ret = buf;
		return 0;
	}

149
	search_start = buf->start & ~((u64)BTRFS_BLOCK_GROUP_SIZE - 1);
150
	ret = __btrfs_cow_block(trans, root, buf, parent,
151
				 parent_slot, cow_ret, search_start, 0);
152
	return ret;
153 154
}

155
static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
156
{
157
	if (blocknr < other && other - (blocknr + blocksize) < 32768)
158
		return 1;
159
	if (blocknr > other && blocknr - (other + blocksize) < 32768)
160 161 162 163 164
		return 1;
	return 0;
}

int btrfs_realloc_node(struct btrfs_trans_handle *trans,
165
		       struct btrfs_root *root, struct extent_buffer *parent,
166 167
		       int start_slot, int cache_only, u64 *last_ret,
		       struct btrfs_key *progress)
168
{
169 170
	struct extent_buffer *cur;
	struct extent_buffer *tmp;
171
	u64 blocknr;
172 173
	u64 search_start = *last_ret;
	u64 last_block = 0;
174 175 176 177 178
	u64 other;
	u32 parent_nritems;
	int end_slot;
	int i;
	int err = 0;
179
	int parent_level;
180 181
	int uptodate;
	u32 blocksize;
182

183 184 185 186
	parent_level = btrfs_header_level(parent);
	if (cache_only && parent_level != 1)
		return 0;

187 188 189 190 191 192 193 194 195 196
	if (trans->transaction != root->fs_info->running_transaction) {
		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
		       root->fs_info->running_transaction->transid);
		WARN_ON(1);
	}
	if (trans->transid != root->fs_info->generation) {
		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
		       root->fs_info->generation);
		WARN_ON(1);
	}
197

198 199
	parent_nritems = btrfs_header_nritems(parent);
	blocksize = btrfs_level_size(root, parent_level - 1);
200 201 202 203 204 205 206
	end_slot = parent_nritems;

	if (parent_nritems == 1)
		return 0;

	for (i = start_slot; i < end_slot; i++) {
		int close = 1;
207

208 209 210 211 212 213 214 215
		if (!parent->map_token) {
			map_extent_buffer(parent,
					btrfs_node_key_ptr_offset(i),
					sizeof(struct btrfs_key_ptr),
					&parent->map_token, &parent->kaddr,
					&parent->map_start, &parent->map_len,
					KM_USER1);
		}
216
		blocknr = btrfs_node_blockptr(parent, i);
217 218
		if (last_block == 0)
			last_block = blocknr;
219

220
		if (i > 0) {
221 222
			other = btrfs_node_blockptr(parent, i - 1);
			close = close_blocks(blocknr, other, blocksize);
223
		}
224
		if (close && i < end_slot - 2) {
225 226
			other = btrfs_node_blockptr(parent, i + 1);
			close = close_blocks(blocknr, other, blocksize);
227
		}
228 229
		if (close) {
			last_block = blocknr;
230
			continue;
231
		}
232 233 234 235 236
		if (parent->map_token) {
			unmap_extent_buffer(parent, parent->map_token,
					    KM_USER1);
			parent->map_token = NULL;
		}
237

238 239 240 241 242
		cur = btrfs_find_tree_block(root, blocknr, blocksize);
		if (cur)
			uptodate = btrfs_buffer_uptodate(cur);
		else
			uptodate = 0;
243
		if (!cur || !uptodate) {
244
			if (cache_only) {
245
				free_extent_buffer(cur);
246 247
				continue;
			}
248 249 250 251 252
			if (!cur) {
				cur = read_tree_block(root, blocknr,
							 blocksize);
			} else if (!uptodate) {
				btrfs_read_buffer(cur);
253
			}
254
		}
255
		if (search_start == 0)
256
			search_start = last_block;
257

258 259 260 261
		err = __btrfs_cow_block(trans, root, cur, parent, i,
					&tmp, search_start,
					min(16 * blocksize,
					    (end_slot - i) * blocksize));
Y
Yan 已提交
262
		if (err) {
263
			free_extent_buffer(cur);
264
			break;
Y
Yan 已提交
265
		}
266
		search_start = tmp->start;
267
		last_block = tmp->start;
268 269
		*last_ret = search_start;
		if (parent_level == 1)
270 271
			btrfs_clear_buffer_defrag(tmp);
		free_extent_buffer(tmp);
272
	}
273 274 275 276 277
	if (parent->map_token) {
		unmap_extent_buffer(parent, parent->map_token,
				    KM_USER1);
		parent->map_token = NULL;
	}
278 279 280
	return err;
}

C
Chris Mason 已提交
281 282 283 284 285
/*
 * The leaf data grows from end-to-front in the node.
 * this returns the address of the start of the last item,
 * which is the stop of the leaf data stack
 */
C
Chris Mason 已提交
286
static inline unsigned int leaf_data_end(struct btrfs_root *root,
287
					 struct extent_buffer *leaf)
288
{
289
	u32 nr = btrfs_header_nritems(leaf);
290
	if (nr == 0)
C
Chris Mason 已提交
291
		return BTRFS_LEAF_DATA_SIZE(root);
292
	return btrfs_item_offset_nr(leaf, nr - 1);
293 294
}

C
Chris Mason 已提交
295 296 297
/*
 * compare two keys in a memcmp fashion
 */
C
Chris Mason 已提交
298
static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
299
{
C
Chris Mason 已提交
300 301 302 303 304
	struct btrfs_key k1;

	btrfs_disk_key_to_cpu(&k1, disk);

	if (k1.objectid > k2->objectid)
305
		return 1;
C
Chris Mason 已提交
306
	if (k1.objectid < k2->objectid)
307
		return -1;
308
	if (k1.type > k2->type)
C
Chris Mason 已提交
309
		return 1;
310
	if (k1.type < k2->type)
C
Chris Mason 已提交
311
		return -1;
312 313 314 315
	if (k1.offset > k2->offset)
		return 1;
	if (k1.offset < k2->offset)
		return -1;
316 317
	return 0;
}
C
Chris Mason 已提交
318

C
Chris Mason 已提交
319 320
static int check_node(struct btrfs_root *root, struct btrfs_path *path,
		      int level)
C
Chris Mason 已提交
321
{
322 323 324 325
	struct extent_buffer *parent = NULL;
	struct extent_buffer *node = path->nodes[level];
	struct btrfs_disk_key parent_key;
	struct btrfs_disk_key node_key;
C
Chris Mason 已提交
326
	int parent_slot;
327 328
	int slot;
	struct btrfs_key cpukey;
329
	u32 nritems = btrfs_header_nritems(node);
C
Chris Mason 已提交
330 331

	if (path->nodes[level + 1])
332
		parent = path->nodes[level + 1];
A
Aneesh 已提交
333

334
	slot = path->slots[level];
335 336
	BUG_ON(nritems == 0);
	if (parent) {
A
Aneesh 已提交
337
		parent_slot = path->slots[level + 1];
338 339 340
		btrfs_node_key(parent, &parent_key, parent_slot);
		btrfs_node_key(node, &node_key, 0);
		BUG_ON(memcmp(&parent_key, &node_key,
C
Chris Mason 已提交
341
			      sizeof(struct btrfs_disk_key)));
342
		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
343
		       btrfs_header_bytenr(node));
C
Chris Mason 已提交
344
	}
C
Chris Mason 已提交
345
	BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
346
	if (slot != 0) {
347 348 349
		btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
		btrfs_node_key(node, &node_key, slot);
		BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
350 351
	}
	if (slot < nritems - 1) {
352 353 354
		btrfs_node_key_to_cpu(node, &cpukey, slot + 1);
		btrfs_node_key(node, &node_key, slot);
		BUG_ON(comp_keys(&node_key, &cpukey) >= 0);
C
Chris Mason 已提交
355 356 357 358
	}
	return 0;
}

C
Chris Mason 已提交
359 360
static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
		      int level)
C
Chris Mason 已提交
361
{
362 363
	struct extent_buffer *leaf = path->nodes[level];
	struct extent_buffer *parent = NULL;
C
Chris Mason 已提交
364
	int parent_slot;
365
	struct btrfs_key cpukey;
366 367 368
	struct btrfs_disk_key parent_key;
	struct btrfs_disk_key leaf_key;
	int slot = path->slots[0];
369

370
	u32 nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
371 372

	if (path->nodes[level + 1])
373
		parent = path->nodes[level + 1];
374 375 376 377 378

	if (nritems == 0)
		return 0;

	if (parent) {
A
Aneesh 已提交
379
		parent_slot = path->slots[level + 1];
380 381
		btrfs_node_key(parent, &parent_key, parent_slot);
		btrfs_item_key(leaf, &leaf_key, 0);
382

383
		BUG_ON(memcmp(&parent_key, &leaf_key,
C
Chris Mason 已提交
384
		       sizeof(struct btrfs_disk_key)));
385
		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
386
		       btrfs_header_bytenr(leaf));
387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411
	}
#if 0
	for (i = 0; nritems > 1 && i < nritems - 2; i++) {
		btrfs_item_key_to_cpu(leaf, &cpukey, i + 1);
		btrfs_item_key(leaf, &leaf_key, i);
		if (comp_keys(&leaf_key, &cpukey) >= 0) {
			btrfs_print_leaf(root, leaf);
			printk("slot %d offset bad key\n", i);
			BUG_ON(1);
		}
		if (btrfs_item_offset_nr(leaf, i) !=
			btrfs_item_end_nr(leaf, i + 1)) {
			btrfs_print_leaf(root, leaf);
			printk("slot %d offset bad\n", i);
			BUG_ON(1);
		}
		if (i == 0) {
			if (btrfs_item_offset_nr(leaf, i) +
			       btrfs_item_size_nr(leaf, i) !=
			       BTRFS_LEAF_DATA_SIZE(root)) {
				btrfs_print_leaf(root, leaf);
				printk("slot %d first offset bad\n", i);
				BUG_ON(1);
			}
		}
C
Chris Mason 已提交
412
	}
413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434
	if (nritems > 0) {
		if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) {
				btrfs_print_leaf(root, leaf);
				printk("slot %d bad size \n", nritems - 1);
				BUG_ON(1);
		}
	}
#endif
	if (slot != 0 && slot < nritems - 1) {
		btrfs_item_key(leaf, &leaf_key, slot);
		btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1);
		if (comp_keys(&leaf_key, &cpukey) <= 0) {
			btrfs_print_leaf(root, leaf);
			printk("slot %d offset bad key\n", slot);
			BUG_ON(1);
		}
		if (btrfs_item_offset_nr(leaf, slot - 1) !=
		       btrfs_item_end_nr(leaf, slot)) {
			btrfs_print_leaf(root, leaf);
			printk("slot %d offset bad\n", slot);
			BUG_ON(1);
		}
435 436
	}
	if (slot < nritems - 1) {
437 438 439 440 441 442 443 444 445
		btrfs_item_key(leaf, &leaf_key, slot);
		btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1);
		BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0);
		if (btrfs_item_offset_nr(leaf, slot) !=
			btrfs_item_end_nr(leaf, slot + 1)) {
			btrfs_print_leaf(root, leaf);
			printk("slot %d offset bad\n", slot);
			BUG_ON(1);
		}
C
Chris Mason 已提交
446
	}
447 448
	BUG_ON(btrfs_item_offset_nr(leaf, 0) +
	       btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
C
Chris Mason 已提交
449 450 451
	return 0;
}

C
Chris Mason 已提交
452 453
static int check_block(struct btrfs_root *root, struct btrfs_path *path,
			int level)
C
Chris Mason 已提交
454
{
455
	return 0;
456
#if 0
457 458
	struct extent_buffer *buf = path->nodes[level];

459 460 461
	if (memcmp_extent_buffer(buf, root->fs_info->fsid,
				 (unsigned long)btrfs_header_fsid(buf),
				 BTRFS_FSID_SIZE)) {
462
		printk("warning bad block %Lu\n", buf->start);
463
		return 1;
464
	}
465
#endif
C
Chris Mason 已提交
466
	if (level == 0)
C
Chris Mason 已提交
467 468
		return check_leaf(root, path, level);
	return check_node(root, path, level);
C
Chris Mason 已提交
469 470
}

C
Chris Mason 已提交
471
/*
472 473 474
 * search for key in the extent_buffer.  The items start at offset p,
 * and they are item_size apart.  There are 'max' items in p.
 *
C
Chris Mason 已提交
475 476 477 478 479 480
 * the slot in the array is returned via slot, and it points to
 * the place where you would insert key if it is not found in
 * the array.
 *
 * slot may point to max if the key is bigger than all of the keys
 */
481 482 483
static int generic_bin_search(struct extent_buffer *eb, unsigned long p,
			      int item_size, struct btrfs_key *key,
			      int max, int *slot)
484 485 486 487 488
{
	int low = 0;
	int high = max;
	int mid;
	int ret;
489
	struct btrfs_disk_key *tmp = NULL;
490 491 492 493 494 495
	struct btrfs_disk_key unaligned;
	unsigned long offset;
	char *map_token = NULL;
	char *kaddr = NULL;
	unsigned long map_start = 0;
	unsigned long map_len = 0;
496
	int err;
497 498 499

	while(low < high) {
		mid = (low + high) / 2;
500 501 502 503 504
		offset = p + mid * item_size;

		if (!map_token || offset < map_start ||
		    (offset + sizeof(struct btrfs_disk_key)) >
		    map_start + map_len) {
505
			if (map_token) {
506
				unmap_extent_buffer(eb, map_token, KM_USER0);
507 508 509 510 511 512 513 514 515 516 517 518 519 520 521
				map_token = NULL;
			}
			err = map_extent_buffer(eb, offset,
						sizeof(struct btrfs_disk_key),
						&map_token, &kaddr,
						&map_start, &map_len, KM_USER0);

			if (!err) {
				tmp = (struct btrfs_disk_key *)(kaddr + offset -
							map_start);
			} else {
				read_extent_buffer(eb, &unaligned,
						   offset, sizeof(unaligned));
				tmp = &unaligned;
			}
522 523 524 525 526

		} else {
			tmp = (struct btrfs_disk_key *)(kaddr + offset -
							map_start);
		}
527 528 529 530 531 532 533 534
		ret = comp_keys(tmp, key);

		if (ret < 0)
			low = mid + 1;
		else if (ret > 0)
			high = mid;
		else {
			*slot = mid;
535 536
			if (map_token)
				unmap_extent_buffer(eb, map_token, KM_USER0);
537 538 539 540
			return 0;
		}
	}
	*slot = low;
541 542
	if (map_token)
		unmap_extent_buffer(eb, map_token, KM_USER0);
543 544 545
	return 1;
}

C
Chris Mason 已提交
546 547 548 549
/*
 * simple bin_search frontend that does the right thing for
 * leaves vs nodes
 */
550 551
static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
		      int level, int *slot)
552
{
553 554 555
	if (level == 0) {
		return generic_bin_search(eb,
					  offsetof(struct btrfs_leaf, items),
C
Chris Mason 已提交
556
					  sizeof(struct btrfs_item),
557
					  key, btrfs_header_nritems(eb),
558
					  slot);
559
	} else {
560 561
		return generic_bin_search(eb,
					  offsetof(struct btrfs_node, ptrs),
C
Chris Mason 已提交
562
					  sizeof(struct btrfs_key_ptr),
563
					  key, btrfs_header_nritems(eb),
564
					  slot);
565 566 567 568
	}
	return -1;
}

569 570
static struct extent_buffer *read_node_slot(struct btrfs_root *root,
				   struct extent_buffer *parent, int slot)
571 572 573
{
	if (slot < 0)
		return NULL;
574
	if (slot >= btrfs_header_nritems(parent))
575
		return NULL;
576 577
	return read_tree_block(root, btrfs_node_blockptr(parent, slot),
		       btrfs_level_size(root, btrfs_header_level(parent) - 1));
578 579
}

580 581
static int balance_level(struct btrfs_trans_handle *trans, struct btrfs_root
			 *root, struct btrfs_path *path, int level)
582
{
583 584 585 586
	struct extent_buffer *right = NULL;
	struct extent_buffer *mid;
	struct extent_buffer *left = NULL;
	struct extent_buffer *parent = NULL;
587 588 589 590
	int ret = 0;
	int wret;
	int pslot;
	int orig_slot = path->slots[level];
591
	int err_on_enospc = 0;
592
	u64 orig_ptr;
593 594 595 596

	if (level == 0)
		return 0;

597
	mid = path->nodes[level];
598
	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
599

C
Chris Mason 已提交
600
	if (level < BTRFS_MAX_LEVEL - 1)
601
		parent = path->nodes[level + 1];
602 603
	pslot = path->slots[level + 1];

C
Chris Mason 已提交
604 605 606 607
	/*
	 * deal with the case where there is only one pointer in the root
	 * by promoting the node below to a root
	 */
608 609
	if (!parent) {
		struct extent_buffer *child;
610

611
		if (btrfs_header_nritems(mid) != 1)
612 613 614
			return 0;

		/* promote the child to a root */
615
		child = read_node_slot(root, mid, 0);
616 617 618
		BUG_ON(!child);
		root->node = child;
		path->nodes[level] = NULL;
619 620
		clean_tree_block(trans, root, mid);
		wait_on_tree_block_writeback(root, mid);
621
		/* once for the path */
622
		free_extent_buffer(mid);
623
		ret = btrfs_free_extent(trans, root, mid->start, mid->len, 1);
624
		/* once for the root ptr */
625
		free_extent_buffer(mid);
626
		return ret;
627
	}
628
	if (btrfs_header_nritems(mid) >
C
Chris Mason 已提交
629
	    BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
630 631
		return 0;

632
	if (btrfs_header_nritems(mid) < 2)
633 634
		err_on_enospc = 1;

635 636 637 638
	left = read_node_slot(root, parent, pslot - 1);
	if (left) {
		wret = btrfs_cow_block(trans, root, left,
				       parent, pslot - 1, &left);
639 640 641 642
		if (wret) {
			ret = wret;
			goto enospc;
		}
643
	}
644 645 646 647
	right = read_node_slot(root, parent, pslot + 1);
	if (right) {
		wret = btrfs_cow_block(trans, root, right,
				       parent, pslot + 1, &right);
648 649 650 651 652 653 654
		if (wret) {
			ret = wret;
			goto enospc;
		}
	}

	/* first, try to make some room in the middle buffer */
655 656 657
	if (left) {
		orig_slot += btrfs_header_nritems(left);
		wret = push_node_left(trans, root, left, mid);
658 659
		if (wret < 0)
			ret = wret;
660
		if (btrfs_header_nritems(mid) < 2)
661
			err_on_enospc = 1;
662
	}
663 664 665 666

	/*
	 * then try to empty the right most buffer into the middle
	 */
667 668
	if (right) {
		wret = push_node_left(trans, root, mid, right);
669
		if (wret < 0 && wret != -ENOSPC)
670
			ret = wret;
671
		if (btrfs_header_nritems(right) == 0) {
672 673 674
			u64 bytenr = right->start;
			u32 blocksize = right->len;

675 676 677
			clean_tree_block(trans, root, right);
			wait_on_tree_block_writeback(root, right);
			free_extent_buffer(right);
678
			right = NULL;
679 680
			wret = del_ptr(trans, root, path, level + 1, pslot +
				       1);
681 682
			if (wret)
				ret = wret;
683 684
			wret = btrfs_free_extent(trans, root, bytenr,
						 blocksize, 1);
685 686 687
			if (wret)
				ret = wret;
		} else {
688 689 690 691
			struct btrfs_disk_key right_key;
			btrfs_node_key(right, &right_key, 0);
			btrfs_set_node_key(parent, &right_key, pslot + 1);
			btrfs_mark_buffer_dirty(parent);
692 693
		}
	}
694
	if (btrfs_header_nritems(mid) == 1) {
695 696 697 698 699 700 701 702 703
		/*
		 * we're not allowed to leave a node with one item in the
		 * tree during a delete.  A deletion from lower in the tree
		 * could try to delete the only pointer in this node.
		 * So, pull some keys from the left.
		 * There has to be a left pointer at this point because
		 * otherwise we would have pulled some pointers from the
		 * right
		 */
704 705
		BUG_ON(!left);
		wret = balance_node_right(trans, root, mid, left);
706
		if (wret < 0) {
707
			ret = wret;
708 709
			goto enospc;
		}
710 711
		BUG_ON(wret == 1);
	}
712
	if (btrfs_header_nritems(mid) == 0) {
713
		/* we've managed to empty the middle node, drop it */
714 715
		u64 bytenr = mid->start;
		u32 blocksize = mid->len;
716 717 718
		clean_tree_block(trans, root, mid);
		wait_on_tree_block_writeback(root, mid);
		free_extent_buffer(mid);
719
		mid = NULL;
720
		wret = del_ptr(trans, root, path, level + 1, pslot);
721 722
		if (wret)
			ret = wret;
723
		wret = btrfs_free_extent(trans, root, bytenr, blocksize, 1);
724 725
		if (wret)
			ret = wret;
726 727
	} else {
		/* update the parent key to reflect our changes */
728 729 730 731
		struct btrfs_disk_key mid_key;
		btrfs_node_key(mid, &mid_key, 0);
		btrfs_set_node_key(parent, &mid_key, pslot);
		btrfs_mark_buffer_dirty(parent);
732
	}
733

734
	/* update the path */
735 736 737 738
	if (left) {
		if (btrfs_header_nritems(left) > orig_slot) {
			extent_buffer_get(left);
			path->nodes[level] = left;
739 740
			path->slots[level + 1] -= 1;
			path->slots[level] = orig_slot;
741 742
			if (mid)
				free_extent_buffer(mid);
743
		} else {
744
			orig_slot -= btrfs_header_nritems(left);
745 746 747
			path->slots[level] = orig_slot;
		}
	}
748
	/* double check we haven't messed things up */
C
Chris Mason 已提交
749
	check_block(root, path, level);
C
Chris Mason 已提交
750
	if (orig_ptr !=
751
	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
752
		BUG();
753
enospc:
754 755 756 757
	if (right)
		free_extent_buffer(right);
	if (left)
		free_extent_buffer(left);
758 759 760
	return ret;
}

761 762 763 764 765
/* returns zero if the push worked, non-zero otherwise */
static int push_nodes_for_insert(struct btrfs_trans_handle *trans,
				struct btrfs_root *root,
				struct btrfs_path *path, int level)
{
766 767 768 769
	struct extent_buffer *right = NULL;
	struct extent_buffer *mid;
	struct extent_buffer *left = NULL;
	struct extent_buffer *parent = NULL;
770 771 772 773 774 775 776 777 778
	int ret = 0;
	int wret;
	int pslot;
	int orig_slot = path->slots[level];
	u64 orig_ptr;

	if (level == 0)
		return 1;

779
	mid = path->nodes[level];
780 781 782
	orig_ptr = btrfs_node_blockptr(mid, orig_slot);

	if (level < BTRFS_MAX_LEVEL - 1)
783
		parent = path->nodes[level + 1];
784 785
	pslot = path->slots[level + 1];

786
	if (!parent)
787 788
		return 1;

789
	left = read_node_slot(root, parent, pslot - 1);
790 791

	/* first, try to make some room in the middle buffer */
792
	if (left) {
793
		u32 left_nr;
794
		left_nr = btrfs_header_nritems(left);
C
Chris Mason 已提交
795 796 797
		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
			wret = 1;
		} else {
798 799
			ret = btrfs_cow_block(trans, root, left, parent,
					      pslot - 1, &left);
800 801 802 803
			if (ret)
				wret = 1;
			else {
				wret = push_node_left(trans, root,
804
						      left, mid);
805
			}
C
Chris Mason 已提交
806
		}
807 808 809
		if (wret < 0)
			ret = wret;
		if (wret == 0) {
810
			struct btrfs_disk_key disk_key;
811
			orig_slot += left_nr;
812 813 814 815 816
			btrfs_node_key(mid, &disk_key, 0);
			btrfs_set_node_key(parent, &disk_key, pslot);
			btrfs_mark_buffer_dirty(parent);
			if (btrfs_header_nritems(left) > orig_slot) {
				path->nodes[level] = left;
817 818
				path->slots[level + 1] -= 1;
				path->slots[level] = orig_slot;
819
				free_extent_buffer(mid);
820 821
			} else {
				orig_slot -=
822
					btrfs_header_nritems(left);
823
				path->slots[level] = orig_slot;
824
				free_extent_buffer(left);
825 826 827
			}
			return 0;
		}
828
		free_extent_buffer(left);
829
	}
830
	right= read_node_slot(root, parent, pslot + 1);
831 832 833 834

	/*
	 * then try to empty the right most buffer into the middle
	 */
835
	if (right) {
C
Chris Mason 已提交
836
		u32 right_nr;
837
		right_nr = btrfs_header_nritems(right);
C
Chris Mason 已提交
838 839 840
		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
			wret = 1;
		} else {
841 842 843
			ret = btrfs_cow_block(trans, root, right,
					      parent, pslot + 1,
					      &right);
844 845 846 847
			if (ret)
				wret = 1;
			else {
				wret = balance_node_right(trans, root,
848
							  right, mid);
849
			}
C
Chris Mason 已提交
850
		}
851 852 853
		if (wret < 0)
			ret = wret;
		if (wret == 0) {
854 855 856 857 858 859 860 861
			struct btrfs_disk_key disk_key;

			btrfs_node_key(right, &disk_key, 0);
			btrfs_set_node_key(parent, &disk_key, pslot + 1);
			btrfs_mark_buffer_dirty(parent);

			if (btrfs_header_nritems(mid) <= orig_slot) {
				path->nodes[level] = right;
862 863
				path->slots[level + 1] += 1;
				path->slots[level] = orig_slot -
864 865
					btrfs_header_nritems(mid);
				free_extent_buffer(mid);
866
			} else {
867
				free_extent_buffer(right);
868 869 870
			}
			return 0;
		}
871
		free_extent_buffer(right);
872 873 874 875
	}
	return 1;
}

876 877 878 879
/*
 * readahead one full node of leaves
 */
static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path,
880
			     int level, int slot)
881
{
882
	struct extent_buffer *node;
883 884
	u32 nritems;
	u64 search;
885 886 887
	u64 lowest_read;
	u64 highest_read;
	u64 nread = 0;
888
	int direction = path->reada;
889
	struct extent_buffer *eb;
890 891 892
	u32 nr;
	u32 blocksize;
	u32 nscan = 0;
893

894
	if (level != 1)
895 896 897
		return;

	if (!path->nodes[level])
898 899
		return;

900
	node = path->nodes[level];
901
	search = btrfs_node_blockptr(node, slot);
902 903
	blocksize = btrfs_level_size(root, level - 1);
	eb = btrfs_find_tree_block(root, search, blocksize);
904 905
	if (eb) {
		free_extent_buffer(eb);
906 907 908
		return;
	}

909 910 911
	highest_read = search;
	lowest_read = search;

912
	nritems = btrfs_header_nritems(node);
913
	nr = slot;
914
	while(1) {
915 916 917 918 919 920 921 922
		if (direction < 0) {
			if (nr == 0)
				break;
			nr--;
		} else if (direction > 0) {
			nr++;
			if (nr >= nritems)
				break;
923
		}
924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940
		search = btrfs_node_blockptr(node, nr);
		if ((search >= lowest_read && search <= highest_read) ||
		    (search < lowest_read && lowest_read - search <= 32768) ||
		    (search > highest_read && search - highest_read <= 32768)) {
			readahead_tree_block(root, search, blocksize);
			nread += blocksize;
		}
		nscan++;
		if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32))
			break;
		if(nread > (1024 * 1024) || nscan > 128)
			break;

		if (search < lowest_read)
			lowest_read = search;
		if (search > highest_read)
			highest_read = search;
941 942
	}
}
C
Chris Mason 已提交
943 944 945 946 947 948
/*
 * look for key in the tree.  path is filled in with nodes along the way
 * if key is found, we return zero and you can find the item in the leaf
 * level of the path (level 0)
 *
 * If the key isn't found, the path points to the slot where it should
C
Chris Mason 已提交
949 950
 * be inserted, and 1 is returned.  If there are other errors during the
 * search a negative error number is returned.
C
Chris Mason 已提交
951 952 953 954
 *
 * if ins_len > 0, nodes and leaves will be split as we walk down the
 * tree.  if ins_len < 0, nodes will be merged as we walk down the tree (if
 * possible)
C
Chris Mason 已提交
955
 */
956 957 958
int btrfs_search_slot(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_key *key, struct btrfs_path *p, int
		      ins_len, int cow)
959
{
960
	struct extent_buffer *b;
961
	u64 bytenr;
962 963 964
	int slot;
	int ret;
	int level;
965
	int should_reada = p->reada;
966 967
	u8 lowest_level = 0;

968 969
	lowest_level = p->lowest_level;
	WARN_ON(lowest_level && ins_len);
C
Chris Mason 已提交
970 971
	WARN_ON(p->nodes[0] != NULL);
	WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex));
972 973
again:
	b = root->node;
974
	extent_buffer_get(b);
975
	while (b) {
976
		level = btrfs_header_level(b);
C
Chris Mason 已提交
977 978
		if (cow) {
			int wret;
C
Chris Mason 已提交
979 980 981
			wret = btrfs_cow_block(trans, root, b,
					       p->nodes[level + 1],
					       p->slots[level + 1],
Y
Yan 已提交
982
					       &b);
983
			if (wret) {
984
				free_extent_buffer(b);
985 986
				return wret;
			}
C
Chris Mason 已提交
987 988
		}
		BUG_ON(!cow && ins_len);
989
		if (level != btrfs_header_level(b))
C
Chris Mason 已提交
990
			WARN_ON(1);
991
		level = btrfs_header_level(b);
992
		p->nodes[level] = b;
C
Chris Mason 已提交
993
		ret = check_block(root, p, level);
C
Chris Mason 已提交
994 995
		if (ret)
			return -1;
996 997
		ret = bin_search(b, key, level, &slot);
		if (level != 0) {
998 999 1000
			if (ret && slot > 0)
				slot -= 1;
			p->slots[level] = slot;
1001
			if (ins_len > 0 && btrfs_header_nritems(b) >=
1002
			    BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
1003
				int sret = split_node(trans, root, p, level);
C
Chris Mason 已提交
1004 1005 1006 1007 1008
				BUG_ON(sret > 0);
				if (sret)
					return sret;
				b = p->nodes[level];
				slot = p->slots[level];
1009
			} else if (ins_len < 0) {
1010 1011
				int sret = balance_level(trans, root, p,
							 level);
1012 1013 1014
				if (sret)
					return sret;
				b = p->nodes[level];
1015 1016
				if (!b) {
					btrfs_release_path(NULL, p);
1017
					goto again;
1018
				}
1019
				slot = p->slots[level];
1020
				BUG_ON(btrfs_header_nritems(b) == 1);
C
Chris Mason 已提交
1021
			}
1022 1023 1024
			/* this is only true while dropping a snapshot */
			if (level == lowest_level)
				break;
1025
			bytenr = btrfs_node_blockptr(b, slot);
1026 1027
			if (should_reada)
				reada_for_search(root, p, level, slot);
1028 1029
			b = read_tree_block(root, bytenr,
					    btrfs_level_size(root, level - 1));
1030 1031
		} else {
			p->slots[level] = slot;
1032
			if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
C
Chris Mason 已提交
1033
			    sizeof(struct btrfs_item) + ins_len) {
1034
				int sret = split_leaf(trans, root, key,
1035
						      p, ins_len, ret == 0);
C
Chris Mason 已提交
1036 1037 1038 1039
				BUG_ON(sret > 0);
				if (sret)
					return sret;
			}
1040 1041 1042
			return ret;
		}
	}
C
Chris Mason 已提交
1043
	return 1;
1044 1045
}

C
Chris Mason 已提交
1046 1047 1048 1049 1050 1051
/*
 * adjust the pointers going up the tree, starting at level
 * making sure the right key of each node is points to 'key'.
 * This is used after shifting pointers to the left, so it stops
 * fixing up pointers when a given leaf/node is not in slot 0 of the
 * higher levels
C
Chris Mason 已提交
1052 1053 1054
 *
 * If this fails to write a tree block, it returns -1, but continues
 * fixing up the blocks in ram so the tree is consistent.
C
Chris Mason 已提交
1055
 */
1056 1057 1058
static int fixup_low_keys(struct btrfs_trans_handle *trans,
			  struct btrfs_root *root, struct btrfs_path *path,
			  struct btrfs_disk_key *key, int level)
1059 1060
{
	int i;
C
Chris Mason 已提交
1061
	int ret = 0;
1062 1063
	struct extent_buffer *t;

C
Chris Mason 已提交
1064
	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1065
		int tslot = path->slots[i];
1066
		if (!path->nodes[i])
1067
			break;
1068 1069
		t = path->nodes[i];
		btrfs_set_node_key(t, key, tslot);
C
Chris Mason 已提交
1070
		btrfs_mark_buffer_dirty(path->nodes[i]);
1071 1072 1073
		if (tslot != 0)
			break;
	}
C
Chris Mason 已提交
1074
	return ret;
1075 1076
}

C
Chris Mason 已提交
1077 1078
/*
 * try to push data from one node into the next node left in the
1079
 * tree.
C
Chris Mason 已提交
1080 1081 1082
 *
 * returns 0 if some ptrs were pushed left, < 0 if there was some horrible
 * error, and > 0 if there was no room in the left hand block.
C
Chris Mason 已提交
1083
 */
1084
static int push_node_left(struct btrfs_trans_handle *trans, struct btrfs_root
1085 1086
			  *root, struct extent_buffer *dst,
			  struct extent_buffer *src)
1087 1088
{
	int push_items = 0;
1089 1090
	int src_nritems;
	int dst_nritems;
C
Chris Mason 已提交
1091
	int ret = 0;
1092

1093 1094
	src_nritems = btrfs_header_nritems(src);
	dst_nritems = btrfs_header_nritems(dst);
C
Chris Mason 已提交
1095
	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
1096

1097
	if (push_items <= 0) {
1098
		return 1;
1099
	}
1100

1101
	if (src_nritems < push_items)
1102 1103
		push_items = src_nritems;

1104 1105 1106 1107 1108
	copy_extent_buffer(dst, src,
			   btrfs_node_key_ptr_offset(dst_nritems),
			   btrfs_node_key_ptr_offset(0),
		           push_items * sizeof(struct btrfs_key_ptr));

1109
	if (push_items < src_nritems) {
1110 1111 1112 1113 1114 1115 1116 1117 1118
		memmove_extent_buffer(src, btrfs_node_key_ptr_offset(0),
				      btrfs_node_key_ptr_offset(push_items),
				      (src_nritems - push_items) *
				      sizeof(struct btrfs_key_ptr));
	}
	btrfs_set_header_nritems(src, src_nritems - push_items);
	btrfs_set_header_nritems(dst, dst_nritems + push_items);
	btrfs_mark_buffer_dirty(src);
	btrfs_mark_buffer_dirty(dst);
1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130
	return ret;
}

/*
 * try to push data from one node into the next node right in the
 * tree.
 *
 * returns 0 if some ptrs were pushed, < 0 if there was some horrible
 * error, and > 0 if there was no room in the right hand block.
 *
 * this will  only push up to 1/2 the contents of the left node over
 */
1131 1132 1133 1134
static int balance_node_right(struct btrfs_trans_handle *trans,
			      struct btrfs_root *root,
			      struct extent_buffer *dst,
			      struct extent_buffer *src)
1135 1136 1137 1138 1139 1140 1141
{
	int push_items = 0;
	int max_push;
	int src_nritems;
	int dst_nritems;
	int ret = 0;

1142 1143
	src_nritems = btrfs_header_nritems(src);
	dst_nritems = btrfs_header_nritems(dst);
C
Chris Mason 已提交
1144
	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
1145
	if (push_items <= 0)
1146 1147 1148 1149
		return 1;

	max_push = src_nritems / 2 + 1;
	/* don't try to empty the node */
Y
Yan 已提交
1150
	if (max_push >= src_nritems)
1151
		return 1;
Y
Yan 已提交
1152

1153 1154 1155
	if (max_push < push_items)
		push_items = max_push;

1156 1157 1158 1159
	memmove_extent_buffer(dst, btrfs_node_key_ptr_offset(push_items),
				      btrfs_node_key_ptr_offset(0),
				      (dst_nritems) *
				      sizeof(struct btrfs_key_ptr));
C
Chris Mason 已提交
1160

1161 1162 1163 1164
	copy_extent_buffer(dst, src,
			   btrfs_node_key_ptr_offset(0),
			   btrfs_node_key_ptr_offset(src_nritems - push_items),
		           push_items * sizeof(struct btrfs_key_ptr));
1165

1166 1167
	btrfs_set_header_nritems(src, src_nritems - push_items);
	btrfs_set_header_nritems(dst, dst_nritems + push_items);
1168

1169 1170
	btrfs_mark_buffer_dirty(src);
	btrfs_mark_buffer_dirty(dst);
C
Chris Mason 已提交
1171
	return ret;
1172 1173
}

C
Chris Mason 已提交
1174 1175 1176 1177
/*
 * helper function to insert a new root level in the tree.
 * A new node is allocated, and a single item is inserted to
 * point to the existing root
C
Chris Mason 已提交
1178 1179
 *
 * returns zero on success or < 0 on failure.
C
Chris Mason 已提交
1180
 */
1181 1182 1183
static int insert_new_root(struct btrfs_trans_handle *trans,
			   struct btrfs_root *root,
			   struct btrfs_path *path, int level)
C
Chris Mason 已提交
1184
{
1185 1186 1187
	struct extent_buffer *lower;
	struct extent_buffer *c;
	struct btrfs_disk_key lower_key;
C
Chris Mason 已提交
1188 1189 1190 1191

	BUG_ON(path->nodes[level]);
	BUG_ON(path->nodes[level-1] != root->node);

1192 1193
	c = btrfs_alloc_free_block(trans, root, root->nodesize,
				   root->node->start, 0);
1194 1195 1196 1197 1198
	if (IS_ERR(c))
		return PTR_ERR(c);
	memset_extent_buffer(c, 0, 0, root->nodesize);
	btrfs_set_header_nritems(c, 1);
	btrfs_set_header_level(c, level);
1199
	btrfs_set_header_bytenr(c, c->start);
1200 1201 1202 1203 1204 1205 1206 1207 1208
	btrfs_set_header_generation(c, trans->transid);
	btrfs_set_header_owner(c, root->root_key.objectid);
	lower = path->nodes[level-1];

	write_extent_buffer(c, root->fs_info->fsid,
			    (unsigned long)btrfs_header_fsid(c),
			    BTRFS_FSID_SIZE);
	if (level == 1)
		btrfs_item_key(lower, &lower_key, 0);
C
Chris Mason 已提交
1209
	else
1210 1211
		btrfs_node_key(lower, &lower_key, 0);
	btrfs_set_node_key(c, &lower_key, 0);
1212
	btrfs_set_node_blockptr(c, 0, lower->start);
1213

1214
	btrfs_mark_buffer_dirty(c);
1215

C
Chris Mason 已提交
1216
	/* the super has an extra ref to root->node */
1217 1218 1219 1220
	free_extent_buffer(root->node);
	root->node = c;
	extent_buffer_get(c);
	path->nodes[level] = c;
C
Chris Mason 已提交
1221 1222 1223 1224
	path->slots[level] = 0;
	return 0;
}

C
Chris Mason 已提交
1225 1226 1227
/*
 * worker function to insert a single pointer in a node.
 * the node should have enough room for the pointer already
C
Chris Mason 已提交
1228
 *
C
Chris Mason 已提交
1229 1230
 * slot and level indicate where you want the key to go, and
 * blocknr is the block the key points to.
C
Chris Mason 已提交
1231 1232
 *
 * returns zero on success and < 0 on any error
C
Chris Mason 已提交
1233
 */
1234 1235
static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_path *path, struct btrfs_disk_key
1236
		      *key, u64 bytenr, int slot, int level)
C
Chris Mason 已提交
1237
{
1238
	struct extent_buffer *lower;
C
Chris Mason 已提交
1239
	int nritems;
C
Chris Mason 已提交
1240 1241

	BUG_ON(!path->nodes[level]);
1242 1243
	lower = path->nodes[level];
	nritems = btrfs_header_nritems(lower);
C
Chris Mason 已提交
1244 1245
	if (slot > nritems)
		BUG();
C
Chris Mason 已提交
1246
	if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
C
Chris Mason 已提交
1247 1248
		BUG();
	if (slot != nritems) {
1249 1250 1251
		memmove_extent_buffer(lower,
			      btrfs_node_key_ptr_offset(slot + 1),
			      btrfs_node_key_ptr_offset(slot),
C
Chris Mason 已提交
1252
			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
C
Chris Mason 已提交
1253
	}
1254
	btrfs_set_node_key(lower, key, slot);
1255
	btrfs_set_node_blockptr(lower, slot, bytenr);
1256 1257
	btrfs_set_header_nritems(lower, nritems + 1);
	btrfs_mark_buffer_dirty(lower);
C
Chris Mason 已提交
1258 1259 1260
	return 0;
}

C
Chris Mason 已提交
1261 1262 1263 1264 1265 1266
/*
 * split the node at the specified level in path in two.
 * The path is corrected to point to the appropriate node after the split
 *
 * Before splitting this tries to make some room in the node by pushing
 * left and right, if either one works, it returns right away.
C
Chris Mason 已提交
1267 1268
 *
 * returns 0 on success and < 0 on failure
C
Chris Mason 已提交
1269
 */
1270 1271
static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_path *path, int level)
1272
{
1273 1274 1275
	struct extent_buffer *c;
	struct extent_buffer *split;
	struct btrfs_disk_key disk_key;
1276
	int mid;
C
Chris Mason 已提交
1277
	int ret;
C
Chris Mason 已提交
1278
	int wret;
1279
	u32 c_nritems;
1280

1281 1282
	c = path->nodes[level];
	if (c == root->node) {
C
Chris Mason 已提交
1283
		/* trying to split the root, lets make a new one */
1284
		ret = insert_new_root(trans, root, path, level + 1);
C
Chris Mason 已提交
1285 1286
		if (ret)
			return ret;
1287 1288
	} else {
		ret = push_nodes_for_insert(trans, root, path, level);
1289 1290
		c = path->nodes[level];
		if (!ret && btrfs_header_nritems(c) <
1291 1292
		    BTRFS_NODEPTRS_PER_BLOCK(root) - 1)
			return 0;
1293 1294
		if (ret < 0)
			return ret;
1295
	}
1296

1297
	c_nritems = btrfs_header_nritems(c);
1298 1299
	split = btrfs_alloc_free_block(trans, root, root->nodesize,
				       c->start, 0);
1300 1301 1302 1303 1304
	if (IS_ERR(split))
		return PTR_ERR(split);

	btrfs_set_header_flags(split, btrfs_header_flags(c));
	btrfs_set_header_level(split, btrfs_header_level(c));
1305
	btrfs_set_header_bytenr(split, split->start);
1306 1307 1308 1309 1310
	btrfs_set_header_generation(split, trans->transid);
	btrfs_set_header_owner(split, root->root_key.objectid);
	write_extent_buffer(split, root->fs_info->fsid,
			    (unsigned long)btrfs_header_fsid(split),
			    BTRFS_FSID_SIZE);
1311

1312
	mid = (c_nritems + 1) / 2;
1313 1314 1315 1316 1317 1318 1319

	copy_extent_buffer(split, c,
			   btrfs_node_key_ptr_offset(0),
			   btrfs_node_key_ptr_offset(mid),
			   (c_nritems - mid) * sizeof(struct btrfs_key_ptr));
	btrfs_set_header_nritems(split, c_nritems - mid);
	btrfs_set_header_nritems(c, mid);
C
Chris Mason 已提交
1320 1321
	ret = 0;

1322 1323 1324 1325
	btrfs_mark_buffer_dirty(c);
	btrfs_mark_buffer_dirty(split);

	btrfs_node_key(split, &disk_key, 0);
1326
	wret = insert_ptr(trans, root, path, &disk_key, split->start,
1327
			  path->slots[level + 1] + 1,
C
Chris Mason 已提交
1328
			  level + 1);
C
Chris Mason 已提交
1329 1330 1331
	if (wret)
		ret = wret;

C
Chris Mason 已提交
1332
	if (path->slots[level] >= mid) {
C
Chris Mason 已提交
1333
		path->slots[level] -= mid;
1334 1335
		free_extent_buffer(c);
		path->nodes[level] = split;
C
Chris Mason 已提交
1336 1337
		path->slots[level + 1] += 1;
	} else {
1338
		free_extent_buffer(split);
1339
	}
C
Chris Mason 已提交
1340
	return ret;
1341 1342
}

C
Chris Mason 已提交
1343 1344 1345 1346 1347
/*
 * how many bytes are required to store the items in a leaf.  start
 * and nr indicate which items in the leaf to check.  This totals up the
 * space used both by the item structs and the item data
 */
1348
static int leaf_space_used(struct extent_buffer *l, int start, int nr)
1349 1350
{
	int data_len;
1351
	int nritems = btrfs_header_nritems(l);
1352
	int end = min(nritems, start + nr) - 1;
1353 1354 1355

	if (!nr)
		return 0;
1356 1357
	data_len = btrfs_item_end_nr(l, start);
	data_len = data_len - btrfs_item_offset_nr(l, end);
C
Chris Mason 已提交
1358
	data_len += sizeof(struct btrfs_item) * nr;
1359
	WARN_ON(data_len < 0);
1360 1361 1362
	return data_len;
}

1363 1364 1365 1366 1367
/*
 * The space between the end of the leaf items and
 * the start of the leaf data.  IOW, how much room
 * the leaf has left for both items and data
 */
1368
int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
1369
{
1370 1371 1372 1373 1374
	int nritems = btrfs_header_nritems(leaf);
	int ret;
	ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
	if (ret < 0) {
		printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
J
Jens Axboe 已提交
1375
		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
1376 1377 1378
		       leaf_space_used(leaf, 0, nritems), nritems);
	}
	return ret;
1379 1380
}

C
Chris Mason 已提交
1381 1382 1383
/*
 * push some data in the path leaf to the right, trying to free up at
 * least data_size bytes.  returns zero if the push worked, nonzero otherwise
C
Chris Mason 已提交
1384 1385 1386
 *
 * returns 1 if the push failed because the other node didn't have enough
 * room, 0 if everything worked out and < 0 if there were major errors.
C
Chris Mason 已提交
1387
 */
1388 1389
static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
			   *root, struct btrfs_path *path, int data_size)
C
Chris Mason 已提交
1390
{
1391 1392 1393 1394
	struct extent_buffer *left = path->nodes[0];
	struct extent_buffer *right;
	struct extent_buffer *upper;
	struct btrfs_disk_key disk_key;
C
Chris Mason 已提交
1395 1396 1397 1398 1399
	int slot;
	int i;
	int free_space;
	int push_space = 0;
	int push_items = 0;
C
Chris Mason 已提交
1400
	struct btrfs_item *item;
1401 1402
	u32 left_nritems;
	u32 right_nritems;
1403
	u32 data_end;
1404
	u32 this_item_size;
1405
	int ret;
C
Chris Mason 已提交
1406 1407 1408 1409 1410 1411

	slot = path->slots[1];
	if (!path->nodes[1]) {
		return 1;
	}
	upper = path->nodes[1];
1412
	if (slot >= btrfs_header_nritems(upper) - 1)
C
Chris Mason 已提交
1413
		return 1;
1414

1415 1416
	right = read_tree_block(root, btrfs_node_blockptr(upper, slot + 1),
				root->leafsize);
C
Chris Mason 已提交
1417
	free_space = btrfs_leaf_free_space(root, right);
C
Chris Mason 已提交
1418
	if (free_space < data_size + sizeof(struct btrfs_item)) {
1419
		free_extent_buffer(right);
C
Chris Mason 已提交
1420 1421
		return 1;
	}
1422

C
Chris Mason 已提交
1423
	/* cow and double check */
1424 1425
	ret = btrfs_cow_block(trans, root, right, upper,
			      slot + 1, &right);
1426
	if (ret) {
1427
		free_extent_buffer(right);
1428 1429
		return 1;
	}
C
Chris Mason 已提交
1430
	free_space = btrfs_leaf_free_space(root, right);
C
Chris Mason 已提交
1431
	if (free_space < data_size + sizeof(struct btrfs_item)) {
1432
		free_extent_buffer(right);
C
Chris Mason 已提交
1433 1434 1435
		return 1;
	}

1436
	left_nritems = btrfs_header_nritems(left);
1437
	if (left_nritems == 0) {
1438
		free_extent_buffer(right);
1439 1440
		return 1;
	}
1441

1442
	for (i = left_nritems - 1; i >= 1; i--) {
1443
		item = btrfs_item_nr(left, i);
1444

C
Chris Mason 已提交
1445 1446
		if (path->slots[0] == i)
			push_space += data_size + sizeof(*item);
1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457

		if (!left->map_token) {
			map_extent_buffer(left, (unsigned long)item,
					sizeof(struct btrfs_item),
					&left->map_token, &left->kaddr,
					&left->map_start, &left->map_len,
					KM_USER1);
		}

		this_item_size = btrfs_item_size(left, item);
		if (this_item_size + sizeof(*item) + push_space > free_space)
C
Chris Mason 已提交
1458 1459
			break;
		push_items++;
1460 1461 1462 1463 1464
		push_space += this_item_size + sizeof(*item);
	}
	if (left->map_token) {
		unmap_extent_buffer(left, left->map_token, KM_USER1);
		left->map_token = NULL;
C
Chris Mason 已提交
1465
	}
1466

C
Chris Mason 已提交
1467
	if (push_items == 0) {
1468
		free_extent_buffer(right);
C
Chris Mason 已提交
1469 1470
		return 1;
	}
1471

1472 1473
	if (push_items == left_nritems)
		WARN_ON(1);
1474

C
Chris Mason 已提交
1475
	/* push left to right */
1476 1477
	right_nritems = btrfs_header_nritems(right);
	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
C
Chris Mason 已提交
1478
	push_space -= leaf_data_end(root, left);
1479

C
Chris Mason 已提交
1480
	/* make room in the right data area */
1481 1482 1483 1484 1485 1486
	data_end = leaf_data_end(root, right);
	memmove_extent_buffer(right,
			      btrfs_leaf_data(right) + data_end - push_space,
			      btrfs_leaf_data(right) + data_end,
			      BTRFS_LEAF_DATA_SIZE(root) - data_end);

C
Chris Mason 已提交
1487
	/* copy from the left data area */
1488
	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
C
Chris Mason 已提交
1489 1490 1491
		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
		     btrfs_leaf_data(left) + leaf_data_end(root, left),
		     push_space);
1492 1493 1494 1495 1496

	memmove_extent_buffer(right, btrfs_item_nr_offset(push_items),
			      btrfs_item_nr_offset(0),
			      right_nritems * sizeof(struct btrfs_item));

C
Chris Mason 已提交
1497
	/* copy the items from left to right */
1498 1499 1500
	copy_extent_buffer(right, left, btrfs_item_nr_offset(0),
		   btrfs_item_nr_offset(left_nritems - push_items),
		   push_items * sizeof(struct btrfs_item));
C
Chris Mason 已提交
1501 1502

	/* update the item pointers */
1503
	right_nritems += push_items;
1504
	btrfs_set_header_nritems(right, right_nritems);
C
Chris Mason 已提交
1505
	push_space = BTRFS_LEAF_DATA_SIZE(root);
1506

1507
	for (i = 0; i < right_nritems; i++) {
1508
		item = btrfs_item_nr(right, i);
1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522
		if (!right->map_token) {
			map_extent_buffer(right, (unsigned long)item,
					sizeof(struct btrfs_item),
					&right->map_token, &right->kaddr,
					&right->map_start, &right->map_len,
					KM_USER1);
		}
		push_space -= btrfs_item_size(right, item);
		btrfs_set_item_offset(right, item, push_space);
	}

	if (right->map_token) {
		unmap_extent_buffer(right, right->map_token, KM_USER1);
		right->map_token = NULL;
C
Chris Mason 已提交
1523
	}
1524
	left_nritems -= push_items;
1525
	btrfs_set_header_nritems(left, left_nritems);
C
Chris Mason 已提交
1526

1527 1528
	btrfs_mark_buffer_dirty(left);
	btrfs_mark_buffer_dirty(right);
1529

1530 1531
	btrfs_item_key(right, &disk_key, 0);
	btrfs_set_node_key(upper, &disk_key, slot + 1);
C
Chris Mason 已提交
1532
	btrfs_mark_buffer_dirty(upper);
C
Chris Mason 已提交
1533

C
Chris Mason 已提交
1534
	/* then fixup the leaf pointer in the path */
1535 1536
	if (path->slots[0] >= left_nritems) {
		path->slots[0] -= left_nritems;
1537 1538
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = right;
C
Chris Mason 已提交
1539 1540
		path->slots[1] += 1;
	} else {
1541
		free_extent_buffer(right);
C
Chris Mason 已提交
1542 1543 1544
	}
	return 0;
}
C
Chris Mason 已提交
1545 1546 1547 1548
/*
 * push some data in the path leaf to the left, trying to free up at
 * least data_size bytes.  returns zero if the push worked, nonzero otherwise
 */
1549 1550
static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
			  *root, struct btrfs_path *path, int data_size)
1551
{
1552 1553 1554
	struct btrfs_disk_key disk_key;
	struct extent_buffer *right = path->nodes[0];
	struct extent_buffer *left;
1555 1556 1557 1558 1559
	int slot;
	int i;
	int free_space;
	int push_space = 0;
	int push_items = 0;
C
Chris Mason 已提交
1560
	struct btrfs_item *item;
1561
	u32 old_left_nritems;
1562
	u32 right_nritems;
C
Chris Mason 已提交
1563 1564
	int ret = 0;
	int wret;
1565 1566
	u32 this_item_size;
	u32 old_left_item_size;
1567 1568

	slot = path->slots[1];
1569
	if (slot == 0)
1570
		return 1;
1571
	if (!path->nodes[1])
1572
		return 1;
1573

1574 1575 1576 1577 1578
	right_nritems = btrfs_header_nritems(right);
	if (right_nritems == 0) {
		return 1;
	}

1579
	left = read_tree_block(root, btrfs_node_blockptr(path->nodes[1],
1580
			       slot - 1), root->leafsize);
C
Chris Mason 已提交
1581
	free_space = btrfs_leaf_free_space(root, left);
C
Chris Mason 已提交
1582
	if (free_space < data_size + sizeof(struct btrfs_item)) {
1583
		free_extent_buffer(left);
1584 1585
		return 1;
	}
C
Chris Mason 已提交
1586 1587

	/* cow and double check */
1588 1589
	ret = btrfs_cow_block(trans, root, left,
			      path->nodes[1], slot - 1, &left);
1590 1591
	if (ret) {
		/* we hit -ENOSPC, but it isn't fatal here */
1592
		free_extent_buffer(left);
1593 1594
		return 1;
	}
1595

C
Chris Mason 已提交
1596
	free_space = btrfs_leaf_free_space(root, left);
C
Chris Mason 已提交
1597
	if (free_space < data_size + sizeof(struct btrfs_item)) {
1598
		free_extent_buffer(left);
C
Chris Mason 已提交
1599 1600 1601
		return 1;
	}

1602 1603
	for (i = 0; i < right_nritems - 1; i++) {
		item = btrfs_item_nr(right, i);
1604 1605 1606 1607 1608 1609 1610 1611
		if (!right->map_token) {
			map_extent_buffer(right, (unsigned long)item,
					sizeof(struct btrfs_item),
					&right->map_token, &right->kaddr,
					&right->map_start, &right->map_len,
					KM_USER1);
		}

1612 1613
		if (path->slots[0] == i)
			push_space += data_size + sizeof(*item);
1614 1615 1616

		this_item_size = btrfs_item_size(right, item);
		if (this_item_size + sizeof(*item) + push_space > free_space)
1617
			break;
1618

1619
		push_items++;
1620 1621 1622 1623 1624 1625
		push_space += this_item_size + sizeof(*item);
	}

	if (right->map_token) {
		unmap_extent_buffer(right, right->map_token, KM_USER1);
		right->map_token = NULL;
1626
	}
1627

1628
	if (push_items == 0) {
1629
		free_extent_buffer(left);
1630 1631
		return 1;
	}
1632
	if (push_items == btrfs_header_nritems(right))
1633
		WARN_ON(1);
1634

1635
	/* push data from right to left */
1636 1637 1638 1639 1640
	copy_extent_buffer(left, right,
			   btrfs_item_nr_offset(btrfs_header_nritems(left)),
			   btrfs_item_nr_offset(0),
			   push_items * sizeof(struct btrfs_item));

C
Chris Mason 已提交
1641
	push_space = BTRFS_LEAF_DATA_SIZE(root) -
1642 1643 1644
		     btrfs_item_offset_nr(right, push_items -1);

	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
C
Chris Mason 已提交
1645 1646
		     leaf_data_end(root, left) - push_space,
		     btrfs_leaf_data(right) +
1647
		     btrfs_item_offset_nr(right, push_items - 1),
C
Chris Mason 已提交
1648
		     push_space);
1649
	old_left_nritems = btrfs_header_nritems(left);
1650 1651
	BUG_ON(old_left_nritems < 0);

1652
	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
C
Chris Mason 已提交
1653
	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
1654
		u32 ioff;
1655

1656
		item = btrfs_item_nr(left, i);
1657 1658 1659 1660 1661 1662 1663 1664
		if (!left->map_token) {
			map_extent_buffer(left, (unsigned long)item,
					sizeof(struct btrfs_item),
					&left->map_token, &left->kaddr,
					&left->map_start, &left->map_len,
					KM_USER1);
		}

1665 1666
		ioff = btrfs_item_offset(left, item);
		btrfs_set_item_offset(left, item,
1667
		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
1668
	}
1669
	btrfs_set_header_nritems(left, old_left_nritems + push_items);
1670 1671 1672 1673
	if (left->map_token) {
		unmap_extent_buffer(left, left->map_token, KM_USER1);
		left->map_token = NULL;
	}
1674 1675

	/* fixup right node */
1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689
	push_space = btrfs_item_offset_nr(right, push_items - 1) -
					  leaf_data_end(root, right);
	memmove_extent_buffer(right, btrfs_leaf_data(right) +
			      BTRFS_LEAF_DATA_SIZE(root) - push_space,
			      btrfs_leaf_data(right) +
			      leaf_data_end(root, right), push_space);

	memmove_extent_buffer(right, btrfs_item_nr_offset(0),
			      btrfs_item_nr_offset(push_items),
			     (btrfs_header_nritems(right) - push_items) *
			     sizeof(struct btrfs_item));

	right_nritems = btrfs_header_nritems(right) - push_items;
	btrfs_set_header_nritems(right, right_nritems);
C
Chris Mason 已提交
1690
	push_space = BTRFS_LEAF_DATA_SIZE(root);
1691

1692 1693
	for (i = 0; i < right_nritems; i++) {
		item = btrfs_item_nr(right, i);
1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708

		if (!right->map_token) {
			map_extent_buffer(right, (unsigned long)item,
					sizeof(struct btrfs_item),
					&right->map_token, &right->kaddr,
					&right->map_start, &right->map_len,
					KM_USER1);
		}

		push_space = push_space - btrfs_item_size(right, item);
		btrfs_set_item_offset(right, item, push_space);
	}
	if (right->map_token) {
		unmap_extent_buffer(right, right->map_token, KM_USER1);
		right->map_token = NULL;
1709
	}
1710

1711 1712
	btrfs_mark_buffer_dirty(left);
	btrfs_mark_buffer_dirty(right);
1713

1714 1715
	btrfs_item_key(right, &disk_key, 0);
	wret = fixup_low_keys(trans, root, path, &disk_key, 1);
C
Chris Mason 已提交
1716 1717
	if (wret)
		ret = wret;
1718 1719 1720 1721

	/* then fixup the leaf pointer in the path */
	if (path->slots[0] < push_items) {
		path->slots[0] += old_left_nritems;
1722 1723
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = left;
1724 1725
		path->slots[1] -= 1;
	} else {
1726
		free_extent_buffer(left);
1727 1728
		path->slots[0] -= push_items;
	}
1729
	BUG_ON(path->slots[0] < 0);
C
Chris Mason 已提交
1730
	return ret;
1731 1732
}

C
Chris Mason 已提交
1733 1734 1735
/*
 * split the path's leaf in two, making sure there is at least data_size
 * available for the resulting leaf level of the path.
C
Chris Mason 已提交
1736 1737
 *
 * returns 0 if all went well and < 0 on failure.
C
Chris Mason 已提交
1738
 */
1739
static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
1740
		      *root, struct btrfs_key *ins_key,
1741
		      struct btrfs_path *path, int data_size, int extend)
1742
{
1743
	struct extent_buffer *l;
1744
	u32 nritems;
1745 1746
	int mid;
	int slot;
1747
	struct extent_buffer *right;
C
Chris Mason 已提交
1748
	int space_needed = data_size + sizeof(struct btrfs_item);
1749 1750 1751
	int data_copy_size;
	int rt_data_off;
	int i;
1752
	int ret = 0;
C
Chris Mason 已提交
1753
	int wret;
1754 1755
	int double_split;
	int num_doubles = 0;
1756
	struct btrfs_disk_key disk_key;
C
Chris Mason 已提交
1757

1758 1759 1760
	if (extend)
		space_needed = data_size;

C
Chris Mason 已提交
1761
	/* first try to make some room by pushing left and right */
1762
	if (ins_key->type != BTRFS_DIR_ITEM_KEY) {
1763
		wret = push_leaf_right(trans, root, path, data_size);
1764
		if (wret < 0) {
C
Chris Mason 已提交
1765
			return wret;
1766 1767 1768 1769 1770 1771 1772
		}
		if (wret) {
			wret = push_leaf_left(trans, root, path, data_size);
			if (wret < 0)
				return wret;
		}
		l = path->nodes[0];
C
Chris Mason 已提交
1773

1774
		/* did the pushes work? */
1775
		if (btrfs_leaf_free_space(root, l) >= space_needed)
1776
			return 0;
1777
	}
C
Chris Mason 已提交
1778

C
Chris Mason 已提交
1779
	if (!path->nodes[1]) {
1780
		ret = insert_new_root(trans, root, path, 1);
C
Chris Mason 已提交
1781 1782 1783
		if (ret)
			return ret;
	}
1784 1785 1786
again:
	double_split = 0;
	l = path->nodes[0];
1787
	slot = path->slots[0];
1788
	nritems = btrfs_header_nritems(l);
1789
	mid = (nritems + 1)/ 2;
1790

1791 1792
	right = btrfs_alloc_free_block(trans, root, root->leafsize,
				       l->start, 0);
1793 1794 1795 1796
	if (IS_ERR(right))
		return PTR_ERR(right);

	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
1797
	btrfs_set_header_bytenr(right, right->start);
1798 1799 1800 1801 1802 1803
	btrfs_set_header_generation(right, trans->transid);
	btrfs_set_header_owner(right, root->root_key.objectid);
	btrfs_set_header_level(right, 0);
	write_extent_buffer(right, root->fs_info->fsid,
			    (unsigned long)btrfs_header_fsid(right),
			    BTRFS_FSID_SIZE);
1804 1805 1806 1807 1808 1809
	if (mid <= slot) {
		if (nritems == 1 ||
		    leaf_space_used(l, mid, nritems - mid) + space_needed >
			BTRFS_LEAF_DATA_SIZE(root)) {
			if (slot >= nritems) {
				btrfs_cpu_key_to_disk(&disk_key, ins_key);
1810
				btrfs_set_header_nritems(right, 0);
1811
				wret = insert_ptr(trans, root, path,
1812
						  &disk_key, right->start,
1813 1814 1815
						  path->slots[1] + 1, 1);
				if (wret)
					ret = wret;
1816 1817
				free_extent_buffer(path->nodes[0]);
				path->nodes[0] = right;
1818 1819 1820 1821 1822
				path->slots[0] = 0;
				path->slots[1] += 1;
				return ret;
			}
			mid = slot;
1823 1824 1825 1826 1827
			if (mid != nritems &&
			    leaf_space_used(l, mid, nritems - mid) +
			    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
				double_split = 1;
			}
1828 1829 1830 1831
		}
	} else {
		if (leaf_space_used(l, 0, mid + 1) + space_needed >
			BTRFS_LEAF_DATA_SIZE(root)) {
1832
			if (!extend && slot == 0) {
1833
				btrfs_cpu_key_to_disk(&disk_key, ins_key);
1834
				btrfs_set_header_nritems(right, 0);
1835 1836
				wret = insert_ptr(trans, root, path,
						  &disk_key,
1837
						  right->start,
1838
						  path->slots[1], 1);
1839 1840
				if (wret)
					ret = wret;
1841 1842
				free_extent_buffer(path->nodes[0]);
				path->nodes[0] = right;
1843
				path->slots[0] = 0;
1844 1845 1846 1847 1848 1849
				if (path->slots[1] == 0) {
					wret = fixup_low_keys(trans, root,
					           path, &disk_key, 1);
					if (wret)
						ret = wret;
				}
1850
				return ret;
1851 1852 1853 1854 1855 1856 1857 1858 1859
			} else if (extend && slot == 0) {
				mid = 1;
			} else {
				mid = slot;
				if (mid != nritems &&
				    leaf_space_used(l, mid, nritems - mid) +
				    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
					double_split = 1;
				}
1860
			}
1861 1862
		}
	}
1863 1864 1865 1866 1867 1868 1869 1870 1871
	nritems = nritems - mid;
	btrfs_set_header_nritems(right, nritems);
	data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);

	copy_extent_buffer(right, l, btrfs_item_nr_offset(0),
			   btrfs_item_nr_offset(mid),
			   nritems * sizeof(struct btrfs_item));

	copy_extent_buffer(right, l,
C
Chris Mason 已提交
1872 1873 1874
		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
		     data_copy_size, btrfs_leaf_data(l) +
		     leaf_data_end(root, l), data_copy_size);
1875

C
Chris Mason 已提交
1876
	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
1877
		      btrfs_item_end_nr(l, mid);
C
Chris Mason 已提交
1878

1879 1880
	for (i = 0; i < nritems; i++) {
		struct btrfs_item *item = btrfs_item_nr(right, i);
1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891
		u32 ioff;

		if (!right->map_token) {
			map_extent_buffer(right, (unsigned long)item,
					sizeof(struct btrfs_item),
					&right->map_token, &right->kaddr,
					&right->map_start, &right->map_len,
					KM_USER1);
		}

		ioff = btrfs_item_offset(right, item);
1892
		btrfs_set_item_offset(right, item, ioff + rt_data_off);
C
Chris Mason 已提交
1893
	}
C
Chris Mason 已提交
1894

1895 1896 1897 1898 1899
	if (right->map_token) {
		unmap_extent_buffer(right, right->map_token, KM_USER1);
		right->map_token = NULL;
	}

1900
	btrfs_set_header_nritems(l, mid);
C
Chris Mason 已提交
1901
	ret = 0;
1902
	btrfs_item_key(right, &disk_key, 0);
1903 1904
	wret = insert_ptr(trans, root, path, &disk_key, right->start,
			  path->slots[1] + 1, 1);
C
Chris Mason 已提交
1905 1906
	if (wret)
		ret = wret;
1907 1908 1909

	btrfs_mark_buffer_dirty(right);
	btrfs_mark_buffer_dirty(l);
1910
	BUG_ON(path->slots[0] != slot);
1911

1912
	if (mid <= slot) {
1913 1914
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = right;
1915 1916
		path->slots[0] -= mid;
		path->slots[1] += 1;
1917
	} else
1918 1919
		free_extent_buffer(right);

1920
	BUG_ON(path->slots[0] < 0);
1921

1922 1923 1924 1925
	if (double_split) {
		BUG_ON(num_doubles != 0);
		num_doubles++;
		goto again;
1926
	}
1927 1928 1929
	return ret;
}

C
Chris Mason 已提交
1930 1931 1932
int btrfs_truncate_item(struct btrfs_trans_handle *trans,
			struct btrfs_root *root,
			struct btrfs_path *path,
1933
			u32 new_size, int from_end)
C
Chris Mason 已提交
1934 1935 1936 1937
{
	int ret = 0;
	int slot;
	int slot_orig;
1938 1939
	struct extent_buffer *leaf;
	struct btrfs_item *item;
C
Chris Mason 已提交
1940 1941 1942 1943 1944 1945 1946 1947
	u32 nritems;
	unsigned int data_end;
	unsigned int old_data_start;
	unsigned int old_size;
	unsigned int size_diff;
	int i;

	slot_orig = path->slots[0];
1948
	leaf = path->nodes[0];
1949 1950 1951 1952 1953
	slot = path->slots[0];

	old_size = btrfs_item_size_nr(leaf, slot);
	if (old_size == new_size)
		return 0;
C
Chris Mason 已提交
1954

1955
	nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
1956 1957
	data_end = leaf_data_end(root, leaf);

1958
	old_data_start = btrfs_item_offset_nr(leaf, slot);
1959

C
Chris Mason 已提交
1960 1961 1962 1963 1964 1965 1966 1967 1968 1969
	size_diff = old_size - new_size;

	BUG_ON(slot < 0);
	BUG_ON(slot >= nritems);

	/*
	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
	 */
	/* first correct the data pointers */
	for (i = slot; i < nritems; i++) {
1970 1971
		u32 ioff;
		item = btrfs_item_nr(leaf, i);
1972 1973 1974 1975 1976 1977 1978 1979 1980

		if (!leaf->map_token) {
			map_extent_buffer(leaf, (unsigned long)item,
					sizeof(struct btrfs_item),
					&leaf->map_token, &leaf->kaddr,
					&leaf->map_start, &leaf->map_len,
					KM_USER1);
		}

1981 1982
		ioff = btrfs_item_offset(leaf, item);
		btrfs_set_item_offset(leaf, item, ioff + size_diff);
C
Chris Mason 已提交
1983
	}
1984 1985 1986 1987 1988 1989

	if (leaf->map_token) {
		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
		leaf->map_token = NULL;
	}

C
Chris Mason 已提交
1990
	/* shift the data */
1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029
	if (from_end) {
		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
			      data_end + size_diff, btrfs_leaf_data(leaf) +
			      data_end, old_data_start + new_size - data_end);
	} else {
		struct btrfs_disk_key disk_key;
		u64 offset;

		btrfs_item_key(leaf, &disk_key, slot);

		if (btrfs_disk_key_type(&disk_key) == BTRFS_EXTENT_DATA_KEY) {
			unsigned long ptr;
			struct btrfs_file_extent_item *fi;

			fi = btrfs_item_ptr(leaf, slot,
					    struct btrfs_file_extent_item);
			fi = (struct btrfs_file_extent_item *)(
			     (unsigned long)fi - size_diff);

			if (btrfs_file_extent_type(leaf, fi) ==
			    BTRFS_FILE_EXTENT_INLINE) {
				ptr = btrfs_item_ptr_offset(leaf, slot);
				memmove_extent_buffer(leaf, ptr,
				        (unsigned long)fi,
				        offsetof(struct btrfs_file_extent_item,
						 disk_bytenr));
			}
		}

		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
			      data_end + size_diff, btrfs_leaf_data(leaf) +
			      data_end, old_data_start - data_end);

		offset = btrfs_disk_key_offset(&disk_key);
		btrfs_set_disk_key_offset(&disk_key, offset + size_diff);
		btrfs_set_item_key(leaf, &disk_key, slot);
		if (slot == 0)
			fixup_low_keys(trans, root, path, &disk_key, 1);
	}
2030 2031 2032 2033

	item = btrfs_item_nr(leaf, slot);
	btrfs_set_item_size(leaf, item, new_size);
	btrfs_mark_buffer_dirty(leaf);
C
Chris Mason 已提交
2034 2035

	ret = 0;
2036 2037
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
C
Chris Mason 已提交
2038
		BUG();
2039
	}
C
Chris Mason 已提交
2040 2041 2042
	return ret;
}

2043 2044 2045
int btrfs_extend_item(struct btrfs_trans_handle *trans,
		      struct btrfs_root *root, struct btrfs_path *path,
		      u32 data_size)
2046 2047 2048 2049
{
	int ret = 0;
	int slot;
	int slot_orig;
2050 2051
	struct extent_buffer *leaf;
	struct btrfs_item *item;
2052 2053 2054 2055 2056 2057 2058
	u32 nritems;
	unsigned int data_end;
	unsigned int old_data;
	unsigned int old_size;
	int i;

	slot_orig = path->slots[0];
2059
	leaf = path->nodes[0];
2060

2061
	nritems = btrfs_header_nritems(leaf);
2062 2063
	data_end = leaf_data_end(root, leaf);

2064 2065
	if (btrfs_leaf_free_space(root, leaf) < data_size) {
		btrfs_print_leaf(root, leaf);
2066
		BUG();
2067
	}
2068
	slot = path->slots[0];
2069
	old_data = btrfs_item_end_nr(leaf, slot);
2070 2071

	BUG_ON(slot < 0);
2072 2073 2074 2075 2076
	if (slot >= nritems) {
		btrfs_print_leaf(root, leaf);
		printk("slot %d too large, nritems %d\n", slot, nritems);
		BUG_ON(1);
	}
2077 2078 2079 2080 2081 2082

	/*
	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
	 */
	/* first correct the data pointers */
	for (i = slot; i < nritems; i++) {
2083 2084
		u32 ioff;
		item = btrfs_item_nr(leaf, i);
2085 2086 2087 2088 2089 2090 2091 2092

		if (!leaf->map_token) {
			map_extent_buffer(leaf, (unsigned long)item,
					sizeof(struct btrfs_item),
					&leaf->map_token, &leaf->kaddr,
					&leaf->map_start, &leaf->map_len,
					KM_USER1);
		}
2093 2094
		ioff = btrfs_item_offset(leaf, item);
		btrfs_set_item_offset(leaf, item, ioff - data_size);
2095
	}
2096

2097 2098 2099 2100 2101
	if (leaf->map_token) {
		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
		leaf->map_token = NULL;
	}

2102
	/* shift the data */
2103
	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2104 2105
		      data_end - data_size, btrfs_leaf_data(leaf) +
		      data_end, old_data - data_end);
2106

2107
	data_end = old_data;
2108 2109 2110 2111
	old_size = btrfs_item_size_nr(leaf, slot);
	item = btrfs_item_nr(leaf, slot);
	btrfs_set_item_size(leaf, item, old_size + data_size);
	btrfs_mark_buffer_dirty(leaf);
2112 2113

	ret = 0;
2114 2115
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
2116
		BUG();
2117
	}
2118 2119 2120
	return ret;
}

C
Chris Mason 已提交
2121 2122 2123 2124
/*
 * Given a key and some data, insert an item into the tree.
 * This does all the path init required, making room in the tree if needed.
 */
2125 2126 2127 2128
int btrfs_insert_empty_item(struct btrfs_trans_handle *trans,
			    struct btrfs_root *root,
			    struct btrfs_path *path,
			    struct btrfs_key *cpu_key, u32 data_size)
2129
{
2130 2131
	struct extent_buffer *leaf;
	struct btrfs_item *item;
C
Chris Mason 已提交
2132
	int ret = 0;
2133
	int slot;
2134
	int slot_orig;
2135
	u32 nritems;
2136
	unsigned int data_end;
C
Chris Mason 已提交
2137 2138 2139
	struct btrfs_disk_key disk_key;

	btrfs_cpu_key_to_disk(&disk_key, cpu_key);
2140

C
Chris Mason 已提交
2141
	/* create a root if there isn't one */
C
Chris Mason 已提交
2142
	if (!root->node)
C
Chris Mason 已提交
2143
		BUG();
2144

2145
	ret = btrfs_search_slot(trans, root, cpu_key, path, data_size, 1);
2146
	if (ret == 0) {
2147
		return -EEXIST;
C
Chris Mason 已提交
2148
	}
2149 2150
	if (ret < 0)
		goto out;
2151

2152
	slot_orig = path->slots[0];
2153
	leaf = path->nodes[0];
C
Chris Mason 已提交
2154

2155
	nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
2156
	data_end = leaf_data_end(root, leaf);
2157

C
Chris Mason 已提交
2158
	if (btrfs_leaf_free_space(root, leaf) <
2159
	    sizeof(struct btrfs_item) + data_size) {
2160 2161 2162
		btrfs_print_leaf(root, leaf);
		printk("not enough freespace need %u have %d\n",
		       data_size, btrfs_leaf_free_space(root, leaf));
2163
		BUG();
2164
	}
2165

2166
	slot = path->slots[0];
2167
	BUG_ON(slot < 0);
2168

2169 2170
	if (slot != nritems) {
		int i;
2171
		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
2172

2173 2174 2175 2176 2177 2178
		if (old_data < data_end) {
			btrfs_print_leaf(root, leaf);
			printk("slot %d old_data %d data_end %d\n",
			       slot, old_data, data_end);
			BUG_ON(1);
		}
2179 2180 2181 2182
		/*
		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
		 */
		/* first correct the data pointers */
2183
		WARN_ON(leaf->map_token);
C
Chris Mason 已提交
2184
		for (i = slot; i < nritems; i++) {
2185
			u32 ioff;
2186

2187
			item = btrfs_item_nr(leaf, i);
2188 2189 2190 2191 2192 2193 2194 2195
			if (!leaf->map_token) {
				map_extent_buffer(leaf, (unsigned long)item,
					sizeof(struct btrfs_item),
					&leaf->map_token, &leaf->kaddr,
					&leaf->map_start, &leaf->map_len,
					KM_USER1);
			}

2196 2197
			ioff = btrfs_item_offset(leaf, item);
			btrfs_set_item_offset(leaf, item, ioff - data_size);
C
Chris Mason 已提交
2198
		}
2199 2200 2201 2202
		if (leaf->map_token) {
			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
			leaf->map_token = NULL;
		}
2203 2204

		/* shift the items */
2205 2206
		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1),
			      btrfs_item_nr_offset(slot),
C
Chris Mason 已提交
2207
			      (nritems - slot) * sizeof(struct btrfs_item));
2208 2209

		/* shift the data */
2210
		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
C
Chris Mason 已提交
2211 2212
			      data_end - data_size, btrfs_leaf_data(leaf) +
			      data_end, old_data - data_end);
2213 2214
		data_end = old_data;
	}
2215

2216
	/* setup the item for the new data */
2217 2218 2219 2220 2221 2222
	btrfs_set_item_key(leaf, &disk_key, slot);
	item = btrfs_item_nr(leaf, slot);
	btrfs_set_item_offset(leaf, item, data_end - data_size);
	btrfs_set_item_size(leaf, item, data_size);
	btrfs_set_header_nritems(leaf, nritems + 1);
	btrfs_mark_buffer_dirty(leaf);
C
Chris Mason 已提交
2223 2224

	ret = 0;
2225
	if (slot == 0)
2226
		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
C
Chris Mason 已提交
2227

2228 2229
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
2230
		BUG();
2231
	}
2232
out:
2233 2234 2235 2236 2237 2238 2239
	return ret;
}

/*
 * Given a key and some data, insert an item into the tree.
 * This does all the path init required, making room in the tree if needed.
 */
2240 2241 2242
int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_key *cpu_key, void *data, u32
		      data_size)
2243 2244
{
	int ret = 0;
C
Chris Mason 已提交
2245
	struct btrfs_path *path;
2246 2247
	struct extent_buffer *leaf;
	unsigned long ptr;
2248

C
Chris Mason 已提交
2249 2250 2251
	path = btrfs_alloc_path();
	BUG_ON(!path);
	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
2252
	if (!ret) {
2253 2254 2255 2256
		leaf = path->nodes[0];
		ptr = btrfs_item_ptr_offset(leaf, path->slots[0]);
		write_extent_buffer(leaf, data, ptr, data_size);
		btrfs_mark_buffer_dirty(leaf);
2257
	}
C
Chris Mason 已提交
2258
	btrfs_free_path(path);
C
Chris Mason 已提交
2259
	return ret;
2260 2261
}

C
Chris Mason 已提交
2262
/*
C
Chris Mason 已提交
2263
 * delete the pointer from a given node.
C
Chris Mason 已提交
2264 2265 2266 2267 2268
 *
 * If the delete empties a node, the node is removed from the tree,
 * continuing all the way the root if required.  The root is converted into
 * a leaf if all the nodes are emptied.
 */
2269 2270
static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
		   struct btrfs_path *path, int level, int slot)
2271
{
2272
	struct extent_buffer *parent = path->nodes[level];
2273
	u32 nritems;
C
Chris Mason 已提交
2274
	int ret = 0;
2275
	int wret;
2276

2277
	nritems = btrfs_header_nritems(parent);
2278
	if (slot != nritems -1) {
2279 2280 2281
		memmove_extent_buffer(parent,
			      btrfs_node_key_ptr_offset(slot),
			      btrfs_node_key_ptr_offset(slot + 1),
C
Chris Mason 已提交
2282 2283
			      sizeof(struct btrfs_key_ptr) *
			      (nritems - slot - 1));
2284
	}
2285
	nritems--;
2286
	btrfs_set_header_nritems(parent, nritems);
2287
	if (nritems == 0 && parent == root->node) {
2288
		BUG_ON(btrfs_header_level(root->node) != 1);
2289
		/* just turn the root into a leaf and break */
2290
		btrfs_set_header_level(root->node, 0);
2291
	} else if (slot == 0) {
2292 2293 2294 2295
		struct btrfs_disk_key disk_key;

		btrfs_node_key(parent, &disk_key, 0);
		wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
C
Chris Mason 已提交
2296 2297
		if (wret)
			ret = wret;
2298
	}
C
Chris Mason 已提交
2299
	btrfs_mark_buffer_dirty(parent);
C
Chris Mason 已提交
2300
	return ret;
2301 2302
}

C
Chris Mason 已提交
2303 2304 2305 2306
/*
 * delete the item at the leaf level in path.  If that empties
 * the leaf, remove it from the tree
 */
2307 2308
int btrfs_del_item(struct btrfs_trans_handle *trans, struct btrfs_root *root,
		   struct btrfs_path *path)
2309 2310
{
	int slot;
2311 2312
	struct extent_buffer *leaf;
	struct btrfs_item *item;
2313 2314
	int doff;
	int dsize;
C
Chris Mason 已提交
2315 2316
	int ret = 0;
	int wret;
2317
	u32 nritems;
2318

2319
	leaf = path->nodes[0];
2320
	slot = path->slots[0];
2321 2322 2323
	doff = btrfs_item_offset_nr(leaf, slot);
	dsize = btrfs_item_size_nr(leaf, slot);
	nritems = btrfs_header_nritems(leaf);
2324

2325
	if (slot != nritems - 1) {
2326
		int i;
C
Chris Mason 已提交
2327
		int data_end = leaf_data_end(root, leaf);
2328 2329

		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
C
Chris Mason 已提交
2330 2331 2332
			      data_end + dsize,
			      btrfs_leaf_data(leaf) + data_end,
			      doff - data_end);
2333

C
Chris Mason 已提交
2334
		for (i = slot + 1; i < nritems; i++) {
2335
			u32 ioff;
2336

2337
			item = btrfs_item_nr(leaf, i);
2338 2339 2340 2341 2342 2343 2344
			if (!leaf->map_token) {
				map_extent_buffer(leaf, (unsigned long)item,
					sizeof(struct btrfs_item),
					&leaf->map_token, &leaf->kaddr,
					&leaf->map_start, &leaf->map_len,
					KM_USER1);
			}
2345 2346
			ioff = btrfs_item_offset(leaf, item);
			btrfs_set_item_offset(leaf, item, ioff + dsize);
C
Chris Mason 已提交
2347
		}
2348 2349 2350 2351 2352 2353

		if (leaf->map_token) {
			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
			leaf->map_token = NULL;
		}

2354 2355
		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
			      btrfs_item_nr_offset(slot + 1),
C
Chris Mason 已提交
2356 2357
			      sizeof(struct btrfs_item) *
			      (nritems - slot - 1));
2358
	}
2359
	btrfs_set_header_nritems(leaf, nritems - 1);
2360
	nritems--;
2361

C
Chris Mason 已提交
2362
	/* delete the leaf if we've emptied it */
2363
	if (nritems == 0) {
2364 2365
		if (leaf == root->node) {
			btrfs_set_header_level(leaf, 0);
2366
		} else {
2367 2368
			clean_tree_block(trans, root, leaf);
			wait_on_tree_block_writeback(root, leaf);
2369
			wret = del_ptr(trans, root, path, 1, path->slots[1]);
C
Chris Mason 已提交
2370 2371
			if (wret)
				ret = wret;
2372
			wret = btrfs_free_extent(trans, root,
2373
						 leaf->start, leaf->len, 1);
C
Chris Mason 已提交
2374 2375
			if (wret)
				ret = wret;
2376
		}
2377
	} else {
2378
		int used = leaf_space_used(leaf, 0, nritems);
C
Chris Mason 已提交
2379
		if (slot == 0) {
2380 2381 2382
			struct btrfs_disk_key disk_key;

			btrfs_item_key(leaf, &disk_key, 0);
2383
			wret = fixup_low_keys(trans, root, path,
2384
					      &disk_key, 1);
C
Chris Mason 已提交
2385 2386 2387 2388
			if (wret)
				ret = wret;
		}

C
Chris Mason 已提交
2389
		/* delete the leaf if it is mostly empty */
2390
		if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) {
2391 2392 2393 2394
			/* push_leaf_left fixes the path.
			 * make sure the path still points to our leaf
			 * for possible call to del_ptr below
			 */
2395
			slot = path->slots[1];
2396 2397
			extent_buffer_get(leaf);

2398
			wret = push_leaf_right(trans, root, path, 1);
2399
			if (wret < 0 && wret != -ENOSPC)
C
Chris Mason 已提交
2400
				ret = wret;
2401 2402 2403

			if (path->nodes[0] == leaf &&
			    btrfs_header_nritems(leaf)) {
2404
				wret = push_leaf_left(trans, root, path, 1);
2405
				if (wret < 0 && wret != -ENOSPC)
C
Chris Mason 已提交
2406 2407
					ret = wret;
			}
2408 2409

			if (btrfs_header_nritems(leaf) == 0) {
2410 2411
				u64 bytenr = leaf->start;
				u32 blocksize = leaf->len;
2412 2413 2414 2415

				clean_tree_block(trans, root, leaf);
				wait_on_tree_block_writeback(root, leaf);

2416
				wret = del_ptr(trans, root, path, 1, slot);
C
Chris Mason 已提交
2417 2418
				if (wret)
					ret = wret;
2419 2420

				free_extent_buffer(leaf);
2421 2422
				wret = btrfs_free_extent(trans, root, bytenr,
							 blocksize, 1);
C
Chris Mason 已提交
2423 2424
				if (wret)
					ret = wret;
C
Chris Mason 已提交
2425
			} else {
2426 2427
				btrfs_mark_buffer_dirty(leaf);
				free_extent_buffer(leaf);
2428
			}
2429
		} else {
2430
			btrfs_mark_buffer_dirty(leaf);
2431 2432
		}
	}
C
Chris Mason 已提交
2433
	return ret;
2434 2435
}

C
Chris Mason 已提交
2436 2437
/*
 * walk up the tree as far as required to find the next leaf.
C
Chris Mason 已提交
2438 2439
 * returns 0 if it found something or 1 if there are no greater leaves.
 * returns < 0 on io errors.
C
Chris Mason 已提交
2440
 */
C
Chris Mason 已提交
2441
int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
2442 2443 2444
{
	int slot;
	int level = 1;
2445
	u64 bytenr;
2446 2447
	struct extent_buffer *c;
	struct extent_buffer *next = NULL;
2448

C
Chris Mason 已提交
2449
	while(level < BTRFS_MAX_LEVEL) {
2450
		if (!path->nodes[level])
C
Chris Mason 已提交
2451
			return 1;
2452

2453 2454
		slot = path->slots[level] + 1;
		c = path->nodes[level];
2455
		if (slot >= btrfs_header_nritems(c)) {
2456 2457 2458
			level++;
			continue;
		}
2459

2460
		bytenr = btrfs_node_blockptr(c, slot);
C
Chris Mason 已提交
2461
		if (next)
2462 2463
			free_extent_buffer(next);

2464 2465
		if (path->reada)
			reada_for_search(root, path, level, slot);
2466

2467 2468
		next = read_tree_block(root, bytenr,
				       btrfs_level_size(root, level -1));
2469 2470 2471 2472 2473 2474
		break;
	}
	path->slots[level] = slot;
	while(1) {
		level--;
		c = path->nodes[level];
2475
		free_extent_buffer(c);
2476 2477 2478 2479
		path->nodes[level] = next;
		path->slots[level] = 0;
		if (!level)
			break;
2480
		if (path->reada)
Y
Yan 已提交
2481
			reada_for_search(root, path, level, 0);
2482 2483
		next = read_tree_block(root, btrfs_node_blockptr(next, 0),
				       btrfs_level_size(root, level - 1));
2484 2485 2486
	}
	return 0;
}