file-item.c 22.5 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 51
	path = btrfs_alloc_path();
	BUG_ON(!path);
C
Chris Mason 已提交
52
	file_key.objectid = objectid;
C
Chris Mason 已提交
53
	file_key.offset = pos;
C
Chris Mason 已提交
54 55
	btrfs_set_key_type(&file_key, BTRFS_EXTENT_DATA_KEY);

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

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

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

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

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

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


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

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

151

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

	WARN_ON(bio->bi_vcnt <= 0);

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

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

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

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

253 254 255 256 257 258 259 260 261 262 263 264
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);
}

265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286
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 已提交
287
	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305
	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 已提交
306
			ret = btrfs_next_leaf(root, path);
307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327
			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;
328 329 330 331
		if (csum_end <= start) {
			path->slots[0]++;
			continue;
		}
332

Y
Yan Zheng 已提交
333
		csum_end = min(csum_end, end + 1);
334 335
		item = btrfs_item_ptr(path->nodes[0], path->slots[0],
				      struct btrfs_csum_item);
Y
Yan Zheng 已提交
336 337 338 339 340 341
		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);
342

Y
Yan Zheng 已提交
343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364
			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);
		}
365 366 367 368 369 370 371 372
		path->slots[0]++;
	}
	ret = 0;
fail:
	btrfs_free_path(path);
	return ret;
}

373
int btrfs_csum_one_bio(struct btrfs_root *root, struct inode *inode,
374
		       struct bio *bio, u64 file_start, int contig)
375
{
376 377
	struct btrfs_ordered_sum *sums;
	struct btrfs_sector_sum *sector_sum;
378
	struct btrfs_ordered_extent *ordered;
379 380 381
	char *data;
	struct bio_vec *bvec = bio->bi_io_vec;
	int bio_index = 0;
382 383 384
	unsigned long total_bytes = 0;
	unsigned long this_sum_bytes = 0;
	u64 offset;
385
	u64 disk_bytenr;
386

387 388
	WARN_ON(bio->bi_vcnt <= 0);
	sums = kzalloc(btrfs_ordered_sum_size(root, bio->bi_size), GFP_NOFS);
389 390
	if (!sums)
		return -ENOMEM;
391

392
	sector_sum = sums->sums;
393
	disk_bytenr = (u64)bio->bi_sector << 9;
394 395
	sums->len = bio->bi_size;
	INIT_LIST_HEAD(&sums->list);
396 397 398 399 400 401 402

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

	ordered = btrfs_lookup_ordered_extent(inode, offset);
403
	BUG_ON(!ordered);
404
	sums->bytenr = ordered->start;
405

C
Chris Mason 已提交
406
	while (bio_index < bio->bi_vcnt) {
407 408 409 410 411
		if (!contig)
			offset = page_offset(bvec->bv_page) + bvec->bv_offset;

		if (!contig && (offset >= ordered->file_offset + ordered->len ||
		    offset < ordered->file_offset)) {
412 413 414 415 416 417 418 419 420 421 422
			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);
423
			sector_sum = sums->sums;
424
			sums->len = bytes_left;
425
			ordered = btrfs_lookup_ordered_extent(inode, offset);
426
			BUG_ON(!ordered);
427
			sums->bytenr = ordered->start;
428 429
		}

430
		data = kmap_atomic(bvec->bv_page, KM_USER0);
431 432 433 434 435
		sector_sum->sum = ~(u32)0;
		sector_sum->sum = btrfs_csum_data(root,
						  data + bvec->bv_offset,
						  sector_sum->sum,
						  bvec->bv_len);
436
		kunmap_atomic(data, KM_USER0);
437 438
		btrfs_csum_final(sector_sum->sum,
				 (char *)&sector_sum->sum);
439
		sector_sum->bytenr = disk_bytenr;
440

441
		sector_sum++;
442
		bio_index++;
443 444
		total_bytes += bvec->bv_len;
		this_sum_bytes += bvec->bv_len;
445 446
		disk_bytenr += bvec->bv_len;
		offset += bvec->bv_len;
447 448
		bvec++;
	}
449
	this_sum_bytes = 0;
450 451
	btrfs_add_ordered_sum(inode, ordered, sums);
	btrfs_put_ordered_extent(ordered);
452 453 454
	return 0;
}

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 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538
/*
 * 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();
539 540
	if (!path)
		return -ENOMEM;
541

C
Chris Mason 已提交
542
	while (1) {
543 544 545 546
		key.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
		key.offset = end_byte - 1;
		key.type = BTRFS_EXTENT_CSUM_KEY;

547
		path->leave_spinning = 1;
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
		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);
577 578
			if (key.offset == bytenr)
				break;
579 580 581 582 583 584 585 586 587 588 589 590 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
		} 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);
625 626
			if (key.offset < bytenr)
				break;
627 628 629 630 631 632 633 634
		}
		btrfs_release_path(root, path);
	}
out:
	btrfs_free_path(path);
	return 0;
}

635
int btrfs_csum_file_blocks(struct btrfs_trans_handle *trans,
636
			   struct btrfs_root *root,
637
			   struct btrfs_ordered_sum *sums)
C
Chris Mason 已提交
638
{
639
	u64 bytenr;
C
Chris Mason 已提交
640 641
	int ret;
	struct btrfs_key file_key;
642
	struct btrfs_key found_key;
643
	u64 next_offset;
644
	u64 total_bytes = 0;
645
	int found_next;
646
	struct btrfs_path *path;
C
Chris Mason 已提交
647
	struct btrfs_csum_item *item;
648
	struct btrfs_csum_item *item_end;
649
	struct extent_buffer *leaf = NULL;
650
	u64 csum_offset;
651
	struct btrfs_sector_sum *sector_sum;
652 653
	u32 nritems;
	u32 ins_size;
654 655 656 657
	char *eb_map;
	char *eb_token;
	unsigned long map_len;
	unsigned long map_start;
658 659
	u16 csum_size =
		btrfs_super_csum_size(&root->fs_info->super_copy);
660

661 662
	path = btrfs_alloc_path();
	BUG_ON(!path);
663
	sector_sum = sums->sums;
664 665 666
again:
	next_offset = (u64)-1;
	found_next = 0;
667 668 669 670
	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);
671

672
	item = btrfs_lookup_csum(trans, root, path, sector_sum->bytenr, 1);
673 674
	if (!IS_ERR(item)) {
		leaf = path->nodes[0];
675
		ret = 0;
676
		goto found;
677
	}
678
	ret = PTR_ERR(item);
679 680 681
	if (ret != -EFBIG && ret != -ENOENT)
		goto fail_unlock;

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

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

731 732
	leaf = path->nodes[0];
	btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
733
	csum_offset = (bytenr - found_key.offset) >>
734
			root->fs_info->sb->s_blocksize_bits;
735

736 737
	if (btrfs_key_type(&found_key) != BTRFS_EXTENT_CSUM_KEY ||
	    found_key.objectid != BTRFS_EXTENT_CSUM_OBJECTID ||
738
	    csum_offset >= MAX_CSUM_ITEMS(root, csum_size)) {
739 740
		goto insert;
	}
741

742
	if (csum_offset >= btrfs_item_size_nr(leaf, path->slots[0]) /
743 744
	    csum_size) {
		u32 diff = (csum_offset + 1) * csum_size;
745 746 747 748 749 750 751 752

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

753
		diff = diff - btrfs_item_size_nr(leaf, path->slots[0]);
C
Chris Mason 已提交
754
		if (diff != csum_size)
C
Chris Mason 已提交
755
			goto insert;
756

757
		ret = btrfs_extend_item(trans, root, path, diff);
758 759 760 761 762
		BUG_ON(ret);
		goto csum;
	}

insert:
763
	btrfs_release_path(root, path);
764
	csum_offset = 0;
765
	if (found_next) {
766 767 768 769
		u64 tmp = total_bytes + root->sectorsize;
		u64 next_sector = sector_sum->bytenr;
		struct btrfs_sector_sum *next = sector_sum + 1;

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

808
	if (!eb_token ||
809
	   (unsigned long)item + csum_size >= map_start + map_len) {
810 811 812
		int err;

		if (eb_token)
813
			unmap_extent_buffer(leaf, eb_token, KM_USER1);
814 815
		eb_token = NULL;
		err = map_private_extent_buffer(leaf, (unsigned long)item,
816
						csum_size,
817
						&eb_token, &eb_map,
818
						&map_start, &map_len, KM_USER1);
819 820 821 822 823
		if (err)
			eb_token = NULL;
	}
	if (eb_token) {
		memcpy(eb_token + ((unsigned long)item & (PAGE_CACHE_SIZE - 1)),
824
		       &sector_sum->sum, csum_size);
825
	} else {
826
		write_extent_buffer(leaf, &sector_sum->sum,
827
				    (unsigned long)item, csum_size);
828
	}
829

830 831 832
	total_bytes += root->sectorsize;
	sector_sum++;
	if (total_bytes < sums->len) {
833
		item = (struct btrfs_csum_item *)((char *)item +
834
						  csum_size);
835 836 837
		if (item < item_end && bytenr + PAGE_CACHE_SIZE ==
		    sector_sum->bytenr) {
			bytenr = sector_sum->bytenr;
838
			goto next_sector;
839
		}
840
	}
841
	if (eb_token) {
842
		unmap_extent_buffer(leaf, eb_token, KM_USER1);
843 844
		eb_token = NULL;
	}
845
	btrfs_mark_buffer_dirty(path->nodes[0]);
846
	if (total_bytes < sums->len) {
847
		btrfs_release_path(root, path);
848
		cond_resched();
849 850
		goto again;
	}
851
out:
852
	btrfs_free_path(path);
C
Chris Mason 已提交
853
	return ret;
854 855 856

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