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

19 20 21
#include <linux/bio.h>
#include <linux/pagemap.h>
#include <linux/highmem.h>
C
Chris Mason 已提交
22
#include "ctree.h"
C
Chris Mason 已提交
23
#include "disk-io.h"
24
#include "transaction.h"
C
Chris Mason 已提交
25
#include "print-tree.h"
C
Chris Mason 已提交
26

C
Chris Mason 已提交
27
#define MAX_CSUM_ITEMS(r, size) ((((BTRFS_LEAF_DATA_SIZE(r) - \
28 29
				   sizeof(struct btrfs_item) * 2) / \
				  size) - 1))
Y
Yan Zheng 已提交
30 31 32 33 34 35

#define MAX_ORDERED_SUM_BYTES(r) ((PAGE_SIZE - \
				   sizeof(struct btrfs_ordered_sum)) / \
				   sizeof(struct btrfs_sector_sum) * \
				   (r)->sectorsize - (r)->sectorsize)

C
Chris Mason 已提交
36
int btrfs_insert_file_extent(struct btrfs_trans_handle *trans,
S
Sage Weil 已提交
37 38 39
			     struct btrfs_root *root,
			     u64 objectid, u64 pos,
			     u64 disk_offset, u64 disk_num_bytes,
C
Chris Mason 已提交
40 41
			     u64 num_bytes, u64 offset, u64 ram_bytes,
			     u8 compression, u8 encryption, u16 other_encoding)
42
{
C
Chris Mason 已提交
43 44 45
	int ret = 0;
	struct btrfs_file_extent_item *item;
	struct btrfs_key file_key;
46
	struct btrfs_path *path;
47
	struct extent_buffer *leaf;
C
Chris Mason 已提交
48

49 50
	path = btrfs_alloc_path();
	BUG_ON(!path);
C
Chris Mason 已提交
51
	file_key.objectid = objectid;
C
Chris Mason 已提交
52
	file_key.offset = pos;
C
Chris Mason 已提交
53 54
	btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY);

55
	ret = btrfs_insert_empty_item(trans, root, path, &file_key,
C
Chris Mason 已提交
56
				      sizeof(*item));
57 58
	if (ret < 0)
		goto out;
C
Chris Mason 已提交
59
	BUG_ON(ret);
60 61
	leaf = path->nodes[0];
	item = btrfs_item_ptr(leaf, path->slots[0],
C
Chris Mason 已提交
62
			      struct btrfs_file_extent_item);
S
Sage Weil 已提交
63
	btrfs_set_file_extent_disk_bytenr(leaf, item, disk_offset);
64
	btrfs_set_file_extent_disk_num_bytes(leaf, item, disk_num_bytes);
S
Sage Weil 已提交
65
	btrfs_set_file_extent_offset(leaf, item, offset);
66
	btrfs_set_file_extent_num_bytes(leaf, item, num_bytes);
C
Chris Mason 已提交
67
	btrfs_set_file_extent_ram_bytes(leaf, item, ram_bytes);
68 69
	btrfs_set_file_extent_generation(leaf, item, trans->transid);
	btrfs_set_file_extent_type(leaf, item, BTRFS_FILE_EXTENT_REG);
C
Chris Mason 已提交
70 71 72 73
	btrfs_set_file_extent_compression(leaf, item, compression);
	btrfs_set_file_extent_encryption(leaf, item, encryption);
	btrfs_set_file_extent_other_encoding(leaf, item, other_encoding);

74
	btrfs_mark_buffer_dirty(leaf);
75
out:
76
	btrfs_free_path(path);
77
	return ret;
78
}
C
Chris Mason 已提交
79

C
Chris Mason 已提交
80 81 82
struct btrfs_csum_item *btrfs_lookup_csum(struct btrfs_trans_handle *trans,
					  struct btrfs_root *root,
					  struct btrfs_path *path,
83
					  u64 bytenr, int cow)
84 85 86 87 88
{
	int ret;
	struct btrfs_key file_key;
	struct btrfs_key found_key;
	struct btrfs_csum_item *item;
89
	struct extent_buffer *leaf;
90
	u64 csum_offset = 0;
91 92
	u16 csum_size =
		btrfs_super_csum_size(&root->fs_info->super_copy);
93
	int csums_in_item;
94

95 96 97
	file_key.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
	file_key.offset = bytenr;
	btrfs_set_key_type(&file_key, BTRFS_EXTENT_CSUM_KEY);
C
Chris Mason 已提交
98
	ret = btrfs_search_slot(trans, root, &file_key, path, 0, cow);
99 100
	if (ret < 0)
		goto fail;
101
	leaf = path->nodes[0];
102 103
	if (ret > 0) {
		ret = 1;
104
		if (path->slots[0] == 0)
105 106
			goto fail;
		path->slots[0]--;
107
		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
108
		if (btrfs_key_type(&found_key) != BTRFS_EXTENT_CSUM_KEY)
109
			goto fail;
110 111

		csum_offset = (bytenr - found_key.offset) >>
112
				root->fs_info->sb->s_blocksize_bits;
113
		csums_in_item = btrfs_item_size_nr(leaf, path->slots[0]);
114
		csums_in_item /= csum_size;
115 116 117

		if (csum_offset >= csums_in_item) {
			ret = -EFBIG;
118 119 120 121
			goto fail;
		}
	}
	item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item);
122
	item = (struct btrfs_csum_item *)((unsigned char *)item +
123
					  csum_offset * csum_size);
124 125 126
	return item;
fail:
	if (ret > 0)
C
Chris Mason 已提交
127
		ret = -ENOENT;
128 129 130 131
	return ERR_PTR(ret);
}


C
Chris Mason 已提交
132 133 134
int btrfs_lookup_file_extent(struct btrfs_trans_handle *trans,
			     struct btrfs_root *root,
			     struct btrfs_path *path, u64 objectid,
C
Chris Mason 已提交
135
			     u64 offset, int mod)
C
Chris Mason 已提交
136 137 138 139 140 141 142
{
	int ret;
	struct btrfs_key file_key;
	int ins_len = mod < 0 ? -1 : 0;
	int cow = mod != 0;

	file_key.objectid = objectid;
143
	file_key.offset = offset;
C
Chris Mason 已提交
144 145 146 147
	btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY);
	ret = btrfs_search_slot(trans, root, &file_key, path, ins_len, cow);
	return ret;
}
C
Chris Mason 已提交
148

149

150
int btrfs_lookup_bio_sums(struct btrfs_root *root, struct inode *inode,
151
			  struct bio *bio, u32 *dst)
152 153 154 155 156 157 158
{
	u32 sum;
	struct bio_vec *bvec = bio->bi_io_vec;
	int bio_index = 0;
	u64 offset;
	u64 item_start_offset = 0;
	u64 item_last_offset = 0;
159
	u64 disk_bytenr;
160
	u32 diff;
161 162
	u16 csum_size =
		btrfs_super_csum_size(&root->fs_info->super_copy);
163 164 165 166 167 168
	int ret;
	struct btrfs_path *path;
	struct btrfs_csum_item *item = NULL;
	struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree;

	path = btrfs_alloc_path();
169 170
	if (bio->bi_size > PAGE_CACHE_SIZE * 8)
		path->reada = 2;
171 172 173

	WARN_ON(bio->bi_vcnt <= 0);

174
	disk_bytenr = (u64)bio->bi_sector << 9;
C
Chris Mason 已提交
175
	while (bio_index < bio->bi_vcnt) {
176
		offset = page_offset(bvec->bv_page) + bvec->bv_offset;
177
		ret = btrfs_find_ordered_sum(inode, offset, disk_bytenr, &sum);
178 179 180
		if (ret == 0)
			goto found;

181 182
		if (!item || disk_bytenr < item_start_offset ||
		    disk_bytenr >= item_last_offset) {
183 184 185 186 187
			struct btrfs_key found_key;
			u32 item_size;

			if (item)
				btrfs_release_path(root, path);
188 189
			item = btrfs_lookup_csum(NULL, root->fs_info->csum_root,
						 path, disk_bytenr, 0);
190 191 192 193 194
			if (IS_ERR(item)) {
				ret = PTR_ERR(item);
				if (ret == -ENOENT || ret == -EFBIG)
					ret = 0;
				sum = 0;
195 196 197 198 199 200
				if (BTRFS_I(inode)->root->root_key.objectid ==
				    BTRFS_DATA_RELOC_TREE_OBJECTID) {
					set_extent_bits(io_tree, offset,
						offset + bvec->bv_len - 1,
						EXTENT_NODATASUM, GFP_NOFS);
				} else {
C
Chris Mason 已提交
201 202 203
					printk(KERN_INFO "btrfs no csum found "
					       "for inode %lu start %llu\n",
					       inode->i_ino,
204 205
					       (unsigned long long)offset);
				}
206
				item = NULL;
207
				btrfs_release_path(root, path);
208 209 210 211 212 213 214 215 216
				goto found;
			}
			btrfs_item_key_to_cpu(path->nodes[0], &found_key,
					      path->slots[0]);

			item_start_offset = found_key.offset;
			item_size = btrfs_item_size_nr(path->nodes[0],
						       path->slots[0]);
			item_last_offset = item_start_offset +
217
				(item_size / csum_size) *
218 219 220 221 222 223 224 225
				root->sectorsize;
			item = btrfs_item_ptr(path->nodes[0], path->slots[0],
					      struct btrfs_csum_item);
		}
		/*
		 * this byte range must be able to fit inside
		 * a single leaf so it will also fit inside a u32
		 */
226
		diff = disk_bytenr - item_start_offset;
227
		diff = diff / root->sectorsize;
228
		diff = diff * csum_size;
229 230

		read_extent_buffer(path->nodes[0], &sum,
231
				   ((unsigned long)item) + diff,
232
				   csum_size);
233
found:
234 235 236 237 238
		if (dst)
			*dst++ = sum;
		else
			set_state_private(io_tree, offset, sum);
		disk_bytenr += bvec->bv_len;
239 240 241 242 243 244 245
		bio_index++;
		bvec++;
	}
	btrfs_free_path(path);
	return 0;
}

246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267
int btrfs_lookup_csums_range(struct btrfs_root *root, u64 start, u64 end,
			     struct list_head *list)
{
	struct btrfs_key key;
	struct btrfs_path *path;
	struct extent_buffer *leaf;
	struct btrfs_ordered_sum *sums;
	struct btrfs_sector_sum *sector_sum;
	struct btrfs_csum_item *item;
	unsigned long offset;
	int ret;
	size_t size;
	u64 csum_end;
	u16 csum_size = btrfs_super_csum_size(&root->fs_info->super_copy);

	path = btrfs_alloc_path();
	BUG_ON(!path);

	key.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
	key.offset = start;
	key.type = BTRFS_EXTENT_CSUM_KEY;

Y
Yan Zheng 已提交
268
	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286
	if (ret < 0)
		goto fail;
	if (ret > 0 && path->slots[0] > 0) {
		leaf = path->nodes[0];
		btrfs_item_key_to_cpu(leaf, &key, path->slots[0] - 1);
		if (key.objectid == BTRFS_EXTENT_CSUM_OBJECTID &&
		    key.type == BTRFS_EXTENT_CSUM_KEY) {
			offset = (start - key.offset) >>
				 root->fs_info->sb->s_blocksize_bits;
			if (offset * csum_size <
			    btrfs_item_size_nr(leaf, path->slots[0] - 1))
				path->slots[0]--;
		}
	}

	while (start <= end) {
		leaf = path->nodes[0];
		if (path->slots[0] >= btrfs_header_nritems(leaf)) {
Y
Yan Zheng 已提交
287
			ret = btrfs_next_leaf(root, path);
288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308
			if (ret < 0)
				goto fail;
			if (ret > 0)
				break;
			leaf = path->nodes[0];
		}

		btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
		if (key.objectid != BTRFS_EXTENT_CSUM_OBJECTID ||
		    key.type != BTRFS_EXTENT_CSUM_KEY)
			break;

		btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
		if (key.offset > end)
			break;

		if (key.offset > start)
			start = key.offset;

		size = btrfs_item_size_nr(leaf, path->slots[0]);
		csum_end = key.offset + (size / csum_size) * root->sectorsize;
309 310 311 312
		if (csum_end <= start) {
			path->slots[0]++;
			continue;
		}
313

Y
Yan Zheng 已提交
314
		csum_end = min(csum_end, end + 1);
315 316
		item = btrfs_item_ptr(path->nodes[0], path->slots[0],
				      struct btrfs_csum_item);
Y
Yan Zheng 已提交
317 318 319 320 321 322
		while (start < csum_end) {
			size = min_t(size_t, csum_end - start,
					MAX_ORDERED_SUM_BYTES(root));
			sums = kzalloc(btrfs_ordered_sum_size(root, size),
					GFP_NOFS);
			BUG_ON(!sums);
323

Y
Yan Zheng 已提交
324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345
			sector_sum = sums->sums;
			sums->bytenr = start;
			sums->len = size;

			offset = (start - key.offset) >>
				root->fs_info->sb->s_blocksize_bits;
			offset *= csum_size;

			while (size > 0) {
				read_extent_buffer(path->nodes[0],
						&sector_sum->sum,
						((unsigned long)item) +
						offset, csum_size);
				sector_sum->bytenr = start;

				size -= root->sectorsize;
				start += root->sectorsize;
				offset += csum_size;
				sector_sum++;
			}
			list_add_tail(&sums->list, list);
		}
346 347 348 349 350 351 352 353
		path->slots[0]++;
	}
	ret = 0;
fail:
	btrfs_free_path(path);
	return ret;
}

354
int btrfs_csum_one_bio(struct btrfs_root *root, struct inode *inode,
355
		       struct bio *bio, u64 file_start, int contig)
356
{
357 358
	struct btrfs_ordered_sum *sums;
	struct btrfs_sector_sum *sector_sum;
359
	struct btrfs_ordered_extent *ordered;
360 361 362
	char *data;
	struct bio_vec *bvec = bio->bi_io_vec;
	int bio_index = 0;
363 364 365
	unsigned long total_bytes = 0;
	unsigned long this_sum_bytes = 0;
	u64 offset;
366
	u64 disk_bytenr;
367

368 369
	WARN_ON(bio->bi_vcnt <= 0);
	sums = kzalloc(btrfs_ordered_sum_size(root, bio->bi_size), GFP_NOFS);
370 371
	if (!sums)
		return -ENOMEM;
372

373
	sector_sum = sums->sums;
374
	disk_bytenr = (u64)bio->bi_sector << 9;
375 376
	sums->len = bio->bi_size;
	INIT_LIST_HEAD(&sums->list);
377 378 379 380 381 382 383

	if (contig)
		offset = file_start;
	else
		offset = page_offset(bvec->bv_page) + bvec->bv_offset;

	ordered = btrfs_lookup_ordered_extent(inode, offset);
384
	BUG_ON(!ordered);
385
	sums->bytenr = ordered->start;
386

C
Chris Mason 已提交
387
	while (bio_index < bio->bi_vcnt) {
388 389 390 391 392
		if (!contig)
			offset = page_offset(bvec->bv_page) + bvec->bv_offset;

		if (!contig && (offset >= ordered->file_offset + ordered->len ||
		    offset < ordered->file_offset)) {
393 394 395 396 397 398 399 400 401 402 403
			unsigned long bytes_left;
			sums->len = this_sum_bytes;
			this_sum_bytes = 0;
			btrfs_add_ordered_sum(inode, ordered, sums);
			btrfs_put_ordered_extent(ordered);

			bytes_left = bio->bi_size - total_bytes;

			sums = kzalloc(btrfs_ordered_sum_size(root, bytes_left),
				       GFP_NOFS);
			BUG_ON(!sums);
404
			sector_sum = sums->sums;
405
			sums->len = bytes_left;
406
			ordered = btrfs_lookup_ordered_extent(inode, offset);
407
			BUG_ON(!ordered);
408
			sums->bytenr = ordered->start;
409 410
		}

411
		data = kmap_atomic(bvec->bv_page, KM_USER0);
412 413 414 415 416
		sector_sum->sum = ~(u32)0;
		sector_sum->sum = btrfs_csum_data(root,
						  data + bvec->bv_offset,
						  sector_sum->sum,
						  bvec->bv_len);
417
		kunmap_atomic(data, KM_USER0);
418 419
		btrfs_csum_final(sector_sum->sum,
				 (char *)&sector_sum->sum);
420
		sector_sum->bytenr = disk_bytenr;
421

422
		sector_sum++;
423
		bio_index++;
424 425
		total_bytes += bvec->bv_len;
		this_sum_bytes += bvec->bv_len;
426 427
		disk_bytenr += bvec->bv_len;
		offset += bvec->bv_len;
428 429
		bvec++;
	}
430
	this_sum_bytes = 0;
431 432
	btrfs_add_ordered_sum(inode, ordered, sums);
	btrfs_put_ordered_extent(ordered);
433 434 435
	return 0;
}

436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520
/*
 * helper function for csum removal, this expects the
 * key to describe the csum pointed to by the path, and it expects
 * the csum to overlap the range [bytenr, len]
 *
 * The csum should not be entirely contained in the range and the
 * range should not be entirely contained in the csum.
 *
 * This calls btrfs_truncate_item with the correct args based on the
 * overlap, and fixes up the key as required.
 */
static noinline int truncate_one_csum(struct btrfs_trans_handle *trans,
				      struct btrfs_root *root,
				      struct btrfs_path *path,
				      struct btrfs_key *key,
				      u64 bytenr, u64 len)
{
	struct extent_buffer *leaf;
	u16 csum_size =
		btrfs_super_csum_size(&root->fs_info->super_copy);
	u64 csum_end;
	u64 end_byte = bytenr + len;
	u32 blocksize_bits = root->fs_info->sb->s_blocksize_bits;
	int ret;

	leaf = path->nodes[0];
	csum_end = btrfs_item_size_nr(leaf, path->slots[0]) / csum_size;
	csum_end <<= root->fs_info->sb->s_blocksize_bits;
	csum_end += key->offset;

	if (key->offset < bytenr && csum_end <= end_byte) {
		/*
		 *         [ bytenr - len ]
		 *         [   ]
		 *   [csum     ]
		 *   A simple truncate off the end of the item
		 */
		u32 new_size = (bytenr - key->offset) >> blocksize_bits;
		new_size *= csum_size;
		ret = btrfs_truncate_item(trans, root, path, new_size, 1);
		BUG_ON(ret);
	} else if (key->offset >= bytenr && csum_end > end_byte &&
		   end_byte > key->offset) {
		/*
		 *         [ bytenr - len ]
		 *                 [ ]
		 *                 [csum     ]
		 * we need to truncate from the beginning of the csum
		 */
		u32 new_size = (csum_end - end_byte) >> blocksize_bits;
		new_size *= csum_size;

		ret = btrfs_truncate_item(trans, root, path, new_size, 0);
		BUG_ON(ret);

		key->offset = end_byte;
		ret = btrfs_set_item_key_safe(trans, root, path, key);
		BUG_ON(ret);
	} else {
		BUG();
	}
	return 0;
}

/*
 * deletes the csum items from the csum tree for a given
 * range of bytes.
 */
int btrfs_del_csums(struct btrfs_trans_handle *trans,
		    struct btrfs_root *root, u64 bytenr, u64 len)
{
	struct btrfs_path *path;
	struct btrfs_key key;
	u64 end_byte = bytenr + len;
	u64 csum_end;
	struct extent_buffer *leaf;
	int ret;
	u16 csum_size =
		btrfs_super_csum_size(&root->fs_info->super_copy);
	int blocksize_bits = root->fs_info->sb->s_blocksize_bits;

	root = root->fs_info->csum_root;

	path = btrfs_alloc_path();

C
Chris Mason 已提交
521
	while (1) {
522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554
		key.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
		key.offset = end_byte - 1;
		key.type = BTRFS_EXTENT_CSUM_KEY;

		ret = btrfs_search_slot(trans, root, &key, path, -1, 1);
		if (ret > 0) {
			if (path->slots[0] == 0)
				goto out;
			path->slots[0]--;
		}
		leaf = path->nodes[0];
		btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);

		if (key.objectid != BTRFS_EXTENT_CSUM_OBJECTID ||
		    key.type != BTRFS_EXTENT_CSUM_KEY) {
			break;
		}

		if (key.offset >= end_byte)
			break;

		csum_end = btrfs_item_size_nr(leaf, path->slots[0]) / csum_size;
		csum_end <<= blocksize_bits;
		csum_end += key.offset;

		/* this csum ends before we start, we're done */
		if (csum_end <= bytenr)
			break;

		/* delete the entire item, it is inside our range */
		if (key.offset >= bytenr && csum_end <= end_byte) {
			ret = btrfs_del_item(trans, root, path);
			BUG_ON(ret);
555 556
			if (key.offset == bytenr)
				break;
557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602
		} else if (key.offset < bytenr && csum_end > end_byte) {
			unsigned long offset;
			unsigned long shift_len;
			unsigned long item_offset;
			/*
			 *        [ bytenr - len ]
			 *     [csum                ]
			 *
			 * Our bytes are in the middle of the csum,
			 * we need to split this item and insert a new one.
			 *
			 * But we can't drop the path because the
			 * csum could change, get removed, extended etc.
			 *
			 * The trick here is the max size of a csum item leaves
			 * enough room in the tree block for a single
			 * item header.  So, we split the item in place,
			 * adding a new header pointing to the existing
			 * bytes.  Then we loop around again and we have
			 * a nicely formed csum item that we can neatly
			 * truncate.
			 */
			offset = (bytenr - key.offset) >> blocksize_bits;
			offset *= csum_size;

			shift_len = (len >> blocksize_bits) * csum_size;

			item_offset = btrfs_item_ptr_offset(leaf,
							    path->slots[0]);

			memset_extent_buffer(leaf, 0, item_offset + offset,
					     shift_len);
			key.offset = bytenr;

			/*
			 * btrfs_split_item returns -EAGAIN when the
			 * item changed size or key
			 */
			ret = btrfs_split_item(trans, root, path, &key, offset);
			BUG_ON(ret && ret != -EAGAIN);

			key.offset = end_byte - 1;
		} else {
			ret = truncate_one_csum(trans, root, path,
						&key, bytenr, len);
			BUG_ON(ret);
603 604
			if (key.offset < bytenr)
				break;
605 606 607 608 609 610 611 612
		}
		btrfs_release_path(root, path);
	}
out:
	btrfs_free_path(path);
	return 0;
}

613
int btrfs_csum_file_blocks(struct btrfs_trans_handle *trans,
614
			   struct btrfs_root *root,
615
			   struct btrfs_ordered_sum *sums)
C
Chris Mason 已提交
616
{
617
	u64 bytenr;
C
Chris Mason 已提交
618 619
	int ret;
	struct btrfs_key file_key;
620
	struct btrfs_key found_key;
621
	u64 next_offset;
622
	u64 total_bytes = 0;
623
	int found_next;
624
	struct btrfs_path *path;
C
Chris Mason 已提交
625
	struct btrfs_csum_item *item;
626
	struct btrfs_csum_item *item_end;
627
	struct extent_buffer *leaf = NULL;
628
	u64 csum_offset;
629
	struct btrfs_sector_sum *sector_sum;
630 631
	u32 nritems;
	u32 ins_size;
632 633 634 635
	char *eb_map;
	char *eb_token;
	unsigned long map_len;
	unsigned long map_start;
636 637
	u16 csum_size =
		btrfs_super_csum_size(&root->fs_info->super_copy);
638

639 640
	path = btrfs_alloc_path();
	BUG_ON(!path);
641
	sector_sum = sums->sums;
642 643 644
again:
	next_offset = (u64)-1;
	found_next = 0;
645 646 647 648
	file_key.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
	file_key.offset = sector_sum->bytenr;
	bytenr = sector_sum->bytenr;
	btrfs_set_key_type(&file_key, BTRFS_EXTENT_CSUM_KEY);
649

650
	item = btrfs_lookup_csum(trans, root, path, sector_sum->bytenr, 1);
651 652
	if (!IS_ERR(item)) {
		leaf = path->nodes[0];
653
		ret = 0;
654
		goto found;
655
	}
656 657 658 659
	ret = PTR_ERR(item);
	if (ret == -EFBIG) {
		u32 item_size;
		/* we found one, but it isn't big enough yet */
660 661
		leaf = path->nodes[0];
		item_size = btrfs_item_size_nr(leaf, path->slots[0]);
662 663
		if ((item_size / csum_size) >=
		    MAX_CSUM_ITEMS(root, csum_size)) {
664 665 666 667
			/* already at max size, make a new one */
			goto insert;
		}
	} else {
668
		int slot = path->slots[0] + 1;
669
		/* we didn't find a csum item, insert one */
670 671 672
		nritems = btrfs_header_nritems(path->nodes[0]);
		if (path->slots[0] >= nritems - 1) {
			ret = btrfs_next_leaf(root, path);
Y
Yan 已提交
673
			if (ret == 1)
674
				found_next = 1;
Y
Yan 已提交
675
			if (ret != 0)
676
				goto insert;
Y
Yan 已提交
677
			slot = 0;
678 679
		}
		btrfs_item_key_to_cpu(path->nodes[0], &found_key, slot);
680 681
		if (found_key.objectid != BTRFS_EXTENT_CSUM_OBJECTID ||
		    found_key.type != BTRFS_EXTENT_CSUM_KEY) {
682 683 684 685 686
			found_next = 1;
			goto insert;
		}
		next_offset = found_key.offset;
		found_next = 1;
687 688 689 690 691 692 693 694
		goto insert;
	}

	/*
	 * at this point, we know the tree has an item, but it isn't big
	 * enough yet to put our csum in.  Grow it
	 */
	btrfs_release_path(root, path);
695
	ret = btrfs_search_slot(trans, root, &file_key, path,
696
				csum_size, 1);
697
	if (ret < 0)
698
		goto fail_unlock;
699 700 701 702 703

	if (ret > 0) {
		if (path->slots[0] == 0)
			goto insert;
		path->slots[0]--;
704
	}
705

706 707
	leaf = path->nodes[0];
	btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
708
	csum_offset = (bytenr - found_key.offset) >>
709
			root->fs_info->sb->s_blocksize_bits;
710

711 712
	if (btrfs_key_type(&found_key) != BTRFS_EXTENT_CSUM_KEY ||
	    found_key.objectid != BTRFS_EXTENT_CSUM_OBJECTID ||
713
	    csum_offset >= MAX_CSUM_ITEMS(root, csum_size)) {
714 715
		goto insert;
	}
716

717
	if (csum_offset >= btrfs_item_size_nr(leaf, path->slots[0]) /
718 719
	    csum_size) {
		u32 diff = (csum_offset + 1) * csum_size;
720 721 722 723 724 725 726 727

		/*
		 * is the item big enough already?  we dropped our lock
		 * before and need to recheck
		 */
		if (diff < btrfs_item_size_nr(leaf, path->slots[0]))
			goto csum;

728
		diff = diff - btrfs_item_size_nr(leaf, path->slots[0]);
C
Chris Mason 已提交
729
		if (diff != csum_size)
C
Chris Mason 已提交
730
			goto insert;
731

732
		ret = btrfs_extend_item(trans, root, path, diff);
733 734 735 736 737
		BUG_ON(ret);
		goto csum;
	}

insert:
738
	btrfs_release_path(root, path);
739
	csum_offset = 0;
740
	if (found_next) {
741 742 743 744
		u64 tmp = total_bytes + root->sectorsize;
		u64 next_sector = sector_sum->bytenr;
		struct btrfs_sector_sum *next = sector_sum + 1;

C
Chris Mason 已提交
745
		while (tmp < sums->len) {
746 747 748 749 750 751 752
			if (next_sector + root->sectorsize != next->bytenr)
				break;
			tmp += root->sectorsize;
			next_sector = next->bytenr;
			next++;
		}
		tmp = min(tmp, next_offset - file_key.offset);
753 754
		tmp >>= root->fs_info->sb->s_blocksize_bits;
		tmp = max((u64)1, tmp);
755 756
		tmp = min(tmp, (u64)MAX_CSUM_ITEMS(root, csum_size));
		ins_size = csum_size * tmp;
757
	} else {
758
		ins_size = csum_size;
759
	}
760
	ret = btrfs_insert_empty_item(trans, root, path, &file_key,
761
				      ins_size);
762
	if (ret < 0)
763
		goto fail_unlock;
764 765
	if (ret != 0) {
		WARN_ON(1);
766
		goto fail_unlock;
767
	}
768
csum:
769 770
	leaf = path->nodes[0];
	item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item);
C
Chris Mason 已提交
771
	ret = 0;
772
	item = (struct btrfs_csum_item *)((unsigned char *)item +
773
					  csum_offset * csum_size);
C
Chris Mason 已提交
774
found:
775 776 777
	item_end = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item);
	item_end = (struct btrfs_csum_item *)((unsigned char *)item_end +
				      btrfs_item_size_nr(leaf, path->slots[0]));
778
	eb_token = NULL;
779
	cond_resched();
780
next_sector:
781

782
	if (!eb_token ||
783
	   (unsigned long)item + csum_size >= map_start + map_len) {
784 785 786
		int err;

		if (eb_token)
787
			unmap_extent_buffer(leaf, eb_token, KM_USER1);
788 789
		eb_token = NULL;
		err = map_private_extent_buffer(leaf, (unsigned long)item,
790
						csum_size,
791
						&eb_token, &eb_map,
792
						&map_start, &map_len, KM_USER1);
793 794 795 796 797
		if (err)
			eb_token = NULL;
	}
	if (eb_token) {
		memcpy(eb_token + ((unsigned long)item & (PAGE_CACHE_SIZE - 1)),
798
		       &sector_sum->sum, csum_size);
799
	} else {
800
		write_extent_buffer(leaf, &sector_sum->sum,
801
				    (unsigned long)item, csum_size);
802
	}
803

804 805 806
	total_bytes += root->sectorsize;
	sector_sum++;
	if (total_bytes < sums->len) {
807
		item = (struct btrfs_csum_item *)((char *)item +
808
						  csum_size);
809 810 811
		if (item < item_end && bytenr + PAGE_CACHE_SIZE ==
		    sector_sum->bytenr) {
			bytenr = sector_sum->bytenr;
812
			goto next_sector;
813
		}
814
	}
815
	if (eb_token) {
816
		unmap_extent_buffer(leaf, eb_token, KM_USER1);
817 818
		eb_token = NULL;
	}
819
	btrfs_mark_buffer_dirty(path->nodes[0]);
820
	cond_resched();
821
	if (total_bytes < sums->len) {
822 823 824
		btrfs_release_path(root, path);
		goto again;
	}
825
out:
826
	btrfs_free_path(path);
C
Chris Mason 已提交
827
	return ret;
828 829 830

fail_unlock:
	goto out;
C
Chris Mason 已提交
831
}