ordered-data.c 21.9 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 20
/*
 * 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/gfp.h>
#include <linux/slab.h>
21
#include <linux/blkdev.h>
22 23
#include <linux/writeback.h>
#include <linux/pagevec.h>
C
Chris Mason 已提交
24 25 26
#include "ctree.h"
#include "transaction.h"
#include "btrfs_inode.h"
27
#include "extent_io.h"
C
Chris Mason 已提交
28

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

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

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

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

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

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

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

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

C
Chris Mason 已提交
91
	while (prev && file_offset >= entry_end(prev_entry)) {
92 93 94 95 96 97 98 99 100 101 102 103 104
		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 已提交
105
	while (prev && file_offset < entry_end(prev_entry)) {
106 107 108 109 110 111
		test = rb_prev(prev);
		if (!test)
			break;
		prev_entry = rb_entry(test, struct btrfs_ordered_extent,
				      rb_node);
		prev = test;
C
Chris Mason 已提交
112 113 114 115 116
	}
	*prev_ret = prev;
	return NULL;
}

C
Chris Mason 已提交
117 118 119
/*
 * helper to check if a given offset is inside a given entry
 */
120 121 122 123 124 125 126 127
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;
}

C
Chris Mason 已提交
128 129 130 131
/*
 * look find the first ordered struct that has this offset, otherwise
 * the first one less than this offset
 */
132 133
static inline struct rb_node *tree_search(struct btrfs_ordered_inode_tree *tree,
					  u64 file_offset)
C
Chris Mason 已提交
134
{
135
	struct rb_root *root = &tree->tree;
C
Chris Mason 已提交
136 137
	struct rb_node *prev;
	struct rb_node *ret;
138 139 140 141 142 143 144 145 146
	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 已提交
147
	if (!ret)
148 149 150
		ret = prev;
	if (ret)
		tree->last = ret;
C
Chris Mason 已提交
151 152 153
	return ret;
}

154 155 156 157 158 159 160 161 162 163 164
/* 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.
 */
165
int btrfs_add_ordered_extent(struct inode *inode, u64 file_offset,
Y
Yan Zheng 已提交
166
			     u64 start, u64 len, u64 disk_len, int type)
C
Chris Mason 已提交
167 168
{
	struct btrfs_ordered_inode_tree *tree;
169 170
	struct rb_node *node;
	struct btrfs_ordered_extent *entry;
C
Chris Mason 已提交
171

172 173
	tree = &BTRFS_I(inode)->ordered_tree;
	entry = kzalloc(sizeof(*entry), GFP_NOFS);
C
Chris Mason 已提交
174 175 176
	if (!entry)
		return -ENOMEM;

177 178 179
	entry->file_offset = file_offset;
	entry->start = start;
	entry->len = len;
C
Chris Mason 已提交
180
	entry->disk_len = disk_len;
181
	entry->bytes_left = len;
182
	entry->inode = inode;
Y
Yan Zheng 已提交
183
	if (type != BTRFS_ORDERED_IO_DONE && type != BTRFS_ORDERED_COMPLETE)
Y
Yan Zheng 已提交
184
		set_bit(type, &entry->flags);
185

186 187 188 189
	/* one ref for the tree */
	atomic_set(&entry->refs, 1);
	init_waitqueue_head(&entry->wait);
	INIT_LIST_HEAD(&entry->list);
190
	INIT_LIST_HEAD(&entry->root_extent_list);
C
Chris Mason 已提交
191

192
	spin_lock(&tree->lock);
193 194
	node = tree_insert(&tree->tree, file_offset,
			   &entry->rb_node);
C
Chris Mason 已提交
195
	BUG_ON(node);
196
	spin_unlock(&tree->lock);
C
Chris Mason 已提交
197

198 199 200 201 202
	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);

203
	BUG_ON(node);
C
Chris Mason 已提交
204 205 206
	return 0;
}

207 208
/*
 * Add a struct btrfs_ordered_sum into the list of checksums to be inserted
209 210
 * when an ordered extent is finished.  If the list covers more than one
 * ordered extent, it is split across multiples.
211
 */
212 213 214
int btrfs_add_ordered_sum(struct inode *inode,
			  struct btrfs_ordered_extent *entry,
			  struct btrfs_ordered_sum *sum)
C
Chris Mason 已提交
215
{
216
	struct btrfs_ordered_inode_tree *tree;
C
Chris Mason 已提交
217

218
	tree = &BTRFS_I(inode)->ordered_tree;
219
	spin_lock(&tree->lock);
220
	list_add_tail(&sum->list, &entry->list);
221
	spin_unlock(&tree->lock);
222
	return 0;
C
Chris Mason 已提交
223 224
}

225 226 227 228 229 230 231 232 233
/*
 * 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.
 */
234
int btrfs_dec_test_ordered_pending(struct inode *inode,
235
				   struct btrfs_ordered_extent **cached,
236
				   u64 file_offset, u64 io_size)
C
Chris Mason 已提交
237
{
238
	struct btrfs_ordered_inode_tree *tree;
C
Chris Mason 已提交
239
	struct rb_node *node;
240
	struct btrfs_ordered_extent *entry = NULL;
241 242 243
	int ret;

	tree = &BTRFS_I(inode)->ordered_tree;
244
	spin_lock(&tree->lock);
245
	node = tree_search(tree, file_offset);
C
Chris Mason 已提交
246
	if (!node) {
247 248
		ret = 1;
		goto out;
C
Chris Mason 已提交
249 250
	}

251 252 253 254
	entry = rb_entry(node, struct btrfs_ordered_extent, rb_node);
	if (!offset_in_entry(entry, file_offset)) {
		ret = 1;
		goto out;
C
Chris Mason 已提交
255
	}
256

257 258 259 260 261 262 263
	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)
264
		ret = test_and_set_bit(BTRFS_ORDERED_IO_DONE, &entry->flags);
265 266
	else
		ret = 1;
267
out:
268 269 270 271
	if (!ret && cached && entry) {
		*cached = entry;
		atomic_inc(&entry->refs);
	}
272
	spin_unlock(&tree->lock);
273 274
	return ret == 0;
}
C
Chris Mason 已提交
275

276 277 278 279
/*
 * used to drop a reference on an ordered extent.  This will free
 * the extent if the last reference is dropped
 */
280 281
int btrfs_put_ordered_extent(struct btrfs_ordered_extent *entry)
{
282 283 284 285
	struct list_head *cur;
	struct btrfs_ordered_sum *sum;

	if (atomic_dec_and_test(&entry->refs)) {
C
Chris Mason 已提交
286
		while (!list_empty(&entry->list)) {
287 288 289 290 291
			cur = entry->list.next;
			sum = list_entry(cur, struct btrfs_ordered_sum, list);
			list_del(&sum->list);
			kfree(sum);
		}
292
		kfree(entry);
293
	}
294
	return 0;
C
Chris Mason 已提交
295
}
296

297 298
/*
 * remove an ordered extent from the tree.  No references are dropped
299
 * and you must wake_up entry->wait.  You must hold the tree lock
300
 * while you call this function.
301
 */
302
static int __btrfs_remove_ordered_extent(struct inode *inode,
303
				struct btrfs_ordered_extent *entry)
304
{
305
	struct btrfs_ordered_inode_tree *tree;
306 307
	struct rb_node *node;

308 309
	tree = &BTRFS_I(inode)->ordered_tree;
	node = &entry->rb_node;
310
	rb_erase(node, &tree->tree);
311 312
	tree->last = NULL;
	set_bit(BTRFS_ORDERED_COMPLETE, &entry->flags);
313

314 315 316 317 318 319
	spin_lock(&BTRFS_I(inode)->accounting_lock);
	BTRFS_I(inode)->outstanding_extents--;
	spin_unlock(&BTRFS_I(inode)->accounting_lock);
	btrfs_unreserve_metadata_for_delalloc(BTRFS_I(inode)->root,
					      inode, 1);

320 321
	spin_lock(&BTRFS_I(inode)->root->fs_info->ordered_extent_lock);
	list_del_init(&entry->root_extent_list);
322 323 324 325 326 327 328 329 330 331

	/*
	 * 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);
	}
332 333
	spin_unlock(&BTRFS_I(inode)->root->fs_info->ordered_extent_lock);

334 335 336 337 338 339 340 341 342 343 344 345 346 347
	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;
348
	spin_lock(&tree->lock);
349
	ret = __btrfs_remove_ordered_extent(inode, entry);
350
	spin_unlock(&tree->lock);
351
	wake_up(&entry->wait);
352 353

	return ret;
354 355
}

C
Chris Mason 已提交
356 357 358 359
/*
 * wait for all the ordered extents in a root.  This is done when balancing
 * space between drives.
 */
Y
Yan, Zheng 已提交
360 361
int btrfs_wait_ordered_extents(struct btrfs_root *root,
			       int nocow_only, int delay_iput)
362 363 364 365 366 367 368 369 370 371
{
	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);
372
	while (!list_empty(&splice)) {
373 374 375
		cur = splice.next;
		ordered = list_entry(cur, struct btrfs_ordered_extent,
				     root_extent_list);
376
		if (nocow_only &&
Y
Yan Zheng 已提交
377 378
		    !test_bit(BTRFS_ORDERED_NOCOW, &ordered->flags) &&
		    !test_bit(BTRFS_ORDERED_PREALLOC, &ordered->flags)) {
379 380
			list_move(&ordered->root_extent_list,
				  &root->fs_info->ordered_extents);
381 382 383 384
			cond_resched_lock(&root->fs_info->ordered_extent_lock);
			continue;
		}

385 386 387 388
		list_del_init(&ordered->root_extent_list);
		atomic_inc(&ordered->refs);

		/*
389
		 * the inode may be getting freed (in sys_unlink path).
390
		 */
391 392
		inode = igrab(ordered->inode);

393 394
		spin_unlock(&root->fs_info->ordered_extent_lock);

395 396 397
		if (inode) {
			btrfs_start_ordered_extent(inode, ordered, 1);
			btrfs_put_ordered_extent(ordered);
Y
Yan, Zheng 已提交
398 399 400 401
			if (delay_iput)
				btrfs_add_delayed_iput(inode);
			else
				iput(inode);
402 403 404
		} else {
			btrfs_put_ordered_extent(ordered);
		}
405 406 407 408 409 410 411

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

412 413 414 415 416 417 418 419 420 421 422 423 424 425 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
/*
 * 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 已提交
459
			btrfs_add_delayed_iput(inode);
460 461 462 463 464 465 466 467 468 469 470 471 472 473
		}

		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;
}

474 475 476 477 478 479 480 481 482 483
/*
 * 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)
484 485 486
{
	u64 start = entry->file_offset;
	u64 end = start + entry->len - 1;
487

488 489 490 491 492
	/*
	 * 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
	 */
493
	filemap_fdatawrite_range(inode->i_mapping, start, end);
C
Chris Mason 已提交
494
	if (wait) {
495 496
		wait_event(entry->wait, test_bit(BTRFS_ORDERED_COMPLETE,
						 &entry->flags));
C
Chris Mason 已提交
497
	}
498
}
499

500 501 502
/*
 * Used to wait on ordered extents across a large range of bytes.
 */
503
int btrfs_wait_ordered_range(struct inode *inode, u64 start, u64 len)
504 505
{
	u64 end;
506 507
	u64 orig_end;
	u64 wait_end;
508
	struct btrfs_ordered_extent *ordered;
509
	int found;
510 511

	if (start + len < start) {
512
		orig_end = INT_LIMIT(loff_t);
513 514
	} else {
		orig_end = start + len - 1;
515 516
		if (orig_end > INT_LIMIT(loff_t))
			orig_end = INT_LIMIT(loff_t);
517
	}
518
	wait_end = orig_end;
C
Chris Mason 已提交
519
again:
520 521 522
	/* start IO across the range first to instantiate any delalloc
	 * extents
	 */
523
	filemap_fdatawrite_range(inode->i_mapping, start, orig_end);
524

525 526 527 528
	/* 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.
	 */
529
	filemap_fdatawrite_range(inode->i_mapping, start, orig_end);
530

531
	filemap_fdatawait_range(inode->i_mapping, start, orig_end);
532

533
	end = orig_end;
534
	found = 0;
C
Chris Mason 已提交
535
	while (1) {
536
		ordered = btrfs_lookup_first_ordered_extent(inode, end);
C
Chris Mason 已提交
537
		if (!ordered)
538
			break;
539
		if (ordered->file_offset > orig_end) {
540 541 542 543 544 545 546
			btrfs_put_ordered_extent(ordered);
			break;
		}
		if (ordered->file_offset + ordered->len < start) {
			btrfs_put_ordered_extent(ordered);
			break;
		}
547
		found++;
548
		btrfs_start_ordered_extent(inode, ordered, 1);
549 550
		end = ordered->file_offset;
		btrfs_put_ordered_extent(ordered);
551
		if (end == 0 || end == start)
552 553 554
			break;
		end--;
	}
555 556
	if (found || test_range_bit(&BTRFS_I(inode)->io_tree, start, orig_end,
			   EXTENT_DELALLOC, 0, NULL)) {
557
		schedule_timeout(1);
C
Chris Mason 已提交
558 559
		goto again;
	}
560
	return 0;
561 562
}

563 564 565 566
/*
 * find an ordered extent corresponding to file_offset.  return NULL if
 * nothing is found, otherwise take a reference on the extent and return it
 */
567 568 569 570 571 572 573 574
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;
575
	spin_lock(&tree->lock);
576 577 578 579 580 581 582 583 584 585
	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:
586
	spin_unlock(&tree->lock);
587 588 589
	return entry;
}

590 591 592 593
/*
 * lookup and return any extent before 'file_offset'.  NULL is returned
 * if none is found
 */
594
struct btrfs_ordered_extent *
C
Chris Mason 已提交
595
btrfs_lookup_first_ordered_extent(struct inode *inode, u64 file_offset)
596 597 598 599 600 601
{
	struct btrfs_ordered_inode_tree *tree;
	struct rb_node *node;
	struct btrfs_ordered_extent *entry = NULL;

	tree = &BTRFS_I(inode)->ordered_tree;
602
	spin_lock(&tree->lock);
603 604 605 606 607 608 609
	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:
610
	spin_unlock(&tree->lock);
611
	return entry;
612
}
613

614 615 616 617
/*
 * 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.
 */
618
int btrfs_ordered_update_i_size(struct inode *inode, u64 offset,
619 620 621 622 623 624 625
				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;
626
	u64 i_size = i_size_read(inode);
627
	struct rb_node *node;
628
	struct rb_node *prev = NULL;
629
	struct btrfs_ordered_extent *test;
630 631 632 633
	int ret = 1;

	if (ordered)
		offset = entry_end(ordered);
634 635
	else
		offset = ALIGN(offset, BTRFS_I(inode)->root->sectorsize);
636

637
	spin_lock(&tree->lock);
638 639
	disk_i_size = BTRFS_I(inode)->disk_i_size;

640 641 642 643 644 645 646
	/* truncate file */
	if (disk_i_size > i_size) {
		BTRFS_I(inode)->disk_i_size = i_size;
		ret = 0;
		goto out;
	}

647 648 649 650
	/*
	 * 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
	 */
651
	if (disk_i_size == i_size || offset <= disk_i_size) {
652 653 654 655 656 657 658
		goto out;
	}

	/*
	 * we can't update the disk_isize if there are delalloc bytes
	 * between disk_i_size and  this ordered extent
	 */
659
	if (test_range_bit(io_tree, disk_i_size, offset - 1,
660
			   EXTENT_DELALLOC, 0, NULL)) {
661 662 663 664 665 666 667
		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
	 */
668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683
	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) {
684 685 686
		test = rb_entry(node, struct btrfs_ordered_extent, rb_node);
		if (test->file_offset + test->len <= disk_i_size)
			break;
687
		if (test->file_offset >= i_size)
688 689 690
			break;
		if (test->file_offset >= disk_i_size)
			goto out;
691
		node = rb_prev(node);
692
	}
693
	new_i_size = min_t(u64, offset, i_size);
694 695 696 697 698 699 700

	/*
	 * 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.
	 */
701 702 703 704 705 706 707 708
	if (ordered) {
		node = rb_next(&ordered->rb_node);
	} else {
		if (prev)
			node = rb_next(prev);
		else
			node = rb_first(&tree->tree);
	}
709 710 711 712 713 714 715
	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);
716
		if (test->file_offset > offset)
Y
Yan Zheng 已提交
717
			i_size_test = test->file_offset;
718
	} else {
719
		i_size_test = i_size;
720 721 722 723 724 725 726 727
	}

	/*
	 * 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.
	 */
728 729 730 731
	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);
732 733
	}
	BTRFS_I(inode)->disk_i_size = new_i_size;
734
	ret = 0;
735
out:
736 737 738 739 740 741 742
	/*
	 * 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);
743
	spin_unlock(&tree->lock);
744 745 746
	if (ordered)
		wake_up(&ordered->wait);
	return ret;
747
}
748

749 750 751 752 753
/*
 * 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
 */
754 755
int btrfs_find_ordered_sum(struct inode *inode, u64 offset, u64 disk_bytenr,
			   u32 *sum)
756 757 758 759 760
{
	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;
761 762 763
	unsigned long num_sectors;
	unsigned long i;
	u32 sectorsize = BTRFS_I(inode)->root->sectorsize;
764 765 766 767 768 769
	int ret = 1;

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

770
	spin_lock(&tree->lock);
Q
Qinghuang Feng 已提交
771
	list_for_each_entry_reverse(ordered_sum, &ordered->list, list) {
772
		if (disk_bytenr >= ordered_sum->bytenr) {
773
			num_sectors = ordered_sum->len / sectorsize;
774
			sector_sums = ordered_sum->sums;
775
			for (i = 0; i < num_sectors; i++) {
776
				if (sector_sums[i].bytenr == disk_bytenr) {
777 778 779 780 781
					*sum = sector_sums[i].sum;
					ret = 0;
					goto out;
				}
			}
782 783 784
		}
	}
out:
785
	spin_unlock(&tree->lock);
786
	btrfs_put_ordered_extent(ordered);
787 788 789
	return ret;
}

790

791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835
/*
 * 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;
}