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

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

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

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

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

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

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

73 74 75 76 77 78 79 80
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);
	}
}

81 82 83 84 85 86 87 88 89 90
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;
91
	struct btrfs_root *new_root;
92

93 94 95 96 97 98
	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;
99 100 101 102 103 104 105 106 107 108 109 110 111 112 113

	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;
	}
114
	cow = __btrfs_alloc_free_block(trans, new_root, buf->len,
115 116 117
				       new_root_objectid,
				       trans->transid, first_key.objectid,
				       level, buf->start, 0);
118 119
	if (IS_ERR(cow)) {
		kfree(new_root);
120
		return PTR_ERR(cow);
121
	}
122 123 124 125 126

	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);
127
	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN);
128 129

	WARN_ON(btrfs_header_generation(buf) > trans->transid);
130 131 132
	ret = btrfs_inc_ref(trans, new_root, buf);
	kfree(new_root);

133 134 135 136 137 138 139 140 141
	if (ret)
		return ret;

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

int __btrfs_cow_block(struct btrfs_trans_handle *trans,
142 143 144 145 146
			     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 已提交
147
{
148
	u64 root_gen;
149
	struct extent_buffer *cow;
150
	u32 nritems;
151 152
	int ret = 0;
	int different_trans = 0;
153 154 155 156 157 158 159 160 161 162
	int level;
	struct btrfs_key first_key;

	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);
163
	WARN_ON(root->ref_cows && trans->transid != root->last_trans);
164

165 166 167 168 169 170 171 172 173 174 175 176 177
	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;
	}
	cow = __btrfs_alloc_free_block(trans, root, buf->len,
				     root->root_key.objectid,
				     root_gen, first_key.objectid, level,
178
				     search_start, empty_size);
179 180
	if (IS_ERR(cow))
		return PTR_ERR(cow);
181

182
	copy_extent_buffer(cow, buf, 0, 0, cow->len);
183
	btrfs_set_header_bytenr(cow, cow->start);
184 185
	btrfs_set_header_generation(cow, trans->transid);
	btrfs_set_header_owner(cow, root->root_key.objectid);
186
	btrfs_clear_header_flag(cow, BTRFS_HEADER_FLAG_WRITTEN);
187

188 189
	WARN_ON(btrfs_header_generation(buf) > trans->transid);
	if (btrfs_header_generation(buf) != trans->transid) {
190 191 192 193 194 195 196 197
		different_trans = 1;
		ret = btrfs_inc_ref(trans, root, buf);
		if (ret)
			return ret;
	} else {
		clean_tree_block(trans, root, buf);
	}

C
Chris Mason 已提交
198
	if (buf == root->node) {
199
		root_gen = btrfs_header_generation(buf);
C
Chris Mason 已提交
200
		root->node = cow;
201
		extent_buffer_get(cow);
C
Chris Mason 已提交
202
		if (buf != root->commit_root) {
203
			btrfs_free_extent(trans, root, buf->start,
204 205
					  buf->len, root->root_key.objectid,
					  root_gen, 0, 0, 1);
C
Chris Mason 已提交
206
		}
207
		free_extent_buffer(buf);
208
		add_root_to_dirty_list(root);
C
Chris Mason 已提交
209
	} else {
210
		root_gen = btrfs_header_generation(parent);
211
		btrfs_set_node_blockptr(parent, parent_slot,
212
					cow->start);
213 214 215
		WARN_ON(trans->transid == 0);
		btrfs_set_node_ptr_generation(parent, parent_slot,
					      trans->transid);
C
Chris Mason 已提交
216
		btrfs_mark_buffer_dirty(parent);
217
		WARN_ON(btrfs_header_generation(parent) != trans->transid);
218 219 220
		btrfs_free_extent(trans, root, buf->start, buf->len,
				  btrfs_header_owner(parent), root_gen,
				  0, 0, 1);
C
Chris Mason 已提交
221
	}
222
	free_extent_buffer(buf);
C
Chris Mason 已提交
223
	btrfs_mark_buffer_dirty(cow);
C
Chris Mason 已提交
224
	*cow_ret = cow;
C
Chris Mason 已提交
225 226 227
	return 0;
}

228 229 230 231
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)
232 233
{
	u64 search_start;
C
Chris Mason 已提交
234
	u64 header_trans;
235
	int ret;
C
Chris Mason 已提交
236

237 238 239 240 241 242 243 244 245 246
	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 已提交
247 248

	header_trans = btrfs_header_generation(buf);
249 250 251
	spin_lock(&root->fs_info->hash_lock);
	if (header_trans == trans->transid &&
	    !btrfs_header_flag(buf, BTRFS_HEADER_FLAG_WRITTEN)) {
252
		*cow_ret = buf;
253
		spin_unlock(&root->fs_info->hash_lock);
254 255
		return 0;
	}
256
	spin_unlock(&root->fs_info->hash_lock);
257
	search_start = buf->start & ~((u64)(1024 * 1024 * 1024) - 1);
258
	ret = __btrfs_cow_block(trans, root, buf, parent,
259
				 parent_slot, cow_ret, search_start, 0);
260
	return ret;
261 262
}

263
static int close_blocks(u64 blocknr, u64 other, u32 blocksize)
264
{
265
	if (blocknr < other && other - (blocknr + blocksize) < 32768)
266
		return 1;
267
	if (blocknr > other && blocknr - (other + blocksize) < 32768)
268 269 270 271
		return 1;
	return 0;
}

272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296
/*
 * 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;
}


297
int btrfs_realloc_node(struct btrfs_trans_handle *trans,
298
		       struct btrfs_root *root, struct extent_buffer *parent,
299 300
		       int start_slot, int cache_only, u64 *last_ret,
		       struct btrfs_key *progress)
301
{
302 303
	struct extent_buffer *cur;
	struct extent_buffer *tmp;
304
	u64 blocknr;
305
	u64 gen;
306 307
	u64 search_start = *last_ret;
	u64 last_block = 0;
308 309 310 311 312
	u64 other;
	u32 parent_nritems;
	int end_slot;
	int i;
	int err = 0;
313
	int parent_level;
314 315
	int uptodate;
	u32 blocksize;
316 317
	int progress_passed = 0;
	struct btrfs_disk_key disk_key;
318

319 320 321 322
	parent_level = btrfs_header_level(parent);
	if (cache_only && parent_level != 1)
		return 0;

323 324 325 326 327 328 329 330 331 332
	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);
	}
333

334 335
	parent_nritems = btrfs_header_nritems(parent);
	blocksize = btrfs_level_size(root, parent_level - 1);
336 337 338 339 340 341 342
	end_slot = parent_nritems;

	if (parent_nritems == 1)
		return 0;

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

344 345 346 347 348 349 350 351
		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);
		}
352 353 354 355 356
		btrfs_node_key(parent, &disk_key, i);
		if (!progress_passed && comp_keys(&disk_key, progress) < 0)
			continue;

		progress_passed = 1;
357
		blocknr = btrfs_node_blockptr(parent, i);
358
		gen = btrfs_node_ptr_generation(parent, i);
359 360
		if (last_block == 0)
			last_block = blocknr;
361

362
		if (i > 0) {
363 364
			other = btrfs_node_blockptr(parent, i - 1);
			close = close_blocks(blocknr, other, blocksize);
365
		}
366
		if (close && i < end_slot - 2) {
367 368
			other = btrfs_node_blockptr(parent, i + 1);
			close = close_blocks(blocknr, other, blocksize);
369
		}
370 371
		if (close) {
			last_block = blocknr;
372
			continue;
373
		}
374 375 376 377 378
		if (parent->map_token) {
			unmap_extent_buffer(parent, parent->map_token,
					    KM_USER1);
			parent->map_token = NULL;
		}
379

380 381 382 383 384
		cur = btrfs_find_tree_block(root, blocknr, blocksize);
		if (cur)
			uptodate = btrfs_buffer_uptodate(cur);
		else
			uptodate = 0;
385
		if (!cur || !uptodate) {
386
			if (cache_only) {
387
				free_extent_buffer(cur);
388 389
				continue;
			}
390 391
			if (!cur) {
				cur = read_tree_block(root, blocknr,
392
							 blocksize, gen);
393
			} else if (!uptodate) {
394
				btrfs_read_buffer(cur, gen);
395
			}
396
		}
397
		if (search_start == 0)
398
			search_start = last_block;
399

400 401 402 403
		err = __btrfs_cow_block(trans, root, cur, parent, i,
					&tmp, search_start,
					min(16 * blocksize,
					    (end_slot - i) * blocksize));
Y
Yan 已提交
404
		if (err) {
405
			free_extent_buffer(cur);
406
			break;
Y
Yan 已提交
407
		}
408
		search_start = tmp->start;
409
		last_block = tmp->start;
410 411
		*last_ret = search_start;
		if (parent_level == 1)
412 413
			btrfs_clear_buffer_defrag(tmp);
		free_extent_buffer(tmp);
414
	}
415 416 417 418 419
	if (parent->map_token) {
		unmap_extent_buffer(parent, parent->map_token,
				    KM_USER1);
		parent->map_token = NULL;
	}
420 421 422
	return err;
}

C
Chris Mason 已提交
423 424 425 426 427
/*
 * 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 已提交
428
static inline unsigned int leaf_data_end(struct btrfs_root *root,
429
					 struct extent_buffer *leaf)
430
{
431
	u32 nr = btrfs_header_nritems(leaf);
432
	if (nr == 0)
C
Chris Mason 已提交
433
		return BTRFS_LEAF_DATA_SIZE(root);
434
	return btrfs_item_offset_nr(leaf, nr - 1);
435 436
}

C
Chris Mason 已提交
437 438
static int check_node(struct btrfs_root *root, struct btrfs_path *path,
		      int level)
C
Chris Mason 已提交
439
{
440 441 442 443
	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 已提交
444
	int parent_slot;
445 446
	int slot;
	struct btrfs_key cpukey;
447
	u32 nritems = btrfs_header_nritems(node);
C
Chris Mason 已提交
448 449

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

452
	slot = path->slots[level];
453 454
	BUG_ON(nritems == 0);
	if (parent) {
A
Aneesh 已提交
455
		parent_slot = path->slots[level + 1];
456 457 458
		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 已提交
459
			      sizeof(struct btrfs_disk_key)));
460
		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
461
		       btrfs_header_bytenr(node));
C
Chris Mason 已提交
462
	}
C
Chris Mason 已提交
463
	BUG_ON(nritems > BTRFS_NODEPTRS_PER_BLOCK(root));
464
	if (slot != 0) {
465 466 467
		btrfs_node_key_to_cpu(node, &cpukey, slot - 1);
		btrfs_node_key(node, &node_key, slot);
		BUG_ON(comp_keys(&node_key, &cpukey) <= 0);
468 469
	}
	if (slot < nritems - 1) {
470 471 472
		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 已提交
473 474 475 476
	}
	return 0;
}

C
Chris Mason 已提交
477 478
static int check_leaf(struct btrfs_root *root, struct btrfs_path *path,
		      int level)
C
Chris Mason 已提交
479
{
480 481
	struct extent_buffer *leaf = path->nodes[level];
	struct extent_buffer *parent = NULL;
C
Chris Mason 已提交
482
	int parent_slot;
483
	struct btrfs_key cpukey;
484 485 486
	struct btrfs_disk_key parent_key;
	struct btrfs_disk_key leaf_key;
	int slot = path->slots[0];
487

488
	u32 nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
489 490

	if (path->nodes[level + 1])
491
		parent = path->nodes[level + 1];
492 493 494 495 496

	if (nritems == 0)
		return 0;

	if (parent) {
A
Aneesh 已提交
497
		parent_slot = path->slots[level + 1];
498 499
		btrfs_node_key(parent, &parent_key, parent_slot);
		btrfs_item_key(leaf, &leaf_key, 0);
500

501
		BUG_ON(memcmp(&parent_key, &leaf_key,
C
Chris Mason 已提交
502
		       sizeof(struct btrfs_disk_key)));
503
		BUG_ON(btrfs_node_blockptr(parent, parent_slot) !=
504
		       btrfs_header_bytenr(leaf));
505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529
	}
#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 已提交
530
	}
531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552
	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);
		}
553 554
	}
	if (slot < nritems - 1) {
555 556 557 558 559 560 561 562 563
		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 已提交
564
	}
565 566
	BUG_ON(btrfs_item_offset_nr(leaf, 0) +
	       btrfs_item_size_nr(leaf, 0) != BTRFS_LEAF_DATA_SIZE(root));
C
Chris Mason 已提交
567 568 569
	return 0;
}

570 571
static int noinline check_block(struct btrfs_root *root,
				struct btrfs_path *path, int level)
C
Chris Mason 已提交
572
{
573
	u64 found_start;
574
	return 0;
575 576 577 578 579 580 581 582 583
	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);
	}
584
#if 0
585 586
	struct extent_buffer *buf = path->nodes[level];

587 588 589
	if (memcmp_extent_buffer(buf, root->fs_info->fsid,
				 (unsigned long)btrfs_header_fsid(buf),
				 BTRFS_FSID_SIZE)) {
590
		printk("warning bad block %Lu\n", buf->start);
591
		return 1;
592
	}
593
#endif
C
Chris Mason 已提交
594
	if (level == 0)
C
Chris Mason 已提交
595 596
		return check_leaf(root, path, level);
	return check_node(root, path, level);
C
Chris Mason 已提交
597 598
}

C
Chris Mason 已提交
599
/*
600 601 602
 * 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 已提交
603 604 605 606 607 608
 * 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
 */
609 610 611
static int generic_bin_search(struct extent_buffer *eb, unsigned long p,
			      int item_size, struct btrfs_key *key,
			      int max, int *slot)
612 613 614 615 616
{
	int low = 0;
	int high = max;
	int mid;
	int ret;
617
	struct btrfs_disk_key *tmp = NULL;
618 619 620 621 622 623
	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;
624
	int err;
625 626 627

	while(low < high) {
		mid = (low + high) / 2;
628 629 630 631 632
		offset = p + mid * item_size;

		if (!map_token || offset < map_start ||
		    (offset + sizeof(struct btrfs_disk_key)) >
		    map_start + map_len) {
633
			if (map_token) {
634
				unmap_extent_buffer(eb, map_token, KM_USER0);
635 636 637 638 639 640 641 642 643 644 645 646 647 648 649
				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;
			}
650 651 652 653 654

		} else {
			tmp = (struct btrfs_disk_key *)(kaddr + offset -
							map_start);
		}
655 656 657 658 659 660 661 662
		ret = comp_keys(tmp, key);

		if (ret < 0)
			low = mid + 1;
		else if (ret > 0)
			high = mid;
		else {
			*slot = mid;
663 664
			if (map_token)
				unmap_extent_buffer(eb, map_token, KM_USER0);
665 666 667 668
			return 0;
		}
	}
	*slot = low;
669 670
	if (map_token)
		unmap_extent_buffer(eb, map_token, KM_USER0);
671 672 673
	return 1;
}

C
Chris Mason 已提交
674 675 676 677
/*
 * simple bin_search frontend that does the right thing for
 * leaves vs nodes
 */
678 679
static int bin_search(struct extent_buffer *eb, struct btrfs_key *key,
		      int level, int *slot)
680
{
681 682 683
	if (level == 0) {
		return generic_bin_search(eb,
					  offsetof(struct btrfs_leaf, items),
C
Chris Mason 已提交
684
					  sizeof(struct btrfs_item),
685
					  key, btrfs_header_nritems(eb),
686
					  slot);
687
	} else {
688 689
		return generic_bin_search(eb,
					  offsetof(struct btrfs_node, ptrs),
C
Chris Mason 已提交
690
					  sizeof(struct btrfs_key_ptr),
691
					  key, btrfs_header_nritems(eb),
692
					  slot);
693 694 695 696
	}
	return -1;
}

697 698
static struct extent_buffer *read_node_slot(struct btrfs_root *root,
				   struct extent_buffer *parent, int slot)
699
{
700
	int level = btrfs_header_level(parent);
701 702
	if (slot < 0)
		return NULL;
703
	if (slot >= btrfs_header_nritems(parent))
704
		return NULL;
705 706 707

	BUG_ON(level == 0);

708
	return read_tree_block(root, btrfs_node_blockptr(parent, slot),
709 710
		       btrfs_level_size(root, level - 1),
		       btrfs_node_ptr_generation(parent, slot));
711 712
}

713 714 715
static int balance_level(struct btrfs_trans_handle *trans,
			 struct btrfs_root *root,
			 struct btrfs_path *path, int level)
716
{
717 718 719 720
	struct extent_buffer *right = NULL;
	struct extent_buffer *mid;
	struct extent_buffer *left = NULL;
	struct extent_buffer *parent = NULL;
721 722 723 724
	int ret = 0;
	int wret;
	int pslot;
	int orig_slot = path->slots[level];
725
	int err_on_enospc = 0;
726
	u64 orig_ptr;
727 728 729 730

	if (level == 0)
		return 0;

731
	mid = path->nodes[level];
732 733
	WARN_ON(btrfs_header_generation(mid) != trans->transid);

734
	orig_ptr = btrfs_node_blockptr(mid, orig_slot);
735

C
Chris Mason 已提交
736
	if (level < BTRFS_MAX_LEVEL - 1)
737
		parent = path->nodes[level + 1];
738 739
	pslot = path->slots[level + 1];

C
Chris Mason 已提交
740 741 742 743
	/*
	 * deal with the case where there is only one pointer in the root
	 * by promoting the node below to a root
	 */
744 745
	if (!parent) {
		struct extent_buffer *child;
746

747
		if (btrfs_header_nritems(mid) != 1)
748 749 750
			return 0;

		/* promote the child to a root */
751
		child = read_node_slot(root, mid, 0);
752
		BUG_ON(!child);
753 754 755
		ret = btrfs_cow_block(trans, root, child, mid, 0, &child);
		BUG_ON(ret);

756
		root->node = child;
757
		add_root_to_dirty_list(root);
758
		path->nodes[level] = NULL;
759
		clean_tree_block(trans, root, mid);
760
		/* once for the path */
761
		free_extent_buffer(mid);
762 763 764
		ret = btrfs_free_extent(trans, root, mid->start, mid->len,
					root->root_key.objectid,
					btrfs_header_generation(mid), 0, 0, 1);
765
		/* once for the root ptr */
766
		free_extent_buffer(mid);
767
		return ret;
768
	}
769
	if (btrfs_header_nritems(mid) >
C
Chris Mason 已提交
770
	    BTRFS_NODEPTRS_PER_BLOCK(root) / 4)
771 772
		return 0;

773
	if (btrfs_header_nritems(mid) < 2)
774 775
		err_on_enospc = 1;

776 777 778 779
	left = read_node_slot(root, parent, pslot - 1);
	if (left) {
		wret = btrfs_cow_block(trans, root, left,
				       parent, pslot - 1, &left);
780 781 782 783
		if (wret) {
			ret = wret;
			goto enospc;
		}
784
	}
785 786 787 788
	right = read_node_slot(root, parent, pslot + 1);
	if (right) {
		wret = btrfs_cow_block(trans, root, right,
				       parent, pslot + 1, &right);
789 790 791 792 793 794 795
		if (wret) {
			ret = wret;
			goto enospc;
		}
	}

	/* first, try to make some room in the middle buffer */
796 797
	if (left) {
		orig_slot += btrfs_header_nritems(left);
798
		wret = push_node_left(trans, root, left, mid, 1);
799 800
		if (wret < 0)
			ret = wret;
801
		if (btrfs_header_nritems(mid) < 2)
802
			err_on_enospc = 1;
803
	}
804 805 806 807

	/*
	 * then try to empty the right most buffer into the middle
	 */
808
	if (right) {
809
		wret = push_node_left(trans, root, mid, right, 1);
810
		if (wret < 0 && wret != -ENOSPC)
811
			ret = wret;
812
		if (btrfs_header_nritems(right) == 0) {
813
			u64 bytenr = right->start;
814
			u64 generation = btrfs_header_generation(parent);
815 816
			u32 blocksize = right->len;

817 818
			clean_tree_block(trans, root, right);
			free_extent_buffer(right);
819
			right = NULL;
820 821
			wret = del_ptr(trans, root, path, level + 1, pslot +
				       1);
822 823
			if (wret)
				ret = wret;
824
			wret = btrfs_free_extent(trans, root, bytenr,
825 826 827
						 blocksize,
						 btrfs_header_owner(parent),
						 generation, 0, 0, 1);
828 829 830
			if (wret)
				ret = wret;
		} else {
831 832 833 834
			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);
835 836
		}
	}
837
	if (btrfs_header_nritems(mid) == 1) {
838 839 840 841 842 843 844 845 846
		/*
		 * 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
		 */
847 848
		BUG_ON(!left);
		wret = balance_node_right(trans, root, mid, left);
849
		if (wret < 0) {
850
			ret = wret;
851 852
			goto enospc;
		}
853 854 855 856 857
		if (wret == 1) {
			wret = push_node_left(trans, root, left, mid, 1);
			if (wret < 0)
				ret = wret;
		}
858 859
		BUG_ON(wret == 1);
	}
860
	if (btrfs_header_nritems(mid) == 0) {
861
		/* we've managed to empty the middle node, drop it */
862
		u64 root_gen = btrfs_header_generation(parent);
863 864
		u64 bytenr = mid->start;
		u32 blocksize = mid->len;
865 866
		clean_tree_block(trans, root, mid);
		free_extent_buffer(mid);
867
		mid = NULL;
868
		wret = del_ptr(trans, root, path, level + 1, pslot);
869 870
		if (wret)
			ret = wret;
871 872 873
		wret = btrfs_free_extent(trans, root, bytenr, blocksize,
					 btrfs_header_owner(parent),
					 root_gen, 0, 0, 1);
874 875
		if (wret)
			ret = wret;
876 877
	} else {
		/* update the parent key to reflect our changes */
878 879 880 881
		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);
882
	}
883

884
	/* update the path */
885 886 887 888
	if (left) {
		if (btrfs_header_nritems(left) > orig_slot) {
			extent_buffer_get(left);
			path->nodes[level] = left;
889 890
			path->slots[level + 1] -= 1;
			path->slots[level] = orig_slot;
891 892
			if (mid)
				free_extent_buffer(mid);
893
		} else {
894
			orig_slot -= btrfs_header_nritems(left);
895 896 897
			path->slots[level] = orig_slot;
		}
	}
898
	/* double check we haven't messed things up */
C
Chris Mason 已提交
899
	check_block(root, path, level);
C
Chris Mason 已提交
900
	if (orig_ptr !=
901
	    btrfs_node_blockptr(path->nodes[level], path->slots[level]))
902
		BUG();
903
enospc:
904 905 906 907
	if (right)
		free_extent_buffer(right);
	if (left)
		free_extent_buffer(left);
908 909 910
	return ret;
}

911
/* returns zero if the push worked, non-zero otherwise */
912 913 914
static int noinline push_nodes_for_insert(struct btrfs_trans_handle *trans,
					  struct btrfs_root *root,
					  struct btrfs_path *path, int level)
915
{
916 917 918 919
	struct extent_buffer *right = NULL;
	struct extent_buffer *mid;
	struct extent_buffer *left = NULL;
	struct extent_buffer *parent = NULL;
920 921 922 923 924 925 926 927 928
	int ret = 0;
	int wret;
	int pslot;
	int orig_slot = path->slots[level];
	u64 orig_ptr;

	if (level == 0)
		return 1;

929
	mid = path->nodes[level];
930
	WARN_ON(btrfs_header_generation(mid) != trans->transid);
931 932 933
	orig_ptr = btrfs_node_blockptr(mid, orig_slot);

	if (level < BTRFS_MAX_LEVEL - 1)
934
		parent = path->nodes[level + 1];
935 936
	pslot = path->slots[level + 1];

937
	if (!parent)
938 939
		return 1;

940
	left = read_node_slot(root, parent, pslot - 1);
941 942

	/* first, try to make some room in the middle buffer */
943
	if (left) {
944
		u32 left_nr;
945
		left_nr = btrfs_header_nritems(left);
C
Chris Mason 已提交
946 947 948
		if (left_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
			wret = 1;
		} else {
949 950
			ret = btrfs_cow_block(trans, root, left, parent,
					      pslot - 1, &left);
951 952 953 954
			if (ret)
				wret = 1;
			else {
				wret = push_node_left(trans, root,
955
						      left, mid, 0);
956
			}
C
Chris Mason 已提交
957
		}
958 959 960
		if (wret < 0)
			ret = wret;
		if (wret == 0) {
961
			struct btrfs_disk_key disk_key;
962
			orig_slot += left_nr;
963 964 965 966 967
			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;
968 969
				path->slots[level + 1] -= 1;
				path->slots[level] = orig_slot;
970
				free_extent_buffer(mid);
971 972
			} else {
				orig_slot -=
973
					btrfs_header_nritems(left);
974
				path->slots[level] = orig_slot;
975
				free_extent_buffer(left);
976 977 978
			}
			return 0;
		}
979
		free_extent_buffer(left);
980
	}
981
	right= read_node_slot(root, parent, pslot + 1);
982 983 984 985

	/*
	 * then try to empty the right most buffer into the middle
	 */
986
	if (right) {
C
Chris Mason 已提交
987
		u32 right_nr;
988
		right_nr = btrfs_header_nritems(right);
C
Chris Mason 已提交
989 990 991
		if (right_nr >= BTRFS_NODEPTRS_PER_BLOCK(root) - 1) {
			wret = 1;
		} else {
992 993 994
			ret = btrfs_cow_block(trans, root, right,
					      parent, pslot + 1,
					      &right);
995 996 997 998
			if (ret)
				wret = 1;
			else {
				wret = balance_node_right(trans, root,
999
							  right, mid);
1000
			}
C
Chris Mason 已提交
1001
		}
1002 1003 1004
		if (wret < 0)
			ret = wret;
		if (wret == 0) {
1005 1006 1007 1008 1009 1010 1011 1012
			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;
1013 1014
				path->slots[level + 1] += 1;
				path->slots[level] = orig_slot -
1015 1016
					btrfs_header_nritems(mid);
				free_extent_buffer(mid);
1017
			} else {
1018
				free_extent_buffer(right);
1019 1020 1021
			}
			return 0;
		}
1022
		free_extent_buffer(right);
1023 1024 1025 1026
	}
	return 1;
}

1027 1028 1029 1030
/*
 * readahead one full node of leaves
 */
static void reada_for_search(struct btrfs_root *root, struct btrfs_path *path,
1031
			     int level, int slot, u64 objectid)
1032
{
1033
	struct extent_buffer *node;
1034
	struct btrfs_disk_key disk_key;
1035 1036
	u32 nritems;
	u64 search;
1037 1038 1039
	u64 lowest_read;
	u64 highest_read;
	u64 nread = 0;
1040
	int direction = path->reada;
1041
	struct extent_buffer *eb;
1042 1043 1044
	u32 nr;
	u32 blocksize;
	u32 nscan = 0;
1045

1046
	if (level != 1)
1047 1048 1049
		return;

	if (!path->nodes[level])
1050 1051
		return;

1052
	node = path->nodes[level];
1053
	search = btrfs_node_blockptr(node, slot);
1054 1055
	blocksize = btrfs_level_size(root, level - 1);
	eb = btrfs_find_tree_block(root, search, blocksize);
1056 1057
	if (eb) {
		free_extent_buffer(eb);
1058 1059 1060
		return;
	}

1061 1062 1063
	highest_read = search;
	lowest_read = search;

1064
	nritems = btrfs_header_nritems(node);
1065
	nr = slot;
1066
	while(1) {
1067 1068 1069 1070 1071 1072 1073 1074
		if (direction < 0) {
			if (nr == 0)
				break;
			nr--;
		} else if (direction > 0) {
			nr++;
			if (nr >= nritems)
				break;
1075
		}
1076 1077 1078 1079 1080
		if (path->reada < 0 && objectid) {
			btrfs_node_key(node, &disk_key, nr);
			if (btrfs_disk_key_objectid(&disk_key) != objectid)
				break;
		}
1081 1082 1083 1084
		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)) {
1085 1086
			readahead_tree_block(root, search, blocksize,
				     btrfs_node_ptr_generation(node, nr));
1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098
			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;
1099 1100
	}
}
C
Chris Mason 已提交
1101 1102 1103 1104 1105 1106
/*
 * 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 已提交
1107 1108
 * be inserted, and 1 is returned.  If there are other errors during the
 * search a negative error number is returned.
C
Chris Mason 已提交
1109 1110 1111 1112
 *
 * 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 已提交
1113
 */
1114 1115 1116
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)
1117
{
1118
	struct extent_buffer *b;
1119 1120 1121
	int slot;
	int ret;
	int level;
1122
	int should_reada = p->reada;
1123 1124
	u8 lowest_level = 0;

1125 1126
	lowest_level = p->lowest_level;
	WARN_ON(lowest_level && ins_len);
C
Chris Mason 已提交
1127 1128
	WARN_ON(p->nodes[0] != NULL);
	WARN_ON(!mutex_is_locked(&root->fs_info->fs_mutex));
1129 1130
again:
	b = root->node;
1131
	extent_buffer_get(b);
1132
	while (b) {
1133
		level = btrfs_header_level(b);
C
Chris Mason 已提交
1134 1135
		if (cow) {
			int wret;
C
Chris Mason 已提交
1136 1137 1138
			wret = btrfs_cow_block(trans, root, b,
					       p->nodes[level + 1],
					       p->slots[level + 1],
Y
Yan 已提交
1139
					       &b);
1140
			if (wret) {
1141
				free_extent_buffer(b);
1142 1143
				return wret;
			}
C
Chris Mason 已提交
1144 1145
		}
		BUG_ON(!cow && ins_len);
1146
		if (level != btrfs_header_level(b))
C
Chris Mason 已提交
1147
			WARN_ON(1);
1148
		level = btrfs_header_level(b);
1149
		p->nodes[level] = b;
C
Chris Mason 已提交
1150
		ret = check_block(root, p, level);
C
Chris Mason 已提交
1151 1152
		if (ret)
			return -1;
1153 1154
		ret = bin_search(b, key, level, &slot);
		if (level != 0) {
1155 1156 1157
			if (ret && slot > 0)
				slot -= 1;
			p->slots[level] = slot;
1158
			if (ins_len > 0 && btrfs_header_nritems(b) >=
1159
			    BTRFS_NODEPTRS_PER_BLOCK(root) - 3) {
1160
				int sret = split_node(trans, root, p, level);
C
Chris Mason 已提交
1161 1162 1163 1164 1165
				BUG_ON(sret > 0);
				if (sret)
					return sret;
				b = p->nodes[level];
				slot = p->slots[level];
1166
			} else if (ins_len < 0) {
1167 1168
				int sret = balance_level(trans, root, p,
							 level);
1169 1170 1171
				if (sret)
					return sret;
				b = p->nodes[level];
1172 1173
				if (!b) {
					btrfs_release_path(NULL, p);
1174
					goto again;
1175
				}
1176
				slot = p->slots[level];
1177
				BUG_ON(btrfs_header_nritems(b) == 1);
C
Chris Mason 已提交
1178
			}
1179 1180 1181
			/* this is only true while dropping a snapshot */
			if (level == lowest_level)
				break;
1182

1183
			if (should_reada)
1184 1185
				reada_for_search(root, p, level, slot,
						 key->objectid);
1186 1187

			b = read_node_slot(root, b, slot);
1188 1189
		} else {
			p->slots[level] = slot;
1190
			if (ins_len > 0 && btrfs_leaf_free_space(root, b) <
C
Chris Mason 已提交
1191
			    sizeof(struct btrfs_item) + ins_len) {
1192
				int sret = split_leaf(trans, root, key,
1193
						      p, ins_len, ret == 0);
C
Chris Mason 已提交
1194 1195 1196 1197
				BUG_ON(sret > 0);
				if (sret)
					return sret;
			}
1198 1199 1200
			return ret;
		}
	}
C
Chris Mason 已提交
1201
	return 1;
1202 1203
}

C
Chris Mason 已提交
1204 1205 1206 1207 1208 1209
/*
 * 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 已提交
1210 1211 1212
 *
 * 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 已提交
1213
 */
1214 1215 1216
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)
1217 1218
{
	int i;
C
Chris Mason 已提交
1219
	int ret = 0;
1220 1221
	struct extent_buffer *t;

C
Chris Mason 已提交
1222
	for (i = level; i < BTRFS_MAX_LEVEL; i++) {
1223
		int tslot = path->slots[i];
1224
		if (!path->nodes[i])
1225
			break;
1226 1227
		t = path->nodes[i];
		btrfs_set_node_key(t, key, tslot);
C
Chris Mason 已提交
1228
		btrfs_mark_buffer_dirty(path->nodes[i]);
1229 1230 1231
		if (tslot != 0)
			break;
	}
C
Chris Mason 已提交
1232
	return ret;
1233 1234
}

C
Chris Mason 已提交
1235 1236
/*
 * try to push data from one node into the next node left in the
1237
 * tree.
C
Chris Mason 已提交
1238 1239 1240
 *
 * 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 已提交
1241
 */
1242 1243
static int push_node_left(struct btrfs_trans_handle *trans,
			  struct btrfs_root *root, struct extent_buffer *dst,
1244
			  struct extent_buffer *src, int empty)
1245 1246
{
	int push_items = 0;
1247 1248
	int src_nritems;
	int dst_nritems;
C
Chris Mason 已提交
1249
	int ret = 0;
1250

1251 1252
	src_nritems = btrfs_header_nritems(src);
	dst_nritems = btrfs_header_nritems(dst);
C
Chris Mason 已提交
1253
	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
1254 1255
	WARN_ON(btrfs_header_generation(src) != trans->transid);
	WARN_ON(btrfs_header_generation(dst) != trans->transid);
1256

1257
	if (!empty && src_nritems <= 8)
1258 1259
		return 1;

1260
	if (push_items <= 0) {
1261
		return 1;
1262
	}
1263

1264
	if (empty) {
1265
		push_items = min(src_nritems, push_items);
1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277
		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);
1278

1279 1280 1281 1282 1283
	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));

1284
	if (push_items < src_nritems) {
1285 1286 1287 1288 1289 1290 1291 1292 1293
		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);
1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305
	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
 */
1306 1307 1308 1309
static int balance_node_right(struct btrfs_trans_handle *trans,
			      struct btrfs_root *root,
			      struct extent_buffer *dst,
			      struct extent_buffer *src)
1310 1311 1312 1313 1314 1315 1316
{
	int push_items = 0;
	int max_push;
	int src_nritems;
	int dst_nritems;
	int ret = 0;

1317 1318 1319
	WARN_ON(btrfs_header_generation(src) != trans->transid);
	WARN_ON(btrfs_header_generation(dst) != trans->transid);

1320 1321
	src_nritems = btrfs_header_nritems(src);
	dst_nritems = btrfs_header_nritems(dst);
C
Chris Mason 已提交
1322
	push_items = BTRFS_NODEPTRS_PER_BLOCK(root) - dst_nritems;
1323
	if (push_items <= 0) {
1324
		return 1;
1325 1326 1327 1328 1329
	}

	if (src_nritems < 4) {
		return 1;
	}
1330 1331 1332

	max_push = src_nritems / 2 + 1;
	/* don't try to empty the node */
1333
	if (max_push >= src_nritems) {
1334
		return 1;
1335
	}
Y
Yan 已提交
1336

1337 1338 1339
	if (max_push < push_items)
		push_items = max_push;

1340 1341 1342 1343
	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 已提交
1344

1345 1346 1347 1348
	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));
1349

1350 1351
	btrfs_set_header_nritems(src, src_nritems - push_items);
	btrfs_set_header_nritems(dst, dst_nritems + push_items);
1352

1353 1354
	btrfs_mark_buffer_dirty(src);
	btrfs_mark_buffer_dirty(dst);
C
Chris Mason 已提交
1355
	return ret;
1356 1357
}

C
Chris Mason 已提交
1358 1359 1360 1361
/*
 * 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 已提交
1362 1363
 *
 * returns zero on success or < 0 on failure.
C
Chris Mason 已提交
1364
 */
1365
static int noinline insert_new_root(struct btrfs_trans_handle *trans,
1366 1367
			   struct btrfs_root *root,
			   struct btrfs_path *path, int level)
C
Chris Mason 已提交
1368
{
1369 1370
	u64 root_gen;
	u64 lower_gen;
1371 1372 1373
	struct extent_buffer *lower;
	struct extent_buffer *c;
	struct btrfs_disk_key lower_key;
C
Chris Mason 已提交
1374 1375 1376 1377

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

1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391
	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);

	c = __btrfs_alloc_free_block(trans, root, root->nodesize,
				   root->root_key.objectid,
				   root_gen, lower_key.objectid, level,
1392
				   root->node->start, 0);
1393 1394 1395 1396 1397
	if (IS_ERR(c))
		return PTR_ERR(c);
	memset_extent_buffer(c, 0, 0, root->nodesize);
	btrfs_set_header_nritems(c, 1);
	btrfs_set_header_level(c, level);
1398
	btrfs_set_header_bytenr(c, c->start);
1399 1400 1401 1402 1403 1404
	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);
1405 1406 1407 1408 1409

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

1410
	btrfs_set_node_key(c, &lower_key, 0);
1411
	btrfs_set_node_blockptr(c, 0, lower->start);
1412 1413 1414 1415
	lower_gen = btrfs_header_generation(lower);
	WARN_ON(lower_gen == 0);

	btrfs_set_node_ptr_generation(c, 0, lower_gen);
1416

1417
	btrfs_mark_buffer_dirty(c);
1418

C
Chris Mason 已提交
1419
	/* the super has an extra ref to root->node */
1420 1421
	free_extent_buffer(root->node);
	root->node = c;
1422
	add_root_to_dirty_list(root);
1423 1424
	extent_buffer_get(c);
	path->nodes[level] = c;
C
Chris Mason 已提交
1425
	path->slots[level] = 0;
1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437

	if (root->ref_cows && lower_gen != trans->transid) {
		struct btrfs_path *back_path = btrfs_alloc_path();
		int ret;
		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);
		btrfs_free_path(back_path);
	}
C
Chris Mason 已提交
1438 1439 1440
	return 0;
}

C
Chris Mason 已提交
1441 1442 1443
/*
 * worker function to insert a single pointer in a node.
 * the node should have enough room for the pointer already
C
Chris Mason 已提交
1444
 *
C
Chris Mason 已提交
1445 1446
 * slot and level indicate where you want the key to go, and
 * blocknr is the block the key points to.
C
Chris Mason 已提交
1447 1448
 *
 * returns zero on success and < 0 on any error
C
Chris Mason 已提交
1449
 */
1450 1451
static int insert_ptr(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_path *path, struct btrfs_disk_key
1452
		      *key, u64 bytenr, int slot, int level)
C
Chris Mason 已提交
1453
{
1454
	struct extent_buffer *lower;
C
Chris Mason 已提交
1455
	int nritems;
C
Chris Mason 已提交
1456 1457

	BUG_ON(!path->nodes[level]);
1458 1459
	lower = path->nodes[level];
	nritems = btrfs_header_nritems(lower);
C
Chris Mason 已提交
1460 1461
	if (slot > nritems)
		BUG();
C
Chris Mason 已提交
1462
	if (nritems == BTRFS_NODEPTRS_PER_BLOCK(root))
C
Chris Mason 已提交
1463 1464
		BUG();
	if (slot != nritems) {
1465 1466 1467
		memmove_extent_buffer(lower,
			      btrfs_node_key_ptr_offset(slot + 1),
			      btrfs_node_key_ptr_offset(slot),
C
Chris Mason 已提交
1468
			      (nritems - slot) * sizeof(struct btrfs_key_ptr));
C
Chris Mason 已提交
1469
	}
1470
	btrfs_set_node_key(lower, key, slot);
1471
	btrfs_set_node_blockptr(lower, slot, bytenr);
1472 1473
	WARN_ON(trans->transid == 0);
	btrfs_set_node_ptr_generation(lower, slot, trans->transid);
1474 1475
	btrfs_set_header_nritems(lower, nritems + 1);
	btrfs_mark_buffer_dirty(lower);
C
Chris Mason 已提交
1476 1477 1478
	return 0;
}

C
Chris Mason 已提交
1479 1480 1481 1482 1483 1484
/*
 * 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 已提交
1485 1486
 *
 * returns 0 on success and < 0 on failure
C
Chris Mason 已提交
1487
 */
1488 1489
static int split_node(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_path *path, int level)
1490
{
1491
	u64 root_gen;
1492 1493 1494
	struct extent_buffer *c;
	struct extent_buffer *split;
	struct btrfs_disk_key disk_key;
1495
	int mid;
C
Chris Mason 已提交
1496
	int ret;
C
Chris Mason 已提交
1497
	int wret;
1498
	u32 c_nritems;
1499

1500
	c = path->nodes[level];
1501
	WARN_ON(btrfs_header_generation(c) != trans->transid);
1502
	if (c == root->node) {
C
Chris Mason 已提交
1503
		/* trying to split the root, lets make a new one */
1504
		ret = insert_new_root(trans, root, path, level + 1);
C
Chris Mason 已提交
1505 1506
		if (ret)
			return ret;
1507 1508
	} else {
		ret = push_nodes_for_insert(trans, root, path, level);
1509 1510
		c = path->nodes[level];
		if (!ret && btrfs_header_nritems(c) <
1511
		    BTRFS_NODEPTRS_PER_BLOCK(root) - 3)
1512
			return 0;
1513 1514
		if (ret < 0)
			return ret;
1515
	}
1516

1517
	c_nritems = btrfs_header_nritems(c);
1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528
	if (root->ref_cows)
		root_gen = trans->transid;
	else
		root_gen = 0;

	btrfs_node_key(c, &disk_key, 0);
	split = __btrfs_alloc_free_block(trans, root, root->nodesize,
					 root->root_key.objectid,
					 root_gen,
					 btrfs_disk_key_objectid(&disk_key),
					 level, c->start, 0);
1529 1530 1531 1532 1533
	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));
1534
	btrfs_set_header_bytenr(split, split->start);
1535 1536
	btrfs_set_header_generation(split, trans->transid);
	btrfs_set_header_owner(split, root->root_key.objectid);
1537
	btrfs_set_header_flags(split, 0);
1538 1539 1540
	write_extent_buffer(split, root->fs_info->fsid,
			    (unsigned long)btrfs_header_fsid(split),
			    BTRFS_FSID_SIZE);
1541 1542 1543
	write_extent_buffer(split, root->fs_info->chunk_tree_uuid,
			    (unsigned long)btrfs_header_chunk_tree_uuid(split),
			    BTRFS_UUID_SIZE);
1544

1545
	mid = (c_nritems + 1) / 2;
1546 1547 1548 1549 1550 1551 1552

	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 已提交
1553 1554
	ret = 0;

1555 1556 1557 1558
	btrfs_mark_buffer_dirty(c);
	btrfs_mark_buffer_dirty(split);

	btrfs_node_key(split, &disk_key, 0);
1559
	wret = insert_ptr(trans, root, path, &disk_key, split->start,
1560
			  path->slots[level + 1] + 1,
C
Chris Mason 已提交
1561
			  level + 1);
C
Chris Mason 已提交
1562 1563 1564
	if (wret)
		ret = wret;

C
Chris Mason 已提交
1565
	if (path->slots[level] >= mid) {
C
Chris Mason 已提交
1566
		path->slots[level] -= mid;
1567 1568
		free_extent_buffer(c);
		path->nodes[level] = split;
C
Chris Mason 已提交
1569 1570
		path->slots[level + 1] += 1;
	} else {
1571
		free_extent_buffer(split);
1572
	}
C
Chris Mason 已提交
1573
	return ret;
1574 1575
}

C
Chris Mason 已提交
1576 1577 1578 1579 1580
/*
 * 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
 */
1581
static int leaf_space_used(struct extent_buffer *l, int start, int nr)
1582 1583
{
	int data_len;
1584
	int nritems = btrfs_header_nritems(l);
1585
	int end = min(nritems, start + nr) - 1;
1586 1587 1588

	if (!nr)
		return 0;
1589 1590
	data_len = btrfs_item_end_nr(l, start);
	data_len = data_len - btrfs_item_offset_nr(l, end);
C
Chris Mason 已提交
1591
	data_len += sizeof(struct btrfs_item) * nr;
1592
	WARN_ON(data_len < 0);
1593 1594 1595
	return data_len;
}

1596 1597 1598 1599 1600
/*
 * 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
 */
1601
int btrfs_leaf_free_space(struct btrfs_root *root, struct extent_buffer *leaf)
1602
{
1603 1604 1605 1606 1607
	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 已提交
1608
		       ret, (unsigned long) BTRFS_LEAF_DATA_SIZE(root),
1609 1610 1611
		       leaf_space_used(leaf, 0, nritems), nritems);
	}
	return ret;
1612 1613
}

C
Chris Mason 已提交
1614 1615 1616
/*
 * 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 已提交
1617 1618 1619
 *
 * 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 已提交
1620
 */
1621
static int push_leaf_right(struct btrfs_trans_handle *trans, struct btrfs_root
1622 1623
			   *root, struct btrfs_path *path, int data_size,
			   int empty)
C
Chris Mason 已提交
1624
{
1625 1626 1627 1628
	struct extent_buffer *left = path->nodes[0];
	struct extent_buffer *right;
	struct extent_buffer *upper;
	struct btrfs_disk_key disk_key;
C
Chris Mason 已提交
1629
	int slot;
1630
	u32 i;
C
Chris Mason 已提交
1631 1632 1633
	int free_space;
	int push_space = 0;
	int push_items = 0;
C
Chris Mason 已提交
1634
	struct btrfs_item *item;
1635
	u32 left_nritems;
1636
	u32 nr;
1637
	u32 right_nritems;
1638
	u32 data_end;
1639
	u32 this_item_size;
1640
	int ret;
C
Chris Mason 已提交
1641 1642 1643 1644 1645 1646

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

1650
	right = read_node_slot(root, upper, slot + 1);
C
Chris Mason 已提交
1651
	free_space = btrfs_leaf_free_space(root, right);
C
Chris Mason 已提交
1652
	if (free_space < data_size + sizeof(struct btrfs_item)) {
1653
		free_extent_buffer(right);
C
Chris Mason 已提交
1654 1655
		return 1;
	}
1656

C
Chris Mason 已提交
1657
	/* cow and double check */
1658 1659
	ret = btrfs_cow_block(trans, root, right, upper,
			      slot + 1, &right);
1660
	if (ret) {
1661
		free_extent_buffer(right);
1662 1663
		return 1;
	}
C
Chris Mason 已提交
1664
	free_space = btrfs_leaf_free_space(root, right);
C
Chris Mason 已提交
1665
	if (free_space < data_size + sizeof(struct btrfs_item)) {
1666
		free_extent_buffer(right);
C
Chris Mason 已提交
1667 1668 1669
		return 1;
	}

1670
	left_nritems = btrfs_header_nritems(left);
1671
	if (left_nritems == 0) {
1672
		free_extent_buffer(right);
1673 1674
		return 1;
	}
1675

1676 1677 1678 1679 1680 1681 1682
	if (empty)
		nr = 0;
	else
		nr = 1;

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

C
Chris Mason 已提交
1685 1686
		if (path->slots[0] == i)
			push_space += data_size + sizeof(*item);
1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697

		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 已提交
1698 1699
			break;
		push_items++;
1700
		push_space += this_item_size + sizeof(*item);
1701 1702 1703
		if (i == 0)
			break;
		i--;
1704 1705 1706 1707
	}
	if (left->map_token) {
		unmap_extent_buffer(left, left->map_token, KM_USER1);
		left->map_token = NULL;
C
Chris Mason 已提交
1708
	}
1709

C
Chris Mason 已提交
1710
	if (push_items == 0) {
1711
		free_extent_buffer(right);
C
Chris Mason 已提交
1712 1713
		return 1;
	}
1714

1715
	if (!empty && push_items == left_nritems)
1716
		WARN_ON(1);
1717

C
Chris Mason 已提交
1718
	/* push left to right */
1719
	right_nritems = btrfs_header_nritems(right);
1720

1721
	push_space = btrfs_item_end_nr(left, left_nritems - push_items);
C
Chris Mason 已提交
1722
	push_space -= leaf_data_end(root, left);
1723

C
Chris Mason 已提交
1724
	/* make room in the right data area */
1725 1726 1727 1728 1729 1730
	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 已提交
1731
	/* copy from the left data area */
1732
	copy_extent_buffer(right, left, btrfs_leaf_data(right) +
C
Chris Mason 已提交
1733 1734 1735
		     BTRFS_LEAF_DATA_SIZE(root) - push_space,
		     btrfs_leaf_data(left) + leaf_data_end(root, left),
		     push_space);
1736 1737 1738 1739 1740

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

C
Chris Mason 已提交
1741
	/* copy the items from left to right */
1742 1743 1744
	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 已提交
1745 1746

	/* update the item pointers */
1747
	right_nritems += push_items;
1748
	btrfs_set_header_nritems(right, right_nritems);
C
Chris Mason 已提交
1749
	push_space = BTRFS_LEAF_DATA_SIZE(root);
1750
	for (i = 0; i < right_nritems; i++) {
1751
		item = btrfs_item_nr(right, i);
1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765
		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 已提交
1766
	}
1767
	left_nritems -= push_items;
1768
	btrfs_set_header_nritems(left, left_nritems);
C
Chris Mason 已提交
1769

1770 1771
	if (left_nritems)
		btrfs_mark_buffer_dirty(left);
1772
	btrfs_mark_buffer_dirty(right);
1773

1774 1775
	btrfs_item_key(right, &disk_key, 0);
	btrfs_set_node_key(upper, &disk_key, slot + 1);
C
Chris Mason 已提交
1776
	btrfs_mark_buffer_dirty(upper);
C
Chris Mason 已提交
1777

C
Chris Mason 已提交
1778
	/* then fixup the leaf pointer in the path */
1779 1780
	if (path->slots[0] >= left_nritems) {
		path->slots[0] -= left_nritems;
1781 1782
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = right;
C
Chris Mason 已提交
1783 1784
		path->slots[1] += 1;
	} else {
1785
		free_extent_buffer(right);
C
Chris Mason 已提交
1786 1787 1788
	}
	return 0;
}
C
Chris Mason 已提交
1789 1790 1791 1792
/*
 * 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
 */
1793
static int push_leaf_left(struct btrfs_trans_handle *trans, struct btrfs_root
1794 1795
			  *root, struct btrfs_path *path, int data_size,
			  int empty)
1796
{
1797 1798 1799
	struct btrfs_disk_key disk_key;
	struct extent_buffer *right = path->nodes[0];
	struct extent_buffer *left;
1800 1801 1802 1803 1804
	int slot;
	int i;
	int free_space;
	int push_space = 0;
	int push_items = 0;
C
Chris Mason 已提交
1805
	struct btrfs_item *item;
1806
	u32 old_left_nritems;
1807
	u32 right_nritems;
1808
	u32 nr;
C
Chris Mason 已提交
1809 1810
	int ret = 0;
	int wret;
1811 1812
	u32 this_item_size;
	u32 old_left_item_size;
1813 1814

	slot = path->slots[1];
1815
	if (slot == 0)
1816
		return 1;
1817
	if (!path->nodes[1])
1818
		return 1;
1819

1820 1821 1822 1823 1824
	right_nritems = btrfs_header_nritems(right);
	if (right_nritems == 0) {
		return 1;
	}

1825
	left = read_node_slot(root, path->nodes[1], slot - 1);
C
Chris Mason 已提交
1826
	free_space = btrfs_leaf_free_space(root, left);
C
Chris Mason 已提交
1827
	if (free_space < data_size + sizeof(struct btrfs_item)) {
1828
		free_extent_buffer(left);
1829 1830
		return 1;
	}
C
Chris Mason 已提交
1831 1832

	/* cow and double check */
1833 1834
	ret = btrfs_cow_block(trans, root, left,
			      path->nodes[1], slot - 1, &left);
1835 1836
	if (ret) {
		/* we hit -ENOSPC, but it isn't fatal here */
1837
		free_extent_buffer(left);
1838 1839
		return 1;
	}
1840

C
Chris Mason 已提交
1841
	free_space = btrfs_leaf_free_space(root, left);
C
Chris Mason 已提交
1842
	if (free_space < data_size + sizeof(struct btrfs_item)) {
1843
		free_extent_buffer(left);
C
Chris Mason 已提交
1844 1845 1846
		return 1;
	}

1847 1848 1849 1850 1851 1852
	if (empty)
		nr = right_nritems;
	else
		nr = right_nritems - 1;

	for (i = 0; i < nr; i++) {
1853
		item = btrfs_item_nr(right, i);
1854 1855 1856 1857 1858 1859 1860 1861
		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);
		}

1862 1863
		if (path->slots[0] == i)
			push_space += data_size + sizeof(*item);
1864 1865 1866

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

1869
		push_items++;
1870 1871 1872 1873 1874 1875
		push_space += this_item_size + sizeof(*item);
	}

	if (right->map_token) {
		unmap_extent_buffer(right, right->map_token, KM_USER1);
		right->map_token = NULL;
1876
	}
1877

1878
	if (push_items == 0) {
1879
		free_extent_buffer(left);
1880 1881
		return 1;
	}
1882
	if (!empty && push_items == btrfs_header_nritems(right))
1883
		WARN_ON(1);
1884

1885
	/* push data from right to left */
1886 1887 1888 1889 1890
	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 已提交
1891
	push_space = BTRFS_LEAF_DATA_SIZE(root) -
1892 1893 1894
		     btrfs_item_offset_nr(right, push_items -1);

	copy_extent_buffer(left, right, btrfs_leaf_data(left) +
C
Chris Mason 已提交
1895 1896
		     leaf_data_end(root, left) - push_space,
		     btrfs_leaf_data(right) +
1897
		     btrfs_item_offset_nr(right, push_items - 1),
C
Chris Mason 已提交
1898
		     push_space);
1899
	old_left_nritems = btrfs_header_nritems(left);
1900 1901
	BUG_ON(old_left_nritems < 0);

1902
	old_left_item_size = btrfs_item_offset_nr(left, old_left_nritems - 1);
C
Chris Mason 已提交
1903
	for (i = old_left_nritems; i < old_left_nritems + push_items; i++) {
1904
		u32 ioff;
1905

1906
		item = btrfs_item_nr(left, i);
1907 1908 1909 1910 1911 1912 1913 1914
		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);
		}

1915 1916
		ioff = btrfs_item_offset(left, item);
		btrfs_set_item_offset(left, item,
1917
		      ioff - (BTRFS_LEAF_DATA_SIZE(root) - old_left_item_size));
1918
	}
1919
	btrfs_set_header_nritems(left, old_left_nritems + push_items);
1920 1921 1922 1923
	if (left->map_token) {
		unmap_extent_buffer(left, left->map_token, KM_USER1);
		left->map_token = NULL;
	}
1924 1925

	/* fixup right node */
1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939
	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),
1940 1941 1942
			      btrfs_item_nr_offset(push_items),
			     (btrfs_header_nritems(right) - push_items) *
			     sizeof(struct btrfs_item));
1943
	}
1944 1945
	right_nritems -= push_items;
	btrfs_set_header_nritems(right, right_nritems);
C
Chris Mason 已提交
1946
	push_space = BTRFS_LEAF_DATA_SIZE(root);
1947 1948
	for (i = 0; i < right_nritems; i++) {
		item = btrfs_item_nr(right, i);
1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963

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

1966
	btrfs_mark_buffer_dirty(left);
1967 1968
	if (right_nritems)
		btrfs_mark_buffer_dirty(right);
1969

1970 1971
	btrfs_item_key(right, &disk_key, 0);
	wret = fixup_low_keys(trans, root, path, &disk_key, 1);
C
Chris Mason 已提交
1972 1973
	if (wret)
		ret = wret;
1974 1975 1976 1977

	/* then fixup the leaf pointer in the path */
	if (path->slots[0] < push_items) {
		path->slots[0] += old_left_nritems;
1978 1979
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = left;
1980 1981
		path->slots[1] -= 1;
	} else {
1982
		free_extent_buffer(left);
1983 1984
		path->slots[0] -= push_items;
	}
1985
	BUG_ON(path->slots[0] < 0);
C
Chris Mason 已提交
1986
	return ret;
1987 1988
}

C
Chris Mason 已提交
1989 1990 1991
/*
 * 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 已提交
1992 1993
 *
 * returns 0 if all went well and < 0 on failure.
C
Chris Mason 已提交
1994
 */
1995
static int split_leaf(struct btrfs_trans_handle *trans, struct btrfs_root
1996
		      *root, struct btrfs_key *ins_key,
1997
		      struct btrfs_path *path, int data_size, int extend)
1998
{
1999
	u64 root_gen;
2000
	struct extent_buffer *l;
2001
	u32 nritems;
2002 2003
	int mid;
	int slot;
2004
	struct extent_buffer *right;
C
Chris Mason 已提交
2005
	int space_needed = data_size + sizeof(struct btrfs_item);
2006 2007 2008
	int data_copy_size;
	int rt_data_off;
	int i;
2009
	int ret = 0;
C
Chris Mason 已提交
2010
	int wret;
2011 2012
	int double_split;
	int num_doubles = 0;
2013
	struct btrfs_disk_key disk_key;
C
Chris Mason 已提交
2014

2015 2016 2017
	if (extend)
		space_needed = data_size;

2018 2019 2020 2021 2022
	if (root->ref_cows)
		root_gen = trans->transid;
	else
		root_gen = 0;

C
Chris Mason 已提交
2023
	/* first try to make some room by pushing left and right */
2024
	if (ins_key->type != BTRFS_DIR_ITEM_KEY) {
2025
		wret = push_leaf_right(trans, root, path, data_size, 0);
2026
		if (wret < 0) {
C
Chris Mason 已提交
2027
			return wret;
2028 2029
		}
		if (wret) {
2030
			wret = push_leaf_left(trans, root, path, data_size, 0);
2031 2032 2033 2034
			if (wret < 0)
				return wret;
		}
		l = path->nodes[0];
C
Chris Mason 已提交
2035

2036
		/* did the pushes work? */
2037
		if (btrfs_leaf_free_space(root, l) >= space_needed)
2038
			return 0;
2039
	}
C
Chris Mason 已提交
2040

C
Chris Mason 已提交
2041
	if (!path->nodes[1]) {
2042
		ret = insert_new_root(trans, root, path, 1);
C
Chris Mason 已提交
2043 2044 2045
		if (ret)
			return ret;
	}
2046 2047 2048
again:
	double_split = 0;
	l = path->nodes[0];
2049
	slot = path->slots[0];
2050
	nritems = btrfs_header_nritems(l);
2051
	mid = (nritems + 1)/ 2;
2052

2053 2054 2055 2056 2057 2058
	btrfs_item_key(l, &disk_key, 0);

	right = __btrfs_alloc_free_block(trans, root, root->leafsize,
					 root->root_key.objectid,
					 root_gen, disk_key.objectid, 0,
					 l->start, 0);
2059 2060
	if (IS_ERR(right)) {
		BUG_ON(1);
2061
		return PTR_ERR(right);
2062
	}
2063 2064

	memset_extent_buffer(right, 0, 0, sizeof(struct btrfs_header));
2065
	btrfs_set_header_bytenr(right, right->start);
2066 2067 2068 2069 2070 2071
	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);
2072 2073 2074 2075

	write_extent_buffer(right, root->fs_info->chunk_tree_uuid,
			    (unsigned long)btrfs_header_chunk_tree_uuid(right),
			    BTRFS_UUID_SIZE);
2076 2077 2078 2079 2080 2081
	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);
2082
				btrfs_set_header_nritems(right, 0);
2083
				wret = insert_ptr(trans, root, path,
2084
						  &disk_key, right->start,
2085 2086 2087
						  path->slots[1] + 1, 1);
				if (wret)
					ret = wret;
2088 2089
				free_extent_buffer(path->nodes[0]);
				path->nodes[0] = right;
2090 2091
				path->slots[0] = 0;
				path->slots[1] += 1;
2092
				btrfs_mark_buffer_dirty(right);
2093 2094 2095
				return ret;
			}
			mid = slot;
2096 2097 2098 2099 2100
			if (mid != nritems &&
			    leaf_space_used(l, mid, nritems - mid) +
			    space_needed > BTRFS_LEAF_DATA_SIZE(root)) {
				double_split = 1;
			}
2101 2102 2103 2104
		}
	} else {
		if (leaf_space_used(l, 0, mid + 1) + space_needed >
			BTRFS_LEAF_DATA_SIZE(root)) {
2105
			if (!extend && slot == 0) {
2106
				btrfs_cpu_key_to_disk(&disk_key, ins_key);
2107
				btrfs_set_header_nritems(right, 0);
2108 2109
				wret = insert_ptr(trans, root, path,
						  &disk_key,
2110
						  right->start,
2111
						  path->slots[1], 1);
2112 2113
				if (wret)
					ret = wret;
2114 2115
				free_extent_buffer(path->nodes[0]);
				path->nodes[0] = right;
2116
				path->slots[0] = 0;
2117 2118 2119 2120 2121 2122
				if (path->slots[1] == 0) {
					wret = fixup_low_keys(trans, root,
					           path, &disk_key, 1);
					if (wret)
						ret = wret;
				}
2123
				btrfs_mark_buffer_dirty(right);
2124
				return ret;
2125 2126 2127 2128 2129 2130 2131 2132 2133
			} 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;
				}
2134
			}
2135 2136
		}
	}
2137 2138 2139 2140 2141 2142 2143 2144 2145
	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 已提交
2146 2147 2148
		     btrfs_leaf_data(right) + BTRFS_LEAF_DATA_SIZE(root) -
		     data_copy_size, btrfs_leaf_data(l) +
		     leaf_data_end(root, l), data_copy_size);
2149

C
Chris Mason 已提交
2150
	rt_data_off = BTRFS_LEAF_DATA_SIZE(root) -
2151
		      btrfs_item_end_nr(l, mid);
C
Chris Mason 已提交
2152

2153 2154
	for (i = 0; i < nritems; i++) {
		struct btrfs_item *item = btrfs_item_nr(right, i);
2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165
		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);
2166
		btrfs_set_item_offset(right, item, ioff + rt_data_off);
C
Chris Mason 已提交
2167
	}
C
Chris Mason 已提交
2168

2169 2170 2171 2172 2173
	if (right->map_token) {
		unmap_extent_buffer(right, right->map_token, KM_USER1);
		right->map_token = NULL;
	}

2174
	btrfs_set_header_nritems(l, mid);
C
Chris Mason 已提交
2175
	ret = 0;
2176
	btrfs_item_key(right, &disk_key, 0);
2177 2178
	wret = insert_ptr(trans, root, path, &disk_key, right->start,
			  path->slots[1] + 1, 1);
C
Chris Mason 已提交
2179 2180
	if (wret)
		ret = wret;
2181 2182 2183

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

2186
	if (mid <= slot) {
2187 2188
		free_extent_buffer(path->nodes[0]);
		path->nodes[0] = right;
2189 2190
		path->slots[0] -= mid;
		path->slots[1] += 1;
2191
	} else
2192 2193
		free_extent_buffer(right);

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

2196 2197 2198 2199
	if (double_split) {
		BUG_ON(num_doubles != 0);
		num_doubles++;
		goto again;
2200
	}
2201 2202 2203
	return ret;
}

C
Chris Mason 已提交
2204 2205 2206
int btrfs_truncate_item(struct btrfs_trans_handle *trans,
			struct btrfs_root *root,
			struct btrfs_path *path,
2207
			u32 new_size, int from_end)
C
Chris Mason 已提交
2208 2209 2210 2211
{
	int ret = 0;
	int slot;
	int slot_orig;
2212 2213
	struct extent_buffer *leaf;
	struct btrfs_item *item;
C
Chris Mason 已提交
2214 2215 2216 2217 2218 2219 2220 2221
	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];
2222
	leaf = path->nodes[0];
2223 2224 2225 2226 2227
	slot = path->slots[0];

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

2229
	nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
2230 2231
	data_end = leaf_data_end(root, leaf);

2232
	old_data_start = btrfs_item_offset_nr(leaf, slot);
2233

C
Chris Mason 已提交
2234 2235 2236 2237 2238 2239 2240 2241 2242 2243
	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++) {
2244 2245
		u32 ioff;
		item = btrfs_item_nr(leaf, i);
2246 2247 2248 2249 2250 2251 2252 2253 2254

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

2255 2256
		ioff = btrfs_item_offset(leaf, item);
		btrfs_set_item_offset(leaf, item, ioff + size_diff);
C
Chris Mason 已提交
2257
	}
2258 2259 2260 2261 2262 2263

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

C
Chris Mason 已提交
2264
	/* shift the data */
2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303
	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);
	}
2304 2305 2306 2307

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

	ret = 0;
2310 2311
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
C
Chris Mason 已提交
2312
		BUG();
2313
	}
C
Chris Mason 已提交
2314 2315 2316
	return ret;
}

2317 2318 2319
int btrfs_extend_item(struct btrfs_trans_handle *trans,
		      struct btrfs_root *root, struct btrfs_path *path,
		      u32 data_size)
2320 2321 2322 2323
{
	int ret = 0;
	int slot;
	int slot_orig;
2324 2325
	struct extent_buffer *leaf;
	struct btrfs_item *item;
2326 2327 2328 2329 2330 2331 2332
	u32 nritems;
	unsigned int data_end;
	unsigned int old_data;
	unsigned int old_size;
	int i;

	slot_orig = path->slots[0];
2333
	leaf = path->nodes[0];
2334

2335
	nritems = btrfs_header_nritems(leaf);
2336 2337
	data_end = leaf_data_end(root, leaf);

2338 2339
	if (btrfs_leaf_free_space(root, leaf) < data_size) {
		btrfs_print_leaf(root, leaf);
2340
		BUG();
2341
	}
2342
	slot = path->slots[0];
2343
	old_data = btrfs_item_end_nr(leaf, slot);
2344 2345

	BUG_ON(slot < 0);
2346 2347 2348 2349 2350
	if (slot >= nritems) {
		btrfs_print_leaf(root, leaf);
		printk("slot %d too large, nritems %d\n", slot, nritems);
		BUG_ON(1);
	}
2351 2352 2353 2354 2355 2356

	/*
	 * item0..itemN ... dataN.offset..dataN.size .. data0.size
	 */
	/* first correct the data pointers */
	for (i = slot; i < nritems; i++) {
2357 2358
		u32 ioff;
		item = btrfs_item_nr(leaf, i);
2359 2360 2361 2362 2363 2364 2365 2366

		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);
		}
2367 2368
		ioff = btrfs_item_offset(leaf, item);
		btrfs_set_item_offset(leaf, item, ioff - data_size);
2369
	}
2370

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

2376
	/* shift the data */
2377
	memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2378 2379
		      data_end - data_size, btrfs_leaf_data(leaf) +
		      data_end, old_data - data_end);
2380

2381
	data_end = old_data;
2382 2383 2384 2385
	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);
2386 2387

	ret = 0;
2388 2389
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
2390
		BUG();
2391
	}
2392 2393 2394
	return ret;
}

C
Chris Mason 已提交
2395 2396 2397 2398
/*
 * 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.
 */
2399
int btrfs_insert_empty_items(struct btrfs_trans_handle *trans,
2400 2401
			    struct btrfs_root *root,
			    struct btrfs_path *path,
2402 2403
			    struct btrfs_key *cpu_key, u32 *data_size,
			    int nr)
2404
{
2405 2406
	struct extent_buffer *leaf;
	struct btrfs_item *item;
C
Chris Mason 已提交
2407
	int ret = 0;
2408
	int slot;
2409
	int slot_orig;
2410
	int i;
2411
	u32 nritems;
2412 2413
	u32 total_size = 0;
	u32 total_data = 0;
2414
	unsigned int data_end;
C
Chris Mason 已提交
2415 2416
	struct btrfs_disk_key disk_key;

2417 2418 2419
	for (i = 0; i < nr; i++) {
		total_data += data_size[i];
	}
2420

C
Chris Mason 已提交
2421
	/* create a root if there isn't one */
C
Chris Mason 已提交
2422
	if (!root->node)
C
Chris Mason 已提交
2423
		BUG();
2424

2425 2426
	total_size = total_data + (nr - 1) * sizeof(struct btrfs_item);
	ret = btrfs_search_slot(trans, root, cpu_key, path, total_size, 1);
2427
	if (ret == 0) {
2428
		return -EEXIST;
C
Chris Mason 已提交
2429
	}
2430 2431
	if (ret < 0)
		goto out;
2432

2433
	slot_orig = path->slots[0];
2434
	leaf = path->nodes[0];
C
Chris Mason 已提交
2435

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

C
Chris Mason 已提交
2439
	if (btrfs_leaf_free_space(root, leaf) <
2440
	    sizeof(struct btrfs_item) + total_size) {
2441 2442
		btrfs_print_leaf(root, leaf);
		printk("not enough freespace need %u have %d\n",
2443
		       total_size, btrfs_leaf_free_space(root, leaf));
2444
		BUG();
2445
	}
2446

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

2450 2451
	if (slot != nritems) {
		int i;
2452
		unsigned int old_data = btrfs_item_end_nr(leaf, slot);
2453

2454 2455 2456 2457 2458 2459
		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);
		}
2460 2461 2462 2463
		/*
		 * item0..itemN ... dataN.offset..dataN.size .. data0.size
		 */
		/* first correct the data pointers */
2464
		WARN_ON(leaf->map_token);
C
Chris Mason 已提交
2465
		for (i = slot; i < nritems; i++) {
2466
			u32 ioff;
2467

2468
			item = btrfs_item_nr(leaf, i);
2469 2470 2471 2472 2473 2474 2475 2476
			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);
			}

2477
			ioff = btrfs_item_offset(leaf, item);
2478
			btrfs_set_item_offset(leaf, item, ioff - total_data);
C
Chris Mason 已提交
2479
		}
2480 2481 2482 2483
		if (leaf->map_token) {
			unmap_extent_buffer(leaf, leaf->map_token, KM_USER1);
			leaf->map_token = NULL;
		}
2484 2485

		/* shift the items */
2486
		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot + nr),
2487
			      btrfs_item_nr_offset(slot),
C
Chris Mason 已提交
2488
			      (nritems - slot) * sizeof(struct btrfs_item));
2489 2490

		/* shift the data */
2491
		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
2492
			      data_end - total_data, btrfs_leaf_data(leaf) +
C
Chris Mason 已提交
2493
			      data_end, old_data - data_end);
2494 2495
		data_end = old_data;
	}
2496

2497
	/* setup the item for the new data */
2498 2499 2500 2501 2502 2503 2504 2505 2506
	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);
2507
	btrfs_mark_buffer_dirty(leaf);
C
Chris Mason 已提交
2508 2509

	ret = 0;
2510 2511
	if (slot == 0) {
		btrfs_cpu_key_to_disk(&disk_key, cpu_key);
2512
		ret = fixup_low_keys(trans, root, path, &disk_key, 1);
2513
	}
C
Chris Mason 已提交
2514

2515 2516
	if (btrfs_leaf_free_space(root, leaf) < 0) {
		btrfs_print_leaf(root, leaf);
2517
		BUG();
2518
	}
2519

2520
out:
2521 2522 2523 2524 2525 2526 2527
	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.
 */
2528 2529 2530
int btrfs_insert_item(struct btrfs_trans_handle *trans, struct btrfs_root
		      *root, struct btrfs_key *cpu_key, void *data, u32
		      data_size)
2531 2532
{
	int ret = 0;
C
Chris Mason 已提交
2533
	struct btrfs_path *path;
2534 2535
	struct extent_buffer *leaf;
	unsigned long ptr;
2536

C
Chris Mason 已提交
2537 2538 2539
	path = btrfs_alloc_path();
	BUG_ON(!path);
	ret = btrfs_insert_empty_item(trans, root, path, cpu_key, data_size);
2540
	if (!ret) {
2541 2542 2543 2544
		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);
2545
	}
C
Chris Mason 已提交
2546
	btrfs_free_path(path);
C
Chris Mason 已提交
2547
	return ret;
2548 2549
}

C
Chris Mason 已提交
2550
/*
C
Chris Mason 已提交
2551
 * delete the pointer from a given node.
C
Chris Mason 已提交
2552 2553 2554 2555 2556
 *
 * 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.
 */
2557 2558
static int del_ptr(struct btrfs_trans_handle *trans, struct btrfs_root *root,
		   struct btrfs_path *path, int level, int slot)
2559
{
2560
	struct extent_buffer *parent = path->nodes[level];
2561
	u32 nritems;
C
Chris Mason 已提交
2562
	int ret = 0;
2563
	int wret;
2564

2565
	nritems = btrfs_header_nritems(parent);
2566
	if (slot != nritems -1) {
2567 2568 2569
		memmove_extent_buffer(parent,
			      btrfs_node_key_ptr_offset(slot),
			      btrfs_node_key_ptr_offset(slot + 1),
C
Chris Mason 已提交
2570 2571
			      sizeof(struct btrfs_key_ptr) *
			      (nritems - slot - 1));
2572
	}
2573
	nritems--;
2574
	btrfs_set_header_nritems(parent, nritems);
2575
	if (nritems == 0 && parent == root->node) {
2576
		BUG_ON(btrfs_header_level(root->node) != 1);
2577
		/* just turn the root into a leaf and break */
2578
		btrfs_set_header_level(root->node, 0);
2579
	} else if (slot == 0) {
2580 2581 2582 2583
		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 已提交
2584 2585
		if (wret)
			ret = wret;
2586
	}
C
Chris Mason 已提交
2587
	btrfs_mark_buffer_dirty(parent);
C
Chris Mason 已提交
2588
	return ret;
2589 2590
}

C
Chris Mason 已提交
2591 2592 2593 2594
/*
 * delete the item at the leaf level in path.  If that empties
 * the leaf, remove it from the tree
 */
2595 2596
int btrfs_del_items(struct btrfs_trans_handle *trans, struct btrfs_root *root,
		    struct btrfs_path *path, int slot, int nr)
2597
{
2598 2599
	struct extent_buffer *leaf;
	struct btrfs_item *item;
2600 2601
	int last_off;
	int dsize = 0;
C
Chris Mason 已提交
2602 2603
	int ret = 0;
	int wret;
2604
	int i;
2605
	u32 nritems;
2606

2607
	leaf = path->nodes[0];
2608 2609 2610 2611 2612
	last_off = btrfs_item_offset_nr(leaf, slot + nr - 1);

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

2613
	nritems = btrfs_header_nritems(leaf);
2614

2615
	if (slot + nr != nritems) {
2616
		int i;
C
Chris Mason 已提交
2617
		int data_end = leaf_data_end(root, leaf);
2618 2619

		memmove_extent_buffer(leaf, btrfs_leaf_data(leaf) +
C
Chris Mason 已提交
2620 2621
			      data_end + dsize,
			      btrfs_leaf_data(leaf) + data_end,
2622
			      last_off - data_end);
2623

2624
		for (i = slot + nr; i < nritems; i++) {
2625
			u32 ioff;
2626

2627
			item = btrfs_item_nr(leaf, i);
2628 2629 2630 2631 2632 2633 2634
			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);
			}
2635 2636
			ioff = btrfs_item_offset(leaf, item);
			btrfs_set_item_offset(leaf, item, ioff + dsize);
C
Chris Mason 已提交
2637
		}
2638 2639 2640 2641 2642 2643

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

2644
		memmove_extent_buffer(leaf, btrfs_item_nr_offset(slot),
2645
			      btrfs_item_nr_offset(slot + nr),
C
Chris Mason 已提交
2646
			      sizeof(struct btrfs_item) *
2647
			      (nritems - slot - nr));
2648
	}
2649 2650
	btrfs_set_header_nritems(leaf, nritems - nr);
	nritems -= nr;
2651

C
Chris Mason 已提交
2652
	/* delete the leaf if we've emptied it */
2653
	if (nritems == 0) {
2654 2655
		if (leaf == root->node) {
			btrfs_set_header_level(leaf, 0);
2656
		} else {
2657
			u64 root_gen = btrfs_header_generation(path->nodes[1]);
2658
			clean_tree_block(trans, root, leaf);
2659
			wret = del_ptr(trans, root, path, 1, path->slots[1]);
C
Chris Mason 已提交
2660 2661
			if (wret)
				ret = wret;
2662
			wret = btrfs_free_extent(trans, root,
2663 2664 2665
					 leaf->start, leaf->len,
					 btrfs_header_owner(path->nodes[1]),
					 root_gen, 0, 0, 1);
C
Chris Mason 已提交
2666 2667
			if (wret)
				ret = wret;
2668
		}
2669
	} else {
2670
		int used = leaf_space_used(leaf, 0, nritems);
C
Chris Mason 已提交
2671
		if (slot == 0) {
2672 2673 2674
			struct btrfs_disk_key disk_key;

			btrfs_item_key(leaf, &disk_key, 0);
2675
			wret = fixup_low_keys(trans, root, path,
2676
					      &disk_key, 1);
C
Chris Mason 已提交
2677 2678 2679 2680
			if (wret)
				ret = wret;
		}

C
Chris Mason 已提交
2681
		/* delete the leaf if it is mostly empty */
2682
		if (used < BTRFS_LEAF_DATA_SIZE(root) / 4) {
2683 2684 2685 2686
			/* push_leaf_left fixes the path.
			 * make sure the path still points to our leaf
			 * for possible call to del_ptr below
			 */
2687
			slot = path->slots[1];
2688 2689
			extent_buffer_get(leaf);

2690
			wret = push_leaf_left(trans, root, path, 1, 1);
2691
			if (wret < 0 && wret != -ENOSPC)
C
Chris Mason 已提交
2692
				ret = wret;
2693 2694 2695

			if (path->nodes[0] == leaf &&
			    btrfs_header_nritems(leaf)) {
2696
				wret = push_leaf_right(trans, root, path, 1, 1);
2697
				if (wret < 0 && wret != -ENOSPC)
C
Chris Mason 已提交
2698 2699
					ret = wret;
			}
2700 2701

			if (btrfs_header_nritems(leaf) == 0) {
2702
				u64 root_gen;
2703 2704
				u64 bytenr = leaf->start;
				u32 blocksize = leaf->len;
2705

2706 2707 2708
				root_gen = btrfs_header_generation(
							   path->nodes[1]);

2709 2710
				clean_tree_block(trans, root, leaf);

2711
				wret = del_ptr(trans, root, path, 1, slot);
C
Chris Mason 已提交
2712 2713
				if (wret)
					ret = wret;
2714 2715

				free_extent_buffer(leaf);
2716
				wret = btrfs_free_extent(trans, root, bytenr,
2717 2718 2719
					     blocksize,
					     btrfs_header_owner(path->nodes[1]),
					     root_gen, 0, 0, 1);
C
Chris Mason 已提交
2720 2721
				if (wret)
					ret = wret;
C
Chris Mason 已提交
2722
			} else {
2723 2724
				btrfs_mark_buffer_dirty(leaf);
				free_extent_buffer(leaf);
2725
			}
2726
		} else {
2727
			btrfs_mark_buffer_dirty(leaf);
2728 2729
		}
	}
C
Chris Mason 已提交
2730
	return ret;
2731 2732
}

2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761
/*
 * walk up the tree as far as required to find the previous leaf.
 * 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)
{
	int slot;
	int level = 1;
	struct extent_buffer *c;
	struct extent_buffer *next = NULL;

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

		slot = path->slots[level];
		c = path->nodes[level];
		if (slot == 0) {
			level++;
			if (level == BTRFS_MAX_LEVEL)
				return 1;
			continue;
		}
		slot--;

		if (next)
			free_extent_buffer(next);

2762
		next = read_node_slot(root, c, slot);
2763 2764 2765 2766 2767 2768 2769
		break;
	}
	path->slots[level] = slot;
	while(1) {
		level--;
		c = path->nodes[level];
		free_extent_buffer(c);
2770 2771 2772
		slot = btrfs_header_nritems(next);
		if (slot != 0)
			slot--;
2773
		path->nodes[level] = next;
2774
		path->slots[level] = slot;
2775 2776
		if (!level)
			break;
2777
		next = read_node_slot(root, next, slot);
2778 2779 2780 2781
	}
	return 0;
}

C
Chris Mason 已提交
2782 2783
/*
 * walk up the tree as far as required to find the next leaf.
C
Chris Mason 已提交
2784 2785
 * returns 0 if it found something or 1 if there are no greater leaves.
 * returns < 0 on io errors.
C
Chris Mason 已提交
2786
 */
C
Chris Mason 已提交
2787
int btrfs_next_leaf(struct btrfs_root *root, struct btrfs_path *path)
2788 2789 2790
{
	int slot;
	int level = 1;
2791 2792
	struct extent_buffer *c;
	struct extent_buffer *next = NULL;
2793

C
Chris Mason 已提交
2794
	while(level < BTRFS_MAX_LEVEL) {
2795
		if (!path->nodes[level])
C
Chris Mason 已提交
2796
			return 1;
2797

2798 2799
		slot = path->slots[level] + 1;
		c = path->nodes[level];
2800
		if (slot >= btrfs_header_nritems(c)) {
2801
			level++;
2802 2803
			if (level == BTRFS_MAX_LEVEL)
				return 1;
2804 2805
			continue;
		}
2806

C
Chris Mason 已提交
2807
		if (next)
2808 2809
			free_extent_buffer(next);

2810
		if (path->reada)
2811
			reada_for_search(root, path, level, slot, 0);
2812

2813
		next = read_node_slot(root, c, slot);
2814 2815 2816 2817 2818 2819
		break;
	}
	path->slots[level] = slot;
	while(1) {
		level--;
		c = path->nodes[level];
2820
		free_extent_buffer(c);
2821 2822 2823 2824
		path->nodes[level] = next;
		path->slots[level] = 0;
		if (!level)
			break;
2825
		if (path->reada)
2826
			reada_for_search(root, path, level, 0, 0);
2827
		next = read_node_slot(root, next, 0);
2828 2829 2830
	}
	return 0;
}
2831 2832 2833 2834 2835 2836 2837 2838 2839 2840 2841 2842 2843 2844 2845 2846 2847 2848 2849 2850 2851 2852 2853 2854 2855

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