file.c 34.6 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 609 610
			btrfs_release_path(root, path);
			ret = btrfs_insert_empty_item(trans, root, path, &ins,
						      sizeof(*extent));
Z
Zheng Yan 已提交
611
			BUG_ON(ret);
612

613 614 615 616 617 618
			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 已提交
619 620 621 622 623 624
			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);
625
			btrfs_set_file_extent_offset(leaf, extent,
626 627 628 629 630
				    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 已提交
631 632 633 634 635 636 637 638 639

			/*
			 * 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 已提交
640
			btrfs_set_file_extent_type(leaf, extent, found_type);
641

C
Chris Mason 已提交
642
			btrfs_mark_buffer_dirty(path->nodes[0]);
Z
Zheng Yan 已提交
643 644

			if (disk_bytenr != 0) {
645 646
				ret = btrfs_update_extent_ref(trans, root,
						disk_bytenr, orig_parent,
C
Chris Mason 已提交
647
						leaf->start,
Z
Zheng Yan 已提交
648
						root->root_key.objectid,
649
						trans->transid, ins.objectid);
650

Z
Zheng Yan 已提交
651 652 653
				BUG_ON(ret);
			}
			btrfs_release_path(root, path);
C
Chris Mason 已提交
654
			if (disk_bytenr != 0)
655
				inode_add_bytes(inode, extent_end - end);
Z
Zheng Yan 已提交
656 657 658
		}

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

661
			if (old_disk_bytenr != 0) {
662 663
				inode_sub_bytes(inode,
						le64_to_cpu(old.num_bytes));
Z
Zheng Yan 已提交
664
				ret = btrfs_free_extent(trans, root,
665
						old_disk_bytenr,
Z
Zheng Yan 已提交
666 667
						le64_to_cpu(old.disk_num_bytes),
						leaf_start, root_owner,
668
						root_gen, key.objectid, 0);
Z
Zheng Yan 已提交
669
				BUG_ON(ret);
670
				*hint_byte = old_disk_bytenr;
Z
Zheng Yan 已提交
671 672 673 674
			}
		}

		if (search_start >= end) {
C
Chris Mason 已提交
675 676 677 678 679 680
			ret = 0;
			goto out;
		}
	}
out:
	btrfs_free_path(path);
Y
Yan Zheng 已提交
681 682 683 684
	if (locked_end > end) {
		unlock_extent(&BTRFS_I(inode)->io_tree, end, locked_end - 1,
			      GFP_NOFS);
	}
C
Chris Mason 已提交
685
	btrfs_check_file(root, inode);
C
Chris Mason 已提交
686 687 688
	return ret;
}

Y
Yan Zheng 已提交
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 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
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;
743
	u64 orig_parent;
Y
Yan Zheng 已提交
744 745 746 747 748 749 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
	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);
888 889 890 891 892 893

	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 已提交
894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913
	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);

914 915 916 917 918 919 920
	if (orig_parent != leaf->start) {
		ret = btrfs_update_extent_ref(trans, root, bytenr,
					      orig_parent, leaf->start,
					      root->root_key.objectid,
					      trans->transid, inode->i_ino);
		BUG_ON(ret);
	}
Y
Yan Zheng 已提交
921 922 923 924 925 926 927 928 929 930 931 932 933 934 935
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 已提交
936
/*
C
Chris Mason 已提交
937 938 939
 * 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 已提交
940
 */
C
Chris Mason 已提交
941
static noinline int prepare_pages(struct btrfs_root *root, struct file *file,
942 943 944
			 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 已提交
945 946 947
{
	int i;
	unsigned long index = pos >> PAGE_CACHE_SHIFT;
948
	struct inode *inode = fdentry(file)->d_inode;
C
Chris Mason 已提交
949
	int err = 0;
950
	u64 start_pos;
951
	u64 last_pos;
952

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

Y
Yan Zheng 已提交
956 957 958 959 960 961
	if (start_pos > inode->i_size) {
		err = btrfs_cont_expand(inode, start_pos);
		if (err)
			return err;
	}

C
Chris Mason 已提交
962
	memset(pages, 0, num_pages * sizeof(struct page *));
963
again:
C
Chris Mason 已提交
964 965 966 967
	for (i = 0; i < num_pages; i++) {
		pages[i] = grab_cache_page(inode->i_mapping, index + i);
		if (!pages[i]) {
			err = -ENOMEM;
968
			BUG_ON(1);
C
Chris Mason 已提交
969
		}
C
Chris Mason 已提交
970
		wait_on_page_writeback(pages[i]);
C
Chris Mason 已提交
971
	}
972
	if (start_pos < inode->i_size) {
973
		struct btrfs_ordered_extent *ordered;
974 975
		lock_extent(&BTRFS_I(inode)->io_tree,
			    start_pos, last_pos - 1, GFP_NOFS);
C
Chris Mason 已提交
976 977
		ordered = btrfs_lookup_first_ordered_extent(inode,
							    last_pos - 1);
978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994
		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);

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

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

	will_write = ((file->f_flags & O_SYNC) || IS_SYNC(inode) ||
		      (file->f_flags & O_DIRECT));
1028 1029 1030

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

C
Chris Mason 已提交
1034
	pos = *ppos;
1035 1036
	start_pos = pos;

C
Chris Mason 已提交
1037 1038 1039 1040
	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)
1041
		goto out_nolock;
C
Chris Mason 已提交
1042
	if (count == 0)
1043
		goto out_nolock;
1044

1045
	err = file_remove_suid(file);
C
Chris Mason 已提交
1046
	if (err)
1047
		goto out_nolock;
C
Chris Mason 已提交
1048 1049
	file_update_time(file);

1050
	pages = kmalloc(nrptrs * sizeof(struct page *), GFP_KERNEL);
C
Chris Mason 已提交
1051 1052

	mutex_lock(&inode->i_mutex);
1053
	BTRFS_I(inode)->sequence++;
C
Chris Mason 已提交
1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064
	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 已提交
1065
			ret = btrfs_readpage(NULL, pinned[0]);
C
Chris Mason 已提交
1066 1067 1068 1069 1070 1071 1072 1073 1074
			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 已提交
1075
			ret = btrfs_readpage(NULL, pinned[1]);
C
Chris Mason 已提交
1076 1077 1078 1079 1080 1081 1082
			BUG_ON(ret);
			wait_on_page_locked(pinned[1]);
		} else {
			unlock_page(pinned[1]);
		}
	}

C
Chris Mason 已提交
1083
	while (count > 0) {
C
Chris Mason 已提交
1084
		size_t offset = pos & (PAGE_CACHE_SIZE - 1);
1085 1086
		size_t write_bytes = min(count, nrptrs *
					(size_t)PAGE_CACHE_SIZE -
1087
					 offset);
C
Chris Mason 已提交
1088 1089 1090
		size_t num_pages = (write_bytes + PAGE_CACHE_SIZE - 1) >>
					PAGE_CACHE_SHIFT;

1091
		WARN_ON(num_pages > nrptrs);
1092
		memset(pages, 0, sizeof(struct page *) * nrptrs);
1093 1094 1095 1096 1097

		ret = btrfs_check_free_space(root, write_bytes, 0);
		if (ret)
			goto out;

C
Chris Mason 已提交
1098 1099
		ret = prepare_pages(root, file, pages, num_pages,
				    pos, first_index, last_index,
1100
				    write_bytes);
1101 1102
		if (ret)
			goto out;
C
Chris Mason 已提交
1103 1104 1105

		ret = btrfs_copy_from_user(pos, num_pages,
					   write_bytes, pages, buf);
1106 1107 1108 1109
		if (ret) {
			btrfs_drop_pages(pages, num_pages);
			goto out;
		}
C
Chris Mason 已提交
1110 1111 1112 1113

		ret = dirty_and_release_pages(NULL, root, file, pages,
					      num_pages, pos, write_bytes);
		btrfs_drop_pages(pages, num_pages);
1114 1115
		if (ret)
			goto out;
C
Chris Mason 已提交
1116

1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129
		if (will_write) {
			btrfs_fdatawrite_range(inode->i_mapping, pos,
					       pos + write_bytes - 1,
					       WB_SYNC_NONE);
		} 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 已提交
1130 1131 1132 1133 1134 1135 1136 1137
		buf += write_bytes;
		count -= write_bytes;
		pos += write_bytes;
		num_written += write_bytes;

		cond_resched();
	}
out:
1138
	mutex_unlock(&inode->i_mutex);
1139

1140
out_nolock:
1141
	kfree(pages);
C
Chris Mason 已提交
1142 1143 1144 1145 1146
	if (pinned[0])
		page_cache_release(pinned[0]);
	if (pinned[1])
		page_cache_release(pinned[1]);
	*ppos = pos;
1147

1148
	if (num_written > 0 && will_write) {
1149 1150
		struct btrfs_trans_handle *trans;

1151 1152
		err = btrfs_wait_ordered_range(inode, start_pos, num_written);
		if (err)
1153
			num_written = err;
1154

1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169
		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) {
				btrfs_sync_log(trans, root);
				btrfs_end_transaction(trans, root);
			} 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);
1170
		}
1171
	}
C
Chris Mason 已提交
1172 1173 1174 1175
	current->backing_dev_info = NULL;
	return num_written ? num_written : err;
}

C
Chris Mason 已提交
1176
int btrfs_release_file(struct inode *inode, struct file *filp)
1177
{
S
Sage Weil 已提交
1178 1179
	if (filp->private_data)
		btrfs_ioctl_trans_end(filp);
1180 1181 1182
	return 0;
}

C
Chris Mason 已提交
1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193
/*
 * 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.
 */
1194
int btrfs_sync_file(struct file *file, struct dentry *dentry, int datasync)
C
Chris Mason 已提交
1195 1196 1197
{
	struct inode *inode = dentry->d_inode;
	struct btrfs_root *root = BTRFS_I(inode)->root;
1198
	int ret = 0;
C
Chris Mason 已提交
1199 1200 1201
	struct btrfs_trans_handle *trans;

	/*
1202 1203
	 * check the transaction that last modified this inode
	 * and see if its already been committed
C
Chris Mason 已提交
1204
	 */
1205 1206
	if (!BTRFS_I(inode)->last_trans)
		goto out;
1207

1208 1209 1210 1211 1212 1213 1214 1215 1216
	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);

C
Chris Mason 已提交
1217
	root->fs_info->tree_log_batch++;
1218 1219
	filemap_fdatawrite(inode->i_mapping);
	btrfs_wait_ordered_range(inode, 0, (u64)-1);
C
Chris Mason 已提交
1220
	root->fs_info->tree_log_batch++;
1221

1222
	/*
1223 1224
	 * ok we haven't committed the transaction yet, lets do a commit
	 */
S
Sage Weil 已提交
1225 1226 1227
	if (file->private_data)
		btrfs_ioctl_trans_end(file);

C
Chris Mason 已提交
1228 1229 1230 1231 1232
	trans = btrfs_start_transaction(root, 1);
	if (!trans) {
		ret = -ENOMEM;
		goto out;
	}
1233 1234

	ret = btrfs_log_dentry_safe(trans, root, file->f_dentry);
C
Chris Mason 已提交
1235
	if (ret < 0)
1236
		goto out;
C
Chris Mason 已提交
1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249

	/* 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.
	 */
	mutex_unlock(&file->f_dentry->d_inode->i_mutex);

1250 1251 1252 1253 1254 1255
	if (ret > 0) {
		ret = btrfs_commit_transaction(trans, root);
	} else {
		btrfs_sync_log(trans, root);
		ret = btrfs_end_transaction(trans, root);
	}
C
Chris Mason 已提交
1256
	mutex_lock(&file->f_dentry->d_inode->i_mutex);
C
Chris Mason 已提交
1257 1258 1259 1260
out:
	return ret > 0 ? EIO : ret;
}

C
Chris Mason 已提交
1261
static struct vm_operations_struct btrfs_file_vm_ops = {
1262
	.fault		= filemap_fault,
C
Chris Mason 已提交
1263 1264 1265 1266 1267 1268 1269 1270 1271 1272
	.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 已提交
1273 1274 1275
struct file_operations btrfs_file_operations = {
	.llseek		= generic_file_llseek,
	.read		= do_sync_read,
C
Chris Mason 已提交
1276
	.aio_read       = generic_file_aio_read,
C
Chris Mason 已提交
1277
	.splice_read	= generic_file_splice_read,
C
Chris Mason 已提交
1278
	.write		= btrfs_file_write,
C
Chris Mason 已提交
1279
	.mmap		= btrfs_file_mmap,
C
Chris Mason 已提交
1280
	.open		= generic_file_open,
1281
	.release	= btrfs_release_file,
C
Chris Mason 已提交
1282
	.fsync		= btrfs_sync_file,
C
Christoph Hellwig 已提交
1283
	.unlocked_ioctl	= btrfs_ioctl,
C
Chris Mason 已提交
1284
#ifdef CONFIG_COMPAT
C
Christoph Hellwig 已提交
1285
	.compat_ioctl	= btrfs_ioctl,
C
Chris Mason 已提交
1286 1287
#endif
};