file.c 36.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
/*
 * 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.
 */

C
Chris Mason 已提交
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37
#include <linux/fs.h>
#include <linux/pagemap.h>
#include <linux/highmem.h>
#include <linux/time.h>
#include <linux/init.h>
#include <linux/string.h>
#include <linux/smp_lock.h>
#include <linux/backing-dev.h>
#include <linux/mpage.h>
#include <linux/swap.h>
#include <linux/writeback.h>
#include <linux/statfs.h>
#include <linux/compat.h>
#include "ctree.h"
#include "disk-io.h"
#include "transaction.h"
#include "btrfs_inode.h"
#include "ioctl.h"
#include "print-tree.h"
38 39
#include "tree-log.h"
#include "locking.h"
40
#include "compat.h"
C
Chris Mason 已提交
41 42


C
Chris Mason 已提交
43 44 45
/* simple helper to fault in pages and copy.  This should go away
 * and be replaced with calls into generic code.
 */
C
Chris Mason 已提交
46
static noinline int btrfs_copy_from_user(loff_t pos, int num_pages,
47 48
					 int write_bytes,
					 struct page **prepared_pages,
C
Chris Mason 已提交
49
					 const char __user *buf)
C
Chris Mason 已提交
50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
{
	long page_fault = 0;
	int i;
	int offset = pos & (PAGE_CACHE_SIZE - 1);

	for (i = 0; i < num_pages && write_bytes > 0; i++, offset = 0) {
		size_t count = min_t(size_t,
				     PAGE_CACHE_SIZE - offset, write_bytes);
		struct page *page = prepared_pages[i];
		fault_in_pages_readable(buf, count);

		/* Copy data from userspace to the current page */
		kmap(page);
		page_fault = __copy_from_user(page_address(page) + offset,
					      buf, count);
		/* Flush processor's dcache for this page */
		flush_dcache_page(page);
		kunmap(page);
		buf += count;
		write_bytes -= count;

		if (page_fault)
			break;
	}
	return page_fault ? -EFAULT : 0;
}

C
Chris Mason 已提交
77 78 79
/*
 * unlocks pages after btrfs_file_write is done with them
 */
C
Chris Mason 已提交
80
static noinline void btrfs_drop_pages(struct page **pages, size_t num_pages)
C
Chris Mason 已提交
81 82 83 84 85
{
	size_t i;
	for (i = 0; i < num_pages; i++) {
		if (!pages[i])
			break;
C
Chris Mason 已提交
86 87 88 89
		/* page checked is some magic around finding pages that
		 * have been modified without going through btrfs_set_page_dirty
		 * clear it here
		 */
C
Chris Mason 已提交
90
		ClearPageChecked(pages[i]);
C
Chris Mason 已提交
91 92 93 94 95 96
		unlock_page(pages[i]);
		mark_page_accessed(pages[i]);
		page_cache_release(pages[i]);
	}
}

C
Chris Mason 已提交
97 98 99 100 101 102 103 104
/*
 * after copy_from_user, pages need to be dirtied and we need to make
 * sure holes are created between the current EOF and the start of
 * any next extents (if required).
 *
 * this also makes the decision about creating an inline extent vs
 * doing real data extents, marking pages dirty and delalloc as required.
 */
C
Chris Mason 已提交
105
static noinline int dirty_and_release_pages(struct btrfs_trans_handle *trans,
C
Chris Mason 已提交
106 107 108 109 110 111 112 113
				   struct btrfs_root *root,
				   struct file *file,
				   struct page **pages,
				   size_t num_pages,
				   loff_t pos,
				   size_t write_bytes)
{
	int err = 0;
114
	int i;
115
	struct inode *inode = fdentry(file)->d_inode;
116
	struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree;
117 118
	u64 hint_byte;
	u64 num_bytes;
119 120 121 122
	u64 start_pos;
	u64 end_of_last_block;
	u64 end_pos = pos + write_bytes;
	loff_t isize = i_size_read(inode);
C
Chris Mason 已提交
123

124
	start_pos = pos & ~((u64)root->sectorsize - 1);
125 126
	num_bytes = (write_bytes + pos - start_pos +
		    root->sectorsize - 1) & ~((u64)root->sectorsize - 1);
C
Chris Mason 已提交
127

128 129
	end_of_last_block = start_pos + num_bytes - 1;

130
	lock_extent(io_tree, start_pos, end_of_last_block, GFP_NOFS);
C
Chris Mason 已提交
131
	trans = btrfs_join_transaction(root, 1);
132 133 134 135 136
	if (!trans) {
		err = -ENOMEM;
		goto out_unlock;
	}
	btrfs_set_trans_block_group(trans, inode);
137
	hint_byte = 0;
138

139
	set_extent_uptodate(io_tree, start_pos, end_of_last_block, GFP_NOFS);
140

C
Chris Mason 已提交
141 142 143
	/* check for reserved extents on each page, we don't want
	 * to reset the delalloc bit on things that already have
	 * extents reserved.
144
	 */
C
Chris Mason 已提交
145 146 147 148 149 150
	btrfs_set_extent_delalloc(inode, start_pos, end_of_last_block);
	for (i = 0; i < num_pages; i++) {
		struct page *p = pages[i];
		SetPageUptodate(p);
		ClearPageChecked(p);
		set_page_dirty(p);
151 152 153 154
	}
	if (end_pos > isize) {
		i_size_write(inode, end_pos);
		btrfs_update_inode(trans, root, inode);
C
Chris Mason 已提交
155
	}
C
Chris Mason 已提交
156
	err = btrfs_end_transaction(trans, root);
157
out_unlock:
158
	unlock_extent(io_tree, start_pos, end_of_last_block, GFP_NOFS);
C
Chris Mason 已提交
159 160 161
	return err;
}

C
Chris Mason 已提交
162 163 164 165
/*
 * this drops all the extents in the cache that intersect the range
 * [start, end].  Existing extents are split as required.
 */
166 167
int btrfs_drop_extent_cache(struct inode *inode, u64 start, u64 end,
			    int skip_pinned)
168 169
{
	struct extent_map *em;
170 171
	struct extent_map *split = NULL;
	struct extent_map *split2 = NULL;
172
	struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree;
173
	u64 len = end - start + 1;
174 175
	int ret;
	int testend = 1;
176
	unsigned long flags;
C
Chris Mason 已提交
177
	int compressed = 0;
178

179
	WARN_ON(end < start);
180
	if (end == (u64)-1) {
181
		len = (u64)-1;
182 183
		testend = 0;
	}
C
Chris Mason 已提交
184
	while (1) {
185 186 187 188 189
		if (!split)
			split = alloc_extent_map(GFP_NOFS);
		if (!split2)
			split2 = alloc_extent_map(GFP_NOFS);

190
		spin_lock(&em_tree->lock);
191
		em = lookup_extent_mapping(em_tree, start, len);
192 193
		if (!em) {
			spin_unlock(&em_tree->lock);
194
			break;
195
		}
196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212
		flags = em->flags;
		if (skip_pinned && test_bit(EXTENT_FLAG_PINNED, &em->flags)) {
			spin_unlock(&em_tree->lock);
			if (em->start <= start &&
			    (!testend || em->start + em->len >= start + len)) {
				free_extent_map(em);
				break;
			}
			if (start < em->start) {
				len = em->start - start;
			} else {
				len = start + len - (em->start + em->len);
				start = em->start + em->len;
			}
			free_extent_map(em);
			continue;
		}
C
Chris Mason 已提交
213
		compressed = test_bit(EXTENT_FLAG_COMPRESSED, &em->flags);
214
		clear_bit(EXTENT_FLAG_PINNED, &em->flags);
215
		remove_extent_mapping(em_tree, em);
216 217 218 219 220

		if (em->block_start < EXTENT_MAP_LAST_BYTE &&
		    em->start < start) {
			split->start = em->start;
			split->len = start - em->start;
221
			split->orig_start = em->orig_start;
222
			split->block_start = em->block_start;
C
Chris Mason 已提交
223 224 225 226 227 228

			if (compressed)
				split->block_len = em->block_len;
			else
				split->block_len = split->len;

229
			split->bdev = em->bdev;
230
			split->flags = flags;
231 232 233 234 235 236 237 238 239 240 241 242 243
			ret = add_extent_mapping(em_tree, split);
			BUG_ON(ret);
			free_extent_map(split);
			split = split2;
			split2 = NULL;
		}
		if (em->block_start < EXTENT_MAP_LAST_BYTE &&
		    testend && em->start + em->len > start + len) {
			u64 diff = start + len - em->start;

			split->start = start + len;
			split->len = em->start + em->len - (start + len);
			split->bdev = em->bdev;
244
			split->flags = flags;
245

C
Chris Mason 已提交
246 247 248
			if (compressed) {
				split->block_len = em->block_len;
				split->block_start = em->block_start;
249
				split->orig_start = em->orig_start;
C
Chris Mason 已提交
250 251 252
			} else {
				split->block_len = split->len;
				split->block_start = em->block_start + diff;
253
				split->orig_start = split->start;
C
Chris Mason 已提交
254
			}
255 256 257 258 259 260

			ret = add_extent_mapping(em_tree, split);
			BUG_ON(ret);
			free_extent_map(split);
			split = NULL;
		}
261 262
		spin_unlock(&em_tree->lock);

263 264 265 266 267
		/* once for us */
		free_extent_map(em);
		/* once for the tree*/
		free_extent_map(em);
	}
268 269 270 271
	if (split)
		free_extent_map(split);
	if (split2)
		free_extent_map(split2);
272 273 274
	return 0;
}

275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293
int btrfs_check_file(struct btrfs_root *root, struct inode *inode)
{
	return 0;
#if 0
	struct btrfs_path *path;
	struct btrfs_key found_key;
	struct extent_buffer *leaf;
	struct btrfs_file_extent_item *extent;
	u64 last_offset = 0;
	int nritems;
	int slot;
	int found_type;
	int ret;
	int err = 0;
	u64 extent_end = 0;

	path = btrfs_alloc_path();
	ret = btrfs_lookup_file_extent(NULL, root, path, inode->i_ino,
				       last_offset, 0);
C
Chris Mason 已提交
294
	while (1) {
295 296 297 298 299 300 301 302 303 304 305 306 307 308 309
		nritems = btrfs_header_nritems(path->nodes[0]);
		if (path->slots[0] >= nritems) {
			ret = btrfs_next_leaf(root, path);
			if (ret)
				goto out;
			nritems = btrfs_header_nritems(path->nodes[0]);
		}
		slot = path->slots[0];
		leaf = path->nodes[0];
		btrfs_item_key_to_cpu(leaf, &found_key, slot);
		if (found_key.objectid != inode->i_ino)
			break;
		if (found_key.type != BTRFS_EXTENT_DATA_KEY)
			goto out;

C
Chris Mason 已提交
310
		if (found_key.offset < last_offset) {
311 312
			WARN_ON(1);
			btrfs_print_leaf(root, leaf);
C
Chris Mason 已提交
313 314 315 316
			printk(KERN_ERR "inode %lu found offset %llu "
			       "expected %llu\n", inode->i_ino,
			       (unsigned long long)found_key.offset,
			       (unsigned long long)last_offset);
317 318 319 320 321 322 323 324 325 326 327 328 329
			err = 1;
			goto out;
		}
		extent = btrfs_item_ptr(leaf, slot,
					struct btrfs_file_extent_item);
		found_type = btrfs_file_extent_type(leaf, extent);
		if (found_type == BTRFS_FILE_EXTENT_REG) {
			extent_end = found_key.offset +
			     btrfs_file_extent_num_bytes(leaf, extent);
		} else if (found_type == BTRFS_FILE_EXTENT_INLINE) {
			struct btrfs_item *item;
			item = btrfs_item_nr(leaf, slot);
			extent_end = found_key.offset +
C
Chris Mason 已提交
330
			     btrfs_file_extent_inline_len(leaf, extent);
331
			extent_end = (extent_end + root->sectorsize - 1) &
C
Chris Mason 已提交
332
				~((u64)root->sectorsize - 1);
333 334 335 336
		}
		last_offset = extent_end;
		path->slots[0]++;
	}
C
Chris Mason 已提交
337
	if (0 && last_offset < inode->i_size) {
338 339
		WARN_ON(1);
		btrfs_print_leaf(root, leaf);
C
Chris Mason 已提交
340 341 342
		printk(KERN_ERR "inode %lu found offset %llu size %llu\n",
		       inode->i_ino, (unsigned long long)last_offset,
		       (unsigned long long)inode->i_size);
343 344 345 346 347 348 349 350 351
		err = 1;

	}
out:
	btrfs_free_path(path);
	return err;
#endif
}

C
Chris Mason 已提交
352 353 354 355 356 357 358 359
/*
 * this is very complex, but the basic idea is to drop all extents
 * in the range start - end.  hint_block is filled in with a block number
 * that would be a good hint to the block allocator for this file.
 *
 * If an extent intersects the range but is not entirely inside the range
 * it is either truncated or split.  Anything entirely inside the range
 * is deleted from the tree.
C
Chris Mason 已提交
360 361 362
 *
 * inline_limit is used to tell this code which offsets in the file to keep
 * if they contain inline extents.
C
Chris Mason 已提交
363
 */
C
Chris Mason 已提交
364
noinline int btrfs_drop_extents(struct btrfs_trans_handle *trans,
C
Chris Mason 已提交
365
		       struct btrfs_root *root, struct inode *inode,
366
		       u64 start, u64 end, u64 inline_limit, u64 *hint_byte)
C
Chris Mason 已提交
367
{
368
	u64 extent_end = 0;
Y
Yan Zheng 已提交
369
	u64 locked_end = end;
370
	u64 search_start = start;
Z
Zheng Yan 已提交
371
	u64 leaf_start;
C
Chris Mason 已提交
372
	u64 ram_bytes = 0;
373 374
	u64 orig_parent = 0;
	u64 disk_bytenr = 0;
C
Chris Mason 已提交
375 376
	u8 compression;
	u8 encryption;
C
Chris Mason 已提交
377
	u16 other_encoding = 0;
Z
Zheng Yan 已提交
378 379
	u64 root_gen;
	u64 root_owner;
380
	struct extent_buffer *leaf;
C
Chris Mason 已提交
381 382
	struct btrfs_file_extent_item *extent;
	struct btrfs_path *path;
383 384 385 386
	struct btrfs_key key;
	struct btrfs_file_extent_item old;
	int keep;
	int slot;
C
Chris Mason 已提交
387
	int bookend;
Y
Yan Zheng 已提交
388
	int found_type = 0;
C
Chris Mason 已提交
389 390
	int found_extent;
	int found_inline;
C
Chris Mason 已提交
391
	int recow;
392
	int ret;
C
Chris Mason 已提交
393

C
Chris Mason 已提交
394
	inline_limit = 0;
395
	btrfs_drop_extent_cache(inode, start, end - 1, 0);
396

C
Chris Mason 已提交
397 398 399
	path = btrfs_alloc_path();
	if (!path)
		return -ENOMEM;
C
Chris Mason 已提交
400
	while (1) {
C
Chris Mason 已提交
401
		recow = 0;
C
Chris Mason 已提交
402 403 404 405 406 407 408 409 410 411 412 413
		btrfs_release_path(root, path);
		ret = btrfs_lookup_file_extent(trans, root, path, inode->i_ino,
					       search_start, -1);
		if (ret < 0)
			goto out;
		if (ret > 0) {
			if (path->slots[0] == 0) {
				ret = 0;
				goto out;
			}
			path->slots[0]--;
		}
414
next_slot:
C
Chris Mason 已提交
415 416 417 418
		keep = 0;
		bookend = 0;
		found_extent = 0;
		found_inline = 0;
Z
Zheng Yan 已提交
419 420 421
		leaf_start = 0;
		root_gen = 0;
		root_owner = 0;
C
Chris Mason 已提交
422 423
		compression = 0;
		encryption = 0;
C
Chris Mason 已提交
424
		extent = NULL;
425
		leaf = path->nodes[0];
C
Chris Mason 已提交
426
		slot = path->slots[0];
427
		ret = 0;
428
		btrfs_item_key_to_cpu(leaf, &key, slot);
429 430
		if (btrfs_key_type(&key) == BTRFS_EXTENT_DATA_KEY &&
		    key.offset >= end) {
C
Chris Mason 已提交
431 432
			goto out;
		}
433 434
		if (btrfs_key_type(&key) > BTRFS_EXTENT_DATA_KEY ||
		    key.objectid != inode->i_ino) {
C
Chris Mason 已提交
435 436
			goto out;
		}
C
Chris Mason 已提交
437
		if (recow) {
438
			search_start = max(key.offset, start);
C
Chris Mason 已提交
439 440
			continue;
		}
441 442 443
		if (btrfs_key_type(&key) == BTRFS_EXTENT_DATA_KEY) {
			extent = btrfs_item_ptr(leaf, slot,
						struct btrfs_file_extent_item);
444
			found_type = btrfs_file_extent_type(leaf, extent);
C
Chris Mason 已提交
445 446 447 448 449 450
			compression = btrfs_file_extent_compression(leaf,
								    extent);
			encryption = btrfs_file_extent_encryption(leaf,
								  extent);
			other_encoding = btrfs_file_extent_other_encoding(leaf,
								  extent);
Y
Yan Zheng 已提交
451 452
			if (found_type == BTRFS_FILE_EXTENT_REG ||
			    found_type == BTRFS_FILE_EXTENT_PREALLOC) {
453 454 455 456 457 458
				extent_end =
				     btrfs_file_extent_disk_bytenr(leaf,
								   extent);
				if (extent_end)
					*hint_byte = extent_end;

459
				extent_end = key.offset +
460
				     btrfs_file_extent_num_bytes(leaf, extent);
C
Chris Mason 已提交
461 462
				ram_bytes = btrfs_file_extent_ram_bytes(leaf,
								extent);
463 464 465 466
				found_extent = 1;
			} else if (found_type == BTRFS_FILE_EXTENT_INLINE) {
				found_inline = 1;
				extent_end = key.offset +
C
Chris Mason 已提交
467
				     btrfs_file_extent_inline_len(leaf, extent);
468 469 470
			}
		} else {
			extent_end = search_start;
C
Chris Mason 已提交
471 472 473
		}

		/* we found nothing we can drop */
474 475 476 477
		if ((!found_extent && !found_inline) ||
		    search_start >= extent_end) {
			int nextret;
			u32 nritems;
478
			nritems = btrfs_header_nritems(leaf);
479 480 481 482
			if (slot >= nritems - 1) {
				nextret = btrfs_next_leaf(root, path);
				if (nextret)
					goto out;
C
Chris Mason 已提交
483
				recow = 1;
484 485 486 487
			} else {
				path->slots[0]++;
			}
			goto next_slot;
C
Chris Mason 已提交
488 489
		}

C
Chris Mason 已提交
490
		if (end <= extent_end && start >= key.offset && found_inline)
491
			*hint_byte = EXTENT_MAP_INLINE;
Z
Zheng Yan 已提交
492 493 494 495 496 497 498

		if (found_extent) {
			read_extent_buffer(leaf, &old, (unsigned long)extent,
					   sizeof(old));
			root_gen = btrfs_header_generation(leaf);
			root_owner = btrfs_header_owner(leaf);
			leaf_start = leaf->start;
499
		}
Z
Zheng Yan 已提交
500

C
Chris Mason 已提交
501
		if (end < extent_end && end >= key.offset) {
502
			bookend = 1;
503
			if (found_inline && start <= key.offset)
504
				keep = 1;
C
Chris Mason 已提交
505
		}
Y
Yan Zheng 已提交
506

507 508 509 510 511 512 513 514 515 516 517 518 519
		if (bookend && found_extent) {
			if (locked_end < extent_end) {
				ret = try_lock_extent(&BTRFS_I(inode)->io_tree,
						locked_end, extent_end - 1,
						GFP_NOFS);
				if (!ret) {
					btrfs_release_path(root, path);
					lock_extent(&BTRFS_I(inode)->io_tree,
						locked_end, extent_end - 1,
						GFP_NOFS);
					locked_end = extent_end;
					continue;
				}
Y
Yan Zheng 已提交
520 521
				locked_end = extent_end;
			}
522 523 524 525 526 527 528 529 530 531
			orig_parent = path->nodes[0]->start;
			disk_bytenr = le64_to_cpu(old.disk_bytenr);
			if (disk_bytenr != 0) {
				ret = btrfs_inc_extent_ref(trans, root,
					   disk_bytenr,
					   le64_to_cpu(old.disk_num_bytes),
					   orig_parent, root->root_key.objectid,
					   trans->transid, inode->i_ino);
				BUG_ON(ret);
			}
Y
Yan Zheng 已提交
532 533 534 535 536 537 538 539
		}

		if (found_inline) {
			u64 mask = root->sectorsize - 1;
			search_start = (extent_end + mask) & ~mask;
		} else
			search_start = extent_end;

C
Chris Mason 已提交
540 541 542 543 544
		/* truncate existing extent */
		if (start > key.offset) {
			u64 new_num;
			u64 old_num;
			keep = 1;
545
			WARN_ON(start & (root->sectorsize - 1));
C
Chris Mason 已提交
546
			if (found_extent) {
547 548 549 550 551 552 553 554
				new_num = start - key.offset;
				old_num = btrfs_file_extent_num_bytes(leaf,
								      extent);
				*hint_byte =
					btrfs_file_extent_disk_bytenr(leaf,
								      extent);
				if (btrfs_file_extent_disk_bytenr(leaf,
								  extent)) {
555 556
					inode_sub_bytes(inode, old_num -
							new_num);
C
Chris Mason 已提交
557
				}
558 559
				btrfs_set_file_extent_num_bytes(leaf,
							extent, new_num);
560
				btrfs_mark_buffer_dirty(leaf);
561 562 563
			} else if (key.offset < inline_limit &&
				   (end > extent_end) &&
				   (inline_limit < extent_end)) {
564 565
				u32 new_size;
				new_size = btrfs_file_extent_calc_inline_size(
566
						   inline_limit - key.offset);
567 568
				inode_sub_bytes(inode, extent_end -
						inline_limit);
C
Chris Mason 已提交
569 570 571 572 573 574
				btrfs_set_file_extent_ram_bytes(leaf, extent,
							new_size);
				if (!compression && !encryption) {
					btrfs_truncate_item(trans, root, path,
							    new_size, 1);
				}
C
Chris Mason 已提交
575 576 577 578
			}
		}
		/* delete the entire extent */
		if (!keep) {
579 580 581
			if (found_inline)
				inode_sub_bytes(inode, extent_end -
						key.offset);
C
Chris Mason 已提交
582
			ret = btrfs_del_item(trans, root, path);
583
			/* TODO update progress marker and return */
C
Chris Mason 已提交
584 585
			BUG_ON(ret);
			extent = NULL;
Z
Zheng Yan 已提交
586 587
			btrfs_release_path(root, path);
			/* the extent will be freed later */
C
Chris Mason 已提交
588
		}
589
		if (bookend && found_inline && start <= key.offset) {
590 591
			u32 new_size;
			new_size = btrfs_file_extent_calc_inline_size(
592
						   extent_end - end);
593
			inode_sub_bytes(inode, end - key.offset);
C
Chris Mason 已提交
594 595 596 597 598
			btrfs_set_file_extent_ram_bytes(leaf, extent,
							new_size);
			if (!compression && !encryption)
				ret = btrfs_truncate_item(trans, root, path,
							  new_size, 0);
Z
Zheng Yan 已提交
599
			BUG_ON(ret);
600
		}
C
Chris Mason 已提交
601 602 603 604 605 606
		/* create bookend, splitting the extent in two */
		if (bookend && found_extent) {
			struct btrfs_key ins;
			ins.objectid = inode->i_ino;
			ins.offset = end;
			btrfs_set_key_type(&ins, BTRFS_EXTENT_DATA_KEY);
607

C
Chris Mason 已提交
608
			btrfs_release_path(root, path);
609
			path->leave_spinning = 1;
C
Chris Mason 已提交
610 611
			ret = btrfs_insert_empty_item(trans, root, path, &ins,
						      sizeof(*extent));
Z
Zheng Yan 已提交
612
			BUG_ON(ret);
613

614 615 616 617 618 619
			leaf = path->nodes[0];
			extent = btrfs_item_ptr(leaf, path->slots[0],
						struct btrfs_file_extent_item);
			write_extent_buffer(leaf, &old,
					    (unsigned long)extent, sizeof(old));

C
Chris Mason 已提交
620 621 622 623 624 625
			btrfs_set_file_extent_compression(leaf, extent,
							  compression);
			btrfs_set_file_extent_encryption(leaf, extent,
							 encryption);
			btrfs_set_file_extent_other_encoding(leaf, extent,
							     other_encoding);
626
			btrfs_set_file_extent_offset(leaf, extent,
627 628 629 630 631
				    le64_to_cpu(old.offset) + end - key.offset);
			WARN_ON(le64_to_cpu(old.num_bytes) <
				(extent_end - end));
			btrfs_set_file_extent_num_bytes(leaf, extent,
							extent_end - end);
C
Chris Mason 已提交
632 633 634 635 636 637 638 639 640

			/*
			 * set the ram bytes to the size of the full extent
			 * before splitting.  This is a worst case flag,
			 * but its the best we can do because we don't know
			 * how splitting affects compression
			 */
			btrfs_set_file_extent_ram_bytes(leaf, extent,
							ram_bytes);
Y
Yan Zheng 已提交
641
			btrfs_set_file_extent_type(leaf, extent, found_type);
642

643
			btrfs_unlock_up_safe(path, 1);
C
Chris Mason 已提交
644
			btrfs_mark_buffer_dirty(path->nodes[0]);
645
			btrfs_set_lock_blocking(path->nodes[0]);
Z
Zheng Yan 已提交
646 647

			if (disk_bytenr != 0) {
648
				ret = btrfs_update_extent_ref(trans, root,
649 650 651
						disk_bytenr,
						le64_to_cpu(old.disk_num_bytes),
						orig_parent,
C
Chris Mason 已提交
652
						leaf->start,
Z
Zheng Yan 已提交
653
						root->root_key.objectid,
654
						trans->transid, ins.objectid);
655

Z
Zheng Yan 已提交
656 657
				BUG_ON(ret);
			}
658
			path->leave_spinning = 0;
Z
Zheng Yan 已提交
659
			btrfs_release_path(root, path);
C
Chris Mason 已提交
660
			if (disk_bytenr != 0)
661
				inode_add_bytes(inode, extent_end - end);
Z
Zheng Yan 已提交
662 663 664
		}

		if (found_extent && !keep) {
665
			u64 old_disk_bytenr = le64_to_cpu(old.disk_bytenr);
Z
Zheng Yan 已提交
666

667
			if (old_disk_bytenr != 0) {
668 669
				inode_sub_bytes(inode,
						le64_to_cpu(old.num_bytes));
Z
Zheng Yan 已提交
670
				ret = btrfs_free_extent(trans, root,
671
						old_disk_bytenr,
Z
Zheng Yan 已提交
672 673
						le64_to_cpu(old.disk_num_bytes),
						leaf_start, root_owner,
674
						root_gen, key.objectid, 0);
Z
Zheng Yan 已提交
675
				BUG_ON(ret);
676
				*hint_byte = old_disk_bytenr;
Z
Zheng Yan 已提交
677 678 679 680
			}
		}

		if (search_start >= end) {
C
Chris Mason 已提交
681 682 683 684 685 686
			ret = 0;
			goto out;
		}
	}
out:
	btrfs_free_path(path);
Y
Yan Zheng 已提交
687 688 689 690
	if (locked_end > end) {
		unlock_extent(&BTRFS_I(inode)->io_tree, end, locked_end - 1,
			      GFP_NOFS);
	}
C
Chris Mason 已提交
691
	btrfs_check_file(root, inode);
C
Chris Mason 已提交
692 693 694
	return ret;
}

Y
Yan Zheng 已提交
695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748
static int extent_mergeable(struct extent_buffer *leaf, int slot,
			    u64 objectid, u64 bytenr, u64 *start, u64 *end)
{
	struct btrfs_file_extent_item *fi;
	struct btrfs_key key;
	u64 extent_end;

	if (slot < 0 || slot >= btrfs_header_nritems(leaf))
		return 0;

	btrfs_item_key_to_cpu(leaf, &key, slot);
	if (key.objectid != objectid || key.type != BTRFS_EXTENT_DATA_KEY)
		return 0;

	fi = btrfs_item_ptr(leaf, slot, struct btrfs_file_extent_item);
	if (btrfs_file_extent_type(leaf, fi) != BTRFS_FILE_EXTENT_REG ||
	    btrfs_file_extent_disk_bytenr(leaf, fi) != bytenr ||
	    btrfs_file_extent_compression(leaf, fi) ||
	    btrfs_file_extent_encryption(leaf, fi) ||
	    btrfs_file_extent_other_encoding(leaf, fi))
		return 0;

	extent_end = key.offset + btrfs_file_extent_num_bytes(leaf, fi);
	if ((*start && *start != key.offset) || (*end && *end != extent_end))
		return 0;

	*start = key.offset;
	*end = extent_end;
	return 1;
}

/*
 * Mark extent in the range start - end as written.
 *
 * This changes extent type from 'pre-allocated' to 'regular'. If only
 * part of extent is marked as written, the extent will be split into
 * two or three.
 */
int btrfs_mark_extent_written(struct btrfs_trans_handle *trans,
			      struct btrfs_root *root,
			      struct inode *inode, u64 start, u64 end)
{
	struct extent_buffer *leaf;
	struct btrfs_path *path;
	struct btrfs_file_extent_item *fi;
	struct btrfs_key key;
	u64 bytenr;
	u64 num_bytes;
	u64 extent_end;
	u64 extent_offset;
	u64 other_start;
	u64 other_end;
	u64 split = start;
	u64 locked_end = end;
749
	u64 orig_parent;
Y
Yan Zheng 已提交
750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 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 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893
	int extent_type;
	int split_end = 1;
	int ret;

	btrfs_drop_extent_cache(inode, start, end - 1, 0);

	path = btrfs_alloc_path();
	BUG_ON(!path);
again:
	key.objectid = inode->i_ino;
	key.type = BTRFS_EXTENT_DATA_KEY;
	if (split == start)
		key.offset = split;
	else
		key.offset = split - 1;

	ret = btrfs_search_slot(trans, root, &key, path, -1, 1);
	if (ret > 0 && path->slots[0] > 0)
		path->slots[0]--;

	leaf = path->nodes[0];
	btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
	BUG_ON(key.objectid != inode->i_ino ||
	       key.type != BTRFS_EXTENT_DATA_KEY);
	fi = btrfs_item_ptr(leaf, path->slots[0],
			    struct btrfs_file_extent_item);
	extent_type = btrfs_file_extent_type(leaf, fi);
	BUG_ON(extent_type != BTRFS_FILE_EXTENT_PREALLOC);
	extent_end = key.offset + btrfs_file_extent_num_bytes(leaf, fi);
	BUG_ON(key.offset > start || extent_end < end);

	bytenr = btrfs_file_extent_disk_bytenr(leaf, fi);
	num_bytes = btrfs_file_extent_disk_num_bytes(leaf, fi);
	extent_offset = btrfs_file_extent_offset(leaf, fi);

	if (key.offset == start)
		split = end;

	if (key.offset == start && extent_end == end) {
		int del_nr = 0;
		int del_slot = 0;
		u64 leaf_owner = btrfs_header_owner(leaf);
		u64 leaf_gen = btrfs_header_generation(leaf);
		other_start = end;
		other_end = 0;
		if (extent_mergeable(leaf, path->slots[0] + 1, inode->i_ino,
				     bytenr, &other_start, &other_end)) {
			extent_end = other_end;
			del_slot = path->slots[0] + 1;
			del_nr++;
			ret = btrfs_free_extent(trans, root, bytenr, num_bytes,
						leaf->start, leaf_owner,
						leaf_gen, inode->i_ino, 0);
			BUG_ON(ret);
		}
		other_start = 0;
		other_end = start;
		if (extent_mergeable(leaf, path->slots[0] - 1, inode->i_ino,
				     bytenr, &other_start, &other_end)) {
			key.offset = other_start;
			del_slot = path->slots[0];
			del_nr++;
			ret = btrfs_free_extent(trans, root, bytenr, num_bytes,
						leaf->start, leaf_owner,
						leaf_gen, inode->i_ino, 0);
			BUG_ON(ret);
		}
		split_end = 0;
		if (del_nr == 0) {
			btrfs_set_file_extent_type(leaf, fi,
						   BTRFS_FILE_EXTENT_REG);
			goto done;
		}

		fi = btrfs_item_ptr(leaf, del_slot - 1,
				    struct btrfs_file_extent_item);
		btrfs_set_file_extent_type(leaf, fi, BTRFS_FILE_EXTENT_REG);
		btrfs_set_file_extent_num_bytes(leaf, fi,
						extent_end - key.offset);
		btrfs_mark_buffer_dirty(leaf);

		ret = btrfs_del_items(trans, root, path, del_slot, del_nr);
		BUG_ON(ret);
		goto done;
	} else if (split == start) {
		if (locked_end < extent_end) {
			ret = try_lock_extent(&BTRFS_I(inode)->io_tree,
					locked_end, extent_end - 1, GFP_NOFS);
			if (!ret) {
				btrfs_release_path(root, path);
				lock_extent(&BTRFS_I(inode)->io_tree,
					locked_end, extent_end - 1, GFP_NOFS);
				locked_end = extent_end;
				goto again;
			}
			locked_end = extent_end;
		}
		btrfs_set_file_extent_num_bytes(leaf, fi, split - key.offset);
		extent_offset += split - key.offset;
	} else  {
		BUG_ON(key.offset != start);
		btrfs_set_file_extent_offset(leaf, fi, extent_offset +
					     split - key.offset);
		btrfs_set_file_extent_num_bytes(leaf, fi, extent_end - split);
		key.offset = split;
		btrfs_set_item_key_safe(trans, root, path, &key);
		extent_end = split;
	}

	if (extent_end == end) {
		split_end = 0;
		extent_type = BTRFS_FILE_EXTENT_REG;
	}
	if (extent_end == end && split == start) {
		other_start = end;
		other_end = 0;
		if (extent_mergeable(leaf, path->slots[0] + 1, inode->i_ino,
				     bytenr, &other_start, &other_end)) {
			path->slots[0]++;
			fi = btrfs_item_ptr(leaf, path->slots[0],
					    struct btrfs_file_extent_item);
			key.offset = split;
			btrfs_set_item_key_safe(trans, root, path, &key);
			btrfs_set_file_extent_offset(leaf, fi, extent_offset);
			btrfs_set_file_extent_num_bytes(leaf, fi,
							other_end - split);
			goto done;
		}
	}
	if (extent_end == end && split == end) {
		other_start = 0;
		other_end = start;
		if (extent_mergeable(leaf, path->slots[0] - 1 , inode->i_ino,
				     bytenr, &other_start, &other_end)) {
			path->slots[0]--;
			fi = btrfs_item_ptr(leaf, path->slots[0],
					    struct btrfs_file_extent_item);
			btrfs_set_file_extent_num_bytes(leaf, fi, extent_end -
							other_start);
			goto done;
		}
	}

	btrfs_mark_buffer_dirty(leaf);
894 895 896 897 898 899

	orig_parent = leaf->start;
	ret = btrfs_inc_extent_ref(trans, root, bytenr, num_bytes,
				   orig_parent, root->root_key.objectid,
				   trans->transid, inode->i_ino);
	BUG_ON(ret);
Y
Yan Zheng 已提交
900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919
	btrfs_release_path(root, path);

	key.offset = start;
	ret = btrfs_insert_empty_item(trans, root, path, &key, sizeof(*fi));
	BUG_ON(ret);

	leaf = path->nodes[0];
	fi = btrfs_item_ptr(leaf, path->slots[0],
			    struct btrfs_file_extent_item);
	btrfs_set_file_extent_generation(leaf, fi, trans->transid);
	btrfs_set_file_extent_type(leaf, fi, extent_type);
	btrfs_set_file_extent_disk_bytenr(leaf, fi, bytenr);
	btrfs_set_file_extent_disk_num_bytes(leaf, fi, num_bytes);
	btrfs_set_file_extent_offset(leaf, fi, extent_offset);
	btrfs_set_file_extent_num_bytes(leaf, fi, extent_end - key.offset);
	btrfs_set_file_extent_ram_bytes(leaf, fi, num_bytes);
	btrfs_set_file_extent_compression(leaf, fi, 0);
	btrfs_set_file_extent_encryption(leaf, fi, 0);
	btrfs_set_file_extent_other_encoding(leaf, fi, 0);

920
	if (orig_parent != leaf->start) {
921
		ret = btrfs_update_extent_ref(trans, root, bytenr, num_bytes,
922 923 924 925 926
					      orig_parent, leaf->start,
					      root->root_key.objectid,
					      trans->transid, inode->i_ino);
		BUG_ON(ret);
	}
Y
Yan Zheng 已提交
927 928 929 930 931 932 933 934 935 936 937 938 939 940 941
done:
	btrfs_mark_buffer_dirty(leaf);
	btrfs_release_path(root, path);
	if (split_end && split == start) {
		split = end;
		goto again;
	}
	if (locked_end > end) {
		unlock_extent(&BTRFS_I(inode)->io_tree, end, locked_end - 1,
			      GFP_NOFS);
	}
	btrfs_free_path(path);
	return 0;
}

C
Chris Mason 已提交
942
/*
C
Chris Mason 已提交
943 944 945
 * this gets pages into the page cache and locks them down, it also properly
 * waits for data=ordered extents to finish before allowing the pages to be
 * modified.
C
Chris Mason 已提交
946
 */
C
Chris Mason 已提交
947
static noinline int prepare_pages(struct btrfs_root *root, struct file *file,
948 949 950
			 struct page **pages, size_t num_pages,
			 loff_t pos, unsigned long first_index,
			 unsigned long last_index, size_t write_bytes)
C
Chris Mason 已提交
951 952 953
{
	int i;
	unsigned long index = pos >> PAGE_CACHE_SHIFT;
954
	struct inode *inode = fdentry(file)->d_inode;
C
Chris Mason 已提交
955
	int err = 0;
956
	u64 start_pos;
957
	u64 last_pos;
958

959
	start_pos = pos & ~((u64)root->sectorsize - 1);
960
	last_pos = ((u64)index + num_pages) << PAGE_CACHE_SHIFT;
C
Chris Mason 已提交
961

Y
Yan Zheng 已提交
962 963 964 965 966 967
	if (start_pos > inode->i_size) {
		err = btrfs_cont_expand(inode, start_pos);
		if (err)
			return err;
	}

C
Chris Mason 已提交
968
	memset(pages, 0, num_pages * sizeof(struct page *));
969
again:
C
Chris Mason 已提交
970 971 972 973
	for (i = 0; i < num_pages; i++) {
		pages[i] = grab_cache_page(inode->i_mapping, index + i);
		if (!pages[i]) {
			err = -ENOMEM;
974
			BUG_ON(1);
C
Chris Mason 已提交
975
		}
C
Chris Mason 已提交
976
		wait_on_page_writeback(pages[i]);
C
Chris Mason 已提交
977
	}
978
	if (start_pos < inode->i_size) {
979
		struct btrfs_ordered_extent *ordered;
980 981
		lock_extent(&BTRFS_I(inode)->io_tree,
			    start_pos, last_pos - 1, GFP_NOFS);
C
Chris Mason 已提交
982 983
		ordered = btrfs_lookup_first_ordered_extent(inode,
							    last_pos - 1);
984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000
		if (ordered &&
		    ordered->file_offset + ordered->len > start_pos &&
		    ordered->file_offset < last_pos) {
			btrfs_put_ordered_extent(ordered);
			unlock_extent(&BTRFS_I(inode)->io_tree,
				      start_pos, last_pos - 1, GFP_NOFS);
			for (i = 0; i < num_pages; i++) {
				unlock_page(pages[i]);
				page_cache_release(pages[i]);
			}
			btrfs_wait_ordered_range(inode, start_pos,
						 last_pos - start_pos);
			goto again;
		}
		if (ordered)
			btrfs_put_ordered_extent(ordered);

1001 1002 1003
		clear_extent_bits(&BTRFS_I(inode)->io_tree, start_pos,
				  last_pos - 1, EXTENT_DIRTY | EXTENT_DELALLOC,
				  GFP_NOFS);
1004 1005
		unlock_extent(&BTRFS_I(inode)->io_tree,
			      start_pos, last_pos - 1, GFP_NOFS);
1006
	}
1007
	for (i = 0; i < num_pages; i++) {
1008
		clear_page_dirty_for_io(pages[i]);
1009 1010 1011
		set_page_extent_mapped(pages[i]);
		WARN_ON(!PageLocked(pages[i]));
	}
C
Chris Mason 已提交
1012 1013 1014 1015 1016 1017 1018
	return 0;
}

static ssize_t btrfs_file_write(struct file *file, const char __user *buf,
				size_t count, loff_t *ppos)
{
	loff_t pos;
1019 1020 1021
	loff_t start_pos;
	ssize_t num_written = 0;
	ssize_t err = 0;
C
Chris Mason 已提交
1022
	int ret = 0;
1023
	struct inode *inode = fdentry(file)->d_inode;
C
Chris Mason 已提交
1024
	struct btrfs_root *root = BTRFS_I(inode)->root;
1025 1026
	struct page **pages = NULL;
	int nrptrs;
C
Chris Mason 已提交
1027 1028 1029
	struct page *pinned[2];
	unsigned long first_index;
	unsigned long last_index;
1030 1031 1032 1033
	int will_write;

	will_write = ((file->f_flags & O_SYNC) || IS_SYNC(inode) ||
		      (file->f_flags & O_DIRECT));
1034 1035 1036

	nrptrs = min((count + PAGE_CACHE_SIZE - 1) / PAGE_CACHE_SIZE,
		     PAGE_CACHE_SIZE / (sizeof(struct page *)));
C
Chris Mason 已提交
1037 1038
	pinned[0] = NULL;
	pinned[1] = NULL;
1039

C
Chris Mason 已提交
1040
	pos = *ppos;
1041 1042
	start_pos = pos;

C
Chris Mason 已提交
1043 1044 1045 1046
	vfs_check_frozen(inode->i_sb, SB_FREEZE_WRITE);
	current->backing_dev_info = inode->i_mapping->backing_dev_info;
	err = generic_write_checks(file, &pos, &count, S_ISBLK(inode->i_mode));
	if (err)
1047
		goto out_nolock;
C
Chris Mason 已提交
1048
	if (count == 0)
1049
		goto out_nolock;
1050

1051
	err = file_remove_suid(file);
C
Chris Mason 已提交
1052
	if (err)
1053
		goto out_nolock;
C
Chris Mason 已提交
1054 1055
	file_update_time(file);

1056
	pages = kmalloc(nrptrs * sizeof(struct page *), GFP_KERNEL);
C
Chris Mason 已提交
1057 1058

	mutex_lock(&inode->i_mutex);
1059
	BTRFS_I(inode)->sequence++;
C
Chris Mason 已提交
1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070
	first_index = pos >> PAGE_CACHE_SHIFT;
	last_index = (pos + count) >> PAGE_CACHE_SHIFT;

	/*
	 * there are lots of better ways to do this, but this code
	 * makes sure the first and last page in the file range are
	 * up to date and ready for cow
	 */
	if ((pos & (PAGE_CACHE_SIZE - 1))) {
		pinned[0] = grab_cache_page(inode->i_mapping, first_index);
		if (!PageUptodate(pinned[0])) {
C
Chris Mason 已提交
1071
			ret = btrfs_readpage(NULL, pinned[0]);
C
Chris Mason 已提交
1072 1073 1074 1075 1076 1077 1078 1079 1080
			BUG_ON(ret);
			wait_on_page_locked(pinned[0]);
		} else {
			unlock_page(pinned[0]);
		}
	}
	if ((pos + count) & (PAGE_CACHE_SIZE - 1)) {
		pinned[1] = grab_cache_page(inode->i_mapping, last_index);
		if (!PageUptodate(pinned[1])) {
C
Chris Mason 已提交
1081
			ret = btrfs_readpage(NULL, pinned[1]);
C
Chris Mason 已提交
1082 1083 1084 1085 1086 1087 1088
			BUG_ON(ret);
			wait_on_page_locked(pinned[1]);
		} else {
			unlock_page(pinned[1]);
		}
	}

C
Chris Mason 已提交
1089
	while (count > 0) {
C
Chris Mason 已提交
1090
		size_t offset = pos & (PAGE_CACHE_SIZE - 1);
1091 1092
		size_t write_bytes = min(count, nrptrs *
					(size_t)PAGE_CACHE_SIZE -
1093
					 offset);
C
Chris Mason 已提交
1094 1095 1096
		size_t num_pages = (write_bytes + PAGE_CACHE_SIZE - 1) >>
					PAGE_CACHE_SHIFT;

1097
		WARN_ON(num_pages > nrptrs);
1098
		memset(pages, 0, sizeof(struct page *) * nrptrs);
1099

J
Josef Bacik 已提交
1100
		ret = btrfs_check_data_free_space(root, inode, write_bytes);
1101 1102 1103
		if (ret)
			goto out;

C
Chris Mason 已提交
1104 1105
		ret = prepare_pages(root, file, pages, num_pages,
				    pos, first_index, last_index,
1106
				    write_bytes);
J
Josef Bacik 已提交
1107 1108 1109
		if (ret) {
			btrfs_free_reserved_data_space(root, inode,
						       write_bytes);
1110
			goto out;
J
Josef Bacik 已提交
1111
		}
C
Chris Mason 已提交
1112 1113 1114

		ret = btrfs_copy_from_user(pos, num_pages,
					   write_bytes, pages, buf);
1115
		if (ret) {
J
Josef Bacik 已提交
1116 1117
			btrfs_free_reserved_data_space(root, inode,
						       write_bytes);
1118 1119 1120
			btrfs_drop_pages(pages, num_pages);
			goto out;
		}
C
Chris Mason 已提交
1121 1122 1123 1124

		ret = dirty_and_release_pages(NULL, root, file, pages,
					      num_pages, pos, write_bytes);
		btrfs_drop_pages(pages, num_pages);
J
Josef Bacik 已提交
1125 1126 1127
		if (ret) {
			btrfs_free_reserved_data_space(root, inode,
						       write_bytes);
1128
			goto out;
J
Josef Bacik 已提交
1129
		}
C
Chris Mason 已提交
1130

1131 1132 1133
		if (will_write) {
			btrfs_fdatawrite_range(inode->i_mapping, pos,
					       pos + write_bytes - 1,
1134
					       WB_SYNC_ALL);
1135 1136 1137 1138 1139 1140 1141 1142 1143
		} else {
			balance_dirty_pages_ratelimited_nr(inode->i_mapping,
							   num_pages);
			if (num_pages <
			    (root->leafsize >> PAGE_CACHE_SHIFT) + 1)
				btrfs_btree_balance_dirty(root, 1);
			btrfs_throttle(root);
		}

C
Chris Mason 已提交
1144 1145 1146 1147 1148 1149 1150 1151
		buf += write_bytes;
		count -= write_bytes;
		pos += write_bytes;
		num_written += write_bytes;

		cond_resched();
	}
out:
1152
	mutex_unlock(&inode->i_mutex);
J
Josef Bacik 已提交
1153 1154
	if (ret)
		err = ret;
1155

1156
out_nolock:
1157
	kfree(pages);
C
Chris Mason 已提交
1158 1159 1160 1161 1162
	if (pinned[0])
		page_cache_release(pinned[0]);
	if (pinned[1])
		page_cache_release(pinned[1]);
	*ppos = pos;
1163

1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177
	/*
	 * we want to make sure fsync finds this change
	 * but we haven't joined a transaction running right now.
	 *
	 * Later on, someone is sure to update the inode and get the
	 * real transid recorded.
	 *
	 * We set last_trans now to the fs_info generation + 1,
	 * this will either be one more than the running transaction
	 * or the generation used for the next transaction if there isn't
	 * one running right now.
	 */
	BTRFS_I(inode)->last_trans = root->fs_info->generation + 1;

1178
	if (num_written > 0 && will_write) {
1179 1180
		struct btrfs_trans_handle *trans;

1181 1182
		err = btrfs_wait_ordered_range(inode, start_pos, num_written);
		if (err)
1183
			num_written = err;
1184

1185 1186 1187 1188 1189
		if ((file->f_flags & O_SYNC) || IS_SYNC(inode)) {
			trans = btrfs_start_transaction(root, 1);
			ret = btrfs_log_dentry_safe(trans, root,
						    file->f_dentry);
			if (ret == 0) {
1190 1191 1192 1193 1194
				ret = btrfs_sync_log(trans, root);
				if (ret == 0)
					btrfs_end_transaction(trans, root);
				else
					btrfs_commit_transaction(trans, root);
1195 1196 1197 1198 1199 1200 1201 1202
			} else {
				btrfs_commit_transaction(trans, root);
			}
		}
		if (file->f_flags & O_DIRECT) {
			invalidate_mapping_pages(inode->i_mapping,
			      start_pos >> PAGE_CACHE_SHIFT,
			     (start_pos + num_written - 1) >> PAGE_CACHE_SHIFT);
1203
		}
1204
	}
C
Chris Mason 已提交
1205 1206 1207 1208
	current->backing_dev_info = NULL;
	return num_written ? num_written : err;
}

C
Chris Mason 已提交
1209
int btrfs_release_file(struct inode *inode, struct file *filp)
1210
{
1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222
	/*
	 * ordered_data_close is set by settattr when we are about to truncate
	 * a file from a non-zero size to a zero size.  This tries to
	 * flush down new bytes that may have been written if the
	 * application were using truncate to replace a file in place.
	 */
	if (BTRFS_I(inode)->ordered_data_close) {
		BTRFS_I(inode)->ordered_data_close = 0;
		btrfs_add_ordered_operation(NULL, BTRFS_I(inode)->root, inode);
		if (inode->i_size > BTRFS_ORDERED_OPERATIONS_FLUSH_LIMIT)
			filemap_flush(inode->i_mapping);
	}
S
Sage Weil 已提交
1223 1224
	if (filp->private_data)
		btrfs_ioctl_trans_end(filp);
1225 1226 1227
	return 0;
}

C
Chris Mason 已提交
1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238
/*
 * fsync call for both files and directories.  This logs the inode into
 * the tree log instead of forcing full commits whenever possible.
 *
 * It needs to call filemap_fdatawait so that all ordered extent updates are
 * in the metadata btree are up to date for copying to the log.
 *
 * It drops the inode mutex before doing the tree log commit.  This is an
 * important optimization for directories because holding the mutex prevents
 * new operations on the dir while we write to disk.
 */
1239
int btrfs_sync_file(struct file *file, struct dentry *dentry, int datasync)
C
Chris Mason 已提交
1240 1241 1242
{
	struct inode *inode = dentry->d_inode;
	struct btrfs_root *root = BTRFS_I(inode)->root;
1243
	int ret = 0;
C
Chris Mason 已提交
1244 1245 1246
	struct btrfs_trans_handle *trans;

	/*
1247 1248
	 * check the transaction that last modified this inode
	 * and see if its already been committed
C
Chris Mason 已提交
1249
	 */
1250 1251
	if (!BTRFS_I(inode)->last_trans)
		goto out;
1252

1253 1254 1255 1256 1257 1258 1259 1260 1261
	mutex_lock(&root->fs_info->trans_mutex);
	if (BTRFS_I(inode)->last_trans <=
	    root->fs_info->last_trans_committed) {
		BTRFS_I(inode)->last_trans = 0;
		mutex_unlock(&root->fs_info->trans_mutex);
		goto out;
	}
	mutex_unlock(&root->fs_info->trans_mutex);

Y
Yan Zheng 已提交
1262
	root->log_batch++;
1263 1264
	filemap_fdatawrite(inode->i_mapping);
	btrfs_wait_ordered_range(inode, 0, (u64)-1);
Y
Yan Zheng 已提交
1265
	root->log_batch++;
1266

1267
	/*
1268 1269
	 * ok we haven't committed the transaction yet, lets do a commit
	 */
1270
	if (file && file->private_data)
S
Sage Weil 已提交
1271 1272
		btrfs_ioctl_trans_end(file);

C
Chris Mason 已提交
1273 1274 1275 1276 1277
	trans = btrfs_start_transaction(root, 1);
	if (!trans) {
		ret = -ENOMEM;
		goto out;
	}
1278

1279
	ret = btrfs_log_dentry_safe(trans, root, dentry);
C
Chris Mason 已提交
1280
	if (ret < 0)
1281
		goto out;
C
Chris Mason 已提交
1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292

	/* we've logged all the items and now have a consistent
	 * version of the file in the log.  It is possible that
	 * someone will come in and modify the file, but that's
	 * fine because the log is consistent on disk, and we
	 * have references to all of the file's extents
	 *
	 * It is possible that someone will come in and log the
	 * file again, but that will end up using the synchronization
	 * inside btrfs_sync_log to keep things safe.
	 */
1293
	mutex_unlock(&dentry->d_inode->i_mutex);
C
Chris Mason 已提交
1294

1295 1296 1297
	if (ret > 0) {
		ret = btrfs_commit_transaction(trans, root);
	} else {
1298 1299 1300 1301 1302
		ret = btrfs_sync_log(trans, root);
		if (ret == 0)
			ret = btrfs_end_transaction(trans, root);
		else
			ret = btrfs_commit_transaction(trans, root);
1303
	}
1304
	mutex_lock(&dentry->d_inode->i_mutex);
C
Chris Mason 已提交
1305 1306 1307 1308
out:
	return ret > 0 ? EIO : ret;
}

C
Chris Mason 已提交
1309
static struct vm_operations_struct btrfs_file_vm_ops = {
1310
	.fault		= filemap_fault,
C
Chris Mason 已提交
1311 1312 1313 1314 1315 1316 1317 1318 1319 1320
	.page_mkwrite	= btrfs_page_mkwrite,
};

static int btrfs_file_mmap(struct file	*filp, struct vm_area_struct *vma)
{
	vma->vm_ops = &btrfs_file_vm_ops;
	file_accessed(filp);
	return 0;
}

C
Chris Mason 已提交
1321 1322 1323
struct file_operations btrfs_file_operations = {
	.llseek		= generic_file_llseek,
	.read		= do_sync_read,
C
Chris Mason 已提交
1324
	.aio_read       = generic_file_aio_read,
C
Chris Mason 已提交
1325
	.splice_read	= generic_file_splice_read,
C
Chris Mason 已提交
1326
	.write		= btrfs_file_write,
C
Chris Mason 已提交
1327
	.mmap		= btrfs_file_mmap,
C
Chris Mason 已提交
1328
	.open		= generic_file_open,
1329
	.release	= btrfs_release_file,
C
Chris Mason 已提交
1330
	.fsync		= btrfs_sync_file,
C
Christoph Hellwig 已提交
1331
	.unlocked_ioctl	= btrfs_ioctl,
C
Chris Mason 已提交
1332
#ifdef CONFIG_COMPAT
C
Christoph Hellwig 已提交
1333
	.compat_ioctl	= btrfs_ioctl,
C
Chris Mason 已提交
1334 1335
#endif
};