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

C
Chris Mason 已提交
19
#include <linux/fs.h>
20
#include <linux/blkdev.h>
21
#include <linux/crc32c.h>
C
Chris Mason 已提交
22
#include <linux/scatterlist.h>
C
Chris Mason 已提交
23
#include <linux/swap.h>
24
#include <linux/radix-tree.h>
C
Chris Mason 已提交
25
#include <linux/writeback.h>
26
#include <linux/buffer_head.h> // for block_sync_page
27 28
#include "ctree.h"
#include "disk-io.h"
29
#include "transaction.h"
30
#include "btrfs_inode.h"
31
#include "print-tree.h"
32

33 34
#if 0
static int check_tree_block(struct btrfs_root *root, struct extent_buffer *buf)
35
{
36 37 38 39
	if (extent_buffer_blocknr(buf) != btrfs_header_blocknr(buf)) {
		printk(KERN_CRIT "buf blocknr(buf) is %llu, header is %llu\n",
		       (unsigned long long)extent_buffer_blocknr(buf),
		       (unsigned long long)btrfs_header_blocknr(buf));
C
Chris Mason 已提交
40
		return 1;
41
	}
42
	return 0;
43
}
44
#endif
45

46
struct extent_buffer *btrfs_find_tree_block(struct btrfs_root *root,
47
					    u64 bytenr, u32 blocksize)
48
{
49
	struct inode *btree_inode = root->fs_info->btree_inode;
50 51
	struct extent_buffer *eb;
	eb = find_extent_buffer(&BTRFS_I(btree_inode)->extent_tree,
52
				bytenr, blocksize, GFP_NOFS);
53
	return eb;
54
}
55

56
struct extent_buffer *btrfs_find_create_tree_block(struct btrfs_root *root,
57
						 u64 bytenr, u32 blocksize)
58 59
{
	struct inode *btree_inode = root->fs_info->btree_inode;
60
	struct extent_buffer *eb;
61

62
	eb = alloc_extent_buffer(&BTRFS_I(btree_inode)->extent_tree,
63
				 bytenr, blocksize, NULL, GFP_NOFS);
64
	return eb;
65 66
}

67 68 69
struct extent_map *btree_get_extent(struct inode *inode, struct page *page,
				    size_t page_offset, u64 start, u64 end,
				    int create)
70
{
71 72 73 74 75 76 77 78
	struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree;
	struct extent_map *em;
	int ret;

again:
	em = lookup_extent_mapping(em_tree, start, end);
	if (em) {
		goto out;
79
	}
80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
	em = alloc_extent_map(GFP_NOFS);
	if (!em) {
		em = ERR_PTR(-ENOMEM);
		goto out;
	}
	em->start = 0;
	em->end = (i_size_read(inode) & ~((u64)PAGE_CACHE_SIZE -1)) - 1;
	em->block_start = 0;
	em->block_end = em->end;
	em->bdev = inode->i_sb->s_bdev;
	ret = add_extent_mapping(em_tree, em);
	if (ret == -EEXIST) {
		free_extent_map(em);
		em = NULL;
		goto again;
	} else if (ret) {
		em = ERR_PTR(ret);
	}
out:
	return em;
100 101
}

102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190
u32 btrfs_csum_data(struct btrfs_root *root, char *data, u32 seed, size_t len)
{
	return crc32c(seed, data, len);
}

void btrfs_csum_final(u32 crc, char *result)
{
	*(__le32 *)result = ~cpu_to_le32(crc);
}

static int csum_tree_block(struct btrfs_root *root, struct extent_buffer *buf,
			   int verify)
{
	char result[BTRFS_CRC32_SIZE];
	unsigned long len;
	unsigned long cur_len;
	unsigned long offset = BTRFS_CSUM_SIZE;
	char *map_token = NULL;
	char *kaddr;
	unsigned long map_start;
	unsigned long map_len;
	int err;
	u32 crc = ~(u32)0;

	len = buf->len - offset;
	while(len > 0) {
		err = map_private_extent_buffer(buf, offset, 32,
					&map_token, &kaddr,
					&map_start, &map_len, KM_USER0);
		if (err) {
			printk("failed to map extent buffer! %lu\n",
			       offset);
			return 1;
		}
		cur_len = min(len, map_len - (offset - map_start));
		crc = btrfs_csum_data(root, kaddr + offset - map_start,
				      crc, cur_len);
		len -= cur_len;
		offset += cur_len;
		unmap_extent_buffer(buf, map_token, KM_USER0);
	}
	btrfs_csum_final(crc, result);

	if (verify) {
		if (memcmp_extent_buffer(buf, result, 0, BTRFS_CRC32_SIZE)) {
			printk("btrfs: %s checksum verify failed on %llu\n",
			       root->fs_info->sb->s_id,
			       buf->start);
			return 1;
		}
	} else {
		write_extent_buffer(buf, result, 0, BTRFS_CRC32_SIZE);
	}
	return 0;
}


int csum_dirty_buffer(struct btrfs_root *root, struct page *page)
{
	struct extent_map_tree *tree;
	u64 start = page->index << PAGE_CACHE_SHIFT;
	u64 found_start;
	int found_level;
	unsigned long len;
	struct extent_buffer *eb;
	tree = &BTRFS_I(page->mapping->host)->extent_tree;

	if (page->private == EXTENT_PAGE_PRIVATE)
		goto out;
	if (!page->private)
		goto out;
	len = page->private >> 2;
	if (len == 0) {
		WARN_ON(1);
	}
	eb = alloc_extent_buffer(tree, start, len, page, GFP_NOFS);
	read_extent_buffer_pages(tree, eb, start + PAGE_CACHE_SIZE, 1);
	found_start = btrfs_header_bytenr(eb);
	if (found_start != start) {
		printk("warning: eb start incorrect %Lu buffer %Lu len %lu\n",
		       start, found_start, len);
	}
	found_level = btrfs_header_level(eb);
	csum_tree_block(root, eb, 0);
	free_extent_buffer(eb);
out:
	return 0;
}

191
static int btree_writepage(struct page *page, struct writeback_control *wbc)
192
{
193
	struct extent_map_tree *tree;
194
	struct btrfs_root *root = BTRFS_I(page->mapping->host)->root;
195
	tree = &BTRFS_I(page->mapping->host)->extent_tree;
196 197

	csum_dirty_buffer(root, page);
198 199 200 201 202 203 204 205
	return extent_write_full_page(tree, page, btree_get_extent, wbc);
}
int btree_readpage(struct file *file, struct page *page)
{
	struct extent_map_tree *tree;
	tree = &BTRFS_I(page->mapping->host)->extent_tree;
	return extent_read_full_page(tree, page, btree_get_extent);
}
C
Chris Mason 已提交
206

207 208 209 210
static int btree_releasepage(struct page *page, gfp_t unused_gfp_flags)
{
	struct extent_map_tree *tree;
	int ret;
211

212 213 214 215 216 217 218
	tree = &BTRFS_I(page->mapping->host)->extent_tree;
	ret = try_release_extent_mapping(tree, page);
	if (ret == 1) {
		ClearPagePrivate(page);
		set_page_private(page, 0);
		page_cache_release(page);
	}
219 220 221
	return ret;
}

222
static void btree_invalidatepage(struct page *page, unsigned long offset)
223
{
224 225 226 227
	struct extent_map_tree *tree;
	tree = &BTRFS_I(page->mapping->host)->extent_tree;
	extent_invalidatepage(tree, page, offset);
	btree_releasepage(page, GFP_NOFS);
228 229
}

230
#if 0
231
static int btree_writepage(struct page *page, struct writeback_control *wbc)
232
{
C
Chris Mason 已提交
233
	struct buffer_head *bh;
234
	struct btrfs_root *root = BTRFS_I(page->mapping->host)->root;
C
Chris Mason 已提交
235 236 237 238 239 240 241 242 243 244 245 246
	struct buffer_head *head;
	if (!page_has_buffers(page)) {
		create_empty_buffers(page, root->fs_info->sb->s_blocksize,
					(1 << BH_Dirty)|(1 << BH_Uptodate));
	}
	head = page_buffers(page);
	bh = head;
	do {
		if (buffer_dirty(bh))
			csum_tree_block(root, bh, 0);
		bh = bh->b_this_page;
	} while (bh != head);
247
	return block_write_full_page(page, btree_get_block, wbc);
248
}
249
#endif
250

251 252 253
static struct address_space_operations btree_aops = {
	.readpage	= btree_readpage,
	.writepage	= btree_writepage,
254 255
	.releasepage	= btree_releasepage,
	.invalidatepage = btree_invalidatepage,
256 257 258
	.sync_page	= block_sync_page,
};

259
int readahead_tree_block(struct btrfs_root *root, u64 bytenr, u32 blocksize)
C
Chris Mason 已提交
260
{
261 262
	struct extent_buffer *buf = NULL;
	struct inode *btree_inode = root->fs_info->btree_inode;
263
	int ret = 0;
C
Chris Mason 已提交
264

265
	buf = btrfs_find_create_tree_block(root, bytenr, blocksize);
266
	if (!buf)
C
Chris Mason 已提交
267
		return 0;
268
	read_extent_buffer_pages(&BTRFS_I(btree_inode)->extent_tree,
269
				 buf, 0, 0);
270
	free_extent_buffer(buf);
271
	return ret;
C
Chris Mason 已提交
272 273
}

274 275
struct extent_buffer *read_tree_block(struct btrfs_root *root, u64 bytenr,
				      u32 blocksize)
276
{
277 278
	struct extent_buffer *buf = NULL;
	struct inode *btree_inode = root->fs_info->btree_inode;
279 280 281 282
	struct extent_map_tree *extent_tree;
	int ret;

	extent_tree = &BTRFS_I(btree_inode)->extent_tree;
283

284
	buf = btrfs_find_create_tree_block(root, bytenr, blocksize);
285 286 287
	if (!buf)
		return NULL;
	read_extent_buffer_pages(&BTRFS_I(btree_inode)->extent_tree,
288 289 290 291 292 293 294 295 296 297 298 299 300 301
				 buf, 0, 1);
	if (buf->flags & EXTENT_CSUM) {
		return buf;
	}
	if (test_range_bit(extent_tree, buf->start, buf->start + buf->len - 1,
			   EXTENT_CSUM, 1)) {
		buf->flags |= EXTENT_CSUM;
		return buf;
	}
	ret = csum_tree_block(root, buf, 1);
	set_extent_bits(extent_tree, buf->start,
			buf->start + buf->len - 1,
			EXTENT_CSUM, GFP_NOFS);
	buf->flags |= EXTENT_CSUM;
302
	return buf;
303 304
}

305
int clean_tree_block(struct btrfs_trans_handle *trans, struct btrfs_root *root,
306
		     struct extent_buffer *buf)
307
{
308 309 310 311 312 313 314 315 316 317 318 319 320 321
	struct inode *btree_inode = root->fs_info->btree_inode;
	clear_extent_buffer_dirty(&BTRFS_I(btree_inode)->extent_tree, buf);
	return 0;
}

int wait_on_tree_block_writeback(struct btrfs_root *root,
				 struct extent_buffer *buf)
{
	struct inode *btree_inode = root->fs_info->btree_inode;
	wait_on_extent_buffer_writeback(&BTRFS_I(btree_inode)->extent_tree,
					buf);
	return 0;
}

322
static int __setup_root(u32 nodesize, u32 leafsize, u32 sectorsize,
323 324
			struct btrfs_root *root,
			struct btrfs_fs_info *fs_info,
C
Chris Mason 已提交
325
			u64 objectid)
326
{
C
Chris Mason 已提交
327
	root->node = NULL;
328
	root->inode = NULL;
329
	root->commit_root = NULL;
330 331 332
	root->sectorsize = sectorsize;
	root->nodesize = nodesize;
	root->leafsize = leafsize;
C
Chris Mason 已提交
333
	root->ref_cows = 0;
334
	root->fs_info = fs_info;
335 336
	root->objectid = objectid;
	root->last_trans = 0;
C
Chris Mason 已提交
337 338
	root->highest_inode = 0;
	root->last_inode_alloc = 0;
339
	root->name = NULL;
340 341
	memset(&root->root_key, 0, sizeof(root->root_key));
	memset(&root->root_item, 0, sizeof(root->root_item));
342
	memset(&root->defrag_progress, 0, sizeof(root->defrag_progress));
343 344
	memset(&root->root_kobj, 0, sizeof(root->root_kobj));
	init_completion(&root->kobj_unregister);
345
	init_rwsem(&root->snap_sem);
346 347
	root->defrag_running = 0;
	root->defrag_level = 0;
348
	root->root_key.objectid = objectid;
349 350 351
	return 0;
}

352
static int find_and_setup_root(struct btrfs_root *tree_root,
353 354
			       struct btrfs_fs_info *fs_info,
			       u64 objectid,
C
Chris Mason 已提交
355
			       struct btrfs_root *root)
356 357
{
	int ret;
358
	u32 blocksize;
359

360 361
	__setup_root(tree_root->nodesize, tree_root->leafsize,
		     tree_root->sectorsize, root, fs_info, objectid);
362 363 364 365
	ret = btrfs_find_last_root(tree_root, objectid,
				   &root->root_item, &root->root_key);
	BUG_ON(ret);

366 367 368
	blocksize = btrfs_level_size(root, btrfs_root_level(&root->root_item));
	root->node = read_tree_block(root, btrfs_root_bytenr(&root->root_item),
				     blocksize);
369
	BUG_ON(!root->node);
370 371 372
	return 0;
}

373 374
struct btrfs_root *btrfs_read_fs_root_no_radix(struct btrfs_fs_info *fs_info,
					       struct btrfs_key *location)
375 376 377 378
{
	struct btrfs_root *root;
	struct btrfs_root *tree_root = fs_info->tree_root;
	struct btrfs_path *path;
379
	struct extent_buffer *l;
C
Chris Mason 已提交
380
	u64 highest_inode;
381
	u32 blocksize;
382 383
	int ret = 0;

384
	root = kzalloc(sizeof(*root), GFP_NOFS);
C
Chris Mason 已提交
385
	if (!root)
386 387
		return ERR_PTR(-ENOMEM);
	if (location->offset == (u64)-1) {
388
		ret = find_and_setup_root(tree_root, fs_info,
389 390 391 392 393 394 395 396
					  location->objectid, root);
		if (ret) {
			kfree(root);
			return ERR_PTR(ret);
		}
		goto insert;
	}

397 398
	__setup_root(tree_root->nodesize, tree_root->leafsize,
		     tree_root->sectorsize, root, fs_info,
399 400 401 402 403 404 405 406 407 408
		     location->objectid);

	path = btrfs_alloc_path();
	BUG_ON(!path);
	ret = btrfs_search_slot(NULL, tree_root, location, path, 0, 0);
	if (ret != 0) {
		if (ret > 0)
			ret = -ENOENT;
		goto out;
	}
409 410 411
	l = path->nodes[0];
	read_extent_buffer(l, &root->root_item,
	       btrfs_item_ptr_offset(l, path->slots[0]),
412 413 414 415 416 417 418 419 420
	       sizeof(root->root_item));
	ret = 0;
out:
	btrfs_release_path(root, path);
	btrfs_free_path(path);
	if (ret) {
		kfree(root);
		return ERR_PTR(ret);
	}
421 422 423
	blocksize = btrfs_level_size(root, btrfs_root_level(&root->root_item));
	root->node = read_tree_block(root, btrfs_root_bytenr(&root->root_item),
				     blocksize);
424 425 426
	BUG_ON(!root->node);
insert:
	root->ref_cows = 1;
427 428 429 430 431 432 433 434 435
	ret = btrfs_find_highest_inode(root, &highest_inode);
	if (ret == 0) {
		root->highest_inode = highest_inode;
		root->last_inode_alloc = highest_inode;
	}
	return root;
}

struct btrfs_root *btrfs_read_fs_root(struct btrfs_fs_info *fs_info,
436 437
				      struct btrfs_key *location,
				      const char *name, int namelen)
438 439 440 441 442 443 444 445 446 447 448 449
{
	struct btrfs_root *root;
	int ret;

	root = radix_tree_lookup(&fs_info->fs_roots_radix,
				 (unsigned long)location->objectid);
	if (root)
		return root;

	root = btrfs_read_fs_root_no_radix(fs_info, location);
	if (IS_ERR(root))
		return root;
C
Chris Mason 已提交
450 451
	ret = radix_tree_insert(&fs_info->fs_roots_radix,
				(unsigned long)root->root_key.objectid,
452 453
				root);
	if (ret) {
454
		free_extent_buffer(root->node);
455 456 457
		kfree(root);
		return ERR_PTR(ret);
	}
458 459 460

	ret = btrfs_set_root_name(root, name, namelen);
	if (ret) {
461
		free_extent_buffer(root->node);
462 463 464 465 466 467
		kfree(root);
		return ERR_PTR(ret);
	}

	ret = btrfs_sysfs_add_root(root);
	if (ret) {
468
		free_extent_buffer(root->node);
469 470 471 472 473
		kfree(root->name);
		kfree(root);
		return ERR_PTR(ret);
	}

474 475 476 477
	ret = btrfs_find_dead_roots(fs_info->tree_root,
				    root->root_key.objectid, root);
	BUG_ON(ret);

478 479
	return root;
}
480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497
#if 0
static int add_hasher(struct btrfs_fs_info *info, char *type) {
	struct btrfs_hasher *hasher;

	hasher = kmalloc(sizeof(*hasher), GFP_NOFS);
	if (!hasher)
		return -ENOMEM;
	hasher->hash_tfm = crypto_alloc_hash(type, 0, CRYPTO_ALG_ASYNC);
	if (!hasher->hash_tfm) {
		kfree(hasher);
		return -EINVAL;
	}
	spin_lock(&info->hash_lock);
	list_add(&hasher->list, &info->hashers);
	spin_unlock(&info->hash_lock);
	return 0;
}
#endif
C
Chris Mason 已提交
498
struct btrfs_root *open_ctree(struct super_block *sb)
499
{
500 501 502 503
	u32 sectorsize;
	u32 nodesize;
	u32 leafsize;
	u32 blocksize;
C
Chris Mason 已提交
504 505 506 507 508 509
	struct btrfs_root *extent_root = kmalloc(sizeof(struct btrfs_root),
						 GFP_NOFS);
	struct btrfs_root *tree_root = kmalloc(sizeof(struct btrfs_root),
					       GFP_NOFS);
	struct btrfs_fs_info *fs_info = kmalloc(sizeof(*fs_info),
						GFP_NOFS);
510
	int ret;
C
Chris Mason 已提交
511
	int err = -EIO;
C
Chris Mason 已提交
512
	struct btrfs_super_block *disk_super;
513

C
Chris Mason 已提交
514 515 516 517
	if (!extent_root || !tree_root || !fs_info) {
		err = -ENOMEM;
		goto fail;
	}
518
	INIT_RADIX_TREE(&fs_info->fs_roots_radix, GFP_NOFS);
C
Chris Mason 已提交
519
	INIT_LIST_HEAD(&fs_info->trans_list);
520
	INIT_LIST_HEAD(&fs_info->dead_roots);
521 522 523
	INIT_LIST_HEAD(&fs_info->hashers);
	spin_lock_init(&fs_info->hash_lock);

524 525
	memset(&fs_info->super_kobj, 0, sizeof(fs_info->super_kobj));
	init_completion(&fs_info->kobj_unregister);
C
Chris Mason 已提交
526
	sb_set_blocksize(sb, 4096);
527
	fs_info->running_transaction = NULL;
528
	fs_info->last_trans_committed = 0;
529 530
	fs_info->tree_root = tree_root;
	fs_info->extent_root = extent_root;
C
Chris Mason 已提交
531
	fs_info->sb = sb;
532 533
	fs_info->btree_inode = new_inode(sb);
	fs_info->btree_inode->i_ino = 1;
C
Chris Mason 已提交
534
	fs_info->btree_inode->i_nlink = 1;
535 536
	fs_info->btree_inode->i_size = sb->s_bdev->bd_inode->i_size;
	fs_info->btree_inode->i_mapping->a_ops = &btree_aops;
537 538 539
	extent_map_tree_init(&BTRFS_I(fs_info->btree_inode)->extent_tree,
			     fs_info->btree_inode->i_mapping,
			     GFP_NOFS);
540 541
	extent_map_tree_init(&fs_info->free_space_cache,
			     fs_info->btree_inode->i_mapping, GFP_NOFS);
542 543
	extent_map_tree_init(&fs_info->block_group_cache,
			     fs_info->btree_inode->i_mapping, GFP_NOFS);
544 545 546 547 548 549
	extent_map_tree_init(&fs_info->pinned_extents,
			     fs_info->btree_inode->i_mapping, GFP_NOFS);
	extent_map_tree_init(&fs_info->pending_del,
			     fs_info->btree_inode->i_mapping, GFP_NOFS);
	extent_map_tree_init(&fs_info->extent_ins,
			     fs_info->btree_inode->i_mapping, GFP_NOFS);
550
	fs_info->do_barriers = 1;
551 552
	fs_info->closing = 0;

C
Chris Mason 已提交
553
	INIT_DELAYED_WORK(&fs_info->trans_work, btrfs_transaction_cleaner);
554 555 556
	BTRFS_I(fs_info->btree_inode)->root = tree_root;
	memset(&BTRFS_I(fs_info->btree_inode)->location, 0,
	       sizeof(struct btrfs_key));
C
Chris Mason 已提交
557
	insert_inode_hash(fs_info->btree_inode);
558
	mapping_set_gfp_mask(fs_info->btree_inode->i_mapping, GFP_NOFS);
C
Chris Mason 已提交
559

C
Chris Mason 已提交
560
	mutex_init(&fs_info->trans_mutex);
C
Chris Mason 已提交
561
	mutex_init(&fs_info->fs_mutex);
562

563 564 565 566 567 568 569 570
#if 0
	ret = add_hasher(fs_info, "crc32c");
	if (ret) {
		printk("btrfs: failed hash setup, modprobe cryptomgr?\n");
		err = -ENOMEM;
		goto fail_iput;
	}
#endif
571
	__setup_root(512, 512, 512, tree_root,
C
Chris Mason 已提交
572
		     fs_info, BTRFS_ROOT_TREE_OBJECTID);
573

C
Chris Mason 已提交
574
	fs_info->sb_buffer = read_tree_block(tree_root,
575 576
					     BTRFS_SUPER_INFO_OFFSET,
					     512);
577

578
	if (!fs_info->sb_buffer)
C
Chris Mason 已提交
579 580
		goto fail_iput;

581 582 583 584 585 586 587
	read_extent_buffer(fs_info->sb_buffer, &fs_info->super_copy, 0,
			   sizeof(fs_info->super_copy));

	read_extent_buffer(fs_info->sb_buffer, fs_info->fsid,
			   (unsigned long)btrfs_super_fsid(fs_info->sb_buffer),
			   BTRFS_FSID_SIZE);
	disk_super = &fs_info->super_copy;
588
	if (!btrfs_super_root(disk_super))
C
Chris Mason 已提交
589
		goto fail_sb_buffer;
590

591 592 593 594 595 596
	nodesize = btrfs_super_nodesize(disk_super);
	leafsize = btrfs_super_leafsize(disk_super);
	sectorsize = btrfs_super_sectorsize(disk_super);
	tree_root->nodesize = nodesize;
	tree_root->leafsize = leafsize;
	tree_root->sectorsize = sectorsize;
597
	sb_set_blocksize(sb, sectorsize);
598

599
	i_size_write(fs_info->btree_inode,
600
		     btrfs_super_total_bytes(disk_super));
601

C
Chris Mason 已提交
602 603 604 605 606
	if (strncmp((char *)(&disk_super->magic), BTRFS_MAGIC,
		    sizeof(disk_super->magic))) {
		printk("btrfs: valid FS not found on %s\n", sb->s_id);
		goto fail_sb_buffer;
	}
607

608 609
	blocksize = btrfs_level_size(tree_root,
				     btrfs_super_root_level(disk_super));
610

C
Chris Mason 已提交
611
	tree_root->node = read_tree_block(tree_root,
612 613
					  btrfs_super_root(disk_super),
					  blocksize);
C
Chris Mason 已提交
614 615
	if (!tree_root->node)
		goto fail_sb_buffer;
616

C
Chris Mason 已提交
617
	mutex_lock(&fs_info->fs_mutex);
618 619

	ret = find_and_setup_root(tree_root, fs_info,
C
Chris Mason 已提交
620
				  BTRFS_EXTENT_TREE_OBJECTID, extent_root);
C
Chris Mason 已提交
621 622 623 624
	if (ret) {
		mutex_unlock(&fs_info->fs_mutex);
		goto fail_tree_root;
	}
625

C
Chris Mason 已提交
626 627
	btrfs_read_block_groups(extent_root);

628
	fs_info->generation = btrfs_super_generation(disk_super) + 1;
C
Chris Mason 已提交
629
	mutex_unlock(&fs_info->fs_mutex);
630
	return tree_root;
C
Chris Mason 已提交
631 632

fail_tree_root:
633
	free_extent_buffer(tree_root->node);
C
Chris Mason 已提交
634
fail_sb_buffer:
635
	free_extent_buffer(fs_info->sb_buffer);
C
Chris Mason 已提交
636 637 638 639 640 641 642
fail_iput:
	iput(fs_info->btree_inode);
fail:
	kfree(extent_root);
	kfree(tree_root);
	kfree(fs_info);
	return ERR_PTR(err);
643 644
}

645
int write_ctree_super(struct btrfs_trans_handle *trans, struct btrfs_root
C
Chris Mason 已提交
646
		      *root)
647
{
648
	int ret;
649 650 651 652 653 654 655
	struct extent_buffer *super = root->fs_info->sb_buffer;
	struct inode *btree_inode = root->fs_info->btree_inode;

	set_extent_buffer_dirty(&BTRFS_I(btree_inode)->extent_tree, super);
	ret = sync_page_range_nolock(btree_inode, btree_inode->i_mapping,
				     super->start, super->len);
	return ret;
C
Chris Mason 已提交
656 657
}

658
int btrfs_free_fs_root(struct btrfs_fs_info *fs_info, struct btrfs_root *root)
C
Chris Mason 已提交
659 660 661
{
	radix_tree_delete(&fs_info->fs_roots_radix,
			  (unsigned long)root->root_key.objectid);
662
	btrfs_sysfs_del_root(root);
C
Chris Mason 已提交
663 664 665
	if (root->inode)
		iput(root->inode);
	if (root->node)
666
		free_extent_buffer(root->node);
C
Chris Mason 已提交
667
	if (root->commit_root)
668
		free_extent_buffer(root->commit_root);
669 670
	if (root->name)
		kfree(root->name);
C
Chris Mason 已提交
671 672 673 674
	kfree(root);
	return 0;
}

C
Chris Mason 已提交
675
static int del_fs_roots(struct btrfs_fs_info *fs_info)
676 677 678 679 680 681 682 683 684 685 686
{
	int ret;
	struct btrfs_root *gang[8];
	int i;

	while(1) {
		ret = radix_tree_gang_lookup(&fs_info->fs_roots_radix,
					     (void **)gang, 0,
					     ARRAY_SIZE(gang));
		if (!ret)
			break;
C
Chris Mason 已提交
687
		for (i = 0; i < ret; i++)
688
			btrfs_free_fs_root(fs_info, gang[i]);
689 690 691
	}
	return 0;
}
692

C
Chris Mason 已提交
693
int close_ctree(struct btrfs_root *root)
C
Chris Mason 已提交
694
{
695
	int ret;
696
	struct btrfs_trans_handle *trans;
697
	struct btrfs_fs_info *fs_info = root->fs_info;
698

699
	fs_info->closing = 1;
C
Chris Mason 已提交
700
	btrfs_transaction_flush_work(root);
701
	mutex_lock(&fs_info->fs_mutex);
702
	btrfs_defrag_dirty_roots(root->fs_info);
C
Chris Mason 已提交
703
	trans = btrfs_start_transaction(root, 1);
704
	ret = btrfs_commit_transaction(trans, root);
C
Chris Mason 已提交
705 706 707 708
	/* run commit again to  drop the original snapshot */
	trans = btrfs_start_transaction(root, 1);
	btrfs_commit_transaction(trans, root);
	ret = btrfs_write_and_wait_transaction(NULL, root);
709
	BUG_ON(ret);
C
Chris Mason 已提交
710
	write_ctree_super(NULL, root);
711 712 713
	mutex_unlock(&fs_info->fs_mutex);

	if (fs_info->extent_root->node)
714
		free_extent_buffer(fs_info->extent_root->node);
715

716
	if (fs_info->tree_root->node)
717
		free_extent_buffer(fs_info->tree_root->node);
718

719
	free_extent_buffer(fs_info->sb_buffer);
720

C
Chris Mason 已提交
721
	btrfs_free_block_groups(root->fs_info);
722
	del_fs_roots(fs_info);
723
	extent_map_tree_empty_lru(&BTRFS_I(fs_info->btree_inode)->extent_tree);
724 725
	truncate_inode_pages(fs_info->btree_inode->i_mapping, 0);
	iput(fs_info->btree_inode);
726 727 728 729 730 731 732 733 734 735
#if 0
	while(!list_empty(&fs_info->hashers)) {
		struct btrfs_hasher *hasher;
		hasher = list_entry(fs_info->hashers.next, struct btrfs_hasher,
				    hashers);
		list_del(&hasher->hashers);
		crypto_free_hash(&fs_info->hash_tfm);
		kfree(hasher);
	}
#endif
736 737
	kfree(fs_info->extent_root);
	kfree(fs_info->tree_root);
738 739 740
	return 0;
}

741 742
int btrfs_buffer_uptodate(struct extent_buffer *buf)
{
743
	struct inode *btree_inode = buf->first_page->mapping->host;
744 745 746 747
	return extent_buffer_uptodate(&BTRFS_I(btree_inode)->extent_tree, buf);
}

int btrfs_set_buffer_uptodate(struct extent_buffer *buf)
C
Chris Mason 已提交
748
{
749
	struct inode *btree_inode = buf->first_page->mapping->host;
750 751 752
	return set_extent_buffer_uptodate(&BTRFS_I(btree_inode)->extent_tree,
					  buf);
}
753

754 755
void btrfs_mark_buffer_dirty(struct extent_buffer *buf)
{
756
	struct btrfs_root *root = BTRFS_I(buf->first_page->mapping->host)->root;
757 758
	u64 transid = btrfs_header_generation(buf);
	struct inode *btree_inode = root->fs_info->btree_inode;
759

C
Chris Mason 已提交
760 761
	if (transid != root->fs_info->generation) {
		printk(KERN_CRIT "transid mismatch buffer %llu, found %Lu running %Lu\n",
762
			(unsigned long long)buf->start,
C
Chris Mason 已提交
763 764 765
			transid, root->fs_info->generation);
		WARN_ON(1);
	}
766
	set_extent_buffer_dirty(&BTRFS_I(btree_inode)->extent_tree, buf);
767 768
}

769
void btrfs_btree_balance_dirty(struct btrfs_root *root, unsigned long nr)
C
Chris Mason 已提交
770
{
771
	balance_dirty_pages_ratelimited_nr(
772
			root->fs_info->btree_inode->i_mapping, 1);
C
Chris Mason 已提交
773
}
774 775 776

void btrfs_set_buffer_defrag(struct extent_buffer *buf)
{
777
	struct btrfs_root *root = BTRFS_I(buf->first_page->mapping->host)->root;
778 779 780 781 782 783 784
	struct inode *btree_inode = root->fs_info->btree_inode;
	set_extent_bits(&BTRFS_I(btree_inode)->extent_tree, buf->start,
			buf->start + buf->len - 1, EXTENT_DEFRAG, GFP_NOFS);
}

void btrfs_set_buffer_defrag_done(struct extent_buffer *buf)
{
785
	struct btrfs_root *root = BTRFS_I(buf->first_page->mapping->host)->root;
786 787 788 789 790 791 792 793
	struct inode *btree_inode = root->fs_info->btree_inode;
	set_extent_bits(&BTRFS_I(btree_inode)->extent_tree, buf->start,
			buf->start + buf->len - 1, EXTENT_DEFRAG_DONE,
			GFP_NOFS);
}

int btrfs_buffer_defrag(struct extent_buffer *buf)
{
794
	struct btrfs_root *root = BTRFS_I(buf->first_page->mapping->host)->root;
795 796 797 798 799 800 801
	struct inode *btree_inode = root->fs_info->btree_inode;
	return test_range_bit(&BTRFS_I(btree_inode)->extent_tree,
		     buf->start, buf->start + buf->len - 1, EXTENT_DEFRAG, 0);
}

int btrfs_buffer_defrag_done(struct extent_buffer *buf)
{
802
	struct btrfs_root *root = BTRFS_I(buf->first_page->mapping->host)->root;
803 804 805 806 807 808 809 810
	struct inode *btree_inode = root->fs_info->btree_inode;
	return test_range_bit(&BTRFS_I(btree_inode)->extent_tree,
		     buf->start, buf->start + buf->len - 1,
		     EXTENT_DEFRAG_DONE, 0);
}

int btrfs_clear_buffer_defrag_done(struct extent_buffer *buf)
{
811
	struct btrfs_root *root = BTRFS_I(buf->first_page->mapping->host)->root;
812 813 814 815 816 817 818 819
	struct inode *btree_inode = root->fs_info->btree_inode;
	return clear_extent_bits(&BTRFS_I(btree_inode)->extent_tree,
		     buf->start, buf->start + buf->len - 1,
		     EXTENT_DEFRAG_DONE, GFP_NOFS);
}

int btrfs_clear_buffer_defrag(struct extent_buffer *buf)
{
820
	struct btrfs_root *root = BTRFS_I(buf->first_page->mapping->host)->root;
821 822 823 824 825 826 827 828
	struct inode *btree_inode = root->fs_info->btree_inode;
	return clear_extent_bits(&BTRFS_I(btree_inode)->extent_tree,
		     buf->start, buf->start + buf->len - 1,
		     EXTENT_DEFRAG, GFP_NOFS);
}

int btrfs_read_buffer(struct extent_buffer *buf)
{
829
	struct btrfs_root *root = BTRFS_I(buf->first_page->mapping->host)->root;
830 831
	struct inode *btree_inode = root->fs_info->btree_inode;
	return read_extent_buffer_pages(&BTRFS_I(btree_inode)->extent_tree,
832
					buf, 0, 1);
833
}