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

19
#include <linux/sched.h>
20 21
#include "ctree.h"
#include "disk-io.h"
22
#include "transaction.h"
23
#include "print-tree.h"
24
#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
			/* this is only true while dropping a snapshot */
1317 1318
			if (level == lowest_level) {
				unlock_up(p, level, lowest_unlock);
1319
				break;
1320
			}
1321

1322
			if (should_reada)
1323 1324
				reada_for_search(root, p, level, slot,
						 key->objectid);
1325

1326 1327 1328 1329 1330 1331
			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)) {
1332 1333 1334 1335 1336 1337 1338 1339 1340
				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);
1341 1342 1343 1344
					if (tmp)
						free_extent_buffer(tmp);
					tmp = read_tree_block(root, blocknr,
							 blocksize, gen);
1345 1346 1347 1348
					if (tmp)
						free_extent_buffer(tmp);
					goto again;
				} else {
1349 1350
					if (tmp)
						free_extent_buffer(tmp);
1351 1352 1353
					b = read_node_slot(root, b, slot);
				}
			}
1354 1355
			if (!p->skip_locking)
				btrfs_tree_lock(b);
1356
			unlock_up(p, level, lowest_unlock);
1357 1358
		} else {
			p->slots[level] = slot;
1359
			if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
C
Chris Mason 已提交
1360
			    sizeof(struct btrfs_item) + ins_len) {
1361
				int sret = split_leaf(trans, root, key,
1362
						      p, ins_len, ret == 0);
C
Chris Mason 已提交
1363 1364 1365 1366
				BUG_ON(sret > 0);
				if (sret)
					return sret;
			}
1367
			unlock_up(p, level, lowest_unlock);
1368 1369 1370
			return ret;
		}
	}
C
Chris Mason 已提交
1371
	return 1;
1372 1373
}

C
Chris Mason 已提交
1374 1375 1376 1377 1378 1379
/*
 * 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 已提交
1380 1381 1382
 *
 * 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 已提交
1383
 */
1384 1385 1386
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)
1387 1388
{
	int i;
C
Chris Mason 已提交
1389
	int ret = 0;
1390 1391
	struct extent_buffer *t;

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

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

1421 1422
	src_nritems = btrfs_header_nritems(src);
	dst_nritems = btrfs_header_nritems(dst);
C
Chris Mason 已提交
1423
	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
1424 1425
	WARN_ON(btrfs_header_generation(src) != trans->transid);
	WARN_ON(btrfs_header_generation(dst) != trans->transid);
1426

1427
	if (!empty && src_nritems <= 8)
1428 1429
		return 1;

1430
	if (push_items <= 0) {
1431
		return 1;
1432
	}
1433

1434
	if (empty) {
1435
		push_items = min(src_nritems, push_items);
1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447
		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);
1448

1449 1450 1451 1452 1453
	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));

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

1487 1488 1489
	WARN_ON(btrfs_header_generation(src) != trans->transid);
	WARN_ON(btrfs_header_generation(dst) != trans->transid);

1490 1491
	src_nritems = btrfs_header_nritems(src);
	dst_nritems = btrfs_header_nritems(dst);
C
Chris Mason 已提交
1492
	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
1493
	if (push_items <= 0) {
1494
		return 1;
1495 1496 1497 1498 1499
	}

	if (src_nritems < 4) {
		return 1;
	}
1500 1501 1502

	max_push = src_nritems / 2 + 1;
	/* don't try to empty the node */
1503
	if (max_push >= src_nritems) {
1504
		return 1;
1505
	}
Y
Yan 已提交
1506

1507 1508 1509
	if (max_push < push_items)
		push_items = max_push;

1510 1511 1512 1513
	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 已提交
1514

1515 1516 1517 1518
	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));
1519

1520 1521
	btrfs_set_header_nritems(src, src_nritems - push_items);
	btrfs_set_header_nritems(dst, dst_nritems + push_items);
1522

1523 1524
	btrfs_mark_buffer_dirty(src);
	btrfs_mark_buffer_dirty(dst);
C
Chris Mason 已提交
1525
	return ret;
1526 1527
}

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

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

1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559
	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);

1560
	c = btrfs_alloc_free_block(trans, root, root->nodesize,
1561 1562
				   root->root_key.objectid,
				   root_gen, lower_key.objectid, level,
1563
				   root->node->start, 0);
1564 1565
	if (IS_ERR(c))
		return PTR_ERR(c);
1566

1567 1568 1569
	memset_extent_buffer(c, 0, 0, root->nodesize);
	btrfs_set_header_nritems(c, 1);
	btrfs_set_header_level(c, level);
1570
	btrfs_set_header_bytenr(c, c->start);
1571 1572 1573 1574 1575 1576
	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);
1577 1578 1579 1580 1581

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

1582
	btrfs_set_node_key(c, &lower_key, 0);
1583
	btrfs_set_node_blockptr(c, 0, lower->start);
1584 1585 1586 1587
	lower_gen = btrfs_header_generation(lower);
	WARN_ON(lower_gen == 0);

	btrfs_set_node_ptr_generation(c, 0, lower_gen);
1588

1589
	btrfs_mark_buffer_dirty(c);
1590

1591 1592
	spin_lock(&root->node_lock);
	old = root->node;
1593
	root->node = c;
1594 1595 1596 1597 1598
	spin_unlock(&root->node_lock);

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

1599
	add_root_to_dirty_list(root);
1600 1601
	extent_buffer_get(c);
	path->nodes[level] = c;
1602
	path->locks[level] = 1;
C
Chris Mason 已提交
1603
	path->slots[level] = 0;
1604 1605 1606 1607

	if (root->ref_cows && lower_gen != trans->transid) {
		struct btrfs_path *back_path = btrfs_alloc_path();
		int ret;
1608
		mutex_lock(&root->fs_info->alloc_mutex);
1609 1610 1611 1612 1613 1614
		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);
1615
		mutex_unlock(&root->fs_info->alloc_mutex);
1616 1617
		btrfs_free_path(back_path);
	}
C
Chris Mason 已提交
1618 1619 1620
	return 0;
}

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

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

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

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

1697
	c_nritems = btrfs_header_nritems(c);
1698 1699 1700 1701 1702 1703
	if (root->ref_cows)
		root_gen = trans->transid;
	else
		root_gen = 0;

	btrfs_node_key(c, &disk_key, 0);
1704
	split = btrfs_alloc_free_block(trans, root, root->nodesize,
1705 1706 1707 1708
					 root->root_key.objectid,
					 root_gen,
					 btrfs_disk_key_objectid(&disk_key),
					 level, c->start, 0);
1709 1710 1711 1712 1713
	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));
1714
	btrfs_set_header_bytenr(split, split->start);
1715 1716
	btrfs_set_header_generation(split, trans->transid);
	btrfs_set_header_owner(split, root->root_key.objectid);
1717
	btrfs_set_header_flags(split, 0);
1718 1719 1720
	write_extent_buffer(split, root->fs_info->fsid,
			    (unsigned long)btrfs_header_fsid(split),
			    BTRFS_FSID_SIZE);
1721 1722 1723
	write_extent_buffer(split, root->fs_info->chunk_tree_uuid,
			    (unsigned long)btrfs_header_chunk_tree_uuid(split),
			    BTRFS_UUID_SIZE);
1724

1725
	mid = (c_nritems + 1) / 2;
1726 1727 1728 1729 1730 1731 1732

	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 已提交
1733 1734
	ret = 0;

1735 1736 1737 1738
	btrfs_mark_buffer_dirty(c);
	btrfs_mark_buffer_dirty(split);

	btrfs_node_key(split, &disk_key, 0);
1739
	wret = insert_ptr(trans, root, path, &disk_key, split->start,
1740
			  path->slots[level + 1] + 1,
C
Chris Mason 已提交
1741
			  level + 1);
C
Chris Mason 已提交
1742 1743 1744
	if (wret)
		ret = wret;

C
Chris Mason 已提交
1745
	if (path->slots[level] >= mid) {
C
Chris Mason 已提交
1746
		path->slots[level] -= mid;
1747
		btrfs_tree_unlock(c);
1748 1749
		free_extent_buffer(c);
		path->nodes[level] = split;
C
Chris Mason 已提交
1750 1751
		path->slots[level + 1] += 1;
	} else {
1752
		btrfs_tree_unlock(split);
1753
		free_extent_buffer(split);
1754
	}
C
Chris Mason 已提交
1755
	return ret;
1756 1757
}

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

	if (!nr)
		return 0;
1771 1772
	data_len = btrfs_item_end_nr(l, start);
	data_len = data_len - btrfs_item_offset_nr(l, end);
C
Chris Mason 已提交
1773
	data_len += sizeof(struct btrfs_item) * nr;
1774
	WARN_ON(data_len < 0);
1775 1776 1777
	return data_len;
}

1778 1779 1780 1781 1782
/*
 * 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
 */
1783
int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
1784
{
1785 1786 1787 1788 1789
	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 已提交
1790
		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
1791 1792 1793
		       leaf_space_used(leaf, 0, nritems), nritems);
	}
	return ret;
1794 1795
}

C
Chris Mason 已提交
1796 1797 1798
/*
 * 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 已提交
1799 1800 1801
 *
 * 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 已提交
1802
 */
1803
static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
1804 1805
			   *root, struct btrfs_path *path, int data_size,
			   int empty)
C
Chris Mason 已提交
1806
{
1807 1808 1809 1810
	struct extent_buffer *left = path->nodes[0];
	struct extent_buffer *right;
	struct extent_buffer *upper;
	struct btrfs_disk_key disk_key;
C
Chris Mason 已提交
1811
	int slot;
1812
	u32 i;
C
Chris Mason 已提交
1813 1814 1815
	int free_space;
	int push_space = 0;
	int push_items = 0;
C
Chris Mason 已提交
1816
	struct btrfs_item *item;
1817
	u32 left_nritems;
1818
	u32 nr;
1819
	u32 right_nritems;
1820
	u32 data_end;
1821
	u32 this_item_size;
1822
	int ret;
C
Chris Mason 已提交
1823 1824 1825 1826 1827 1828

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

1832 1833
	WARN_ON(!btrfs_tree_locked(path->nodes[1]));

1834
	right = read_node_slot(root, upper, slot + 1);
1835
	btrfs_tree_lock(right);
C
Chris Mason 已提交
1836
	free_space = btrfs_leaf_free_space(root, right);
1837 1838
	if (free_space < data_size + sizeof(struct btrfs_item))
		goto out_unlock;
1839

C
Chris Mason 已提交
1840
	/* cow and double check */
1841 1842
	ret = btrfs_cow_block(trans, root, right, upper,
			      slot + 1, &right);
1843 1844 1845
	if (ret)
		goto out_unlock;

C
Chris Mason 已提交
1846
	free_space = btrfs_leaf_free_space(root, right);
1847 1848
	if (free_space < data_size + sizeof(struct btrfs_item))
		goto out_unlock;
C
Chris Mason 已提交
1849

1850
	left_nritems = btrfs_header_nritems(left);
1851 1852
	if (left_nritems == 0)
		goto out_unlock;
1853

1854 1855 1856 1857 1858 1859 1860
	if (empty)
		nr = 0;
	else
		nr = 1;

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

C
Chris Mason 已提交
1863 1864
		if (path->slots[0] == i)
			push_space += data_size + sizeof(*item);
1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875

		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 已提交
1876 1877
			break;
		push_items++;
1878
		push_space += this_item_size + sizeof(*item);
1879 1880 1881
		if (i == 0)
			break;
		i--;
1882 1883 1884 1885
	}
	if (left->map_token) {
		unmap_extent_buffer(left, left->map_token, KM_USER1);
		left->map_token = NULL;
C
Chris Mason 已提交
1886
	}
1887

1888 1889
	if (push_items == 0)
		goto out_unlock;
1890

1891
	if (!empty && push_items == left_nritems)
1892
		WARN_ON(1);
1893

C
Chris Mason 已提交
1894
	/* push left to right */
1895
	right_nritems = btrfs_header_nritems(right);
1896

1897
	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
C
Chris Mason 已提交
1898
	push_space -= leaf_data_end(root, left);
1899

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

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

C
Chris Mason 已提交
1917
	/* copy the items from left to right */
1918 1919 1920
	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 已提交
1921 1922

	/* update the item pointers */
1923
	right_nritems += push_items;
1924
	btrfs_set_header_nritems(right, right_nritems);
C
Chris Mason 已提交
1925
	push_space = BTRFS_LEAF_DATA_SIZE(root);
1926
	for (i = 0; i < right_nritems; i++) {
1927
		item = btrfs_item_nr(right, i);
1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941
		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 已提交
1942
	}
1943
	left_nritems -= push_items;
1944
	btrfs_set_header_nritems(left, left_nritems);
C
Chris Mason 已提交
1945

1946 1947
	if (left_nritems)
		btrfs_mark_buffer_dirty(left);
1948
	btrfs_mark_buffer_dirty(right);
1949

1950 1951
	btrfs_item_key(right, &disk_key, 0);
	btrfs_set_node_key(upper, &disk_key, slot + 1);
C
Chris Mason 已提交
1952
	btrfs_mark_buffer_dirty(upper);
C
Chris Mason 已提交
1953

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

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

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

	slot = path->slots[1];
2001
	if (slot == 0)
2002
		return 1;
2003
	if (!path->nodes[1])
2004
		return 1;
2005

2006 2007 2008 2009 2010
	right_nritems = btrfs_header_nritems(right);
	if (right_nritems == 0) {
		return 1;
	}

2011 2012
	WARN_ON(!btrfs_tree_locked(path->nodes[1]));

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

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

C
Chris Mason 已提交
2030
	free_space = btrfs_leaf_free_space(root, left);
C
Chris Mason 已提交
2031
	if (free_space < data_size + sizeof(struct btrfs_item)) {
2032 2033
		ret = 1;
		goto out;
C
Chris Mason 已提交
2034 2035
	}

2036 2037 2038 2039 2040 2041
	if (empty)
		nr = right_nritems;
	else
		nr = right_nritems - 1;

	for (i = 0; i < nr; i++) {
2042
		item = btrfs_item_nr(right, i);
2043 2044 2045 2046 2047 2048 2049 2050
		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);
		}

2051 2052
		if (path->slots[0] == i)
			push_space += data_size + sizeof(*item);
2053 2054 2055

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

2058
		push_items++;
2059 2060 2061 2062 2063 2064
		push_space += this_item_size + sizeof(*item);
	}

	if (right->map_token) {
		unmap_extent_buffer(right, right->map_token, KM_USER1);
		right->map_token = NULL;
2065
	}
2066

2067
	if (push_items == 0) {
2068 2069
		ret = 1;
		goto out;
2070
	}
2071
	if (!empty && push_items == btrfs_header_nritems(right))
2072
		WARN_ON(1);
2073

2074
	/* push data from right to left */
2075 2076 2077 2078 2079
	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 已提交
2080
	push_space = BTRFS_LEAF_DATA_SIZE(root) -
2081 2082 2083
		     btrfs_item_offset_nr(right, push_items -1);

	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
C
Chris Mason 已提交
2084 2085
		     leaf_data_end(root, left) - push_space,
		     btrfs_leaf_data(right) +
2086
		     btrfs_item_offset_nr(right, push_items - 1),
C
Chris Mason 已提交
2087
		     push_space);
2088
	old_left_nritems = btrfs_header_nritems(left);
2089 2090
	BUG_ON(old_left_nritems < 0);

2091
	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
C
Chris Mason 已提交
2092
	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
2093
		u32 ioff;
2094

2095
		item = btrfs_item_nr(left, i);
2096 2097 2098 2099 2100 2101 2102 2103
		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);
		}

2104 2105
		ioff = btrfs_item_offset(left, item);
		btrfs_set_item_offset(left, item,
2106
		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
2107
	}
2108
	btrfs_set_header_nritems(left, old_left_nritems + push_items);
2109 2110 2111 2112
	if (left->map_token) {
		unmap_extent_buffer(left, left->map_token, KM_USER1);
		left->map_token = NULL;
	}
2113 2114

	/* fixup right node */
2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128
	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),
2129 2130 2131
			      btrfs_item_nr_offset(push_items),
			     (btrfs_header_nritems(right) - push_items) *
			     sizeof(struct btrfs_item));
2132
	}
2133 2134
	right_nritems -= push_items;
	btrfs_set_header_nritems(right, right_nritems);
C
Chris Mason 已提交
2135
	push_space = BTRFS_LEAF_DATA_SIZE(root);
2136 2137
	for (i = 0; i < right_nritems; i++) {
		item = btrfs_item_nr(right, i);
2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152

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

2155
	btrfs_mark_buffer_dirty(left);
2156 2157
	if (right_nritems)
		btrfs_mark_buffer_dirty(right);
2158

2159 2160
	btrfs_item_key(right, &disk_key, 0);
	wret = fixup_low_keys(trans, root, path, &disk_key, 1);
C
Chris Mason 已提交
2161 2162
	if (wret)
		ret = wret;
2163 2164 2165 2166

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

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

2212 2213 2214
	if (extend)
		space_needed = data_size;

2215 2216 2217 2218 2219
	if (root->ref_cows)
		root_gen = trans->transid;
	else
		root_gen = 0;

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

2233
		/* did the pushes work? */
2234
		if (btrfs_leaf_free_space(root, l) >= space_needed)
2235
			return 0;
2236
	}
C
Chris Mason 已提交
2237

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

2250 2251
	btrfs_item_key(l, &disk_key, 0);

2252
	right = btrfs_alloc_free_block(trans, root, root->leafsize,
2253 2254 2255
					 root->root_key.objectid,
					 root_gen, disk_key.objectid, 0,
					 l->start, 0);
2256 2257
	if (IS_ERR(right)) {
		BUG_ON(1);
2258
		return PTR_ERR(right);
2259
	}
2260 2261

	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
2262
	btrfs_set_header_bytenr(right, right->start);
2263 2264 2265 2266 2267 2268
	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);
2269 2270 2271 2272

	write_extent_buffer(right, root->fs_info->chunk_tree_uuid,
			    (unsigned long)btrfs_header_chunk_tree_uuid(right),
			    BTRFS_UUID_SIZE);
2273 2274 2275 2276 2277 2278
	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);
2279
				btrfs_set_header_nritems(right, 0);
2280
				wret = insert_ptr(trans, root, path,
2281
						  &disk_key, right->start,
2282 2283 2284
						  path->slots[1] + 1, 1);
				if (wret)
					ret = wret;
2285 2286

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

C
Chris Mason 已提交
2350
	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
2351
		      btrfs_item_end_nr(l, mid);
C
Chris Mason 已提交
2352

2353 2354
	for (i = 0; i < nritems; i++) {
		struct btrfs_item *item = btrfs_item_nr(right, i);
2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365
		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);
2366
		btrfs_set_item_offset(right, item, ioff + rt_data_off);
C
Chris Mason 已提交
2367
	}
C
Chris Mason 已提交
2368

2369 2370 2371 2372 2373
	if (right->map_token) {
		unmap_extent_buffer(right, right->map_token, KM_USER1);
		right->map_token = NULL;
	}

2374
	btrfs_set_header_nritems(l, mid);
C
Chris Mason 已提交
2375
	ret = 0;
2376
	btrfs_item_key(right, &disk_key, 0);
2377 2378
	wret = insert_ptr(trans, root, path, &disk_key, right->start,
			  path->slots[1] + 1, 1);
C
Chris Mason 已提交
2379 2380
	if (wret)
		ret = wret;
2381 2382 2383

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

2386
	if (mid <= slot) {
2387
		btrfs_tree_unlock(path->nodes[0]);
2388 2389
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = right;
2390 2391
		path->slots[0] -= mid;
		path->slots[1] += 1;
2392 2393
	} else {
		btrfs_tree_unlock(right);
2394
		free_extent_buffer(right);
2395
	}
2396

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

2399 2400 2401 2402
	if (double_split) {
		BUG_ON(num_doubles != 0);
		num_doubles++;
		goto again;
2403
	}
2404 2405 2406
	return ret;
}

C
Chris Mason 已提交
2407 2408 2409
int btrfs_truncate_item(struct btrfs_trans_handle *trans,
			struct btrfs_root *root,
			struct btrfs_path *path,
2410
			u32 new_size, int from_end)
C
Chris Mason 已提交
2411 2412 2413 2414
{
	int ret = 0;
	int slot;
	int slot_orig;
2415 2416
	struct extent_buffer *leaf;
	struct btrfs_item *item;
C
Chris Mason 已提交
2417 2418 2419 2420 2421 2422 2423 2424
	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];
2425
	leaf = path->nodes[0];
2426 2427 2428 2429 2430
	slot = path->slots[0];

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

2432
	nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
2433 2434
	data_end = leaf_data_end(root, leaf);

2435
	old_data_start = btrfs_item_offset_nr(leaf, slot);
2436

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

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

2458 2459
		ioff = btrfs_item_offset(leaf, item);
		btrfs_set_item_offset(leaf, item, ioff + size_diff);
C
Chris Mason 已提交
2460
	}
2461 2462 2463 2464 2465 2466

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

C
Chris Mason 已提交
2467
	/* shift the data */
2468 2469 2470 2471 2472 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
	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);
	}
2507 2508 2509 2510

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

	ret = 0;
2513 2514
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
C
Chris Mason 已提交
2515
		BUG();
2516
	}
C
Chris Mason 已提交
2517 2518 2519
	return ret;
}

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

	slot_orig = path->slots[0];
2536
	leaf = path->nodes[0];
2537

2538
	nritems = btrfs_header_nritems(leaf);
2539 2540
	data_end = leaf_data_end(root, leaf);

2541 2542
	if (btrfs_leaf_free_space(root, leaf) < data_size) {
		btrfs_print_leaf(root, leaf);
2543
		BUG();
2544
	}
2545
	slot = path->slots[0];
2546
	old_data = btrfs_item_end_nr(leaf, slot);
2547 2548

	BUG_ON(slot < 0);
2549 2550 2551 2552 2553
	if (slot >= nritems) {
		btrfs_print_leaf(root, leaf);
		printk("slot %d too large, nritems %d\n", slot, nritems);
		BUG_ON(1);
	}
2554 2555 2556 2557 2558 2559

	/*
	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
	 */
	/* first correct the data pointers */
	for (i = slot; i < nritems; i++) {
2560 2561
		u32 ioff;
		item = btrfs_item_nr(leaf, i);
2562 2563 2564 2565 2566 2567 2568 2569

		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);
		}
2570 2571
		ioff = btrfs_item_offset(leaf, item);
		btrfs_set_item_offset(leaf, item, ioff - data_size);
2572
	}
2573

2574 2575 2576 2577 2578
	if (leaf->map_token) {
		unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
		leaf->map_token = NULL;
	}

2579
	/* shift the data */
2580
	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2581 2582
		      data_end - data_size, btrfs_leaf_data(leaf) +
		      data_end, old_data - data_end);
2583

2584
	data_end = old_data;
2585 2586 2587 2588
	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);
2589 2590

	ret = 0;
2591 2592
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
2593
		BUG();
2594
	}
2595 2596 2597
	return ret;
}

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

2620 2621 2622
	for (i = 0; i < nr; i++) {
		total_data += data_size[i];
	}
2623

2624 2625
	total_size = total_data + (nr - 1) * sizeof(struct btrfs_item);
	ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1);
2626
	if (ret == 0) {
2627
		return -EEXIST;
C
Chris Mason 已提交
2628
	}
2629 2630
	if (ret < 0)
		goto out;
2631

2632
	slot_orig = path->slots[0];
2633
	leaf = path->nodes[0];
C
Chris Mason 已提交
2634

2635
	nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
2636
	data_end = leaf_data_end(root, leaf);
2637

C
Chris Mason 已提交
2638
	if (btrfs_leaf_free_space(root, leaf) <
2639
	    sizeof(struct btrfs_item) + total_size) {
2640 2641
		btrfs_print_leaf(root, leaf);
		printk("not enough freespace need %u have %d\n",
2642
		       total_size, btrfs_leaf_free_space(root, leaf));
2643
		BUG();
2644
	}
2645

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

2649 2650
	if (slot != nritems) {
		int i;
2651
		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
2652

2653 2654 2655 2656 2657 2658
		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);
		}
2659 2660 2661 2662
		/*
		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
		 */
		/* first correct the data pointers */
2663
		WARN_ON(leaf->map_token);
C
Chris Mason 已提交
2664
		for (i = slot; i < nritems; i++) {
2665
			u32 ioff;
2666

2667
			item = btrfs_item_nr(leaf, i);
2668 2669 2670 2671 2672 2673 2674 2675
			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);
			}

2676
			ioff = btrfs_item_offset(leaf, item);
2677
			btrfs_set_item_offset(leaf, item, ioff - total_data);
C
Chris Mason 已提交
2678
		}
2679 2680 2681 2682
		if (leaf->map_token) {
			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
			leaf->map_token = NULL;
		}
2683 2684

		/* shift the items */
2685
		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr),
2686
			      btrfs_item_nr_offset(slot),
C
Chris Mason 已提交
2687
			      (nritems - slot) * sizeof(struct btrfs_item));
2688 2689

		/* shift the data */
2690
		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2691
			      data_end - total_data, btrfs_leaf_data(leaf) +
C
Chris Mason 已提交
2692
			      data_end, old_data - data_end);
2693 2694
		data_end = old_data;
	}
2695

2696
	/* setup the item for the new data */
2697 2698 2699 2700 2701 2702 2703 2704 2705
	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);
2706
	btrfs_mark_buffer_dirty(leaf);
C
Chris Mason 已提交
2707 2708

	ret = 0;
2709 2710
	if (slot == 0) {
		btrfs_cpu_key_to_disk(&disk_key, cpu_key);
2711
		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
2712
	}
C
Chris Mason 已提交
2713

2714 2715
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
2716
		BUG();
2717
	}
2718
out:
2719 2720 2721 2722 2723 2724 2725
	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.
 */
2726 2727 2728
int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_key *cpu_key, void *data, u32
		      data_size)
2729 2730
{
	int ret = 0;
C
Chris Mason 已提交
2731
	struct btrfs_path *path;
2732 2733
	struct extent_buffer *leaf;
	unsigned long ptr;
2734

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

C
Chris Mason 已提交
2748
/*
C
Chris Mason 已提交
2749
 * delete the pointer from a given node.
C
Chris Mason 已提交
2750 2751 2752 2753 2754
 *
 * 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.
 */
2755 2756
static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
		   struct btrfs_path *path, int level, int slot)
2757
{
2758
	struct extent_buffer *parent = path->nodes[level];
2759
	u32 nritems;
C
Chris Mason 已提交
2760
	int ret = 0;
2761
	int wret;
2762

2763
	nritems = btrfs_header_nritems(parent);
2764
	if (slot != nritems -1) {
2765 2766 2767
		memmove_extent_buffer(parent,
			      btrfs_node_key_ptr_offset(slot),
			      btrfs_node_key_ptr_offset(slot + 1),
C
Chris Mason 已提交
2768 2769
			      sizeof(struct btrfs_key_ptr) *
			      (nritems - slot - 1));
2770
	}
2771
	nritems--;
2772
	btrfs_set_header_nritems(parent, nritems);
2773
	if (nritems == 0 && parent == root->node) {
2774
		BUG_ON(btrfs_header_level(root->node) != 1);
2775
		/* just turn the root into a leaf and break */
2776
		btrfs_set_header_level(root->node, 0);
2777
	} else if (slot == 0) {
2778 2779 2780 2781
		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 已提交
2782 2783
		if (wret)
			ret = wret;
2784
	}
C
Chris Mason 已提交
2785
	btrfs_mark_buffer_dirty(parent);
C
Chris Mason 已提交
2786
	return ret;
2787 2788
}

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

2805
	leaf = path->nodes[0];
2806 2807 2808 2809 2810
	last_off = btrfs_item_offset_nr(leaf, slot + nr - 1);

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

2811
	nritems = btrfs_header_nritems(leaf);
2812

2813
	if (slot + nr != nritems) {
2814
		int i;
C
Chris Mason 已提交
2815
		int data_end = leaf_data_end(root, leaf);
2816 2817

		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
C
Chris Mason 已提交
2818 2819
			      data_end + dsize,
			      btrfs_leaf_data(leaf) + data_end,
2820
			      last_off - data_end);
2821

2822
		for (i = slot + nr; i < nritems; i++) {
2823
			u32 ioff;
2824

2825
			item = btrfs_item_nr(leaf, i);
2826 2827 2828 2829 2830 2831 2832
			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);
			}
2833 2834
			ioff = btrfs_item_offset(leaf, item);
			btrfs_set_item_offset(leaf, item, ioff + dsize);
C
Chris Mason 已提交
2835
		}
2836 2837 2838 2839 2840 2841

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

2842
		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
2843
			      btrfs_item_nr_offset(slot + nr),
C
Chris Mason 已提交
2844
			      sizeof(struct btrfs_item) *
2845
			      (nritems - slot - nr));
2846
	}
2847 2848
	btrfs_set_header_nritems(leaf, nritems - nr);
	nritems -= nr;
2849

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

			btrfs_item_key(leaf, &disk_key, 0);
2872
			wret = fixup_low_keys(trans, root, path,
2873
					      &disk_key, 1);
C
Chris Mason 已提交
2874 2875 2876 2877
			if (wret)
				ret = wret;
		}

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

2887
			wret = push_leaf_left(trans, root, path, 1, 1);
2888
			if (wret < 0 && wret != -ENOSPC)
C
Chris Mason 已提交
2889
				ret = wret;
2890 2891 2892

			if (path->nodes[0] == leaf &&
			    btrfs_header_nritems(leaf)) {
2893
				wret = push_leaf_right(trans, root, path, 1, 1);
2894
				if (wret < 0 && wret != -ENOSPC)
C
Chris Mason 已提交
2895 2896
					ret = wret;
			}
2897 2898

			if (btrfs_header_nritems(leaf) == 0) {
2899
				u64 root_gen;
2900 2901
				u64 bytenr = leaf->start;
				u32 blocksize = leaf->len;
2902

2903 2904 2905
				root_gen = btrfs_header_generation(
							   path->nodes[1]);

2906
				wret = del_ptr(trans, root, path, 1, slot);
C
Chris Mason 已提交
2907 2908
				if (wret)
					ret = wret;
2909 2910

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

2934
/*
2935
 * search the tree again to find a leaf with lesser keys
2936 2937 2938 2939 2940
 * 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)
{
2941 2942 2943
	struct btrfs_key key;
	struct btrfs_disk_key found_key;
	int ret;
2944

2945
	btrfs_item_key_to_cpu(path->nodes[0], &key, 0);
2946

2947 2948 2949 2950 2951 2952 2953 2954
	if (key.offset > 0)
		key.offset--;
	else if (key.type > 0)
		key.type--;
	else if (key.objectid > 0)
		key.objectid--;
	else
		return 1;
2955

2956 2957 2958 2959 2960 2961 2962 2963 2964
	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;
2965 2966
}

2967 2968 2969 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
/*
 * 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.
 */
3096
int btrfs_find_next_key(struct btrfs_root *root, struct btrfs_path *path,
3097 3098
			struct btrfs_key *key, int lowest_level,
			int cache_only, u64 min_trans)
3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109
{
	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];
3110
next:
3111 3112 3113 3114 3115 3116 3117 3118 3119
		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);
3120 3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139
		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;
			}
3140
			btrfs_node_key_to_cpu(c, key, slot);
3141
		}
3142 3143 3144 3145 3146
		return 0;
	}
	return 1;
}

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

	if (ret < 0)
		return ret;

3177
	nritems = btrfs_header_nritems(path->nodes[0]);
3178 3179 3180 3181 3182 3183
	/*
	 * 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.
	 */
3184
	if (nritems > 0 && path->slots[0] < nritems - 1) {
3185
		path->slots[0]++;
3186 3187
		goto done;
	}
3188

C
Chris Mason 已提交
3189
	while(level < BTRFS_MAX_LEVEL) {
3190
		if (!path->nodes[level])
C
Chris Mason 已提交
3191
			return 1;
3192

3193 3194
		slot = path->slots[level] + 1;
		c = path->nodes[level];
3195
		if (slot >= btrfs_header_nritems(c)) {
3196
			level++;
3197
			if (level == BTRFS_MAX_LEVEL) {
3198
				return 1;
3199
			}
3200 3201
			continue;
		}
3202

3203 3204
		if (next) {
			btrfs_tree_unlock(next);
3205
			free_extent_buffer(next);
3206
		}
3207

3208
		if (level == 1 && path->locks[1] && path->reada)
3209
			reada_for_search(root, path, level, slot, 0);
3210

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

3244 3245 3246 3247 3248 3249
/*
 * 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
 */
3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273
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;
}