ordered-data.c 25.2 KB
Newer Older
C
Chris Mason 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
/*
 * 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.
 */

#include <linux/slab.h>
20
#include <linux/blkdev.h>
21 22
#include <linux/writeback.h>
#include <linux/pagevec.h>
C
Chris Mason 已提交
23 24 25
#include "ctree.h"
#include "transaction.h"
#include "btrfs_inode.h"
26
#include "extent_io.h"
C
Chris Mason 已提交
27

28
static u64 entry_end(struct btrfs_ordered_extent *entry)
C
Chris Mason 已提交
29
{
30 31 32
	if (entry->file_offset + entry->len < entry->file_offset)
		return (u64)-1;
	return entry->file_offset + entry->len;
C
Chris Mason 已提交
33 34
}

C
Chris Mason 已提交
35 36 37
/* returns NULL if the insertion worked, or it returns the node it did find
 * in the tree
 */
38 39
static struct rb_node *tree_insert(struct rb_root *root, u64 file_offset,
				   struct rb_node *node)
C
Chris Mason 已提交
40
{
C
Chris Mason 已提交
41 42
	struct rb_node **p = &root->rb_node;
	struct rb_node *parent = NULL;
43
	struct btrfs_ordered_extent *entry;
C
Chris Mason 已提交
44

C
Chris Mason 已提交
45
	while (*p) {
C
Chris Mason 已提交
46
		parent = *p;
47
		entry = rb_entry(parent, struct btrfs_ordered_extent, rb_node);
C
Chris Mason 已提交
48

49
		if (file_offset < entry->file_offset)
C
Chris Mason 已提交
50
			p = &(*p)->rb_left;
51
		else if (file_offset >= entry_end(entry))
C
Chris Mason 已提交
52 53 54 55 56 57 58 59 60 61
			p = &(*p)->rb_right;
		else
			return parent;
	}

	rb_link_node(node, parent, p);
	rb_insert_color(node, root);
	return NULL;
}

C
Chris Mason 已提交
62 63 64 65
/*
 * look for a given offset in the tree, and if it can't be found return the
 * first lesser offset
 */
66 67
static struct rb_node *__tree_search(struct rb_root *root, u64 file_offset,
				     struct rb_node **prev_ret)
C
Chris Mason 已提交
68
{
C
Chris Mason 已提交
69
	struct rb_node *n = root->rb_node;
C
Chris Mason 已提交
70
	struct rb_node *prev = NULL;
71 72 73
	struct rb_node *test;
	struct btrfs_ordered_extent *entry;
	struct btrfs_ordered_extent *prev_entry = NULL;
C
Chris Mason 已提交
74

C
Chris Mason 已提交
75
	while (n) {
76
		entry = rb_entry(n, struct btrfs_ordered_extent, rb_node);
C
Chris Mason 已提交
77 78 79
		prev = n;
		prev_entry = entry;

80
		if (file_offset < entry->file_offset)
C
Chris Mason 已提交
81
			n = n->rb_left;
82
		else if (file_offset >= entry_end(entry))
C
Chris Mason 已提交
83 84 85 86 87 88 89
			n = n->rb_right;
		else
			return n;
	}
	if (!prev_ret)
		return NULL;

C
Chris Mason 已提交
90
	while (prev && file_offset >= entry_end(prev_entry)) {
91 92 93 94 95 96 97 98 99 100 101 102 103
		test = rb_next(prev);
		if (!test)
			break;
		prev_entry = rb_entry(test, struct btrfs_ordered_extent,
				      rb_node);
		if (file_offset < entry_end(prev_entry))
			break;

		prev = test;
	}
	if (prev)
		prev_entry = rb_entry(prev, struct btrfs_ordered_extent,
				      rb_node);
C
Chris Mason 已提交
104
	while (prev && file_offset < entry_end(prev_entry)) {
105 106 107 108 109 110
		test = rb_prev(prev);
		if (!test)
			break;
		prev_entry = rb_entry(test, struct btrfs_ordered_extent,
				      rb_node);
		prev = test;
C
Chris Mason 已提交
111 112 113 114 115
	}
	*prev_ret = prev;
	return NULL;
}

C
Chris Mason 已提交
116 117 118
/*
 * helper to check if a given offset is inside a given entry
 */
119 120 121 122 123 124 125 126
static int offset_in_entry(struct btrfs_ordered_extent *entry, u64 file_offset)
{
	if (file_offset < entry->file_offset ||
	    entry->file_offset + entry->len <= file_offset)
		return 0;
	return 1;
}

127 128 129 130 131 132 133 134 135
static int range_overlaps(struct btrfs_ordered_extent *entry, u64 file_offset,
			  u64 len)
{
	if (file_offset + len <= entry->file_offset ||
	    entry->file_offset + entry->len <= file_offset)
		return 0;
	return 1;
}

C
Chris Mason 已提交
136 137 138 139
/*
 * look find the first ordered struct that has this offset, otherwise
 * the first one less than this offset
 */
140 141
static inline struct rb_node *tree_search(struct btrfs_ordered_inode_tree *tree,
					  u64 file_offset)
C
Chris Mason 已提交
142
{
143
	struct rb_root *root = &tree->tree;
C
Chris Mason 已提交
144 145
	struct rb_node *prev;
	struct rb_node *ret;
146 147 148 149 150 151 152 153 154
	struct btrfs_ordered_extent *entry;

	if (tree->last) {
		entry = rb_entry(tree->last, struct btrfs_ordered_extent,
				 rb_node);
		if (offset_in_entry(entry, file_offset))
			return tree->last;
	}
	ret = __tree_search(root, file_offset, &prev);
C
Chris Mason 已提交
155
	if (!ret)
156 157 158
		ret = prev;
	if (ret)
		tree->last = ret;
C
Chris Mason 已提交
159 160 161
	return ret;
}

162 163 164 165 166 167 168 169 170 171 172
/* allocate and add a new ordered_extent into the per-inode tree.
 * file_offset is the logical offset in the file
 *
 * start is the disk block number of an extent already reserved in the
 * extent allocation tree
 *
 * len is the length of the extent
 *
 * The tree is given a single reference on the ordered extent that was
 * inserted.
 */
173 174 175
static int __btrfs_add_ordered_extent(struct inode *inode, u64 file_offset,
				      u64 start, u64 len, u64 disk_len,
				      int type, int dio)
C
Chris Mason 已提交
176 177
{
	struct btrfs_ordered_inode_tree *tree;
178 179
	struct rb_node *node;
	struct btrfs_ordered_extent *entry;
C
Chris Mason 已提交
180

181 182
	tree = &BTRFS_I(inode)->ordered_tree;
	entry = kzalloc(sizeof(*entry), GFP_NOFS);
C
Chris Mason 已提交
183 184 185
	if (!entry)
		return -ENOMEM;

186 187 188
	entry->file_offset = file_offset;
	entry->start = start;
	entry->len = len;
C
Chris Mason 已提交
189
	entry->disk_len = disk_len;
190
	entry->bytes_left = len;
191
	entry->inode = inode;
Y
Yan Zheng 已提交
192
	if (type != BTRFS_ORDERED_IO_DONE && type != BTRFS_ORDERED_COMPLETE)
Y
Yan Zheng 已提交
193
		set_bit(type, &entry->flags);
194

195 196 197
	if (dio)
		set_bit(BTRFS_ORDERED_DIRECT, &entry->flags);

198 199 200 201
	/* one ref for the tree */
	atomic_set(&entry->refs, 1);
	init_waitqueue_head(&entry->wait);
	INIT_LIST_HEAD(&entry->list);
202
	INIT_LIST_HEAD(&entry->root_extent_list);
C
Chris Mason 已提交
203

204
	spin_lock(&tree->lock);
205 206
	node = tree_insert(&tree->tree, file_offset,
			   &entry->rb_node);
C
Chris Mason 已提交
207
	BUG_ON(node);
208
	spin_unlock(&tree->lock);
C
Chris Mason 已提交
209

210 211 212 213 214
	spin_lock(&BTRFS_I(inode)->root->fs_info->ordered_extent_lock);
	list_add_tail(&entry->root_extent_list,
		      &BTRFS_I(inode)->root->fs_info->ordered_extents);
	spin_unlock(&BTRFS_I(inode)->root->fs_info->ordered_extent_lock);

215
	BUG_ON(node);
C
Chris Mason 已提交
216 217 218
	return 0;
}

219 220 221 222 223 224 225 226 227 228 229 230 231 232
int btrfs_add_ordered_extent(struct inode *inode, u64 file_offset,
			     u64 start, u64 len, u64 disk_len, int type)
{
	return __btrfs_add_ordered_extent(inode, file_offset, start, len,
					  disk_len, type, 0);
}

int btrfs_add_ordered_extent_dio(struct inode *inode, u64 file_offset,
				 u64 start, u64 len, u64 disk_len, int type)
{
	return __btrfs_add_ordered_extent(inode, file_offset, start, len,
					  disk_len, type, 1);
}

233 234
/*
 * Add a struct btrfs_ordered_sum into the list of checksums to be inserted
235 236
 * when an ordered extent is finished.  If the list covers more than one
 * ordered extent, it is split across multiples.
237
 */
238 239 240
int btrfs_add_ordered_sum(struct inode *inode,
			  struct btrfs_ordered_extent *entry,
			  struct btrfs_ordered_sum *sum)
C
Chris Mason 已提交
241
{
242
	struct btrfs_ordered_inode_tree *tree;
C
Chris Mason 已提交
243

244
	tree = &BTRFS_I(inode)->ordered_tree;
245
	spin_lock(&tree->lock);
246
	list_add_tail(&sum->list, &entry->list);
247
	spin_unlock(&tree->lock);
248
	return 0;
C
Chris Mason 已提交
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 304 305 306 307 308 309 310 311 312 313 314 315 316 317
/*
 * this is used to account for finished IO across a given range
 * of the file.  The IO may span ordered extents.  If
 * a given ordered_extent is completely done, 1 is returned, otherwise
 * 0.
 *
 * test_and_set_bit on a flag in the struct btrfs_ordered_extent is used
 * to make sure this function only returns 1 once for a given ordered extent.
 *
 * file_offset is updated to one byte past the range that is recorded as
 * complete.  This allows you to walk forward in the file.
 */
int btrfs_dec_test_first_ordered_pending(struct inode *inode,
				   struct btrfs_ordered_extent **cached,
				   u64 *file_offset, u64 io_size)
{
	struct btrfs_ordered_inode_tree *tree;
	struct rb_node *node;
	struct btrfs_ordered_extent *entry = NULL;
	int ret;
	u64 dec_end;
	u64 dec_start;
	u64 to_dec;

	tree = &BTRFS_I(inode)->ordered_tree;
	spin_lock(&tree->lock);
	node = tree_search(tree, *file_offset);
	if (!node) {
		ret = 1;
		goto out;
	}

	entry = rb_entry(node, struct btrfs_ordered_extent, rb_node);
	if (!offset_in_entry(entry, *file_offset)) {
		ret = 1;
		goto out;
	}

	dec_start = max(*file_offset, entry->file_offset);
	dec_end = min(*file_offset + io_size, entry->file_offset +
		      entry->len);
	*file_offset = dec_end;
	if (dec_start > dec_end) {
		printk(KERN_CRIT "bad ordering dec_start %llu end %llu\n",
		       (unsigned long long)dec_start,
		       (unsigned long long)dec_end);
	}
	to_dec = dec_end - dec_start;
	if (to_dec > entry->bytes_left) {
		printk(KERN_CRIT "bad ordered accounting left %llu size %llu\n",
		       (unsigned long long)entry->bytes_left,
		       (unsigned long long)to_dec);
	}
	entry->bytes_left -= to_dec;
	if (entry->bytes_left == 0)
		ret = test_and_set_bit(BTRFS_ORDERED_IO_DONE, &entry->flags);
	else
		ret = 1;
out:
	if (!ret && cached && entry) {
		*cached = entry;
		atomic_inc(&entry->refs);
	}
	spin_unlock(&tree->lock);
	return ret == 0;
}

318 319 320 321 322 323 324 325 326
/*
 * this is used to account for finished IO across a given range
 * of the file.  The IO should not span ordered extents.  If
 * a given ordered_extent is completely done, 1 is returned, otherwise
 * 0.
 *
 * test_and_set_bit on a flag in the struct btrfs_ordered_extent is used
 * to make sure this function only returns 1 once for a given ordered extent.
 */
327
int btrfs_dec_test_ordered_pending(struct inode *inode,
328
				   struct btrfs_ordered_extent **cached,
329
				   u64 file_offset, u64 io_size)
C
Chris Mason 已提交
330
{
331
	struct btrfs_ordered_inode_tree *tree;
C
Chris Mason 已提交
332
	struct rb_node *node;
333
	struct btrfs_ordered_extent *entry = NULL;
334 335 336
	int ret;

	tree = &BTRFS_I(inode)->ordered_tree;
337
	spin_lock(&tree->lock);
338
	node = tree_search(tree, file_offset);
C
Chris Mason 已提交
339
	if (!node) {
340 341
		ret = 1;
		goto out;
C
Chris Mason 已提交
342 343
	}

344 345 346 347
	entry = rb_entry(node, struct btrfs_ordered_extent, rb_node);
	if (!offset_in_entry(entry, file_offset)) {
		ret = 1;
		goto out;
C
Chris Mason 已提交
348
	}
349

350 351 352 353 354 355 356
	if (io_size > entry->bytes_left) {
		printk(KERN_CRIT "bad ordered accounting left %llu size %llu\n",
		       (unsigned long long)entry->bytes_left,
		       (unsigned long long)io_size);
	}
	entry->bytes_left -= io_size;
	if (entry->bytes_left == 0)
357
		ret = test_and_set_bit(BTRFS_ORDERED_IO_DONE, &entry->flags);
358 359
	else
		ret = 1;
360
out:
361 362 363 364
	if (!ret && cached && entry) {
		*cached = entry;
		atomic_inc(&entry->refs);
	}
365
	spin_unlock(&tree->lock);
366 367
	return ret == 0;
}
C
Chris Mason 已提交
368

369 370 371 372
/*
 * used to drop a reference on an ordered extent.  This will free
 * the extent if the last reference is dropped
 */
373 374
int btrfs_put_ordered_extent(struct btrfs_ordered_extent *entry)
{
375 376 377 378
	struct list_head *cur;
	struct btrfs_ordered_sum *sum;

	if (atomic_dec_and_test(&entry->refs)) {
C
Chris Mason 已提交
379
		while (!list_empty(&entry->list)) {
380 381 382 383 384
			cur = entry->list.next;
			sum = list_entry(cur, struct btrfs_ordered_sum, list);
			list_del(&sum->list);
			kfree(sum);
		}
385
		kfree(entry);
386
	}
387
	return 0;
C
Chris Mason 已提交
388
}
389

390 391
/*
 * remove an ordered extent from the tree.  No references are dropped
392
 * and you must wake_up entry->wait.  You must hold the tree lock
393
 * while you call this function.
394
 */
395
static int __btrfs_remove_ordered_extent(struct inode *inode,
396
				struct btrfs_ordered_extent *entry)
397
{
398
	struct btrfs_ordered_inode_tree *tree;
399
	struct btrfs_root *root = BTRFS_I(inode)->root;
400 401
	struct rb_node *node;

402 403
	tree = &BTRFS_I(inode)->ordered_tree;
	node = &entry->rb_node;
404
	rb_erase(node, &tree->tree);
405 406
	tree->last = NULL;
	set_bit(BTRFS_ORDERED_COMPLETE, &entry->flags);
407

408
	spin_lock(&root->fs_info->ordered_extent_lock);
409
	list_del_init(&entry->root_extent_list);
410 411 412 413 414 415 416 417 418 419

	/*
	 * we have no more ordered extents for this inode and
	 * no dirty pages.  We can safely remove it from the
	 * list of ordered extents
	 */
	if (RB_EMPTY_ROOT(&tree->tree) &&
	    !mapping_tagged(inode->i_mapping, PAGECACHE_TAG_DIRTY)) {
		list_del_init(&BTRFS_I(inode)->ordered_operations);
	}
420
	spin_unlock(&root->fs_info->ordered_extent_lock);
421

422 423 424 425 426 427 428 429 430 431 432 433 434 435
	return 0;
}

/*
 * remove an ordered extent from the tree.  No references are dropped
 * but any waiters are woken.
 */
int btrfs_remove_ordered_extent(struct inode *inode,
				struct btrfs_ordered_extent *entry)
{
	struct btrfs_ordered_inode_tree *tree;
	int ret;

	tree = &BTRFS_I(inode)->ordered_tree;
436
	spin_lock(&tree->lock);
437
	ret = __btrfs_remove_ordered_extent(inode, entry);
438
	spin_unlock(&tree->lock);
439
	wake_up(&entry->wait);
440 441

	return ret;
442 443
}

C
Chris Mason 已提交
444 445 446 447
/*
 * wait for all the ordered extents in a root.  This is done when balancing
 * space between drives.
 */
Y
Yan, Zheng 已提交
448 449
int btrfs_wait_ordered_extents(struct btrfs_root *root,
			       int nocow_only, int delay_iput)
450 451 452 453 454 455 456 457 458 459
{
	struct list_head splice;
	struct list_head *cur;
	struct btrfs_ordered_extent *ordered;
	struct inode *inode;

	INIT_LIST_HEAD(&splice);

	spin_lock(&root->fs_info->ordered_extent_lock);
	list_splice_init(&root->fs_info->ordered_extents, &splice);
460
	while (!list_empty(&splice)) {
461 462 463
		cur = splice.next;
		ordered = list_entry(cur, struct btrfs_ordered_extent,
				     root_extent_list);
464
		if (nocow_only &&
Y
Yan Zheng 已提交
465 466
		    !test_bit(BTRFS_ORDERED_NOCOW, &ordered->flags) &&
		    !test_bit(BTRFS_ORDERED_PREALLOC, &ordered->flags)) {
467 468
			list_move(&ordered->root_extent_list,
				  &root->fs_info->ordered_extents);
469 470 471 472
			cond_resched_lock(&root->fs_info->ordered_extent_lock);
			continue;
		}

473 474 475 476
		list_del_init(&ordered->root_extent_list);
		atomic_inc(&ordered->refs);

		/*
477
		 * the inode may be getting freed (in sys_unlink path).
478
		 */
479 480
		inode = igrab(ordered->inode);

481 482
		spin_unlock(&root->fs_info->ordered_extent_lock);

483 484 485
		if (inode) {
			btrfs_start_ordered_extent(inode, ordered, 1);
			btrfs_put_ordered_extent(ordered);
Y
Yan, Zheng 已提交
486 487 488 489
			if (delay_iput)
				btrfs_add_delayed_iput(inode);
			else
				iput(inode);
490 491 492
		} else {
			btrfs_put_ordered_extent(ordered);
		}
493 494 495 496 497 498 499

		spin_lock(&root->fs_info->ordered_extent_lock);
	}
	spin_unlock(&root->fs_info->ordered_extent_lock);
	return 0;
}

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
/*
 * this is used during transaction commit to write all the inodes
 * added to the ordered operation list.  These files must be fully on
 * disk before the transaction commits.
 *
 * we have two modes here, one is to just start the IO via filemap_flush
 * and the other is to wait for all the io.  When we wait, we have an
 * extra check to make sure the ordered operation list really is empty
 * before we return
 */
int btrfs_run_ordered_operations(struct btrfs_root *root, int wait)
{
	struct btrfs_inode *btrfs_inode;
	struct inode *inode;
	struct list_head splice;

	INIT_LIST_HEAD(&splice);

	mutex_lock(&root->fs_info->ordered_operations_mutex);
	spin_lock(&root->fs_info->ordered_extent_lock);
again:
	list_splice_init(&root->fs_info->ordered_operations, &splice);

	while (!list_empty(&splice)) {
		btrfs_inode = list_entry(splice.next, struct btrfs_inode,
				   ordered_operations);

		inode = &btrfs_inode->vfs_inode;

		list_del_init(&btrfs_inode->ordered_operations);

		/*
		 * the inode may be getting freed (in sys_unlink path).
		 */
		inode = igrab(inode);

		if (!wait && inode) {
			list_add_tail(&BTRFS_I(inode)->ordered_operations,
			      &root->fs_info->ordered_operations);
		}
		spin_unlock(&root->fs_info->ordered_extent_lock);

		if (inode) {
			if (wait)
				btrfs_wait_ordered_range(inode, 0, (u64)-1);
			else
				filemap_flush(inode->i_mapping);
Y
Yan, Zheng 已提交
547
			btrfs_add_delayed_iput(inode);
548 549 550 551 552 553 554 555 556 557 558 559 560 561
		}

		cond_resched();
		spin_lock(&root->fs_info->ordered_extent_lock);
	}
	if (wait && !list_empty(&root->fs_info->ordered_operations))
		goto again;

	spin_unlock(&root->fs_info->ordered_extent_lock);
	mutex_unlock(&root->fs_info->ordered_operations_mutex);

	return 0;
}

562 563 564 565 566 567 568 569 570 571
/*
 * Used to start IO or wait for a given ordered extent to finish.
 *
 * If wait is one, this effectively waits on page writeback for all the pages
 * in the extent, and it waits on the io completion code to insert
 * metadata into the btree corresponding to the extent
 */
void btrfs_start_ordered_extent(struct inode *inode,
				       struct btrfs_ordered_extent *entry,
				       int wait)
572 573 574
{
	u64 start = entry->file_offset;
	u64 end = start + entry->len - 1;
575

576 577 578 579 580
	/*
	 * pages in the range can be dirty, clean or writeback.  We
	 * start IO on any dirty ones so the wait doesn't stall waiting
	 * for pdflush to find them
	 */
581 582
	if (!test_bit(BTRFS_ORDERED_DIRECT, &entry->flags))
		filemap_fdatawrite_range(inode->i_mapping, start, end);
C
Chris Mason 已提交
583
	if (wait) {
584 585
		wait_event(entry->wait, test_bit(BTRFS_ORDERED_COMPLETE,
						 &entry->flags));
C
Chris Mason 已提交
586
	}
587
}
588

589 590 591
/*
 * Used to wait on ordered extents across a large range of bytes.
 */
592
int btrfs_wait_ordered_range(struct inode *inode, u64 start, u64 len)
593 594
{
	u64 end;
595
	u64 orig_end;
596
	struct btrfs_ordered_extent *ordered;
597
	int found;
598 599

	if (start + len < start) {
600
		orig_end = INT_LIMIT(loff_t);
601 602
	} else {
		orig_end = start + len - 1;
603 604
		if (orig_end > INT_LIMIT(loff_t))
			orig_end = INT_LIMIT(loff_t);
605
	}
C
Chris Mason 已提交
606
again:
607 608 609
	/* start IO across the range first to instantiate any delalloc
	 * extents
	 */
610
	filemap_fdatawrite_range(inode->i_mapping, start, orig_end);
611

612 613 614 615
	/* The compression code will leave pages locked but return from
	 * writepage without setting the page writeback.  Starting again
	 * with WB_SYNC_ALL will end up waiting for the IO to actually start.
	 */
616
	filemap_fdatawrite_range(inode->i_mapping, start, orig_end);
617

618
	filemap_fdatawait_range(inode->i_mapping, start, orig_end);
619

620
	end = orig_end;
621
	found = 0;
C
Chris Mason 已提交
622
	while (1) {
623
		ordered = btrfs_lookup_first_ordered_extent(inode, end);
C
Chris Mason 已提交
624
		if (!ordered)
625
			break;
626
		if (ordered->file_offset > orig_end) {
627 628 629 630 631 632 633
			btrfs_put_ordered_extent(ordered);
			break;
		}
		if (ordered->file_offset + ordered->len < start) {
			btrfs_put_ordered_extent(ordered);
			break;
		}
634
		found++;
635
		btrfs_start_ordered_extent(inode, ordered, 1);
636 637
		end = ordered->file_offset;
		btrfs_put_ordered_extent(ordered);
638
		if (end == 0 || end == start)
639 640 641
			break;
		end--;
	}
642 643
	if (found || test_range_bit(&BTRFS_I(inode)->io_tree, start, orig_end,
			   EXTENT_DELALLOC, 0, NULL)) {
644
		schedule_timeout(1);
C
Chris Mason 已提交
645 646
		goto again;
	}
647
	return 0;
648 649
}

650 651 652 653
/*
 * find an ordered extent corresponding to file_offset.  return NULL if
 * nothing is found, otherwise take a reference on the extent and return it
 */
654 655 656 657 658 659 660 661
struct btrfs_ordered_extent *btrfs_lookup_ordered_extent(struct inode *inode,
							 u64 file_offset)
{
	struct btrfs_ordered_inode_tree *tree;
	struct rb_node *node;
	struct btrfs_ordered_extent *entry = NULL;

	tree = &BTRFS_I(inode)->ordered_tree;
662
	spin_lock(&tree->lock);
663 664 665 666 667 668 669 670 671 672
	node = tree_search(tree, file_offset);
	if (!node)
		goto out;

	entry = rb_entry(node, struct btrfs_ordered_extent, rb_node);
	if (!offset_in_entry(entry, file_offset))
		entry = NULL;
	if (entry)
		atomic_inc(&entry->refs);
out:
673
	spin_unlock(&tree->lock);
674 675 676
	return entry;
}

677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717
/* Since the DIO code tries to lock a wide area we need to look for any ordered
 * extents that exist in the range, rather than just the start of the range.
 */
struct btrfs_ordered_extent *btrfs_lookup_ordered_range(struct inode *inode,
							u64 file_offset,
							u64 len)
{
	struct btrfs_ordered_inode_tree *tree;
	struct rb_node *node;
	struct btrfs_ordered_extent *entry = NULL;

	tree = &BTRFS_I(inode)->ordered_tree;
	spin_lock(&tree->lock);
	node = tree_search(tree, file_offset);
	if (!node) {
		node = tree_search(tree, file_offset + len);
		if (!node)
			goto out;
	}

	while (1) {
		entry = rb_entry(node, struct btrfs_ordered_extent, rb_node);
		if (range_overlaps(entry, file_offset, len))
			break;

		if (entry->file_offset >= file_offset + len) {
			entry = NULL;
			break;
		}
		entry = NULL;
		node = rb_next(node);
		if (!node)
			break;
	}
out:
	if (entry)
		atomic_inc(&entry->refs);
	spin_unlock(&tree->lock);
	return entry;
}

718 719 720 721
/*
 * lookup and return any extent before 'file_offset'.  NULL is returned
 * if none is found
 */
722
struct btrfs_ordered_extent *
C
Chris Mason 已提交
723
btrfs_lookup_first_ordered_extent(struct inode *inode, u64 file_offset)
724 725 726 727 728 729
{
	struct btrfs_ordered_inode_tree *tree;
	struct rb_node *node;
	struct btrfs_ordered_extent *entry = NULL;

	tree = &BTRFS_I(inode)->ordered_tree;
730
	spin_lock(&tree->lock);
731 732 733 734 735 736 737
	node = tree_search(tree, file_offset);
	if (!node)
		goto out;

	entry = rb_entry(node, struct btrfs_ordered_extent, rb_node);
	atomic_inc(&entry->refs);
out:
738
	spin_unlock(&tree->lock);
739
	return entry;
740
}
741

742 743 744 745
/*
 * After an extent is done, call this to conditionally update the on disk
 * i_size.  i_size is updated to cover any fully written part of the file.
 */
746
int btrfs_ordered_update_i_size(struct inode *inode, u64 offset,
747 748 749 750 751 752 753
				struct btrfs_ordered_extent *ordered)
{
	struct btrfs_ordered_inode_tree *tree = &BTRFS_I(inode)->ordered_tree;
	struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree;
	u64 disk_i_size;
	u64 new_i_size;
	u64 i_size_test;
754
	u64 i_size = i_size_read(inode);
755
	struct rb_node *node;
756
	struct rb_node *prev = NULL;
757
	struct btrfs_ordered_extent *test;
758 759 760 761
	int ret = 1;

	if (ordered)
		offset = entry_end(ordered);
762 763
	else
		offset = ALIGN(offset, BTRFS_I(inode)->root->sectorsize);
764

765
	spin_lock(&tree->lock);
766 767
	disk_i_size = BTRFS_I(inode)->disk_i_size;

768 769 770 771 772 773 774
	/* truncate file */
	if (disk_i_size > i_size) {
		BTRFS_I(inode)->disk_i_size = i_size;
		ret = 0;
		goto out;
	}

775 776 777 778
	/*
	 * if the disk i_size is already at the inode->i_size, or
	 * this ordered extent is inside the disk i_size, we're done
	 */
779
	if (disk_i_size == i_size || offset <= disk_i_size) {
780 781 782 783 784 785 786
		goto out;
	}

	/*
	 * we can't update the disk_isize if there are delalloc bytes
	 * between disk_i_size and  this ordered extent
	 */
787
	if (test_range_bit(io_tree, disk_i_size, offset - 1,
788
			   EXTENT_DELALLOC, 0, NULL)) {
789 790 791 792 793 794 795
		goto out;
	}
	/*
	 * walk backward from this ordered extent to disk_i_size.
	 * if we find an ordered extent then we can't update disk i_size
	 * yet
	 */
796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811
	if (ordered) {
		node = rb_prev(&ordered->rb_node);
	} else {
		prev = tree_search(tree, offset);
		/*
		 * we insert file extents without involving ordered struct,
		 * so there should be no ordered struct cover this offset
		 */
		if (prev) {
			test = rb_entry(prev, struct btrfs_ordered_extent,
					rb_node);
			BUG_ON(offset_in_entry(test, offset));
		}
		node = prev;
	}
	while (node) {
812 813 814
		test = rb_entry(node, struct btrfs_ordered_extent, rb_node);
		if (test->file_offset + test->len <= disk_i_size)
			break;
815
		if (test->file_offset >= i_size)
816 817 818
			break;
		if (test->file_offset >= disk_i_size)
			goto out;
819
		node = rb_prev(node);
820
	}
821
	new_i_size = min_t(u64, offset, i_size);
822 823 824 825 826 827 828

	/*
	 * at this point, we know we can safely update i_size to at least
	 * the offset from this ordered extent.  But, we need to
	 * walk forward and see if ios from higher up in the file have
	 * finished.
	 */
829 830 831 832 833 834 835 836
	if (ordered) {
		node = rb_next(&ordered->rb_node);
	} else {
		if (prev)
			node = rb_next(prev);
		else
			node = rb_first(&tree->tree);
	}
837 838 839 840 841 842 843
	i_size_test = 0;
	if (node) {
		/*
		 * do we have an area where IO might have finished
		 * between our ordered extent and the next one.
		 */
		test = rb_entry(node, struct btrfs_ordered_extent, rb_node);
844
		if (test->file_offset > offset)
Y
Yan Zheng 已提交
845
			i_size_test = test->file_offset;
846
	} else {
847
		i_size_test = i_size;
848 849 850 851 852 853 854 855
	}

	/*
	 * i_size_test is the end of a region after this ordered
	 * extent where there are no ordered extents.  As long as there
	 * are no delalloc bytes in this area, it is safe to update
	 * disk_i_size to the end of the region.
	 */
856 857 858 859
	if (i_size_test > offset &&
	    !test_range_bit(io_tree, offset, i_size_test - 1,
			    EXTENT_DELALLOC, 0, NULL)) {
		new_i_size = min_t(u64, i_size_test, i_size);
860 861
	}
	BTRFS_I(inode)->disk_i_size = new_i_size;
862
	ret = 0;
863
out:
864 865 866 867 868 869 870
	/*
	 * we need to remove the ordered extent with the tree lock held
	 * so that other people calling this function don't find our fully
	 * processed ordered entry and skip updating the i_size
	 */
	if (ordered)
		__btrfs_remove_ordered_extent(inode, ordered);
871
	spin_unlock(&tree->lock);
872 873 874
	if (ordered)
		wake_up(&ordered->wait);
	return ret;
875
}
876

877 878 879 880 881
/*
 * search the ordered extents for one corresponding to 'offset' and
 * try to find a checksum.  This is used because we allow pages to
 * be reclaimed before their checksum is actually put into the btree
 */
882 883
int btrfs_find_ordered_sum(struct inode *inode, u64 offset, u64 disk_bytenr,
			   u32 *sum)
884 885 886 887 888
{
	struct btrfs_ordered_sum *ordered_sum;
	struct btrfs_sector_sum *sector_sums;
	struct btrfs_ordered_extent *ordered;
	struct btrfs_ordered_inode_tree *tree = &BTRFS_I(inode)->ordered_tree;
889 890 891
	unsigned long num_sectors;
	unsigned long i;
	u32 sectorsize = BTRFS_I(inode)->root->sectorsize;
892 893 894 895 896 897
	int ret = 1;

	ordered = btrfs_lookup_ordered_extent(inode, offset);
	if (!ordered)
		return 1;

898
	spin_lock(&tree->lock);
Q
Qinghuang Feng 已提交
899
	list_for_each_entry_reverse(ordered_sum, &ordered->list, list) {
900
		if (disk_bytenr >= ordered_sum->bytenr) {
901
			num_sectors = ordered_sum->len / sectorsize;
902
			sector_sums = ordered_sum->sums;
903
			for (i = 0; i < num_sectors; i++) {
904
				if (sector_sums[i].bytenr == disk_bytenr) {
905 906 907 908 909
					*sum = sector_sums[i].sum;
					ret = 0;
					goto out;
				}
			}
910 911 912
		}
	}
out:
913
	spin_unlock(&tree->lock);
914
	btrfs_put_ordered_extent(ordered);
915 916 917
	return ret;
}

918

919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963
/*
 * add a given inode to the list of inodes that must be fully on
 * disk before a transaction commit finishes.
 *
 * This basically gives us the ext3 style data=ordered mode, and it is mostly
 * used to make sure renamed files are fully on disk.
 *
 * It is a noop if the inode is already fully on disk.
 *
 * If trans is not null, we'll do a friendly check for a transaction that
 * is already flushing things and force the IO down ourselves.
 */
int btrfs_add_ordered_operation(struct btrfs_trans_handle *trans,
				struct btrfs_root *root,
				struct inode *inode)
{
	u64 last_mod;

	last_mod = max(BTRFS_I(inode)->generation, BTRFS_I(inode)->last_trans);

	/*
	 * if this file hasn't been changed since the last transaction
	 * commit, we can safely return without doing anything
	 */
	if (last_mod < root->fs_info->last_trans_committed)
		return 0;

	/*
	 * the transaction is already committing.  Just start the IO and
	 * don't bother with all of this list nonsense
	 */
	if (trans && root->fs_info->running_transaction->blocked) {
		btrfs_wait_ordered_range(inode, 0, (u64)-1);
		return 0;
	}

	spin_lock(&root->fs_info->ordered_extent_lock);
	if (list_empty(&BTRFS_I(inode)->ordered_operations)) {
		list_add_tail(&BTRFS_I(inode)->ordered_operations,
			      &root->fs_info->ordered_operations);
	}
	spin_unlock(&root->fs_info->ordered_extent_lock);

	return 0;
}