ctree.c 122.3 KB
Newer Older
C
Chris Mason 已提交
1
/*
C
Chris Mason 已提交
2
 * Copyright (C) 2007,2008 Oracle.  All rights reserved.
C
Chris Mason 已提交
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
 *
 * 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
#include <linux/slab.h>
21
#include <linux/rbtree.h>
22 23
#include "ctree.h"
#include "disk-io.h"
24
#include "transaction.h"
25
#include "print-tree.h"
26
#include "locking.h"
27

28 29 30
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
31
		      *root, struct btrfs_key *ins_key,
32
		      struct btrfs_path *path, int data_size, int extend);
33 34
static int push_node_left(struct btrfs_trans_handle *trans,
			  struct btrfs_root *root, struct extent_buffer *dst,
35
			  struct extent_buffer *src, int empty);
36 37 38 39
static int balance_node_right(struct btrfs_trans_handle *trans,
			      struct btrfs_root *root,
			      struct extent_buffer *dst_buf,
			      struct extent_buffer *src_buf);
40
static void del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
41
		   struct btrfs_path *path, int level, int slot);
42

C
Chris Mason 已提交
43
struct btrfs_path *btrfs_alloc_path(void)
C
Chris Mason 已提交
44
{
C
Chris Mason 已提交
45
	struct btrfs_path *path;
J
Jeff Mahoney 已提交
46
	path = kmem_cache_zalloc(btrfs_path_cachep, GFP_NOFS);
C
Chris Mason 已提交
47
	return path;
C
Chris Mason 已提交
48 49
}

50 51 52 53 54 55 56 57
/*
 * set all locked nodes in the path to blocking locks.  This should
 * be done before scheduling
 */
noinline void btrfs_set_path_blocking(struct btrfs_path *p)
{
	int i;
	for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
58 59 60 61 62 63 64
		if (!p->nodes[i] || !p->locks[i])
			continue;
		btrfs_set_lock_blocking_rw(p->nodes[i], p->locks[i]);
		if (p->locks[i] == BTRFS_READ_LOCK)
			p->locks[i] = BTRFS_READ_LOCK_BLOCKING;
		else if (p->locks[i] == BTRFS_WRITE_LOCK)
			p->locks[i] = BTRFS_WRITE_LOCK_BLOCKING;
65 66 67 68 69
	}
}

/*
 * reset all the locked nodes in the patch to spinning locks.
70 71 72 73 74
 *
 * held is used to keep lockdep happy, when lockdep is enabled
 * we set held to a blocking lock before we go around and
 * retake all the spinlocks in the path.  You can safely use NULL
 * for held
75
 */
76
noinline void btrfs_clear_path_blocking(struct btrfs_path *p,
77
					struct extent_buffer *held, int held_rw)
78 79
{
	int i;
80 81 82 83 84 85 86 87

#ifdef CONFIG_DEBUG_LOCK_ALLOC
	/* lockdep really cares that we take all of these spinlocks
	 * in the right order.  If any of the locks in the path are not
	 * currently blocking, it is going to complain.  So, make really
	 * really sure by forcing the path to blocking before we clear
	 * the path blocking.
	 */
88 89 90 91 92 93 94
	if (held) {
		btrfs_set_lock_blocking_rw(held, held_rw);
		if (held_rw == BTRFS_WRITE_LOCK)
			held_rw = BTRFS_WRITE_LOCK_BLOCKING;
		else if (held_rw == BTRFS_READ_LOCK)
			held_rw = BTRFS_READ_LOCK_BLOCKING;
	}
95 96 97 98
	btrfs_set_path_blocking(p);
#endif

	for (i = BTRFS_MAX_LEVEL - 1; i >= 0; i--) {
99 100 101 102 103 104 105
		if (p->nodes[i] && p->locks[i]) {
			btrfs_clear_lock_blocking_rw(p->nodes[i], p->locks[i]);
			if (p->locks[i] == BTRFS_WRITE_LOCK_BLOCKING)
				p->locks[i] = BTRFS_WRITE_LOCK;
			else if (p->locks[i] == BTRFS_READ_LOCK_BLOCKING)
				p->locks[i] = BTRFS_READ_LOCK;
		}
106
	}
107 108 109

#ifdef CONFIG_DEBUG_LOCK_ALLOC
	if (held)
110
		btrfs_clear_lock_blocking_rw(held, held_rw);
111
#endif
112 113
}

C
Chris Mason 已提交
114
/* this also releases the path */
C
Chris Mason 已提交
115
void btrfs_free_path(struct btrfs_path *p)
116
{
117 118
	if (!p)
		return;
119
	btrfs_release_path(p);
C
Chris Mason 已提交
120
	kmem_cache_free(btrfs_path_cachep, p);
121 122
}

C
Chris Mason 已提交
123 124 125 126 127 128
/*
 * path release drops references on the extent buffers in the path
 * and it drops any locks held by this path
 *
 * It is safe to call this on paths that no locks or extent buffers held.
 */
129
noinline void btrfs_release_path(struct btrfs_path *p)
130 131
{
	int i;
132

C
Chris Mason 已提交
133
	for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
134
		p->slots[i] = 0;
135
		if (!p->nodes[i])
136 137
			continue;
		if (p->locks[i]) {
138
			btrfs_tree_unlock_rw(p->nodes[i], p->locks[i]);
139 140
			p->locks[i] = 0;
		}
141
		free_extent_buffer(p->nodes[i]);
142
		p->nodes[i] = NULL;
143 144 145
	}
}

C
Chris Mason 已提交
146 147 148 149 150 151 152 153 154 155
/*
 * safely gets a reference on the root node of a tree.  A lock
 * is not taken, so a concurrent writer may put a different node
 * at the root of the tree.  See btrfs_lock_root_node for the
 * looping required.
 *
 * The extent buffer returned by this has a reference taken, so
 * it won't disappear.  It may stop being the root of the tree
 * at any time because there are no locks held.
 */
156 157 158
struct extent_buffer *btrfs_root_node(struct btrfs_root *root)
{
	struct extent_buffer *eb;
159

160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176
	while (1) {
		rcu_read_lock();
		eb = rcu_dereference(root->node);

		/*
		 * RCU really hurts here, we could free up the root node because
		 * it was cow'ed but we may not get the new root node yet so do
		 * the inc_not_zero dance and if it doesn't work then
		 * synchronize_rcu and try again.
		 */
		if (atomic_inc_not_zero(&eb->refs)) {
			rcu_read_unlock();
			break;
		}
		rcu_read_unlock();
		synchronize_rcu();
	}
177 178 179
	return eb;
}

C
Chris Mason 已提交
180 181 182 183
/* loop around taking references on and locking the root node of the
 * tree until you end up with a lock on the root.  A locked buffer
 * is returned, with a reference held.
 */
184 185 186 187
struct extent_buffer *btrfs_lock_root_node(struct btrfs_root *root)
{
	struct extent_buffer *eb;

C
Chris Mason 已提交
188
	while (1) {
189 190
		eb = btrfs_root_node(root);
		btrfs_tree_lock(eb);
191
		if (eb == root->node)
192 193 194 195 196 197 198
			break;
		btrfs_tree_unlock(eb);
		free_extent_buffer(eb);
	}
	return eb;
}

199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217
/* loop around taking references on and locking the root node of the
 * tree until you end up with a lock on the root.  A locked buffer
 * is returned, with a reference held.
 */
struct extent_buffer *btrfs_read_lock_root_node(struct btrfs_root *root)
{
	struct extent_buffer *eb;

	while (1) {
		eb = btrfs_root_node(root);
		btrfs_tree_read_lock(eb);
		if (eb == root->node)
			break;
		btrfs_tree_read_unlock(eb);
		free_extent_buffer(eb);
	}
	return eb;
}

C
Chris Mason 已提交
218 219 220 221
/* cowonly root (everything not a reference counted cow subvolume), just get
 * put onto a simple dirty list.  transaction.c walks this to make sure they
 * get properly updated on disk.
 */
222 223
static void add_root_to_dirty_list(struct btrfs_root *root)
{
224
	spin_lock(&root->fs_info->trans_lock);
225 226 227 228
	if (root->track_dirty && list_empty(&root->dirty_list)) {
		list_add(&root->dirty_list,
			 &root->fs_info->dirty_cowonly_roots);
	}
229
	spin_unlock(&root->fs_info->trans_lock);
230 231
}

C
Chris Mason 已提交
232 233 234 235 236
/*
 * used by snapshot creation to make a copy of a root for a tree with
 * a given objectid.  The buffer with the new root node is returned in
 * cow_ret, and this func returns zero on success or a negative error code.
 */
237 238 239 240 241 242 243 244
int btrfs_copy_root(struct btrfs_trans_handle *trans,
		      struct btrfs_root *root,
		      struct extent_buffer *buf,
		      struct extent_buffer **cow_ret, u64 new_root_objectid)
{
	struct extent_buffer *cow;
	int ret = 0;
	int level;
245
	struct btrfs_disk_key disk_key;
246 247 248 249 250 251

	WARN_ON(root->ref_cows && trans->transid !=
		root->fs_info->running_transaction->transid);
	WARN_ON(root->ref_cows && trans->transid != root->last_trans);

	level = btrfs_header_level(buf);
252 253 254 255
	if (level == 0)
		btrfs_item_key(buf, &disk_key, 0);
	else
		btrfs_node_key(buf, &disk_key, 0);
Z
Zheng Yan 已提交
256

257 258
	cow = btrfs_alloc_free_block(trans, root, buf->len, 0,
				     new_root_objectid, &disk_key, level,
259
				     buf->start, 0);
260
	if (IS_ERR(cow))
261 262 263 264 265
		return PTR_ERR(cow);

	copy_extent_buffer(cow, buf, 0, 0, cow->len);
	btrfs_set_header_bytenr(cow, cow->start);
	btrfs_set_header_generation(cow, trans->transid);
266 267 268 269 270 271 272
	btrfs_set_header_backref_rev(cow, BTRFS_MIXED_BACKREF_REV);
	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN |
				     BTRFS_HEADER_FLAG_RELOC);
	if (new_root_objectid == BTRFS_TREE_RELOC_OBJECTID)
		btrfs_set_header_flag(cow, BTRFS_HEADER_FLAG_RELOC);
	else
		btrfs_set_header_owner(cow, new_root_objectid);
273

Y
Yan Zheng 已提交
274 275 276 277
	write_extent_buffer(cow, root->fs_info->fsid,
			    (unsigned long)btrfs_header_fsid(cow),
			    BTRFS_FSID_SIZE);

278
	WARN_ON(btrfs_header_generation(buf) > trans->transid);
279
	if (new_root_objectid == BTRFS_TREE_RELOC_OBJECTID)
A
Arne Jansen 已提交
280
		ret = btrfs_inc_ref(trans, root, cow, 1, 1);
281
	else
A
Arne Jansen 已提交
282
		ret = btrfs_inc_ref(trans, root, cow, 0, 1);
283

284 285 286 287 288 289 290 291
	if (ret)
		return ret;

	btrfs_mark_buffer_dirty(cow);
	*cow_ret = cow;
	return 0;
}

292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 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 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697
enum mod_log_op {
	MOD_LOG_KEY_REPLACE,
	MOD_LOG_KEY_ADD,
	MOD_LOG_KEY_REMOVE,
	MOD_LOG_KEY_REMOVE_WHILE_FREEING,
	MOD_LOG_KEY_REMOVE_WHILE_MOVING,
	MOD_LOG_MOVE_KEYS,
	MOD_LOG_ROOT_REPLACE,
};

struct tree_mod_move {
	int dst_slot;
	int nr_items;
};

struct tree_mod_root {
	u64 logical;
	u8 level;
};

struct tree_mod_elem {
	struct rb_node node;
	u64 index;		/* shifted logical */
	struct seq_list elem;
	enum mod_log_op op;

	/* this is used for MOD_LOG_KEY_* and MOD_LOG_MOVE_KEYS operations */
	int slot;

	/* this is used for MOD_LOG_KEY* and MOD_LOG_ROOT_REPLACE */
	u64 generation;

	/* those are used for op == MOD_LOG_KEY_{REPLACE,REMOVE} */
	struct btrfs_disk_key key;
	u64 blockptr;

	/* this is used for op == MOD_LOG_MOVE_KEYS */
	struct tree_mod_move move;

	/* this is used for op == MOD_LOG_ROOT_REPLACE */
	struct tree_mod_root old_root;
};

static inline void
__get_tree_mod_seq(struct btrfs_fs_info *fs_info, struct seq_list *elem)
{
	elem->seq = atomic_inc_return(&fs_info->tree_mod_seq);
	list_add_tail(&elem->list, &fs_info->tree_mod_seq_list);
}

void btrfs_get_tree_mod_seq(struct btrfs_fs_info *fs_info,
			    struct seq_list *elem)
{
	elem->flags = 1;
	spin_lock(&fs_info->tree_mod_seq_lock);
	__get_tree_mod_seq(fs_info, elem);
	spin_unlock(&fs_info->tree_mod_seq_lock);
}

void btrfs_put_tree_mod_seq(struct btrfs_fs_info *fs_info,
			    struct seq_list *elem)
{
	struct rb_root *tm_root;
	struct rb_node *node;
	struct rb_node *next;
	struct seq_list *cur_elem;
	struct tree_mod_elem *tm;
	u64 min_seq = (u64)-1;
	u64 seq_putting = elem->seq;

	if (!seq_putting)
		return;

	BUG_ON(!(elem->flags & 1));
	spin_lock(&fs_info->tree_mod_seq_lock);
	list_del(&elem->list);

	list_for_each_entry(cur_elem, &fs_info->tree_mod_seq_list, list) {
		if ((cur_elem->flags & 1) && cur_elem->seq < min_seq) {
			if (seq_putting > cur_elem->seq) {
				/*
				 * blocker with lower sequence number exists, we
				 * cannot remove anything from the log
				 */
				goto out;
			}
			min_seq = cur_elem->seq;
		}
	}

	/*
	 * anything that's lower than the lowest existing (read: blocked)
	 * sequence number can be removed from the tree.
	 */
	write_lock(&fs_info->tree_mod_log_lock);
	tm_root = &fs_info->tree_mod_log;
	for (node = rb_first(tm_root); node; node = next) {
		next = rb_next(node);
		tm = container_of(node, struct tree_mod_elem, node);
		if (tm->elem.seq > min_seq)
			continue;
		rb_erase(node, tm_root);
		list_del(&tm->elem.list);
		kfree(tm);
	}
	write_unlock(&fs_info->tree_mod_log_lock);
out:
	spin_unlock(&fs_info->tree_mod_seq_lock);
}

/*
 * key order of the log:
 *       index -> sequence
 *
 * the index is the shifted logical of the *new* root node for root replace
 * operations, or the shifted logical of the affected block for all other
 * operations.
 */
static noinline int
__tree_mod_log_insert(struct btrfs_fs_info *fs_info, struct tree_mod_elem *tm)
{
	struct rb_root *tm_root;
	struct rb_node **new;
	struct rb_node *parent = NULL;
	struct tree_mod_elem *cur;
	int ret = 0;

	BUG_ON(!tm || !tm->elem.seq);

	write_lock(&fs_info->tree_mod_log_lock);
	tm_root = &fs_info->tree_mod_log;
	new = &tm_root->rb_node;
	while (*new) {
		cur = container_of(*new, struct tree_mod_elem, node);
		parent = *new;
		if (cur->index < tm->index)
			new = &((*new)->rb_left);
		else if (cur->index > tm->index)
			new = &((*new)->rb_right);
		else if (cur->elem.seq < tm->elem.seq)
			new = &((*new)->rb_left);
		else if (cur->elem.seq > tm->elem.seq)
			new = &((*new)->rb_right);
		else {
			kfree(tm);
			ret = -EEXIST;
			goto unlock;
		}
	}

	rb_link_node(&tm->node, parent, new);
	rb_insert_color(&tm->node, tm_root);
unlock:
	write_unlock(&fs_info->tree_mod_log_lock);
	return ret;
}

int tree_mod_alloc(struct btrfs_fs_info *fs_info, gfp_t flags,
		   struct tree_mod_elem **tm_ret)
{
	struct tree_mod_elem *tm;
	u64 seq = 0;

	smp_mb();
	if (list_empty(&fs_info->tree_mod_seq_list))
		return 0;

	tm = *tm_ret = kzalloc(sizeof(*tm), flags);
	if (!tm)
		return -ENOMEM;

	__get_tree_mod_seq(fs_info, &tm->elem);
	seq = tm->elem.seq;
	tm->elem.flags = 0;

	return seq;
}

static noinline int
tree_mod_log_insert_key_mask(struct btrfs_fs_info *fs_info,
			     struct extent_buffer *eb, int slot,
			     enum mod_log_op op, gfp_t flags)
{
	struct tree_mod_elem *tm;
	int ret;

	ret = tree_mod_alloc(fs_info, flags, &tm);
	if (ret <= 0)
		return ret;

	tm->index = eb->start >> PAGE_CACHE_SHIFT;
	if (op != MOD_LOG_KEY_ADD) {
		btrfs_node_key(eb, &tm->key, slot);
		tm->blockptr = btrfs_node_blockptr(eb, slot);
	}
	tm->op = op;
	tm->slot = slot;
	tm->generation = btrfs_node_ptr_generation(eb, slot);

	return __tree_mod_log_insert(fs_info, tm);
}

static noinline int
tree_mod_log_insert_key(struct btrfs_fs_info *fs_info, struct extent_buffer *eb,
			int slot, enum mod_log_op op)
{
	return tree_mod_log_insert_key_mask(fs_info, eb, slot, op, GFP_NOFS);
}

static noinline int
tree_mod_log_insert_move(struct btrfs_fs_info *fs_info,
			 struct extent_buffer *eb, int dst_slot, int src_slot,
			 int nr_items, gfp_t flags)
{
	struct tree_mod_elem *tm;
	int ret;
	int i;

	ret = tree_mod_alloc(fs_info, flags, &tm);
	if (ret <= 0)
		return ret;

	for (i = 0; i + dst_slot < src_slot && i < nr_items; i++) {
		ret = tree_mod_log_insert_key(fs_info, eb, i + dst_slot,
					      MOD_LOG_KEY_REMOVE_WHILE_MOVING);
		BUG_ON(ret < 0);
	}

	tm->index = eb->start >> PAGE_CACHE_SHIFT;
	tm->slot = src_slot;
	tm->move.dst_slot = dst_slot;
	tm->move.nr_items = nr_items;
	tm->op = MOD_LOG_MOVE_KEYS;

	return __tree_mod_log_insert(fs_info, tm);
}

static noinline int
tree_mod_log_insert_root(struct btrfs_fs_info *fs_info,
			 struct extent_buffer *old_root,
			 struct extent_buffer *new_root, gfp_t flags)
{
	struct tree_mod_elem *tm;
	int ret;

	ret = tree_mod_alloc(fs_info, flags, &tm);
	if (ret <= 0)
		return ret;

	tm->index = new_root->start >> PAGE_CACHE_SHIFT;
	tm->old_root.logical = old_root->start;
	tm->old_root.level = btrfs_header_level(old_root);
	tm->generation = btrfs_header_generation(old_root);
	tm->op = MOD_LOG_ROOT_REPLACE;

	return __tree_mod_log_insert(fs_info, tm);
}

static struct tree_mod_elem *
__tree_mod_log_search(struct btrfs_fs_info *fs_info, u64 start, u64 min_seq,
		      int smallest)
{
	struct rb_root *tm_root;
	struct rb_node *node;
	struct tree_mod_elem *cur = NULL;
	struct tree_mod_elem *found = NULL;
	u64 index = start >> PAGE_CACHE_SHIFT;

	read_lock(&fs_info->tree_mod_log_lock);
	tm_root = &fs_info->tree_mod_log;
	node = tm_root->rb_node;
	while (node) {
		cur = container_of(node, struct tree_mod_elem, node);
		if (cur->index < index) {
			node = node->rb_left;
		} else if (cur->index > index) {
			node = node->rb_right;
		} else if (cur->elem.seq < min_seq) {
			node = node->rb_left;
		} else if (!smallest) {
			/* we want the node with the highest seq */
			if (found)
				BUG_ON(found->elem.seq > cur->elem.seq);
			found = cur;
			node = node->rb_left;
		} else if (cur->elem.seq > min_seq) {
			/* we want the node with the smallest seq */
			if (found)
				BUG_ON(found->elem.seq < cur->elem.seq);
			found = cur;
			node = node->rb_right;
		} else {
			found = cur;
			break;
		}
	}
	read_unlock(&fs_info->tree_mod_log_lock);

	return found;
}

/*
 * this returns the element from the log with the smallest time sequence
 * value that's in the log (the oldest log item). any element with a time
 * sequence lower than min_seq will be ignored.
 */
static struct tree_mod_elem *
tree_mod_log_search_oldest(struct btrfs_fs_info *fs_info, u64 start,
			   u64 min_seq)
{
	return __tree_mod_log_search(fs_info, start, min_seq, 1);
}

/*
 * this returns the element from the log with the largest time sequence
 * value that's in the log (the most recent log item). any element with
 * a time sequence lower than min_seq will be ignored.
 */
static struct tree_mod_elem *
tree_mod_log_search(struct btrfs_fs_info *fs_info, u64 start, u64 min_seq)
{
	return __tree_mod_log_search(fs_info, start, min_seq, 0);
}

static inline void
tree_mod_log_eb_copy(struct btrfs_fs_info *fs_info, struct extent_buffer *dst,
		     struct extent_buffer *src, unsigned long dst_offset,
		     unsigned long src_offset, int nr_items)
{
	int ret;
	int i;

	smp_mb();
	if (list_empty(&fs_info->tree_mod_seq_list))
		return;

	if (btrfs_header_level(dst) == 0 && btrfs_header_level(src) == 0)
		return;

	/* speed this up by single seq for all operations? */
	for (i = 0; i < nr_items; i++) {
		ret = tree_mod_log_insert_key(fs_info, src, i + src_offset,
					      MOD_LOG_KEY_REMOVE);
		BUG_ON(ret < 0);
		ret = tree_mod_log_insert_key(fs_info, dst, i + dst_offset,
					      MOD_LOG_KEY_ADD);
		BUG_ON(ret < 0);
	}
}

static inline void
tree_mod_log_eb_move(struct btrfs_fs_info *fs_info, struct extent_buffer *dst,
		     int dst_offset, int src_offset, int nr_items)
{
	int ret;
	ret = tree_mod_log_insert_move(fs_info, dst, dst_offset, src_offset,
				       nr_items, GFP_NOFS);
	BUG_ON(ret < 0);
}

static inline void
tree_mod_log_set_node_key(struct btrfs_fs_info *fs_info,
			  struct extent_buffer *eb,
			  struct btrfs_disk_key *disk_key, int slot, int atomic)
{
	int ret;

	ret = tree_mod_log_insert_key_mask(fs_info, eb, slot,
					   MOD_LOG_KEY_REPLACE,
					   atomic ? GFP_ATOMIC : GFP_NOFS);
	BUG_ON(ret < 0);
}

static void tree_mod_log_free_eb(struct btrfs_fs_info *fs_info,
				 struct extent_buffer *eb)
{
	int i;
	int ret;
	u32 nritems;

	smp_mb();
	if (list_empty(&fs_info->tree_mod_seq_list))
		return;

	if (btrfs_header_level(eb) == 0)
		return;

	nritems = btrfs_header_nritems(eb);
	for (i = nritems - 1; i >= 0; i--) {
		ret = tree_mod_log_insert_key(fs_info, eb, i,
					      MOD_LOG_KEY_REMOVE_WHILE_FREEING);
		BUG_ON(ret < 0);
	}
}

static inline void
tree_mod_log_set_root_pointer(struct btrfs_root *root,
			      struct extent_buffer *new_root_node)
{
	int ret;
	tree_mod_log_free_eb(root->fs_info, root->node);
	ret = tree_mod_log_insert_root(root->fs_info, root->node,
				       new_root_node, GFP_NOFS);
	BUG_ON(ret < 0);
}

698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726
/*
 * check if the tree block can be shared by multiple trees
 */
int btrfs_block_can_be_shared(struct btrfs_root *root,
			      struct extent_buffer *buf)
{
	/*
	 * Tree blocks not in refernece counted trees and tree roots
	 * are never shared. If a block was allocated after the last
	 * snapshot and the block was not allocated by tree relocation,
	 * we know the block is not shared.
	 */
	if (root->ref_cows &&
	    buf != root->node && buf != root->commit_root &&
	    (btrfs_header_generation(buf) <=
	     btrfs_root_last_snapshot(&root->root_item) ||
	     btrfs_header_flag(buf, BTRFS_HEADER_FLAG_RELOC)))
		return 1;
#ifdef BTRFS_COMPAT_EXTENT_TREE_V0
	if (root->ref_cows &&
	    btrfs_header_backref_rev(buf) < BTRFS_MIXED_BACKREF_REV)
		return 1;
#endif
	return 0;
}

static noinline int update_ref_for_cow(struct btrfs_trans_handle *trans,
				       struct btrfs_root *root,
				       struct extent_buffer *buf,
727 728
				       struct extent_buffer *cow,
				       int *last_ref)
729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755
{
	u64 refs;
	u64 owner;
	u64 flags;
	u64 new_flags = 0;
	int ret;

	/*
	 * Backrefs update rules:
	 *
	 * Always use full backrefs for extent pointers in tree block
	 * allocated by tree relocation.
	 *
	 * If a shared tree block is no longer referenced by its owner
	 * tree (btrfs_header_owner(buf) == root->root_key.objectid),
	 * use full backrefs for extent pointers in tree block.
	 *
	 * If a tree block is been relocating
	 * (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID),
	 * use full backrefs for extent pointers in tree block.
	 * The reason for this is some operations (such as drop tree)
	 * are only allowed for blocks use full backrefs.
	 */

	if (btrfs_block_can_be_shared(root, buf)) {
		ret = btrfs_lookup_extent_info(trans, root, buf->start,
					       buf->len, &refs, &flags);
756 757
		if (ret)
			return ret;
758 759 760 761 762
		if (refs == 0) {
			ret = -EROFS;
			btrfs_std_error(root->fs_info, ret);
			return ret;
		}
763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779
	} else {
		refs = 1;
		if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID ||
		    btrfs_header_backref_rev(buf) < BTRFS_MIXED_BACKREF_REV)
			flags = BTRFS_BLOCK_FLAG_FULL_BACKREF;
		else
			flags = 0;
	}

	owner = btrfs_header_owner(buf);
	BUG_ON(owner == BTRFS_TREE_RELOC_OBJECTID &&
	       !(flags & BTRFS_BLOCK_FLAG_FULL_BACKREF));

	if (refs > 1) {
		if ((owner == root->root_key.objectid ||
		     root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) &&
		    !(flags & BTRFS_BLOCK_FLAG_FULL_BACKREF)) {
A
Arne Jansen 已提交
780
			ret = btrfs_inc_ref(trans, root, buf, 1, 1);
781
			BUG_ON(ret); /* -ENOMEM */
782 783 784

			if (root->root_key.objectid ==
			    BTRFS_TREE_RELOC_OBJECTID) {
A
Arne Jansen 已提交
785
				ret = btrfs_dec_ref(trans, root, buf, 0, 1);
786
				BUG_ON(ret); /* -ENOMEM */
A
Arne Jansen 已提交
787
				ret = btrfs_inc_ref(trans, root, cow, 1, 1);
788
				BUG_ON(ret); /* -ENOMEM */
789 790 791 792 793 794
			}
			new_flags |= BTRFS_BLOCK_FLAG_FULL_BACKREF;
		} else {

			if (root->root_key.objectid ==
			    BTRFS_TREE_RELOC_OBJECTID)
A
Arne Jansen 已提交
795
				ret = btrfs_inc_ref(trans, root, cow, 1, 1);
796
			else
A
Arne Jansen 已提交
797
				ret = btrfs_inc_ref(trans, root, cow, 0, 1);
798
			BUG_ON(ret); /* -ENOMEM */
799 800 801 802 803 804
		}
		if (new_flags != 0) {
			ret = btrfs_set_disk_extent_flags(trans, root,
							  buf->start,
							  buf->len,
							  new_flags, 0);
805 806
			if (ret)
				return ret;
807 808 809 810 811
		}
	} else {
		if (flags & BTRFS_BLOCK_FLAG_FULL_BACKREF) {
			if (root->root_key.objectid ==
			    BTRFS_TREE_RELOC_OBJECTID)
A
Arne Jansen 已提交
812
				ret = btrfs_inc_ref(trans, root, cow, 1, 1);
813
			else
A
Arne Jansen 已提交
814
				ret = btrfs_inc_ref(trans, root, cow, 0, 1);
815
			BUG_ON(ret); /* -ENOMEM */
A
Arne Jansen 已提交
816
			ret = btrfs_dec_ref(trans, root, buf, 1, 1);
817
			BUG_ON(ret); /* -ENOMEM */
818 819
		}
		clean_tree_block(trans, root, buf);
820
		*last_ref = 1;
821 822 823 824
	}
	return 0;
}

C
Chris Mason 已提交
825
/*
C
Chris Mason 已提交
826 827 828 829
 * does the dirty work in cow of a single block.  The parent block (if
 * supplied) is updated to point to the new cow copy.  The new buffer is marked
 * dirty and returned locked.  If you modify the block it needs to be marked
 * dirty again.
C
Chris Mason 已提交
830 831 832
 *
 * search_start -- an allocation hint for the new block
 *
C
Chris Mason 已提交
833 834 835
 * empty_size -- a hint that you plan on doing more cow.  This is the size in
 * bytes the allocator should try to find free next to the block it returns.
 * This is just a hint and may be ignored by the allocator.
C
Chris Mason 已提交
836
 */
C
Chris Mason 已提交
837
static noinline int __btrfs_cow_block(struct btrfs_trans_handle *trans,
838 839 840 841
			     struct btrfs_root *root,
			     struct extent_buffer *buf,
			     struct extent_buffer *parent, int parent_slot,
			     struct extent_buffer **cow_ret,
842
			     u64 search_start, u64 empty_size)
C
Chris Mason 已提交
843
{
844
	struct btrfs_disk_key disk_key;
845
	struct extent_buffer *cow;
846
	int level, ret;
847
	int last_ref = 0;
848
	int unlock_orig = 0;
849
	u64 parent_start;
850

851 852 853
	if (*cow_ret == buf)
		unlock_orig = 1;

854
	btrfs_assert_tree_locked(buf);
855

856 857
	WARN_ON(root->ref_cows && trans->transid !=
		root->fs_info->running_transaction->transid);
858
	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
859

860
	level = btrfs_header_level(buf);
Z
Zheng Yan 已提交
861

862 863 864 865 866 867 868 869 870 871 872 873 874 875 876
	if (level == 0)
		btrfs_item_key(buf, &disk_key, 0);
	else
		btrfs_node_key(buf, &disk_key, 0);

	if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID) {
		if (parent)
			parent_start = parent->start;
		else
			parent_start = 0;
	} else
		parent_start = 0;

	cow = btrfs_alloc_free_block(trans, root, buf->len, parent_start,
				     root->root_key.objectid, &disk_key,
877
				     level, search_start, empty_size);
878 879
	if (IS_ERR(cow))
		return PTR_ERR(cow);
880

881 882
	/* cow is set to blocking by btrfs_init_new_buffer */

883
	copy_extent_buffer(cow, buf, 0, 0, cow->len);
884
	btrfs_set_header_bytenr(cow, cow->start);
885
	btrfs_set_header_generation(cow, trans->transid);
886 887 888 889 890 891 892
	btrfs_set_header_backref_rev(cow, BTRFS_MIXED_BACKREF_REV);
	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN |
				     BTRFS_HEADER_FLAG_RELOC);
	if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID)
		btrfs_set_header_flag(cow, BTRFS_HEADER_FLAG_RELOC);
	else
		btrfs_set_header_owner(cow, root->root_key.objectid);
893

Y
Yan Zheng 已提交
894 895 896 897
	write_extent_buffer(cow, root->fs_info->fsid,
			    (unsigned long)btrfs_header_fsid(cow),
			    BTRFS_FSID_SIZE);

898
	ret = update_ref_for_cow(trans, root, buf, cow, &last_ref);
899
	if (ret) {
900
		btrfs_abort_transaction(trans, root, ret);
901 902
		return ret;
	}
Z
Zheng Yan 已提交
903

904 905 906
	if (root->ref_cows)
		btrfs_reloc_cow_block(trans, root, buf, cow);

C
Chris Mason 已提交
907
	if (buf == root->node) {
908
		WARN_ON(parent && parent != buf);
909 910 911 912 913
		if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID ||
		    btrfs_header_backref_rev(buf) < BTRFS_MIXED_BACKREF_REV)
			parent_start = buf->start;
		else
			parent_start = 0;
914

915
		extent_buffer_get(cow);
916
		rcu_assign_pointer(root->node, cow);
917

918
		btrfs_free_tree_block(trans, root, buf, parent_start,
919
				      last_ref);
920
		free_extent_buffer(buf);
921
		add_root_to_dirty_list(root);
C
Chris Mason 已提交
922
	} else {
923 924 925 926 927 928
		if (root->root_key.objectid == BTRFS_TREE_RELOC_OBJECTID)
			parent_start = parent->start;
		else
			parent_start = 0;

		WARN_ON(trans->transid != btrfs_header_generation(parent));
929
		btrfs_set_node_blockptr(parent, parent_slot,
930
					cow->start);
931 932
		btrfs_set_node_ptr_generation(parent, parent_slot,
					      trans->transid);
C
Chris Mason 已提交
933
		btrfs_mark_buffer_dirty(parent);
934
		btrfs_free_tree_block(trans, root, buf, parent_start,
935
				      last_ref);
C
Chris Mason 已提交
936
	}
937 938
	if (unlock_orig)
		btrfs_tree_unlock(buf);
939
	free_extent_buffer_stale(buf);
C
Chris Mason 已提交
940
	btrfs_mark_buffer_dirty(cow);
C
Chris Mason 已提交
941
	*cow_ret = cow;
C
Chris Mason 已提交
942 943 944
	return 0;
}

945 946 947 948
static inline int should_cow_block(struct btrfs_trans_handle *trans,
				   struct btrfs_root *root,
				   struct extent_buffer *buf)
{
949 950 951 952 953 954 955 956 957 958 959 960 961 962
	/* ensure we can see the force_cow */
	smp_rmb();

	/*
	 * We do not need to cow a block if
	 * 1) this block is not created or changed in this transaction;
	 * 2) this block does not belong to TREE_RELOC tree;
	 * 3) the root is not forced COW.
	 *
	 * What is forced COW:
	 *    when we create snapshot during commiting the transaction,
	 *    after we've finished coping src root, we must COW the shared
	 *    block to ensure the metadata consistency.
	 */
963 964 965
	if (btrfs_header_generation(buf) == trans->transid &&
	    !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN) &&
	    !(root->root_key.objectid != BTRFS_TREE_RELOC_OBJECTID &&
966 967
	      btrfs_header_flag(buf, BTRFS_HEADER_FLAG_RELOC)) &&
	    !root->force_cow)
968 969 970 971
		return 0;
	return 1;
}

C
Chris Mason 已提交
972 973 974 975 976
/*
 * cows a single block, see __btrfs_cow_block for the real work.
 * This version of it has extra checks so that a block isn't cow'd more than
 * once per transaction, as long as it hasn't been written yet
 */
C
Chris Mason 已提交
977
noinline int btrfs_cow_block(struct btrfs_trans_handle *trans,
978 979
		    struct btrfs_root *root, struct extent_buffer *buf,
		    struct extent_buffer *parent, int parent_slot,
980
		    struct extent_buffer **cow_ret)
981 982
{
	u64 search_start;
983
	int ret;
C
Chris Mason 已提交
984

985
	if (trans->transaction != root->fs_info->running_transaction) {
C
Chris Mason 已提交
986 987 988
		printk(KERN_CRIT "trans %llu running %llu\n",
		       (unsigned long long)trans->transid,
		       (unsigned long long)
989 990 991 992
		       root->fs_info->running_transaction->transid);
		WARN_ON(1);
	}
	if (trans->transid != root->fs_info->generation) {
C
Chris Mason 已提交
993 994 995
		printk(KERN_CRIT "trans %llu running %llu\n",
		       (unsigned long long)trans->transid,
		       (unsigned long long)root->fs_info->generation);
996 997
		WARN_ON(1);
	}
C
Chris Mason 已提交
998

999
	if (!should_cow_block(trans, root, buf)) {
1000 1001 1002
		*cow_ret = buf;
		return 0;
	}
1003

1004
	search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1);
1005 1006 1007 1008 1009

	if (parent)
		btrfs_set_lock_blocking(parent);
	btrfs_set_lock_blocking(buf);

1010
	ret = __btrfs_cow_block(trans, root, buf, parent,
1011
				 parent_slot, cow_ret, search_start, 0);
1012 1013 1014

	trace_btrfs_cow_block(root, buf, *cow_ret);

1015
	return ret;
1016 1017
}

C
Chris Mason 已提交
1018 1019 1020 1021
/*
 * helper function for defrag to decide if two blocks pointed to by a
 * node are actually close by
 */
1022
static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
1023
{
1024
	if (blocknr < other && other - (blocknr + blocksize) < 32768)
1025
		return 1;
1026
	if (blocknr > other && blocknr - (other + blocksize) < 32768)
1027 1028 1029 1030
		return 1;
	return 0;
}

1031 1032 1033 1034 1035 1036 1037 1038 1039
/*
 * compare two keys in a memcmp fashion
 */
static int comp_keys(struct btrfs_disk_key *disk, struct btrfs_key *k2)
{
	struct btrfs_key k1;

	btrfs_disk_key_to_cpu(&k1, disk);

1040
	return btrfs_comp_cpu_keys(&k1, k2);
1041 1042
}

1043 1044 1045
/*
 * same as comp_keys only with two btrfs_key's
 */
1046
int btrfs_comp_cpu_keys(struct btrfs_key *k1, struct btrfs_key *k2)
1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061
{
	if (k1->objectid > k2->objectid)
		return 1;
	if (k1->objectid < k2->objectid)
		return -1;
	if (k1->type > k2->type)
		return 1;
	if (k1->type < k2->type)
		return -1;
	if (k1->offset > k2->offset)
		return 1;
	if (k1->offset < k2->offset)
		return -1;
	return 0;
}
1062

C
Chris Mason 已提交
1063 1064 1065 1066 1067
/*
 * this is used by the defrag code to go through all the
 * leaves pointed to by a node and reallocate them so that
 * disk order is close to key order
 */
1068
int btrfs_realloc_node(struct btrfs_trans_handle *trans,
1069
		       struct btrfs_root *root, struct extent_buffer *parent,
1070 1071
		       int start_slot, int cache_only, u64 *last_ret,
		       struct btrfs_key *progress)
1072
{
1073
	struct extent_buffer *cur;
1074
	u64 blocknr;
1075
	u64 gen;
1076 1077
	u64 search_start = *last_ret;
	u64 last_block = 0;
1078 1079 1080 1081 1082
	u64 other;
	u32 parent_nritems;
	int end_slot;
	int i;
	int err = 0;
1083
	int parent_level;
1084 1085
	int uptodate;
	u32 blocksize;
1086 1087
	int progress_passed = 0;
	struct btrfs_disk_key disk_key;
1088

1089 1090 1091 1092
	parent_level = btrfs_header_level(parent);
	if (cache_only && parent_level != 1)
		return 0;

C
Chris Mason 已提交
1093
	if (trans->transaction != root->fs_info->running_transaction)
1094
		WARN_ON(1);
C
Chris Mason 已提交
1095
	if (trans->transid != root->fs_info->generation)
1096
		WARN_ON(1);
1097

1098 1099
	parent_nritems = btrfs_header_nritems(parent);
	blocksize = btrfs_level_size(root, parent_level - 1);
1100 1101 1102 1103 1104
	end_slot = parent_nritems;

	if (parent_nritems == 1)
		return 0;

1105 1106
	btrfs_set_lock_blocking(parent);

1107 1108
	for (i = start_slot; i < end_slot; i++) {
		int close = 1;
1109

1110 1111 1112 1113 1114
		btrfs_node_key(parent, &disk_key, i);
		if (!progress_passed && comp_keys(&disk_key, progress) < 0)
			continue;

		progress_passed = 1;
1115
		blocknr = btrfs_node_blockptr(parent, i);
1116
		gen = btrfs_node_ptr_generation(parent, i);
1117 1118
		if (last_block == 0)
			last_block = blocknr;
1119

1120
		if (i > 0) {
1121 1122
			other = btrfs_node_blockptr(parent, i - 1);
			close = close_blocks(blocknr, other, blocksize);
1123
		}
C
Chris Mason 已提交
1124
		if (!close && i < end_slot - 2) {
1125 1126
			other = btrfs_node_blockptr(parent, i + 1);
			close = close_blocks(blocknr, other, blocksize);
1127
		}
1128 1129
		if (close) {
			last_block = blocknr;
1130
			continue;
1131
		}
1132

1133 1134
		cur = btrfs_find_tree_block(root, blocknr, blocksize);
		if (cur)
1135
			uptodate = btrfs_buffer_uptodate(cur, gen, 0);
1136 1137
		else
			uptodate = 0;
1138
		if (!cur || !uptodate) {
1139
			if (cache_only) {
1140
				free_extent_buffer(cur);
1141 1142
				continue;
			}
1143 1144
			if (!cur) {
				cur = read_tree_block(root, blocknr,
1145
							 blocksize, gen);
1146 1147
				if (!cur)
					return -EIO;
1148
			} else if (!uptodate) {
1149
				btrfs_read_buffer(cur, gen);
1150
			}
1151
		}
1152
		if (search_start == 0)
1153
			search_start = last_block;
1154

1155
		btrfs_tree_lock(cur);
1156
		btrfs_set_lock_blocking(cur);
1157
		err = __btrfs_cow_block(trans, root, cur, parent, i,
1158
					&cur, search_start,
1159
					min(16 * blocksize,
1160
					    (end_slot - i) * blocksize));
Y
Yan 已提交
1161
		if (err) {
1162
			btrfs_tree_unlock(cur);
1163
			free_extent_buffer(cur);
1164
			break;
Y
Yan 已提交
1165
		}
1166 1167
		search_start = cur->start;
		last_block = cur->start;
1168
		*last_ret = search_start;
1169 1170
		btrfs_tree_unlock(cur);
		free_extent_buffer(cur);
1171 1172 1173 1174
	}
	return err;
}

C
Chris Mason 已提交
1175 1176 1177 1178 1179
/*
 * 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 已提交
1180
static inline unsigned int leaf_data_end(struct btrfs_root *root,
1181
					 struct extent_buffer *leaf)
1182
{
1183
	u32 nr = btrfs_header_nritems(leaf);
1184
	if (nr == 0)
C
Chris Mason 已提交
1185
		return BTRFS_LEAF_DATA_SIZE(root);
1186
	return btrfs_item_offset_nr(leaf, nr - 1);
1187 1188
}

C
Chris Mason 已提交
1189

C
Chris Mason 已提交
1190
/*
1191 1192 1193
 * 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 已提交
1194 1195 1196 1197 1198 1199
 * 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
 */
1200 1201 1202 1203
static noinline int generic_bin_search(struct extent_buffer *eb,
				       unsigned long p,
				       int item_size, struct btrfs_key *key,
				       int max, int *slot)
1204 1205 1206 1207 1208
{
	int low = 0;
	int high = max;
	int mid;
	int ret;
1209
	struct btrfs_disk_key *tmp = NULL;
1210 1211 1212 1213 1214
	struct btrfs_disk_key unaligned;
	unsigned long offset;
	char *kaddr = NULL;
	unsigned long map_start = 0;
	unsigned long map_len = 0;
1215
	int err;
1216

C
Chris Mason 已提交
1217
	while (low < high) {
1218
		mid = (low + high) / 2;
1219 1220
		offset = p + mid * item_size;

1221
		if (!kaddr || offset < map_start ||
1222 1223
		    (offset + sizeof(struct btrfs_disk_key)) >
		    map_start + map_len) {
1224 1225

			err = map_private_extent_buffer(eb, offset,
1226
						sizeof(struct btrfs_disk_key),
1227
						&kaddr, &map_start, &map_len);
1228 1229 1230 1231 1232 1233 1234 1235 1236

			if (!err) {
				tmp = (struct btrfs_disk_key *)(kaddr + offset -
							map_start);
			} else {
				read_extent_buffer(eb, &unaligned,
						   offset, sizeof(unaligned));
				tmp = &unaligned;
			}
1237 1238 1239 1240 1241

		} else {
			tmp = (struct btrfs_disk_key *)(kaddr + offset -
							map_start);
		}
1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256
		ret = comp_keys(tmp, key);

		if (ret < 0)
			low = mid + 1;
		else if (ret > 0)
			high = mid;
		else {
			*slot = mid;
			return 0;
		}
	}
	*slot = low;
	return 1;
}

C
Chris Mason 已提交
1257 1258 1259 1260
/*
 * simple bin_search frontend that does the right thing for
 * leaves vs nodes
 */
1261 1262
static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
		      int level, int *slot)
1263
{
1264 1265 1266
	if (level == 0) {
		return generic_bin_search(eb,
					  offsetof(struct btrfs_leaf, items),
C
Chris Mason 已提交
1267
					  sizeof(struct btrfs_item),
1268
					  key, btrfs_header_nritems(eb),
1269
					  slot);
1270
	} else {
1271 1272
		return generic_bin_search(eb,
					  offsetof(struct btrfs_node, ptrs),
C
Chris Mason 已提交
1273
					  sizeof(struct btrfs_key_ptr),
1274
					  key, btrfs_header_nritems(eb),
1275
					  slot);
1276 1277 1278 1279
	}
	return -1;
}

1280 1281 1282 1283 1284 1285
int btrfs_bin_search(struct extent_buffer *eb, struct btrfs_key *key,
		     int level, int *slot)
{
	return bin_search(eb, key, level, slot);
}

1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301
static void root_add_used(struct btrfs_root *root, u32 size)
{
	spin_lock(&root->accounting_lock);
	btrfs_set_root_used(&root->root_item,
			    btrfs_root_used(&root->root_item) + size);
	spin_unlock(&root->accounting_lock);
}

static void root_sub_used(struct btrfs_root *root, u32 size)
{
	spin_lock(&root->accounting_lock);
	btrfs_set_root_used(&root->root_item,
			    btrfs_root_used(&root->root_item) - size);
	spin_unlock(&root->accounting_lock);
}

C
Chris Mason 已提交
1302 1303 1304 1305
/* given a node and slot number, this reads the blocks it points to.  The
 * extent buffer is returned with a reference taken (but unlocked).
 * NULL is returned on error.
 */
1306
static noinline struct extent_buffer *read_node_slot(struct btrfs_root *root,
1307
				   struct extent_buffer *parent, int slot)
1308
{
1309
	int level = btrfs_header_level(parent);
1310 1311
	if (slot < 0)
		return NULL;
1312
	if (slot >= btrfs_header_nritems(parent))
1313
		return NULL;
1314 1315 1316

	BUG_ON(level == 0);

1317
	return read_tree_block(root, btrfs_node_blockptr(parent, slot),
1318 1319
		       btrfs_level_size(root, level - 1),
		       btrfs_node_ptr_generation(parent, slot));
1320 1321
}

C
Chris Mason 已提交
1322 1323 1324 1325 1326
/*
 * node level balancing, used to make sure nodes are in proper order for
 * item deletion.  We balance from the top down, so we have to make sure
 * that a deletion won't leave an node completely empty later on.
 */
1327
static noinline int balance_level(struct btrfs_trans_handle *trans,
1328 1329
			 struct btrfs_root *root,
			 struct btrfs_path *path, int level)
1330
{
1331 1332 1333 1334
	struct extent_buffer *right = NULL;
	struct extent_buffer *mid;
	struct extent_buffer *left = NULL;
	struct extent_buffer *parent = NULL;
1335 1336 1337 1338
	int ret = 0;
	int wret;
	int pslot;
	int orig_slot = path->slots[level];
1339
	u64 orig_ptr;
1340 1341 1342 1343

	if (level == 0)
		return 0;

1344
	mid = path->nodes[level];
1345

1346 1347
	WARN_ON(path->locks[level] != BTRFS_WRITE_LOCK &&
		path->locks[level] != BTRFS_WRITE_LOCK_BLOCKING);
1348 1349
	WARN_ON(btrfs_header_generation(mid) != trans->transid);

1350
	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
1351

L
Li Zefan 已提交
1352
	if (level < BTRFS_MAX_LEVEL - 1) {
1353
		parent = path->nodes[level + 1];
L
Li Zefan 已提交
1354 1355
		pslot = path->slots[level + 1];
	}
1356

C
Chris Mason 已提交
1357 1358 1359 1360
	/*
	 * deal with the case where there is only one pointer in the root
	 * by promoting the node below to a root
	 */
1361 1362
	if (!parent) {
		struct extent_buffer *child;
1363

1364
		if (btrfs_header_nritems(mid) != 1)
1365 1366 1367
			return 0;

		/* promote the child to a root */
1368
		child = read_node_slot(root, mid, 0);
1369 1370 1371 1372 1373 1374
		if (!child) {
			ret = -EROFS;
			btrfs_std_error(root->fs_info, ret);
			goto enospc;
		}

1375
		btrfs_tree_lock(child);
1376
		btrfs_set_lock_blocking(child);
1377
		ret = btrfs_cow_block(trans, root, child, mid, 0, &child);
1378 1379 1380 1381 1382
		if (ret) {
			btrfs_tree_unlock(child);
			free_extent_buffer(child);
			goto enospc;
		}
1383

1384
		rcu_assign_pointer(root->node, child);
1385

1386
		add_root_to_dirty_list(root);
1387
		btrfs_tree_unlock(child);
1388

1389
		path->locks[level] = 0;
1390
		path->nodes[level] = NULL;
1391
		clean_tree_block(trans, root, mid);
1392
		btrfs_tree_unlock(mid);
1393
		/* once for the path */
1394
		free_extent_buffer(mid);
1395 1396

		root_sub_used(root, mid->len);
1397
		btrfs_free_tree_block(trans, root, mid, 0, 1);
1398
		/* once for the root ptr */
1399
		free_extent_buffer_stale(mid);
1400
		return 0;
1401
	}
1402
	if (btrfs_header_nritems(mid) >
C
Chris Mason 已提交
1403
	    BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
1404 1405
		return 0;

1406
	btrfs_header_nritems(mid);
1407

1408 1409
	left = read_node_slot(root, parent, pslot - 1);
	if (left) {
1410
		btrfs_tree_lock(left);
1411
		btrfs_set_lock_blocking(left);
1412
		wret = btrfs_cow_block(trans, root, left,
1413
				       parent, pslot - 1, &left);
1414 1415 1416 1417
		if (wret) {
			ret = wret;
			goto enospc;
		}
1418
	}
1419 1420
	right = read_node_slot(root, parent, pslot + 1);
	if (right) {
1421
		btrfs_tree_lock(right);
1422
		btrfs_set_lock_blocking(right);
1423
		wret = btrfs_cow_block(trans, root, right,
1424
				       parent, pslot + 1, &right);
1425 1426 1427 1428 1429 1430 1431
		if (wret) {
			ret = wret;
			goto enospc;
		}
	}

	/* first, try to make some room in the middle buffer */
1432 1433
	if (left) {
		orig_slot += btrfs_header_nritems(left);
1434
		wret = push_node_left(trans, root, left, mid, 1);
1435 1436
		if (wret < 0)
			ret = wret;
1437
		btrfs_header_nritems(mid);
1438
	}
1439 1440 1441 1442

	/*
	 * then try to empty the right most buffer into the middle
	 */
1443
	if (right) {
1444
		wret = push_node_left(trans, root, mid, right, 1);
1445
		if (wret < 0 && wret != -ENOSPC)
1446
			ret = wret;
1447 1448
		if (btrfs_header_nritems(right) == 0) {
			clean_tree_block(trans, root, right);
1449
			btrfs_tree_unlock(right);
1450
			del_ptr(trans, root, path, level + 1, pslot + 1);
1451
			root_sub_used(root, right->len);
1452
			btrfs_free_tree_block(trans, root, right, 0, 1);
1453
			free_extent_buffer_stale(right);
1454
			right = NULL;
1455
		} else {
1456 1457 1458 1459
			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);
1460 1461
		}
	}
1462
	if (btrfs_header_nritems(mid) == 1) {
1463 1464 1465 1466 1467 1468 1469 1470 1471
		/*
		 * 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
		 */
1472 1473 1474 1475 1476
		if (!left) {
			ret = -EROFS;
			btrfs_std_error(root->fs_info, ret);
			goto enospc;
		}
1477
		wret = balance_node_right(trans, root, mid, left);
1478
		if (wret < 0) {
1479
			ret = wret;
1480 1481
			goto enospc;
		}
1482 1483 1484 1485 1486
		if (wret == 1) {
			wret = push_node_left(trans, root, left, mid, 1);
			if (wret < 0)
				ret = wret;
		}
1487 1488
		BUG_ON(wret == 1);
	}
1489 1490
	if (btrfs_header_nritems(mid) == 0) {
		clean_tree_block(trans, root, mid);
1491
		btrfs_tree_unlock(mid);
1492
		del_ptr(trans, root, path, level + 1, pslot);
1493
		root_sub_used(root, mid->len);
1494
		btrfs_free_tree_block(trans, root, mid, 0, 1);
1495
		free_extent_buffer_stale(mid);
1496
		mid = NULL;
1497 1498
	} else {
		/* update the parent key to reflect our changes */
1499 1500 1501 1502
		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);
1503
	}
1504

1505
	/* update the path */
1506 1507 1508
	if (left) {
		if (btrfs_header_nritems(left) > orig_slot) {
			extent_buffer_get(left);
1509
			/* left was locked after cow */
1510
			path->nodes[level] = left;
1511 1512
			path->slots[level + 1] -= 1;
			path->slots[level] = orig_slot;
1513 1514
			if (mid) {
				btrfs_tree_unlock(mid);
1515
				free_extent_buffer(mid);
1516
			}
1517
		} else {
1518
			orig_slot -= btrfs_header_nritems(left);
1519 1520 1521
			path->slots[level] = orig_slot;
		}
	}
1522
	/* double check we haven't messed things up */
C
Chris Mason 已提交
1523
	if (orig_ptr !=
1524
	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
1525
		BUG();
1526
enospc:
1527 1528
	if (right) {
		btrfs_tree_unlock(right);
1529
		free_extent_buffer(right);
1530 1531 1532 1533
	}
	if (left) {
		if (path->nodes[level] != left)
			btrfs_tree_unlock(left);
1534
		free_extent_buffer(left);
1535
	}
1536 1537 1538
	return ret;
}

C
Chris Mason 已提交
1539 1540 1541 1542
/* Node balancing for insertion.  Here we only split or push nodes around
 * when they are completely full.  This is also done top down, so we
 * have to be pessimistic.
 */
C
Chris Mason 已提交
1543
static noinline int push_nodes_for_insert(struct btrfs_trans_handle *trans,
1544 1545
					  struct btrfs_root *root,
					  struct btrfs_path *path, int level)
1546
{
1547 1548 1549 1550
	struct extent_buffer *right = NULL;
	struct extent_buffer *mid;
	struct extent_buffer *left = NULL;
	struct extent_buffer *parent = NULL;
1551 1552 1553 1554 1555 1556 1557 1558
	int ret = 0;
	int wret;
	int pslot;
	int orig_slot = path->slots[level];

	if (level == 0)
		return 1;

1559
	mid = path->nodes[level];
1560
	WARN_ON(btrfs_header_generation(mid) != trans->transid);
1561

L
Li Zefan 已提交
1562
	if (level < BTRFS_MAX_LEVEL - 1) {
1563
		parent = path->nodes[level + 1];
L
Li Zefan 已提交
1564 1565
		pslot = path->slots[level + 1];
	}
1566

1567
	if (!parent)
1568 1569
		return 1;

1570
	left = read_node_slot(root, parent, pslot - 1);
1571 1572

	/* first, try to make some room in the middle buffer */
1573
	if (left) {
1574
		u32 left_nr;
1575 1576

		btrfs_tree_lock(left);
1577 1578
		btrfs_set_lock_blocking(left);

1579
		left_nr = btrfs_header_nritems(left);
C
Chris Mason 已提交
1580 1581 1582
		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
			wret = 1;
		} else {
1583
			ret = btrfs_cow_block(trans, root, left, parent,
1584
					      pslot - 1, &left);
1585 1586 1587 1588
			if (ret)
				wret = 1;
			else {
				wret = push_node_left(trans, root,
1589
						      left, mid, 0);
1590
			}
C
Chris Mason 已提交
1591
		}
1592 1593 1594
		if (wret < 0)
			ret = wret;
		if (wret == 0) {
1595
			struct btrfs_disk_key disk_key;
1596
			orig_slot += left_nr;
1597 1598 1599 1600 1601
			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;
1602 1603
				path->slots[level + 1] -= 1;
				path->slots[level] = orig_slot;
1604
				btrfs_tree_unlock(mid);
1605
				free_extent_buffer(mid);
1606 1607
			} else {
				orig_slot -=
1608
					btrfs_header_nritems(left);
1609
				path->slots[level] = orig_slot;
1610
				btrfs_tree_unlock(left);
1611
				free_extent_buffer(left);
1612 1613 1614
			}
			return 0;
		}
1615
		btrfs_tree_unlock(left);
1616
		free_extent_buffer(left);
1617
	}
1618
	right = read_node_slot(root, parent, pslot + 1);
1619 1620 1621 1622

	/*
	 * then try to empty the right most buffer into the middle
	 */
1623
	if (right) {
C
Chris Mason 已提交
1624
		u32 right_nr;
1625

1626
		btrfs_tree_lock(right);
1627 1628
		btrfs_set_lock_blocking(right);

1629
		right_nr = btrfs_header_nritems(right);
C
Chris Mason 已提交
1630 1631 1632
		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
			wret = 1;
		} else {
1633 1634
			ret = btrfs_cow_block(trans, root, right,
					      parent, pslot + 1,
1635
					      &right);
1636 1637 1638 1639
			if (ret)
				wret = 1;
			else {
				wret = balance_node_right(trans, root,
1640
							  right, mid);
1641
			}
C
Chris Mason 已提交
1642
		}
1643 1644 1645
		if (wret < 0)
			ret = wret;
		if (wret == 0) {
1646 1647 1648 1649 1650 1651 1652 1653
			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;
1654 1655
				path->slots[level + 1] += 1;
				path->slots[level] = orig_slot -
1656
					btrfs_header_nritems(mid);
1657
				btrfs_tree_unlock(mid);
1658
				free_extent_buffer(mid);
1659
			} else {
1660
				btrfs_tree_unlock(right);
1661
				free_extent_buffer(right);
1662 1663 1664
			}
			return 0;
		}
1665
		btrfs_tree_unlock(right);
1666
		free_extent_buffer(right);
1667 1668 1669 1670
	}
	return 1;
}

1671
/*
C
Chris Mason 已提交
1672 1673
 * readahead one full node of leaves, finding things that are close
 * to the block in 'slot', and triggering ra on them.
1674
 */
1675 1676 1677
static void reada_for_search(struct btrfs_root *root,
			     struct btrfs_path *path,
			     int level, int slot, u64 objectid)
1678
{
1679
	struct extent_buffer *node;
1680
	struct btrfs_disk_key disk_key;
1681 1682
	u32 nritems;
	u64 search;
1683
	u64 target;
1684
	u64 nread = 0;
1685
	u64 gen;
1686
	int direction = path->reada;
1687
	struct extent_buffer *eb;
1688 1689 1690
	u32 nr;
	u32 blocksize;
	u32 nscan = 0;
1691

1692
	if (level != 1)
1693 1694 1695
		return;

	if (!path->nodes[level])
1696 1697
		return;

1698
	node = path->nodes[level];
1699

1700
	search = btrfs_node_blockptr(node, slot);
1701 1702
	blocksize = btrfs_level_size(root, level - 1);
	eb = btrfs_find_tree_block(root, search, blocksize);
1703 1704
	if (eb) {
		free_extent_buffer(eb);
1705 1706 1707
		return;
	}

1708
	target = search;
1709

1710
	nritems = btrfs_header_nritems(node);
1711
	nr = slot;
1712

C
Chris Mason 已提交
1713
	while (1) {
1714 1715 1716 1717 1718 1719 1720 1721
		if (direction < 0) {
			if (nr == 0)
				break;
			nr--;
		} else if (direction > 0) {
			nr++;
			if (nr >= nritems)
				break;
1722
		}
1723 1724 1725 1726 1727
		if (path->reada < 0 && objectid) {
			btrfs_node_key(node, &disk_key, nr);
			if (btrfs_disk_key_objectid(&disk_key) != objectid)
				break;
		}
1728
		search = btrfs_node_blockptr(node, nr);
1729 1730
		if ((search <= target && target - search <= 65536) ||
		    (search > target && search - target <= 65536)) {
1731 1732
			gen = btrfs_node_ptr_generation(node, nr);
			readahead_tree_block(root, search, blocksize, gen);
1733 1734 1735
			nread += blocksize;
		}
		nscan++;
1736
		if ((nread > 65536 || nscan > 32))
1737
			break;
1738 1739
	}
}
1740

1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757
/*
 * returns -EAGAIN if it had to drop the path, or zero if everything was in
 * cache
 */
static noinline int reada_for_balance(struct btrfs_root *root,
				      struct btrfs_path *path, int level)
{
	int slot;
	int nritems;
	struct extent_buffer *parent;
	struct extent_buffer *eb;
	u64 gen;
	u64 block1 = 0;
	u64 block2 = 0;
	int ret = 0;
	int blocksize;

1758
	parent = path->nodes[level + 1];
1759 1760 1761 1762
	if (!parent)
		return 0;

	nritems = btrfs_header_nritems(parent);
1763
	slot = path->slots[level + 1];
1764 1765 1766 1767 1768 1769
	blocksize = btrfs_level_size(root, level);

	if (slot > 0) {
		block1 = btrfs_node_blockptr(parent, slot - 1);
		gen = btrfs_node_ptr_generation(parent, slot - 1);
		eb = btrfs_find_tree_block(root, block1, blocksize);
1770 1771 1772 1773 1774 1775
		/*
		 * if we get -eagain from btrfs_buffer_uptodate, we
		 * don't want to return eagain here.  That will loop
		 * forever
		 */
		if (eb && btrfs_buffer_uptodate(eb, gen, 1) != 0)
1776 1777 1778
			block1 = 0;
		free_extent_buffer(eb);
	}
1779
	if (slot + 1 < nritems) {
1780 1781 1782
		block2 = btrfs_node_blockptr(parent, slot + 1);
		gen = btrfs_node_ptr_generation(parent, slot + 1);
		eb = btrfs_find_tree_block(root, block2, blocksize);
1783
		if (eb && btrfs_buffer_uptodate(eb, gen, 1) != 0)
1784 1785 1786 1787 1788
			block2 = 0;
		free_extent_buffer(eb);
	}
	if (block1 || block2) {
		ret = -EAGAIN;
1789 1790

		/* release the whole path */
1791
		btrfs_release_path(path);
1792 1793

		/* read the blocks */
1794 1795 1796 1797 1798 1799 1800 1801 1802
		if (block1)
			readahead_tree_block(root, block1, blocksize, 0);
		if (block2)
			readahead_tree_block(root, block2, blocksize, 0);

		if (block1) {
			eb = read_tree_block(root, block1, blocksize, 0);
			free_extent_buffer(eb);
		}
1803
		if (block2) {
1804 1805 1806 1807 1808 1809 1810 1811
			eb = read_tree_block(root, block2, blocksize, 0);
			free_extent_buffer(eb);
		}
	}
	return ret;
}


C
Chris Mason 已提交
1812
/*
C
Chris Mason 已提交
1813 1814 1815 1816
 * when we walk down the tree, it is usually safe to unlock the higher layers
 * in the tree.  The exceptions are when our path goes through slot 0, because
 * operations on the tree might require changing key pointers higher up in the
 * tree.
C
Chris Mason 已提交
1817
 *
C
Chris Mason 已提交
1818 1819 1820
 * callers might also have set path->keep_locks, which tells this code to keep
 * the lock if the path points to the last slot in the block.  This is part of
 * walking through the tree, and selecting the next slot in the higher block.
C
Chris Mason 已提交
1821
 *
C
Chris Mason 已提交
1822 1823
 * lowest_unlock sets the lowest level in the tree we're allowed to unlock.  so
 * if lowest_unlock is 1, level 0 won't be unlocked
C
Chris Mason 已提交
1824
 */
1825
static noinline void unlock_up(struct btrfs_path *path, int level,
1826 1827
			       int lowest_unlock, int min_write_lock_level,
			       int *write_lock_level)
1828 1829 1830
{
	int i;
	int skip_level = level;
1831
	int no_skips = 0;
1832 1833 1834 1835 1836 1837 1838
	struct extent_buffer *t;

	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
		if (!path->nodes[i])
			break;
		if (!path->locks[i])
			break;
1839
		if (!no_skips && path->slots[i] == 0) {
1840 1841 1842
			skip_level = i + 1;
			continue;
		}
1843
		if (!no_skips && path->keep_locks) {
1844 1845 1846
			u32 nritems;
			t = path->nodes[i];
			nritems = btrfs_header_nritems(t);
1847
			if (nritems < 1 || path->slots[i] >= nritems - 1) {
1848 1849 1850 1851
				skip_level = i + 1;
				continue;
			}
		}
1852 1853 1854
		if (skip_level < i && i >= lowest_unlock)
			no_skips = 1;

1855 1856
		t = path->nodes[i];
		if (i >= lowest_unlock && i > skip_level && path->locks[i]) {
1857
			btrfs_tree_unlock_rw(t, path->locks[i]);
1858
			path->locks[i] = 0;
1859 1860 1861 1862 1863
			if (write_lock_level &&
			    i > min_write_lock_level &&
			    i <= *write_lock_level) {
				*write_lock_level = i - 1;
			}
1864 1865 1866 1867
		}
	}
}

1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880
/*
 * This releases any locks held in the path starting at level and
 * going all the way up to the root.
 *
 * btrfs_search_slot will keep the lock held on higher nodes in a few
 * corner cases, such as COW of the block at slot zero in the node.  This
 * ignores those rules, and it should only be called when there are no
 * more updates to be done higher up in the tree.
 */
noinline void btrfs_unlock_up_safe(struct btrfs_path *path, int level)
{
	int i;

1881
	if (path->keep_locks)
1882 1883 1884 1885
		return;

	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
		if (!path->nodes[i])
1886
			continue;
1887
		if (!path->locks[i])
1888
			continue;
1889
		btrfs_tree_unlock_rw(path->nodes[i], path->locks[i]);
1890 1891 1892 1893
		path->locks[i] = 0;
	}
}

1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912
/*
 * helper function for btrfs_search_slot.  The goal is to find a block
 * in cache without setting the path to blocking.  If we find the block
 * we return zero and the path is unchanged.
 *
 * If we can't find the block, we set the path blocking and do some
 * reada.  -EAGAIN is returned and the search must be repeated.
 */
static int
read_block_for_search(struct btrfs_trans_handle *trans,
		       struct btrfs_root *root, struct btrfs_path *p,
		       struct extent_buffer **eb_ret, int level, int slot,
		       struct btrfs_key *key)
{
	u64 blocknr;
	u64 gen;
	u32 blocksize;
	struct extent_buffer *b = *eb_ret;
	struct extent_buffer *tmp;
1913
	int ret;
1914 1915 1916 1917 1918 1919

	blocknr = btrfs_node_blockptr(b, slot);
	gen = btrfs_node_ptr_generation(b, slot);
	blocksize = btrfs_level_size(root, level - 1);

	tmp = btrfs_find_tree_block(root, blocknr, blocksize);
1920
	if (tmp) {
1921 1922 1923
		/* first we do an atomic uptodate check */
		if (btrfs_buffer_uptodate(tmp, 0, 1) > 0) {
			if (btrfs_buffer_uptodate(tmp, gen, 1) > 0) {
1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938
				/*
				 * we found an up to date block without
				 * sleeping, return
				 * right away
				 */
				*eb_ret = tmp;
				return 0;
			}
			/* the pages were up to date, but we failed
			 * the generation number check.  Do a full
			 * read for the generation number that is correct.
			 * We must do this without dropping locks so
			 * we can trust our generation number
			 */
			free_extent_buffer(tmp);
1939 1940
			btrfs_set_path_blocking(p);

1941
			/* now we're allowed to do a blocking uptodate check */
1942
			tmp = read_tree_block(root, blocknr, blocksize, gen);
1943
			if (tmp && btrfs_buffer_uptodate(tmp, gen, 0) > 0) {
1944 1945 1946 1947
				*eb_ret = tmp;
				return 0;
			}
			free_extent_buffer(tmp);
1948
			btrfs_release_path(p);
1949 1950
			return -EIO;
		}
1951 1952 1953 1954 1955
	}

	/*
	 * reduce lock contention at high levels
	 * of the btree by dropping locks before
1956 1957 1958
	 * we read.  Don't release the lock on the current
	 * level because we need to walk this node to figure
	 * out which blocks to read.
1959
	 */
1960 1961 1962
	btrfs_unlock_up_safe(p, level + 1);
	btrfs_set_path_blocking(p);

1963
	free_extent_buffer(tmp);
1964 1965 1966
	if (p->reada)
		reada_for_search(root, p, level, slot, key->objectid);

1967
	btrfs_release_path(p);
1968 1969

	ret = -EAGAIN;
1970
	tmp = read_tree_block(root, blocknr, blocksize, 0);
1971 1972 1973 1974 1975 1976 1977
	if (tmp) {
		/*
		 * If the read above didn't mark this buffer up to date,
		 * it will never end up being up to date.  Set ret to EIO now
		 * and give up so that our caller doesn't loop forever
		 * on our EAGAINs.
		 */
1978
		if (!btrfs_buffer_uptodate(tmp, 0, 0))
1979
			ret = -EIO;
1980
		free_extent_buffer(tmp);
1981 1982
	}
	return ret;
1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996
}

/*
 * helper function for btrfs_search_slot.  This does all of the checks
 * for node-level blocks and does any balancing required based on
 * the ins_len.
 *
 * If no extra work was required, zero is returned.  If we had to
 * drop the path, -EAGAIN is returned and btrfs_search_slot must
 * start over
 */
static int
setup_nodes_for_search(struct btrfs_trans_handle *trans,
		       struct btrfs_root *root, struct btrfs_path *p,
1997 1998
		       struct extent_buffer *b, int level, int ins_len,
		       int *write_lock_level)
1999 2000 2001 2002 2003 2004
{
	int ret;
	if ((p->search_for_split || ins_len > 0) && btrfs_header_nritems(b) >=
	    BTRFS_NODEPTRS_PER_BLOCK(root) - 3) {
		int sret;

2005 2006 2007 2008 2009 2010
		if (*write_lock_level < level + 1) {
			*write_lock_level = level + 1;
			btrfs_release_path(p);
			goto again;
		}

2011 2012 2013 2014 2015 2016
		sret = reada_for_balance(root, p, level);
		if (sret)
			goto again;

		btrfs_set_path_blocking(p);
		sret = split_node(trans, root, p, level);
2017
		btrfs_clear_path_blocking(p, NULL, 0);
2018 2019 2020 2021 2022 2023 2024 2025

		BUG_ON(sret > 0);
		if (sret) {
			ret = sret;
			goto done;
		}
		b = p->nodes[level];
	} else if (ins_len < 0 && btrfs_header_nritems(b) <
C
Chris Mason 已提交
2026
		   BTRFS_NODEPTRS_PER_BLOCK(root) / 2) {
2027 2028
		int sret;

2029 2030 2031 2032 2033 2034
		if (*write_lock_level < level + 1) {
			*write_lock_level = level + 1;
			btrfs_release_path(p);
			goto again;
		}

2035 2036 2037 2038 2039 2040
		sret = reada_for_balance(root, p, level);
		if (sret)
			goto again;

		btrfs_set_path_blocking(p);
		sret = balance_level(trans, root, p, level);
2041
		btrfs_clear_path_blocking(p, NULL, 0);
2042 2043 2044 2045 2046 2047 2048

		if (sret) {
			ret = sret;
			goto done;
		}
		b = p->nodes[level];
		if (!b) {
2049
			btrfs_release_path(p);
2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061
			goto again;
		}
		BUG_ON(btrfs_header_nritems(b) == 1);
	}
	return 0;

again:
	ret = -EAGAIN;
done:
	return ret;
}

C
Chris Mason 已提交
2062 2063 2064 2065 2066 2067
/*
 * 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 已提交
2068 2069
 * be inserted, and 1 is returned.  If there are other errors during the
 * search a negative error number is returned.
C
Chris Mason 已提交
2070 2071 2072 2073
 *
 * 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 已提交
2074
 */
2075 2076 2077
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)
2078
{
2079
	struct extent_buffer *b;
2080 2081
	int slot;
	int ret;
2082
	int err;
2083
	int level;
2084
	int lowest_unlock = 1;
2085 2086 2087
	int root_lock;
	/* everything at write_lock_level or lower must be write locked */
	int write_lock_level = 0;
2088
	u8 lowest_level = 0;
2089
	int min_write_lock_level;
2090

2091
	lowest_level = p->lowest_level;
2092
	WARN_ON(lowest_level && ins_len > 0);
C
Chris Mason 已提交
2093
	WARN_ON(p->nodes[0] != NULL);
2094

2095
	if (ins_len < 0) {
2096
		lowest_unlock = 2;
2097

2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116
		/* when we are removing items, we might have to go up to level
		 * two as we update tree pointers  Make sure we keep write
		 * for those levels as well
		 */
		write_lock_level = 2;
	} else if (ins_len > 0) {
		/*
		 * for inserting items, make sure we have a write lock on
		 * level 1 so we can update keys
		 */
		write_lock_level = 1;
	}

	if (!cow)
		write_lock_level = -1;

	if (cow && (p->keep_locks || p->lowest_level))
		write_lock_level = BTRFS_MAX_LEVEL;

2117 2118
	min_write_lock_level = write_lock_level;

2119
again:
2120 2121 2122 2123 2124
	/*
	 * we try very hard to do read locks on the root
	 */
	root_lock = BTRFS_READ_LOCK;
	level = 0;
2125
	if (p->search_commit_root) {
2126 2127 2128 2129
		/*
		 * the commit roots are read only
		 * so we always do read locks
		 */
2130 2131
		b = root->commit_root;
		extent_buffer_get(b);
2132
		level = btrfs_header_level(b);
2133
		if (!p->skip_locking)
2134
			btrfs_tree_read_lock(b);
2135
	} else {
2136
		if (p->skip_locking) {
2137
			b = btrfs_root_node(root);
2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155
			level = btrfs_header_level(b);
		} else {
			/* we don't know the level of the root node
			 * until we actually have it read locked
			 */
			b = btrfs_read_lock_root_node(root);
			level = btrfs_header_level(b);
			if (level <= write_lock_level) {
				/* whoops, must trade for write lock */
				btrfs_tree_read_unlock(b);
				free_extent_buffer(b);
				b = btrfs_lock_root_node(root);
				root_lock = BTRFS_WRITE_LOCK;

				/* the level might have changed, check again */
				level = btrfs_header_level(b);
			}
		}
2156
	}
2157 2158 2159
	p->nodes[level] = b;
	if (!p->skip_locking)
		p->locks[level] = root_lock;
2160

2161
	while (b) {
2162
		level = btrfs_header_level(b);
2163 2164 2165 2166 2167

		/*
		 * setup the path here so we can release it under lock
		 * contention with the cow code
		 */
C
Chris Mason 已提交
2168
		if (cow) {
2169 2170 2171 2172 2173
			/*
			 * if we don't really need to cow this block
			 * then we don't want to set the path blocking,
			 * so we test it here
			 */
2174
			if (!should_cow_block(trans, root, b))
2175
				goto cow_done;
2176

2177 2178
			btrfs_set_path_blocking(p);

2179 2180 2181 2182 2183 2184 2185 2186 2187 2188
			/*
			 * must have write locks on this node and the
			 * parent
			 */
			if (level + 1 > write_lock_level) {
				write_lock_level = level + 1;
				btrfs_release_path(p);
				goto again;
			}

2189 2190 2191 2192 2193
			err = btrfs_cow_block(trans, root, b,
					      p->nodes[level + 1],
					      p->slots[level + 1], &b);
			if (err) {
				ret = err;
2194
				goto done;
2195
			}
C
Chris Mason 已提交
2196
		}
2197
cow_done:
C
Chris Mason 已提交
2198
		BUG_ON(!cow && ins_len);
2199

2200
		p->nodes[level] = b;
2201
		btrfs_clear_path_blocking(p, NULL, 0);
2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216

		/*
		 * we have a lock on b and as long as we aren't changing
		 * the tree, there is no way to for the items in b to change.
		 * It is safe to drop the lock on our parent before we
		 * go through the expensive btree search on b.
		 *
		 * If cow is true, then we might be changing slot zero,
		 * which may require changing the parent.  So, we can't
		 * drop the lock until after we know which slot we're
		 * operating on.
		 */
		if (!cow)
			btrfs_unlock_up_safe(p, level + 1);

2217
		ret = bin_search(b, key, level, &slot);
2218

2219
		if (level != 0) {
2220 2221 2222
			int dec = 0;
			if (ret && slot > 0) {
				dec = 1;
2223
				slot -= 1;
2224
			}
2225
			p->slots[level] = slot;
2226
			err = setup_nodes_for_search(trans, root, p, b, level,
2227
					     ins_len, &write_lock_level);
2228
			if (err == -EAGAIN)
2229
				goto again;
2230 2231
			if (err) {
				ret = err;
2232
				goto done;
2233
			}
2234 2235
			b = p->nodes[level];
			slot = p->slots[level];
2236

2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249
			/*
			 * slot 0 is special, if we change the key
			 * we have to update the parent pointer
			 * which means we must have a write lock
			 * on the parent
			 */
			if (slot == 0 && cow &&
			    write_lock_level < level + 1) {
				write_lock_level = level + 1;
				btrfs_release_path(p);
				goto again;
			}

2250 2251
			unlock_up(p, level, lowest_unlock,
				  min_write_lock_level, &write_lock_level);
2252

2253
			if (level == lowest_level) {
2254 2255
				if (dec)
					p->slots[level]++;
2256
				goto done;
2257
			}
2258

2259
			err = read_block_for_search(trans, root, p,
2260
						    &b, level, slot, key);
2261
			if (err == -EAGAIN)
2262
				goto again;
2263 2264
			if (err) {
				ret = err;
2265
				goto done;
2266
			}
2267

2268
			if (!p->skip_locking) {
2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287
				level = btrfs_header_level(b);
				if (level <= write_lock_level) {
					err = btrfs_try_tree_write_lock(b);
					if (!err) {
						btrfs_set_path_blocking(p);
						btrfs_tree_lock(b);
						btrfs_clear_path_blocking(p, b,
								  BTRFS_WRITE_LOCK);
					}
					p->locks[level] = BTRFS_WRITE_LOCK;
				} else {
					err = btrfs_try_tree_read_lock(b);
					if (!err) {
						btrfs_set_path_blocking(p);
						btrfs_tree_read_lock(b);
						btrfs_clear_path_blocking(p, b,
								  BTRFS_READ_LOCK);
					}
					p->locks[level] = BTRFS_READ_LOCK;
2288
				}
2289
				p->nodes[level] = b;
2290
			}
2291 2292
		} else {
			p->slots[level] = slot;
2293 2294
			if (ins_len > 0 &&
			    btrfs_leaf_free_space(root, b) < ins_len) {
2295 2296 2297 2298 2299 2300
				if (write_lock_level < 1) {
					write_lock_level = 1;
					btrfs_release_path(p);
					goto again;
				}

2301
				btrfs_set_path_blocking(p);
2302 2303
				err = split_leaf(trans, root, key,
						 p, ins_len, ret == 0);
2304
				btrfs_clear_path_blocking(p, NULL, 0);
2305

2306 2307 2308
				BUG_ON(err > 0);
				if (err) {
					ret = err;
2309 2310
					goto done;
				}
C
Chris Mason 已提交
2311
			}
2312
			if (!p->search_for_split)
2313 2314
				unlock_up(p, level, lowest_unlock,
					  min_write_lock_level, &write_lock_level);
2315
			goto done;
2316 2317
		}
	}
2318 2319
	ret = 1;
done:
2320 2321 2322 2323
	/*
	 * we don't really know what they plan on doing with the path
	 * from here on, so for now just mark it as blocking
	 */
2324 2325
	if (!p->leave_spinning)
		btrfs_set_path_blocking(p);
2326
	if (ret < 0)
2327
		btrfs_release_path(p);
2328
	return ret;
2329 2330
}

C
Chris Mason 已提交
2331 2332 2333 2334 2335 2336
/*
 * 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 已提交
2337
 *
C
Chris Mason 已提交
2338
 */
2339 2340 2341
static void fixup_low_keys(struct btrfs_trans_handle *trans,
			   struct btrfs_root *root, struct btrfs_path *path,
			   struct btrfs_disk_key *key, int level)
2342 2343
{
	int i;
2344 2345
	struct extent_buffer *t;

C
Chris Mason 已提交
2346
	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
2347
		int tslot = path->slots[i];
2348
		if (!path->nodes[i])
2349
			break;
2350 2351
		t = path->nodes[i];
		btrfs_set_node_key(t, key, tslot);
C
Chris Mason 已提交
2352
		btrfs_mark_buffer_dirty(path->nodes[i]);
2353 2354 2355 2356 2357
		if (tslot != 0)
			break;
	}
}

Z
Zheng Yan 已提交
2358 2359 2360 2361 2362 2363
/*
 * update item key.
 *
 * This function isn't completely safe. It's the caller's responsibility
 * that the new key won't break the order
 */
2364 2365 2366
void btrfs_set_item_key_safe(struct btrfs_trans_handle *trans,
			     struct btrfs_root *root, struct btrfs_path *path,
			     struct btrfs_key *new_key)
Z
Zheng Yan 已提交
2367 2368 2369 2370 2371 2372 2373 2374 2375
{
	struct btrfs_disk_key disk_key;
	struct extent_buffer *eb;
	int slot;

	eb = path->nodes[0];
	slot = path->slots[0];
	if (slot > 0) {
		btrfs_item_key(eb, &disk_key, slot - 1);
2376
		BUG_ON(comp_keys(&disk_key, new_key) >= 0);
Z
Zheng Yan 已提交
2377 2378 2379
	}
	if (slot < btrfs_header_nritems(eb) - 1) {
		btrfs_item_key(eb, &disk_key, slot + 1);
2380
		BUG_ON(comp_keys(&disk_key, new_key) <= 0);
Z
Zheng Yan 已提交
2381 2382 2383 2384 2385 2386 2387 2388 2389
	}

	btrfs_cpu_key_to_disk(&disk_key, new_key);
	btrfs_set_item_key(eb, &disk_key, slot);
	btrfs_mark_buffer_dirty(eb);
	if (slot == 0)
		fixup_low_keys(trans, root, path, &disk_key, 1);
}

C
Chris Mason 已提交
2390 2391
/*
 * try to push data from one node into the next node left in the
2392
 * tree.
C
Chris Mason 已提交
2393 2394 2395
 *
 * 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 已提交
2396
 */
2397 2398
static int push_node_left(struct btrfs_trans_handle *trans,
			  struct btrfs_root *root, struct extent_buffer *dst,
2399
			  struct extent_buffer *src, int empty)
2400 2401
{
	int push_items = 0;
2402 2403
	int src_nritems;
	int dst_nritems;
C
Chris Mason 已提交
2404
	int ret = 0;
2405

2406 2407
	src_nritems = btrfs_header_nritems(src);
	dst_nritems = btrfs_header_nritems(dst);
C
Chris Mason 已提交
2408
	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
2409 2410
	WARN_ON(btrfs_header_generation(src) != trans->transid);
	WARN_ON(btrfs_header_generation(dst) != trans->transid);
2411

2412
	if (!empty && src_nritems <= 8)
2413 2414
		return 1;

C
Chris Mason 已提交
2415
	if (push_items <= 0)
2416 2417
		return 1;

2418
	if (empty) {
2419
		push_items = min(src_nritems, push_items);
2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431
		if (push_items < src_nritems) {
			/* leave at least 8 pointers in the node if
			 * we aren't going to empty it
			 */
			if (src_nritems - push_items < 8) {
				if (push_items <= 8)
					return 1;
				push_items -= 8;
			}
		}
	} else
		push_items = min(src_nritems - 8, push_items);
2432

2433 2434 2435
	copy_extent_buffer(dst, src,
			   btrfs_node_key_ptr_offset(dst_nritems),
			   btrfs_node_key_ptr_offset(0),
C
Chris Mason 已提交
2436
			   push_items * sizeof(struct btrfs_key_ptr));
2437

2438
	if (push_items < src_nritems) {
2439 2440 2441 2442 2443 2444 2445 2446 2447
		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);
Z
Zheng Yan 已提交
2448

2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460
	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
 */
2461 2462 2463 2464
static int balance_node_right(struct btrfs_trans_handle *trans,
			      struct btrfs_root *root,
			      struct extent_buffer *dst,
			      struct extent_buffer *src)
2465 2466 2467 2468 2469 2470 2471
{
	int push_items = 0;
	int max_push;
	int src_nritems;
	int dst_nritems;
	int ret = 0;

2472 2473 2474
	WARN_ON(btrfs_header_generation(src) != trans->transid);
	WARN_ON(btrfs_header_generation(dst) != trans->transid);

2475 2476
	src_nritems = btrfs_header_nritems(src);
	dst_nritems = btrfs_header_nritems(dst);
C
Chris Mason 已提交
2477
	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
C
Chris Mason 已提交
2478
	if (push_items <= 0)
2479
		return 1;
2480

C
Chris Mason 已提交
2481
	if (src_nritems < 4)
2482
		return 1;
2483 2484 2485

	max_push = src_nritems / 2 + 1;
	/* don't try to empty the node */
C
Chris Mason 已提交
2486
	if (max_push >= src_nritems)
2487
		return 1;
Y
Yan 已提交
2488

2489 2490 2491
	if (max_push < push_items)
		push_items = max_push;

2492 2493 2494 2495
	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 已提交
2496

2497 2498 2499
	copy_extent_buffer(dst, src,
			   btrfs_node_key_ptr_offset(0),
			   btrfs_node_key_ptr_offset(src_nritems - push_items),
C
Chris Mason 已提交
2500
			   push_items * sizeof(struct btrfs_key_ptr));
2501

2502 2503
	btrfs_set_header_nritems(src, src_nritems - push_items);
	btrfs_set_header_nritems(dst, dst_nritems + push_items);
2504

2505 2506
	btrfs_mark_buffer_dirty(src);
	btrfs_mark_buffer_dirty(dst);
Z
Zheng Yan 已提交
2507

C
Chris Mason 已提交
2508
	return ret;
2509 2510
}

C
Chris Mason 已提交
2511 2512 2513 2514
/*
 * 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 已提交
2515 2516
 *
 * returns zero on success or < 0 on failure.
C
Chris Mason 已提交
2517
 */
C
Chris Mason 已提交
2518
static noinline int insert_new_root(struct btrfs_trans_handle *trans,
2519 2520
			   struct btrfs_root *root,
			   struct btrfs_path *path, int level)
C
Chris Mason 已提交
2521
{
2522
	u64 lower_gen;
2523 2524
	struct extent_buffer *lower;
	struct extent_buffer *c;
2525
	struct extent_buffer *old;
2526
	struct btrfs_disk_key lower_key;
C
Chris Mason 已提交
2527 2528 2529 2530

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

2531 2532 2533 2534 2535 2536
	lower = path->nodes[level-1];
	if (level == 1)
		btrfs_item_key(lower, &lower_key, 0);
	else
		btrfs_node_key(lower, &lower_key, 0);

Z
Zheng Yan 已提交
2537
	c = btrfs_alloc_free_block(trans, root, root->nodesize, 0,
2538
				   root->root_key.objectid, &lower_key,
2539
				   level, root->node->start, 0);
2540 2541
	if (IS_ERR(c))
		return PTR_ERR(c);
2542

2543 2544
	root_add_used(root, root->nodesize);

2545
	memset_extent_buffer(c, 0, 0, sizeof(struct btrfs_header));
2546 2547
	btrfs_set_header_nritems(c, 1);
	btrfs_set_header_level(c, level);
2548
	btrfs_set_header_bytenr(c, c->start);
2549
	btrfs_set_header_generation(c, trans->transid);
2550
	btrfs_set_header_backref_rev(c, BTRFS_MIXED_BACKREF_REV);
2551 2552 2553 2554 2555
	btrfs_set_header_owner(c, root->root_key.objectid);

	write_extent_buffer(c, root->fs_info->fsid,
			    (unsigned long)btrfs_header_fsid(c),
			    BTRFS_FSID_SIZE);
2556 2557 2558 2559 2560

	write_extent_buffer(c, root->fs_info->chunk_tree_uuid,
			    (unsigned long)btrfs_header_chunk_tree_uuid(c),
			    BTRFS_UUID_SIZE);

2561
	btrfs_set_node_key(c, &lower_key, 0);
2562
	btrfs_set_node_blockptr(c, 0, lower->start);
2563
	lower_gen = btrfs_header_generation(lower);
Z
Zheng Yan 已提交
2564
	WARN_ON(lower_gen != trans->transid);
2565 2566

	btrfs_set_node_ptr_generation(c, 0, lower_gen);
2567

2568
	btrfs_mark_buffer_dirty(c);
2569

2570
	old = root->node;
2571
	rcu_assign_pointer(root->node, c);
2572 2573 2574 2575

	/* the super has an extra ref to root->node */
	free_extent_buffer(old);

2576
	add_root_to_dirty_list(root);
2577 2578
	extent_buffer_get(c);
	path->nodes[level] = c;
2579
	path->locks[level] = BTRFS_WRITE_LOCK;
C
Chris Mason 已提交
2580 2581 2582 2583
	path->slots[level] = 0;
	return 0;
}

C
Chris Mason 已提交
2584 2585 2586
/*
 * worker function to insert a single pointer in a node.
 * the node should have enough room for the pointer already
C
Chris Mason 已提交
2587
 *
C
Chris Mason 已提交
2588 2589 2590
 * slot and level indicate where you want the key to go, and
 * blocknr is the block the key points to.
 */
2591 2592 2593 2594
static void insert_ptr(struct btrfs_trans_handle *trans,
		       struct btrfs_root *root, struct btrfs_path *path,
		       struct btrfs_disk_key *key, u64 bytenr,
		       int slot, int level)
C
Chris Mason 已提交
2595
{
2596
	struct extent_buffer *lower;
C
Chris Mason 已提交
2597
	int nritems;
C
Chris Mason 已提交
2598 2599

	BUG_ON(!path->nodes[level]);
2600
	btrfs_assert_tree_locked(path->nodes[level]);
2601 2602
	lower = path->nodes[level];
	nritems = btrfs_header_nritems(lower);
S
Stoyan Gaydarov 已提交
2603
	BUG_ON(slot > nritems);
2604
	BUG_ON(nritems == BTRFS_NODEPTRS_PER_BLOCK(root));
C
Chris Mason 已提交
2605
	if (slot != nritems) {
2606 2607 2608
		memmove_extent_buffer(lower,
			      btrfs_node_key_ptr_offset(slot + 1),
			      btrfs_node_key_ptr_offset(slot),
C
Chris Mason 已提交
2609
			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
C
Chris Mason 已提交
2610
	}
2611
	btrfs_set_node_key(lower, key, slot);
2612
	btrfs_set_node_blockptr(lower, slot, bytenr);
2613 2614
	WARN_ON(trans->transid == 0);
	btrfs_set_node_ptr_generation(lower, slot, trans->transid);
2615 2616
	btrfs_set_header_nritems(lower, nritems + 1);
	btrfs_mark_buffer_dirty(lower);
C
Chris Mason 已提交
2617 2618
}

C
Chris Mason 已提交
2619 2620 2621 2622 2623 2624
/*
 * 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 已提交
2625 2626
 *
 * returns 0 on success and < 0 on failure
C
Chris Mason 已提交
2627
 */
2628 2629 2630
static noinline int split_node(struct btrfs_trans_handle *trans,
			       struct btrfs_root *root,
			       struct btrfs_path *path, int level)
2631
{
2632 2633 2634
	struct extent_buffer *c;
	struct extent_buffer *split;
	struct btrfs_disk_key disk_key;
2635
	int mid;
C
Chris Mason 已提交
2636
	int ret;
2637
	u32 c_nritems;
2638

2639
	c = path->nodes[level];
2640
	WARN_ON(btrfs_header_generation(c) != trans->transid);
2641
	if (c == root->node) {
C
Chris Mason 已提交
2642
		/* trying to split the root, lets make a new one */
2643
		ret = insert_new_root(trans, root, path, level + 1);
C
Chris Mason 已提交
2644 2645
		if (ret)
			return ret;
2646
	} else {
2647
		ret = push_nodes_for_insert(trans, root, path, level);
2648 2649
		c = path->nodes[level];
		if (!ret && btrfs_header_nritems(c) <
2650
		    BTRFS_NODEPTRS_PER_BLOCK(root) - 3)
2651
			return 0;
2652 2653
		if (ret < 0)
			return ret;
2654
	}
2655

2656
	c_nritems = btrfs_header_nritems(c);
2657 2658
	mid = (c_nritems + 1) / 2;
	btrfs_node_key(c, &disk_key, mid);
2659

2660
	split = btrfs_alloc_free_block(trans, root, root->nodesize, 0,
Z
Zheng Yan 已提交
2661
					root->root_key.objectid,
2662
					&disk_key, level, c->start, 0);
2663 2664 2665
	if (IS_ERR(split))
		return PTR_ERR(split);

2666 2667
	root_add_used(root, root->nodesize);

2668
	memset_extent_buffer(split, 0, 0, sizeof(struct btrfs_header));
2669
	btrfs_set_header_level(split, btrfs_header_level(c));
2670
	btrfs_set_header_bytenr(split, split->start);
2671
	btrfs_set_header_generation(split, trans->transid);
2672
	btrfs_set_header_backref_rev(split, BTRFS_MIXED_BACKREF_REV);
2673 2674 2675 2676
	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);
2677 2678 2679
	write_extent_buffer(split, root->fs_info->chunk_tree_uuid,
			    (unsigned long)btrfs_header_chunk_tree_uuid(split),
			    BTRFS_UUID_SIZE);
2680

2681 2682 2683 2684 2685 2686
	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 已提交
2687 2688
	ret = 0;

2689 2690 2691
	btrfs_mark_buffer_dirty(c);
	btrfs_mark_buffer_dirty(split);

2692 2693
	insert_ptr(trans, root, path, &disk_key, split->start,
		   path->slots[level + 1] + 1, level + 1);
C
Chris Mason 已提交
2694

C
Chris Mason 已提交
2695
	if (path->slots[level] >= mid) {
C
Chris Mason 已提交
2696
		path->slots[level] -= mid;
2697
		btrfs_tree_unlock(c);
2698 2699
		free_extent_buffer(c);
		path->nodes[level] = split;
C
Chris Mason 已提交
2700 2701
		path->slots[level + 1] += 1;
	} else {
2702
		btrfs_tree_unlock(split);
2703
		free_extent_buffer(split);
2704
	}
C
Chris Mason 已提交
2705
	return ret;
2706 2707
}

C
Chris Mason 已提交
2708 2709 2710 2711 2712
/*
 * 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
 */
2713
static int leaf_space_used(struct extent_buffer *l, int start, int nr)
2714 2715
{
	int data_len;
2716
	int nritems = btrfs_header_nritems(l);
2717
	int end = min(nritems, start + nr) - 1;
2718 2719 2720

	if (!nr)
		return 0;
2721 2722
	data_len = btrfs_item_end_nr(l, start);
	data_len = data_len - btrfs_item_offset_nr(l, end);
C
Chris Mason 已提交
2723
	data_len += sizeof(struct btrfs_item) * nr;
2724
	WARN_ON(data_len < 0);
2725 2726 2727
	return data_len;
}

2728 2729 2730 2731 2732
/*
 * 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
 */
C
Chris Mason 已提交
2733
noinline int btrfs_leaf_free_space(struct btrfs_root *root,
2734
				   struct extent_buffer *leaf)
2735
{
2736 2737 2738 2739
	int nritems = btrfs_header_nritems(leaf);
	int ret;
	ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
	if (ret < 0) {
C
Chris Mason 已提交
2740 2741
		printk(KERN_CRIT "leaf free space ret %d, leaf data size %lu, "
		       "used %d nritems %d\n",
J
Jens Axboe 已提交
2742
		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
2743 2744 2745
		       leaf_space_used(leaf, 0, nritems), nritems);
	}
	return ret;
2746 2747
}

2748 2749 2750 2751
/*
 * min slot controls the lowest index we're willing to push to the
 * right.  We'll push up to and including min_slot, but no lower
 */
2752 2753 2754 2755 2756
static noinline int __push_leaf_right(struct btrfs_trans_handle *trans,
				      struct btrfs_root *root,
				      struct btrfs_path *path,
				      int data_size, int empty,
				      struct extent_buffer *right,
2757 2758
				      int free_space, u32 left_nritems,
				      u32 min_slot)
C
Chris Mason 已提交
2759
{
2760
	struct extent_buffer *left = path->nodes[0];
2761
	struct extent_buffer *upper = path->nodes[1];
2762
	struct btrfs_map_token token;
2763
	struct btrfs_disk_key disk_key;
C
Chris Mason 已提交
2764
	int slot;
2765
	u32 i;
C
Chris Mason 已提交
2766 2767
	int push_space = 0;
	int push_items = 0;
C
Chris Mason 已提交
2768
	struct btrfs_item *item;
2769
	u32 nr;
2770
	u32 right_nritems;
2771
	u32 data_end;
2772
	u32 this_item_size;
C
Chris Mason 已提交
2773

2774 2775
	btrfs_init_map_token(&token);

2776 2777 2778
	if (empty)
		nr = 0;
	else
2779
		nr = max_t(u32, 1, min_slot);
2780

Z
Zheng Yan 已提交
2781
	if (path->slots[0] >= left_nritems)
2782
		push_space += data_size;
Z
Zheng Yan 已提交
2783

2784
	slot = path->slots[1];
2785 2786
	i = left_nritems - 1;
	while (i >= nr) {
2787
		item = btrfs_item_nr(left, i);
2788

Z
Zheng Yan 已提交
2789 2790 2791 2792 2793 2794 2795 2796 2797 2798
		if (!empty && push_items > 0) {
			if (path->slots[0] > i)
				break;
			if (path->slots[0] == i) {
				int space = btrfs_leaf_free_space(root, left);
				if (space + push_space * 2 > free_space)
					break;
			}
		}

C
Chris Mason 已提交
2799
		if (path->slots[0] == i)
2800
			push_space += data_size;
2801 2802 2803

		this_item_size = btrfs_item_size(left, item);
		if (this_item_size + sizeof(*item) + push_space > free_space)
C
Chris Mason 已提交
2804
			break;
Z
Zheng Yan 已提交
2805

C
Chris Mason 已提交
2806
		push_items++;
2807
		push_space += this_item_size + sizeof(*item);
2808 2809 2810
		if (i == 0)
			break;
		i--;
2811
	}
2812

2813 2814
	if (push_items == 0)
		goto out_unlock;
2815

2816
	if (!empty && push_items == left_nritems)
2817
		WARN_ON(1);
2818

C
Chris Mason 已提交
2819
	/* push left to right */
2820
	right_nritems = btrfs_header_nritems(right);
2821

2822
	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
C
Chris Mason 已提交
2823
	push_space -= leaf_data_end(root, left);
2824

C
Chris Mason 已提交
2825
	/* make room in the right data area */
2826 2827 2828 2829 2830 2831
	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 已提交
2832
	/* copy from the left data area */
2833
	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
C
Chris Mason 已提交
2834 2835 2836
		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
		     btrfs_leaf_data(left) + leaf_data_end(root, left),
		     push_space);
2837 2838 2839 2840 2841

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

C
Chris Mason 已提交
2842
	/* copy the items from left to right */
2843 2844 2845
	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 已提交
2846 2847

	/* update the item pointers */
2848
	right_nritems += push_items;
2849
	btrfs_set_header_nritems(right, right_nritems);
C
Chris Mason 已提交
2850
	push_space = BTRFS_LEAF_DATA_SIZE(root);
2851
	for (i = 0; i < right_nritems; i++) {
2852
		item = btrfs_item_nr(right, i);
2853 2854
		push_space -= btrfs_token_item_size(right, item, &token);
		btrfs_set_token_item_offset(right, item, push_space, &token);
2855 2856
	}

2857
	left_nritems -= push_items;
2858
	btrfs_set_header_nritems(left, left_nritems);
C
Chris Mason 已提交
2859

2860 2861
	if (left_nritems)
		btrfs_mark_buffer_dirty(left);
2862 2863 2864
	else
		clean_tree_block(trans, root, left);

2865
	btrfs_mark_buffer_dirty(right);
2866

2867 2868
	btrfs_item_key(right, &disk_key, 0);
	btrfs_set_node_key(upper, &disk_key, slot + 1);
C
Chris Mason 已提交
2869
	btrfs_mark_buffer_dirty(upper);
C
Chris Mason 已提交
2870

C
Chris Mason 已提交
2871
	/* then fixup the leaf pointer in the path */
2872 2873
	if (path->slots[0] >= left_nritems) {
		path->slots[0] -= left_nritems;
2874 2875 2876
		if (btrfs_header_nritems(path->nodes[0]) == 0)
			clean_tree_block(trans, root, path->nodes[0]);
		btrfs_tree_unlock(path->nodes[0]);
2877 2878
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = right;
C
Chris Mason 已提交
2879 2880
		path->slots[1] += 1;
	} else {
2881
		btrfs_tree_unlock(right);
2882
		free_extent_buffer(right);
C
Chris Mason 已提交
2883 2884
	}
	return 0;
2885 2886 2887 2888 2889

out_unlock:
	btrfs_tree_unlock(right);
	free_extent_buffer(right);
	return 1;
C
Chris Mason 已提交
2890
}
2891

2892 2893 2894 2895 2896 2897
/*
 * 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
 *
 * 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.
2898 2899 2900
 *
 * this will push starting from min_slot to the end of the leaf.  It won't
 * push any slot lower than min_slot
2901 2902
 */
static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
2903 2904 2905
			   *root, struct btrfs_path *path,
			   int min_data_size, int data_size,
			   int empty, u32 min_slot)
2906 2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925
{
	struct extent_buffer *left = path->nodes[0];
	struct extent_buffer *right;
	struct extent_buffer *upper;
	int slot;
	int free_space;
	u32 left_nritems;
	int ret;

	if (!path->nodes[1])
		return 1;

	slot = path->slots[1];
	upper = path->nodes[1];
	if (slot >= btrfs_header_nritems(upper) - 1)
		return 1;

	btrfs_assert_tree_locked(path->nodes[1]);

	right = read_node_slot(root, upper, slot + 1);
T
Tsutomu Itoh 已提交
2926 2927 2928
	if (right == NULL)
		return 1;

2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949
	btrfs_tree_lock(right);
	btrfs_set_lock_blocking(right);

	free_space = btrfs_leaf_free_space(root, right);
	if (free_space < data_size)
		goto out_unlock;

	/* cow and double check */
	ret = btrfs_cow_block(trans, root, right, upper,
			      slot + 1, &right);
	if (ret)
		goto out_unlock;

	free_space = btrfs_leaf_free_space(root, right);
	if (free_space < data_size)
		goto out_unlock;

	left_nritems = btrfs_header_nritems(left);
	if (left_nritems == 0)
		goto out_unlock;

2950 2951
	return __push_leaf_right(trans, root, path, min_data_size, empty,
				right, free_space, left_nritems, min_slot);
2952 2953 2954 2955 2956 2957
out_unlock:
	btrfs_tree_unlock(right);
	free_extent_buffer(right);
	return 1;
}

C
Chris Mason 已提交
2958 2959 2960
/*
 * 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
2961 2962 2963 2964
 *
 * max_slot can put a limit on how far into the leaf we'll push items.  The
 * item at 'max_slot' won't be touched.  Use (u32)-1 to make us do all the
 * items
C
Chris Mason 已提交
2965
 */
2966 2967 2968 2969
static noinline int __push_leaf_left(struct btrfs_trans_handle *trans,
				     struct btrfs_root *root,
				     struct btrfs_path *path, int data_size,
				     int empty, struct extent_buffer *left,
2970 2971
				     int free_space, u32 right_nritems,
				     u32 max_slot)
2972
{
2973 2974
	struct btrfs_disk_key disk_key;
	struct extent_buffer *right = path->nodes[0];
2975 2976 2977
	int i;
	int push_space = 0;
	int push_items = 0;
C
Chris Mason 已提交
2978
	struct btrfs_item *item;
2979
	u32 old_left_nritems;
2980
	u32 nr;
C
Chris Mason 已提交
2981
	int ret = 0;
2982 2983
	u32 this_item_size;
	u32 old_left_item_size;
2984 2985 2986
	struct btrfs_map_token token;

	btrfs_init_map_token(&token);
2987

2988
	if (empty)
2989
		nr = min(right_nritems, max_slot);
2990
	else
2991
		nr = min(right_nritems - 1, max_slot);
2992 2993

	for (i = 0; i < nr; i++) {
2994
		item = btrfs_item_nr(right, i);
2995

Z
Zheng Yan 已提交
2996 2997 2998 2999 3000 3001 3002 3003 3004 3005
		if (!empty && push_items > 0) {
			if (path->slots[0] < i)
				break;
			if (path->slots[0] == i) {
				int space = btrfs_leaf_free_space(root, right);
				if (space + push_space * 2 > free_space)
					break;
			}
		}

3006
		if (path->slots[0] == i)
3007
			push_space += data_size;
3008 3009 3010

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

3013
		push_items++;
3014 3015 3016
		push_space += this_item_size + sizeof(*item);
	}

3017
	if (push_items == 0) {
3018 3019
		ret = 1;
		goto out;
3020
	}
3021
	if (!empty && push_items == btrfs_header_nritems(right))
3022
		WARN_ON(1);
3023

3024
	/* push data from right to left */
3025 3026 3027 3028 3029
	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 已提交
3030
	push_space = BTRFS_LEAF_DATA_SIZE(root) -
C
Chris Mason 已提交
3031
		     btrfs_item_offset_nr(right, push_items - 1);
3032 3033

	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
C
Chris Mason 已提交
3034 3035
		     leaf_data_end(root, left) - push_space,
		     btrfs_leaf_data(right) +
3036
		     btrfs_item_offset_nr(right, push_items - 1),
C
Chris Mason 已提交
3037
		     push_space);
3038
	old_left_nritems = btrfs_header_nritems(left);
3039
	BUG_ON(old_left_nritems <= 0);
3040

3041
	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
C
Chris Mason 已提交
3042
	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
3043
		u32 ioff;
3044

3045
		item = btrfs_item_nr(left, i);
3046

3047 3048 3049 3050
		ioff = btrfs_token_item_offset(left, item, &token);
		btrfs_set_token_item_offset(left, item,
		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size),
		      &token);
3051
	}
3052
	btrfs_set_header_nritems(left, old_left_nritems + push_items);
3053 3054

	/* fixup right node */
3055
	if (push_items > right_nritems) {
C
Chris Mason 已提交
3056 3057
		printk(KERN_CRIT "push items %d nr %u\n", push_items,
		       right_nritems);
3058 3059 3060 3061 3062 3063 3064 3065 3066 3067 3068 3069
		WARN_ON(1);
	}

	if (push_items < right_nritems) {
		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),
3070 3071 3072
			      btrfs_item_nr_offset(push_items),
			     (btrfs_header_nritems(right) - push_items) *
			     sizeof(struct btrfs_item));
3073
	}
3074 3075
	right_nritems -= push_items;
	btrfs_set_header_nritems(right, right_nritems);
C
Chris Mason 已提交
3076
	push_space = BTRFS_LEAF_DATA_SIZE(root);
3077 3078
	for (i = 0; i < right_nritems; i++) {
		item = btrfs_item_nr(right, i);
3079

3080 3081 3082
		push_space = push_space - btrfs_token_item_size(right,
								item, &token);
		btrfs_set_token_item_offset(right, item, push_space, &token);
3083
	}
3084

3085
	btrfs_mark_buffer_dirty(left);
3086 3087
	if (right_nritems)
		btrfs_mark_buffer_dirty(right);
3088 3089
	else
		clean_tree_block(trans, root, right);
3090

3091
	btrfs_item_key(right, &disk_key, 0);
3092
	fixup_low_keys(trans, root, path, &disk_key, 1);
3093 3094 3095 3096

	/* then fixup the leaf pointer in the path */
	if (path->slots[0] < push_items) {
		path->slots[0] += old_left_nritems;
3097
		btrfs_tree_unlock(path->nodes[0]);
3098 3099
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = left;
3100 3101
		path->slots[1] -= 1;
	} else {
3102
		btrfs_tree_unlock(left);
3103
		free_extent_buffer(left);
3104 3105
		path->slots[0] -= push_items;
	}
3106
	BUG_ON(path->slots[0] < 0);
C
Chris Mason 已提交
3107
	return ret;
3108 3109 3110 3111
out:
	btrfs_tree_unlock(left);
	free_extent_buffer(left);
	return ret;
3112 3113
}

3114 3115 3116
/*
 * 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
3117 3118 3119 3120
 *
 * max_slot can put a limit on how far into the leaf we'll push items.  The
 * item at 'max_slot' won't be touched.  Use (u32)-1 to make us push all the
 * items
3121 3122
 */
static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
3123 3124
			  *root, struct btrfs_path *path, int min_data_size,
			  int data_size, int empty, u32 max_slot)
3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145
{
	struct extent_buffer *right = path->nodes[0];
	struct extent_buffer *left;
	int slot;
	int free_space;
	u32 right_nritems;
	int ret = 0;

	slot = path->slots[1];
	if (slot == 0)
		return 1;
	if (!path->nodes[1])
		return 1;

	right_nritems = btrfs_header_nritems(right);
	if (right_nritems == 0)
		return 1;

	btrfs_assert_tree_locked(path->nodes[1]);

	left = read_node_slot(root, path->nodes[1], slot - 1);
T
Tsutomu Itoh 已提交
3146 3147 3148
	if (left == NULL)
		return 1;

3149 3150 3151 3152 3153 3154 3155 3156 3157 3158 3159 3160 3161 3162
	btrfs_tree_lock(left);
	btrfs_set_lock_blocking(left);

	free_space = btrfs_leaf_free_space(root, left);
	if (free_space < data_size) {
		ret = 1;
		goto out;
	}

	/* cow and double check */
	ret = btrfs_cow_block(trans, root, left,
			      path->nodes[1], slot - 1, &left);
	if (ret) {
		/* we hit -ENOSPC, but it isn't fatal here */
3163 3164
		if (ret == -ENOSPC)
			ret = 1;
3165 3166 3167 3168 3169 3170 3171 3172 3173
		goto out;
	}

	free_space = btrfs_leaf_free_space(root, left);
	if (free_space < data_size) {
		ret = 1;
		goto out;
	}

3174 3175 3176
	return __push_leaf_left(trans, root, path, min_data_size,
			       empty, left, free_space, right_nritems,
			       max_slot);
3177 3178 3179 3180 3181 3182 3183 3184 3185 3186
out:
	btrfs_tree_unlock(left);
	free_extent_buffer(left);
	return ret;
}

/*
 * split the path's leaf in two, making sure there is at least data_size
 * available for the resulting leaf level of the path.
 */
3187 3188 3189 3190 3191 3192
static noinline void copy_for_split(struct btrfs_trans_handle *trans,
				    struct btrfs_root *root,
				    struct btrfs_path *path,
				    struct extent_buffer *l,
				    struct extent_buffer *right,
				    int slot, int mid, int nritems)
3193 3194 3195 3196 3197
{
	int data_copy_size;
	int rt_data_off;
	int i;
	struct btrfs_disk_key disk_key;
3198 3199 3200
	struct btrfs_map_token token;

	btrfs_init_map_token(&token);
3201 3202 3203 3204 3205 3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221

	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,
		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
		     data_copy_size, btrfs_leaf_data(l) +
		     leaf_data_end(root, l), data_copy_size);

	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
		      btrfs_item_end_nr(l, mid);

	for (i = 0; i < nritems; i++) {
		struct btrfs_item *item = btrfs_item_nr(right, i);
		u32 ioff;

3222 3223 3224
		ioff = btrfs_token_item_offset(right, item, &token);
		btrfs_set_token_item_offset(right, item,
					    ioff + rt_data_off, &token);
3225 3226 3227 3228
	}

	btrfs_set_header_nritems(l, mid);
	btrfs_item_key(right, &disk_key, 0);
3229 3230
	insert_ptr(trans, root, path, &disk_key, right->start,
		   path->slots[1] + 1, 1);
3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249

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

	if (mid <= slot) {
		btrfs_tree_unlock(path->nodes[0]);
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = right;
		path->slots[0] -= mid;
		path->slots[1] += 1;
	} else {
		btrfs_tree_unlock(right);
		free_extent_buffer(right);
	}

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

3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298 3299 3300 3301 3302 3303 3304 3305 3306 3307
/*
 * double splits happen when we need to insert a big item in the middle
 * of a leaf.  A double split can leave us with 3 mostly empty leaves:
 * leaf: [ slots 0 - N] [ our target ] [ N + 1 - total in leaf ]
 *          A                 B                 C
 *
 * We avoid this by trying to push the items on either side of our target
 * into the adjacent leaves.  If all goes well we can avoid the double split
 * completely.
 */
static noinline int push_for_double_split(struct btrfs_trans_handle *trans,
					  struct btrfs_root *root,
					  struct btrfs_path *path,
					  int data_size)
{
	int ret;
	int progress = 0;
	int slot;
	u32 nritems;

	slot = path->slots[0];

	/*
	 * try to push all the items after our slot into the
	 * right leaf
	 */
	ret = push_leaf_right(trans, root, path, 1, data_size, 0, slot);
	if (ret < 0)
		return ret;

	if (ret == 0)
		progress++;

	nritems = btrfs_header_nritems(path->nodes[0]);
	/*
	 * our goal is to get our slot at the start or end of a leaf.  If
	 * we've done so we're done
	 */
	if (path->slots[0] == 0 || path->slots[0] == nritems)
		return 0;

	if (btrfs_leaf_free_space(root, path->nodes[0]) >= data_size)
		return 0;

	/* try to push all the items before our slot into the next leaf */
	slot = path->slots[0];
	ret = push_leaf_left(trans, root, path, 1, data_size, 0, slot);
	if (ret < 0)
		return ret;

	if (ret == 0)
		progress++;

	if (progress)
		return 0;
	return 1;
}

C
Chris Mason 已提交
3308 3309 3310
/*
 * 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 已提交
3311 3312
 *
 * returns 0 if all went well and < 0 on failure.
C
Chris Mason 已提交
3313
 */
3314 3315 3316 3317 3318
static noinline int split_leaf(struct btrfs_trans_handle *trans,
			       struct btrfs_root *root,
			       struct btrfs_key *ins_key,
			       struct btrfs_path *path, int data_size,
			       int extend)
3319
{
3320
	struct btrfs_disk_key disk_key;
3321
	struct extent_buffer *l;
3322
	u32 nritems;
3323 3324
	int mid;
	int slot;
3325
	struct extent_buffer *right;
3326
	int ret = 0;
C
Chris Mason 已提交
3327
	int wret;
3328
	int split;
3329
	int num_doubles = 0;
3330
	int tried_avoid_double = 0;
C
Chris Mason 已提交
3331

3332 3333 3334 3335 3336 3337
	l = path->nodes[0];
	slot = path->slots[0];
	if (extend && data_size + btrfs_item_size_nr(l, slot) +
	    sizeof(struct btrfs_item) > BTRFS_LEAF_DATA_SIZE(root))
		return -EOVERFLOW;

C
Chris Mason 已提交
3338
	/* first try to make some room by pushing left and right */
3339 3340 3341
	if (data_size) {
		wret = push_leaf_right(trans, root, path, data_size,
				       data_size, 0, 0);
C
Chris Mason 已提交
3342
		if (wret < 0)
C
Chris Mason 已提交
3343
			return wret;
3344
		if (wret) {
3345 3346
			wret = push_leaf_left(trans, root, path, data_size,
					      data_size, 0, (u32)-1);
3347 3348 3349 3350
			if (wret < 0)
				return wret;
		}
		l = path->nodes[0];
C
Chris Mason 已提交
3351

3352
		/* did the pushes work? */
3353
		if (btrfs_leaf_free_space(root, l) >= data_size)
3354
			return 0;
3355
	}
C
Chris Mason 已提交
3356

C
Chris Mason 已提交
3357
	if (!path->nodes[1]) {
3358
		ret = insert_new_root(trans, root, path, 1);
C
Chris Mason 已提交
3359 3360 3361
		if (ret)
			return ret;
	}
3362
again:
3363
	split = 1;
3364
	l = path->nodes[0];
3365
	slot = path->slots[0];
3366
	nritems = btrfs_header_nritems(l);
C
Chris Mason 已提交
3367
	mid = (nritems + 1) / 2;
3368

3369 3370 3371 3372 3373 3374 3375 3376 3377 3378 3379
	if (mid <= slot) {
		if (nritems == 1 ||
		    leaf_space_used(l, mid, nritems - mid) + data_size >
			BTRFS_LEAF_DATA_SIZE(root)) {
			if (slot >= nritems) {
				split = 0;
			} else {
				mid = slot;
				if (mid != nritems &&
				    leaf_space_used(l, mid, nritems - mid) +
				    data_size > BTRFS_LEAF_DATA_SIZE(root)) {
3380 3381
					if (data_size && !tried_avoid_double)
						goto push_for_double;
3382 3383 3384 3385 3386 3387 3388 3389 3390 3391 3392 3393 3394 3395 3396 3397
					split = 2;
				}
			}
		}
	} else {
		if (leaf_space_used(l, 0, mid) + data_size >
			BTRFS_LEAF_DATA_SIZE(root)) {
			if (!extend && data_size && slot == 0) {
				split = 0;
			} else if ((extend || !data_size) && slot == 0) {
				mid = 1;
			} else {
				mid = slot;
				if (mid != nritems &&
				    leaf_space_used(l, mid, nritems - mid) +
				    data_size > BTRFS_LEAF_DATA_SIZE(root)) {
3398 3399
					if (data_size && !tried_avoid_double)
						goto push_for_double;
3400 3401 3402 3403 3404 3405 3406 3407 3408 3409 3410 3411
					split = 2 ;
				}
			}
		}
	}

	if (split == 0)
		btrfs_cpu_key_to_disk(&disk_key, ins_key);
	else
		btrfs_item_key(l, &disk_key, mid);

	right = btrfs_alloc_free_block(trans, root, root->leafsize, 0,
Z
Zheng Yan 已提交
3412
					root->root_key.objectid,
3413
					&disk_key, 0, l->start, 0);
3414
	if (IS_ERR(right))
3415
		return PTR_ERR(right);
3416 3417

	root_add_used(root, root->leafsize);
3418 3419

	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
3420
	btrfs_set_header_bytenr(right, right->start);
3421
	btrfs_set_header_generation(right, trans->transid);
3422
	btrfs_set_header_backref_rev(right, BTRFS_MIXED_BACKREF_REV);
3423 3424 3425 3426 3427
	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);
3428 3429 3430 3431

	write_extent_buffer(right, root->fs_info->chunk_tree_uuid,
			    (unsigned long)btrfs_header_chunk_tree_uuid(right),
			    BTRFS_UUID_SIZE);
3432

3433 3434 3435
	if (split == 0) {
		if (mid <= slot) {
			btrfs_set_header_nritems(right, 0);
3436 3437
			insert_ptr(trans, root, path, &disk_key, right->start,
				   path->slots[1] + 1, 1);
3438 3439 3440 3441 3442 3443 3444
			btrfs_tree_unlock(path->nodes[0]);
			free_extent_buffer(path->nodes[0]);
			path->nodes[0] = right;
			path->slots[0] = 0;
			path->slots[1] += 1;
		} else {
			btrfs_set_header_nritems(right, 0);
3445
			insert_ptr(trans, root, path, &disk_key, right->start,
3446 3447 3448 3449 3450
					  path->slots[1], 1);
			btrfs_tree_unlock(path->nodes[0]);
			free_extent_buffer(path->nodes[0]);
			path->nodes[0] = right;
			path->slots[0] = 0;
3451 3452 3453
			if (path->slots[1] == 0)
				fixup_low_keys(trans, root, path,
					       &disk_key, 1);
3454
		}
3455 3456
		btrfs_mark_buffer_dirty(right);
		return ret;
3457
	}
C
Chris Mason 已提交
3458

3459
	copy_for_split(trans, root, path, l, right, slot, mid, nritems);
Z
Zheng Yan 已提交
3460

3461
	if (split == 2) {
3462 3463 3464
		BUG_ON(num_doubles != 0);
		num_doubles++;
		goto again;
3465
	}
3466

3467
	return 0;
3468 3469 3470 3471 3472 3473 3474

push_for_double:
	push_for_double_split(trans, root, path, data_size);
	tried_avoid_double = 1;
	if (btrfs_leaf_free_space(root, path->nodes[0]) >= data_size)
		return 0;
	goto again;
3475 3476
}

Y
Yan, Zheng 已提交
3477 3478 3479
static noinline int setup_leaf_for_split(struct btrfs_trans_handle *trans,
					 struct btrfs_root *root,
					 struct btrfs_path *path, int ins_len)
3480
{
Y
Yan, Zheng 已提交
3481
	struct btrfs_key key;
3482
	struct extent_buffer *leaf;
Y
Yan, Zheng 已提交
3483 3484 3485 3486
	struct btrfs_file_extent_item *fi;
	u64 extent_len = 0;
	u32 item_size;
	int ret;
3487 3488

	leaf = path->nodes[0];
Y
Yan, Zheng 已提交
3489 3490 3491 3492 3493 3494 3495
	btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);

	BUG_ON(key.type != BTRFS_EXTENT_DATA_KEY &&
	       key.type != BTRFS_EXTENT_CSUM_KEY);

	if (btrfs_leaf_free_space(root, leaf) >= ins_len)
		return 0;
3496 3497

	item_size = btrfs_item_size_nr(leaf, path->slots[0]);
Y
Yan, Zheng 已提交
3498 3499 3500 3501 3502
	if (key.type == BTRFS_EXTENT_DATA_KEY) {
		fi = btrfs_item_ptr(leaf, path->slots[0],
				    struct btrfs_file_extent_item);
		extent_len = btrfs_file_extent_num_bytes(leaf, fi);
	}
3503
	btrfs_release_path(path);
3504 3505

	path->keep_locks = 1;
Y
Yan, Zheng 已提交
3506 3507
	path->search_for_split = 1;
	ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
3508
	path->search_for_split = 0;
Y
Yan, Zheng 已提交
3509 3510
	if (ret < 0)
		goto err;
3511

Y
Yan, Zheng 已提交
3512 3513
	ret = -EAGAIN;
	leaf = path->nodes[0];
3514
	/* if our item isn't there or got smaller, return now */
Y
Yan, Zheng 已提交
3515 3516 3517
	if (ret > 0 || item_size != btrfs_item_size_nr(leaf, path->slots[0]))
		goto err;

3518 3519 3520 3521
	/* the leaf has  changed, it now has room.  return now */
	if (btrfs_leaf_free_space(root, path->nodes[0]) >= ins_len)
		goto err;

Y
Yan, Zheng 已提交
3522 3523 3524 3525 3526
	if (key.type == BTRFS_EXTENT_DATA_KEY) {
		fi = btrfs_item_ptr(leaf, path->slots[0],
				    struct btrfs_file_extent_item);
		if (extent_len != btrfs_file_extent_num_bytes(leaf, fi))
			goto err;
3527 3528
	}

3529
	btrfs_set_path_blocking(path);
Y
Yan, Zheng 已提交
3530
	ret = split_leaf(trans, root, &key, path, ins_len, 1);
3531 3532
	if (ret)
		goto err;
3533

Y
Yan, Zheng 已提交
3534
	path->keep_locks = 0;
3535
	btrfs_unlock_up_safe(path, 1);
Y
Yan, Zheng 已提交
3536 3537 3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555 3556 3557
	return 0;
err:
	path->keep_locks = 0;
	return ret;
}

static noinline int split_item(struct btrfs_trans_handle *trans,
			       struct btrfs_root *root,
			       struct btrfs_path *path,
			       struct btrfs_key *new_key,
			       unsigned long split_offset)
{
	struct extent_buffer *leaf;
	struct btrfs_item *item;
	struct btrfs_item *new_item;
	int slot;
	char *buf;
	u32 nritems;
	u32 item_size;
	u32 orig_offset;
	struct btrfs_disk_key disk_key;

3558 3559 3560
	leaf = path->nodes[0];
	BUG_ON(btrfs_leaf_free_space(root, leaf) < sizeof(struct btrfs_item));

3561 3562
	btrfs_set_path_blocking(path);

3563 3564 3565 3566 3567
	item = btrfs_item_nr(leaf, path->slots[0]);
	orig_offset = btrfs_item_offset(leaf, item);
	item_size = btrfs_item_size(leaf, item);

	buf = kmalloc(item_size, GFP_NOFS);
Y
Yan, Zheng 已提交
3568 3569 3570
	if (!buf)
		return -ENOMEM;

3571 3572 3573
	read_extent_buffer(leaf, buf, btrfs_item_ptr_offset(leaf,
			    path->slots[0]), item_size);

Y
Yan, Zheng 已提交
3574
	slot = path->slots[0] + 1;
3575 3576 3577 3578
	nritems = btrfs_header_nritems(leaf);
	if (slot != nritems) {
		/* shift the items */
		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + 1),
Y
Yan, Zheng 已提交
3579 3580
				btrfs_item_nr_offset(slot),
				(nritems - slot) * sizeof(struct btrfs_item));
3581 3582 3583 3584 3585 3586 3587 3588 3589 3590 3591 3592 3593 3594 3595 3596 3597 3598 3599 3600 3601 3602 3603 3604 3605 3606 3607
	}

	btrfs_cpu_key_to_disk(&disk_key, new_key);
	btrfs_set_item_key(leaf, &disk_key, slot);

	new_item = btrfs_item_nr(leaf, slot);

	btrfs_set_item_offset(leaf, new_item, orig_offset);
	btrfs_set_item_size(leaf, new_item, item_size - split_offset);

	btrfs_set_item_offset(leaf, item,
			      orig_offset + item_size - split_offset);
	btrfs_set_item_size(leaf, item, split_offset);

	btrfs_set_header_nritems(leaf, nritems + 1);

	/* write the data for the start of the original item */
	write_extent_buffer(leaf, buf,
			    btrfs_item_ptr_offset(leaf, path->slots[0]),
			    split_offset);

	/* write the data for the new item */
	write_extent_buffer(leaf, buf + split_offset,
			    btrfs_item_ptr_offset(leaf, slot),
			    item_size - split_offset);
	btrfs_mark_buffer_dirty(leaf);

Y
Yan, Zheng 已提交
3608
	BUG_ON(btrfs_leaf_free_space(root, leaf) < 0);
3609
	kfree(buf);
Y
Yan, Zheng 已提交
3610 3611 3612 3613 3614 3615 3616 3617 3618 3619 3620 3621 3622 3623 3624 3625 3626 3627 3628 3629 3630 3631 3632 3633 3634 3635 3636 3637 3638 3639 3640
	return 0;
}

/*
 * This function splits a single item into two items,
 * giving 'new_key' to the new item and splitting the
 * old one at split_offset (from the start of the item).
 *
 * The path may be released by this operation.  After
 * the split, the path is pointing to the old item.  The
 * new item is going to be in the same node as the old one.
 *
 * Note, the item being split must be smaller enough to live alone on
 * a tree block with room for one extra struct btrfs_item
 *
 * This allows us to split the item in place, keeping a lock on the
 * leaf the entire time.
 */
int btrfs_split_item(struct btrfs_trans_handle *trans,
		     struct btrfs_root *root,
		     struct btrfs_path *path,
		     struct btrfs_key *new_key,
		     unsigned long split_offset)
{
	int ret;
	ret = setup_leaf_for_split(trans, root, path,
				   sizeof(struct btrfs_item));
	if (ret)
		return ret;

	ret = split_item(trans, root, path, new_key, split_offset);
3641 3642 3643
	return ret;
}

Y
Yan, Zheng 已提交
3644 3645 3646 3647 3648 3649 3650 3651 3652 3653 3654 3655 3656 3657 3658 3659 3660 3661 3662 3663 3664 3665 3666 3667 3668
/*
 * This function duplicate a item, giving 'new_key' to the new item.
 * It guarantees both items live in the same tree leaf and the new item
 * is contiguous with the original item.
 *
 * This allows us to split file extent in place, keeping a lock on the
 * leaf the entire time.
 */
int btrfs_duplicate_item(struct btrfs_trans_handle *trans,
			 struct btrfs_root *root,
			 struct btrfs_path *path,
			 struct btrfs_key *new_key)
{
	struct extent_buffer *leaf;
	int ret;
	u32 item_size;

	leaf = path->nodes[0];
	item_size = btrfs_item_size_nr(leaf, path->slots[0]);
	ret = setup_leaf_for_split(trans, root, path,
				   item_size + sizeof(struct btrfs_item));
	if (ret)
		return ret;

	path->slots[0]++;
3669 3670 3671
	setup_items_for_insert(trans, root, path, new_key, &item_size,
			       item_size, item_size +
			       sizeof(struct btrfs_item), 1);
Y
Yan, Zheng 已提交
3672 3673 3674 3675 3676 3677 3678 3679
	leaf = path->nodes[0];
	memcpy_extent_buffer(leaf,
			     btrfs_item_ptr_offset(leaf, path->slots[0]),
			     btrfs_item_ptr_offset(leaf, path->slots[0] - 1),
			     item_size);
	return 0;
}

C
Chris Mason 已提交
3680 3681 3682 3683 3684 3685
/*
 * make the item pointed to by the path smaller.  new_size indicates
 * how small to make it, and from_end tells us if we just chop bytes
 * off the end of the item or if we shift the item to chop bytes off
 * the front.
 */
3686 3687 3688 3689
void btrfs_truncate_item(struct btrfs_trans_handle *trans,
			 struct btrfs_root *root,
			 struct btrfs_path *path,
			 u32 new_size, int from_end)
C
Chris Mason 已提交
3690 3691
{
	int slot;
3692 3693
	struct extent_buffer *leaf;
	struct btrfs_item *item;
C
Chris Mason 已提交
3694 3695 3696 3697 3698 3699
	u32 nritems;
	unsigned int data_end;
	unsigned int old_data_start;
	unsigned int old_size;
	unsigned int size_diff;
	int i;
3700 3701 3702
	struct btrfs_map_token token;

	btrfs_init_map_token(&token);
C
Chris Mason 已提交
3703

3704
	leaf = path->nodes[0];
3705 3706 3707 3708
	slot = path->slots[0];

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

3711
	nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
3712 3713
	data_end = leaf_data_end(root, leaf);

3714
	old_data_start = btrfs_item_offset_nr(leaf, slot);
3715

C
Chris Mason 已提交
3716 3717 3718 3719 3720 3721 3722 3723 3724 3725
	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++) {
3726 3727
		u32 ioff;
		item = btrfs_item_nr(leaf, i);
3728

3729 3730 3731
		ioff = btrfs_token_item_offset(leaf, item, &token);
		btrfs_set_token_item_offset(leaf, item,
					    ioff + size_diff, &token);
C
Chris Mason 已提交
3732
	}
3733

C
Chris Mason 已提交
3734
	/* shift the data */
3735 3736 3737 3738 3739 3740 3741 3742 3743 3744 3745 3746 3747 3748 3749 3750 3751 3752 3753 3754 3755 3756 3757
	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,
C
Chris Mason 已提交
3758 3759
				      (unsigned long)fi,
				      offsetof(struct btrfs_file_extent_item,
3760 3761 3762 3763 3764 3765 3766 3767 3768 3769 3770 3771 3772 3773
						 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);
	}
3774 3775 3776 3777

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

3779 3780
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
C
Chris Mason 已提交
3781
		BUG();
3782
	}
C
Chris Mason 已提交
3783 3784
}

C
Chris Mason 已提交
3785 3786 3787
/*
 * make the item pointed to by the path bigger, data_size is the new size.
 */
3788 3789 3790
void btrfs_extend_item(struct btrfs_trans_handle *trans,
		       struct btrfs_root *root, struct btrfs_path *path,
		       u32 data_size)
3791 3792
{
	int slot;
3793 3794
	struct extent_buffer *leaf;
	struct btrfs_item *item;
3795 3796 3797 3798 3799
	u32 nritems;
	unsigned int data_end;
	unsigned int old_data;
	unsigned int old_size;
	int i;
3800 3801 3802
	struct btrfs_map_token token;

	btrfs_init_map_token(&token);
3803

3804
	leaf = path->nodes[0];
3805

3806
	nritems = btrfs_header_nritems(leaf);
3807 3808
	data_end = leaf_data_end(root, leaf);

3809 3810
	if (btrfs_leaf_free_space(root, leaf) < data_size) {
		btrfs_print_leaf(root, leaf);
3811
		BUG();
3812
	}
3813
	slot = path->slots[0];
3814
	old_data = btrfs_item_end_nr(leaf, slot);
3815 3816

	BUG_ON(slot < 0);
3817 3818
	if (slot >= nritems) {
		btrfs_print_leaf(root, leaf);
C
Chris Mason 已提交
3819 3820
		printk(KERN_CRIT "slot %d too large, nritems %d\n",
		       slot, nritems);
3821 3822
		BUG_ON(1);
	}
3823 3824 3825 3826 3827 3828

	/*
	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
	 */
	/* first correct the data pointers */
	for (i = slot; i < nritems; i++) {
3829 3830
		u32 ioff;
		item = btrfs_item_nr(leaf, i);
3831

3832 3833 3834
		ioff = btrfs_token_item_offset(leaf, item, &token);
		btrfs_set_token_item_offset(leaf, item,
					    ioff - data_size, &token);
3835
	}
3836

3837
	/* shift the data */
3838
	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
3839 3840
		      data_end - data_size, btrfs_leaf_data(leaf) +
		      data_end, old_data - data_end);
3841

3842
	data_end = old_data;
3843 3844 3845 3846
	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);
3847

3848 3849
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
3850
		BUG();
3851
	}
3852 3853
}

3854 3855 3856 3857 3858 3859 3860 3861 3862 3863 3864 3865 3866 3867 3868 3869 3870 3871 3872 3873 3874 3875
/*
 * Given a key and some data, insert items into the tree.
 * This does all the path init required, making room in the tree if needed.
 * Returns the number of keys that were inserted.
 */
int btrfs_insert_some_items(struct btrfs_trans_handle *trans,
			    struct btrfs_root *root,
			    struct btrfs_path *path,
			    struct btrfs_key *cpu_key, u32 *data_size,
			    int nr)
{
	struct extent_buffer *leaf;
	struct btrfs_item *item;
	int ret = 0;
	int slot;
	int i;
	u32 nritems;
	u32 total_data = 0;
	u32 total_size = 0;
	unsigned int data_end;
	struct btrfs_disk_key disk_key;
	struct btrfs_key found_key;
3876 3877 3878
	struct btrfs_map_token token;

	btrfs_init_map_token(&token);
3879

3880 3881 3882 3883 3884 3885
	for (i = 0; i < nr; i++) {
		if (total_size + data_size[i] + sizeof(struct btrfs_item) >
		    BTRFS_LEAF_DATA_SIZE(root)) {
			break;
			nr = i;
		}
3886
		total_data += data_size[i];
3887 3888 3889
		total_size += data_size[i] + sizeof(struct btrfs_item);
	}
	BUG_ON(nr == 0);
3890 3891 3892 3893 3894 3895 3896 3897 3898 3899 3900 3901 3902 3903 3904 3905 3906 3907 3908 3909 3910 3911 3912 3913 3914 3915 3916 3917 3918 3919 3920 3921 3922 3923

	ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1);
	if (ret == 0)
		return -EEXIST;
	if (ret < 0)
		goto out;

	leaf = path->nodes[0];

	nritems = btrfs_header_nritems(leaf);
	data_end = leaf_data_end(root, leaf);

	if (btrfs_leaf_free_space(root, leaf) < total_size) {
		for (i = nr; i >= 0; i--) {
			total_data -= data_size[i];
			total_size -= data_size[i] + sizeof(struct btrfs_item);
			if (total_size < btrfs_leaf_free_space(root, leaf))
				break;
		}
		nr = i;
	}

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

	if (slot != nritems) {
		unsigned int old_data = btrfs_item_end_nr(leaf, slot);

		item = btrfs_item_nr(leaf, slot);
		btrfs_item_key_to_cpu(leaf, &found_key, slot);

		/* figure out how many keys we can insert in here */
		total_data = data_size[0];
		for (i = 1; i < nr; i++) {
3924
			if (btrfs_comp_cpu_keys(&found_key, cpu_key + i) <= 0)
3925 3926 3927 3928 3929 3930 3931
				break;
			total_data += data_size[i];
		}
		nr = i;

		if (old_data < data_end) {
			btrfs_print_leaf(root, leaf);
C
Chris Mason 已提交
3932
			printk(KERN_CRIT "slot %d old_data %d data_end %d\n",
3933 3934 3935 3936 3937 3938 3939 3940 3941 3942 3943
			       slot, old_data, data_end);
			BUG_ON(1);
		}
		/*
		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
		 */
		/* first correct the data pointers */
		for (i = slot; i < nritems; i++) {
			u32 ioff;

			item = btrfs_item_nr(leaf, i);
3944 3945 3946
			ioff = btrfs_token_item_offset(leaf, item, &token);
			btrfs_set_token_item_offset(leaf, item,
						    ioff - total_data, &token);
3947 3948 3949 3950 3951 3952 3953 3954 3955 3956 3957 3958 3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970 3971 3972
		}
		/* shift the items */
		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr),
			      btrfs_item_nr_offset(slot),
			      (nritems - slot) * sizeof(struct btrfs_item));

		/* shift the data */
		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
			      data_end - total_data, btrfs_leaf_data(leaf) +
			      data_end, old_data - data_end);
		data_end = old_data;
	} else {
		/*
		 * this sucks but it has to be done, if we are inserting at
		 * the end of the leaf only insert 1 of the items, since we
		 * have no way of knowing whats on the next leaf and we'd have
		 * to drop our current locks to figure it out
		 */
		nr = 1;
	}

	/* setup the item for the new data */
	for (i = 0; i < nr; i++) {
		btrfs_cpu_key_to_disk(&disk_key, cpu_key + i);
		btrfs_set_item_key(leaf, &disk_key, slot + i);
		item = btrfs_item_nr(leaf, slot + i);
3973 3974
		btrfs_set_token_item_offset(leaf, item,
					    data_end - data_size[i], &token);
3975
		data_end -= data_size[i];
3976
		btrfs_set_token_item_size(leaf, item, data_size[i], &token);
3977 3978 3979 3980 3981 3982 3983
	}
	btrfs_set_header_nritems(leaf, nritems + nr);
	btrfs_mark_buffer_dirty(leaf);

	ret = 0;
	if (slot == 0) {
		btrfs_cpu_key_to_disk(&disk_key, cpu_key);
3984
		fixup_low_keys(trans, root, path, &disk_key, 1);
3985 3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996
	}

	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
		BUG();
	}
out:
	if (!ret)
		ret = nr;
	return ret;
}

C
Chris Mason 已提交
3997
/*
3998 3999 4000
 * this is a helper for btrfs_insert_empty_items, the main goal here is
 * to save stack depth by doing the bulk of the work in a function
 * that doesn't call btrfs_search_slot
C
Chris Mason 已提交
4001
 */
4002 4003 4004 4005
void setup_items_for_insert(struct btrfs_trans_handle *trans,
			    struct btrfs_root *root, struct btrfs_path *path,
			    struct btrfs_key *cpu_key, u32 *data_size,
			    u32 total_data, u32 total_size, int nr)
4006
{
4007
	struct btrfs_item *item;
4008
	int i;
4009
	u32 nritems;
4010
	unsigned int data_end;
C
Chris Mason 已提交
4011
	struct btrfs_disk_key disk_key;
4012 4013
	struct extent_buffer *leaf;
	int slot;
4014 4015 4016
	struct btrfs_map_token token;

	btrfs_init_map_token(&token);
C
Chris Mason 已提交
4017

4018
	leaf = path->nodes[0];
4019
	slot = path->slots[0];
C
Chris Mason 已提交
4020

4021
	nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
4022
	data_end = leaf_data_end(root, leaf);
4023

4024
	if (btrfs_leaf_free_space(root, leaf) < total_size) {
4025
		btrfs_print_leaf(root, leaf);
C
Chris Mason 已提交
4026
		printk(KERN_CRIT "not enough freespace need %u have %d\n",
4027
		       total_size, btrfs_leaf_free_space(root, leaf));
4028
		BUG();
4029
	}
4030

4031
	if (slot != nritems) {
4032
		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
4033

4034 4035
		if (old_data < data_end) {
			btrfs_print_leaf(root, leaf);
C
Chris Mason 已提交
4036
			printk(KERN_CRIT "slot %d old_data %d data_end %d\n",
4037 4038 4039
			       slot, old_data, data_end);
			BUG_ON(1);
		}
4040 4041 4042 4043
		/*
		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
		 */
		/* first correct the data pointers */
C
Chris Mason 已提交
4044
		for (i = slot; i < nritems; i++) {
4045
			u32 ioff;
4046

4047
			item = btrfs_item_nr(leaf, i);
4048 4049 4050
			ioff = btrfs_token_item_offset(leaf, item, &token);
			btrfs_set_token_item_offset(leaf, item,
						    ioff - total_data, &token);
C
Chris Mason 已提交
4051
		}
4052
		/* shift the items */
4053
		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr),
4054
			      btrfs_item_nr_offset(slot),
C
Chris Mason 已提交
4055
			      (nritems - slot) * sizeof(struct btrfs_item));
4056 4057

		/* shift the data */
4058
		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
4059
			      data_end - total_data, btrfs_leaf_data(leaf) +
C
Chris Mason 已提交
4060
			      data_end, old_data - data_end);
4061 4062
		data_end = old_data;
	}
4063

4064
	/* setup the item for the new data */
4065 4066 4067 4068
	for (i = 0; i < nr; i++) {
		btrfs_cpu_key_to_disk(&disk_key, cpu_key + i);
		btrfs_set_item_key(leaf, &disk_key, slot + i);
		item = btrfs_item_nr(leaf, slot + i);
4069 4070
		btrfs_set_token_item_offset(leaf, item,
					    data_end - data_size[i], &token);
4071
		data_end -= data_size[i];
4072
		btrfs_set_token_item_size(leaf, item, data_size[i], &token);
4073
	}
4074

4075
	btrfs_set_header_nritems(leaf, nritems + nr);
C
Chris Mason 已提交
4076

4077 4078
	if (slot == 0) {
		btrfs_cpu_key_to_disk(&disk_key, cpu_key);
4079
		fixup_low_keys(trans, root, path, &disk_key, 1);
4080
	}
4081 4082
	btrfs_unlock_up_safe(path, 1);
	btrfs_mark_buffer_dirty(leaf);
C
Chris Mason 已提交
4083

4084 4085
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
4086
		BUG();
4087
	}
4088 4089 4090 4091 4092 4093 4094 4095 4096 4097 4098 4099 4100 4101 4102 4103 4104 4105 4106 4107 4108 4109 4110 4111 4112 4113
}

/*
 * Given a key and some data, insert items into the tree.
 * This does all the path init required, making room in the tree if needed.
 */
int btrfs_insert_empty_items(struct btrfs_trans_handle *trans,
			    struct btrfs_root *root,
			    struct btrfs_path *path,
			    struct btrfs_key *cpu_key, u32 *data_size,
			    int nr)
{
	int ret = 0;
	int slot;
	int i;
	u32 total_size = 0;
	u32 total_data = 0;

	for (i = 0; i < nr; i++)
		total_data += data_size[i];

	total_size = total_data + (nr * sizeof(struct btrfs_item));
	ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1);
	if (ret == 0)
		return -EEXIST;
	if (ret < 0)
4114
		return ret;
4115 4116 4117 4118

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

4119
	setup_items_for_insert(trans, root, path, cpu_key, data_size,
4120
			       total_data, total_size, nr);
4121
	return 0;
4122 4123 4124 4125 4126 4127
}

/*
 * 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.
 */
4128 4129 4130
int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_key *cpu_key, void *data, u32
		      data_size)
4131 4132
{
	int ret = 0;
C
Chris Mason 已提交
4133
	struct btrfs_path *path;
4134 4135
	struct extent_buffer *leaf;
	unsigned long ptr;
4136

C
Chris Mason 已提交
4137
	path = btrfs_alloc_path();
T
Tsutomu Itoh 已提交
4138 4139
	if (!path)
		return -ENOMEM;
C
Chris Mason 已提交
4140
	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
4141
	if (!ret) {
4142 4143 4144 4145
		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);
4146
	}
C
Chris Mason 已提交
4147
	btrfs_free_path(path);
C
Chris Mason 已提交
4148
	return ret;
4149 4150
}

C
Chris Mason 已提交
4151
/*
C
Chris Mason 已提交
4152
 * delete the pointer from a given node.
C
Chris Mason 已提交
4153
 *
C
Chris Mason 已提交
4154 4155
 * the tree should have been previously balanced so the deletion does not
 * empty a node.
C
Chris Mason 已提交
4156
 */
4157 4158
static void del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
		    struct btrfs_path *path, int level, int slot)
4159
{
4160
	struct extent_buffer *parent = path->nodes[level];
4161
	u32 nritems;
4162

4163
	nritems = btrfs_header_nritems(parent);
C
Chris Mason 已提交
4164
	if (slot != nritems - 1) {
4165 4166 4167
		memmove_extent_buffer(parent,
			      btrfs_node_key_ptr_offset(slot),
			      btrfs_node_key_ptr_offset(slot + 1),
C
Chris Mason 已提交
4168 4169
			      sizeof(struct btrfs_key_ptr) *
			      (nritems - slot - 1));
4170
	}
4171
	nritems--;
4172
	btrfs_set_header_nritems(parent, nritems);
4173
	if (nritems == 0 && parent == root->node) {
4174
		BUG_ON(btrfs_header_level(root->node) != 1);
4175
		/* just turn the root into a leaf and break */
4176
		btrfs_set_header_level(root->node, 0);
4177
	} else if (slot == 0) {
4178 4179 4180
		struct btrfs_disk_key disk_key;

		btrfs_node_key(parent, &disk_key, 0);
4181
		fixup_low_keys(trans, root, path, &disk_key, level + 1);
4182
	}
C
Chris Mason 已提交
4183
	btrfs_mark_buffer_dirty(parent);
4184 4185
}

4186 4187
/*
 * a helper function to delete the leaf pointed to by path->slots[1] and
4188
 * path->nodes[1].
4189 4190 4191 4192 4193 4194 4195
 *
 * This deletes the pointer in path->nodes[1] and frees the leaf
 * block extent.  zero is returned if it all worked out, < 0 otherwise.
 *
 * The path must have already been setup for deleting the leaf, including
 * all the proper balancing.  path->nodes[1] must be locked.
 */
4196 4197 4198 4199
static noinline void btrfs_del_leaf(struct btrfs_trans_handle *trans,
				    struct btrfs_root *root,
				    struct btrfs_path *path,
				    struct extent_buffer *leaf)
4200
{
4201
	WARN_ON(btrfs_header_generation(leaf) != trans->transid);
4202
	del_ptr(trans, root, path, 1, path->slots[1]);
4203

4204 4205 4206 4207 4208 4209
	/*
	 * btrfs_free_extent is expensive, we want to make sure we
	 * aren't holding any locks when we call it
	 */
	btrfs_unlock_up_safe(path, 0);

4210 4211
	root_sub_used(root, leaf->len);

4212
	extent_buffer_get(leaf);
4213
	btrfs_free_tree_block(trans, root, leaf, 0, 1);
4214
	free_extent_buffer_stale(leaf);
4215
}
C
Chris Mason 已提交
4216 4217 4218 4219
/*
 * delete the item at the leaf level in path.  If that empties
 * the leaf, remove it from the tree
 */
4220 4221
int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root,
		    struct btrfs_path *path, int slot, int nr)
4222
{
4223 4224
	struct extent_buffer *leaf;
	struct btrfs_item *item;
4225 4226
	int last_off;
	int dsize = 0;
C
Chris Mason 已提交
4227 4228
	int ret = 0;
	int wret;
4229
	int i;
4230
	u32 nritems;
4231 4232 4233
	struct btrfs_map_token token;

	btrfs_init_map_token(&token);
4234

4235
	leaf = path->nodes[0];
4236 4237 4238 4239 4240
	last_off = btrfs_item_offset_nr(leaf, slot + nr - 1);

	for (i = 0; i < nr; i++)
		dsize += btrfs_item_size_nr(leaf, slot + i);

4241
	nritems = btrfs_header_nritems(leaf);
4242

4243
	if (slot + nr != nritems) {
C
Chris Mason 已提交
4244
		int data_end = leaf_data_end(root, leaf);
4245 4246

		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
C
Chris Mason 已提交
4247 4248
			      data_end + dsize,
			      btrfs_leaf_data(leaf) + data_end,
4249
			      last_off - data_end);
4250

4251
		for (i = slot + nr; i < nritems; i++) {
4252
			u32 ioff;
4253

4254
			item = btrfs_item_nr(leaf, i);
4255 4256 4257
			ioff = btrfs_token_item_offset(leaf, item, &token);
			btrfs_set_token_item_offset(leaf, item,
						    ioff + dsize, &token);
C
Chris Mason 已提交
4258
		}
4259

4260
		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
4261
			      btrfs_item_nr_offset(slot + nr),
C
Chris Mason 已提交
4262
			      sizeof(struct btrfs_item) *
4263
			      (nritems - slot - nr));
4264
	}
4265 4266
	btrfs_set_header_nritems(leaf, nritems - nr);
	nritems -= nr;
4267

C
Chris Mason 已提交
4268
	/* delete the leaf if we've emptied it */
4269
	if (nritems == 0) {
4270 4271
		if (leaf == root->node) {
			btrfs_set_header_level(leaf, 0);
4272
		} else {
4273 4274
			btrfs_set_path_blocking(path);
			clean_tree_block(trans, root, leaf);
4275
			btrfs_del_leaf(trans, root, path, leaf);
4276
		}
4277
	} else {
4278
		int used = leaf_space_used(leaf, 0, nritems);
C
Chris Mason 已提交
4279
		if (slot == 0) {
4280 4281 4282
			struct btrfs_disk_key disk_key;

			btrfs_item_key(leaf, &disk_key, 0);
4283
			fixup_low_keys(trans, root, path, &disk_key, 1);
C
Chris Mason 已提交
4284 4285
		}

C
Chris Mason 已提交
4286
		/* delete the leaf if it is mostly empty */
4287
		if (used < BTRFS_LEAF_DATA_SIZE(root) / 3) {
4288 4289 4290 4291
			/* push_leaf_left fixes the path.
			 * make sure the path still points to our leaf
			 * for possible call to del_ptr below
			 */
4292
			slot = path->slots[1];
4293 4294
			extent_buffer_get(leaf);

4295
			btrfs_set_path_blocking(path);
4296 4297
			wret = push_leaf_left(trans, root, path, 1, 1,
					      1, (u32)-1);
4298
			if (wret < 0 && wret != -ENOSPC)
C
Chris Mason 已提交
4299
				ret = wret;
4300 4301 4302

			if (path->nodes[0] == leaf &&
			    btrfs_header_nritems(leaf)) {
4303 4304
				wret = push_leaf_right(trans, root, path, 1,
						       1, 1, 0);
4305
				if (wret < 0 && wret != -ENOSPC)
C
Chris Mason 已提交
4306 4307
					ret = wret;
			}
4308 4309

			if (btrfs_header_nritems(leaf) == 0) {
4310
				path->slots[1] = slot;
4311
				btrfs_del_leaf(trans, root, path, leaf);
4312
				free_extent_buffer(leaf);
4313
				ret = 0;
C
Chris Mason 已提交
4314
			} else {
4315 4316 4317 4318 4319 4320 4321
				/* if we're still in the path, make sure
				 * we're dirty.  Otherwise, one of the
				 * push_leaf functions must have already
				 * dirtied this buffer
				 */
				if (path->nodes[0] == leaf)
					btrfs_mark_buffer_dirty(leaf);
4322
				free_extent_buffer(leaf);
4323
			}
4324
		} else {
4325
			btrfs_mark_buffer_dirty(leaf);
4326 4327
		}
	}
C
Chris Mason 已提交
4328
	return ret;
4329 4330
}

4331
/*
4332
 * search the tree again to find a leaf with lesser keys
4333 4334
 * returns 0 if it found something or 1 if there are no lesser leaves.
 * returns < 0 on io errors.
C
Chris Mason 已提交
4335 4336 4337
 *
 * This may release the path, and so you may lose any locks held at the
 * time you call it.
4338 4339 4340
 */
int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path)
{
4341 4342 4343
	struct btrfs_key key;
	struct btrfs_disk_key found_key;
	int ret;
4344

4345
	btrfs_item_key_to_cpu(path->nodes[0], &key, 0);
4346

4347 4348 4349 4350 4351 4352 4353 4354
	if (key.offset > 0)
		key.offset--;
	else if (key.type > 0)
		key.type--;
	else if (key.objectid > 0)
		key.objectid--;
	else
		return 1;
4355

4356
	btrfs_release_path(path);
4357 4358 4359 4360 4361 4362 4363 4364
	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
	if (ret < 0)
		return ret;
	btrfs_item_key(path->nodes[0], &found_key, 0);
	ret = comp_keys(&found_key, &key);
	if (ret < 0)
		return 0;
	return 1;
4365 4366
}

4367 4368 4369
/*
 * A helper function to walk down the tree starting at min_key, and looking
 * for nodes or leaves that are either in cache or have a minimum
C
Chris Mason 已提交
4370
 * transaction id.  This is used by the btree defrag code, and tree logging
4371 4372 4373 4374 4375 4376 4377 4378 4379 4380 4381
 *
 * This does not cow, but it does stuff the starting key it finds back
 * into min_key, so you can call btrfs_search_slot with cow=1 on the
 * key and get a writable path.
 *
 * This does lock as it descends, and path->keep_locks should be set
 * to 1 by the caller.
 *
 * This honors path->lowest_level to prevent descent past a given level
 * of the tree.
 *
C
Chris Mason 已提交
4382 4383 4384 4385
 * min_trans indicates the oldest transaction that you are interested
 * in walking through.  Any nodes or leaves older than min_trans are
 * skipped over (without reading them).
 *
4386 4387 4388 4389
 * returns zero if something useful was found, < 0 on error and 1 if there
 * was nothing in the tree that matched the search criteria.
 */
int btrfs_search_forward(struct btrfs_root *root, struct btrfs_key *min_key,
4390
			 struct btrfs_key *max_key,
4391 4392 4393 4394 4395 4396
			 struct btrfs_path *path, int cache_only,
			 u64 min_trans)
{
	struct extent_buffer *cur;
	struct btrfs_key found_key;
	int slot;
4397
	int sret;
4398 4399 4400 4401
	u32 nritems;
	int level;
	int ret = 1;

4402
	WARN_ON(!path->keep_locks);
4403
again:
4404
	cur = btrfs_read_lock_root_node(root);
4405
	level = btrfs_header_level(cur);
4406
	WARN_ON(path->nodes[level]);
4407
	path->nodes[level] = cur;
4408
	path->locks[level] = BTRFS_READ_LOCK;
4409 4410 4411 4412 4413

	if (btrfs_header_generation(cur) < min_trans) {
		ret = 1;
		goto out;
	}
C
Chris Mason 已提交
4414
	while (1) {
4415 4416
		nritems = btrfs_header_nritems(cur);
		level = btrfs_header_level(cur);
4417
		sret = bin_search(cur, min_key, level, &slot);
4418

4419 4420
		/* at the lowest level, we're done, setup the path and exit */
		if (level == path->lowest_level) {
4421 4422
			if (slot >= nritems)
				goto find_next_key;
4423 4424 4425 4426 4427
			ret = 0;
			path->slots[level] = slot;
			btrfs_item_key_to_cpu(cur, &found_key, slot);
			goto out;
		}
4428 4429
		if (sret && slot > 0)
			slot--;
4430 4431 4432 4433 4434
		/*
		 * check this node pointer against the cache_only and
		 * min_trans parameters.  If it isn't in cache or is too
		 * old, skip to the next one.
		 */
C
Chris Mason 已提交
4435
		while (slot < nritems) {
4436 4437 4438
			u64 blockptr;
			u64 gen;
			struct extent_buffer *tmp;
4439 4440
			struct btrfs_disk_key disk_key;

4441 4442 4443 4444 4445 4446 4447 4448 4449
			blockptr = btrfs_node_blockptr(cur, slot);
			gen = btrfs_node_ptr_generation(cur, slot);
			if (gen < min_trans) {
				slot++;
				continue;
			}
			if (!cache_only)
				break;

4450 4451 4452 4453 4454 4455 4456 4457
			if (max_key) {
				btrfs_node_key(cur, &disk_key, slot);
				if (comp_keys(&disk_key, max_key) >= 0) {
					ret = 1;
					goto out;
				}
			}

4458 4459 4460
			tmp = btrfs_find_tree_block(root, blockptr,
					    btrfs_level_size(root, level - 1));

4461
			if (tmp && btrfs_buffer_uptodate(tmp, gen, 1) > 0) {
4462 4463 4464 4465 4466 4467 4468
				free_extent_buffer(tmp);
				break;
			}
			if (tmp)
				free_extent_buffer(tmp);
			slot++;
		}
4469
find_next_key:
4470 4471 4472 4473 4474
		/*
		 * we didn't find a candidate key in this node, walk forward
		 * and find another one
		 */
		if (slot >= nritems) {
4475
			path->slots[level] = slot;
4476
			btrfs_set_path_blocking(path);
4477
			sret = btrfs_find_next_key(root, path, min_key, level,
4478
						  cache_only, min_trans);
4479
			if (sret == 0) {
4480
				btrfs_release_path(path);
4481 4482 4483 4484 4485 4486 4487 4488 4489 4490
				goto again;
			} else {
				goto out;
			}
		}
		/* save our key for returning back */
		btrfs_node_key_to_cpu(cur, &found_key, slot);
		path->slots[level] = slot;
		if (level == path->lowest_level) {
			ret = 0;
4491
			unlock_up(path, level, 1, 0, NULL);
4492 4493
			goto out;
		}
4494
		btrfs_set_path_blocking(path);
4495
		cur = read_node_slot(root, cur, slot);
4496
		BUG_ON(!cur); /* -ENOMEM */
4497

4498
		btrfs_tree_read_lock(cur);
4499

4500
		path->locks[level - 1] = BTRFS_READ_LOCK;
4501
		path->nodes[level - 1] = cur;
4502
		unlock_up(path, level, 1, 0, NULL);
4503
		btrfs_clear_path_blocking(path, NULL, 0);
4504 4505 4506 4507
	}
out:
	if (ret == 0)
		memcpy(min_key, &found_key, sizeof(found_key));
4508
	btrfs_set_path_blocking(path);
4509 4510 4511 4512 4513 4514 4515 4516 4517 4518 4519 4520 4521 4522 4523
	return ret;
}

/*
 * this is similar to btrfs_next_leaf, but does not try to preserve
 * and fixup the path.  It looks for and returns the next key in the
 * tree based on the current path and the cache_only and min_trans
 * parameters.
 *
 * 0 is returned if another key is found, < 0 if there are any errors
 * and 1 is returned if there are no higher keys in the tree
 *
 * path->keep_locks should be set to 1 on the search made before
 * calling this function.
 */
4524
int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path,
4525
			struct btrfs_key *key, int level,
4526
			int cache_only, u64 min_trans)
4527 4528 4529 4530
{
	int slot;
	struct extent_buffer *c;

4531
	WARN_ON(!path->keep_locks);
C
Chris Mason 已提交
4532
	while (level < BTRFS_MAX_LEVEL) {
4533 4534 4535 4536 4537
		if (!path->nodes[level])
			return 1;

		slot = path->slots[level] + 1;
		c = path->nodes[level];
4538
next:
4539
		if (slot >= btrfs_header_nritems(c)) {
4540 4541 4542 4543 4544
			int ret;
			int orig_lowest;
			struct btrfs_key cur_key;
			if (level + 1 >= BTRFS_MAX_LEVEL ||
			    !path->nodes[level + 1])
4545
				return 1;
4546 4547 4548 4549 4550 4551 4552 4553 4554 4555 4556 4557 4558

			if (path->locks[level + 1]) {
				level++;
				continue;
			}

			slot = btrfs_header_nritems(c) - 1;
			if (level == 0)
				btrfs_item_key_to_cpu(c, &cur_key, slot);
			else
				btrfs_node_key_to_cpu(c, &cur_key, slot);

			orig_lowest = path->lowest_level;
4559
			btrfs_release_path(path);
4560 4561 4562 4563 4564 4565 4566 4567 4568 4569 4570 4571
			path->lowest_level = level;
			ret = btrfs_search_slot(NULL, root, &cur_key, path,
						0, 0);
			path->lowest_level = orig_lowest;
			if (ret < 0)
				return ret;

			c = path->nodes[level];
			slot = path->slots[level];
			if (ret == 0)
				slot++;
			goto next;
4572
		}
4573

4574 4575
		if (level == 0)
			btrfs_item_key_to_cpu(c, key, slot);
4576 4577 4578 4579 4580 4581 4582 4583
		else {
			u64 blockptr = btrfs_node_blockptr(c, slot);
			u64 gen = btrfs_node_ptr_generation(c, slot);

			if (cache_only) {
				struct extent_buffer *cur;
				cur = btrfs_find_tree_block(root, blockptr,
					    btrfs_level_size(root, level - 1));
4584 4585
				if (!cur ||
				    btrfs_buffer_uptodate(cur, gen, 1) <= 0) {
4586 4587 4588 4589 4590 4591 4592 4593 4594 4595 4596
					slot++;
					if (cur)
						free_extent_buffer(cur);
					goto next;
				}
				free_extent_buffer(cur);
			}
			if (gen < min_trans) {
				slot++;
				goto next;
			}
4597
			btrfs_node_key_to_cpu(c, key, slot);
4598
		}
4599 4600 4601 4602 4603
		return 0;
	}
	return 1;
}

C
Chris Mason 已提交
4604
/*
4605
 * search the tree again to find a leaf with greater keys
C
Chris Mason 已提交
4606 4607
 * returns 0 if it found something or 1 if there are no greater leaves.
 * returns < 0 on io errors.
C
Chris Mason 已提交
4608
 */
C
Chris Mason 已提交
4609
int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
4610 4611
{
	int slot;
4612
	int level;
4613
	struct extent_buffer *c;
4614
	struct extent_buffer *next;
4615 4616 4617
	struct btrfs_key key;
	u32 nritems;
	int ret;
4618
	int old_spinning = path->leave_spinning;
4619
	int next_rw_lock = 0;
4620 4621

	nritems = btrfs_header_nritems(path->nodes[0]);
C
Chris Mason 已提交
4622
	if (nritems == 0)
4623 4624
		return 1;

4625 4626 4627 4628
	btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1);
again:
	level = 1;
	next = NULL;
4629
	next_rw_lock = 0;
4630
	btrfs_release_path(path);
4631

4632
	path->keep_locks = 1;
4633
	path->leave_spinning = 1;
4634

4635 4636 4637 4638 4639 4640
	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
	path->keep_locks = 0;

	if (ret < 0)
		return ret;

4641
	nritems = btrfs_header_nritems(path->nodes[0]);
4642 4643 4644 4645 4646 4647
	/*
	 * by releasing the path above we dropped all our locks.  A balance
	 * could have added more items next to the key that used to be
	 * at the very end of the block.  So, check again here and
	 * advance the path if there are now more items available.
	 */
4648
	if (nritems > 0 && path->slots[0] < nritems - 1) {
4649 4650
		if (ret == 0)
			path->slots[0]++;
4651
		ret = 0;
4652 4653
		goto done;
	}
4654

C
Chris Mason 已提交
4655
	while (level < BTRFS_MAX_LEVEL) {
4656 4657 4658 4659
		if (!path->nodes[level]) {
			ret = 1;
			goto done;
		}
4660

4661 4662
		slot = path->slots[level] + 1;
		c = path->nodes[level];
4663
		if (slot >= btrfs_header_nritems(c)) {
4664
			level++;
4665 4666 4667 4668
			if (level == BTRFS_MAX_LEVEL) {
				ret = 1;
				goto done;
			}
4669 4670
			continue;
		}
4671

4672
		if (next) {
4673
			btrfs_tree_unlock_rw(next, next_rw_lock);
4674
			free_extent_buffer(next);
4675
		}
4676

4677
		next = c;
4678
		next_rw_lock = path->locks[level];
4679 4680 4681 4682
		ret = read_block_for_search(NULL, root, path, &next, level,
					    slot, &key);
		if (ret == -EAGAIN)
			goto again;
4683

4684
		if (ret < 0) {
4685
			btrfs_release_path(path);
4686 4687 4688
			goto done;
		}

4689
		if (!path->skip_locking) {
4690
			ret = btrfs_try_tree_read_lock(next);
4691 4692
			if (!ret) {
				btrfs_set_path_blocking(path);
4693
				btrfs_tree_read_lock(next);
4694
				btrfs_clear_path_blocking(path, next,
4695
							  BTRFS_READ_LOCK);
4696
			}
4697
			next_rw_lock = BTRFS_READ_LOCK;
4698
		}
4699 4700 4701
		break;
	}
	path->slots[level] = slot;
C
Chris Mason 已提交
4702
	while (1) {
4703 4704
		level--;
		c = path->nodes[level];
4705
		if (path->locks[level])
4706
			btrfs_tree_unlock_rw(c, path->locks[level]);
4707

4708
		free_extent_buffer(c);
4709 4710
		path->nodes[level] = next;
		path->slots[level] = 0;
4711
		if (!path->skip_locking)
4712
			path->locks[level] = next_rw_lock;
4713 4714
		if (!level)
			break;
4715

4716 4717 4718 4719 4720
		ret = read_block_for_search(NULL, root, path, &next, level,
					    0, &key);
		if (ret == -EAGAIN)
			goto again;

4721
		if (ret < 0) {
4722
			btrfs_release_path(path);
4723 4724 4725
			goto done;
		}

4726
		if (!path->skip_locking) {
4727
			ret = btrfs_try_tree_read_lock(next);
4728 4729
			if (!ret) {
				btrfs_set_path_blocking(path);
4730
				btrfs_tree_read_lock(next);
4731
				btrfs_clear_path_blocking(path, next,
4732 4733
							  BTRFS_READ_LOCK);
			}
4734
			next_rw_lock = BTRFS_READ_LOCK;
4735
		}
4736
	}
4737
	ret = 0;
4738
done:
4739
	unlock_up(path, 0, 1, 0, NULL);
4740 4741 4742 4743 4744
	path->leave_spinning = old_spinning;
	if (!old_spinning)
		btrfs_set_path_blocking(path);

	return ret;
4745
}
4746

4747 4748 4749 4750 4751 4752
/*
 * this uses btrfs_prev_leaf to walk backwards in the tree, and keeps
 * searching until it gets past min_objectid or finds an item of 'type'
 *
 * returns 0 if something is found, 1 if nothing was found and < 0 on error
 */
4753 4754 4755 4756 4757 4758
int btrfs_previous_item(struct btrfs_root *root,
			struct btrfs_path *path, u64 min_objectid,
			int type)
{
	struct btrfs_key found_key;
	struct extent_buffer *leaf;
4759
	u32 nritems;
4760 4761
	int ret;

C
Chris Mason 已提交
4762
	while (1) {
4763
		if (path->slots[0] == 0) {
4764
			btrfs_set_path_blocking(path);
4765 4766 4767 4768 4769 4770 4771
			ret = btrfs_prev_leaf(root, path);
			if (ret != 0)
				return ret;
		} else {
			path->slots[0]--;
		}
		leaf = path->nodes[0];
4772 4773 4774 4775 4776 4777
		nritems = btrfs_header_nritems(leaf);
		if (nritems == 0)
			return 1;
		if (path->slots[0] == nritems)
			path->slots[0]--;

4778
		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
4779 4780
		if (found_key.objectid < min_objectid)
			break;
4781 4782
		if (found_key.type == type)
			return 0;
4783 4784 4785
		if (found_key.objectid == min_objectid &&
		    found_key.type < type)
			break;
4786 4787 4788
	}
	return 1;
}