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
		!mutex_is_locked(&root->fs_info->alloc_mutex));
	if (ins_len < 0)
		lowest_unlock = 2;
1256
again:
1257 1258 1259 1260
	if (p->skip_locking)
		b = btrfs_root_node(root);
	else
		b = btrfs_lock_root_node(root);
1261

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

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

1314
			/* this is only true while dropping a snapshot */
1315
			if (level == lowest_level) {
1316
				break;
1317
			}
1318

1319 1320 1321 1322 1323 1324
			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)) {
1325 1326 1327 1328 1329 1330 1331 1332 1333
				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);
1334 1335
					if (tmp)
						free_extent_buffer(tmp);
1336 1337 1338 1339 1340
					if (should_reada)
						reada_for_search(root, p,
								 level, slot,
								 key->objectid);

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

	btrfs_set_node_ptr_generation(c, 0, lower_gen);
1589

1590
	btrfs_mark_buffer_dirty(c);
1591

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

2213 2214 2215
	if (extend)
		space_needed = data_size;

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

2436
	old_data_start = btrfs_item_offset_nr(leaf, slot);
2437

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

2812
	nritems = btrfs_header_nritems(leaf);
2813

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

	if (ret < 0)
		return ret;

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

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

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

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

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

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

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