file-item.c 22.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/bio.h>
20
#include <linux/slab.h>
21 22
#include <linux/pagemap.h>
#include <linux/highmem.h>
C
Chris Mason 已提交
23
#include "ctree.h"
C
Chris Mason 已提交
24
#include "disk-io.h"
25
#include "transaction.h"
C
Chris Mason 已提交
26
#include "print-tree.h"
C
Chris Mason 已提交
27

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

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

50
	path = btrfs_alloc_path();
T
Tsutomu Itoh 已提交
51 52
	if (!path)
		return -ENOMEM;
C
Chris Mason 已提交
53
	file_key.objectid = objectid;
C
Chris Mason 已提交
54
	file_key.offset = pos;
C
Chris Mason 已提交
55 56
	btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY);

57
	path->leave_spinning = 1;
58
	ret = btrfs_insert_empty_item(trans, root, path, &file_key,
C
Chris Mason 已提交
59
				      sizeof(*item));
60 61
	if (ret < 0)
		goto out;
C
Chris Mason 已提交
62
	BUG_ON(ret);
63 64
	leaf = path->nodes[0];
	item = btrfs_item_ptr(leaf, path->slots[0],
C
Chris Mason 已提交
65
			      struct btrfs_file_extent_item);
S
Sage Weil 已提交
66
	btrfs_set_file_extent_disk_bytenr(leaf, item, disk_offset);
67
	btrfs_set_file_extent_disk_num_bytes(leaf, item, disk_num_bytes);
S
Sage Weil 已提交
68
	btrfs_set_file_extent_offset(leaf, item, offset);
69
	btrfs_set_file_extent_num_bytes(leaf, item, num_bytes);
C
Chris Mason 已提交
70
	btrfs_set_file_extent_ram_bytes(leaf, item, ram_bytes);
71 72
	btrfs_set_file_extent_generation(leaf, item, trans->transid);
	btrfs_set_file_extent_type(leaf, item, BTRFS_FILE_EXTENT_REG);
C
Chris Mason 已提交
73 74 75 76
	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);

77
	btrfs_mark_buffer_dirty(leaf);
78
out:
79
	btrfs_free_path(path);
80
	return ret;
81
}
C
Chris Mason 已提交
82

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

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

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

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


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

	file_key.objectid = objectid;
146
	file_key.offset = offset;
C
Chris Mason 已提交
147 148 149 150
	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 已提交
151

152

153 154 155
static int __btrfs_lookup_bio_sums(struct btrfs_root *root,
				   struct inode *inode, struct bio *bio,
				   u64 logical_offset, u32 *dst, int dio)
156 157 158 159
{
	u32 sum;
	struct bio_vec *bvec = bio->bi_io_vec;
	int bio_index = 0;
160
	u64 offset = 0;
161 162
	u64 item_start_offset = 0;
	u64 item_last_offset = 0;
163
	u64 disk_bytenr;
164
	u32 diff;
165 166
	u16 csum_size =
		btrfs_super_csum_size(&root->fs_info->super_copy);
167 168 169 170 171 172
	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();
173 174
	if (!path)
		return -ENOMEM;
175 176
	if (bio->bi_size > PAGE_CACHE_SIZE * 8)
		path->reada = 2;
177 178 179

	WARN_ON(bio->bi_vcnt <= 0);

180
	disk_bytenr = (u64)bio->bi_sector << 9;
181 182
	if (dio)
		offset = logical_offset;
C
Chris Mason 已提交
183
	while (bio_index < bio->bi_vcnt) {
184 185
		if (!dio)
			offset = page_offset(bvec->bv_page) + bvec->bv_offset;
186
		ret = btrfs_find_ordered_sum(inode, offset, disk_bytenr, &sum);
187 188 189
		if (ret == 0)
			goto found;

190 191
		if (!item || disk_bytenr < item_start_offset ||
		    disk_bytenr >= item_last_offset) {
192 193 194 195
			struct btrfs_key found_key;
			u32 item_size;

			if (item)
196
				btrfs_release_path(path);
197 198
			item = btrfs_lookup_csum(NULL, root->fs_info->csum_root,
						 path, disk_bytenr, 0);
199 200 201 202 203
			if (IS_ERR(item)) {
				ret = PTR_ERR(item);
				if (ret == -ENOENT || ret == -EFBIG)
					ret = 0;
				sum = 0;
204 205 206 207 208 209
				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 已提交
210
					printk(KERN_INFO "btrfs no csum found "
L
Li Zefan 已提交
211 212 213
					       "for inode %llu start %llu\n",
					       (unsigned long long)
					       btrfs_ino(inode),
214 215
					       (unsigned long long)offset);
				}
216
				item = NULL;
217
				btrfs_release_path(path);
218 219 220 221 222 223 224 225 226
				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 +
227
				(item_size / csum_size) *
228 229 230 231 232 233 234 235
				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
		 */
236
		diff = disk_bytenr - item_start_offset;
237
		diff = diff / root->sectorsize;
238
		diff = diff * csum_size;
239 240

		read_extent_buffer(path->nodes[0], &sum,
241
				   ((unsigned long)item) + diff,
242
				   csum_size);
243
found:
244 245 246 247 248
		if (dst)
			*dst++ = sum;
		else
			set_state_private(io_tree, offset, sum);
		disk_bytenr += bvec->bv_len;
249
		offset += bvec->bv_len;
250 251 252 253 254 255 256
		bio_index++;
		bvec++;
	}
	btrfs_free_path(path);
	return 0;
}

257 258 259 260 261 262 263 264 265 266 267 268
int btrfs_lookup_bio_sums(struct btrfs_root *root, struct inode *inode,
			  struct bio *bio, u32 *dst)
{
	return __btrfs_lookup_bio_sums(root, inode, bio, 0, dst, 0);
}

int btrfs_lookup_bio_sums_dio(struct btrfs_root *root, struct inode *inode,
			      struct bio *bio, u64 offset, u32 *dst)
{
	return __btrfs_lookup_bio_sums(root, inode, bio, offset, dst, 1);
}

269
int btrfs_lookup_csums_range(struct btrfs_root *root, u64 start, u64 end,
A
Arne Jansen 已提交
270
			     struct list_head *list, int search_commit)
271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286
{
	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);

A
Arne Jansen 已提交
287 288 289 290 291 292
	if (search_commit) {
		path->skip_locking = 1;
		path->reada = 2;
		path->search_commit_root = 1;
	}

293 294 295 296
	key.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
	key.offset = start;
	key.type = BTRFS_EXTENT_CSUM_KEY;

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

Y
Yan Zheng 已提交
343
		csum_end = min(csum_end, end + 1);
344 345
		item = btrfs_item_ptr(path->nodes[0], path->slots[0],
				      struct btrfs_csum_item);
Y
Yan Zheng 已提交
346 347 348 349 350 351
		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);
352

Y
Yan Zheng 已提交
353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374
			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);
		}
375 376 377 378 379 380 381 382
		path->slots[0]++;
	}
	ret = 0;
fail:
	btrfs_free_path(path);
	return ret;
}

383
int btrfs_csum_one_bio(struct btrfs_root *root, struct inode *inode,
384
		       struct bio *bio, u64 file_start, int contig)
385
{
386 387
	struct btrfs_ordered_sum *sums;
	struct btrfs_sector_sum *sector_sum;
388
	struct btrfs_ordered_extent *ordered;
389 390 391
	char *data;
	struct bio_vec *bvec = bio->bi_io_vec;
	int bio_index = 0;
392 393 394
	unsigned long total_bytes = 0;
	unsigned long this_sum_bytes = 0;
	u64 offset;
395
	u64 disk_bytenr;
396

397 398
	WARN_ON(bio->bi_vcnt <= 0);
	sums = kzalloc(btrfs_ordered_sum_size(root, bio->bi_size), GFP_NOFS);
399 400
	if (!sums)
		return -ENOMEM;
401

402
	sector_sum = sums->sums;
403
	disk_bytenr = (u64)bio->bi_sector << 9;
404 405
	sums->len = bio->bi_size;
	INIT_LIST_HEAD(&sums->list);
406 407 408 409 410 411 412

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

	ordered = btrfs_lookup_ordered_extent(inode, offset);
413
	BUG_ON(!ordered);
414
	sums->bytenr = ordered->start;
415

C
Chris Mason 已提交
416
	while (bio_index < bio->bi_vcnt) {
417 418 419 420 421
		if (!contig)
			offset = page_offset(bvec->bv_page) + bvec->bv_offset;

		if (!contig && (offset >= ordered->file_offset + ordered->len ||
		    offset < ordered->file_offset)) {
422 423 424 425 426 427 428 429 430 431 432
			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);
433
			sector_sum = sums->sums;
434
			sums->len = bytes_left;
435
			ordered = btrfs_lookup_ordered_extent(inode, offset);
436
			BUG_ON(!ordered);
437
			sums->bytenr = ordered->start;
438 439
		}

440
		data = kmap_atomic(bvec->bv_page, KM_USER0);
441 442 443 444 445
		sector_sum->sum = ~(u32)0;
		sector_sum->sum = btrfs_csum_data(root,
						  data + bvec->bv_offset,
						  sector_sum->sum,
						  bvec->bv_len);
446
		kunmap_atomic(data, KM_USER0);
447 448
		btrfs_csum_final(sector_sum->sum,
				 (char *)&sector_sum->sum);
449
		sector_sum->bytenr = disk_bytenr;
450

451
		sector_sum++;
452
		bio_index++;
453 454
		total_bytes += bvec->bv_len;
		this_sum_bytes += bvec->bv_len;
455 456
		disk_bytenr += bvec->bv_len;
		offset += bvec->bv_len;
457 458
		bvec++;
	}
459
	this_sum_bytes = 0;
460 461
	btrfs_add_ordered_sum(inode, ordered, sums);
	btrfs_put_ordered_extent(ordered);
462 463 464
	return 0;
}

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 521 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
/*
 * 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);
	} 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);

		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();
547 548
	if (!path)
		return -ENOMEM;
549

C
Chris Mason 已提交
550
	while (1) {
551 552 553 554
		key.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
		key.offset = end_byte - 1;
		key.type = BTRFS_EXTENT_CSUM_KEY;

555
		path->leave_spinning = 1;
556 557 558
		ret = btrfs_search_slot(trans, root, &key, path, -1, 1);
		if (ret > 0) {
			if (path->slots[0] == 0)
559
				break;
560
			path->slots[0]--;
561
		} else if (ret < 0) {
562
			break;
563
		}
564

565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586
		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);
587 588
			if (ret)
				goto out;
589 590
			if (key.offset == bytenr)
				break;
591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636
		} 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);
637 638
			if (key.offset < bytenr)
				break;
639
		}
640
		btrfs_release_path(path);
641
	}
642
	ret = 0;
643 644
out:
	btrfs_free_path(path);
645
	return ret;
646 647
}

648
int btrfs_csum_file_blocks(struct btrfs_trans_handle *trans,
649
			   struct btrfs_root *root,
650
			   struct btrfs_ordered_sum *sums)
C
Chris Mason 已提交
651
{
652
	u64 bytenr;
C
Chris Mason 已提交
653 654
	int ret;
	struct btrfs_key file_key;
655
	struct btrfs_key found_key;
656
	u64 next_offset;
657
	u64 total_bytes = 0;
658
	int found_next;
659
	struct btrfs_path *path;
C
Chris Mason 已提交
660
	struct btrfs_csum_item *item;
661
	struct btrfs_csum_item *item_end;
662
	struct extent_buffer *leaf = NULL;
663
	u64 csum_offset;
664
	struct btrfs_sector_sum *sector_sum;
665 666
	u32 nritems;
	u32 ins_size;
667 668 669 670
	char *eb_map;
	char *eb_token;
	unsigned long map_len;
	unsigned long map_start;
671 672
	u16 csum_size =
		btrfs_super_csum_size(&root->fs_info->super_copy);
673

674 675
	path = btrfs_alloc_path();
	BUG_ON(!path);
676
	sector_sum = sums->sums;
677 678 679
again:
	next_offset = (u64)-1;
	found_next = 0;
680 681 682 683
	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);
684

685
	item = btrfs_lookup_csum(trans, root, path, sector_sum->bytenr, 1);
686 687
	if (!IS_ERR(item)) {
		leaf = path->nodes[0];
688
		ret = 0;
689
		goto found;
690
	}
691
	ret = PTR_ERR(item);
692 693 694
	if (ret != -EFBIG && ret != -ENOENT)
		goto fail_unlock;

695 696 697
	if (ret == -EFBIG) {
		u32 item_size;
		/* we found one, but it isn't big enough yet */
698 699
		leaf = path->nodes[0];
		item_size = btrfs_item_size_nr(leaf, path->slots[0]);
700 701
		if ((item_size / csum_size) >=
		    MAX_CSUM_ITEMS(root, csum_size)) {
702 703 704 705
			/* already at max size, make a new one */
			goto insert;
		}
	} else {
706
		int slot = path->slots[0] + 1;
707
		/* we didn't find a csum item, insert one */
708 709 710
		nritems = btrfs_header_nritems(path->nodes[0]);
		if (path->slots[0] >= nritems - 1) {
			ret = btrfs_next_leaf(root, path);
Y
Yan 已提交
711
			if (ret == 1)
712
				found_next = 1;
Y
Yan 已提交
713
			if (ret != 0)
714
				goto insert;
Y
Yan 已提交
715
			slot = 0;
716 717
		}
		btrfs_item_key_to_cpu(path->nodes[0], &found_key, slot);
718 719
		if (found_key.objectid != BTRFS_EXTENT_CSUM_OBJECTID ||
		    found_key.type != BTRFS_EXTENT_CSUM_KEY) {
720 721 722 723 724
			found_next = 1;
			goto insert;
		}
		next_offset = found_key.offset;
		found_next = 1;
725 726 727 728 729 730 731
		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
	 */
732
	btrfs_release_path(path);
733
	ret = btrfs_search_slot(trans, root, &file_key, path,
734
				csum_size, 1);
735
	if (ret < 0)
736
		goto fail_unlock;
737 738 739 740 741

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

744 745
	leaf = path->nodes[0];
	btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
746
	csum_offset = (bytenr - found_key.offset) >>
747
			root->fs_info->sb->s_blocksize_bits;
748

749 750
	if (btrfs_key_type(&found_key) != BTRFS_EXTENT_CSUM_KEY ||
	    found_key.objectid != BTRFS_EXTENT_CSUM_OBJECTID ||
751
	    csum_offset >= MAX_CSUM_ITEMS(root, csum_size)) {
752 753
		goto insert;
	}
754

755
	if (csum_offset >= btrfs_item_size_nr(leaf, path->slots[0]) /
756 757
	    csum_size) {
		u32 diff = (csum_offset + 1) * csum_size;
758 759 760 761 762 763 764 765

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

766
		diff = diff - btrfs_item_size_nr(leaf, path->slots[0]);
C
Chris Mason 已提交
767
		if (diff != csum_size)
C
Chris Mason 已提交
768
			goto insert;
769

770
		ret = btrfs_extend_item(trans, root, path, diff);
771 772 773 774
		goto csum;
	}

insert:
775
	btrfs_release_path(path);
776
	csum_offset = 0;
777
	if (found_next) {
778 779 780 781
		u64 tmp = total_bytes + root->sectorsize;
		u64 next_sector = sector_sum->bytenr;
		struct btrfs_sector_sum *next = sector_sum + 1;

C
Chris Mason 已提交
782
		while (tmp < sums->len) {
783 784 785 786 787 788 789
			if (next_sector + root->sectorsize != next->bytenr)
				break;
			tmp += root->sectorsize;
			next_sector = next->bytenr;
			next++;
		}
		tmp = min(tmp, next_offset - file_key.offset);
790 791
		tmp >>= root->fs_info->sb->s_blocksize_bits;
		tmp = max((u64)1, tmp);
792 793
		tmp = min(tmp, (u64)MAX_CSUM_ITEMS(root, csum_size));
		ins_size = csum_size * tmp;
794
	} else {
795
		ins_size = csum_size;
796
	}
797
	path->leave_spinning = 1;
798
	ret = btrfs_insert_empty_item(trans, root, path, &file_key,
799
				      ins_size);
800
	path->leave_spinning = 0;
801
	if (ret < 0)
802
		goto fail_unlock;
803 804
	if (ret != 0) {
		WARN_ON(1);
805
		goto fail_unlock;
806
	}
807
csum:
808 809
	leaf = path->nodes[0];
	item = btrfs_item_ptr(leaf, path->slots[0], struct btrfs_csum_item);
C
Chris Mason 已提交
810
	ret = 0;
811
	item = (struct btrfs_csum_item *)((unsigned char *)item +
812
					  csum_offset * csum_size);
C
Chris Mason 已提交
813
found:
814 815 816
	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]));
817
	eb_token = NULL;
818
next_sector:
819

820
	if (!eb_token ||
821
	   (unsigned long)item + csum_size >= map_start + map_len) {
822 823 824
		int err;

		if (eb_token)
825
			unmap_extent_buffer(leaf, eb_token, KM_USER1);
826 827
		eb_token = NULL;
		err = map_private_extent_buffer(leaf, (unsigned long)item,
828
						csum_size,
829
						&eb_token, &eb_map,
830
						&map_start, &map_len, KM_USER1);
831 832 833 834 835
		if (err)
			eb_token = NULL;
	}
	if (eb_token) {
		memcpy(eb_token + ((unsigned long)item & (PAGE_CACHE_SIZE - 1)),
836
		       &sector_sum->sum, csum_size);
837
	} else {
838
		write_extent_buffer(leaf, &sector_sum->sum,
839
				    (unsigned long)item, csum_size);
840
	}
841

842 843 844
	total_bytes += root->sectorsize;
	sector_sum++;
	if (total_bytes < sums->len) {
845
		item = (struct btrfs_csum_item *)((char *)item +
846
						  csum_size);
847 848 849
		if (item < item_end && bytenr + PAGE_CACHE_SIZE ==
		    sector_sum->bytenr) {
			bytenr = sector_sum->bytenr;
850
			goto next_sector;
851
		}
852
	}
853
	if (eb_token) {
854
		unmap_extent_buffer(leaf, eb_token, KM_USER1);
855 856
		eb_token = NULL;
	}
857
	btrfs_mark_buffer_dirty(path->nodes[0]);
858
	if (total_bytes < sums->len) {
859
		btrfs_release_path(path);
860
		cond_resched();
861 862
		goto again;
	}
863
out:
864
	btrfs_free_path(path);
C
Chris Mason 已提交
865
	return ret;
866 867 868

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