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
	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);
833
		goto release;
Y
Yan Zheng 已提交
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
	} 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
done:
	btrfs_mark_buffer_dirty(leaf);
929 930

release:
Y
Yan Zheng 已提交
931 932 933 934 935 936 937 938 939 940 941 942 943
	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 已提交
944
/*
C
Chris Mason 已提交
945 946 947
 * 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 已提交
948
 */
C
Chris Mason 已提交
949
static noinline int prepare_pages(struct btrfs_root *root, struct file *file,
950 951 952
			 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 已提交
953 954 955
{
	int i;
	unsigned long index = pos >> PAGE_CACHE_SHIFT;
956
	struct inode *inode = fdentry(file)->d_inode;
C
Chris Mason 已提交
957
	int err = 0;
958
	u64 start_pos;
959
	u64 last_pos;
960

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

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

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

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

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

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

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

C
Chris Mason 已提交
1042
	pos = *ppos;
1043 1044
	start_pos = pos;

C
Chris Mason 已提交
1045 1046 1047 1048
	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)
1049
		goto out_nolock;
C
Chris Mason 已提交
1050
	if (count == 0)
1051
		goto out_nolock;
1052

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

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

	mutex_lock(&inode->i_mutex);
1061
	BTRFS_I(inode)->sequence++;
C
Chris Mason 已提交
1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072
	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 已提交
1073
			ret = btrfs_readpage(NULL, pinned[0]);
C
Chris Mason 已提交
1074 1075 1076 1077 1078 1079 1080 1081 1082
			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 已提交
1083
			ret = btrfs_readpage(NULL, pinned[1]);
C
Chris Mason 已提交
1084 1085 1086 1087 1088 1089 1090
			BUG_ON(ret);
			wait_on_page_locked(pinned[1]);
		} else {
			unlock_page(pinned[1]);
		}
	}

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

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

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

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

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

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

1133 1134 1135
		if (will_write) {
			btrfs_fdatawrite_range(inode->i_mapping, pos,
					       pos + write_bytes - 1,
1136
					       WB_SYNC_ALL);
1137 1138 1139 1140 1141 1142 1143 1144 1145
		} 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 已提交
1146 1147 1148 1149 1150 1151 1152 1153
		buf += write_bytes;
		count -= write_bytes;
		pos += write_bytes;
		num_written += write_bytes;

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

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

1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179
	/*
	 * 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;

1180
	if (num_written > 0 && will_write) {
1181 1182
		struct btrfs_trans_handle *trans;

1183 1184
		err = btrfs_wait_ordered_range(inode, start_pos, num_written);
		if (err)
1185
			num_written = err;
1186

1187 1188 1189 1190 1191
		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) {
1192 1193 1194 1195 1196
				ret = btrfs_sync_log(trans, root);
				if (ret == 0)
					btrfs_end_transaction(trans, root);
				else
					btrfs_commit_transaction(trans, root);
1197 1198 1199 1200 1201 1202 1203 1204
			} 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);
1205
		}
1206
	}
C
Chris Mason 已提交
1207 1208 1209 1210
	current->backing_dev_info = NULL;
	return num_written ? num_written : err;
}

C
Chris Mason 已提交
1211
int btrfs_release_file(struct inode *inode, struct file *filp)
1212
{
1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224
	/*
	 * 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 已提交
1225 1226
	if (filp->private_data)
		btrfs_ioctl_trans_end(filp);
1227 1228 1229
	return 0;
}

C
Chris Mason 已提交
1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240
/*
 * 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.
 */
1241
int btrfs_sync_file(struct file *file, struct dentry *dentry, int datasync)
C
Chris Mason 已提交
1242 1243 1244
{
	struct inode *inode = dentry->d_inode;
	struct btrfs_root *root = BTRFS_I(inode)->root;
1245
	int ret = 0;
C
Chris Mason 已提交
1246 1247 1248
	struct btrfs_trans_handle *trans;

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

1255 1256 1257 1258 1259 1260 1261 1262 1263
	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 已提交
1264
	root->log_batch++;
1265 1266
	filemap_fdatawrite(inode->i_mapping);
	btrfs_wait_ordered_range(inode, 0, (u64)-1);
Y
Yan Zheng 已提交
1267
	root->log_batch++;
1268

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

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

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

	/* 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.
	 */
1295
	mutex_unlock(&dentry->d_inode->i_mutex);
C
Chris Mason 已提交
1296

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

C
Chris Mason 已提交
1311
static struct vm_operations_struct btrfs_file_vm_ops = {
1312
	.fault		= filemap_fault,
C
Chris Mason 已提交
1313 1314 1315 1316 1317 1318 1319 1320 1321 1322
	.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 已提交
1323 1324 1325
struct file_operations btrfs_file_operations = {
	.llseek		= generic_file_llseek,
	.read		= do_sync_read,
C
Chris Mason 已提交
1326
	.aio_read       = generic_file_aio_read,
C
Chris Mason 已提交
1327
	.splice_read	= generic_file_splice_read,
C
Chris Mason 已提交
1328
	.write		= btrfs_file_write,
C
Chris Mason 已提交
1329
	.mmap		= btrfs_file_mmap,
C
Chris Mason 已提交
1330
	.open		= generic_file_open,
1331
	.release	= btrfs_release_file,
C
Chris Mason 已提交
1332
	.fsync		= btrfs_sync_file,
C
Christoph Hellwig 已提交
1333
	.unlocked_ioctl	= btrfs_ioctl,
C
Chris Mason 已提交
1334
#ifdef CONFIG_COMPAT
C
Christoph Hellwig 已提交
1335
	.compat_ioctl	= btrfs_ioctl,
C
Chris Mason 已提交
1336 1337
#endif
};