file-item.c 21.6 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))
C
Chris Mason 已提交
30
int btrfs_insert_file_extent(struct btrfs_trans_handle *trans,
S
Sage Weil 已提交
31 32 33
			     struct btrfs_root *root,
			     u64 objectid, u64 pos,
			     u64 disk_offset, u64 disk_num_bytes,
C
Chris Mason 已提交
34 35
			     u64 num_bytes, u64 offset, u64 ram_bytes,
			     u8 compression, u8 encryption, u16 other_encoding)
36
{
C
Chris Mason 已提交
37 38 39
	int ret = 0;
	struct btrfs_file_extent_item *item;
	struct btrfs_key file_key;
40
	struct btrfs_path *path;
41
	struct extent_buffer *leaf;
C
Chris Mason 已提交
42

43 44
	path = btrfs_alloc_path();
	BUG_ON(!path);
C
Chris Mason 已提交
45
	file_key.objectid = objectid;
C
Chris Mason 已提交
46
	file_key.offset = pos;
C
Chris Mason 已提交
47 48
	btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY);

49
	ret = btrfs_insert_empty_item(trans, root, path, &file_key,
C
Chris Mason 已提交
50
				      sizeof(*item));
51 52
	if (ret < 0)
		goto out;
C
Chris Mason 已提交
53
	BUG_ON(ret);
54 55
	leaf = path->nodes[0];
	item = btrfs_item_ptr(leaf, path->slots[0],
C
Chris Mason 已提交
56
			      struct btrfs_file_extent_item);
S
Sage Weil 已提交
57
	btrfs_set_file_extent_disk_bytenr(leaf, item, disk_offset);
58
	btrfs_set_file_extent_disk_num_bytes(leaf, item, disk_num_bytes);
S
Sage Weil 已提交
59
	btrfs_set_file_extent_offset(leaf, item, offset);
60
	btrfs_set_file_extent_num_bytes(leaf, item, num_bytes);
C
Chris Mason 已提交
61
	btrfs_set_file_extent_ram_bytes(leaf, item, ram_bytes);
62 63
	btrfs_set_file_extent_generation(leaf, item, trans->transid);
	btrfs_set_file_extent_type(leaf, item, BTRFS_FILE_EXTENT_REG);
C
Chris Mason 已提交
64 65 66 67
	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);

68
	btrfs_mark_buffer_dirty(leaf);
69
out:
70
	btrfs_free_path(path);
71
	return ret;
72
}
C
Chris Mason 已提交
73

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

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

		csum_offset = (bytenr - found_key.offset) >>
106
				root->fs_info->sb->s_blocksize_bits;
107
		csums_in_item = btrfs_item_size_nr(leaf, path->slots[0]);
108
		csums_in_item /= csum_size;
109 110 111

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


C
Chris Mason 已提交
126 127 128
int btrfs_lookup_file_extent(struct btrfs_trans_handle *trans,
			     struct btrfs_root *root,
			     struct btrfs_path *path, u64 objectid,
C
Chris Mason 已提交
129
			     u64 offset, int mod)
C
Chris Mason 已提交
130 131 132 133 134 135 136
{
	int ret;
	struct btrfs_key file_key;
	int ins_len = mod < 0 ? -1 : 0;
	int cow = mod != 0;

	file_key.objectid = objectid;
137
	file_key.offset = offset;
C
Chris Mason 已提交
138 139 140 141
	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 已提交
142

143

144
int btrfs_lookup_bio_sums(struct btrfs_root *root, struct inode *inode,
145
			  struct bio *bio, u32 *dst)
146 147 148 149 150 151 152
{
	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;
153
	u64 disk_bytenr;
154
	u32 diff;
155 156
	u16 csum_size =
		btrfs_super_csum_size(&root->fs_info->super_copy);
157 158 159 160 161 162
	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();
163 164
	if (bio->bi_size > PAGE_CACHE_SIZE * 8)
		path->reada = 2;
165 166 167

	WARN_ON(bio->bi_vcnt <= 0);

168
	disk_bytenr = (u64)bio->bi_sector << 9;
C
Chris Mason 已提交
169
	while (bio_index < bio->bi_vcnt) {
170
		offset = page_offset(bvec->bv_page) + bvec->bv_offset;
171
		ret = btrfs_find_ordered_sum(inode, offset, disk_bytenr, &sum);
172 173 174
		if (ret == 0)
			goto found;

175 176
		if (!item || disk_bytenr < item_start_offset ||
		    disk_bytenr >= item_last_offset) {
177 178 179 180 181
			struct btrfs_key found_key;
			u32 item_size;

			if (item)
				btrfs_release_path(root, path);
182 183
			item = btrfs_lookup_csum(NULL, root->fs_info->csum_root,
						 path, disk_bytenr, 0);
184 185 186 187 188
			if (IS_ERR(item)) {
				ret = PTR_ERR(item);
				if (ret == -ENOENT || ret == -EFBIG)
					ret = 0;
				sum = 0;
189 190 191 192 193 194
				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 已提交
195 196 197
					printk(KERN_INFO "btrfs no csum found "
					       "for inode %lu start %llu\n",
					       inode->i_ino,
198 199
					       (unsigned long long)offset);
				}
200
				item = NULL;
201
				btrfs_release_path(root, path);
202 203 204 205 206 207 208 209 210
				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 +
211
				(item_size / csum_size) *
212 213 214 215 216 217 218 219
				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
		 */
220
		diff = disk_bytenr - item_start_offset;
221
		diff = diff / root->sectorsize;
222
		diff = diff * csum_size;
223 224

		read_extent_buffer(path->nodes[0], &sum,
225
				   ((unsigned long)item) + diff,
226
				   csum_size);
227
found:
228 229 230 231 232
		if (dst)
			*dst++ = sum;
		else
			set_state_private(io_tree, offset, sum);
		disk_bytenr += bvec->bv_len;
233 234 235 236 237 238 239
		bio_index++;
		bvec++;
	}
	btrfs_free_path(path);
	return 0;
}

240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 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 297 298 299 300 301 302 303
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;

	ret = btrfs_search_slot(NULL, root->fs_info->csum_root,
				&key, path, 0, 0);
	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)) {
			ret = btrfs_next_leaf(root->fs_info->csum_root, path);
			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;
304 305 306 307
		if (csum_end <= start) {
			path->slots[0]++;
			continue;
		}
308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343

		size = min(csum_end, end + 1) - start;
		sums = kzalloc(btrfs_ordered_sum_size(root, size), GFP_NOFS);
		BUG_ON(!sums);

		sector_sum = sums->sums;
		sums->bytenr = start;
		sums->len = size;

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

		item = btrfs_item_ptr(path->nodes[0], path->slots[0],
				      struct btrfs_csum_item);
		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);

		path->slots[0]++;
	}
	ret = 0;
fail:
	btrfs_free_path(path);
	return ret;
}

344
int btrfs_csum_one_bio(struct btrfs_root *root, struct inode *inode,
345
		       struct bio *bio, u64 file_start, int contig)
346
{
347 348
	struct btrfs_ordered_sum *sums;
	struct btrfs_sector_sum *sector_sum;
349
	struct btrfs_ordered_extent *ordered;
350 351 352
	char *data;
	struct bio_vec *bvec = bio->bi_io_vec;
	int bio_index = 0;
353 354 355
	unsigned long total_bytes = 0;
	unsigned long this_sum_bytes = 0;
	u64 offset;
356
	u64 disk_bytenr;
357

358 359
	WARN_ON(bio->bi_vcnt <= 0);
	sums = kzalloc(btrfs_ordered_sum_size(root, bio->bi_size), GFP_NOFS);
360 361
	if (!sums)
		return -ENOMEM;
362

363
	sector_sum = sums->sums;
364
	disk_bytenr = (u64)bio->bi_sector << 9;
365 366
	sums->len = bio->bi_size;
	INIT_LIST_HEAD(&sums->list);
367 368 369 370 371 372 373

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

	ordered = btrfs_lookup_ordered_extent(inode, offset);
374
	BUG_ON(!ordered);
375
	sums->bytenr = ordered->start;
376

C
Chris Mason 已提交
377
	while (bio_index < bio->bi_vcnt) {
378 379 380 381 382
		if (!contig)
			offset = page_offset(bvec->bv_page) + bvec->bv_offset;

		if (!contig && (offset >= ordered->file_offset + ordered->len ||
		    offset < ordered->file_offset)) {
383 384 385 386 387 388 389 390 391 392 393
			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);
394
			sector_sum = sums->sums;
395
			sums->len = bytes_left;
396
			ordered = btrfs_lookup_ordered_extent(inode, offset);
397
			BUG_ON(!ordered);
398
			sums->bytenr = ordered->start;
399 400
		}

401
		data = kmap_atomic(bvec->bv_page, KM_USER0);
402 403 404 405 406
		sector_sum->sum = ~(u32)0;
		sector_sum->sum = btrfs_csum_data(root,
						  data + bvec->bv_offset,
						  sector_sum->sum,
						  bvec->bv_len);
407
		kunmap_atomic(data, KM_USER0);
408 409
		btrfs_csum_final(sector_sum->sum,
				 (char *)&sector_sum->sum);
410
		sector_sum->bytenr = disk_bytenr;
411

412
		sector_sum++;
413
		bio_index++;
414 415
		total_bytes += bvec->bv_len;
		this_sum_bytes += bvec->bv_len;
416 417
		disk_bytenr += bvec->bv_len;
		offset += bvec->bv_len;
418 419
		bvec++;
	}
420
	this_sum_bytes = 0;
421 422
	btrfs_add_ordered_sum(inode, ordered, sums);
	btrfs_put_ordered_extent(ordered);
423 424 425
	return 0;
}

426 427 428 429 430 431 432 433 434 435 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
/*
 * 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 已提交
511
	while (1) {
512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544
		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);
545 546
			if (key.offset == bytenr)
				break;
547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592
		} 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);
593 594
			if (key.offset < bytenr)
				break;
595 596 597 598 599 600 601 602
		}
		btrfs_release_path(root, path);
	}
out:
	btrfs_free_path(path);
	return 0;
}

603
int btrfs_csum_file_blocks(struct btrfs_trans_handle *trans,
604
			   struct btrfs_root *root,
605
			   struct btrfs_ordered_sum *sums)
C
Chris Mason 已提交
606
{
607
	u64 bytenr;
C
Chris Mason 已提交
608 609
	int ret;
	struct btrfs_key file_key;
610
	struct btrfs_key found_key;
611
	u64 next_offset;
612
	u64 total_bytes = 0;
613
	int found_next;
614
	struct btrfs_path *path;
C
Chris Mason 已提交
615
	struct btrfs_csum_item *item;
616
	struct btrfs_csum_item *item_end;
617
	struct extent_buffer *leaf = NULL;
618
	u64 csum_offset;
619
	struct btrfs_sector_sum *sector_sum;
620 621
	u32 nritems;
	u32 ins_size;
622 623 624 625
	char *eb_map;
	char *eb_token;
	unsigned long map_len;
	unsigned long map_start;
626 627
	u16 csum_size =
		btrfs_super_csum_size(&root->fs_info->super_copy);
628

629 630
	path = btrfs_alloc_path();
	BUG_ON(!path);
631
	sector_sum = sums->sums;
632 633 634
again:
	next_offset = (u64)-1;
	found_next = 0;
635 636 637 638
	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);
639

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

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

696 697
	leaf = path->nodes[0];
	btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
698
	csum_offset = (bytenr - found_key.offset) >>
699
			root->fs_info->sb->s_blocksize_bits;
700

701 702
	if (btrfs_key_type(&found_key) != BTRFS_EXTENT_CSUM_KEY ||
	    found_key.objectid != BTRFS_EXTENT_CSUM_OBJECTID ||
703
	    csum_offset >= MAX_CSUM_ITEMS(root, csum_size)) {
704 705
		goto insert;
	}
706

707
	if (csum_offset >= btrfs_item_size_nr(leaf, path->slots[0]) /
708 709
	    csum_size) {
		u32 diff = (csum_offset + 1) * csum_size;
710 711 712 713 714 715 716 717

		/*
		 * 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;

718
		diff = diff - btrfs_item_size_nr(leaf, path->slots[0]);
C
Chris Mason 已提交
719
		if (diff != csum_size)
C
Chris Mason 已提交
720
			goto insert;
721

722
		ret = btrfs_extend_item(trans, root, path, diff);
723 724 725 726 727
		BUG_ON(ret);
		goto csum;
	}

insert:
728
	btrfs_release_path(root, path);
729
	csum_offset = 0;
730
	if (found_next) {
731 732 733 734
		u64 tmp = total_bytes + root->sectorsize;
		u64 next_sector = sector_sum->bytenr;
		struct btrfs_sector_sum *next = sector_sum + 1;

C
Chris Mason 已提交
735
		while (tmp < sums->len) {
736 737 738 739 740 741 742
			if (next_sector + root->sectorsize != next->bytenr)
				break;
			tmp += root->sectorsize;
			next_sector = next->bytenr;
			next++;
		}
		tmp = min(tmp, next_offset - file_key.offset);
743 744
		tmp >>= root->fs_info->sb->s_blocksize_bits;
		tmp = max((u64)1, tmp);
745 746
		tmp = min(tmp, (u64)MAX_CSUM_ITEMS(root, csum_size));
		ins_size = csum_size * tmp;
747
	} else {
748
		ins_size = csum_size;
749
	}
750
	ret = btrfs_insert_empty_item(trans, root, path, &file_key,
751
				      ins_size);
752
	if (ret < 0)
753
		goto fail_unlock;
754 755
	if (ret != 0) {
		WARN_ON(1);
756
		goto fail_unlock;
757
	}
758
csum:
759 760
	leaf = path->nodes[0];
	item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item);
C
Chris Mason 已提交
761
	ret = 0;
762
	item = (struct btrfs_csum_item *)((unsigned char *)item +
763
					  csum_offset * csum_size);
C
Chris Mason 已提交
764
found:
765 766 767
	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]));
768
	eb_token = NULL;
769
	cond_resched();
770
next_sector:
771

772
	if (!eb_token ||
773
	   (unsigned long)item + csum_size >= map_start + map_len) {
774 775 776
		int err;

		if (eb_token)
777
			unmap_extent_buffer(leaf, eb_token, KM_USER1);
778 779
		eb_token = NULL;
		err = map_private_extent_buffer(leaf, (unsigned long)item,
780
						csum_size,
781
						&eb_token, &eb_map,
782
						&map_start, &map_len, KM_USER1);
783 784 785 786 787
		if (err)
			eb_token = NULL;
	}
	if (eb_token) {
		memcpy(eb_token + ((unsigned long)item & (PAGE_CACHE_SIZE - 1)),
788
		       &sector_sum->sum, csum_size);
789
	} else {
790
		write_extent_buffer(leaf, &sector_sum->sum,
791
				    (unsigned long)item, csum_size);
792
	}
793

794 795 796
	total_bytes += root->sectorsize;
	sector_sum++;
	if (total_bytes < sums->len) {
797
		item = (struct btrfs_csum_item *)((char *)item +
798
						  csum_size);
799 800 801
		if (item < item_end && bytenr + PAGE_CACHE_SIZE ==
		    sector_sum->bytenr) {
			bytenr = sector_sum->bytenr;
802
			goto next_sector;
803
		}
804
	}
805
	if (eb_token) {
806
		unmap_extent_buffer(leaf, eb_token, KM_USER1);
807 808
		eb_token = NULL;
	}
809
	btrfs_mark_buffer_dirty(path->nodes[0]);
810
	cond_resched();
811
	if (total_bytes < sums->len) {
812 813 814
		btrfs_release_path(root, path);
		goto again;
	}
815
out:
816
	btrfs_free_path(path);
C
Chris Mason 已提交
817
	return ret;
818 819 820

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