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

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

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

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

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

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

C
Chris Mason 已提交
63
void btrfs_release_path(struct btrfs_root *root, struct btrfs_path *p)
64 65
{
	int i;
66

C
Chris Mason 已提交
67
	for (i = 0; i < BTRFS_MAX_LEVEL; i++) {
68
		p->slots[i] = 0;
69
		if (!p->nodes[i])
70 71 72 73 74
			continue;
		if (p->locks[i]) {
			btrfs_tree_unlock(p->nodes[i]);
			p->locks[i] = 0;
		}
75
		free_extent_buffer(p->nodes[i]);
76
		p->nodes[i] = NULL;
77 78 79
	}
}

80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110
struct extent_buffer *btrfs_root_node(struct btrfs_root *root)
{
	struct extent_buffer *eb;
	spin_lock(&root->node_lock);
	eb = root->node;
	extent_buffer_get(eb);
	spin_unlock(&root->node_lock);
	return eb;
}

struct extent_buffer *btrfs_lock_root_node(struct btrfs_root *root)
{
	struct extent_buffer *eb;

	while(1) {
		eb = btrfs_root_node(root);
		btrfs_tree_lock(eb);

		spin_lock(&root->node_lock);
		if (eb == root->node) {
			spin_unlock(&root->node_lock);
			break;
		}
		spin_unlock(&root->node_lock);

		btrfs_tree_unlock(eb);
		free_extent_buffer(eb);
	}
	return eb;
}

111 112 113 114 115 116 117 118
static void add_root_to_dirty_list(struct btrfs_root *root)
{
	if (root->track_dirty && list_empty(&root->dirty_list)) {
		list_add(&root->dirty_list,
			 &root->fs_info->dirty_cowonly_roots);
	}
}

119 120 121 122 123 124 125 126 127 128
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;
	u32 nritems;
	int ret = 0;
	int level;
	struct btrfs_key first_key;
129
	struct btrfs_root *new_root;
130

131 132 133 134 135 136
	new_root = kmalloc(sizeof(*new_root), GFP_NOFS);
	if (!new_root)
		return -ENOMEM;

	memcpy(new_root, root, sizeof(*new_root));
	new_root->root_key.objectid = new_root_objectid;
137 138 139 140 141 142 143 144 145 146 147 148 149 150 151

	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);
	nritems = btrfs_header_nritems(buf);
	if (nritems) {
		if (level == 0)
			btrfs_item_key_to_cpu(buf, &first_key, 0);
		else
			btrfs_node_key_to_cpu(buf, &first_key, 0);
	} else {
		first_key.objectid = 0;
	}
152
	cow = btrfs_alloc_free_block(trans, new_root, buf->len,
153 154 155
				       new_root_objectid,
				       trans->transid, first_key.objectid,
				       level, buf->start, 0);
156 157
	if (IS_ERR(cow)) {
		kfree(new_root);
158
		return PTR_ERR(cow);
159
	}
160 161 162 163 164

	copy_extent_buffer(cow, buf, 0, 0, cow->len);
	btrfs_set_header_bytenr(cow, cow->start);
	btrfs_set_header_generation(cow, trans->transid);
	btrfs_set_header_owner(cow, new_root_objectid);
165
	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN);
166 167

	WARN_ON(btrfs_header_generation(buf) > trans->transid);
168 169 170
	ret = btrfs_inc_ref(trans, new_root, buf);
	kfree(new_root);

171 172 173 174 175 176 177 178 179
	if (ret)
		return ret;

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

int __btrfs_cow_block(struct btrfs_trans_handle *trans,
180 181 182 183 184
			     struct btrfs_root *root,
			     struct extent_buffer *buf,
			     struct extent_buffer *parent, int parent_slot,
			     struct extent_buffer **cow_ret,
			     u64 search_start, u64 empty_size)
C
Chris Mason 已提交
185
{
186
	u64 root_gen;
187
	struct extent_buffer *cow;
188
	u32 nritems;
189 190
	int ret = 0;
	int different_trans = 0;
191
	int level;
192
	int unlock_orig = 0;
193 194
	struct btrfs_key first_key;

195 196 197 198 199
	if (*cow_ret == buf)
		unlock_orig = 1;

	WARN_ON(!btrfs_tree_locked(buf));

200 201 202 203 204 205 206
	if (root->ref_cows) {
		root_gen = trans->transid;
	} else {
		root_gen = 0;
	}
	WARN_ON(root->ref_cows && trans->transid !=
		root->fs_info->running_transaction->transid);
207
	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
208

209 210 211 212 213 214 215 216 217 218
	level = btrfs_header_level(buf);
	nritems = btrfs_header_nritems(buf);
	if (nritems) {
		if (level == 0)
			btrfs_item_key_to_cpu(buf, &first_key, 0);
		else
			btrfs_node_key_to_cpu(buf, &first_key, 0);
	} else {
		first_key.objectid = 0;
	}
219
	cow = btrfs_alloc_free_block(trans, root, buf->len,
220 221
				     root->root_key.objectid,
				     root_gen, first_key.objectid, level,
222
				     search_start, empty_size);
223 224
	if (IS_ERR(cow))
		return PTR_ERR(cow);
225

226
	copy_extent_buffer(cow, buf, 0, 0, cow->len);
227
	btrfs_set_header_bytenr(cow, cow->start);
228 229
	btrfs_set_header_generation(cow, trans->transid);
	btrfs_set_header_owner(cow, root->root_key.objectid);
230
	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN);
231

232 233
	WARN_ON(btrfs_header_generation(buf) > trans->transid);
	if (btrfs_header_generation(buf) != trans->transid) {
234 235 236 237 238 239 240 241
		different_trans = 1;
		ret = btrfs_inc_ref(trans, root, buf);
		if (ret)
			return ret;
	} else {
		clean_tree_block(trans, root, buf);
	}

C
Chris Mason 已提交
242
	if (buf == root->node) {
243
		WARN_ON(parent && parent != buf);
244
		root_gen = btrfs_header_generation(buf);
245 246

		spin_lock(&root->node_lock);
C
Chris Mason 已提交
247
		root->node = cow;
248
		extent_buffer_get(cow);
249 250
		spin_unlock(&root->node_lock);

C
Chris Mason 已提交
251
		if (buf != root->commit_root) {
252
			btrfs_free_extent(trans, root, buf->start,
253 254
					  buf->len, root->root_key.objectid,
					  root_gen, 0, 0, 1);
C
Chris Mason 已提交
255
		}
256
		free_extent_buffer(buf);
257
		add_root_to_dirty_list(root);
C
Chris Mason 已提交
258
	} else {
259
		root_gen = btrfs_header_generation(parent);
260
		btrfs_set_node_blockptr(parent, parent_slot,
261
					cow->start);
262 263 264
		WARN_ON(trans->transid == 0);
		btrfs_set_node_ptr_generation(parent, parent_slot,
					      trans->transid);
C
Chris Mason 已提交
265
		btrfs_mark_buffer_dirty(parent);
266
		WARN_ON(btrfs_header_generation(parent) != trans->transid);
267 268 269
		btrfs_free_extent(trans, root, buf->start, buf->len,
				  btrfs_header_owner(parent), root_gen,
				  0, 0, 1);
C
Chris Mason 已提交
270
	}
271 272
	if (unlock_orig)
		btrfs_tree_unlock(buf);
273
	free_extent_buffer(buf);
C
Chris Mason 已提交
274
	btrfs_mark_buffer_dirty(cow);
C
Chris Mason 已提交
275
	*cow_ret = cow;
C
Chris Mason 已提交
276 277 278
	return 0;
}

279 280 281 282
int btrfs_cow_block(struct btrfs_trans_handle *trans,
		    struct btrfs_root *root, struct extent_buffer *buf,
		    struct extent_buffer *parent, int parent_slot,
		    struct extent_buffer **cow_ret)
283 284
{
	u64 search_start;
C
Chris Mason 已提交
285
	u64 header_trans;
286
	int ret;
C
Chris Mason 已提交
287

288 289 290 291 292 293 294 295 296 297
	if (trans->transaction != root->fs_info->running_transaction) {
		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
		       root->fs_info->running_transaction->transid);
		WARN_ON(1);
	}
	if (trans->transid != root->fs_info->generation) {
		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
		       root->fs_info->generation);
		WARN_ON(1);
	}
C
Chris Mason 已提交
298 299

	header_trans = btrfs_header_generation(buf);
300 301 302
	spin_lock(&root->fs_info->hash_lock);
	if (header_trans == trans->transid &&
	    !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN)) {
303
		*cow_ret = buf;
304
		spin_unlock(&root->fs_info->hash_lock);
305 306
		return 0;
	}
307
	spin_unlock(&root->fs_info->hash_lock);
308
	search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1);
309
	ret = __btrfs_cow_block(trans, root, buf, parent,
310
				 parent_slot, cow_ret, search_start, 0);
311
	return ret;
312 313
}

314
static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
315
{
316
	if (blocknr < other && other - (blocknr + blocksize) < 32768)
317
		return 1;
318
	if (blocknr > other && blocknr - (other + blocksize) < 32768)
319 320 321 322
		return 1;
	return 0;
}

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
/*
 * 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);

	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;
}


348
int btrfs_realloc_node(struct btrfs_trans_handle *trans,
349
		       struct btrfs_root *root, struct extent_buffer *parent,
350 351
		       int start_slot, int cache_only, u64 *last_ret,
		       struct btrfs_key *progress)
352
{
353
	struct extent_buffer *cur;
354
	u64 blocknr;
355
	u64 gen;
356 357
	u64 search_start = *last_ret;
	u64 last_block = 0;
358 359 360 361 362
	u64 other;
	u32 parent_nritems;
	int end_slot;
	int i;
	int err = 0;
363
	int parent_level;
364 365
	int uptodate;
	u32 blocksize;
366 367
	int progress_passed = 0;
	struct btrfs_disk_key disk_key;
368

369 370 371 372
	parent_level = btrfs_header_level(parent);
	if (cache_only && parent_level != 1)
		return 0;

373 374 375 376 377 378 379 380 381 382
	if (trans->transaction != root->fs_info->running_transaction) {
		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
		       root->fs_info->running_transaction->transid);
		WARN_ON(1);
	}
	if (trans->transid != root->fs_info->generation) {
		printk(KERN_CRIT "trans %Lu running %Lu\n", trans->transid,
		       root->fs_info->generation);
		WARN_ON(1);
	}
383

384 385
	parent_nritems = btrfs_header_nritems(parent);
	blocksize = btrfs_level_size(root, parent_level - 1);
386 387 388 389 390 391 392
	end_slot = parent_nritems;

	if (parent_nritems == 1)
		return 0;

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

394 395 396 397 398 399 400 401
		if (!parent->map_token) {
			map_extent_buffer(parent,
					btrfs_node_key_ptr_offset(i),
					sizeof(struct btrfs_key_ptr),
					&parent->map_token, &parent->kaddr,
					&parent->map_start, &parent->map_len,
					KM_USER1);
		}
402 403 404 405 406
		btrfs_node_key(parent, &disk_key, i);
		if (!progress_passed && comp_keys(&disk_key, progress) < 0)
			continue;

		progress_passed = 1;
407
		blocknr = btrfs_node_blockptr(parent, i);
408
		gen = btrfs_node_ptr_generation(parent, i);
409 410
		if (last_block == 0)
			last_block = blocknr;
411

412
		if (i > 0) {
413 414
			other = btrfs_node_blockptr(parent, i - 1);
			close = close_blocks(blocknr, other, blocksize);
415
		}
C
Chris Mason 已提交
416
		if (!close && i < end_slot - 2) {
417 418
			other = btrfs_node_blockptr(parent, i + 1);
			close = close_blocks(blocknr, other, blocksize);
419
		}
420 421
		if (close) {
			last_block = blocknr;
422
			continue;
423
		}
424 425 426 427 428
		if (parent->map_token) {
			unmap_extent_buffer(parent, parent->map_token,
					    KM_USER1);
			parent->map_token = NULL;
		}
429

430 431
		cur = btrfs_find_tree_block(root, blocknr, blocksize);
		if (cur)
432
			uptodate = btrfs_buffer_uptodate(cur, gen);
433 434
		else
			uptodate = 0;
435
		if (!cur || !uptodate) {
436
			if (cache_only) {
437
				free_extent_buffer(cur);
438 439
				continue;
			}
440 441
			if (!cur) {
				cur = read_tree_block(root, blocknr,
442
							 blocksize, gen);
443
			} else if (!uptodate) {
444
				btrfs_read_buffer(cur, gen);
445
			}
446
		}
447
		if (search_start == 0)
448
			search_start = last_block;
449

450
		btrfs_tree_lock(cur);
451
		err = __btrfs_cow_block(trans, root, cur, parent, i,
452
					&cur, search_start,
453 454
					min(16 * blocksize,
					    (end_slot - i) * blocksize));
Y
Yan 已提交
455
		if (err) {
456
			btrfs_tree_unlock(cur);
457
			free_extent_buffer(cur);
458
			break;
Y
Yan 已提交
459
		}
460 461
		search_start = cur->start;
		last_block = cur->start;
462
		*last_ret = search_start;
463 464
		btrfs_tree_unlock(cur);
		free_extent_buffer(cur);
465
	}
466 467 468 469 470
	if (parent->map_token) {
		unmap_extent_buffer(parent, parent->map_token,
				    KM_USER1);
		parent->map_token = NULL;
	}
471 472 473
	return err;
}

C
Chris Mason 已提交
474 475 476 477 478
/*
 * 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 已提交
479
static inline unsigned int leaf_data_end(struct btrfs_root *root,
480
					 struct extent_buffer *leaf)
481
{
482
	u32 nr = btrfs_header_nritems(leaf);
483
	if (nr == 0)
C
Chris Mason 已提交
484
		return BTRFS_LEAF_DATA_SIZE(root);
485
	return btrfs_item_offset_nr(leaf, nr - 1);
486 487
}

C
Chris Mason 已提交
488 489
static int check_node(struct btrfs_root *root, struct btrfs_path *path,
		      int level)
C
Chris Mason 已提交
490
{
491 492 493 494
	struct extent_buffer *parent = NULL;
	struct extent_buffer *node = path->nodes[level];
	struct btrfs_disk_key parent_key;
	struct btrfs_disk_key node_key;
C
Chris Mason 已提交
495
	int parent_slot;
496 497
	int slot;
	struct btrfs_key cpukey;
498
	u32 nritems = btrfs_header_nritems(node);
C
Chris Mason 已提交
499 500

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

503
	slot = path->slots[level];
504 505
	BUG_ON(nritems == 0);
	if (parent) {
A
Aneesh 已提交
506
		parent_slot = path->slots[level + 1];
507 508 509
		btrfs_node_key(parent, &parent_key, parent_slot);
		btrfs_node_key(node, &node_key, 0);
		BUG_ON(memcmp(&parent_key, &node_key,
C
Chris Mason 已提交
510
			      sizeof(struct btrfs_disk_key)));
511
		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
512
		       btrfs_header_bytenr(node));
C
Chris Mason 已提交
513
	}
C
Chris Mason 已提交
514
	BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
515
	if (slot != 0) {
516 517 518
		btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
		btrfs_node_key(node, &node_key, slot);
		BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
519 520
	}
	if (slot < nritems - 1) {
521 522 523
		btrfs_node_key_to_cpu(node, &cpukey, slot + 1);
		btrfs_node_key(node, &node_key, slot);
		BUG_ON(comp_keys(&node_key, &cpukey) >= 0);
C
Chris Mason 已提交
524 525 526 527
	}
	return 0;
}

C
Chris Mason 已提交
528 529
static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
		      int level)
C
Chris Mason 已提交
530
{
531 532
	struct extent_buffer *leaf = path->nodes[level];
	struct extent_buffer *parent = NULL;
C
Chris Mason 已提交
533
	int parent_slot;
534
	struct btrfs_key cpukey;
535 536 537
	struct btrfs_disk_key parent_key;
	struct btrfs_disk_key leaf_key;
	int slot = path->slots[0];
538

539
	u32 nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
540 541

	if (path->nodes[level + 1])
542
		parent = path->nodes[level + 1];
543 544 545 546 547

	if (nritems == 0)
		return 0;

	if (parent) {
A
Aneesh 已提交
548
		parent_slot = path->slots[level + 1];
549 550
		btrfs_node_key(parent, &parent_key, parent_slot);
		btrfs_item_key(leaf, &leaf_key, 0);
551

552
		BUG_ON(memcmp(&parent_key, &leaf_key,
C
Chris Mason 已提交
553
		       sizeof(struct btrfs_disk_key)));
554
		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
555
		       btrfs_header_bytenr(leaf));
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
	}
#if 0
	for (i = 0; nritems > 1 && i < nritems - 2; i++) {
		btrfs_item_key_to_cpu(leaf, &cpukey, i + 1);
		btrfs_item_key(leaf, &leaf_key, i);
		if (comp_keys(&leaf_key, &cpukey) >= 0) {
			btrfs_print_leaf(root, leaf);
			printk("slot %d offset bad key\n", i);
			BUG_ON(1);
		}
		if (btrfs_item_offset_nr(leaf, i) !=
			btrfs_item_end_nr(leaf, i + 1)) {
			btrfs_print_leaf(root, leaf);
			printk("slot %d offset bad\n", i);
			BUG_ON(1);
		}
		if (i == 0) {
			if (btrfs_item_offset_nr(leaf, i) +
			       btrfs_item_size_nr(leaf, i) !=
			       BTRFS_LEAF_DATA_SIZE(root)) {
				btrfs_print_leaf(root, leaf);
				printk("slot %d first offset bad\n", i);
				BUG_ON(1);
			}
		}
C
Chris Mason 已提交
581
	}
582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603
	if (nritems > 0) {
		if (btrfs_item_size_nr(leaf, nritems - 1) > 4096) {
				btrfs_print_leaf(root, leaf);
				printk("slot %d bad size \n", nritems - 1);
				BUG_ON(1);
		}
	}
#endif
	if (slot != 0 && slot < nritems - 1) {
		btrfs_item_key(leaf, &leaf_key, slot);
		btrfs_item_key_to_cpu(leaf, &cpukey, slot - 1);
		if (comp_keys(&leaf_key, &cpukey) <= 0) {
			btrfs_print_leaf(root, leaf);
			printk("slot %d offset bad key\n", slot);
			BUG_ON(1);
		}
		if (btrfs_item_offset_nr(leaf, slot - 1) !=
		       btrfs_item_end_nr(leaf, slot)) {
			btrfs_print_leaf(root, leaf);
			printk("slot %d offset bad\n", slot);
			BUG_ON(1);
		}
604 605
	}
	if (slot < nritems - 1) {
606 607 608 609 610 611 612 613 614
		btrfs_item_key(leaf, &leaf_key, slot);
		btrfs_item_key_to_cpu(leaf, &cpukey, slot + 1);
		BUG_ON(comp_keys(&leaf_key, &cpukey) >= 0);
		if (btrfs_item_offset_nr(leaf, slot) !=
			btrfs_item_end_nr(leaf, slot + 1)) {
			btrfs_print_leaf(root, leaf);
			printk("slot %d offset bad\n", slot);
			BUG_ON(1);
		}
C
Chris Mason 已提交
615
	}
616 617
	BUG_ON(btrfs_item_offset_nr(leaf, 0) +
	       btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
C
Chris Mason 已提交
618 619 620
	return 0;
}

621 622
static int noinline check_block(struct btrfs_root *root,
				struct btrfs_path *path, int level)
C
Chris Mason 已提交
623
{
624
	u64 found_start;
625
	return 0;
626 627 628 629 630 631 632 633 634
	if (btrfs_header_level(path->nodes[level]) != level)
	    printk("warning: bad level %Lu wanted %d found %d\n",
		   path->nodes[level]->start, level,
		   btrfs_header_level(path->nodes[level]));
	found_start = btrfs_header_bytenr(path->nodes[level]);
	if (found_start != path->nodes[level]->start) {
	    printk("warning: bad bytentr %Lu found %Lu\n",
		   path->nodes[level]->start, found_start);
	}
635
#if 0
636 637
	struct extent_buffer *buf = path->nodes[level];

638 639 640
	if (memcmp_extent_buffer(buf, root->fs_info->fsid,
				 (unsigned long)btrfs_header_fsid(buf),
				 BTRFS_FSID_SIZE)) {
641
		printk("warning bad block %Lu\n", buf->start);
642
		return 1;
643
	}
644
#endif
C
Chris Mason 已提交
645
	if (level == 0)
C
Chris Mason 已提交
646 647
		return check_leaf(root, path, level);
	return check_node(root, path, level);
C
Chris Mason 已提交
648 649
}

C
Chris Mason 已提交
650
/*
651 652 653
 * 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 已提交
654 655 656 657 658 659
 * 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
 */
660 661 662
static int generic_bin_search(struct extent_buffer *eb, unsigned long p,
			      int item_size, struct btrfs_key *key,
			      int max, int *slot)
663 664 665 666 667
{
	int low = 0;
	int high = max;
	int mid;
	int ret;
668
	struct btrfs_disk_key *tmp = NULL;
669 670 671 672 673 674
	struct btrfs_disk_key unaligned;
	unsigned long offset;
	char *map_token = NULL;
	char *kaddr = NULL;
	unsigned long map_start = 0;
	unsigned long map_len = 0;
675
	int err;
676 677 678

	while(low < high) {
		mid = (low + high) / 2;
679 680 681 682 683
		offset = p + mid * item_size;

		if (!map_token || offset < map_start ||
		    (offset + sizeof(struct btrfs_disk_key)) >
		    map_start + map_len) {
684
			if (map_token) {
685
				unmap_extent_buffer(eb, map_token, KM_USER0);
686 687 688 689 690 691 692 693 694 695 696 697 698 699 700
				map_token = NULL;
			}
			err = map_extent_buffer(eb, offset,
						sizeof(struct btrfs_disk_key),
						&map_token, &kaddr,
						&map_start, &map_len, KM_USER0);

			if (!err) {
				tmp = (struct btrfs_disk_key *)(kaddr + offset -
							map_start);
			} else {
				read_extent_buffer(eb, &unaligned,
						   offset, sizeof(unaligned));
				tmp = &unaligned;
			}
701 702 703 704 705

		} else {
			tmp = (struct btrfs_disk_key *)(kaddr + offset -
							map_start);
		}
706 707 708 709 710 711 712 713
		ret = comp_keys(tmp, key);

		if (ret < 0)
			low = mid + 1;
		else if (ret > 0)
			high = mid;
		else {
			*slot = mid;
714 715
			if (map_token)
				unmap_extent_buffer(eb, map_token, KM_USER0);
716 717 718 719
			return 0;
		}
	}
	*slot = low;
720 721
	if (map_token)
		unmap_extent_buffer(eb, map_token, KM_USER0);
722 723 724
	return 1;
}

C
Chris Mason 已提交
725 726 727 728
/*
 * simple bin_search frontend that does the right thing for
 * leaves vs nodes
 */
729 730
static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
		      int level, int *slot)
731
{
732 733 734
	if (level == 0) {
		return generic_bin_search(eb,
					  offsetof(struct btrfs_leaf, items),
C
Chris Mason 已提交
735
					  sizeof(struct btrfs_item),
736
					  key, btrfs_header_nritems(eb),
737
					  slot);
738
	} else {
739 740
		return generic_bin_search(eb,
					  offsetof(struct btrfs_node, ptrs),
C
Chris Mason 已提交
741
					  sizeof(struct btrfs_key_ptr),
742
					  key, btrfs_header_nritems(eb),
743
					  slot);
744 745 746 747
	}
	return -1;
}

748 749
static struct extent_buffer *read_node_slot(struct btrfs_root *root,
				   struct extent_buffer *parent, int slot)
750
{
751
	int level = btrfs_header_level(parent);
752 753
	if (slot < 0)
		return NULL;
754
	if (slot >= btrfs_header_nritems(parent))
755
		return NULL;
756 757 758

	BUG_ON(level == 0);

759
	return read_tree_block(root, btrfs_node_blockptr(parent, slot),
760 761
		       btrfs_level_size(root, level - 1),
		       btrfs_node_ptr_generation(parent, slot));
762 763
}

764 765 766
static int balance_level(struct btrfs_trans_handle *trans,
			 struct btrfs_root *root,
			 struct btrfs_path *path, int level)
767
{
768 769 770 771
	struct extent_buffer *right = NULL;
	struct extent_buffer *mid;
	struct extent_buffer *left = NULL;
	struct extent_buffer *parent = NULL;
772 773 774 775
	int ret = 0;
	int wret;
	int pslot;
	int orig_slot = path->slots[level];
776
	int err_on_enospc = 0;
777
	u64 orig_ptr;
778 779 780 781

	if (level == 0)
		return 0;

782
	mid = path->nodes[level];
783
	WARN_ON(!path->locks[level]);
784 785
	WARN_ON(btrfs_header_generation(mid) != trans->transid);

786
	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
787

C
Chris Mason 已提交
788
	if (level < BTRFS_MAX_LEVEL - 1)
789
		parent = path->nodes[level + 1];
790 791
	pslot = path->slots[level + 1];

C
Chris Mason 已提交
792 793 794 795
	/*
	 * deal with the case where there is only one pointer in the root
	 * by promoting the node below to a root
	 */
796 797
	if (!parent) {
		struct extent_buffer *child;
798

799
		if (btrfs_header_nritems(mid) != 1)
800 801 802
			return 0;

		/* promote the child to a root */
803
		child = read_node_slot(root, mid, 0);
804
		btrfs_tree_lock(child);
805
		BUG_ON(!child);
806 807 808
		ret = btrfs_cow_block(trans, root, child, mid, 0, &child);
		BUG_ON(ret);

809
		spin_lock(&root->node_lock);
810
		root->node = child;
811 812
		spin_unlock(&root->node_lock);

813
		add_root_to_dirty_list(root);
814 815
		btrfs_tree_unlock(child);
		path->locks[level] = 0;
816
		path->nodes[level] = NULL;
817
		clean_tree_block(trans, root, mid);
818
		btrfs_tree_unlock(mid);
819
		/* once for the path */
820
		free_extent_buffer(mid);
821 822 823
		ret = btrfs_free_extent(trans, root, mid->start, mid->len,
					root->root_key.objectid,
					btrfs_header_generation(mid), 0, 0, 1);
824
		/* once for the root ptr */
825
		free_extent_buffer(mid);
826
		return ret;
827
	}
828
	if (btrfs_header_nritems(mid) >
C
Chris Mason 已提交
829
	    BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
830 831
		return 0;

832
	if (btrfs_header_nritems(mid) < 2)
833 834
		err_on_enospc = 1;

835 836
	left = read_node_slot(root, parent, pslot - 1);
	if (left) {
837
		btrfs_tree_lock(left);
838 839
		wret = btrfs_cow_block(trans, root, left,
				       parent, pslot - 1, &left);
840 841 842 843
		if (wret) {
			ret = wret;
			goto enospc;
		}
844
	}
845 846
	right = read_node_slot(root, parent, pslot + 1);
	if (right) {
847
		btrfs_tree_lock(right);
848 849
		wret = btrfs_cow_block(trans, root, right,
				       parent, pslot + 1, &right);
850 851 852 853 854 855 856
		if (wret) {
			ret = wret;
			goto enospc;
		}
	}

	/* first, try to make some room in the middle buffer */
857 858
	if (left) {
		orig_slot += btrfs_header_nritems(left);
859
		wret = push_node_left(trans, root, left, mid, 1);
860 861
		if (wret < 0)
			ret = wret;
862
		if (btrfs_header_nritems(mid) < 2)
863
			err_on_enospc = 1;
864
	}
865 866 867 868

	/*
	 * then try to empty the right most buffer into the middle
	 */
869
	if (right) {
870
		wret = push_node_left(trans, root, mid, right, 1);
871
		if (wret < 0 && wret != -ENOSPC)
872
			ret = wret;
873
		if (btrfs_header_nritems(right) == 0) {
874
			u64 bytenr = right->start;
875
			u64 generation = btrfs_header_generation(parent);
876 877
			u32 blocksize = right->len;

878
			clean_tree_block(trans, root, right);
879
			btrfs_tree_unlock(right);
880
			free_extent_buffer(right);
881
			right = NULL;
882 883
			wret = del_ptr(trans, root, path, level + 1, pslot +
				       1);
884 885
			if (wret)
				ret = wret;
886
			wret = btrfs_free_extent(trans, root, bytenr,
887 888 889
						 blocksize,
						 btrfs_header_owner(parent),
						 generation, 0, 0, 1);
890 891 892
			if (wret)
				ret = wret;
		} else {
893 894 895 896
			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);
897 898
		}
	}
899
	if (btrfs_header_nritems(mid) == 1) {
900 901 902 903 904 905 906 907 908
		/*
		 * 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
		 */
909 910
		BUG_ON(!left);
		wret = balance_node_right(trans, root, mid, left);
911
		if (wret < 0) {
912
			ret = wret;
913 914
			goto enospc;
		}
915 916 917 918 919
		if (wret == 1) {
			wret = push_node_left(trans, root, left, mid, 1);
			if (wret < 0)
				ret = wret;
		}
920 921
		BUG_ON(wret == 1);
	}
922
	if (btrfs_header_nritems(mid) == 0) {
923
		/* we've managed to empty the middle node, drop it */
924
		u64 root_gen = btrfs_header_generation(parent);
925 926
		u64 bytenr = mid->start;
		u32 blocksize = mid->len;
927

928
		clean_tree_block(trans, root, mid);
929
		btrfs_tree_unlock(mid);
930
		free_extent_buffer(mid);
931
		mid = NULL;
932
		wret = del_ptr(trans, root, path, level + 1, pslot);
933 934
		if (wret)
			ret = wret;
935 936 937
		wret = btrfs_free_extent(trans, root, bytenr, blocksize,
					 btrfs_header_owner(parent),
					 root_gen, 0, 0, 1);
938 939
		if (wret)
			ret = wret;
940 941
	} else {
		/* update the parent key to reflect our changes */
942 943 944 945
		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);
946
	}
947

948
	/* update the path */
949 950 951
	if (left) {
		if (btrfs_header_nritems(left) > orig_slot) {
			extent_buffer_get(left);
952
			/* left was locked after cow */
953
			path->nodes[level] = left;
954 955
			path->slots[level + 1] -= 1;
			path->slots[level] = orig_slot;
956 957
			if (mid) {
				btrfs_tree_unlock(mid);
958
				free_extent_buffer(mid);
959
			}
960
		} else {
961
			orig_slot -= btrfs_header_nritems(left);
962 963 964
			path->slots[level] = orig_slot;
		}
	}
965
	/* double check we haven't messed things up */
C
Chris Mason 已提交
966
	check_block(root, path, level);
C
Chris Mason 已提交
967
	if (orig_ptr !=
968
	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
969
		BUG();
970
enospc:
971 972
	if (right) {
		btrfs_tree_unlock(right);
973
		free_extent_buffer(right);
974 975 976 977
	}
	if (left) {
		if (path->nodes[level] != left)
			btrfs_tree_unlock(left);
978
		free_extent_buffer(left);
979
	}
980 981 982
	return ret;
}

983
/* returns zero if the push worked, non-zero otherwise */
984 985 986
static int noinline push_nodes_for_insert(struct btrfs_trans_handle *trans,
					  struct btrfs_root *root,
					  struct btrfs_path *path, int level)
987
{
988 989 990 991
	struct extent_buffer *right = NULL;
	struct extent_buffer *mid;
	struct extent_buffer *left = NULL;
	struct extent_buffer *parent = NULL;
992 993 994 995 996 997 998 999 1000
	int ret = 0;
	int wret;
	int pslot;
	int orig_slot = path->slots[level];
	u64 orig_ptr;

	if (level == 0)
		return 1;

1001
	mid = path->nodes[level];
1002
	WARN_ON(btrfs_header_generation(mid) != trans->transid);
1003 1004 1005
	orig_ptr = btrfs_node_blockptr(mid, orig_slot);

	if (level < BTRFS_MAX_LEVEL - 1)
1006
		parent = path->nodes[level + 1];
1007 1008
	pslot = path->slots[level + 1];

1009
	if (!parent)
1010 1011
		return 1;

1012
	left = read_node_slot(root, parent, pslot - 1);
1013 1014

	/* first, try to make some room in the middle buffer */
1015
	if (left) {
1016
		u32 left_nr;
1017 1018

		btrfs_tree_lock(left);
1019
		left_nr = btrfs_header_nritems(left);
C
Chris Mason 已提交
1020 1021 1022
		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
			wret = 1;
		} else {
1023 1024
			ret = btrfs_cow_block(trans, root, left, parent,
					      pslot - 1, &left);
1025 1026 1027 1028
			if (ret)
				wret = 1;
			else {
				wret = push_node_left(trans, root,
1029
						      left, mid, 0);
1030
			}
C
Chris Mason 已提交
1031
		}
1032 1033 1034
		if (wret < 0)
			ret = wret;
		if (wret == 0) {
1035
			struct btrfs_disk_key disk_key;
1036
			orig_slot += left_nr;
1037 1038 1039 1040 1041
			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;
1042 1043
				path->slots[level + 1] -= 1;
				path->slots[level] = orig_slot;
1044
				btrfs_tree_unlock(mid);
1045
				free_extent_buffer(mid);
1046 1047
			} else {
				orig_slot -=
1048
					btrfs_header_nritems(left);
1049
				path->slots[level] = orig_slot;
1050
				btrfs_tree_unlock(left);
1051
				free_extent_buffer(left);
1052 1053 1054
			}
			return 0;
		}
1055
		btrfs_tree_unlock(left);
1056
		free_extent_buffer(left);
1057
	}
1058
	right = read_node_slot(root, parent, pslot + 1);
1059 1060 1061 1062

	/*
	 * then try to empty the right most buffer into the middle
	 */
1063
	if (right) {
C
Chris Mason 已提交
1064
		u32 right_nr;
1065
		btrfs_tree_lock(right);
1066
		right_nr = btrfs_header_nritems(right);
C
Chris Mason 已提交
1067 1068 1069
		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
			wret = 1;
		} else {
1070 1071 1072
			ret = btrfs_cow_block(trans, root, right,
					      parent, pslot + 1,
					      &right);
1073 1074 1075 1076
			if (ret)
				wret = 1;
			else {
				wret = balance_node_right(trans, root,
1077
							  right, mid);
1078
			}
C
Chris Mason 已提交
1079
		}
1080 1081 1082
		if (wret < 0)
			ret = wret;
		if (wret == 0) {
1083 1084 1085 1086 1087 1088 1089 1090
			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;
1091 1092
				path->slots[level + 1] += 1;
				path->slots[level] = orig_slot -
1093
					btrfs_header_nritems(mid);
1094
				btrfs_tree_unlock(mid);
1095
				free_extent_buffer(mid);
1096
			} else {
1097
				btrfs_tree_unlock(right);
1098
				free_extent_buffer(right);
1099 1100 1101
			}
			return 0;
		}
1102
		btrfs_tree_unlock(right);
1103
		free_extent_buffer(right);
1104 1105 1106 1107
	}
	return 1;
}

1108 1109 1110 1111
/*
 * readahead one full node of leaves
 */
static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path,
1112
			     int level, int slot, u64 objectid)
1113
{
1114
	struct extent_buffer *node;
1115
	struct btrfs_disk_key disk_key;
1116 1117
	u32 nritems;
	u64 search;
1118 1119 1120
	u64 lowest_read;
	u64 highest_read;
	u64 nread = 0;
1121
	int direction = path->reada;
1122
	struct extent_buffer *eb;
1123 1124 1125
	u32 nr;
	u32 blocksize;
	u32 nscan = 0;
1126

1127
	if (level != 1)
1128 1129 1130
		return;

	if (!path->nodes[level])
1131 1132
		return;

1133
	node = path->nodes[level];
1134

1135
	search = btrfs_node_blockptr(node, slot);
1136 1137
	blocksize = btrfs_level_size(root, level - 1);
	eb = btrfs_find_tree_block(root, search, blocksize);
1138 1139
	if (eb) {
		free_extent_buffer(eb);
1140 1141 1142
		return;
	}

1143 1144 1145
	highest_read = search;
	lowest_read = search;

1146
	nritems = btrfs_header_nritems(node);
1147
	nr = slot;
1148
	while(1) {
1149 1150 1151 1152 1153 1154 1155 1156
		if (direction < 0) {
			if (nr == 0)
				break;
			nr--;
		} else if (direction > 0) {
			nr++;
			if (nr >= nritems)
				break;
1157
		}
1158 1159 1160 1161 1162
		if (path->reada < 0 && objectid) {
			btrfs_node_key(node, &disk_key, nr);
			if (btrfs_disk_key_objectid(&disk_key) != objectid)
				break;
		}
1163 1164 1165 1166
		search = btrfs_node_blockptr(node, nr);
		if ((search >= lowest_read && search <= highest_read) ||
		    (search < lowest_read && lowest_read - search <= 32768) ||
		    (search > highest_read && search - highest_read <= 32768)) {
1167 1168
			readahead_tree_block(root, search, blocksize,
				     btrfs_node_ptr_generation(node, nr));
1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180
			nread += blocksize;
		}
		nscan++;
		if (path->reada < 2 && (nread > (256 * 1024) || nscan > 32))
			break;
		if(nread > (1024 * 1024) || nscan > 128)
			break;

		if (search < lowest_read)
			lowest_read = search;
		if (search > highest_read)
			highest_read = search;
1181 1182
	}
}
1183 1184 1185 1186 1187

static void unlock_up(struct btrfs_path *path, int level, int lowest_unlock)
{
	int i;
	int skip_level = level;
1188
	int no_skips = 0;
1189 1190 1191 1192 1193 1194 1195
	struct extent_buffer *t;

	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
		if (!path->nodes[i])
			break;
		if (!path->locks[i])
			break;
1196
		if (!no_skips && path->slots[i] == 0) {
1197 1198 1199
			skip_level = i + 1;
			continue;
		}
1200
		if (!no_skips && path->keep_locks) {
1201 1202 1203
			u32 nritems;
			t = path->nodes[i];
			nritems = btrfs_header_nritems(t);
1204
			if (nritems < 1 || path->slots[i] >= nritems - 1) {
1205 1206 1207 1208
				skip_level = i + 1;
				continue;
			}
		}
1209 1210 1211
		if (skip_level < i && i >= lowest_unlock)
			no_skips = 1;

1212 1213 1214 1215 1216 1217 1218 1219
		t = path->nodes[i];
		if (i >= lowest_unlock && i > skip_level && path->locks[i]) {
			btrfs_tree_unlock(t);
			path->locks[i] = 0;
		}
	}
}

C
Chris Mason 已提交
1220 1221 1222 1223 1224 1225
/*
 * 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 已提交
1226 1227
 * be inserted, and 1 is returned.  If there are other errors during the
 * search a negative error number is returned.
C
Chris Mason 已提交
1228 1229 1230 1231
 *
 * 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 已提交
1232
 */
1233 1234 1235
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)
1236
{
1237
	struct extent_buffer *b;
1238
	struct extent_buffer *tmp;
1239 1240 1241
	int slot;
	int ret;
	int level;
1242
	int should_reada = p->reada;
1243
	int lowest_unlock = 1;
1244
	int blocksize;
1245
	u8 lowest_level = 0;
1246 1247
	u64 blocknr;
	u64 gen;
1248

1249 1250
	lowest_level = p->lowest_level;
	WARN_ON(lowest_level && ins_len);
C
Chris Mason 已提交
1251
	WARN_ON(p->nodes[0] != NULL);
1252
	WARN_ON(cow && root == root->fs_info->extent_root &&
1253 1254 1255 1256 1257 1258 1259
		!mutex_is_locked(&root->fs_info->alloc_mutex));
	WARN_ON(root == root->fs_info->chunk_root &&
		!mutex_is_locked(&root->fs_info->chunk_mutex));
	WARN_ON(root == root->fs_info->dev_root &&
		!mutex_is_locked(&root->fs_info->chunk_mutex));
	if (ins_len < 0)
		lowest_unlock = 2;
1260
again:
1261 1262 1263 1264
	if (p->skip_locking)
		b = btrfs_root_node(root);
	else
		b = btrfs_lock_root_node(root);
1265

1266
	while (b) {
1267
		level = btrfs_header_level(b);
C
Chris Mason 已提交
1268 1269
		if (cow) {
			int wret;
C
Chris Mason 已提交
1270 1271 1272
			wret = btrfs_cow_block(trans, root, b,
					       p->nodes[level + 1],
					       p->slots[level + 1],
Y
Yan 已提交
1273
					       &b);
1274
			if (wret) {
1275
				free_extent_buffer(b);
1276 1277
				return wret;
			}
C
Chris Mason 已提交
1278 1279
		}
		BUG_ON(!cow && ins_len);
1280
		if (level != btrfs_header_level(b))
C
Chris Mason 已提交
1281
			WARN_ON(1);
1282
		level = btrfs_header_level(b);
1283
		p->nodes[level] = b;
1284 1285
		if (!p->skip_locking)
			p->locks[level] = 1;
C
Chris Mason 已提交
1286
		ret = check_block(root, p, level);
C
Chris Mason 已提交
1287 1288
		if (ret)
			return -1;
1289

1290 1291
		ret = bin_search(b, key, level, &slot);
		if (level != 0) {
1292 1293 1294
			if (ret && slot > 0)
				slot -= 1;
			p->slots[level] = slot;
1295
			if (ins_len > 0 && btrfs_header_nritems(b) >=
1296
			    BTRFS_NODEPTRS_PER_BLOCK(root) - 3) {
1297
				int sret = split_node(trans, root, p, level);
C
Chris Mason 已提交
1298 1299 1300 1301 1302
				BUG_ON(sret > 0);
				if (sret)
					return sret;
				b = p->nodes[level];
				slot = p->slots[level];
1303
			} else if (ins_len < 0) {
1304 1305
				int sret = balance_level(trans, root, p,
							 level);
1306 1307 1308
				if (sret)
					return sret;
				b = p->nodes[level];
1309 1310
				if (!b) {
					btrfs_release_path(NULL, p);
1311
					goto again;
1312
				}
1313
				slot = p->slots[level];
1314
				BUG_ON(btrfs_header_nritems(b) == 1);
C
Chris Mason 已提交
1315
			}
1316 1317
			unlock_up(p, level, lowest_unlock);

1318
			/* this is only true while dropping a snapshot */
1319
			if (level == lowest_level) {
1320
				break;
1321
			}
1322

1323 1324 1325 1326 1327 1328
			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);
			if (tmp && btrfs_buffer_uptodate(tmp, gen)) {
1329 1330 1331 1332 1333 1334 1335 1336 1337
				b = tmp;
			} else {
				/*
				 * reduce lock contention at high levels
				 * of the btree by dropping locks before
				 * we read.
				 */
				if (level > 1) {
					btrfs_release_path(NULL, p);
1338 1339
					if (tmp)
						free_extent_buffer(tmp);
1340 1341 1342 1343 1344
					if (should_reada)
						reada_for_search(root, p,
								 level, slot,
								 key->objectid);

1345 1346
					tmp = read_tree_block(root, blocknr,
							 blocksize, gen);
1347 1348 1349 1350
					if (tmp)
						free_extent_buffer(tmp);
					goto again;
				} else {
1351 1352
					if (tmp)
						free_extent_buffer(tmp);
1353 1354 1355 1356
					if (should_reada)
						reada_for_search(root, p,
								 level, slot,
								 key->objectid);
1357 1358 1359
					b = read_node_slot(root, b, slot);
				}
			}
1360 1361
			if (!p->skip_locking)
				btrfs_tree_lock(b);
1362 1363
		} else {
			p->slots[level] = slot;
1364
			if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
C
Chris Mason 已提交
1365
			    sizeof(struct btrfs_item) + ins_len) {
1366
				int sret = split_leaf(trans, root, key,
1367
						      p, ins_len, ret == 0);
C
Chris Mason 已提交
1368 1369 1370 1371
				BUG_ON(sret > 0);
				if (sret)
					return sret;
			}
1372
			unlock_up(p, level, lowest_unlock);
1373 1374 1375
			return ret;
		}
	}
C
Chris Mason 已提交
1376
	return 1;
1377 1378
}

C
Chris Mason 已提交
1379 1380 1381 1382 1383 1384
/*
 * 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 已提交
1385 1386 1387
 *
 * If this fails to write a tree block, it returns -1, but continues
 * fixing up the blocks in ram so the tree is consistent.
C
Chris Mason 已提交
1388
 */
1389 1390 1391
static int fixup_low_keys(struct btrfs_trans_handle *trans,
			  struct btrfs_root *root, struct btrfs_path *path,
			  struct btrfs_disk_key *key, int level)
1392 1393
{
	int i;
C
Chris Mason 已提交
1394
	int ret = 0;
1395 1396
	struct extent_buffer *t;

C
Chris Mason 已提交
1397
	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1398
		int tslot = path->slots[i];
1399
		if (!path->nodes[i])
1400
			break;
1401 1402
		t = path->nodes[i];
		btrfs_set_node_key(t, key, tslot);
C
Chris Mason 已提交
1403
		btrfs_mark_buffer_dirty(path->nodes[i]);
1404 1405 1406
		if (tslot != 0)
			break;
	}
C
Chris Mason 已提交
1407
	return ret;
1408 1409
}

C
Chris Mason 已提交
1410 1411
/*
 * try to push data from one node into the next node left in the
1412
 * tree.
C
Chris Mason 已提交
1413 1414 1415
 *
 * 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 已提交
1416
 */
1417 1418
static int push_node_left(struct btrfs_trans_handle *trans,
			  struct btrfs_root *root, struct extent_buffer *dst,
1419
			  struct extent_buffer *src, int empty)
1420 1421
{
	int push_items = 0;
1422 1423
	int src_nritems;
	int dst_nritems;
C
Chris Mason 已提交
1424
	int ret = 0;
1425

1426 1427
	src_nritems = btrfs_header_nritems(src);
	dst_nritems = btrfs_header_nritems(dst);
C
Chris Mason 已提交
1428
	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
1429 1430
	WARN_ON(btrfs_header_generation(src) != trans->transid);
	WARN_ON(btrfs_header_generation(dst) != trans->transid);
1431

1432
	if (!empty && src_nritems <= 8)
1433 1434
		return 1;

1435
	if (push_items <= 0) {
1436
		return 1;
1437
	}
1438

1439
	if (empty) {
1440
		push_items = min(src_nritems, push_items);
1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452
		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);
1453

1454 1455 1456 1457 1458
	copy_extent_buffer(dst, src,
			   btrfs_node_key_ptr_offset(dst_nritems),
			   btrfs_node_key_ptr_offset(0),
		           push_items * sizeof(struct btrfs_key_ptr));

1459
	if (push_items < src_nritems) {
1460 1461 1462 1463 1464 1465 1466 1467 1468
		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);
1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480
	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
 */
1481 1482 1483 1484
static int balance_node_right(struct btrfs_trans_handle *trans,
			      struct btrfs_root *root,
			      struct extent_buffer *dst,
			      struct extent_buffer *src)
1485 1486 1487 1488 1489 1490 1491
{
	int push_items = 0;
	int max_push;
	int src_nritems;
	int dst_nritems;
	int ret = 0;

1492 1493 1494
	WARN_ON(btrfs_header_generation(src) != trans->transid);
	WARN_ON(btrfs_header_generation(dst) != trans->transid);

1495 1496
	src_nritems = btrfs_header_nritems(src);
	dst_nritems = btrfs_header_nritems(dst);
C
Chris Mason 已提交
1497
	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
1498
	if (push_items <= 0) {
1499
		return 1;
1500 1501 1502 1503 1504
	}

	if (src_nritems < 4) {
		return 1;
	}
1505 1506 1507

	max_push = src_nritems / 2 + 1;
	/* don't try to empty the node */
1508
	if (max_push >= src_nritems) {
1509
		return 1;
1510
	}
Y
Yan 已提交
1511

1512 1513 1514
	if (max_push < push_items)
		push_items = max_push;

1515 1516 1517 1518
	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 已提交
1519

1520 1521 1522 1523
	copy_extent_buffer(dst, src,
			   btrfs_node_key_ptr_offset(0),
			   btrfs_node_key_ptr_offset(src_nritems - push_items),
		           push_items * sizeof(struct btrfs_key_ptr));
1524

1525 1526
	btrfs_set_header_nritems(src, src_nritems - push_items);
	btrfs_set_header_nritems(dst, dst_nritems + push_items);
1527

1528 1529
	btrfs_mark_buffer_dirty(src);
	btrfs_mark_buffer_dirty(dst);
C
Chris Mason 已提交
1530
	return ret;
1531 1532
}

C
Chris Mason 已提交
1533 1534 1535 1536
/*
 * 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 已提交
1537 1538
 *
 * returns zero on success or < 0 on failure.
C
Chris Mason 已提交
1539
 */
1540
static int noinline insert_new_root(struct btrfs_trans_handle *trans,
1541 1542
			   struct btrfs_root *root,
			   struct btrfs_path *path, int level)
C
Chris Mason 已提交
1543
{
1544 1545
	u64 root_gen;
	u64 lower_gen;
1546 1547
	struct extent_buffer *lower;
	struct extent_buffer *c;
1548
	struct extent_buffer *old;
1549
	struct btrfs_disk_key lower_key;
C
Chris Mason 已提交
1550 1551 1552 1553

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

1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564
	if (root->ref_cows)
		root_gen = trans->transid;
	else
		root_gen = 0;

	lower = path->nodes[level-1];
	if (level == 1)
		btrfs_item_key(lower, &lower_key, 0);
	else
		btrfs_node_key(lower, &lower_key, 0);

1565
	c = btrfs_alloc_free_block(trans, root, root->nodesize,
1566 1567
				   root->root_key.objectid,
				   root_gen, lower_key.objectid, level,
1568
				   root->node->start, 0);
1569 1570
	if (IS_ERR(c))
		return PTR_ERR(c);
1571

1572 1573 1574
	memset_extent_buffer(c, 0, 0, root->nodesize);
	btrfs_set_header_nritems(c, 1);
	btrfs_set_header_level(c, level);
1575
	btrfs_set_header_bytenr(c, c->start);
1576 1577 1578 1579 1580 1581
	btrfs_set_header_generation(c, trans->transid);
	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);
1582 1583 1584 1585 1586

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

1587
	btrfs_set_node_key(c, &lower_key, 0);
1588
	btrfs_set_node_blockptr(c, 0, lower->start);
1589 1590 1591 1592
	lower_gen = btrfs_header_generation(lower);
	WARN_ON(lower_gen == 0);

	btrfs_set_node_ptr_generation(c, 0, lower_gen);
1593

1594
	btrfs_mark_buffer_dirty(c);
1595

1596 1597
	spin_lock(&root->node_lock);
	old = root->node;
1598
	root->node = c;
1599 1600 1601 1602 1603
	spin_unlock(&root->node_lock);

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

1604
	add_root_to_dirty_list(root);
1605 1606
	extent_buffer_get(c);
	path->nodes[level] = c;
1607
	path->locks[level] = 1;
C
Chris Mason 已提交
1608
	path->slots[level] = 0;
1609 1610 1611 1612

	if (root->ref_cows && lower_gen != trans->transid) {
		struct btrfs_path *back_path = btrfs_alloc_path();
		int ret;
1613
		mutex_lock(&root->fs_info->alloc_mutex);
1614 1615 1616 1617 1618 1619
		ret = btrfs_insert_extent_backref(trans,
						  root->fs_info->extent_root,
						  path, lower->start,
						  root->root_key.objectid,
						  trans->transid, 0, 0);
		BUG_ON(ret);
1620
		mutex_unlock(&root->fs_info->alloc_mutex);
1621 1622
		btrfs_free_path(back_path);
	}
C
Chris Mason 已提交
1623 1624 1625
	return 0;
}

C
Chris Mason 已提交
1626 1627 1628
/*
 * worker function to insert a single pointer in a node.
 * the node should have enough room for the pointer already
C
Chris Mason 已提交
1629
 *
C
Chris Mason 已提交
1630 1631
 * slot and level indicate where you want the key to go, and
 * blocknr is the block the key points to.
C
Chris Mason 已提交
1632 1633
 *
 * returns zero on success and < 0 on any error
C
Chris Mason 已提交
1634
 */
1635 1636
static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_path *path, struct btrfs_disk_key
1637
		      *key, u64 bytenr, int slot, int level)
C
Chris Mason 已提交
1638
{
1639
	struct extent_buffer *lower;
C
Chris Mason 已提交
1640
	int nritems;
C
Chris Mason 已提交
1641 1642

	BUG_ON(!path->nodes[level]);
1643 1644
	lower = path->nodes[level];
	nritems = btrfs_header_nritems(lower);
C
Chris Mason 已提交
1645 1646
	if (slot > nritems)
		BUG();
C
Chris Mason 已提交
1647
	if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
C
Chris Mason 已提交
1648 1649
		BUG();
	if (slot != nritems) {
1650 1651 1652
		memmove_extent_buffer(lower,
			      btrfs_node_key_ptr_offset(slot + 1),
			      btrfs_node_key_ptr_offset(slot),
C
Chris Mason 已提交
1653
			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
C
Chris Mason 已提交
1654
	}
1655
	btrfs_set_node_key(lower, key, slot);
1656
	btrfs_set_node_blockptr(lower, slot, bytenr);
1657 1658
	WARN_ON(trans->transid == 0);
	btrfs_set_node_ptr_generation(lower, slot, trans->transid);
1659 1660
	btrfs_set_header_nritems(lower, nritems + 1);
	btrfs_mark_buffer_dirty(lower);
C
Chris Mason 已提交
1661 1662 1663
	return 0;
}

C
Chris Mason 已提交
1664 1665 1666 1667 1668 1669
/*
 * 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 已提交
1670 1671
 *
 * returns 0 on success and < 0 on failure
C
Chris Mason 已提交
1672
 */
1673 1674
static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_path *path, int level)
1675
{
1676
	u64 root_gen;
1677 1678 1679
	struct extent_buffer *c;
	struct extent_buffer *split;
	struct btrfs_disk_key disk_key;
1680
	int mid;
C
Chris Mason 已提交
1681
	int ret;
C
Chris Mason 已提交
1682
	int wret;
1683
	u32 c_nritems;
1684

1685
	c = path->nodes[level];
1686
	WARN_ON(btrfs_header_generation(c) != trans->transid);
1687
	if (c == root->node) {
C
Chris Mason 已提交
1688
		/* trying to split the root, lets make a new one */
1689
		ret = insert_new_root(trans, root, path, level + 1);
C
Chris Mason 已提交
1690 1691
		if (ret)
			return ret;
1692 1693
	} else {
		ret = push_nodes_for_insert(trans, root, path, level);
1694 1695
		c = path->nodes[level];
		if (!ret && btrfs_header_nritems(c) <
1696
		    BTRFS_NODEPTRS_PER_BLOCK(root) - 3)
1697
			return 0;
1698 1699
		if (ret < 0)
			return ret;
1700
	}
1701

1702
	c_nritems = btrfs_header_nritems(c);
1703 1704 1705 1706 1707 1708
	if (root->ref_cows)
		root_gen = trans->transid;
	else
		root_gen = 0;

	btrfs_node_key(c, &disk_key, 0);
1709
	split = btrfs_alloc_free_block(trans, root, root->nodesize,
1710 1711 1712 1713
					 root->root_key.objectid,
					 root_gen,
					 btrfs_disk_key_objectid(&disk_key),
					 level, c->start, 0);
1714 1715 1716 1717 1718
	if (IS_ERR(split))
		return PTR_ERR(split);

	btrfs_set_header_flags(split, btrfs_header_flags(c));
	btrfs_set_header_level(split, btrfs_header_level(c));
1719
	btrfs_set_header_bytenr(split, split->start);
1720 1721
	btrfs_set_header_generation(split, trans->transid);
	btrfs_set_header_owner(split, root->root_key.objectid);
1722
	btrfs_set_header_flags(split, 0);
1723 1724 1725
	write_extent_buffer(split, root->fs_info->fsid,
			    (unsigned long)btrfs_header_fsid(split),
			    BTRFS_FSID_SIZE);
1726 1727 1728
	write_extent_buffer(split, root->fs_info->chunk_tree_uuid,
			    (unsigned long)btrfs_header_chunk_tree_uuid(split),
			    BTRFS_UUID_SIZE);
1729

1730
	mid = (c_nritems + 1) / 2;
1731 1732 1733 1734 1735 1736 1737

	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 已提交
1738 1739
	ret = 0;

1740 1741 1742 1743
	btrfs_mark_buffer_dirty(c);
	btrfs_mark_buffer_dirty(split);

	btrfs_node_key(split, &disk_key, 0);
1744
	wret = insert_ptr(trans, root, path, &disk_key, split->start,
1745
			  path->slots[level + 1] + 1,
C
Chris Mason 已提交
1746
			  level + 1);
C
Chris Mason 已提交
1747 1748 1749
	if (wret)
		ret = wret;

C
Chris Mason 已提交
1750
	if (path->slots[level] >= mid) {
C
Chris Mason 已提交
1751
		path->slots[level] -= mid;
1752
		btrfs_tree_unlock(c);
1753 1754
		free_extent_buffer(c);
		path->nodes[level] = split;
C
Chris Mason 已提交
1755 1756
		path->slots[level + 1] += 1;
	} else {
1757
		btrfs_tree_unlock(split);
1758
		free_extent_buffer(split);
1759
	}
C
Chris Mason 已提交
1760
	return ret;
1761 1762
}

C
Chris Mason 已提交
1763 1764 1765 1766 1767
/*
 * 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
 */
1768
static int leaf_space_used(struct extent_buffer *l, int start, int nr)
1769 1770
{
	int data_len;
1771
	int nritems = btrfs_header_nritems(l);
1772
	int end = min(nritems, start + nr) - 1;
1773 1774 1775

	if (!nr)
		return 0;
1776 1777
	data_len = btrfs_item_end_nr(l, start);
	data_len = data_len - btrfs_item_offset_nr(l, end);
C
Chris Mason 已提交
1778
	data_len += sizeof(struct btrfs_item) * nr;
1779
	WARN_ON(data_len < 0);
1780 1781 1782
	return data_len;
}

1783 1784 1785 1786 1787
/*
 * 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
 */
1788
int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
1789
{
1790 1791 1792 1793 1794
	int nritems = btrfs_header_nritems(leaf);
	int ret;
	ret = BTRFS_LEAF_DATA_SIZE(root) - leaf_space_used(leaf, 0, nritems);
	if (ret < 0) {
		printk("leaf free space ret %d, leaf data size %lu, used %d nritems %d\n",
J
Jens Axboe 已提交
1795
		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
1796 1797 1798
		       leaf_space_used(leaf, 0, nritems), nritems);
	}
	return ret;
1799 1800
}

C
Chris Mason 已提交
1801 1802 1803
/*
 * push some data in the path leaf to the right, trying to free up at
 * least data_size bytes.  returns zero if the push worked, nonzero otherwise
C
Chris Mason 已提交
1804 1805 1806
 *
 * returns 1 if the push failed because the other node didn't have enough
 * room, 0 if everything worked out and < 0 if there were major errors.
C
Chris Mason 已提交
1807
 */
1808
static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
1809 1810
			   *root, struct btrfs_path *path, int data_size,
			   int empty)
C
Chris Mason 已提交
1811
{
1812 1813 1814 1815
	struct extent_buffer *left = path->nodes[0];
	struct extent_buffer *right;
	struct extent_buffer *upper;
	struct btrfs_disk_key disk_key;
C
Chris Mason 已提交
1816
	int slot;
1817
	u32 i;
C
Chris Mason 已提交
1818 1819 1820
	int free_space;
	int push_space = 0;
	int push_items = 0;
C
Chris Mason 已提交
1821
	struct btrfs_item *item;
1822
	u32 left_nritems;
1823
	u32 nr;
1824
	u32 right_nritems;
1825
	u32 data_end;
1826
	u32 this_item_size;
1827
	int ret;
C
Chris Mason 已提交
1828 1829 1830 1831 1832 1833

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

1837 1838
	WARN_ON(!btrfs_tree_locked(path->nodes[1]));

1839
	right = read_node_slot(root, upper, slot + 1);
1840
	btrfs_tree_lock(right);
C
Chris Mason 已提交
1841
	free_space = btrfs_leaf_free_space(root, right);
1842 1843
	if (free_space < data_size + sizeof(struct btrfs_item))
		goto out_unlock;
1844

C
Chris Mason 已提交
1845
	/* cow and double check */
1846 1847
	ret = btrfs_cow_block(trans, root, right, upper,
			      slot + 1, &right);
1848 1849 1850
	if (ret)
		goto out_unlock;

C
Chris Mason 已提交
1851
	free_space = btrfs_leaf_free_space(root, right);
1852 1853
	if (free_space < data_size + sizeof(struct btrfs_item))
		goto out_unlock;
C
Chris Mason 已提交
1854

1855
	left_nritems = btrfs_header_nritems(left);
1856 1857
	if (left_nritems == 0)
		goto out_unlock;
1858

1859 1860 1861 1862 1863 1864 1865
	if (empty)
		nr = 0;
	else
		nr = 1;

	i = left_nritems - 1;
	while (i >= nr) {
1866
		item = btrfs_item_nr(left, i);
1867

C
Chris Mason 已提交
1868 1869
		if (path->slots[0] == i)
			push_space += data_size + sizeof(*item);
1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880

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

		this_item_size = btrfs_item_size(left, item);
		if (this_item_size + sizeof(*item) + push_space > free_space)
C
Chris Mason 已提交
1881 1882
			break;
		push_items++;
1883
		push_space += this_item_size + sizeof(*item);
1884 1885 1886
		if (i == 0)
			break;
		i--;
1887 1888 1889 1890
	}
	if (left->map_token) {
		unmap_extent_buffer(left, left->map_token, KM_USER1);
		left->map_token = NULL;
C
Chris Mason 已提交
1891
	}
1892

1893 1894
	if (push_items == 0)
		goto out_unlock;
1895

1896
	if (!empty && push_items == left_nritems)
1897
		WARN_ON(1);
1898

C
Chris Mason 已提交
1899
	/* push left to right */
1900
	right_nritems = btrfs_header_nritems(right);
1901

1902
	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
C
Chris Mason 已提交
1903
	push_space -= leaf_data_end(root, left);
1904

C
Chris Mason 已提交
1905
	/* make room in the right data area */
1906 1907 1908 1909 1910 1911
	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 已提交
1912
	/* copy from the left data area */
1913
	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
C
Chris Mason 已提交
1914 1915 1916
		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
		     btrfs_leaf_data(left) + leaf_data_end(root, left),
		     push_space);
1917 1918 1919 1920 1921

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

C
Chris Mason 已提交
1922
	/* copy the items from left to right */
1923 1924 1925
	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 已提交
1926 1927

	/* update the item pointers */
1928
	right_nritems += push_items;
1929
	btrfs_set_header_nritems(right, right_nritems);
C
Chris Mason 已提交
1930
	push_space = BTRFS_LEAF_DATA_SIZE(root);
1931
	for (i = 0; i < right_nritems; i++) {
1932
		item = btrfs_item_nr(right, i);
1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946
		if (!right->map_token) {
			map_extent_buffer(right, (unsigned long)item,
					sizeof(struct btrfs_item),
					&right->map_token, &right->kaddr,
					&right->map_start, &right->map_len,
					KM_USER1);
		}
		push_space -= btrfs_item_size(right, item);
		btrfs_set_item_offset(right, item, push_space);
	}

	if (right->map_token) {
		unmap_extent_buffer(right, right->map_token, KM_USER1);
		right->map_token = NULL;
C
Chris Mason 已提交
1947
	}
1948
	left_nritems -= push_items;
1949
	btrfs_set_header_nritems(left, left_nritems);
C
Chris Mason 已提交
1950

1951 1952
	if (left_nritems)
		btrfs_mark_buffer_dirty(left);
1953
	btrfs_mark_buffer_dirty(right);
1954

1955 1956
	btrfs_item_key(right, &disk_key, 0);
	btrfs_set_node_key(upper, &disk_key, slot + 1);
C
Chris Mason 已提交
1957
	btrfs_mark_buffer_dirty(upper);
C
Chris Mason 已提交
1958

C
Chris Mason 已提交
1959
	/* then fixup the leaf pointer in the path */
1960 1961
	if (path->slots[0] >= left_nritems) {
		path->slots[0] -= left_nritems;
1962 1963 1964
		if (btrfs_header_nritems(path->nodes[0]) == 0)
			clean_tree_block(trans, root, path->nodes[0]);
		btrfs_tree_unlock(path->nodes[0]);
1965 1966
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = right;
C
Chris Mason 已提交
1967 1968
		path->slots[1] += 1;
	} else {
1969
		btrfs_tree_unlock(right);
1970
		free_extent_buffer(right);
C
Chris Mason 已提交
1971 1972
	}
	return 0;
1973 1974 1975 1976 1977

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

C
Chris Mason 已提交
1980 1981 1982 1983
/*
 * 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
 */
1984
static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
1985 1986
			  *root, struct btrfs_path *path, int data_size,
			  int empty)
1987
{
1988 1989 1990
	struct btrfs_disk_key disk_key;
	struct extent_buffer *right = path->nodes[0];
	struct extent_buffer *left;
1991 1992 1993 1994 1995
	int slot;
	int i;
	int free_space;
	int push_space = 0;
	int push_items = 0;
C
Chris Mason 已提交
1996
	struct btrfs_item *item;
1997
	u32 old_left_nritems;
1998
	u32 right_nritems;
1999
	u32 nr;
C
Chris Mason 已提交
2000 2001
	int ret = 0;
	int wret;
2002 2003
	u32 this_item_size;
	u32 old_left_item_size;
2004 2005

	slot = path->slots[1];
2006
	if (slot == 0)
2007
		return 1;
2008
	if (!path->nodes[1])
2009
		return 1;
2010

2011 2012 2013 2014 2015
	right_nritems = btrfs_header_nritems(right);
	if (right_nritems == 0) {
		return 1;
	}

2016 2017
	WARN_ON(!btrfs_tree_locked(path->nodes[1]));

2018
	left = read_node_slot(root, path->nodes[1], slot - 1);
2019
	btrfs_tree_lock(left);
C
Chris Mason 已提交
2020
	free_space = btrfs_leaf_free_space(root, left);
C
Chris Mason 已提交
2021
	if (free_space < data_size + sizeof(struct btrfs_item)) {
2022 2023
		ret = 1;
		goto out;
2024
	}
C
Chris Mason 已提交
2025 2026

	/* cow and double check */
2027 2028
	ret = btrfs_cow_block(trans, root, left,
			      path->nodes[1], slot - 1, &left);
2029 2030
	if (ret) {
		/* we hit -ENOSPC, but it isn't fatal here */
2031 2032
		ret = 1;
		goto out;
2033
	}
2034

C
Chris Mason 已提交
2035
	free_space = btrfs_leaf_free_space(root, left);
C
Chris Mason 已提交
2036
	if (free_space < data_size + sizeof(struct btrfs_item)) {
2037 2038
		ret = 1;
		goto out;
C
Chris Mason 已提交
2039 2040
	}

2041 2042 2043 2044 2045 2046
	if (empty)
		nr = right_nritems;
	else
		nr = right_nritems - 1;

	for (i = 0; i < nr; i++) {
2047
		item = btrfs_item_nr(right, i);
2048 2049 2050 2051 2052 2053 2054 2055
		if (!right->map_token) {
			map_extent_buffer(right, (unsigned long)item,
					sizeof(struct btrfs_item),
					&right->map_token, &right->kaddr,
					&right->map_start, &right->map_len,
					KM_USER1);
		}

2056 2057
		if (path->slots[0] == i)
			push_space += data_size + sizeof(*item);
2058 2059 2060

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

2063
		push_items++;
2064 2065 2066 2067 2068 2069
		push_space += this_item_size + sizeof(*item);
	}

	if (right->map_token) {
		unmap_extent_buffer(right, right->map_token, KM_USER1);
		right->map_token = NULL;
2070
	}
2071

2072
	if (push_items == 0) {
2073 2074
		ret = 1;
		goto out;
2075
	}
2076
	if (!empty && push_items == btrfs_header_nritems(right))
2077
		WARN_ON(1);
2078

2079
	/* push data from right to left */
2080 2081 2082 2083 2084
	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 已提交
2085
	push_space = BTRFS_LEAF_DATA_SIZE(root) -
2086 2087 2088
		     btrfs_item_offset_nr(right, push_items -1);

	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
C
Chris Mason 已提交
2089 2090
		     leaf_data_end(root, left) - push_space,
		     btrfs_leaf_data(right) +
2091
		     btrfs_item_offset_nr(right, push_items - 1),
C
Chris Mason 已提交
2092
		     push_space);
2093
	old_left_nritems = btrfs_header_nritems(left);
2094 2095
	BUG_ON(old_left_nritems < 0);

2096
	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
C
Chris Mason 已提交
2097
	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
2098
		u32 ioff;
2099

2100
		item = btrfs_item_nr(left, i);
2101 2102 2103 2104 2105 2106 2107 2108
		if (!left->map_token) {
			map_extent_buffer(left, (unsigned long)item,
					sizeof(struct btrfs_item),
					&left->map_token, &left->kaddr,
					&left->map_start, &left->map_len,
					KM_USER1);
		}

2109 2110
		ioff = btrfs_item_offset(left, item);
		btrfs_set_item_offset(left, item,
2111
		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
2112
	}
2113
	btrfs_set_header_nritems(left, old_left_nritems + push_items);
2114 2115 2116 2117
	if (left->map_token) {
		unmap_extent_buffer(left, left->map_token, KM_USER1);
		left->map_token = NULL;
	}
2118 2119

	/* fixup right node */
2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133
	if (push_items > right_nritems) {
		printk("push items %d nr %u\n", push_items, right_nritems);
		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),
2134 2135 2136
			      btrfs_item_nr_offset(push_items),
			     (btrfs_header_nritems(right) - push_items) *
			     sizeof(struct btrfs_item));
2137
	}
2138 2139
	right_nritems -= push_items;
	btrfs_set_header_nritems(right, right_nritems);
C
Chris Mason 已提交
2140
	push_space = BTRFS_LEAF_DATA_SIZE(root);
2141 2142
	for (i = 0; i < right_nritems; i++) {
		item = btrfs_item_nr(right, i);
2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157

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

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

2160
	btrfs_mark_buffer_dirty(left);
2161 2162
	if (right_nritems)
		btrfs_mark_buffer_dirty(right);
2163

2164 2165
	btrfs_item_key(right, &disk_key, 0);
	wret = fixup_low_keys(trans, root, path, &disk_key, 1);
C
Chris Mason 已提交
2166 2167
	if (wret)
		ret = wret;
2168 2169 2170 2171

	/* then fixup the leaf pointer in the path */
	if (path->slots[0] < push_items) {
		path->slots[0] += old_left_nritems;
2172 2173 2174
		if (btrfs_header_nritems(path->nodes[0]) == 0)
			clean_tree_block(trans, root, path->nodes[0]);
		btrfs_tree_unlock(path->nodes[0]);
2175 2176
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = left;
2177 2178
		path->slots[1] -= 1;
	} else {
2179
		btrfs_tree_unlock(left);
2180
		free_extent_buffer(left);
2181 2182
		path->slots[0] -= push_items;
	}
2183
	BUG_ON(path->slots[0] < 0);
C
Chris Mason 已提交
2184
	return ret;
2185 2186 2187 2188
out:
	btrfs_tree_unlock(left);
	free_extent_buffer(left);
	return ret;
2189 2190
}

C
Chris Mason 已提交
2191 2192 2193
/*
 * 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 已提交
2194 2195
 *
 * returns 0 if all went well and < 0 on failure.
C
Chris Mason 已提交
2196
 */
2197
static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
2198
		      *root, struct btrfs_key *ins_key,
2199
		      struct btrfs_path *path, int data_size, int extend)
2200
{
2201
	u64 root_gen;
2202
	struct extent_buffer *l;
2203
	u32 nritems;
2204 2205
	int mid;
	int slot;
2206
	struct extent_buffer *right;
C
Chris Mason 已提交
2207
	int space_needed = data_size + sizeof(struct btrfs_item);
2208 2209 2210
	int data_copy_size;
	int rt_data_off;
	int i;
2211
	int ret = 0;
C
Chris Mason 已提交
2212
	int wret;
2213 2214
	int double_split;
	int num_doubles = 0;
2215
	struct btrfs_disk_key disk_key;
C
Chris Mason 已提交
2216

2217 2218 2219
	if (extend)
		space_needed = data_size;

2220 2221 2222 2223 2224
	if (root->ref_cows)
		root_gen = trans->transid;
	else
		root_gen = 0;

C
Chris Mason 已提交
2225
	/* first try to make some room by pushing left and right */
2226
	if (ins_key->type != BTRFS_DIR_ITEM_KEY) {
2227
		wret = push_leaf_right(trans, root, path, data_size, 0);
2228
		if (wret < 0) {
C
Chris Mason 已提交
2229
			return wret;
2230 2231
		}
		if (wret) {
2232
			wret = push_leaf_left(trans, root, path, data_size, 0);
2233 2234 2235 2236
			if (wret < 0)
				return wret;
		}
		l = path->nodes[0];
C
Chris Mason 已提交
2237

2238
		/* did the pushes work? */
2239
		if (btrfs_leaf_free_space(root, l) >= space_needed)
2240
			return 0;
2241
	}
C
Chris Mason 已提交
2242

C
Chris Mason 已提交
2243
	if (!path->nodes[1]) {
2244
		ret = insert_new_root(trans, root, path, 1);
C
Chris Mason 已提交
2245 2246 2247
		if (ret)
			return ret;
	}
2248 2249 2250
again:
	double_split = 0;
	l = path->nodes[0];
2251
	slot = path->slots[0];
2252
	nritems = btrfs_header_nritems(l);
2253
	mid = (nritems + 1)/ 2;
2254

2255 2256
	btrfs_item_key(l, &disk_key, 0);

2257
	right = btrfs_alloc_free_block(trans, root, root->leafsize,
2258 2259 2260
					 root->root_key.objectid,
					 root_gen, disk_key.objectid, 0,
					 l->start, 0);
2261 2262
	if (IS_ERR(right)) {
		BUG_ON(1);
2263
		return PTR_ERR(right);
2264
	}
2265 2266

	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
2267
	btrfs_set_header_bytenr(right, right->start);
2268 2269 2270 2271 2272 2273
	btrfs_set_header_generation(right, trans->transid);
	btrfs_set_header_owner(right, root->root_key.objectid);
	btrfs_set_header_level(right, 0);
	write_extent_buffer(right, root->fs_info->fsid,
			    (unsigned long)btrfs_header_fsid(right),
			    BTRFS_FSID_SIZE);
2274 2275 2276 2277

	write_extent_buffer(right, root->fs_info->chunk_tree_uuid,
			    (unsigned long)btrfs_header_chunk_tree_uuid(right),
			    BTRFS_UUID_SIZE);
2278 2279 2280 2281 2282 2283
	if (mid <= slot) {
		if (nritems == 1 ||
		    leaf_space_used(l, mid, nritems - mid) + space_needed >
			BTRFS_LEAF_DATA_SIZE(root)) {
			if (slot >= nritems) {
				btrfs_cpu_key_to_disk(&disk_key, ins_key);
2284
				btrfs_set_header_nritems(right, 0);
2285
				wret = insert_ptr(trans, root, path,
2286
						  &disk_key, right->start,
2287 2288 2289
						  path->slots[1] + 1, 1);
				if (wret)
					ret = wret;
2290 2291

				btrfs_tree_unlock(path->nodes[0]);
2292 2293
				free_extent_buffer(path->nodes[0]);
				path->nodes[0] = right;
2294 2295
				path->slots[0] = 0;
				path->slots[1] += 1;
2296
				btrfs_mark_buffer_dirty(right);
2297 2298 2299
				return ret;
			}
			mid = slot;
2300 2301 2302 2303 2304
			if (mid != nritems &&
			    leaf_space_used(l, mid, nritems - mid) +
			    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
				double_split = 1;
			}
2305 2306 2307 2308
		}
	} else {
		if (leaf_space_used(l, 0, mid + 1) + space_needed >
			BTRFS_LEAF_DATA_SIZE(root)) {
2309
			if (!extend && slot == 0) {
2310
				btrfs_cpu_key_to_disk(&disk_key, ins_key);
2311
				btrfs_set_header_nritems(right, 0);
2312 2313
				wret = insert_ptr(trans, root, path,
						  &disk_key,
2314
						  right->start,
2315
						  path->slots[1], 1);
2316 2317
				if (wret)
					ret = wret;
2318
				btrfs_tree_unlock(path->nodes[0]);
2319 2320
				free_extent_buffer(path->nodes[0]);
				path->nodes[0] = right;
2321
				path->slots[0] = 0;
2322 2323 2324 2325 2326 2327
				if (path->slots[1] == 0) {
					wret = fixup_low_keys(trans, root,
					           path, &disk_key, 1);
					if (wret)
						ret = wret;
				}
2328
				btrfs_mark_buffer_dirty(right);
2329
				return ret;
2330 2331 2332 2333 2334 2335 2336 2337 2338
			} else if (extend && slot == 0) {
				mid = 1;
			} else {
				mid = slot;
				if (mid != nritems &&
				    leaf_space_used(l, mid, nritems - mid) +
				    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
					double_split = 1;
				}
2339
			}
2340 2341
		}
	}
2342 2343 2344 2345 2346 2347 2348 2349 2350
	nritems = nritems - mid;
	btrfs_set_header_nritems(right, nritems);
	data_copy_size = btrfs_item_end_nr(l, mid) - leaf_data_end(root, l);

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

	copy_extent_buffer(right, l,
C
Chris Mason 已提交
2351 2352 2353
		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
		     data_copy_size, btrfs_leaf_data(l) +
		     leaf_data_end(root, l), data_copy_size);
2354

C
Chris Mason 已提交
2355
	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
2356
		      btrfs_item_end_nr(l, mid);
C
Chris Mason 已提交
2357

2358 2359
	for (i = 0; i < nritems; i++) {
		struct btrfs_item *item = btrfs_item_nr(right, i);
2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370
		u32 ioff;

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

		ioff = btrfs_item_offset(right, item);
2371
		btrfs_set_item_offset(right, item, ioff + rt_data_off);
C
Chris Mason 已提交
2372
	}
C
Chris Mason 已提交
2373

2374 2375 2376 2377 2378
	if (right->map_token) {
		unmap_extent_buffer(right, right->map_token, KM_USER1);
		right->map_token = NULL;
	}

2379
	btrfs_set_header_nritems(l, mid);
C
Chris Mason 已提交
2380
	ret = 0;
2381
	btrfs_item_key(right, &disk_key, 0);
2382 2383
	wret = insert_ptr(trans, root, path, &disk_key, right->start,
			  path->slots[1] + 1, 1);
C
Chris Mason 已提交
2384 2385
	if (wret)
		ret = wret;
2386 2387 2388

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

2391
	if (mid <= slot) {
2392
		btrfs_tree_unlock(path->nodes[0]);
2393 2394
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = right;
2395 2396
		path->slots[0] -= mid;
		path->slots[1] += 1;
2397 2398
	} else {
		btrfs_tree_unlock(right);
2399
		free_extent_buffer(right);
2400
	}
2401

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

2404 2405 2406 2407
	if (double_split) {
		BUG_ON(num_doubles != 0);
		num_doubles++;
		goto again;
2408
	}
2409 2410 2411
	return ret;
}

C
Chris Mason 已提交
2412 2413 2414
int btrfs_truncate_item(struct btrfs_trans_handle *trans,
			struct btrfs_root *root,
			struct btrfs_path *path,
2415
			u32 new_size, int from_end)
C
Chris Mason 已提交
2416 2417 2418 2419
{
	int ret = 0;
	int slot;
	int slot_orig;
2420 2421
	struct extent_buffer *leaf;
	struct btrfs_item *item;
C
Chris Mason 已提交
2422 2423 2424 2425 2426 2427 2428 2429
	u32 nritems;
	unsigned int data_end;
	unsigned int old_data_start;
	unsigned int old_size;
	unsigned int size_diff;
	int i;

	slot_orig = path->slots[0];
2430
	leaf = path->nodes[0];
2431 2432 2433 2434 2435
	slot = path->slots[0];

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

2437
	nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
2438 2439
	data_end = leaf_data_end(root, leaf);

2440
	old_data_start = btrfs_item_offset_nr(leaf, slot);
2441

C
Chris Mason 已提交
2442 2443 2444 2445 2446 2447 2448 2449 2450 2451
	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++) {
2452 2453
		u32 ioff;
		item = btrfs_item_nr(leaf, i);
2454 2455 2456 2457 2458 2459 2460 2461 2462

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

2463 2464
		ioff = btrfs_item_offset(leaf, item);
		btrfs_set_item_offset(leaf, item, ioff + size_diff);
C
Chris Mason 已提交
2465
	}
2466 2467 2468 2469 2470 2471

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

C
Chris Mason 已提交
2472
	/* shift the data */
2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511
	if (from_end) {
		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
			      data_end + size_diff, btrfs_leaf_data(leaf) +
			      data_end, old_data_start + new_size - data_end);
	} else {
		struct btrfs_disk_key disk_key;
		u64 offset;

		btrfs_item_key(leaf, &disk_key, slot);

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

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

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

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

		offset = btrfs_disk_key_offset(&disk_key);
		btrfs_set_disk_key_offset(&disk_key, offset + size_diff);
		btrfs_set_item_key(leaf, &disk_key, slot);
		if (slot == 0)
			fixup_low_keys(trans, root, path, &disk_key, 1);
	}
2512 2513 2514 2515

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

	ret = 0;
2518 2519
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
C
Chris Mason 已提交
2520
		BUG();
2521
	}
C
Chris Mason 已提交
2522 2523 2524
	return ret;
}

2525 2526 2527
int btrfs_extend_item(struct btrfs_trans_handle *trans,
		      struct btrfs_root *root, struct btrfs_path *path,
		      u32 data_size)
2528 2529 2530 2531
{
	int ret = 0;
	int slot;
	int slot_orig;
2532 2533
	struct extent_buffer *leaf;
	struct btrfs_item *item;
2534 2535 2536 2537 2538 2539 2540
	u32 nritems;
	unsigned int data_end;
	unsigned int old_data;
	unsigned int old_size;
	int i;

	slot_orig = path->slots[0];
2541
	leaf = path->nodes[0];
2542

2543
	nritems = btrfs_header_nritems(leaf);
2544 2545
	data_end = leaf_data_end(root, leaf);

2546 2547
	if (btrfs_leaf_free_space(root, leaf) < data_size) {
		btrfs_print_leaf(root, leaf);
2548
		BUG();
2549
	}
2550
	slot = path->slots[0];
2551
	old_data = btrfs_item_end_nr(leaf, slot);
2552 2553

	BUG_ON(slot < 0);
2554 2555 2556 2557 2558
	if (slot >= nritems) {
		btrfs_print_leaf(root, leaf);
		printk("slot %d too large, nritems %d\n", slot, nritems);
		BUG_ON(1);
	}
2559 2560 2561 2562 2563 2564

	/*
	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
	 */
	/* first correct the data pointers */
	for (i = slot; i < nritems; i++) {
2565 2566
		u32 ioff;
		item = btrfs_item_nr(leaf, i);
2567 2568 2569 2570 2571 2572 2573 2574

		if (!leaf->map_token) {
			map_extent_buffer(leaf, (unsigned long)item,
					sizeof(struct btrfs_item),
					&leaf->map_token, &leaf->kaddr,
					&leaf->map_start, &leaf->map_len,
					KM_USER1);
		}
2575 2576
		ioff = btrfs_item_offset(leaf, item);
		btrfs_set_item_offset(leaf, item, ioff - data_size);
2577
	}
2578

2579 2580 2581 2582 2583
	if (leaf->map_token) {
		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
		leaf->map_token = NULL;
	}

2584
	/* shift the data */
2585
	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2586 2587
		      data_end - data_size, btrfs_leaf_data(leaf) +
		      data_end, old_data - data_end);
2588

2589
	data_end = old_data;
2590 2591 2592 2593
	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);
2594 2595

	ret = 0;
2596 2597
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
2598
		BUG();
2599
	}
2600 2601 2602
	return ret;
}

C
Chris Mason 已提交
2603 2604 2605 2606
/*
 * 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.
 */
2607
int btrfs_insert_empty_items(struct btrfs_trans_handle *trans,
2608 2609
			    struct btrfs_root *root,
			    struct btrfs_path *path,
2610 2611
			    struct btrfs_key *cpu_key, u32 *data_size,
			    int nr)
2612
{
2613 2614
	struct extent_buffer *leaf;
	struct btrfs_item *item;
C
Chris Mason 已提交
2615
	int ret = 0;
2616
	int slot;
2617
	int slot_orig;
2618
	int i;
2619
	u32 nritems;
2620 2621
	u32 total_size = 0;
	u32 total_data = 0;
2622
	unsigned int data_end;
C
Chris Mason 已提交
2623 2624
	struct btrfs_disk_key disk_key;

2625 2626 2627
	for (i = 0; i < nr; i++) {
		total_data += data_size[i];
	}
2628

2629 2630
	total_size = total_data + (nr - 1) * sizeof(struct btrfs_item);
	ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1);
2631
	if (ret == 0) {
2632
		return -EEXIST;
C
Chris Mason 已提交
2633
	}
2634 2635
	if (ret < 0)
		goto out;
2636

2637
	slot_orig = path->slots[0];
2638
	leaf = path->nodes[0];
C
Chris Mason 已提交
2639

2640
	nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
2641
	data_end = leaf_data_end(root, leaf);
2642

C
Chris Mason 已提交
2643
	if (btrfs_leaf_free_space(root, leaf) <
2644
	    sizeof(struct btrfs_item) + total_size) {
2645 2646
		btrfs_print_leaf(root, leaf);
		printk("not enough freespace need %u have %d\n",
2647
		       total_size, btrfs_leaf_free_space(root, leaf));
2648
		BUG();
2649
	}
2650

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

2654 2655
	if (slot != nritems) {
		int i;
2656
		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
2657

2658 2659 2660 2661 2662 2663
		if (old_data < data_end) {
			btrfs_print_leaf(root, leaf);
			printk("slot %d old_data %d data_end %d\n",
			       slot, old_data, data_end);
			BUG_ON(1);
		}
2664 2665 2666 2667
		/*
		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
		 */
		/* first correct the data pointers */
2668
		WARN_ON(leaf->map_token);
C
Chris Mason 已提交
2669
		for (i = slot; i < nritems; i++) {
2670
			u32 ioff;
2671

2672
			item = btrfs_item_nr(leaf, i);
2673 2674 2675 2676 2677 2678 2679 2680
			if (!leaf->map_token) {
				map_extent_buffer(leaf, (unsigned long)item,
					sizeof(struct btrfs_item),
					&leaf->map_token, &leaf->kaddr,
					&leaf->map_start, &leaf->map_len,
					KM_USER1);
			}

2681
			ioff = btrfs_item_offset(leaf, item);
2682
			btrfs_set_item_offset(leaf, item, ioff - total_data);
C
Chris Mason 已提交
2683
		}
2684 2685 2686 2687
		if (leaf->map_token) {
			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
			leaf->map_token = NULL;
		}
2688 2689

		/* shift the items */
2690
		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr),
2691
			      btrfs_item_nr_offset(slot),
C
Chris Mason 已提交
2692
			      (nritems - slot) * sizeof(struct btrfs_item));
2693 2694

		/* shift the data */
2695
		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2696
			      data_end - total_data, btrfs_leaf_data(leaf) +
C
Chris Mason 已提交
2697
			      data_end, old_data - data_end);
2698 2699
		data_end = old_data;
	}
2700

2701
	/* setup the item for the new data */
2702 2703 2704 2705 2706 2707 2708 2709 2710
	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);
		btrfs_set_item_offset(leaf, item, data_end - data_size[i]);
		data_end -= data_size[i];
		btrfs_set_item_size(leaf, item, data_size[i]);
	}
	btrfs_set_header_nritems(leaf, nritems + nr);
2711
	btrfs_mark_buffer_dirty(leaf);
C
Chris Mason 已提交
2712 2713

	ret = 0;
2714 2715
	if (slot == 0) {
		btrfs_cpu_key_to_disk(&disk_key, cpu_key);
2716
		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
2717
	}
C
Chris Mason 已提交
2718

2719 2720
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
2721
		BUG();
2722
	}
2723
out:
2724 2725 2726 2727 2728 2729 2730
	return ret;
}

/*
 * Given a key and some data, insert an item into the tree.
 * This does all the path init required, making room in the tree if needed.
 */
2731 2732 2733
int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_key *cpu_key, void *data, u32
		      data_size)
2734 2735
{
	int ret = 0;
C
Chris Mason 已提交
2736
	struct btrfs_path *path;
2737 2738
	struct extent_buffer *leaf;
	unsigned long ptr;
2739

C
Chris Mason 已提交
2740 2741 2742
	path = btrfs_alloc_path();
	BUG_ON(!path);
	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
2743
	if (!ret) {
2744 2745 2746 2747
		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);
2748
	}
C
Chris Mason 已提交
2749
	btrfs_free_path(path);
C
Chris Mason 已提交
2750
	return ret;
2751 2752
}

C
Chris Mason 已提交
2753
/*
C
Chris Mason 已提交
2754
 * delete the pointer from a given node.
C
Chris Mason 已提交
2755 2756 2757 2758 2759
 *
 * If the delete empties a node, the node is removed from the tree,
 * continuing all the way the root if required.  The root is converted into
 * a leaf if all the nodes are emptied.
 */
2760 2761
static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
		   struct btrfs_path *path, int level, int slot)
2762
{
2763
	struct extent_buffer *parent = path->nodes[level];
2764
	u32 nritems;
C
Chris Mason 已提交
2765
	int ret = 0;
2766
	int wret;
2767

2768
	nritems = btrfs_header_nritems(parent);
2769
	if (slot != nritems -1) {
2770 2771 2772
		memmove_extent_buffer(parent,
			      btrfs_node_key_ptr_offset(slot),
			      btrfs_node_key_ptr_offset(slot + 1),
C
Chris Mason 已提交
2773 2774
			      sizeof(struct btrfs_key_ptr) *
			      (nritems - slot - 1));
2775
	}
2776
	nritems--;
2777
	btrfs_set_header_nritems(parent, nritems);
2778
	if (nritems == 0 && parent == root->node) {
2779
		BUG_ON(btrfs_header_level(root->node) != 1);
2780
		/* just turn the root into a leaf and break */
2781
		btrfs_set_header_level(root->node, 0);
2782
	} else if (slot == 0) {
2783 2784 2785 2786
		struct btrfs_disk_key disk_key;

		btrfs_node_key(parent, &disk_key, 0);
		wret = fixup_low_keys(trans, root, path, &disk_key, level + 1);
C
Chris Mason 已提交
2787 2788
		if (wret)
			ret = wret;
2789
	}
C
Chris Mason 已提交
2790
	btrfs_mark_buffer_dirty(parent);
C
Chris Mason 已提交
2791
	return ret;
2792 2793
}

C
Chris Mason 已提交
2794 2795 2796 2797
/*
 * delete the item at the leaf level in path.  If that empties
 * the leaf, remove it from the tree
 */
2798 2799
int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root,
		    struct btrfs_path *path, int slot, int nr)
2800
{
2801 2802
	struct extent_buffer *leaf;
	struct btrfs_item *item;
2803 2804
	int last_off;
	int dsize = 0;
C
Chris Mason 已提交
2805 2806
	int ret = 0;
	int wret;
2807
	int i;
2808
	u32 nritems;
2809

2810
	leaf = path->nodes[0];
2811 2812 2813 2814 2815
	last_off = btrfs_item_offset_nr(leaf, slot + nr - 1);

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

2816
	nritems = btrfs_header_nritems(leaf);
2817

2818
	if (slot + nr != nritems) {
2819
		int i;
C
Chris Mason 已提交
2820
		int data_end = leaf_data_end(root, leaf);
2821 2822

		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
C
Chris Mason 已提交
2823 2824
			      data_end + dsize,
			      btrfs_leaf_data(leaf) + data_end,
2825
			      last_off - data_end);
2826

2827
		for (i = slot + nr; i < nritems; i++) {
2828
			u32 ioff;
2829

2830
			item = btrfs_item_nr(leaf, i);
2831 2832 2833 2834 2835 2836 2837
			if (!leaf->map_token) {
				map_extent_buffer(leaf, (unsigned long)item,
					sizeof(struct btrfs_item),
					&leaf->map_token, &leaf->kaddr,
					&leaf->map_start, &leaf->map_len,
					KM_USER1);
			}
2838 2839
			ioff = btrfs_item_offset(leaf, item);
			btrfs_set_item_offset(leaf, item, ioff + dsize);
C
Chris Mason 已提交
2840
		}
2841 2842 2843 2844 2845 2846

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

2847
		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
2848
			      btrfs_item_nr_offset(slot + nr),
C
Chris Mason 已提交
2849
			      sizeof(struct btrfs_item) *
2850
			      (nritems - slot - nr));
2851
	}
2852 2853
	btrfs_set_header_nritems(leaf, nritems - nr);
	nritems -= nr;
2854

C
Chris Mason 已提交
2855
	/* delete the leaf if we've emptied it */
2856
	if (nritems == 0) {
2857 2858
		if (leaf == root->node) {
			btrfs_set_header_level(leaf, 0);
2859
		} else {
2860
			u64 root_gen = btrfs_header_generation(path->nodes[1]);
2861
			wret = del_ptr(trans, root, path, 1, path->slots[1]);
C
Chris Mason 已提交
2862 2863
			if (wret)
				ret = wret;
2864
			wret = btrfs_free_extent(trans, root,
2865 2866 2867
					 leaf->start, leaf->len,
					 btrfs_header_owner(path->nodes[1]),
					 root_gen, 0, 0, 1);
C
Chris Mason 已提交
2868 2869
			if (wret)
				ret = wret;
2870
		}
2871
	} else {
2872
		int used = leaf_space_used(leaf, 0, nritems);
C
Chris Mason 已提交
2873
		if (slot == 0) {
2874 2875 2876
			struct btrfs_disk_key disk_key;

			btrfs_item_key(leaf, &disk_key, 0);
2877
			wret = fixup_low_keys(trans, root, path,
2878
					      &disk_key, 1);
C
Chris Mason 已提交
2879 2880 2881 2882
			if (wret)
				ret = wret;
		}

C
Chris Mason 已提交
2883
		/* delete the leaf if it is mostly empty */
2884
		if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) {
2885 2886 2887 2888
			/* push_leaf_left fixes the path.
			 * make sure the path still points to our leaf
			 * for possible call to del_ptr below
			 */
2889
			slot = path->slots[1];
2890 2891
			extent_buffer_get(leaf);

2892
			wret = push_leaf_left(trans, root, path, 1, 1);
2893
			if (wret < 0 && wret != -ENOSPC)
C
Chris Mason 已提交
2894
				ret = wret;
2895 2896 2897

			if (path->nodes[0] == leaf &&
			    btrfs_header_nritems(leaf)) {
2898
				wret = push_leaf_right(trans, root, path, 1, 1);
2899
				if (wret < 0 && wret != -ENOSPC)
C
Chris Mason 已提交
2900 2901
					ret = wret;
			}
2902 2903

			if (btrfs_header_nritems(leaf) == 0) {
2904
				u64 root_gen;
2905 2906
				u64 bytenr = leaf->start;
				u32 blocksize = leaf->len;
2907

2908 2909 2910
				root_gen = btrfs_header_generation(
							   path->nodes[1]);

2911
				wret = del_ptr(trans, root, path, 1, slot);
C
Chris Mason 已提交
2912 2913
				if (wret)
					ret = wret;
2914 2915

				free_extent_buffer(leaf);
2916
				wret = btrfs_free_extent(trans, root, bytenr,
2917 2918 2919
					     blocksize,
					     btrfs_header_owner(path->nodes[1]),
					     root_gen, 0, 0, 1);
C
Chris Mason 已提交
2920 2921
				if (wret)
					ret = wret;
C
Chris Mason 已提交
2922
			} else {
2923 2924 2925 2926 2927 2928 2929
				/* 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);
2930
				free_extent_buffer(leaf);
2931
			}
2932
		} else {
2933
			btrfs_mark_buffer_dirty(leaf);
2934 2935
		}
	}
C
Chris Mason 已提交
2936
	return ret;
2937 2938
}

2939
/*
2940
 * search the tree again to find a leaf with lesser keys
2941 2942 2943 2944 2945
 * returns 0 if it found something or 1 if there are no lesser leaves.
 * returns < 0 on io errors.
 */
int btrfs_prev_leaf(struct btrfs_root *root, struct btrfs_path *path)
{
2946 2947 2948
	struct btrfs_key key;
	struct btrfs_disk_key found_key;
	int ret;
2949

2950
	btrfs_item_key_to_cpu(path->nodes[0], &key, 0);
2951

2952 2953 2954 2955 2956 2957 2958 2959
	if (key.offset > 0)
		key.offset--;
	else if (key.type > 0)
		key.type--;
	else if (key.objectid > 0)
		key.objectid--;
	else
		return 1;
2960

2961 2962 2963 2964 2965 2966 2967 2968 2969
	btrfs_release_path(root, path);
	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;
2970 2971
}

2972 2973 2974 2975 2976 2977 2978 2979 2980 2981 2982 2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998 2999 3000 3001 3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031 3032 3033 3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056 3057 3058 3059 3060 3061 3062 3063 3064 3065 3066 3067 3068 3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079 3080 3081 3082 3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095 3096 3097 3098 3099 3100
/*
 * 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
 * transaction id.  This is used by the btree defrag code, but could
 * also be used to search for blocks that have changed since a given
 * transaction id.
 *
 * 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.
 *
 * 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,
			 struct btrfs_path *path, int cache_only,
			 u64 min_trans)
{
	struct extent_buffer *cur;
	struct btrfs_key found_key;
	int slot;
	u32 nritems;
	int level;
	int ret = 1;

again:
	cur = btrfs_lock_root_node(root);
	level = btrfs_header_level(cur);
	path->nodes[level] = cur;
	path->locks[level] = 1;

	if (btrfs_header_generation(cur) < min_trans) {
		ret = 1;
		goto out;
	}
	while(1) {
		nritems = btrfs_header_nritems(cur);
		level = btrfs_header_level(cur);
		bin_search(cur, min_key, level, &slot);

		/* at level = 0, we're done, setup the path and exit */
		if (level == 0) {
			ret = 0;
			path->slots[level] = slot;
			btrfs_item_key_to_cpu(cur, &found_key, slot);
			goto out;
		}
		/*
		 * 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.
		 */
		while(slot < nritems) {
			u64 blockptr;
			u64 gen;
			struct extent_buffer *tmp;
			blockptr = btrfs_node_blockptr(cur, slot);
			gen = btrfs_node_ptr_generation(cur, slot);
			if (gen < min_trans) {
				slot++;
				continue;
			}
			if (!cache_only)
				break;

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

			if (tmp && btrfs_buffer_uptodate(tmp, gen)) {
				free_extent_buffer(tmp);
				break;
			}
			if (tmp)
				free_extent_buffer(tmp);
			slot++;
		}
		/*
		 * we didn't find a candidate key in this node, walk forward
		 * and find another one
		 */
		if (slot >= nritems) {
			ret = btrfs_find_next_key(root, path, min_key, level,
						  cache_only, min_trans);
			if (ret == 0) {
				btrfs_release_path(root, path);
				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;
			unlock_up(path, level, 1);
			goto out;
		}
		cur = read_node_slot(root, cur, slot);

		btrfs_tree_lock(cur);
		path->locks[level - 1] = 1;
		path->nodes[level - 1] = cur;
		unlock_up(path, level, 1);
	}
out:
	if (ret == 0)
		memcpy(min_key, &found_key, sizeof(found_key));
	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.
 */
3101
int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path,
3102 3103
			struct btrfs_key *key, int lowest_level,
			int cache_only, u64 min_trans)
3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114
{
	int level = lowest_level;
	int slot;
	struct extent_buffer *c;

	while(level < BTRFS_MAX_LEVEL) {
		if (!path->nodes[level])
			return 1;

		slot = path->slots[level] + 1;
		c = path->nodes[level];
3115
next:
3116 3117 3118 3119 3120 3121 3122 3123 3124
		if (slot >= btrfs_header_nritems(c)) {
			level++;
			if (level == BTRFS_MAX_LEVEL) {
				return 1;
			}
			continue;
		}
		if (level == 0)
			btrfs_item_key_to_cpu(c, key, slot);
3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144
		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));
				if (!cur || !btrfs_buffer_uptodate(cur, gen)) {
					slot++;
					if (cur)
						free_extent_buffer(cur);
					goto next;
				}
				free_extent_buffer(cur);
			}
			if (gen < min_trans) {
				slot++;
				goto next;
			}
3145
			btrfs_node_key_to_cpu(c, key, slot);
3146
		}
3147 3148 3149 3150 3151
		return 0;
	}
	return 1;
}

C
Chris Mason 已提交
3152
/*
3153
 * search the tree again to find a leaf with greater keys
C
Chris Mason 已提交
3154 3155
 * returns 0 if it found something or 1 if there are no greater leaves.
 * returns < 0 on io errors.
C
Chris Mason 已提交
3156
 */
C
Chris Mason 已提交
3157
int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
3158 3159 3160
{
	int slot;
	int level = 1;
3161 3162
	struct extent_buffer *c;
	struct extent_buffer *next = NULL;
3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173 3174
	struct btrfs_key key;
	u32 nritems;
	int ret;

	nritems = btrfs_header_nritems(path->nodes[0]);
	if (nritems == 0) {
		return 1;
	}

	btrfs_item_key_to_cpu(path->nodes[0], &key, nritems - 1);

	btrfs_release_path(root, path);
3175
	path->keep_locks = 1;
3176 3177 3178 3179 3180 3181
	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
	path->keep_locks = 0;

	if (ret < 0)
		return ret;

3182
	nritems = btrfs_header_nritems(path->nodes[0]);
3183 3184 3185 3186 3187 3188
	/*
	 * 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.
	 */
3189
	if (nritems > 0 && path->slots[0] < nritems - 1) {
3190
		path->slots[0]++;
3191 3192
		goto done;
	}
3193

C
Chris Mason 已提交
3194
	while(level < BTRFS_MAX_LEVEL) {
3195
		if (!path->nodes[level])
C
Chris Mason 已提交
3196
			return 1;
3197

3198 3199
		slot = path->slots[level] + 1;
		c = path->nodes[level];
3200
		if (slot >= btrfs_header_nritems(c)) {
3201
			level++;
3202
			if (level == BTRFS_MAX_LEVEL) {
3203
				return 1;
3204
			}
3205 3206
			continue;
		}
3207

3208 3209
		if (next) {
			btrfs_tree_unlock(next);
3210
			free_extent_buffer(next);
3211
		}
3212

3213
		if (level == 1 && path->locks[1] && path->reada)
3214
			reada_for_search(root, path, level, slot, 0);
3215

3216
		next = read_node_slot(root, c, slot);
3217 3218 3219 3220
		if (!path->skip_locking) {
			WARN_ON(!btrfs_tree_locked(c));
			btrfs_tree_lock(next);
		}
3221 3222 3223 3224 3225 3226
		break;
	}
	path->slots[level] = slot;
	while(1) {
		level--;
		c = path->nodes[level];
3227 3228
		if (path->locks[level])
			btrfs_tree_unlock(c);
3229
		free_extent_buffer(c);
3230 3231
		path->nodes[level] = next;
		path->slots[level] = 0;
3232 3233
		if (!path->skip_locking)
			path->locks[level] = 1;
3234 3235
		if (!level)
			break;
3236 3237
		if (level == 1 && path->locks[1] && path->reada)
			reada_for_search(root, path, level, slot, 0);
3238
		next = read_node_slot(root, next, 0);
3239 3240 3241 3242
		if (!path->skip_locking) {
			WARN_ON(!btrfs_tree_locked(path->nodes[level]));
			btrfs_tree_lock(next);
		}
3243
	}
3244 3245
done:
	unlock_up(path, 0, 1);
3246 3247
	return 0;
}
3248

3249 3250 3251 3252 3253 3254
/*
 * 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
 */
3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278
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;
	int ret;

	while(1) {
		if (path->slots[0] == 0) {
			ret = btrfs_prev_leaf(root, path);
			if (ret != 0)
				return ret;
		} else {
			path->slots[0]--;
		}
		leaf = path->nodes[0];
		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
		if (found_key.type == type)
			return 0;
	}
	return 1;
}