inode.c 143.5 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.
 */

19
#include <linux/kernel.h>
20
#include <linux/bio.h>
C
Chris Mason 已提交
21
#include <linux/buffer_head.h>
S
Sage Weil 已提交
22
#include <linux/file.h>
C
Chris Mason 已提交
23 24 25 26 27 28 29 30 31 32 33 34
#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/backing-dev.h>
#include <linux/mpage.h>
#include <linux/swap.h>
#include <linux/writeback.h>
#include <linux/statfs.h>
#include <linux/compat.h>
C
Chris Mason 已提交
35
#include <linux/bit_spinlock.h>
J
Josef Bacik 已提交
36
#include <linux/xattr.h>
J
Josef Bacik 已提交
37
#include <linux/posix_acl.h>
Y
Yan Zheng 已提交
38
#include <linux/falloc.h>
C
Chris Mason 已提交
39
#include "compat.h"
C
Chris Mason 已提交
40 41 42 43 44 45
#include "ctree.h"
#include "disk-io.h"
#include "transaction.h"
#include "btrfs_inode.h"
#include "ioctl.h"
#include "print-tree.h"
46
#include "volumes.h"
47
#include "ordered-data.h"
48
#include "xattr.h"
49
#include "tree-log.h"
C
Chris Mason 已提交
50
#include "compression.h"
51
#include "locking.h"
C
Chris Mason 已提交
52 53 54 55 56 57 58 59 60

struct btrfs_iget_args {
	u64 ino;
	struct btrfs_root *root;
};

static struct inode_operations btrfs_dir_inode_operations;
static struct inode_operations btrfs_symlink_inode_operations;
static struct inode_operations btrfs_dir_ro_inode_operations;
J
Josef Bacik 已提交
61
static struct inode_operations btrfs_special_inode_operations;
C
Chris Mason 已提交
62 63 64 65
static struct inode_operations btrfs_file_inode_operations;
static struct address_space_operations btrfs_aops;
static struct address_space_operations btrfs_symlink_aops;
static struct file_operations btrfs_dir_file_operations;
66
static struct extent_io_ops btrfs_extent_io_ops;
C
Chris Mason 已提交
67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83

static struct kmem_cache *btrfs_inode_cachep;
struct kmem_cache *btrfs_trans_handle_cachep;
struct kmem_cache *btrfs_transaction_cachep;
struct kmem_cache *btrfs_path_cachep;

#define S_SHIFT 12
static unsigned char btrfs_type_by_mode[S_IFMT >> S_SHIFT] = {
	[S_IFREG >> S_SHIFT]	= BTRFS_FT_REG_FILE,
	[S_IFDIR >> S_SHIFT]	= BTRFS_FT_DIR,
	[S_IFCHR >> S_SHIFT]	= BTRFS_FT_CHRDEV,
	[S_IFBLK >> S_SHIFT]	= BTRFS_FT_BLKDEV,
	[S_IFIFO >> S_SHIFT]	= BTRFS_FT_FIFO,
	[S_IFSOCK >> S_SHIFT]	= BTRFS_FT_SOCK,
	[S_IFLNK >> S_SHIFT]	= BTRFS_FT_SYMLINK,
};

84
static void btrfs_truncate(struct inode *inode);
C
Chris Mason 已提交
85
static int btrfs_finish_ordered_io(struct inode *inode, u64 start, u64 end);
86 87 88 89
static noinline int cow_file_range(struct inode *inode,
				   struct page *locked_page,
				   u64 start, u64 end, int *page_started,
				   unsigned long *nr_written, int unlock);
90

J
Jim Owens 已提交
91 92 93 94 95 96 97 98 99 100
static int btrfs_init_inode_security(struct inode *inode,  struct inode *dir)
{
	int err;

	err = btrfs_init_acl(inode, dir);
	if (!err)
		err = btrfs_xattr_security_init(inode, dir);
	return err;
}

C
Chris Mason 已提交
101 102 103 104 105
/*
 * this does all the hard work for inserting an inline extent into
 * the btree.  The caller should have done a btrfs_drop_extents so that
 * no overlapping inline items exist in the btree
 */
C
Chris Mason 已提交
106
static noinline int insert_inline_extent(struct btrfs_trans_handle *trans,
C
Chris Mason 已提交
107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129
				struct btrfs_root *root, struct inode *inode,
				u64 start, size_t size, size_t compressed_size,
				struct page **compressed_pages)
{
	struct btrfs_key key;
	struct btrfs_path *path;
	struct extent_buffer *leaf;
	struct page *page = NULL;
	char *kaddr;
	unsigned long ptr;
	struct btrfs_file_extent_item *ei;
	int err = 0;
	int ret;
	size_t cur_size = size;
	size_t datasize;
	unsigned long offset;
	int use_compress = 0;

	if (compressed_size && compressed_pages) {
		use_compress = 1;
		cur_size = compressed_size;
	}

C
Chris Mason 已提交
130 131
	path = btrfs_alloc_path();
	if (!path)
C
Chris Mason 已提交
132 133
		return -ENOMEM;

134
	path->leave_spinning = 1;
C
Chris Mason 已提交
135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162
	btrfs_set_trans_block_group(trans, inode);

	key.objectid = inode->i_ino;
	key.offset = start;
	btrfs_set_key_type(&key, BTRFS_EXTENT_DATA_KEY);
	datasize = btrfs_file_extent_calc_inline_size(cur_size);

	inode_add_bytes(inode, size);
	ret = btrfs_insert_empty_item(trans, root, path, &key,
				      datasize);
	BUG_ON(ret);
	if (ret) {
		err = ret;
		goto fail;
	}
	leaf = path->nodes[0];
	ei = btrfs_item_ptr(leaf, path->slots[0],
			    struct btrfs_file_extent_item);
	btrfs_set_file_extent_generation(leaf, ei, trans->transid);
	btrfs_set_file_extent_type(leaf, ei, BTRFS_FILE_EXTENT_INLINE);
	btrfs_set_file_extent_encryption(leaf, ei, 0);
	btrfs_set_file_extent_other_encoding(leaf, ei, 0);
	btrfs_set_file_extent_ram_bytes(leaf, ei, size);
	ptr = btrfs_file_extent_inline_start(ei);

	if (use_compress) {
		struct page *cpage;
		int i = 0;
C
Chris Mason 已提交
163
		while (compressed_size > 0) {
C
Chris Mason 已提交
164
			cpage = compressed_pages[i];
165
			cur_size = min_t(unsigned long, compressed_size,
C
Chris Mason 已提交
166 167
				       PAGE_CACHE_SIZE);

168
			kaddr = kmap_atomic(cpage, KM_USER0);
C
Chris Mason 已提交
169
			write_extent_buffer(leaf, kaddr, ptr, cur_size);
170
			kunmap_atomic(kaddr, KM_USER0);
C
Chris Mason 已提交
171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204

			i++;
			ptr += cur_size;
			compressed_size -= cur_size;
		}
		btrfs_set_file_extent_compression(leaf, ei,
						  BTRFS_COMPRESS_ZLIB);
	} else {
		page = find_get_page(inode->i_mapping,
				     start >> PAGE_CACHE_SHIFT);
		btrfs_set_file_extent_compression(leaf, ei, 0);
		kaddr = kmap_atomic(page, KM_USER0);
		offset = start & (PAGE_CACHE_SIZE - 1);
		write_extent_buffer(leaf, kaddr + offset, ptr, size);
		kunmap_atomic(kaddr, KM_USER0);
		page_cache_release(page);
	}
	btrfs_mark_buffer_dirty(leaf);
	btrfs_free_path(path);

	BTRFS_I(inode)->disk_i_size = inode->i_size;
	btrfs_update_inode(trans, root, inode);
	return 0;
fail:
	btrfs_free_path(path);
	return err;
}


/*
 * conditionally insert an inline extent into the file.  This
 * does the checks required to make sure the data is small enough
 * to fit as an inline extent.
 */
205
static noinline int cow_file_range_inline(struct btrfs_trans_handle *trans,
C
Chris Mason 已提交
206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223
				 struct btrfs_root *root,
				 struct inode *inode, u64 start, u64 end,
				 size_t compressed_size,
				 struct page **compressed_pages)
{
	u64 isize = i_size_read(inode);
	u64 actual_end = min(end + 1, isize);
	u64 inline_len = actual_end - start;
	u64 aligned_end = (end + root->sectorsize - 1) &
			~((u64)root->sectorsize - 1);
	u64 hint_byte;
	u64 data_len = inline_len;
	int ret;

	if (compressed_size)
		data_len = compressed_size;

	if (start > 0 ||
C
Chris Mason 已提交
224
	    actual_end >= PAGE_CACHE_SIZE ||
C
Chris Mason 已提交
225 226 227 228 229 230 231 232 233
	    data_len >= BTRFS_MAX_INLINE_DATA_SIZE(root) ||
	    (!compressed_size &&
	    (actual_end & (root->sectorsize - 1)) == 0) ||
	    end + 1 < isize ||
	    data_len > root->fs_info->max_inline) {
		return 1;
	}

	ret = btrfs_drop_extents(trans, root, inode, start,
C
Chris Mason 已提交
234 235
				 aligned_end, aligned_end, start,
				 &hint_byte, 1);
C
Chris Mason 已提交
236 237 238 239 240 241 242 243
	BUG_ON(ret);

	if (isize > actual_end)
		inline_len = min_t(u64, isize, actual_end);
	ret = insert_inline_extent(trans, root, inode, start,
				   inline_len, compressed_size,
				   compressed_pages);
	BUG_ON(ret);
C
Chris Mason 已提交
244
	btrfs_drop_extent_cache(inode, start, aligned_end - 1, 0);
C
Chris Mason 已提交
245 246 247
	return 0;
}

248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284
struct async_extent {
	u64 start;
	u64 ram_size;
	u64 compressed_size;
	struct page **pages;
	unsigned long nr_pages;
	struct list_head list;
};

struct async_cow {
	struct inode *inode;
	struct btrfs_root *root;
	struct page *locked_page;
	u64 start;
	u64 end;
	struct list_head extents;
	struct btrfs_work work;
};

static noinline int add_async_extent(struct async_cow *cow,
				     u64 start, u64 ram_size,
				     u64 compressed_size,
				     struct page **pages,
				     unsigned long nr_pages)
{
	struct async_extent *async_extent;

	async_extent = kmalloc(sizeof(*async_extent), GFP_NOFS);
	async_extent->start = start;
	async_extent->ram_size = ram_size;
	async_extent->compressed_size = compressed_size;
	async_extent->pages = pages;
	async_extent->nr_pages = nr_pages;
	list_add_tail(&async_extent->list, &cow->extents);
	return 0;
}

C
Chris Mason 已提交
285
/*
286 287 288
 * we create compressed extents in two phases.  The first
 * phase compresses a range of pages that have already been
 * locked (both pages and state bits are locked).
C
Chris Mason 已提交
289
 *
290 291 292 293 294
 * This is done inside an ordered work queue, and the compression
 * is spread across many cpus.  The actual IO submission is step
 * two, and the ordered work queue takes care of making sure that
 * happens in the same order things were put onto the queue by
 * writepages and friends.
C
Chris Mason 已提交
295
 *
296 297 298 299
 * If this code finds it can't get good compression, it puts an
 * entry onto the work queue to write the uncompressed bytes.  This
 * makes sure that both compressed inodes and uncompressed inodes
 * are written in the same order that pdflush sent them down.
C
Chris Mason 已提交
300
 */
301 302 303 304 305
static noinline int compress_file_range(struct inode *inode,
					struct page *locked_page,
					u64 start, u64 end,
					struct async_cow *async_cow,
					int *num_added)
306 307 308
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct btrfs_trans_handle *trans;
309
	u64 num_bytes;
C
Chris Mason 已提交
310 311
	u64 orig_start;
	u64 disk_num_bytes;
312
	u64 blocksize = root->sectorsize;
C
Chris Mason 已提交
313
	u64 actual_end;
314
	u64 isize = i_size_read(inode);
315
	int ret = 0;
C
Chris Mason 已提交
316 317 318 319 320 321
	struct page **pages = NULL;
	unsigned long nr_pages;
	unsigned long nr_pages_ret = 0;
	unsigned long total_compressed = 0;
	unsigned long total_in = 0;
	unsigned long max_compressed = 128 * 1024;
322
	unsigned long max_uncompressed = 128 * 1024;
C
Chris Mason 已提交
323 324
	int i;
	int will_compress;
325

C
Chris Mason 已提交
326 327
	orig_start = start;

328
	actual_end = min_t(u64, isize, end + 1);
C
Chris Mason 已提交
329 330 331 332
again:
	will_compress = 0;
	nr_pages = (end >> PAGE_CACHE_SHIFT) - (start >> PAGE_CACHE_SHIFT) + 1;
	nr_pages = min(nr_pages, (128 * 1024UL) / PAGE_CACHE_SIZE);
333

334 335 336 337 338 339 340 341 342 343 344 345 346
	/*
	 * we don't want to send crud past the end of i_size through
	 * compression, that's just a waste of CPU time.  So, if the
	 * end of the file is before the start of our current
	 * requested range of bytes, we bail out to the uncompressed
	 * cleanup code that can deal with all of this.
	 *
	 * It isn't really the fastest way to fix things, but this is a
	 * very uncommon corner.
	 */
	if (actual_end <= start)
		goto cleanup_and_bail_uncompressed;

C
Chris Mason 已提交
347 348 349 350
	total_compressed = actual_end - start;

	/* we want to make sure that amount of ram required to uncompress
	 * an extent is reasonable, so we limit the total size in ram
351 352 353 354 355 356 357
	 * of a compressed extent to 128k.  This is a crucial number
	 * because it also controls how easily we can spread reads across
	 * cpus for decompression.
	 *
	 * We also want to make sure the amount of IO required to do
	 * a random read is reasonably small, so we limit the size of
	 * a compressed extent to 128k.
C
Chris Mason 已提交
358 359
	 */
	total_compressed = min(total_compressed, max_uncompressed);
360
	num_bytes = (end - start + blocksize) & ~(blocksize - 1);
361
	num_bytes = max(blocksize,  num_bytes);
C
Chris Mason 已提交
362 363 364
	disk_num_bytes = num_bytes;
	total_in = 0;
	ret = 0;
365

366 367 368 369
	/*
	 * we do compression for mount -o compress and when the
	 * inode has not been flagged as nocompress.  This flag can
	 * change at any time if we discover bad compression ratios.
C
Chris Mason 已提交
370
	 */
371
	if (!(BTRFS_I(inode)->flags & BTRFS_INODE_NOCOMPRESS) &&
C
Chris Mason 已提交
372 373
	    btrfs_test_opt(root, COMPRESS)) {
		WARN_ON(pages);
374
		pages = kzalloc(sizeof(struct page *) * nr_pages, GFP_NOFS);
C
Chris Mason 已提交
375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401

		ret = btrfs_zlib_compress_pages(inode->i_mapping, start,
						total_compressed, pages,
						nr_pages, &nr_pages_ret,
						&total_in,
						&total_compressed,
						max_compressed);

		if (!ret) {
			unsigned long offset = total_compressed &
				(PAGE_CACHE_SIZE - 1);
			struct page *page = pages[nr_pages_ret - 1];
			char *kaddr;

			/* zero the tail end of the last page, we might be
			 * sending it down to disk
			 */
			if (offset) {
				kaddr = kmap_atomic(page, KM_USER0);
				memset(kaddr + offset, 0,
				       PAGE_CACHE_SIZE - offset);
				kunmap_atomic(kaddr, KM_USER0);
			}
			will_compress = 1;
		}
	}
	if (start == 0) {
402 403 404 405
		trans = btrfs_join_transaction(root, 1);
		BUG_ON(!trans);
		btrfs_set_trans_block_group(trans, inode);

C
Chris Mason 已提交
406
		/* lets try to make an inline extent */
407
		if (ret || total_in < (actual_end - start)) {
C
Chris Mason 已提交
408
			/* we didn't compress the entire range, try
409
			 * to make an uncompressed inline extent.
C
Chris Mason 已提交
410 411 412 413
			 */
			ret = cow_file_range_inline(trans, root, inode,
						    start, end, 0, NULL);
		} else {
414
			/* try making a compressed inline extent */
C
Chris Mason 已提交
415 416 417 418
			ret = cow_file_range_inline(trans, root, inode,
						    start, end,
						    total_compressed, pages);
		}
419
		btrfs_end_transaction(trans, root);
C
Chris Mason 已提交
420
		if (ret == 0) {
421 422 423 424 425
			/*
			 * inline extent creation worked, we don't need
			 * to create any more async work items.  Unlock
			 * and free up our temp pages.
			 */
C
Chris Mason 已提交
426 427
			extent_clear_unlock_delalloc(inode,
						     &BTRFS_I(inode)->io_tree,
428
						     start, end, NULL, 1, 0,
429
						     0, 1, 1, 1, 0);
C
Chris Mason 已提交
430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462
			ret = 0;
			goto free_pages_out;
		}
	}

	if (will_compress) {
		/*
		 * we aren't doing an inline extent round the compressed size
		 * up to a block size boundary so the allocator does sane
		 * things
		 */
		total_compressed = (total_compressed + blocksize - 1) &
			~(blocksize - 1);

		/*
		 * one last check to make sure the compression is really a
		 * win, compare the page count read with the blocks on disk
		 */
		total_in = (total_in + PAGE_CACHE_SIZE - 1) &
			~(PAGE_CACHE_SIZE - 1);
		if (total_compressed >= total_in) {
			will_compress = 0;
		} else {
			disk_num_bytes = total_compressed;
			num_bytes = total_in;
		}
	}
	if (!will_compress && pages) {
		/*
		 * the compression code ran but failed to make things smaller,
		 * free any pages it allocated and our page pointer array
		 */
		for (i = 0; i < nr_pages_ret; i++) {
C
Chris Mason 已提交
463
			WARN_ON(pages[i]->mapping);
C
Chris Mason 已提交
464 465 466 467 468 469 470 471
			page_cache_release(pages[i]);
		}
		kfree(pages);
		pages = NULL;
		total_compressed = 0;
		nr_pages_ret = 0;

		/* flag the file so we don't compress in the future */
472
		BTRFS_I(inode)->flags |= BTRFS_INODE_NOCOMPRESS;
C
Chris Mason 已提交
473
	}
474 475
	if (will_compress) {
		*num_added += 1;
C
Chris Mason 已提交
476

477 478 479 480 481 482
		/* the async work queues will take care of doing actual
		 * allocation on disk for these compressed pages,
		 * and will submit them to the elevator.
		 */
		add_async_extent(async_cow, start, num_bytes,
				 total_compressed, pages, nr_pages_ret);
483

484
		if (start + num_bytes < end && start + num_bytes < actual_end) {
485 486 487 488 489 490
			start += num_bytes;
			pages = NULL;
			cond_resched();
			goto again;
		}
	} else {
491
cleanup_and_bail_uncompressed:
492 493 494 495 496 497 498 499 500 501 502 503 504 505 506
		/*
		 * No compression, but we still need to write the pages in
		 * the file we've been given so far.  redirty the locked
		 * page if it corresponds to our extent and set things up
		 * for the async work queue to run cow_file_range to do
		 * the normal delalloc dance
		 */
		if (page_offset(locked_page) >= start &&
		    page_offset(locked_page) <= end) {
			__set_page_dirty_nobuffers(locked_page);
			/* unlocked later on in the async handlers */
		}
		add_async_extent(async_cow, start, end - start + 1, 0, NULL, 0);
		*num_added += 1;
	}
507

508 509 510 511 512 513 514 515
out:
	return 0;

free_pages_out:
	for (i = 0; i < nr_pages_ret; i++) {
		WARN_ON(pages[i]->mapping);
		page_cache_release(pages[i]);
	}
C
Chris Mason 已提交
516
	kfree(pages);
517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544

	goto out;
}

/*
 * phase two of compressed writeback.  This is the ordered portion
 * of the code, which only gets called in the order the work was
 * queued.  We walk all the async extents created by compress_file_range
 * and send them down to the disk.
 */
static noinline int submit_compressed_extents(struct inode *inode,
					      struct async_cow *async_cow)
{
	struct async_extent *async_extent;
	u64 alloc_hint = 0;
	struct btrfs_trans_handle *trans;
	struct btrfs_key ins;
	struct extent_map *em;
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree;
	struct extent_io_tree *io_tree;
	int ret;

	if (list_empty(&async_cow->extents))
		return 0;

	trans = btrfs_join_transaction(root, 1);

C
Chris Mason 已提交
545
	while (!list_empty(&async_cow->extents)) {
546 547 548
		async_extent = list_entry(async_cow->extents.next,
					  struct async_extent, list);
		list_del(&async_extent->list);
C
Chris Mason 已提交
549

550 551 552 553 554 555 556 557
		io_tree = &BTRFS_I(inode)->io_tree;

		/* did the compression code fall back to uncompressed IO? */
		if (!async_extent->pages) {
			int page_started = 0;
			unsigned long nr_written = 0;

			lock_extent(io_tree, async_extent->start,
C
Chris Mason 已提交
558 559
				    async_extent->start +
				    async_extent->ram_size - 1, GFP_NOFS);
560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576

			/* allocate blocks */
			cow_file_range(inode, async_cow->locked_page,
				       async_extent->start,
				       async_extent->start +
				       async_extent->ram_size - 1,
				       &page_started, &nr_written, 0);

			/*
			 * if page_started, cow_file_range inserted an
			 * inline extent and took care of all the unlocking
			 * and IO for us.  Otherwise, we need to submit
			 * all those pages down to the drive.
			 */
			if (!page_started)
				extent_write_locked_range(io_tree,
						  inode, async_extent->start,
C
Chris Mason 已提交
577
						  async_extent->start +
578 579 580 581 582 583 584 585 586 587 588
						  async_extent->ram_size - 1,
						  btrfs_get_extent,
						  WB_SYNC_ALL);
			kfree(async_extent);
			cond_resched();
			continue;
		}

		lock_extent(io_tree, async_extent->start,
			    async_extent->start + async_extent->ram_size - 1,
			    GFP_NOFS);
C
Chris Mason 已提交
589
		/*
590 591
		 * here we're doing allocation and writeback of the
		 * compressed pages
C
Chris Mason 已提交
592
		 */
593 594 595 596 597 598 599 600 601 602 603 604 605
		btrfs_drop_extent_cache(inode, async_extent->start,
					async_extent->start +
					async_extent->ram_size - 1, 0);

		ret = btrfs_reserve_extent(trans, root,
					   async_extent->compressed_size,
					   async_extent->compressed_size,
					   0, alloc_hint,
					   (u64)-1, &ins, 1);
		BUG_ON(ret);
		em = alloc_extent_map(GFP_NOFS);
		em->start = async_extent->start;
		em->len = async_extent->ram_size;
606
		em->orig_start = em->start;
C
Chris Mason 已提交
607

608 609 610 611 612 613
		em->block_start = ins.objectid;
		em->block_len = ins.offset;
		em->bdev = root->fs_info->fs_devices->latest_bdev;
		set_bit(EXTENT_FLAG_PINNED, &em->flags);
		set_bit(EXTENT_FLAG_COMPRESSED, &em->flags);

C
Chris Mason 已提交
614
		while (1) {
615
			write_lock(&em_tree->lock);
616
			ret = add_extent_mapping(em_tree, em);
617
			write_unlock(&em_tree->lock);
618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643
			if (ret != -EEXIST) {
				free_extent_map(em);
				break;
			}
			btrfs_drop_extent_cache(inode, async_extent->start,
						async_extent->start +
						async_extent->ram_size - 1, 0);
		}

		ret = btrfs_add_ordered_extent(inode, async_extent->start,
					       ins.objectid,
					       async_extent->ram_size,
					       ins.offset,
					       BTRFS_ORDERED_COMPRESSED);
		BUG_ON(ret);

		btrfs_end_transaction(trans, root);

		/*
		 * clear dirty, set writeback and unlock the pages.
		 */
		extent_clear_unlock_delalloc(inode,
					     &BTRFS_I(inode)->io_tree,
					     async_extent->start,
					     async_extent->start +
					     async_extent->ram_size - 1,
644
					     NULL, 1, 1, 0, 1, 1, 0, 0);
645 646

		ret = btrfs_submit_compressed_write(inode,
C
Chris Mason 已提交
647 648 649 650 651
				    async_extent->start,
				    async_extent->ram_size,
				    ins.objectid,
				    ins.offset, async_extent->pages,
				    async_extent->nr_pages);
652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691

		BUG_ON(ret);
		trans = btrfs_join_transaction(root, 1);
		alloc_hint = ins.objectid + ins.offset;
		kfree(async_extent);
		cond_resched();
	}

	btrfs_end_transaction(trans, root);
	return 0;
}

/*
 * when extent_io.c finds a delayed allocation range in the file,
 * the call backs end up in this code.  The basic idea is to
 * allocate extents on disk for the range, and create ordered data structs
 * in ram to track those extents.
 *
 * locked_page is the page that writepage had locked already.  We use
 * it to make sure we don't do extra locks or unlocks.
 *
 * *page_started is set to one if we unlock locked_page and do everything
 * required to start IO on it.  It may be clean and already done with
 * IO when we return.
 */
static noinline int cow_file_range(struct inode *inode,
				   struct page *locked_page,
				   u64 start, u64 end, int *page_started,
				   unsigned long *nr_written,
				   int unlock)
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct btrfs_trans_handle *trans;
	u64 alloc_hint = 0;
	u64 num_bytes;
	unsigned long ram_size;
	u64 disk_num_bytes;
	u64 cur_alloc_size;
	u64 blocksize = root->sectorsize;
	u64 actual_end;
692
	u64 isize = i_size_read(inode);
693 694 695 696 697 698 699 700 701
	struct btrfs_key ins;
	struct extent_map *em;
	struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree;
	int ret = 0;

	trans = btrfs_join_transaction(root, 1);
	BUG_ON(!trans);
	btrfs_set_trans_block_group(trans, inode);

702
	actual_end = min_t(u64, isize, end + 1);
703 704 705 706 707 708 709 710 711 712 713 714 715 716

	num_bytes = (end - start + blocksize) & ~(blocksize - 1);
	num_bytes = max(blocksize,  num_bytes);
	disk_num_bytes = num_bytes;
	ret = 0;

	if (start == 0) {
		/* lets try to make an inline extent */
		ret = cow_file_range_inline(trans, root, inode,
					    start, end, 0, NULL);
		if (ret == 0) {
			extent_clear_unlock_delalloc(inode,
						     &BTRFS_I(inode)->io_tree,
						     start, end, NULL, 1, 1,
717
						     1, 1, 1, 1, 0);
718 719 720 721 722 723 724 725 726 727 728 729 730
			*nr_written = *nr_written +
			     (end - start + PAGE_CACHE_SIZE) / PAGE_CACHE_SIZE;
			*page_started = 1;
			ret = 0;
			goto out;
		}
	}

	BUG_ON(disk_num_bytes >
	       btrfs_super_total_bytes(&root->fs_info->super_copy));

	btrfs_drop_extent_cache(inode, start, start + num_bytes - 1, 0);

C
Chris Mason 已提交
731
	while (disk_num_bytes > 0) {
C
Chris Mason 已提交
732
		cur_alloc_size = min(disk_num_bytes, root->fs_info->max_extent);
733
		ret = btrfs_reserve_extent(trans, root, cur_alloc_size,
734
					   root->sectorsize, 0, alloc_hint,
735
					   (u64)-1, &ins, 1);
C
Chris Mason 已提交
736 737
		BUG_ON(ret);

738 739
		em = alloc_extent_map(GFP_NOFS);
		em->start = start;
740
		em->orig_start = em->start;
C
Chris Mason 已提交
741

742 743
		ram_size = ins.offset;
		em->len = ins.offset;
C
Chris Mason 已提交
744

745
		em->block_start = ins.objectid;
C
Chris Mason 已提交
746
		em->block_len = ins.offset;
747
		em->bdev = root->fs_info->fs_devices->latest_bdev;
748
		set_bit(EXTENT_FLAG_PINNED, &em->flags);
C
Chris Mason 已提交
749

C
Chris Mason 已提交
750
		while (1) {
751
			write_lock(&em_tree->lock);
752
			ret = add_extent_mapping(em_tree, em);
753
			write_unlock(&em_tree->lock);
754 755 756 757 758
			if (ret != -EEXIST) {
				free_extent_map(em);
				break;
			}
			btrfs_drop_extent_cache(inode, start,
C
Chris Mason 已提交
759
						start + ram_size - 1, 0);
760 761
		}

762
		cur_alloc_size = ins.offset;
763
		ret = btrfs_add_ordered_extent(inode, start, ins.objectid,
764
					       ram_size, cur_alloc_size, 0);
765
		BUG_ON(ret);
C
Chris Mason 已提交
766

767 768 769 770 771 772 773
		if (root->root_key.objectid ==
		    BTRFS_DATA_RELOC_TREE_OBJECTID) {
			ret = btrfs_reloc_clone_csums(inode, start,
						      cur_alloc_size);
			BUG_ON(ret);
		}

C
Chris Mason 已提交
774
		if (disk_num_bytes < cur_alloc_size)
775
			break;
C
Chris Mason 已提交
776

C
Chris Mason 已提交
777 778 779
		/* we're not doing compressed IO, don't unlock the first
		 * page (which the caller expects to stay locked), don't
		 * clear any dirty bits and don't set any writeback bits
780 781 782
		 *
		 * Do set the Private2 bit so we know this page was properly
		 * setup for writepage
C
Chris Mason 已提交
783 784 785
		 */
		extent_clear_unlock_delalloc(inode, &BTRFS_I(inode)->io_tree,
					     start, start + ram_size - 1,
786
					     locked_page, unlock, 1,
787
					     1, 0, 0, 0, 1);
C
Chris Mason 已提交
788
		disk_num_bytes -= cur_alloc_size;
789 790 791
		num_bytes -= cur_alloc_size;
		alloc_hint = ins.objectid + ins.offset;
		start += cur_alloc_size;
792 793
	}
out:
794
	ret = 0;
795
	btrfs_end_transaction(trans, root);
C
Chris Mason 已提交
796

797
	return ret;
798
}
C
Chris Mason 已提交
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
/*
 * work queue call back to started compression on a file and pages
 */
static noinline void async_cow_start(struct btrfs_work *work)
{
	struct async_cow *async_cow;
	int num_added = 0;
	async_cow = container_of(work, struct async_cow, work);

	compress_file_range(async_cow->inode, async_cow->locked_page,
			    async_cow->start, async_cow->end, async_cow,
			    &num_added);
	if (num_added == 0)
		async_cow->inode = NULL;
}

/*
 * work queue call back to submit previously compressed pages
 */
static noinline void async_cow_submit(struct btrfs_work *work)
{
	struct async_cow *async_cow;
	struct btrfs_root *root;
	unsigned long nr_pages;

	async_cow = container_of(work, struct async_cow, work);

	root = async_cow->root;
	nr_pages = (async_cow->end - async_cow->start + PAGE_CACHE_SIZE) >>
		PAGE_CACHE_SHIFT;

	atomic_sub(nr_pages, &root->fs_info->async_delalloc_pages);

	if (atomic_read(&root->fs_info->async_delalloc_pages) <
	    5 * 1042 * 1024 &&
	    waitqueue_active(&root->fs_info->async_submit_wait))
		wake_up(&root->fs_info->async_submit_wait);

C
Chris Mason 已提交
838
	if (async_cow->inode)
839 840
		submit_compressed_extents(async_cow->inode, async_cow);
}
C
Chris Mason 已提交
841

842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859
static noinline void async_cow_free(struct btrfs_work *work)
{
	struct async_cow *async_cow;
	async_cow = container_of(work, struct async_cow, work);
	kfree(async_cow);
}

static int cow_file_range_async(struct inode *inode, struct page *locked_page,
				u64 start, u64 end, int *page_started,
				unsigned long *nr_written)
{
	struct async_cow *async_cow;
	struct btrfs_root *root = BTRFS_I(inode)->root;
	unsigned long nr_pages;
	u64 cur_end;
	int limit = 10 * 1024 * 1042;

	clear_extent_bit(&BTRFS_I(inode)->io_tree, start, end, EXTENT_LOCKED |
860
			 EXTENT_DELALLOC, 1, 0, NULL, GFP_NOFS);
C
Chris Mason 已提交
861
	while (start < end) {
862 863 864 865 866 867
		async_cow = kmalloc(sizeof(*async_cow), GFP_NOFS);
		async_cow->inode = inode;
		async_cow->root = root;
		async_cow->locked_page = locked_page;
		async_cow->start = start;

868
		if (BTRFS_I(inode)->flags & BTRFS_INODE_NOCOMPRESS)
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
			cur_end = end;
		else
			cur_end = min(end, start + 512 * 1024 - 1);

		async_cow->end = cur_end;
		INIT_LIST_HEAD(&async_cow->extents);

		async_cow->work.func = async_cow_start;
		async_cow->work.ordered_func = async_cow_submit;
		async_cow->work.ordered_free = async_cow_free;
		async_cow->work.flags = 0;

		nr_pages = (cur_end - start + PAGE_CACHE_SIZE) >>
			PAGE_CACHE_SHIFT;
		atomic_add(nr_pages, &root->fs_info->async_delalloc_pages);

		btrfs_queue_worker(&root->fs_info->delalloc_workers,
				   &async_cow->work);

		if (atomic_read(&root->fs_info->async_delalloc_pages) > limit) {
			wait_event(root->fs_info->async_submit_wait,
			   (atomic_read(&root->fs_info->async_delalloc_pages) <
			    limit));
		}

C
Chris Mason 已提交
894
		while (atomic_read(&root->fs_info->async_submit_draining) &&
895 896 897 898 899 900 901 902 903 904 905
		      atomic_read(&root->fs_info->async_delalloc_pages)) {
			wait_event(root->fs_info->async_submit_wait,
			  (atomic_read(&root->fs_info->async_delalloc_pages) ==
			   0));
		}

		*nr_written += nr_pages;
		start = cur_end + 1;
	}
	*page_started = 1;
	return 0;
906 907
}

C
Chris Mason 已提交
908
static noinline int csum_exist_in_range(struct btrfs_root *root,
909 910 911 912 913 914
					u64 bytenr, u64 num_bytes)
{
	int ret;
	struct btrfs_ordered_sum *sums;
	LIST_HEAD(list);

Y
Yan Zheng 已提交
915 916
	ret = btrfs_lookup_csums_range(root->fs_info->csum_root, bytenr,
				       bytenr + num_bytes - 1, &list);
917 918 919 920 921 922 923 924 925 926 927
	if (ret == 0 && list_empty(&list))
		return 0;

	while (!list_empty(&list)) {
		sums = list_entry(list.next, struct btrfs_ordered_sum, list);
		list_del(&sums->list);
		kfree(sums);
	}
	return 1;
}

C
Chris Mason 已提交
928 929 930 931 932 933 934
/*
 * when nowcow writeback call back.  This checks for snapshots or COW copies
 * of the extents that exist in the file, and COWs the file as required.
 *
 * If no cow copies or snapshots exist, we write directly to the existing
 * blocks on disk
 */
935 936
static noinline int run_delalloc_nocow(struct inode *inode,
				       struct page *locked_page,
937 938
			      u64 start, u64 end, int *page_started, int force,
			      unsigned long *nr_written)
939 940
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
941
	struct btrfs_trans_handle *trans;
942 943
	struct extent_buffer *leaf;
	struct btrfs_path *path;
Y
Yan Zheng 已提交
944
	struct btrfs_file_extent_item *fi;
945
	struct btrfs_key found_key;
Y
Yan Zheng 已提交
946 947 948
	u64 cow_start;
	u64 cur_offset;
	u64 extent_end;
949
	u64 extent_offset;
Y
Yan Zheng 已提交
950 951 952 953
	u64 disk_bytenr;
	u64 num_bytes;
	int extent_type;
	int ret;
Y
Yan Zheng 已提交
954
	int type;
Y
Yan Zheng 已提交
955 956
	int nocow;
	int check_prev = 1;
957 958 959

	path = btrfs_alloc_path();
	BUG_ON(!path);
960 961
	trans = btrfs_join_transaction(root, 1);
	BUG_ON(!trans);
962

Y
Yan Zheng 已提交
963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987
	cow_start = (u64)-1;
	cur_offset = start;
	while (1) {
		ret = btrfs_lookup_file_extent(trans, root, path, inode->i_ino,
					       cur_offset, 0);
		BUG_ON(ret < 0);
		if (ret > 0 && path->slots[0] > 0 && check_prev) {
			leaf = path->nodes[0];
			btrfs_item_key_to_cpu(leaf, &found_key,
					      path->slots[0] - 1);
			if (found_key.objectid == inode->i_ino &&
			    found_key.type == BTRFS_EXTENT_DATA_KEY)
				path->slots[0]--;
		}
		check_prev = 0;
next_slot:
		leaf = path->nodes[0];
		if (path->slots[0] >= btrfs_header_nritems(leaf)) {
			ret = btrfs_next_leaf(root, path);
			if (ret < 0)
				BUG_ON(1);
			if (ret > 0)
				break;
			leaf = path->nodes[0];
		}
988

Y
Yan Zheng 已提交
989 990
		nocow = 0;
		disk_bytenr = 0;
991
		num_bytes = 0;
Y
Yan Zheng 已提交
992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007
		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);

		if (found_key.objectid > inode->i_ino ||
		    found_key.type > BTRFS_EXTENT_DATA_KEY ||
		    found_key.offset > end)
			break;

		if (found_key.offset > cur_offset) {
			extent_end = found_key.offset;
			goto out_check;
		}

		fi = btrfs_item_ptr(leaf, path->slots[0],
				    struct btrfs_file_extent_item);
		extent_type = btrfs_file_extent_type(leaf, fi);

Y
Yan Zheng 已提交
1008 1009
		if (extent_type == BTRFS_FILE_EXTENT_REG ||
		    extent_type == BTRFS_FILE_EXTENT_PREALLOC) {
Y
Yan Zheng 已提交
1010
			disk_bytenr = btrfs_file_extent_disk_bytenr(leaf, fi);
1011
			extent_offset = btrfs_file_extent_offset(leaf, fi);
Y
Yan Zheng 已提交
1012 1013 1014 1015 1016 1017
			extent_end = found_key.offset +
				btrfs_file_extent_num_bytes(leaf, fi);
			if (extent_end <= start) {
				path->slots[0]++;
				goto next_slot;
			}
1018 1019
			if (disk_bytenr == 0)
				goto out_check;
Y
Yan Zheng 已提交
1020 1021 1022 1023
			if (btrfs_file_extent_compression(leaf, fi) ||
			    btrfs_file_extent_encryption(leaf, fi) ||
			    btrfs_file_extent_other_encoding(leaf, fi))
				goto out_check;
Y
Yan Zheng 已提交
1024 1025
			if (extent_type == BTRFS_FILE_EXTENT_REG && !force)
				goto out_check;
1026
			if (btrfs_extent_readonly(root, disk_bytenr))
Y
Yan Zheng 已提交
1027
				goto out_check;
1028
			if (btrfs_cross_ref_exist(trans, root, inode->i_ino,
1029 1030
						  found_key.offset -
						  extent_offset, disk_bytenr))
1031
				goto out_check;
1032
			disk_bytenr += extent_offset;
1033 1034 1035 1036 1037 1038 1039 1040 1041
			disk_bytenr += cur_offset - found_key.offset;
			num_bytes = min(end + 1, extent_end) - cur_offset;
			/*
			 * force cow if csum exists in the range.
			 * this ensure that csum for a given extent are
			 * either valid or do not exist.
			 */
			if (csum_exist_in_range(root, disk_bytenr, num_bytes))
				goto out_check;
Y
Yan Zheng 已提交
1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062
			nocow = 1;
		} else if (extent_type == BTRFS_FILE_EXTENT_INLINE) {
			extent_end = found_key.offset +
				btrfs_file_extent_inline_len(leaf, fi);
			extent_end = ALIGN(extent_end, root->sectorsize);
		} else {
			BUG_ON(1);
		}
out_check:
		if (extent_end <= start) {
			path->slots[0]++;
			goto next_slot;
		}
		if (!nocow) {
			if (cow_start == (u64)-1)
				cow_start = cur_offset;
			cur_offset = extent_end;
			if (cur_offset > end)
				break;
			path->slots[0]++;
			goto next_slot;
1063 1064 1065
		}

		btrfs_release_path(root, path);
Y
Yan Zheng 已提交
1066 1067
		if (cow_start != (u64)-1) {
			ret = cow_file_range(inode, locked_page, cow_start,
1068 1069
					found_key.offset - 1, page_started,
					nr_written, 1);
Y
Yan Zheng 已提交
1070 1071
			BUG_ON(ret);
			cow_start = (u64)-1;
1072
		}
Y
Yan Zheng 已提交
1073

Y
Yan Zheng 已提交
1074 1075 1076 1077 1078 1079
		if (extent_type == BTRFS_FILE_EXTENT_PREALLOC) {
			struct extent_map *em;
			struct extent_map_tree *em_tree;
			em_tree = &BTRFS_I(inode)->extent_tree;
			em = alloc_extent_map(GFP_NOFS);
			em->start = cur_offset;
1080
			em->orig_start = em->start;
Y
Yan Zheng 已提交
1081 1082 1083 1084 1085 1086
			em->len = num_bytes;
			em->block_len = num_bytes;
			em->block_start = disk_bytenr;
			em->bdev = root->fs_info->fs_devices->latest_bdev;
			set_bit(EXTENT_FLAG_PINNED, &em->flags);
			while (1) {
1087
				write_lock(&em_tree->lock);
Y
Yan Zheng 已提交
1088
				ret = add_extent_mapping(em_tree, em);
1089
				write_unlock(&em_tree->lock);
Y
Yan Zheng 已提交
1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100
				if (ret != -EEXIST) {
					free_extent_map(em);
					break;
				}
				btrfs_drop_extent_cache(inode, em->start,
						em->start + em->len - 1, 0);
			}
			type = BTRFS_ORDERED_PREALLOC;
		} else {
			type = BTRFS_ORDERED_NOCOW;
		}
Y
Yan Zheng 已提交
1101 1102

		ret = btrfs_add_ordered_extent(inode, cur_offset, disk_bytenr,
Y
Yan Zheng 已提交
1103 1104
					       num_bytes, num_bytes, type);
		BUG_ON(ret);
1105

Y
Yan Zheng 已提交
1106 1107
		extent_clear_unlock_delalloc(inode, &BTRFS_I(inode)->io_tree,
					cur_offset, cur_offset + num_bytes - 1,
1108
					locked_page, 1, 1, 1, 0, 0, 0, 1);
Y
Yan Zheng 已提交
1109 1110 1111
		cur_offset = extent_end;
		if (cur_offset > end)
			break;
1112
	}
Y
Yan Zheng 已提交
1113 1114 1115 1116 1117 1118
	btrfs_release_path(root, path);

	if (cur_offset <= end && cow_start == (u64)-1)
		cow_start = cur_offset;
	if (cow_start != (u64)-1) {
		ret = cow_file_range(inode, locked_page, cow_start, end,
1119
				     page_started, nr_written, 1);
Y
Yan Zheng 已提交
1120 1121 1122 1123 1124
		BUG_ON(ret);
	}

	ret = btrfs_end_transaction(trans, root);
	BUG_ON(ret);
1125
	btrfs_free_path(path);
Y
Yan Zheng 已提交
1126
	return 0;
1127 1128
}

C
Chris Mason 已提交
1129 1130 1131
/*
 * extent_io.c call back to do delayed allocation processing
 */
C
Chris Mason 已提交
1132
static int run_delalloc_range(struct inode *inode, struct page *locked_page,
1133 1134
			      u64 start, u64 end, int *page_started,
			      unsigned long *nr_written)
1135 1136
{
	int ret;
1137
	struct btrfs_root *root = BTRFS_I(inode)->root;
1138

1139
	if (BTRFS_I(inode)->flags & BTRFS_INODE_NODATACOW)
C
Chris Mason 已提交
1140
		ret = run_delalloc_nocow(inode, locked_page, start, end,
C
Chris Mason 已提交
1141
					 page_started, 1, nr_written);
1142
	else if (BTRFS_I(inode)->flags & BTRFS_INODE_PREALLOC)
Y
Yan Zheng 已提交
1143
		ret = run_delalloc_nocow(inode, locked_page, start, end,
C
Chris Mason 已提交
1144
					 page_started, 0, nr_written);
1145 1146 1147
	else if (!btrfs_test_opt(root, COMPRESS))
		ret = cow_file_range(inode, locked_page, start, end,
				      page_started, nr_written, 1);
1148
	else
1149
		ret = cow_file_range_async(inode, locked_page, start, end,
C
Chris Mason 已提交
1150
					   page_started, nr_written);
1151 1152 1153
	return ret;
}

C
Chris Mason 已提交
1154 1155 1156 1157 1158
/*
 * extent_io.c set_bit_hook, used to track delayed allocation
 * bytes in this file, and to maintain the list of inodes that
 * have pending delalloc work to be done.
 */
1159
static int btrfs_set_bit_hook(struct inode *inode, u64 start, u64 end,
C
Chris Mason 已提交
1160
		       unsigned long old, unsigned long bits)
1161
{
1162 1163 1164 1165 1166
	/*
	 * set_bit and clear bit hooks normally require _irqsave/restore
	 * but in this case, we are only testeing for the DELALLOC
	 * bit, which is only set or cleared with irqs on
	 */
C
Chris Mason 已提交
1167
	if (!(old & EXTENT_DELALLOC) && (bits & EXTENT_DELALLOC)) {
1168
		struct btrfs_root *root = BTRFS_I(inode)->root;
J
Josef Bacik 已提交
1169
		btrfs_delalloc_reserve_space(root, inode, end - start + 1);
1170
		spin_lock(&root->fs_info->delalloc_lock);
C
Chris Mason 已提交
1171
		BTRFS_I(inode)->delalloc_bytes += end - start + 1;
1172
		root->fs_info->delalloc_bytes += end - start + 1;
1173 1174 1175 1176
		if (list_empty(&BTRFS_I(inode)->delalloc_inodes)) {
			list_add_tail(&BTRFS_I(inode)->delalloc_inodes,
				      &root->fs_info->delalloc_inodes);
		}
1177
		spin_unlock(&root->fs_info->delalloc_lock);
1178 1179 1180 1181
	}
	return 0;
}

C
Chris Mason 已提交
1182 1183 1184
/*
 * extent_io.c clear_bit_hook, see set_bit_hook for why
 */
1185
static int btrfs_clear_bit_hook(struct inode *inode, u64 start, u64 end,
C
Chris Mason 已提交
1186
			 unsigned long old, unsigned long bits)
1187
{
1188 1189 1190 1191 1192
	/*
	 * set_bit and clear bit hooks normally require _irqsave/restore
	 * but in this case, we are only testeing for the DELALLOC
	 * bit, which is only set or cleared with irqs on
	 */
C
Chris Mason 已提交
1193
	if ((old & EXTENT_DELALLOC) && (bits & EXTENT_DELALLOC)) {
1194
		struct btrfs_root *root = BTRFS_I(inode)->root;
1195

1196
		spin_lock(&root->fs_info->delalloc_lock);
C
Chris Mason 已提交
1197
		if (end - start + 1 > root->fs_info->delalloc_bytes) {
C
Chris Mason 已提交
1198 1199 1200 1201 1202
			printk(KERN_INFO "btrfs warning: delalloc account "
			       "%llu %llu\n",
			       (unsigned long long)end - start + 1,
			       (unsigned long long)
			       root->fs_info->delalloc_bytes);
J
Josef Bacik 已提交
1203
			btrfs_delalloc_free_space(root, inode, (u64)-1);
C
Chris Mason 已提交
1204
			root->fs_info->delalloc_bytes = 0;
C
Chris Mason 已提交
1205
			BTRFS_I(inode)->delalloc_bytes = 0;
C
Chris Mason 已提交
1206
		} else {
J
Josef Bacik 已提交
1207 1208
			btrfs_delalloc_free_space(root, inode,
						  end - start + 1);
C
Chris Mason 已提交
1209
			root->fs_info->delalloc_bytes -= end - start + 1;
C
Chris Mason 已提交
1210
			BTRFS_I(inode)->delalloc_bytes -= end - start + 1;
C
Chris Mason 已提交
1211
		}
1212 1213 1214 1215
		if (BTRFS_I(inode)->delalloc_bytes == 0 &&
		    !list_empty(&BTRFS_I(inode)->delalloc_inodes)) {
			list_del_init(&BTRFS_I(inode)->delalloc_inodes);
		}
1216
		spin_unlock(&root->fs_info->delalloc_lock);
1217 1218 1219 1220
	}
	return 0;
}

C
Chris Mason 已提交
1221 1222 1223 1224
/*
 * extent_io.c merge_bio_hook, this must check the chunk tree to make sure
 * we don't create bios that span stripes or chunks
 */
1225
int btrfs_merge_bio_hook(struct page *page, unsigned long offset,
C
Chris Mason 已提交
1226 1227
			 size_t size, struct bio *bio,
			 unsigned long bio_flags)
1228 1229 1230
{
	struct btrfs_root *root = BTRFS_I(page->mapping->host)->root;
	struct btrfs_mapping_tree *map_tree;
1231
	u64 logical = (u64)bio->bi_sector << 9;
1232 1233 1234 1235
	u64 length = 0;
	u64 map_length;
	int ret;

1236 1237 1238
	if (bio_flags & EXTENT_BIO_COMPRESSED)
		return 0;

1239
	length = bio->bi_size;
1240 1241
	map_tree = &root->fs_info->mapping_tree;
	map_length = length;
1242
	ret = btrfs_map_block(map_tree, READ, logical,
1243
			      &map_length, NULL, 0);
1244

C
Chris Mason 已提交
1245
	if (map_length < length + size)
1246 1247 1248 1249
		return 1;
	return 0;
}

C
Chris Mason 已提交
1250 1251 1252 1253 1254 1255 1256 1257
/*
 * in order to insert checksums into the metadata in large chunks,
 * we wait until bio submission time.   All the pages in the bio are
 * checksummed and sums are attached onto the ordered extent record.
 *
 * At IO completion time the cums attached on the ordered extent record
 * are inserted into the btree
 */
C
Chris Mason 已提交
1258 1259 1260
static int __btrfs_submit_bio_start(struct inode *inode, int rw,
				    struct bio *bio, int mirror_num,
				    unsigned long bio_flags)
1261 1262 1263
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	int ret = 0;
1264

1265
	ret = btrfs_csum_one_bio(root, inode, bio, 0, 0);
1266
	BUG_ON(ret);
C
Chris Mason 已提交
1267 1268
	return 0;
}
1269

C
Chris Mason 已提交
1270 1271 1272 1273 1274 1275 1276 1277
/*
 * in order to insert checksums into the metadata in large chunks,
 * we wait until bio submission time.   All the pages in the bio are
 * checksummed and sums are attached onto the ordered extent record.
 *
 * At IO completion time the cums attached on the ordered extent record
 * are inserted into the btree
 */
1278
static int __btrfs_submit_bio_done(struct inode *inode, int rw, struct bio *bio,
C
Chris Mason 已提交
1279 1280 1281
			  int mirror_num, unsigned long bio_flags)
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
1282
	return btrfs_map_bio(root, rw, bio, mirror_num, 1);
1283 1284
}

C
Chris Mason 已提交
1285
/*
1286 1287
 * extent_io.c submission hook. This does the right thing for csum calculation
 * on write, or reading the csums from the tree before a read
C
Chris Mason 已提交
1288
 */
1289
static int btrfs_submit_bio_hook(struct inode *inode, int rw, struct bio *bio,
C
Chris Mason 已提交
1290
			  int mirror_num, unsigned long bio_flags)
1291 1292 1293
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	int ret = 0;
1294
	int skip_sum;
1295

1296
	skip_sum = BTRFS_I(inode)->flags & BTRFS_INODE_NODATASUM;
1297

1298 1299
	ret = btrfs_bio_wq_end_io(root->fs_info, bio, 0);
	BUG_ON(ret);
1300

1301
	if (!(rw & (1 << BIO_RW))) {
1302
		if (bio_flags & EXTENT_BIO_COMPRESSED) {
C
Chris Mason 已提交
1303 1304
			return btrfs_submit_compressed_read(inode, bio,
						    mirror_num, bio_flags);
1305 1306
		} else if (!skip_sum)
			btrfs_lookup_bio_sums(root, inode, bio, NULL);
1307
		goto mapit;
1308
	} else if (!skip_sum) {
1309 1310 1311
		/* csum items have already been cloned */
		if (root->root_key.objectid == BTRFS_DATA_RELOC_TREE_OBJECTID)
			goto mapit;
1312 1313
		/* we're doing a write, do the async checksumming */
		return btrfs_wq_submit_bio(BTRFS_I(inode)->root->fs_info,
1314
				   inode, rw, bio, mirror_num,
C
Chris Mason 已提交
1315 1316
				   bio_flags, __btrfs_submit_bio_start,
				   __btrfs_submit_bio_done);
1317 1318
	}

1319
mapit:
1320
	return btrfs_map_bio(root, rw, bio, mirror_num, 0);
1321
}
C
Chris Mason 已提交
1322

C
Chris Mason 已提交
1323 1324 1325 1326
/*
 * given a list of ordered sums record them in the inode.  This happens
 * at IO completion time based on sums calculated at bio submission time.
 */
1327
static noinline int add_pending_csums(struct btrfs_trans_handle *trans,
1328 1329 1330 1331 1332 1333
			     struct inode *inode, u64 file_offset,
			     struct list_head *list)
{
	struct btrfs_ordered_sum *sum;

	btrfs_set_trans_block_group(trans, inode);
Q
Qinghuang Feng 已提交
1334 1335

	list_for_each_entry(sum, list, list) {
1336 1337
		btrfs_csum_file_blocks(trans,
		       BTRFS_I(inode)->root->fs_info->csum_root, sum);
1338 1339 1340 1341
	}
	return 0;
}

1342 1343
int btrfs_set_extent_delalloc(struct inode *inode, u64 start, u64 end)
{
C
Chris Mason 已提交
1344
	if ((end & (PAGE_CACHE_SIZE - 1)) == 0)
1345
		WARN_ON(1);
1346 1347 1348 1349
	return set_extent_delalloc(&BTRFS_I(inode)->io_tree, start, end,
				   GFP_NOFS);
}

C
Chris Mason 已提交
1350
/* see btrfs_writepage_start_hook for details on why this is required */
1351 1352 1353 1354 1355
struct btrfs_writepage_fixup {
	struct page *page;
	struct btrfs_work work;
};

1356
static void btrfs_writepage_fixup_worker(struct btrfs_work *work)
1357 1358 1359 1360 1361 1362 1363 1364 1365 1366
{
	struct btrfs_writepage_fixup *fixup;
	struct btrfs_ordered_extent *ordered;
	struct page *page;
	struct inode *inode;
	u64 page_start;
	u64 page_end;

	fixup = container_of(work, struct btrfs_writepage_fixup, work);
	page = fixup->page;
C
Chris Mason 已提交
1367
again:
1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378
	lock_page(page);
	if (!page->mapping || !PageDirty(page) || !PageChecked(page)) {
		ClearPageChecked(page);
		goto out_page;
	}

	inode = page->mapping->host;
	page_start = page_offset(page);
	page_end = page_offset(page) + PAGE_CACHE_SIZE - 1;

	lock_extent(&BTRFS_I(inode)->io_tree, page_start, page_end, GFP_NOFS);
C
Chris Mason 已提交
1379 1380

	/* already ordered? We're done */
1381
	if (PagePrivate2(page))
1382
		goto out;
C
Chris Mason 已提交
1383 1384 1385 1386 1387 1388 1389 1390 1391

	ordered = btrfs_lookup_ordered_extent(inode, page_start);
	if (ordered) {
		unlock_extent(&BTRFS_I(inode)->io_tree, page_start,
			      page_end, GFP_NOFS);
		unlock_page(page);
		btrfs_start_ordered_extent(inode, ordered, 1);
		goto again;
	}
1392

1393
	btrfs_set_extent_delalloc(inode, page_start, page_end);
1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407
	ClearPageChecked(page);
out:
	unlock_extent(&BTRFS_I(inode)->io_tree, page_start, page_end, GFP_NOFS);
out_page:
	unlock_page(page);
	page_cache_release(page);
}

/*
 * There are a few paths in the higher layers of the kernel that directly
 * set the page dirty bit without asking the filesystem if it is a
 * good idea.  This causes problems because we want to make sure COW
 * properly happens and the data=ordered rules are followed.
 *
C
Chris Mason 已提交
1408
 * In our case any range that doesn't have the ORDERED bit set
1409 1410 1411 1412
 * hasn't been properly setup for IO.  We kick off an async process
 * to fix it up.  The async helper will wait for ordered extents, set
 * the delalloc bit and make it safe to write the page.
 */
1413
static int btrfs_writepage_start_hook(struct page *page, u64 start, u64 end)
1414 1415 1416 1417 1418
{
	struct inode *inode = page->mapping->host;
	struct btrfs_writepage_fixup *fixup;
	struct btrfs_root *root = BTRFS_I(inode)->root;

1419 1420
	/* this page is properly in the ordered list */
	if (TestClearPagePrivate2(page))
1421 1422 1423 1424 1425 1426 1427 1428
		return 0;

	if (PageChecked(page))
		return -EAGAIN;

	fixup = kzalloc(sizeof(*fixup), GFP_NOFS);
	if (!fixup)
		return -EAGAIN;
1429

1430 1431 1432 1433 1434 1435 1436 1437
	SetPageChecked(page);
	page_cache_get(page);
	fixup->work.func = btrfs_writepage_fixup_worker;
	fixup->page = page;
	btrfs_queue_worker(&root->fs_info->fixup_workers, &fixup->work);
	return -EAGAIN;
}

Y
Yan Zheng 已提交
1438 1439 1440 1441
static int insert_reserved_file_extent(struct btrfs_trans_handle *trans,
				       struct inode *inode, u64 file_pos,
				       u64 disk_bytenr, u64 disk_num_bytes,
				       u64 num_bytes, u64 ram_bytes,
1442
				       u64 locked_end,
Y
Yan Zheng 已提交
1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456
				       u8 compression, u8 encryption,
				       u16 other_encoding, int extent_type)
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct btrfs_file_extent_item *fi;
	struct btrfs_path *path;
	struct extent_buffer *leaf;
	struct btrfs_key ins;
	u64 hint;
	int ret;

	path = btrfs_alloc_path();
	BUG_ON(!path);

1457
	path->leave_spinning = 1;
C
Chris Mason 已提交
1458 1459 1460 1461 1462 1463 1464 1465 1466 1467

	/*
	 * we may be replacing one extent in the tree with another.
	 * The new extent is pinned in the extent map, and we don't want
	 * to drop it from the cache until it is completely in the btree.
	 *
	 * So, tell btrfs_drop_extents to leave this extent in the cache.
	 * the caller is expected to unpin it and allow it to be merged
	 * with the others.
	 */
Y
Yan Zheng 已提交
1468
	ret = btrfs_drop_extents(trans, root, inode, file_pos,
1469
				 file_pos + num_bytes, locked_end,
C
Chris Mason 已提交
1470
				 file_pos, &hint, 0);
Y
Yan Zheng 已提交
1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490
	BUG_ON(ret);

	ins.objectid = inode->i_ino;
	ins.offset = file_pos;
	ins.type = BTRFS_EXTENT_DATA_KEY;
	ret = btrfs_insert_empty_item(trans, root, path, &ins, 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, disk_bytenr);
	btrfs_set_file_extent_disk_num_bytes(leaf, fi, disk_num_bytes);
	btrfs_set_file_extent_offset(leaf, fi, 0);
	btrfs_set_file_extent_num_bytes(leaf, fi, num_bytes);
	btrfs_set_file_extent_ram_bytes(leaf, fi, ram_bytes);
	btrfs_set_file_extent_compression(leaf, fi, compression);
	btrfs_set_file_extent_encryption(leaf, fi, encryption);
	btrfs_set_file_extent_other_encoding(leaf, fi, other_encoding);
1491 1492 1493 1494

	btrfs_unlock_up_safe(path, 1);
	btrfs_set_lock_blocking(leaf);

Y
Yan Zheng 已提交
1495 1496 1497 1498 1499 1500 1501
	btrfs_mark_buffer_dirty(leaf);

	inode_add_bytes(inode, num_bytes);

	ins.objectid = disk_bytenr;
	ins.offset = disk_num_bytes;
	ins.type = BTRFS_EXTENT_ITEM_KEY;
1502 1503 1504
	ret = btrfs_alloc_reserved_file_extent(trans, root,
					root->root_key.objectid,
					inode->i_ino, file_pos, &ins);
Y
Yan Zheng 已提交
1505 1506
	BUG_ON(ret);
	btrfs_free_path(path);
1507

Y
Yan Zheng 已提交
1508 1509 1510
	return 0;
}

1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534
/*
 * helper function for btrfs_finish_ordered_io, this
 * just reads in some of the csum leaves to prime them into ram
 * before we start the transaction.  It limits the amount of btree
 * reads required while inside the transaction.
 */
static noinline void reada_csum(struct btrfs_root *root,
				struct btrfs_path *path,
				struct btrfs_ordered_extent *ordered_extent)
{
	struct btrfs_ordered_sum *sum;
	u64 bytenr;

	sum = list_entry(ordered_extent->list.next, struct btrfs_ordered_sum,
			 list);
	bytenr = sum->sums[0].bytenr;

	/*
	 * we don't care about the results, the point of this search is
	 * just to get the btree leaves into ram
	 */
	btrfs_lookup_csum(NULL, root->fs_info->csum_root, path, bytenr, 0);
}

C
Chris Mason 已提交
1535 1536 1537 1538
/* as ordered data IO finishes, this gets called so we can finish
 * an ordered extent if the range of bytes in the file it covers are
 * fully written.
 */
1539
static int btrfs_finish_ordered_io(struct inode *inode, u64 start, u64 end)
1540 1541 1542
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct btrfs_trans_handle *trans;
1543
	struct btrfs_ordered_extent *ordered_extent = NULL;
1544
	struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree;
1545
	struct btrfs_path *path;
Y
Yan Zheng 已提交
1546
	int compressed = 0;
1547 1548 1549
	int ret;

	ret = btrfs_dec_test_ordered_pending(inode, start, end - start + 1);
1550
	if (!ret)
1551 1552
		return 0;

1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565
	/*
	 * before we join the transaction, try to do some of our IO.
	 * This will limit the amount of IO that we have to do with
	 * the transaction running.  We're unlikely to need to do any
	 * IO if the file extents are new, the disk_i_size checks
	 * covers the most common case.
	 */
	if (start < BTRFS_I(inode)->disk_i_size) {
		path = btrfs_alloc_path();
		if (path) {
			ret = btrfs_lookup_file_extent(NULL, root, path,
						       inode->i_ino,
						       start, 0);
1566 1567 1568 1569 1570 1571
			ordered_extent = btrfs_lookup_ordered_extent(inode,
								     start);
			if (!list_empty(&ordered_extent->list)) {
				btrfs_release_path(root, path);
				reada_csum(root, path, ordered_extent);
			}
1572 1573 1574 1575
			btrfs_free_path(path);
		}
	}

1576
	trans = btrfs_join_transaction(root, 1);
1577

1578 1579
	if (!ordered_extent)
		ordered_extent = btrfs_lookup_ordered_extent(inode, start);
1580
	BUG_ON(!ordered_extent);
1581 1582
	if (test_bit(BTRFS_ORDERED_NOCOW, &ordered_extent->flags))
		goto nocow;
1583 1584 1585 1586 1587

	lock_extent(io_tree, ordered_extent->file_offset,
		    ordered_extent->file_offset + ordered_extent->len - 1,
		    GFP_NOFS);

C
Chris Mason 已提交
1588
	if (test_bit(BTRFS_ORDERED_COMPRESSED, &ordered_extent->flags))
Y
Yan Zheng 已提交
1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603
		compressed = 1;
	if (test_bit(BTRFS_ORDERED_PREALLOC, &ordered_extent->flags)) {
		BUG_ON(compressed);
		ret = btrfs_mark_extent_written(trans, root, inode,
						ordered_extent->file_offset,
						ordered_extent->file_offset +
						ordered_extent->len);
		BUG_ON(ret);
	} else {
		ret = insert_reserved_file_extent(trans, inode,
						ordered_extent->file_offset,
						ordered_extent->start,
						ordered_extent->disk_len,
						ordered_extent->len,
						ordered_extent->len,
1604 1605
						ordered_extent->file_offset +
						ordered_extent->len,
Y
Yan Zheng 已提交
1606 1607
						compressed, 0, 0,
						BTRFS_FILE_EXTENT_REG);
C
Chris Mason 已提交
1608 1609 1610
		unpin_extent_cache(&BTRFS_I(inode)->extent_tree,
				   ordered_extent->file_offset,
				   ordered_extent->len);
Y
Yan Zheng 已提交
1611 1612
		BUG_ON(ret);
	}
1613 1614 1615
	unlock_extent(io_tree, ordered_extent->file_offset,
		    ordered_extent->file_offset + ordered_extent->len - 1,
		    GFP_NOFS);
1616
nocow:
1617 1618 1619
	add_pending_csums(trans, inode, ordered_extent->file_offset,
			  &ordered_extent->list);

1620
	mutex_lock(&BTRFS_I(inode)->extent_mutex);
1621
	btrfs_ordered_update_i_size(inode, ordered_extent);
1622
	btrfs_update_inode(trans, root, inode);
1623
	btrfs_remove_ordered_extent(inode, ordered_extent);
1624
	mutex_unlock(&BTRFS_I(inode)->extent_mutex);
1625

1626 1627 1628 1629 1630 1631 1632 1633 1634
	/* once for us */
	btrfs_put_ordered_extent(ordered_extent);
	/* once for the tree */
	btrfs_put_ordered_extent(ordered_extent);

	btrfs_end_transaction(trans, root);
	return 0;
}

1635
static int btrfs_writepage_end_io_hook(struct page *page, u64 start, u64 end,
1636 1637
				struct extent_state *state, int uptodate)
{
1638
	ClearPagePrivate2(page);
1639 1640 1641
	return btrfs_finish_ordered_io(page->mapping->host, start, end);
}

C
Chris Mason 已提交
1642 1643 1644 1645 1646 1647 1648 1649
/*
 * When IO fails, either with EIO or csum verification fails, we
 * try other mirrors that might have a good copy of the data.  This
 * io_failure_record is used to record state as we go through all the
 * mirrors.  If another mirror has good data, the page is set up to date
 * and things continue.  If a good mirror can't be found, the original
 * bio end_io callback is called to indicate things have failed.
 */
1650 1651 1652 1653 1654
struct io_failure_record {
	struct page *page;
	u64 start;
	u64 len;
	u64 logical;
1655
	unsigned long bio_flags;
1656 1657 1658
	int last_mirror;
};

1659
static int btrfs_io_failed_hook(struct bio *failed_bio,
1660 1661
			 struct page *page, u64 start, u64 end,
			 struct extent_state *state)
1662 1663 1664 1665 1666 1667
{
	struct io_failure_record *failrec = NULL;
	u64 private;
	struct extent_map *em;
	struct inode *inode = page->mapping->host;
	struct extent_io_tree *failure_tree = &BTRFS_I(inode)->io_failure_tree;
1668
	struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree;
1669 1670 1671
	struct bio *bio;
	int num_copies;
	int ret;
1672
	int rw;
1673 1674 1675 1676 1677 1678 1679 1680 1681 1682
	u64 logical;

	ret = get_state_private(failure_tree, start, &private);
	if (ret) {
		failrec = kmalloc(sizeof(*failrec), GFP_NOFS);
		if (!failrec)
			return -ENOMEM;
		failrec->start = start;
		failrec->len = end - start + 1;
		failrec->last_mirror = 0;
1683
		failrec->bio_flags = 0;
1684

1685
		read_lock(&em_tree->lock);
1686 1687 1688 1689 1690
		em = lookup_extent_mapping(em_tree, start, failrec->len);
		if (em->start > start || em->start + em->len < start) {
			free_extent_map(em);
			em = NULL;
		}
1691
		read_unlock(&em_tree->lock);
1692 1693 1694 1695 1696 1697 1698

		if (!em || IS_ERR(em)) {
			kfree(failrec);
			return -EIO;
		}
		logical = start - em->start;
		logical = em->block_start + logical;
1699 1700 1701 1702
		if (test_bit(EXTENT_FLAG_COMPRESSED, &em->flags)) {
			logical = em->block_start;
			failrec->bio_flags = EXTENT_BIO_COMPRESSED;
		}
1703 1704 1705 1706
		failrec->logical = logical;
		free_extent_map(em);
		set_extent_bits(failure_tree, start, end, EXTENT_LOCKED |
				EXTENT_DIRTY, GFP_NOFS);
1707 1708
		set_state_private(failure_tree, start,
				 (u64)(unsigned long)failrec);
1709
	} else {
1710
		failrec = (struct io_failure_record *)(unsigned long)private;
1711 1712 1713 1714 1715 1716
	}
	num_copies = btrfs_num_copies(
			      &BTRFS_I(inode)->root->fs_info->mapping_tree,
			      failrec->logical, failrec->len);
	failrec->last_mirror++;
	if (!state) {
1717
		spin_lock(&BTRFS_I(inode)->io_tree.lock);
1718 1719 1720 1721 1722
		state = find_first_extent_bit_state(&BTRFS_I(inode)->io_tree,
						    failrec->start,
						    EXTENT_LOCKED);
		if (state && state->start != failrec->start)
			state = NULL;
1723
		spin_unlock(&BTRFS_I(inode)->io_tree.lock);
1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737
	}
	if (!state || failrec->last_mirror > num_copies) {
		set_state_private(failure_tree, failrec->start, 0);
		clear_extent_bits(failure_tree, failrec->start,
				  failrec->start + failrec->len - 1,
				  EXTENT_LOCKED | EXTENT_DIRTY, GFP_NOFS);
		kfree(failrec);
		return -EIO;
	}
	bio = bio_alloc(GFP_NOFS, 1);
	bio->bi_private = state;
	bio->bi_end_io = failed_bio->bi_end_io;
	bio->bi_sector = failrec->logical >> 9;
	bio->bi_bdev = failed_bio->bi_bdev;
1738
	bio->bi_size = 0;
1739

1740
	bio_add_page(bio, page, failrec->len, start - page_offset(page));
1741 1742 1743 1744 1745 1746
	if (failed_bio->bi_rw & (1 << BIO_RW))
		rw = WRITE;
	else
		rw = READ;

	BTRFS_I(inode)->io_tree.ops->submit_bio_hook(inode, rw, bio,
C
Chris Mason 已提交
1747
						      failrec->last_mirror,
1748
						      failrec->bio_flags);
1749 1750 1751
	return 0;
}

C
Chris Mason 已提交
1752 1753 1754 1755
/*
 * each time an IO finishes, we do a fast check in the IO failure tree
 * to see if we need to process or clean up an io_failure_record
 */
1756
static int btrfs_clean_io_failures(struct inode *inode, u64 start)
1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780
{
	u64 private;
	u64 private_failure;
	struct io_failure_record *failure;
	int ret;

	private = 0;
	if (count_range_bits(&BTRFS_I(inode)->io_failure_tree, &private,
			     (u64)-1, 1, EXTENT_DIRTY)) {
		ret = get_state_private(&BTRFS_I(inode)->io_failure_tree,
					start, &private_failure);
		if (ret == 0) {
			failure = (struct io_failure_record *)(unsigned long)
				   private_failure;
			set_state_private(&BTRFS_I(inode)->io_failure_tree,
					  failure->start, 0);
			clear_extent_bits(&BTRFS_I(inode)->io_failure_tree,
					  failure->start,
					  failure->start + failure->len - 1,
					  EXTENT_DIRTY | EXTENT_LOCKED,
					  GFP_NOFS);
			kfree(failure);
		}
	}
1781 1782 1783
	return 0;
}

C
Chris Mason 已提交
1784 1785 1786 1787 1788
/*
 * when reads are done, we need to check csums to verify the data is correct
 * if there's a match, we allow the bio to finish.  If not, we go through
 * the io_failure_record routines to find good copies
 */
1789
static int btrfs_readpage_end_io_hook(struct page *page, u64 start, u64 end,
1790
			       struct extent_state *state)
1791
{
1792
	size_t offset = start - ((u64)page->index << PAGE_CACHE_SHIFT);
1793
	struct inode *inode = page->mapping->host;
1794
	struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree;
1795
	char *kaddr;
1796
	u64 private = ~(u32)0;
1797
	int ret;
1798 1799
	struct btrfs_root *root = BTRFS_I(inode)->root;
	u32 csum = ~(u32)0;
1800

1801 1802 1803 1804
	if (PageChecked(page)) {
		ClearPageChecked(page);
		goto good;
	}
1805 1806

	if (BTRFS_I(inode)->flags & BTRFS_INODE_NODATASUM)
1807 1808 1809
		return 0;

	if (root->root_key.objectid == BTRFS_DATA_RELOC_TREE_OBJECTID &&
1810
	    test_range_bit(io_tree, start, end, EXTENT_NODATASUM, 1, NULL)) {
1811 1812
		clear_extent_bits(io_tree, start, end, EXTENT_NODATASUM,
				  GFP_NOFS);
1813
		return 0;
1814
	}
1815

Y
Yan 已提交
1816
	if (state && state->start == start) {
1817 1818 1819 1820 1821
		private = state->private;
		ret = 0;
	} else {
		ret = get_state_private(io_tree, start, &private);
	}
1822
	kaddr = kmap_atomic(page, KM_USER0);
C
Chris Mason 已提交
1823
	if (ret)
1824
		goto zeroit;
C
Chris Mason 已提交
1825

1826 1827
	csum = btrfs_csum_data(root, kaddr + offset, csum,  end - start + 1);
	btrfs_csum_final(csum, (char *)&csum);
C
Chris Mason 已提交
1828
	if (csum != private)
1829
		goto zeroit;
C
Chris Mason 已提交
1830

1831
	kunmap_atomic(kaddr, KM_USER0);
1832
good:
1833 1834 1835
	/* if the io failure tree for this inode is non-empty,
	 * check to see if we've recovered from a failed IO
	 */
1836
	btrfs_clean_io_failures(inode, start);
1837 1838 1839
	return 0;

zeroit:
C
Chris Mason 已提交
1840 1841 1842 1843 1844 1845
	if (printk_ratelimit()) {
		printk(KERN_INFO "btrfs csum failed ino %lu off %llu csum %u "
		       "private %llu\n", page->mapping->host->i_ino,
		       (unsigned long long)start, csum,
		       (unsigned long long)private);
	}
1846 1847
	memset(kaddr + offset, 1, end - start + 1);
	flush_dcache_page(page);
1848
	kunmap_atomic(kaddr, KM_USER0);
1849 1850
	if (private == 0)
		return 0;
1851
	return -EIO;
1852
}
1853

1854 1855 1856 1857 1858 1859 1860 1861 1862
/*
 * This creates an orphan entry for the given inode in case something goes
 * wrong in the middle of an unlink/truncate.
 */
int btrfs_orphan_add(struct btrfs_trans_handle *trans, struct inode *inode)
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	int ret = 0;

1863
	spin_lock(&root->list_lock);
1864 1865 1866

	/* already on the orphan list, we're good */
	if (!list_empty(&BTRFS_I(inode)->i_orphan)) {
1867
		spin_unlock(&root->list_lock);
1868 1869 1870 1871 1872
		return 0;
	}

	list_add(&BTRFS_I(inode)->i_orphan, &root->orphan_list);

1873
	spin_unlock(&root->list_lock);
1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891

	/*
	 * insert an orphan item to track this unlinked/truncated file
	 */
	ret = btrfs_insert_orphan_item(trans, root, inode->i_ino);

	return ret;
}

/*
 * We have done the truncate/delete so we can go ahead and remove the orphan
 * item for this particular inode.
 */
int btrfs_orphan_del(struct btrfs_trans_handle *trans, struct inode *inode)
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	int ret = 0;

1892
	spin_lock(&root->list_lock);
1893 1894

	if (list_empty(&BTRFS_I(inode)->i_orphan)) {
1895
		spin_unlock(&root->list_lock);
1896 1897 1898 1899 1900
		return 0;
	}

	list_del_init(&BTRFS_I(inode)->i_orphan);
	if (!trans) {
1901
		spin_unlock(&root->list_lock);
1902 1903 1904
		return 0;
	}

1905
	spin_unlock(&root->list_lock);
1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973

	ret = btrfs_del_orphan_item(trans, root, inode->i_ino);

	return ret;
}

/*
 * this cleans up any orphans that may be left on the list from the last use
 * of this root.
 */
void btrfs_orphan_cleanup(struct btrfs_root *root)
{
	struct btrfs_path *path;
	struct extent_buffer *leaf;
	struct btrfs_item *item;
	struct btrfs_key key, found_key;
	struct btrfs_trans_handle *trans;
	struct inode *inode;
	int ret = 0, nr_unlink = 0, nr_truncate = 0;

	path = btrfs_alloc_path();
	if (!path)
		return;
	path->reada = -1;

	key.objectid = BTRFS_ORPHAN_OBJECTID;
	btrfs_set_key_type(&key, BTRFS_ORPHAN_ITEM_KEY);
	key.offset = (u64)-1;


	while (1) {
		ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
		if (ret < 0) {
			printk(KERN_ERR "Error searching slot for orphan: %d"
			       "\n", ret);
			break;
		}

		/*
		 * if ret == 0 means we found what we were searching for, which
		 * is weird, but possible, so only screw with path if we didnt
		 * find the key and see if we have stuff that matches
		 */
		if (ret > 0) {
			if (path->slots[0] == 0)
				break;
			path->slots[0]--;
		}

		/* pull out the item */
		leaf = path->nodes[0];
		item = btrfs_item_nr(leaf, path->slots[0]);
		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);

		/* make sure the item matches what we want */
		if (found_key.objectid != BTRFS_ORPHAN_OBJECTID)
			break;
		if (btrfs_key_type(&found_key) != BTRFS_ORPHAN_ITEM_KEY)
			break;

		/* release the path since we're done with it */
		btrfs_release_path(root, path);

		/*
		 * this is where we are basically btrfs_lookup, without the
		 * crossing root thing.  we store the inode number in the
		 * offset of the orphan item.
		 */
1974 1975 1976 1977 1978
		found_key.objectid = found_key.offset;
		found_key.type = BTRFS_INODE_ITEM_KEY;
		found_key.offset = 0;
		inode = btrfs_iget(root->fs_info->sb, &found_key, root);
		if (IS_ERR(inode))
1979 1980 1981 1982 1983 1984
			break;

		/*
		 * add this inode to the orphan list so btrfs_orphan_del does
		 * the proper thing when we hit it
		 */
1985
		spin_lock(&root->list_lock);
1986
		list_add(&BTRFS_I(inode)->i_orphan, &root->orphan_list);
1987
		spin_unlock(&root->list_lock);
1988 1989 1990 1991 1992 1993 1994 1995

		/*
		 * if this is a bad inode, means we actually succeeded in
		 * removing the inode, but not the orphan record, which means
		 * we need to manually delete the orphan since iput will just
		 * do a destroy_inode
		 */
		if (is_bad_inode(inode)) {
1996
			trans = btrfs_start_transaction(root, 1);
1997
			btrfs_orphan_del(trans, inode);
1998
			btrfs_end_transaction(trans, root);
1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022
			iput(inode);
			continue;
		}

		/* if we have links, this was a truncate, lets do that */
		if (inode->i_nlink) {
			nr_truncate++;
			btrfs_truncate(inode);
		} else {
			nr_unlink++;
		}

		/* this will do delete_inode and everything for us */
		iput(inode);
	}

	if (nr_unlink)
		printk(KERN_INFO "btrfs: unlinked %d orphans\n", nr_unlink);
	if (nr_truncate)
		printk(KERN_INFO "btrfs: truncated %d orphans\n", nr_truncate);

	btrfs_free_path(path);
}

2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073
/*
 * very simple check to peek ahead in the leaf looking for xattrs.  If we
 * don't find any xattrs, we know there can't be any acls.
 *
 * slot is the slot the inode is in, objectid is the objectid of the inode
 */
static noinline int acls_after_inode_item(struct extent_buffer *leaf,
					  int slot, u64 objectid)
{
	u32 nritems = btrfs_header_nritems(leaf);
	struct btrfs_key found_key;
	int scanned = 0;

	slot++;
	while (slot < nritems) {
		btrfs_item_key_to_cpu(leaf, &found_key, slot);

		/* we found a different objectid, there must not be acls */
		if (found_key.objectid != objectid)
			return 0;

		/* we found an xattr, assume we've got an acl */
		if (found_key.type == BTRFS_XATTR_ITEM_KEY)
			return 1;

		/*
		 * we found a key greater than an xattr key, there can't
		 * be any acls later on
		 */
		if (found_key.type > BTRFS_XATTR_ITEM_KEY)
			return 0;

		slot++;
		scanned++;

		/*
		 * it goes inode, inode backrefs, xattrs, extents,
		 * so if there are a ton of hard links to an inode there can
		 * be a lot of backrefs.  Don't waste time searching too hard,
		 * this is just an optimization
		 */
		if (scanned >= 8)
			break;
	}
	/* we hit the end of the leaf before we found an xattr or
	 * something larger than an xattr.  We have to assume the inode
	 * has acls
	 */
	return 1;
}

C
Chris Mason 已提交
2074 2075 2076
/*
 * read an inode from the btree into the in-memory inode
 */
2077
static void btrfs_read_locked_inode(struct inode *inode)
C
Chris Mason 已提交
2078 2079
{
	struct btrfs_path *path;
2080
	struct extent_buffer *leaf;
C
Chris Mason 已提交
2081
	struct btrfs_inode_item *inode_item;
2082
	struct btrfs_timespec *tspec;
C
Chris Mason 已提交
2083 2084
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct btrfs_key location;
2085
	int maybe_acls;
C
Chris Mason 已提交
2086
	u64 alloc_group_block;
J
Josef Bacik 已提交
2087
	u32 rdev;
C
Chris Mason 已提交
2088 2089 2090 2091 2092
	int ret;

	path = btrfs_alloc_path();
	BUG_ON(!path);
	memcpy(&location, &BTRFS_I(inode)->location, sizeof(location));
C
Chris Mason 已提交
2093

C
Chris Mason 已提交
2094
	ret = btrfs_lookup_inode(NULL, root, path, &location, 0);
2095
	if (ret)
C
Chris Mason 已提交
2096 2097
		goto make_bad;

2098 2099 2100 2101 2102 2103 2104 2105
	leaf = path->nodes[0];
	inode_item = btrfs_item_ptr(leaf, path->slots[0],
				    struct btrfs_inode_item);

	inode->i_mode = btrfs_inode_mode(leaf, inode_item);
	inode->i_nlink = btrfs_inode_nlink(leaf, inode_item);
	inode->i_uid = btrfs_inode_uid(leaf, inode_item);
	inode->i_gid = btrfs_inode_gid(leaf, inode_item);
2106
	btrfs_i_size_write(inode, btrfs_inode_size(leaf, inode_item));
2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119

	tspec = btrfs_inode_atime(inode_item);
	inode->i_atime.tv_sec = btrfs_timespec_sec(leaf, tspec);
	inode->i_atime.tv_nsec = btrfs_timespec_nsec(leaf, tspec);

	tspec = btrfs_inode_mtime(inode_item);
	inode->i_mtime.tv_sec = btrfs_timespec_sec(leaf, tspec);
	inode->i_mtime.tv_nsec = btrfs_timespec_nsec(leaf, tspec);

	tspec = btrfs_inode_ctime(inode_item);
	inode->i_ctime.tv_sec = btrfs_timespec_sec(leaf, tspec);
	inode->i_ctime.tv_nsec = btrfs_timespec_nsec(leaf, tspec);

2120
	inode_set_bytes(inode, btrfs_inode_nbytes(leaf, inode_item));
2121
	BTRFS_I(inode)->generation = btrfs_inode_generation(leaf, inode_item);
2122
	BTRFS_I(inode)->sequence = btrfs_inode_sequence(leaf, inode_item);
2123
	inode->i_generation = BTRFS_I(inode)->generation;
J
Josef Bacik 已提交
2124
	inode->i_rdev = 0;
2125 2126
	rdev = btrfs_inode_rdev(leaf, inode_item);

2127
	BTRFS_I(inode)->index_cnt = (u64)-1;
2128
	BTRFS_I(inode)->flags = btrfs_inode_flags(leaf, inode_item);
2129

2130
	alloc_group_block = btrfs_inode_block_group(leaf, inode_item);
2131

2132 2133 2134 2135 2136
	/*
	 * try to precache a NULL acl entry for files that don't have
	 * any xattrs or acls
	 */
	maybe_acls = acls_after_inode_item(leaf, path->slots[0], inode->i_ino);
2137 2138
	if (!maybe_acls)
		cache_no_acl(inode);
2139

2140 2141
	BTRFS_I(inode)->block_group = btrfs_find_block_group(root, 0,
						alloc_group_block, 0);
C
Chris Mason 已提交
2142 2143 2144 2145 2146 2147
	btrfs_free_path(path);
	inode_item = NULL;

	switch (inode->i_mode & S_IFMT) {
	case S_IFREG:
		inode->i_mapping->a_ops = &btrfs_aops;
C
Chris Mason 已提交
2148
		inode->i_mapping->backing_dev_info = &root->fs_info->bdi;
2149
		BTRFS_I(inode)->io_tree.ops = &btrfs_extent_io_ops;
C
Chris Mason 已提交
2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162
		inode->i_fop = &btrfs_file_operations;
		inode->i_op = &btrfs_file_inode_operations;
		break;
	case S_IFDIR:
		inode->i_fop = &btrfs_dir_file_operations;
		if (root == root->fs_info->tree_root)
			inode->i_op = &btrfs_dir_ro_inode_operations;
		else
			inode->i_op = &btrfs_dir_inode_operations;
		break;
	case S_IFLNK:
		inode->i_op = &btrfs_symlink_inode_operations;
		inode->i_mapping->a_ops = &btrfs_symlink_aops;
C
Chris Mason 已提交
2163
		inode->i_mapping->backing_dev_info = &root->fs_info->bdi;
C
Chris Mason 已提交
2164
		break;
J
Josef Bacik 已提交
2165
	default:
J
Jim Owens 已提交
2166
		inode->i_op = &btrfs_special_inode_operations;
J
Josef Bacik 已提交
2167 2168
		init_special_inode(inode, inode->i_mode, rdev);
		break;
C
Chris Mason 已提交
2169
	}
2170 2171

	btrfs_update_iflags(inode);
C
Chris Mason 已提交
2172 2173 2174 2175 2176 2177 2178
	return;

make_bad:
	btrfs_free_path(path);
	make_bad_inode(inode);
}

C
Chris Mason 已提交
2179 2180 2181
/*
 * given a leaf and an inode, copy the inode fields into the leaf
 */
2182 2183
static void fill_inode_item(struct btrfs_trans_handle *trans,
			    struct extent_buffer *leaf,
2184
			    struct btrfs_inode_item *item,
C
Chris Mason 已提交
2185 2186
			    struct inode *inode)
{
2187 2188
	btrfs_set_inode_uid(leaf, item, inode->i_uid);
	btrfs_set_inode_gid(leaf, item, inode->i_gid);
2189
	btrfs_set_inode_size(leaf, item, BTRFS_I(inode)->disk_i_size);
2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207
	btrfs_set_inode_mode(leaf, item, inode->i_mode);
	btrfs_set_inode_nlink(leaf, item, inode->i_nlink);

	btrfs_set_timespec_sec(leaf, btrfs_inode_atime(item),
			       inode->i_atime.tv_sec);
	btrfs_set_timespec_nsec(leaf, btrfs_inode_atime(item),
				inode->i_atime.tv_nsec);

	btrfs_set_timespec_sec(leaf, btrfs_inode_mtime(item),
			       inode->i_mtime.tv_sec);
	btrfs_set_timespec_nsec(leaf, btrfs_inode_mtime(item),
				inode->i_mtime.tv_nsec);

	btrfs_set_timespec_sec(leaf, btrfs_inode_ctime(item),
			       inode->i_ctime.tv_sec);
	btrfs_set_timespec_nsec(leaf, btrfs_inode_ctime(item),
				inode->i_ctime.tv_nsec);

2208
	btrfs_set_inode_nbytes(leaf, item, inode_get_bytes(inode));
2209
	btrfs_set_inode_generation(leaf, item, BTRFS_I(inode)->generation);
2210
	btrfs_set_inode_sequence(leaf, item, BTRFS_I(inode)->sequence);
2211
	btrfs_set_inode_transid(leaf, item, trans->transid);
2212
	btrfs_set_inode_rdev(leaf, item, inode->i_rdev);
Y
Yan 已提交
2213
	btrfs_set_inode_flags(leaf, item, BTRFS_I(inode)->flags);
2214
	btrfs_set_inode_block_group(leaf, item, BTRFS_I(inode)->block_group);
C
Chris Mason 已提交
2215 2216
}

C
Chris Mason 已提交
2217 2218 2219
/*
 * copy everything in the in-memory inode into the btree.
 */
C
Chris Mason 已提交
2220 2221
noinline int btrfs_update_inode(struct btrfs_trans_handle *trans,
				struct btrfs_root *root, struct inode *inode)
C
Chris Mason 已提交
2222 2223 2224
{
	struct btrfs_inode_item *inode_item;
	struct btrfs_path *path;
2225
	struct extent_buffer *leaf;
C
Chris Mason 已提交
2226 2227 2228 2229
	int ret;

	path = btrfs_alloc_path();
	BUG_ON(!path);
2230
	path->leave_spinning = 1;
C
Chris Mason 已提交
2231 2232 2233 2234 2235 2236 2237 2238
	ret = btrfs_lookup_inode(trans, root, path,
				 &BTRFS_I(inode)->location, 1);
	if (ret) {
		if (ret > 0)
			ret = -ENOENT;
		goto failed;
	}

2239
	btrfs_unlock_up_safe(path, 1);
2240 2241
	leaf = path->nodes[0];
	inode_item = btrfs_item_ptr(leaf, path->slots[0],
C
Chris Mason 已提交
2242 2243
				  struct btrfs_inode_item);

2244
	fill_inode_item(trans, leaf, inode_item, inode);
2245
	btrfs_mark_buffer_dirty(leaf);
2246
	btrfs_set_inode_last_trans(trans, inode);
C
Chris Mason 已提交
2247 2248 2249 2250 2251 2252 2253
	ret = 0;
failed:
	btrfs_free_path(path);
	return ret;
}


C
Chris Mason 已提交
2254 2255 2256 2257 2258
/*
 * unlink helper that gets used here in inode.c and in the tree logging
 * recovery code.  It remove a link in a directory with a given name, and
 * also drops the back refs in the inode to the directory
 */
2259 2260 2261 2262
int btrfs_unlink_inode(struct btrfs_trans_handle *trans,
		       struct btrfs_root *root,
		       struct inode *dir, struct inode *inode,
		       const char *name, int name_len)
C
Chris Mason 已提交
2263 2264 2265
{
	struct btrfs_path *path;
	int ret = 0;
2266
	struct extent_buffer *leaf;
C
Chris Mason 已提交
2267
	struct btrfs_dir_item *di;
2268
	struct btrfs_key key;
2269
	u64 index;
C
Chris Mason 已提交
2270 2271

	path = btrfs_alloc_path();
2272 2273 2274 2275 2276
	if (!path) {
		ret = -ENOMEM;
		goto err;
	}

2277
	path->leave_spinning = 1;
C
Chris Mason 已提交
2278 2279 2280 2281 2282 2283 2284 2285 2286 2287
	di = btrfs_lookup_dir_item(trans, root, path, dir->i_ino,
				    name, name_len, -1);
	if (IS_ERR(di)) {
		ret = PTR_ERR(di);
		goto err;
	}
	if (!di) {
		ret = -ENOENT;
		goto err;
	}
2288 2289
	leaf = path->nodes[0];
	btrfs_dir_item_key_to_cpu(leaf, di, &key);
C
Chris Mason 已提交
2290
	ret = btrfs_delete_one_dir_name(trans, root, path, di);
2291 2292
	if (ret)
		goto err;
C
Chris Mason 已提交
2293 2294
	btrfs_release_path(root, path);

2295
	ret = btrfs_del_inode_ref(trans, root, name, name_len,
2296 2297
				  inode->i_ino,
				  dir->i_ino, &index);
2298
	if (ret) {
C
Chris Mason 已提交
2299
		printk(KERN_INFO "btrfs failed to delete reference to %.*s, "
2300
		       "inode %lu parent %lu\n", name_len, name,
2301
		       inode->i_ino, dir->i_ino);
2302 2303 2304
		goto err;
	}

C
Chris Mason 已提交
2305
	di = btrfs_lookup_dir_index_item(trans, root, path, dir->i_ino,
2306
					 index, name, name_len, -1);
C
Chris Mason 已提交
2307 2308 2309 2310 2311 2312 2313 2314 2315
	if (IS_ERR(di)) {
		ret = PTR_ERR(di);
		goto err;
	}
	if (!di) {
		ret = -ENOENT;
		goto err;
	}
	ret = btrfs_delete_one_dir_name(trans, root, path, di);
2316
	btrfs_release_path(root, path);
C
Chris Mason 已提交
2317

2318 2319
	ret = btrfs_del_inode_ref_in_log(trans, root, name, name_len,
					 inode, dir->i_ino);
C
Chris Mason 已提交
2320
	BUG_ON(ret != 0 && ret != -ENOENT);
2321 2322 2323 2324

	ret = btrfs_del_dir_entries_in_log(trans, root, name, name_len,
					   dir, index);
	BUG_ON(ret);
C
Chris Mason 已提交
2325 2326
err:
	btrfs_free_path(path);
2327 2328 2329 2330 2331 2332 2333 2334 2335
	if (ret)
		goto out;

	btrfs_i_size_write(dir, dir->i_size - name_len * 2);
	inode->i_ctime = dir->i_mtime = dir->i_ctime = CURRENT_TIME;
	btrfs_update_inode(trans, root, dir);
	btrfs_drop_nlink(inode);
	ret = btrfs_update_inode(trans, root, inode);
out:
C
Chris Mason 已提交
2336 2337 2338 2339 2340 2341 2342
	return ret;
}

static int btrfs_unlink(struct inode *dir, struct dentry *dentry)
{
	struct btrfs_root *root;
	struct btrfs_trans_handle *trans;
2343
	struct inode *inode = dentry->d_inode;
C
Chris Mason 已提交
2344
	int ret;
2345
	unsigned long nr = 0;
C
Chris Mason 已提交
2346 2347

	root = BTRFS_I(dir)->root;
2348

C
Chris Mason 已提交
2349
	trans = btrfs_start_transaction(root, 1);
2350

C
Chris Mason 已提交
2351
	btrfs_set_trans_block_group(trans, dir);
2352 2353 2354

	btrfs_record_unlink_dir(trans, dir, dentry->d_inode, 0);

2355 2356
	ret = btrfs_unlink_inode(trans, root, dir, dentry->d_inode,
				 dentry->d_name.name, dentry->d_name.len);
2357 2358 2359 2360

	if (inode->i_nlink == 0)
		ret = btrfs_orphan_add(trans, inode);

2361
	nr = trans->blocks_used;
2362

2363
	btrfs_end_transaction_throttle(trans, root);
2364
	btrfs_btree_balance_dirty(root, nr);
C
Chris Mason 已提交
2365 2366 2367 2368 2369 2370
	return ret;
}

static int btrfs_rmdir(struct inode *dir, struct dentry *dentry)
{
	struct inode *inode = dentry->d_inode;
2371
	int err = 0;
C
Chris Mason 已提交
2372 2373 2374
	int ret;
	struct btrfs_root *root = BTRFS_I(dir)->root;
	struct btrfs_trans_handle *trans;
2375
	unsigned long nr = 0;
C
Chris Mason 已提交
2376

2377 2378 2379 2380 2381 2382
	/*
	 * the FIRST_FREE_OBJECTID check makes sure we don't try to rmdir
	 * the root of a subvolume or snapshot
	 */
	if (inode->i_size > BTRFS_EMPTY_DIR_SIZE ||
	    inode->i_ino == BTRFS_FIRST_FREE_OBJECTID) {
Y
Yan 已提交
2383
		return -ENOTEMPTY;
2384
	}
Y
Yan 已提交
2385

C
Chris Mason 已提交
2386 2387 2388
	trans = btrfs_start_transaction(root, 1);
	btrfs_set_trans_block_group(trans, dir);

2389 2390 2391 2392
	err = btrfs_orphan_add(trans, inode);
	if (err)
		goto fail_trans;

C
Chris Mason 已提交
2393
	/* now the directory is empty */
2394 2395
	err = btrfs_unlink_inode(trans, root, dir, dentry->d_inode,
				 dentry->d_name.name, dentry->d_name.len);
C
Chris Mason 已提交
2396
	if (!err)
2397
		btrfs_i_size_write(inode, 0);
2398

2399
fail_trans:
2400
	nr = trans->blocks_used;
2401
	ret = btrfs_end_transaction_throttle(trans, root);
2402
	btrfs_btree_balance_dirty(root, nr);
2403

C
Chris Mason 已提交
2404 2405 2406 2407 2408
	if (ret && !err)
		err = ret;
	return err;
}

2409
#if 0
2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437
/*
 * when truncating bytes in a file, it is possible to avoid reading
 * the leaves that contain only checksum items.  This can be the
 * majority of the IO required to delete a large file, but it must
 * be done carefully.
 *
 * The keys in the level just above the leaves are checked to make sure
 * the lowest key in a given leaf is a csum key, and starts at an offset
 * after the new  size.
 *
 * Then the key for the next leaf is checked to make sure it also has
 * a checksum item for the same file.  If it does, we know our target leaf
 * contains only checksum items, and it can be safely freed without reading
 * it.
 *
 * This is just an optimization targeted at large files.  It may do
 * nothing.  It will return 0 unless things went badly.
 */
static noinline int drop_csum_leaves(struct btrfs_trans_handle *trans,
				     struct btrfs_root *root,
				     struct btrfs_path *path,
				     struct inode *inode, u64 new_size)
{
	struct btrfs_key key;
	int ret;
	int nritems;
	struct btrfs_key found_key;
	struct btrfs_key other_key;
Y
Yan Zheng 已提交
2438 2439 2440
	struct btrfs_leaf_ref *ref;
	u64 leaf_gen;
	u64 leaf_start;
2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533

	path->lowest_level = 1;
	key.objectid = inode->i_ino;
	key.type = BTRFS_CSUM_ITEM_KEY;
	key.offset = new_size;
again:
	ret = btrfs_search_slot(trans, root, &key, path, -1, 1);
	if (ret < 0)
		goto out;

	if (path->nodes[1] == NULL) {
		ret = 0;
		goto out;
	}
	ret = 0;
	btrfs_node_key_to_cpu(path->nodes[1], &found_key, path->slots[1]);
	nritems = btrfs_header_nritems(path->nodes[1]);

	if (!nritems)
		goto out;

	if (path->slots[1] >= nritems)
		goto next_node;

	/* did we find a key greater than anything we want to delete? */
	if (found_key.objectid > inode->i_ino ||
	   (found_key.objectid == inode->i_ino && found_key.type > key.type))
		goto out;

	/* we check the next key in the node to make sure the leave contains
	 * only checksum items.  This comparison doesn't work if our
	 * leaf is the last one in the node
	 */
	if (path->slots[1] + 1 >= nritems) {
next_node:
		/* search forward from the last key in the node, this
		 * will bring us into the next node in the tree
		 */
		btrfs_node_key_to_cpu(path->nodes[1], &found_key, nritems - 1);

		/* unlikely, but we inc below, so check to be safe */
		if (found_key.offset == (u64)-1)
			goto out;

		/* search_forward needs a path with locks held, do the
		 * search again for the original key.  It is possible
		 * this will race with a balance and return a path that
		 * we could modify, but this drop is just an optimization
		 * and is allowed to miss some leaves.
		 */
		btrfs_release_path(root, path);
		found_key.offset++;

		/* setup a max key for search_forward */
		other_key.offset = (u64)-1;
		other_key.type = key.type;
		other_key.objectid = key.objectid;

		path->keep_locks = 1;
		ret = btrfs_search_forward(root, &found_key, &other_key,
					   path, 0, 0);
		path->keep_locks = 0;
		if (ret || found_key.objectid != key.objectid ||
		    found_key.type != key.type) {
			ret = 0;
			goto out;
		}

		key.offset = found_key.offset;
		btrfs_release_path(root, path);
		cond_resched();
		goto again;
	}

	/* we know there's one more slot after us in the tree,
	 * read that key so we can verify it is also a checksum item
	 */
	btrfs_node_key_to_cpu(path->nodes[1], &other_key, path->slots[1] + 1);

	if (found_key.objectid < inode->i_ino)
		goto next_key;

	if (found_key.type != key.type || found_key.offset < new_size)
		goto next_key;

	/*
	 * if the key for the next leaf isn't a csum key from this objectid,
	 * we can't be sure there aren't good items inside this leaf.
	 * Bail out
	 */
	if (other_key.objectid != inode->i_ino || other_key.type != key.type)
		goto out;

Y
Yan Zheng 已提交
2534 2535
	leaf_start = btrfs_node_blockptr(path->nodes[1], path->slots[1]);
	leaf_gen = btrfs_node_ptr_generation(path->nodes[1], path->slots[1]);
2536 2537 2538 2539
	/*
	 * it is safe to delete this leaf, it contains only
	 * csum items from this inode at an offset >= new_size
	 */
Y
Yan Zheng 已提交
2540
	ret = btrfs_del_leaf(trans, root, path, leaf_start);
2541 2542
	BUG_ON(ret);

Y
Yan Zheng 已提交
2543 2544 2545 2546 2547 2548 2549 2550 2551
	if (root->ref_cows && leaf_gen < trans->transid) {
		ref = btrfs_alloc_leaf_ref(root, 0);
		if (ref) {
			ref->root_gen = root->root_key.offset;
			ref->bytenr = leaf_start;
			ref->owner = 0;
			ref->generation = leaf_gen;
			ref->nritems = 0;

2552 2553
			btrfs_sort_leaf_ref(ref);

Y
Yan Zheng 已提交
2554 2555 2556 2557 2558 2559 2560
			ret = btrfs_add_leaf_ref(root, ref, 0);
			WARN_ON(ret);
			btrfs_free_leaf_ref(root, ref);
		} else {
			WARN_ON(1);
		}
	}
2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578
next_key:
	btrfs_release_path(root, path);

	if (other_key.objectid == inode->i_ino &&
	    other_key.type == key.type && other_key.offset > key.offset) {
		key.offset = other_key.offset;
		cond_resched();
		goto again;
	}
	ret = 0;
out:
	/* fixup any changes we've made to the path */
	path->lowest_level = 0;
	path->keep_locks = 0;
	btrfs_release_path(root, path);
	return ret;
}

2579 2580
#endif

C
Chris Mason 已提交
2581 2582 2583
/*
 * this can truncate away extent items, csum items and directory items.
 * It starts at a high offset and removes keys until it can't find
C
Chris Mason 已提交
2584
 * any higher than new_size
C
Chris Mason 已提交
2585 2586 2587
 *
 * csum items that cross the new i_size are truncated to the new size
 * as well.
2588 2589 2590
 *
 * min_type is the minimum key type to truncate down to.  If set to 0, this
 * will kill all the items on this inode, including the INODE_ITEM_KEY.
C
Chris Mason 已提交
2591
 */
2592 2593 2594 2595
noinline int btrfs_truncate_inode_items(struct btrfs_trans_handle *trans,
					struct btrfs_root *root,
					struct inode *inode,
					u64 new_size, u32 min_type)
C
Chris Mason 已提交
2596 2597 2598 2599
{
	int ret;
	struct btrfs_path *path;
	struct btrfs_key key;
2600
	struct btrfs_key found_key;
2601
	u32 found_type = (u8)-1;
2602
	struct extent_buffer *leaf;
C
Chris Mason 已提交
2603 2604
	struct btrfs_file_extent_item *fi;
	u64 extent_start = 0;
2605
	u64 extent_num_bytes = 0;
2606
	u64 extent_offset = 0;
C
Chris Mason 已提交
2607 2608 2609
	u64 item_end = 0;
	int found_extent;
	int del_item;
2610 2611
	int pending_del_nr = 0;
	int pending_del_slot = 0;
2612
	int extent_type = -1;
2613
	int encoding;
2614
	u64 mask = root->sectorsize - 1;
C
Chris Mason 已提交
2615

2616
	if (root->ref_cows)
2617
		btrfs_drop_extent_cache(inode, new_size & (~mask), (u64)-1, 0);
C
Chris Mason 已提交
2618 2619
	path = btrfs_alloc_path();
	BUG_ON(!path);
J
Julia Lawall 已提交
2620
	path->reada = -1;
2621

C
Chris Mason 已提交
2622 2623 2624
	/* FIXME, add redo link to tree so we don't leak on crash */
	key.objectid = inode->i_ino;
	key.offset = (u64)-1;
2625 2626
	key.type = (u8)-1;

2627
search_again:
2628
	path->leave_spinning = 1;
2629
	ret = btrfs_search_slot(trans, root, &key, path, -1, 1);
C
Chris Mason 已提交
2630
	if (ret < 0)
2631
		goto error;
C
Chris Mason 已提交
2632

2633
	if (ret > 0) {
2634 2635 2636 2637 2638 2639 2640
		/* there are no items in the tree for us to truncate, we're
		 * done
		 */
		if (path->slots[0] == 0) {
			ret = 0;
			goto error;
		}
2641 2642 2643
		path->slots[0]--;
	}

C
Chris Mason 已提交
2644
	while (1) {
C
Chris Mason 已提交
2645
		fi = NULL;
2646 2647 2648
		leaf = path->nodes[0];
		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
		found_type = btrfs_key_type(&found_key);
2649
		encoding = 0;
C
Chris Mason 已提交
2650

2651
		if (found_key.objectid != inode->i_ino)
C
Chris Mason 已提交
2652
			break;
2653

2654
		if (found_type < min_type)
C
Chris Mason 已提交
2655 2656
			break;

2657
		item_end = found_key.offset;
C
Chris Mason 已提交
2658
		if (found_type == BTRFS_EXTENT_DATA_KEY) {
2659
			fi = btrfs_item_ptr(leaf, path->slots[0],
C
Chris Mason 已提交
2660
					    struct btrfs_file_extent_item);
2661
			extent_type = btrfs_file_extent_type(leaf, fi);
2662 2663 2664 2665
			encoding = btrfs_file_extent_compression(leaf, fi);
			encoding |= btrfs_file_extent_encryption(leaf, fi);
			encoding |= btrfs_file_extent_other_encoding(leaf, fi);

2666
			if (extent_type != BTRFS_FILE_EXTENT_INLINE) {
2667
				item_end +=
2668
				    btrfs_file_extent_num_bytes(leaf, fi);
2669 2670
			} else if (extent_type == BTRFS_FILE_EXTENT_INLINE) {
				item_end += btrfs_file_extent_inline_len(leaf,
C
Chris Mason 已提交
2671
									 fi);
C
Chris Mason 已提交
2672
			}
2673
			item_end--;
C
Chris Mason 已提交
2674
		}
2675
		if (item_end < new_size) {
C
Chris Mason 已提交
2676
			if (found_type == BTRFS_DIR_ITEM_KEY)
2677
				found_type = BTRFS_INODE_ITEM_KEY;
C
Chris Mason 已提交
2678
			else if (found_type == BTRFS_EXTENT_ITEM_KEY)
2679
				found_type = BTRFS_EXTENT_DATA_KEY;
C
Chris Mason 已提交
2680
			else if (found_type == BTRFS_EXTENT_DATA_KEY)
2681
				found_type = BTRFS_XATTR_ITEM_KEY;
C
Chris Mason 已提交
2682
			else if (found_type == BTRFS_XATTR_ITEM_KEY)
2683
				found_type = BTRFS_INODE_REF_KEY;
C
Chris Mason 已提交
2684
			else if (found_type)
2685
				found_type--;
C
Chris Mason 已提交
2686
			else
2687
				break;
2688
			btrfs_set_key_type(&key, found_type);
2689
			goto next;
C
Chris Mason 已提交
2690
		}
2691
		if (found_key.offset >= new_size)
C
Chris Mason 已提交
2692 2693 2694 2695 2696 2697
			del_item = 1;
		else
			del_item = 0;
		found_extent = 0;

		/* FIXME, shrink the extent if the ref count is only 1 */
2698 2699 2700 2701
		if (found_type != BTRFS_EXTENT_DATA_KEY)
			goto delete;

		if (extent_type != BTRFS_FILE_EXTENT_INLINE) {
C
Chris Mason 已提交
2702
			u64 num_dec;
2703
			extent_start = btrfs_file_extent_disk_bytenr(leaf, fi);
2704
			if (!del_item && !encoding) {
2705 2706
				u64 orig_num_bytes =
					btrfs_file_extent_num_bytes(leaf, fi);
2707
				extent_num_bytes = new_size -
2708
					found_key.offset + root->sectorsize - 1;
2709 2710
				extent_num_bytes = extent_num_bytes &
					~((u64)root->sectorsize - 1);
2711 2712 2713
				btrfs_set_file_extent_num_bytes(leaf, fi,
							 extent_num_bytes);
				num_dec = (orig_num_bytes -
C
Chris Mason 已提交
2714
					   extent_num_bytes);
2715
				if (root->ref_cows && extent_start != 0)
2716
					inode_sub_bytes(inode, num_dec);
2717
				btrfs_mark_buffer_dirty(leaf);
C
Chris Mason 已提交
2718
			} else {
2719 2720 2721
				extent_num_bytes =
					btrfs_file_extent_disk_num_bytes(leaf,
									 fi);
2722 2723 2724
				extent_offset = found_key.offset -
					btrfs_file_extent_offset(leaf, fi);

C
Chris Mason 已提交
2725
				/* FIXME blocksize != 4096 */
C
Chris Mason 已提交
2726
				num_dec = btrfs_file_extent_num_bytes(leaf, fi);
C
Chris Mason 已提交
2727 2728
				if (extent_start != 0) {
					found_extent = 1;
2729
					if (root->ref_cows)
2730
						inode_sub_bytes(inode, num_dec);
2731
				}
C
Chris Mason 已提交
2732
			}
C
Chris Mason 已提交
2733
		} else if (extent_type == BTRFS_FILE_EXTENT_INLINE) {
C
Chris Mason 已提交
2734 2735 2736 2737 2738 2739 2740 2741
			/*
			 * we can't truncate inline items that have had
			 * special encodings
			 */
			if (!del_item &&
			    btrfs_file_extent_compression(leaf, fi) == 0 &&
			    btrfs_file_extent_encryption(leaf, fi) == 0 &&
			    btrfs_file_extent_other_encoding(leaf, fi) == 0) {
2742 2743 2744
				u32 size = new_size - found_key.offset;

				if (root->ref_cows) {
2745 2746
					inode_sub_bytes(inode, item_end + 1 -
							new_size);
2747 2748 2749
				}
				size =
				    btrfs_file_extent_calc_inline_size(size);
C
Chris Mason 已提交
2750
				ret = btrfs_truncate_item(trans, root, path,
2751
							  size, 1);
C
Chris Mason 已提交
2752
				BUG_ON(ret);
2753
			} else if (root->ref_cows) {
2754 2755
				inode_sub_bytes(inode, item_end + 1 -
						found_key.offset);
C
Chris Mason 已提交
2756
			}
C
Chris Mason 已提交
2757
		}
2758
delete:
C
Chris Mason 已提交
2759
		if (del_item) {
2760 2761 2762 2763 2764 2765 2766 2767 2768 2769
			if (!pending_del_nr) {
				/* no pending yet, add ourselves */
				pending_del_slot = path->slots[0];
				pending_del_nr = 1;
			} else if (pending_del_nr &&
				   path->slots[0] + 1 == pending_del_slot) {
				/* hop on the pending chunk */
				pending_del_nr++;
				pending_del_slot = path->slots[0];
			} else {
C
Chris Mason 已提交
2770
				BUG();
2771
			}
C
Chris Mason 已提交
2772 2773 2774
		} else {
			break;
		}
2775
		if (found_extent && root->ref_cows) {
2776
			btrfs_set_path_blocking(path);
C
Chris Mason 已提交
2777
			ret = btrfs_free_extent(trans, root, extent_start,
2778 2779 2780
						extent_num_bytes, 0,
						btrfs_header_owner(leaf),
						inode->i_ino, extent_offset);
C
Chris Mason 已提交
2781 2782
			BUG_ON(ret);
		}
2783 2784 2785 2786 2787
next:
		if (path->slots[0] == 0) {
			if (pending_del_nr)
				goto del_pending;
			btrfs_release_path(root, path);
2788 2789
			if (found_type == BTRFS_INODE_ITEM_KEY)
				break;
2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805
			goto search_again;
		}

		path->slots[0]--;
		if (pending_del_nr &&
		    path->slots[0] + 1 != pending_del_slot) {
			struct btrfs_key debug;
del_pending:
			btrfs_item_key_to_cpu(path->nodes[0], &debug,
					      pending_del_slot);
			ret = btrfs_del_items(trans, root, path,
					      pending_del_slot,
					      pending_del_nr);
			BUG_ON(ret);
			pending_del_nr = 0;
			btrfs_release_path(root, path);
2806 2807
			if (found_type == BTRFS_INODE_ITEM_KEY)
				break;
2808 2809
			goto search_again;
		}
C
Chris Mason 已提交
2810 2811 2812
	}
	ret = 0;
error:
2813 2814 2815 2816
	if (pending_del_nr) {
		ret = btrfs_del_items(trans, root, path, pending_del_slot,
				      pending_del_nr);
	}
C
Chris Mason 已提交
2817 2818 2819 2820 2821 2822 2823 2824 2825 2826 2827
	btrfs_free_path(path);
	return ret;
}

/*
 * taken from block_truncate_page, but does cow as it zeros out
 * any bytes left in the last page in the file.
 */
static int btrfs_truncate_page(struct address_space *mapping, loff_t from)
{
	struct inode *inode = mapping->host;
2828
	struct btrfs_root *root = BTRFS_I(inode)->root;
2829 2830 2831
	struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree;
	struct btrfs_ordered_extent *ordered;
	char *kaddr;
2832
	u32 blocksize = root->sectorsize;
C
Chris Mason 已提交
2833 2834 2835 2836
	pgoff_t index = from >> PAGE_CACHE_SHIFT;
	unsigned offset = from & (PAGE_CACHE_SIZE-1);
	struct page *page;
	int ret = 0;
2837
	u64 page_start;
2838
	u64 page_end;
C
Chris Mason 已提交
2839 2840 2841 2842 2843

	if ((offset & (blocksize - 1)) == 0)
		goto out;

	ret = -ENOMEM;
2844
again:
C
Chris Mason 已提交
2845 2846 2847
	page = grab_cache_page(mapping, index);
	if (!page)
		goto out;
2848 2849 2850 2851

	page_start = page_offset(page);
	page_end = page_start + PAGE_CACHE_SIZE - 1;

C
Chris Mason 已提交
2852
	if (!PageUptodate(page)) {
C
Chris Mason 已提交
2853
		ret = btrfs_readpage(NULL, page);
C
Chris Mason 已提交
2854
		lock_page(page);
2855 2856 2857 2858 2859
		if (page->mapping != mapping) {
			unlock_page(page);
			page_cache_release(page);
			goto again;
		}
C
Chris Mason 已提交
2860 2861
		if (!PageUptodate(page)) {
			ret = -EIO;
2862
			goto out_unlock;
C
Chris Mason 已提交
2863 2864
		}
	}
2865
	wait_on_page_writeback(page);
2866 2867 2868 2869 2870 2871 2872 2873 2874

	lock_extent(io_tree, page_start, page_end, GFP_NOFS);
	set_page_extent_mapped(page);

	ordered = btrfs_lookup_ordered_extent(inode, page_start);
	if (ordered) {
		unlock_extent(io_tree, page_start, page_end, GFP_NOFS);
		unlock_page(page);
		page_cache_release(page);
2875
		btrfs_start_ordered_extent(inode, ordered, 1);
2876 2877 2878 2879
		btrfs_put_ordered_extent(ordered);
		goto again;
	}

2880
	btrfs_set_extent_delalloc(inode, page_start, page_end);
2881 2882 2883 2884 2885 2886 2887
	ret = 0;
	if (offset != PAGE_CACHE_SIZE) {
		kaddr = kmap(page);
		memset(kaddr + offset, 0, PAGE_CACHE_SIZE - offset);
		flush_dcache_page(page);
		kunmap(page);
	}
2888
	ClearPageChecked(page);
2889 2890
	set_page_dirty(page);
	unlock_extent(io_tree, page_start, page_end, GFP_NOFS);
C
Chris Mason 已提交
2891

2892
out_unlock:
C
Chris Mason 已提交
2893 2894 2895 2896 2897 2898
	unlock_page(page);
	page_cache_release(page);
out:
	return ret;
}

Y
Yan Zheng 已提交
2899
int btrfs_cont_expand(struct inode *inode, loff_t size)
C
Chris Mason 已提交
2900
{
Y
Yan Zheng 已提交
2901 2902 2903 2904 2905 2906 2907 2908 2909 2910
	struct btrfs_trans_handle *trans;
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree;
	struct extent_map *em;
	u64 mask = root->sectorsize - 1;
	u64 hole_start = (inode->i_size + mask) & ~mask;
	u64 block_end = (size + mask) & ~mask;
	u64 last_byte;
	u64 cur_offset;
	u64 hole_size;
C
Chris Mason 已提交
2911 2912
	int err;

Y
Yan Zheng 已提交
2913 2914 2915
	if (size <= hole_start)
		return 0;

J
Josef Bacik 已提交
2916
	err = btrfs_check_metadata_free_space(root);
C
Chris Mason 已提交
2917 2918 2919
	if (err)
		return err;

Y
Yan Zheng 已提交
2920
	btrfs_truncate_page(inode->i_mapping, inode->i_size);
C
Chris Mason 已提交
2921

Y
Yan Zheng 已提交
2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932
	while (1) {
		struct btrfs_ordered_extent *ordered;
		btrfs_wait_ordered_range(inode, hole_start,
					 block_end - hole_start);
		lock_extent(io_tree, hole_start, block_end - 1, GFP_NOFS);
		ordered = btrfs_lookup_ordered_extent(inode, hole_start);
		if (!ordered)
			break;
		unlock_extent(io_tree, hole_start, block_end - 1, GFP_NOFS);
		btrfs_put_ordered_extent(ordered);
	}
C
Chris Mason 已提交
2933

Y
Yan Zheng 已提交
2934 2935
	trans = btrfs_start_transaction(root, 1);
	btrfs_set_trans_block_group(trans, inode);
C
Chris Mason 已提交
2936

Y
Yan Zheng 已提交
2937 2938 2939 2940 2941 2942 2943 2944
	cur_offset = hole_start;
	while (1) {
		em = btrfs_get_extent(inode, NULL, 0, cur_offset,
				block_end - cur_offset, 0);
		BUG_ON(IS_ERR(em) || !em);
		last_byte = min(extent_map_end(em), block_end);
		last_byte = (last_byte + mask) & ~mask;
		if (test_bit(EXTENT_FLAG_VACANCY, &em->flags)) {
2945
			u64 hint_byte = 0;
Y
Yan Zheng 已提交
2946
			hole_size = last_byte - cur_offset;
2947 2948 2949
			err = btrfs_drop_extents(trans, root, inode,
						 cur_offset,
						 cur_offset + hole_size,
2950
						 block_end,
C
Chris Mason 已提交
2951
						 cur_offset, &hint_byte, 1);
2952 2953
			if (err)
				break;
Y
Yan Zheng 已提交
2954 2955 2956 2957 2958 2959 2960 2961 2962 2963 2964 2965
			err = btrfs_insert_file_extent(trans, root,
					inode->i_ino, cur_offset, 0,
					0, hole_size, 0, hole_size,
					0, 0, 0);
			btrfs_drop_extent_cache(inode, hole_start,
					last_byte - 1, 0);
		}
		free_extent_map(em);
		cur_offset = last_byte;
		if (err || cur_offset >= block_end)
			break;
	}
2966

Y
Yan Zheng 已提交
2967 2968 2969 2970
	btrfs_end_transaction(trans, root);
	unlock_extent(io_tree, hole_start, block_end - 1, GFP_NOFS);
	return err;
}
C
Chris Mason 已提交
2971

Y
Yan Zheng 已提交
2972 2973 2974 2975
static int btrfs_setattr(struct dentry *dentry, struct iattr *attr)
{
	struct inode *inode = dentry->d_inode;
	int err;
C
Chris Mason 已提交
2976

Y
Yan Zheng 已提交
2977 2978 2979
	err = inode_change_ok(inode, attr);
	if (err)
		return err;
C
Chris Mason 已提交
2980

2981 2982 2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995
	if (S_ISREG(inode->i_mode) && (attr->ia_valid & ATTR_SIZE)) {
		if (attr->ia_size > inode->i_size) {
			err = btrfs_cont_expand(inode, attr->ia_size);
			if (err)
				return err;
		} else if (inode->i_size > 0 &&
			   attr->ia_size == 0) {

			/* we're truncating a file that used to have good
			 * data down to zero.  Make sure it gets into
			 * the ordered flush list so that any new writes
			 * get down to disk quickly.
			 */
			BTRFS_I(inode)->ordered_data_close = 1;
		}
C
Chris Mason 已提交
2996
	}
Y
Yan Zheng 已提交
2997

C
Chris Mason 已提交
2998
	err = inode_setattr(inode, attr);
J
Josef Bacik 已提交
2999 3000 3001

	if (!err && ((attr->ia_valid & ATTR_MODE)))
		err = btrfs_acl_chmod(inode);
C
Chris Mason 已提交
3002 3003
	return err;
}
3004

C
Chris Mason 已提交
3005 3006 3007 3008
void btrfs_delete_inode(struct inode *inode)
{
	struct btrfs_trans_handle *trans;
	struct btrfs_root *root = BTRFS_I(inode)->root;
3009
	unsigned long nr;
C
Chris Mason 已提交
3010 3011 3012 3013
	int ret;

	truncate_inode_pages(&inode->i_data, 0);
	if (is_bad_inode(inode)) {
3014
		btrfs_orphan_del(NULL, inode);
C
Chris Mason 已提交
3015 3016
		goto no_delete;
	}
C
Chris Mason 已提交
3017
	btrfs_wait_ordered_range(inode, 0, (u64)-1);
3018

3019
	btrfs_i_size_write(inode, 0);
3020
	trans = btrfs_join_transaction(root, 1);
3021

C
Chris Mason 已提交
3022
	btrfs_set_trans_block_group(trans, inode);
3023
	ret = btrfs_truncate_inode_items(trans, root, inode, inode->i_size, 0);
3024 3025
	if (ret) {
		btrfs_orphan_del(NULL, inode);
3026
		goto no_delete_lock;
3027 3028 3029
	}

	btrfs_orphan_del(trans, inode);
3030

3031
	nr = trans->blocks_used;
3032
	clear_inode(inode);
3033

C
Chris Mason 已提交
3034
	btrfs_end_transaction(trans, root);
3035
	btrfs_btree_balance_dirty(root, nr);
C
Chris Mason 已提交
3036
	return;
3037 3038

no_delete_lock:
3039
	nr = trans->blocks_used;
3040
	btrfs_end_transaction(trans, root);
3041
	btrfs_btree_balance_dirty(root, nr);
C
Chris Mason 已提交
3042 3043 3044 3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056 3057
no_delete:
	clear_inode(inode);
}

/*
 * this returns the key found in the dir entry in the location pointer.
 * If no dir entries were found, location->objectid is 0.
 */
static int btrfs_inode_by_name(struct inode *dir, struct dentry *dentry,
			       struct btrfs_key *location)
{
	const char *name = dentry->d_name.name;
	int namelen = dentry->d_name.len;
	struct btrfs_dir_item *di;
	struct btrfs_path *path;
	struct btrfs_root *root = BTRFS_I(dir)->root;
Y
Yan 已提交
3058
	int ret = 0;
C
Chris Mason 已提交
3059 3060 3061

	path = btrfs_alloc_path();
	BUG_ON(!path);
3062

C
Chris Mason 已提交
3063 3064
	di = btrfs_lookup_dir_item(NULL, root, path, dir->i_ino, name,
				    namelen, 0);
Y
Yan 已提交
3065 3066
	if (IS_ERR(di))
		ret = PTR_ERR(di);
C
Chris Mason 已提交
3067 3068

	if (!di || IS_ERR(di))
3069
		goto out_err;
C
Chris Mason 已提交
3070

3071
	btrfs_dir_item_key_to_cpu(path->nodes[0], di, location);
C
Chris Mason 已提交
3072 3073 3074
out:
	btrfs_free_path(path);
	return ret;
3075 3076 3077
out_err:
	location->objectid = 0;
	goto out;
C
Chris Mason 已提交
3078 3079 3080 3081 3082 3083 3084 3085 3086
}

/*
 * when we hit a tree root in a directory, the btrfs part of the inode
 * needs to be changed to reflect the root directory of the tree root.  This
 * is kind of like crossing a mount point.
 */
static int fixup_tree_root_location(struct btrfs_root *root,
			     struct btrfs_key *location,
3087 3088
			     struct btrfs_root **sub_root,
			     struct dentry *dentry)
C
Chris Mason 已提交
3089 3090 3091 3092 3093 3094 3095 3096
{
	struct btrfs_root_item *ri;

	if (btrfs_key_type(location) != BTRFS_ROOT_ITEM_KEY)
		return 0;
	if (location->objectid == BTRFS_ROOT_TREE_OBJECTID)
		return 0;

3097 3098 3099
	*sub_root = btrfs_read_fs_root(root->fs_info, location,
					dentry->d_name.name,
					dentry->d_name.len);
C
Chris Mason 已提交
3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110
	if (IS_ERR(*sub_root))
		return PTR_ERR(*sub_root);

	ri = &(*sub_root)->root_item;
	location->objectid = btrfs_root_dirid(ri);
	btrfs_set_key_type(location, BTRFS_INODE_ITEM_KEY);
	location->offset = 0;

	return 0;
}

3111 3112 3113 3114
static void inode_tree_add(struct inode *inode)
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct btrfs_inode *entry;
3115 3116 3117 3118 3119 3120
	struct rb_node **p;
	struct rb_node *parent;

again:
	p = &root->inode_tree.rb_node;
	parent = NULL;
3121 3122 3123 3124 3125 3126 3127

	spin_lock(&root->inode_lock);
	while (*p) {
		parent = *p;
		entry = rb_entry(parent, struct btrfs_inode, rb_node);

		if (inode->i_ino < entry->vfs_inode.i_ino)
3128
			p = &parent->rb_left;
3129
		else if (inode->i_ino > entry->vfs_inode.i_ino)
3130
			p = &parent->rb_right;
3131 3132 3133
		else {
			WARN_ON(!(entry->vfs_inode.i_state &
				  (I_WILL_FREE | I_FREEING | I_CLEAR)));
3134 3135 3136 3137
			rb_erase(parent, &root->inode_tree);
			RB_CLEAR_NODE(parent);
			spin_unlock(&root->inode_lock);
			goto again;
3138 3139 3140 3141 3142 3143 3144 3145 3146 3147 3148
		}
	}
	rb_link_node(&BTRFS_I(inode)->rb_node, parent, p);
	rb_insert_color(&BTRFS_I(inode)->rb_node, &root->inode_tree);
	spin_unlock(&root->inode_lock);
}

static void inode_tree_del(struct inode *inode)
{
	struct btrfs_root *root = BTRFS_I(inode)->root;

3149
	spin_lock(&root->inode_lock);
3150 3151 3152 3153
	if (!RB_EMPTY_NODE(&BTRFS_I(inode)->rb_node)) {
		rb_erase(&BTRFS_I(inode)->rb_node, &root->inode_tree);
		RB_CLEAR_NODE(&BTRFS_I(inode)->rb_node);
	}
3154
	spin_unlock(&root->inode_lock);
3155 3156
}

3157
static noinline void init_btrfs_i(struct inode *inode)
C
Chris Mason 已提交
3158
{
3159 3160 3161
	struct btrfs_inode *bi = BTRFS_I(inode);

	bi->generation = 0;
3162
	bi->sequence = 0;
3163 3164 3165
	bi->last_trans = 0;
	bi->logged_trans = 0;
	bi->delalloc_bytes = 0;
J
Josef Bacik 已提交
3166
	bi->reserved_bytes = 0;
3167 3168 3169
	bi->disk_i_size = 0;
	bi->flags = 0;
	bi->index_cnt = (u64)-1;
3170
	bi->last_unlink_trans = 0;
3171
	bi->ordered_data_close = 0;
3172 3173
	extent_map_tree_init(&BTRFS_I(inode)->extent_tree, GFP_NOFS);
	extent_io_tree_init(&BTRFS_I(inode)->io_tree,
3174
			     inode->i_mapping, GFP_NOFS);
3175 3176
	extent_io_tree_init(&BTRFS_I(inode)->io_failure_tree,
			     inode->i_mapping, GFP_NOFS);
3177
	INIT_LIST_HEAD(&BTRFS_I(inode)->delalloc_inodes);
3178
	INIT_LIST_HEAD(&BTRFS_I(inode)->ordered_operations);
3179
	RB_CLEAR_NODE(&BTRFS_I(inode)->rb_node);
3180
	btrfs_ordered_inode_tree_init(&BTRFS_I(inode)->ordered_tree);
3181
	mutex_init(&BTRFS_I(inode)->extent_mutex);
3182 3183 3184 3185 3186 3187 3188 3189 3190
	mutex_init(&BTRFS_I(inode)->log_mutex);
}

static int btrfs_init_locked_inode(struct inode *inode, void *p)
{
	struct btrfs_iget_args *args = p;
	inode->i_ino = args->ino;
	init_btrfs_i(inode);
	BTRFS_I(inode)->root = args->root;
J
Josef Bacik 已提交
3191
	btrfs_set_inode_space_info(args->root, inode);
C
Chris Mason 已提交
3192 3193 3194 3195 3196 3197
	return 0;
}

static int btrfs_find_actor(struct inode *inode, void *opaque)
{
	struct btrfs_iget_args *args = opaque;
C
Chris Mason 已提交
3198 3199
	return args->ino == inode->i_ino &&
		args->root == BTRFS_I(inode)->root;
C
Chris Mason 已提交
3200 3201
}

3202 3203 3204
static struct inode *btrfs_iget_locked(struct super_block *s,
				       u64 objectid,
				       struct btrfs_root *root)
C
Chris Mason 已提交
3205 3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216
{
	struct inode *inode;
	struct btrfs_iget_args args;
	args.ino = objectid;
	args.root = root;

	inode = iget5_locked(s, objectid, btrfs_find_actor,
			     btrfs_init_locked_inode,
			     (void *)&args);
	return inode;
}

B
Balaji Rao 已提交
3217 3218 3219 3220
/* Get an inode object given its location and corresponding root.
 * Returns in *is_new if the inode was read from disk
 */
struct inode *btrfs_iget(struct super_block *s, struct btrfs_key *location,
3221
			 struct btrfs_root *root)
B
Balaji Rao 已提交
3222 3223 3224 3225 3226
{
	struct inode *inode;

	inode = btrfs_iget_locked(s, location->objectid, root);
	if (!inode)
3227
		return ERR_PTR(-ENOMEM);
B
Balaji Rao 已提交
3228 3229 3230 3231 3232

	if (inode->i_state & I_NEW) {
		BTRFS_I(inode)->root = root;
		memcpy(&BTRFS_I(inode)->location, location, sizeof(*location));
		btrfs_read_locked_inode(inode);
3233 3234

		inode_tree_add(inode);
B
Balaji Rao 已提交
3235 3236 3237 3238 3239 3240
		unlock_new_inode(inode);
	}

	return inode;
}

3241
struct inode *btrfs_lookup_dentry(struct inode *dir, struct dentry *dentry)
C
Chris Mason 已提交
3242
{
C
Chris Mason 已提交
3243
	struct inode *inode;
C
Chris Mason 已提交
3244 3245 3246 3247
	struct btrfs_inode *bi = BTRFS_I(dir);
	struct btrfs_root *root = bi->root;
	struct btrfs_root *sub_root = root;
	struct btrfs_key location;
3248
	int ret;
C
Chris Mason 已提交
3249 3250 3251

	if (dentry->d_name.len > BTRFS_NAME_LEN)
		return ERR_PTR(-ENAMETOOLONG);
3252

C
Chris Mason 已提交
3253
	ret = btrfs_inode_by_name(dir, dentry, &location);
3254

C
Chris Mason 已提交
3255 3256
	if (ret < 0)
		return ERR_PTR(ret);
3257

C
Chris Mason 已提交
3258 3259
	inode = NULL;
	if (location.objectid) {
3260 3261
		ret = fixup_tree_root_location(root, &location, &sub_root,
						dentry);
C
Chris Mason 已提交
3262 3263 3264 3265
		if (ret < 0)
			return ERR_PTR(ret);
		if (ret > 0)
			return ERR_PTR(-ENOENT);
3266
		inode = btrfs_iget(dir->i_sb, &location, sub_root);
B
Balaji Rao 已提交
3267 3268
		if (IS_ERR(inode))
			return ERR_CAST(inode);
C
Chris Mason 已提交
3269
	}
3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283
	return inode;
}

static struct dentry *btrfs_lookup(struct inode *dir, struct dentry *dentry,
				   struct nameidata *nd)
{
	struct inode *inode;

	if (dentry->d_name.len > BTRFS_NAME_LEN)
		return ERR_PTR(-ENAMETOOLONG);

	inode = btrfs_lookup_dentry(dir, dentry);
	if (IS_ERR(inode))
		return ERR_CAST(inode);
3284

C
Chris Mason 已提交
3285 3286 3287 3288 3289 3290 3291
	return d_splice_alias(inode, dentry);
}

static unsigned char btrfs_filetype_table[] = {
	DT_UNKNOWN, DT_REG, DT_DIR, DT_CHR, DT_BLK, DT_FIFO, DT_SOCK, DT_LNK
};

3292 3293
static int btrfs_real_readdir(struct file *filp, void *dirent,
			      filldir_t filldir)
C
Chris Mason 已提交
3294
{
3295
	struct inode *inode = filp->f_dentry->d_inode;
C
Chris Mason 已提交
3296 3297 3298 3299
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct btrfs_item *item;
	struct btrfs_dir_item *di;
	struct btrfs_key key;
3300
	struct btrfs_key found_key;
C
Chris Mason 已提交
3301 3302 3303
	struct btrfs_path *path;
	int ret;
	u32 nritems;
3304
	struct extent_buffer *leaf;
C
Chris Mason 已提交
3305 3306 3307 3308 3309 3310 3311 3312
	int slot;
	int advance;
	unsigned char d_type;
	int over = 0;
	u32 di_cur;
	u32 di_total;
	u32 di_len;
	int key_type = BTRFS_DIR_INDEX_KEY;
3313 3314 3315
	char tmp_name[32];
	char *name_ptr;
	int name_len;
C
Chris Mason 已提交
3316 3317 3318 3319

	/* FIXME, use a real flag for deciding about the key type */
	if (root->fs_info->tree_root == root)
		key_type = BTRFS_DIR_ITEM_KEY;
3320

3321 3322 3323 3324 3325 3326 3327 3328 3329 3330 3331
	/* special case for "." */
	if (filp->f_pos == 0) {
		over = filldir(dirent, ".", 1,
			       1, inode->i_ino,
			       DT_DIR);
		if (over)
			return 0;
		filp->f_pos = 1;
	}
	/* special case for .., just use the back ref */
	if (filp->f_pos == 1) {
3332
		u64 pino = parent_ino(filp->f_path.dentry);
3333
		over = filldir(dirent, "..", 2,
3334
			       2, pino, DT_DIR);
3335
		if (over)
3336
			return 0;
3337 3338
		filp->f_pos = 2;
	}
3339 3340 3341
	path = btrfs_alloc_path();
	path->reada = 2;

C
Chris Mason 已提交
3342 3343
	btrfs_set_key_type(&key, key_type);
	key.offset = filp->f_pos;
3344
	key.objectid = inode->i_ino;
3345

C
Chris Mason 已提交
3346 3347 3348 3349
	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
	if (ret < 0)
		goto err;
	advance = 0;
3350 3351

	while (1) {
3352 3353
		leaf = path->nodes[0];
		nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
3354 3355
		slot = path->slots[0];
		if (advance || slot >= nritems) {
3356
			if (slot >= nritems - 1) {
C
Chris Mason 已提交
3357 3358 3359
				ret = btrfs_next_leaf(root, path);
				if (ret)
					break;
3360 3361
				leaf = path->nodes[0];
				nritems = btrfs_header_nritems(leaf);
C
Chris Mason 已提交
3362 3363 3364 3365 3366 3367
				slot = path->slots[0];
			} else {
				slot++;
				path->slots[0]++;
			}
		}
3368

C
Chris Mason 已提交
3369
		advance = 1;
3370 3371 3372 3373
		item = btrfs_item_nr(leaf, slot);
		btrfs_item_key_to_cpu(leaf, &found_key, slot);

		if (found_key.objectid != key.objectid)
C
Chris Mason 已提交
3374
			break;
3375
		if (btrfs_key_type(&found_key) != key_type)
C
Chris Mason 已提交
3376
			break;
3377
		if (found_key.offset < filp->f_pos)
C
Chris Mason 已提交
3378
			continue;
3379 3380

		filp->f_pos = found_key.offset;
3381

C
Chris Mason 已提交
3382 3383
		di = btrfs_item_ptr(leaf, slot, struct btrfs_dir_item);
		di_cur = 0;
3384
		di_total = btrfs_item_size(leaf, item);
3385 3386

		while (di_cur < di_total) {
3387 3388 3389
			struct btrfs_key location;

			name_len = btrfs_dir_name_len(leaf, di);
3390
			if (name_len <= sizeof(tmp_name)) {
3391 3392 3393
				name_ptr = tmp_name;
			} else {
				name_ptr = kmalloc(name_len, GFP_NOFS);
3394 3395 3396 3397
				if (!name_ptr) {
					ret = -ENOMEM;
					goto err;
				}
3398 3399 3400 3401 3402 3403
			}
			read_extent_buffer(leaf, name_ptr,
					   (unsigned long)(di + 1), name_len);

			d_type = btrfs_filetype_table[btrfs_dir_type(leaf, di)];
			btrfs_dir_item_key_to_cpu(leaf, di, &location);
3404 3405 3406 3407 3408 3409 3410 3411 3412

			/* is this a reference to our own snapshot? If so
			 * skip it
			 */
			if (location.type == BTRFS_ROOT_ITEM_KEY &&
			    location.objectid == root->root_key.objectid) {
				over = 0;
				goto skip;
			}
3413
			over = filldir(dirent, name_ptr, name_len,
3414
				       found_key.offset, location.objectid,
C
Chris Mason 已提交
3415
				       d_type);
3416

3417
skip:
3418 3419 3420
			if (name_ptr != tmp_name)
				kfree(name_ptr);

C
Chris Mason 已提交
3421 3422
			if (over)
				goto nopos;
J
Josef Bacik 已提交
3423
			di_len = btrfs_dir_name_len(leaf, di) +
3424
				 btrfs_dir_data_len(leaf, di) + sizeof(*di);
C
Chris Mason 已提交
3425 3426 3427 3428
			di_cur += di_len;
			di = (struct btrfs_dir_item *)((char *)di + di_len);
		}
	}
3429 3430

	/* Reached end of directory/root. Bump pos past the last item. */
3431
	if (key_type == BTRFS_DIR_INDEX_KEY)
3432
		filp->f_pos = INT_LIMIT(off_t);
3433 3434
	else
		filp->f_pos++;
C
Chris Mason 已提交
3435 3436 3437 3438 3439 3440 3441 3442 3443 3444 3445 3446 3447
nopos:
	ret = 0;
err:
	btrfs_free_path(path);
	return ret;
}

int btrfs_write_inode(struct inode *inode, int wait)
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct btrfs_trans_handle *trans;
	int ret = 0;

Y
Yan Zheng 已提交
3448
	if (root->fs_info->btree_inode == inode)
3449 3450
		return 0;

C
Chris Mason 已提交
3451
	if (wait) {
3452
		trans = btrfs_join_transaction(root, 1);
C
Chris Mason 已提交
3453 3454 3455 3456 3457 3458 3459
		btrfs_set_trans_block_group(trans, inode);
		ret = btrfs_commit_transaction(trans, root);
	}
	return ret;
}

/*
3460
 * This is somewhat expensive, updating the tree every time the
C
Chris Mason 已提交
3461 3462 3463 3464 3465 3466 3467 3468 3469
 * inode changes.  But, it is most likely to find the inode in cache.
 * FIXME, needs more benchmarking...there are no reasons other than performance
 * to keep or drop this code.
 */
void btrfs_dirty_inode(struct inode *inode)
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct btrfs_trans_handle *trans;

3470
	trans = btrfs_join_transaction(root, 1);
C
Chris Mason 已提交
3471 3472 3473 3474 3475
	btrfs_set_trans_block_group(trans, inode);
	btrfs_update_inode(trans, root, inode);
	btrfs_end_transaction(trans, root);
}

C
Chris Mason 已提交
3476 3477 3478 3479 3480
/*
 * find the highest existing sequence number in a directory
 * and then set the in-memory index_cnt variable to reflect
 * free sequence numbers
 */
3481 3482 3483 3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502 3503 3504 3505 3506 3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530 3531 3532
static int btrfs_set_inode_index_count(struct inode *inode)
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct btrfs_key key, found_key;
	struct btrfs_path *path;
	struct extent_buffer *leaf;
	int ret;

	key.objectid = inode->i_ino;
	btrfs_set_key_type(&key, BTRFS_DIR_INDEX_KEY);
	key.offset = (u64)-1;

	path = btrfs_alloc_path();
	if (!path)
		return -ENOMEM;

	ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
	if (ret < 0)
		goto out;
	/* FIXME: we should be able to handle this */
	if (ret == 0)
		goto out;
	ret = 0;

	/*
	 * MAGIC NUMBER EXPLANATION:
	 * since we search a directory based on f_pos we have to start at 2
	 * since '.' and '..' have f_pos of 0 and 1 respectively, so everybody
	 * else has to start at 2
	 */
	if (path->slots[0] == 0) {
		BTRFS_I(inode)->index_cnt = 2;
		goto out;
	}

	path->slots[0]--;

	leaf = path->nodes[0];
	btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);

	if (found_key.objectid != inode->i_ino ||
	    btrfs_key_type(&found_key) != BTRFS_DIR_INDEX_KEY) {
		BTRFS_I(inode)->index_cnt = 2;
		goto out;
	}

	BTRFS_I(inode)->index_cnt = found_key.offset + 1;
out:
	btrfs_free_path(path);
	return ret;
}

C
Chris Mason 已提交
3533 3534 3535 3536
/*
 * helper to find a free sequence number in a given directory.  This current
 * code is very simple, later versions will do smarter things in the btree
 */
3537
int btrfs_set_inode_index(struct inode *dir, u64 *index)
3538 3539 3540 3541 3542
{
	int ret = 0;

	if (BTRFS_I(dir)->index_cnt == (u64)-1) {
		ret = btrfs_set_inode_index_count(dir);
C
Chris Mason 已提交
3543
		if (ret)
3544 3545 3546
			return ret;
	}

3547
	*index = BTRFS_I(dir)->index_cnt;
3548 3549 3550 3551 3552
	BTRFS_I(dir)->index_cnt++;

	return ret;
}

C
Chris Mason 已提交
3553 3554
static struct inode *btrfs_new_inode(struct btrfs_trans_handle *trans,
				     struct btrfs_root *root,
3555
				     struct inode *dir,
3556
				     const char *name, int name_len,
3557 3558
				     u64 ref_objectid, u64 objectid,
				     u64 alloc_hint, int mode, u64 *index)
C
Chris Mason 已提交
3559 3560
{
	struct inode *inode;
3561
	struct btrfs_inode_item *inode_item;
C
Chris Mason 已提交
3562
	struct btrfs_key *location;
3563
	struct btrfs_path *path;
3564 3565 3566 3567
	struct btrfs_inode_ref *ref;
	struct btrfs_key key[2];
	u32 sizes[2];
	unsigned long ptr;
C
Chris Mason 已提交
3568 3569 3570
	int ret;
	int owner;

3571 3572 3573
	path = btrfs_alloc_path();
	BUG_ON(!path);

C
Chris Mason 已提交
3574 3575 3576 3577
	inode = new_inode(root->fs_info->sb);
	if (!inode)
		return ERR_PTR(-ENOMEM);

3578
	if (dir) {
3579
		ret = btrfs_set_inode_index(dir, index);
3580 3581
		if (ret) {
			iput(inode);
3582
			return ERR_PTR(ret);
3583
		}
3584 3585 3586 3587 3588 3589
	}
	/*
	 * index_cnt is ignored for everything but a dir,
	 * btrfs_get_inode_index_count has an explanation for the magic
	 * number
	 */
3590
	init_btrfs_i(inode);
3591
	BTRFS_I(inode)->index_cnt = 2;
C
Chris Mason 已提交
3592
	BTRFS_I(inode)->root = root;
3593
	BTRFS_I(inode)->generation = trans->transid;
J
Josef Bacik 已提交
3594
	btrfs_set_inode_space_info(root, inode);
3595

C
Chris Mason 已提交
3596 3597 3598 3599
	if (mode & S_IFDIR)
		owner = 0;
	else
		owner = 1;
3600 3601
	BTRFS_I(inode)->block_group =
			btrfs_find_block_group(root, 0, alloc_hint, owner);
3602 3603 3604 3605 3606 3607 3608 3609 3610 3611 3612 3613

	key[0].objectid = objectid;
	btrfs_set_key_type(&key[0], BTRFS_INODE_ITEM_KEY);
	key[0].offset = 0;

	key[1].objectid = objectid;
	btrfs_set_key_type(&key[1], BTRFS_INODE_REF_KEY);
	key[1].offset = ref_objectid;

	sizes[0] = sizeof(struct btrfs_inode_item);
	sizes[1] = name_len + sizeof(*ref);

3614
	path->leave_spinning = 1;
3615 3616
	ret = btrfs_insert_empty_items(trans, root, path, key, sizes, 2);
	if (ret != 0)
3617 3618
		goto fail;

3619 3620 3621
	if (objectid > root->highest_inode)
		root->highest_inode = objectid;

C
Chris Mason 已提交
3622
	inode->i_uid = current_fsuid();
3623

3624
	if (dir && (dir->i_mode & S_ISGID)) {
3625 3626 3627 3628 3629 3630
		inode->i_gid = dir->i_gid;
		if (S_ISDIR(mode))
			mode |= S_ISGID;
	} else
		inode->i_gid = current_fsgid();

C
Chris Mason 已提交
3631 3632
	inode->i_mode = mode;
	inode->i_ino = objectid;
3633
	inode_set_bytes(inode, 0);
C
Chris Mason 已提交
3634
	inode->i_mtime = inode->i_atime = inode->i_ctime = CURRENT_TIME;
3635 3636
	inode_item = btrfs_item_ptr(path->nodes[0], path->slots[0],
				  struct btrfs_inode_item);
3637
	fill_inode_item(trans, path->nodes[0], inode_item, inode);
3638 3639 3640 3641

	ref = btrfs_item_ptr(path->nodes[0], path->slots[0] + 1,
			     struct btrfs_inode_ref);
	btrfs_set_inode_ref_name_len(path->nodes[0], ref, name_len);
3642
	btrfs_set_inode_ref_index(path->nodes[0], ref, *index);
3643 3644 3645
	ptr = (unsigned long)(ref + 1);
	write_extent_buffer(path->nodes[0], name, ptr, name_len);

3646 3647 3648
	btrfs_mark_buffer_dirty(path->nodes[0]);
	btrfs_free_path(path);

C
Chris Mason 已提交
3649 3650 3651 3652 3653
	location = &BTRFS_I(inode)->location;
	location->objectid = objectid;
	location->offset = 0;
	btrfs_set_key_type(location, BTRFS_INODE_ITEM_KEY);

3654 3655
	btrfs_inherit_iflags(inode, dir);

3656 3657 3658 3659 3660 3661 3662
	if ((mode & S_IFREG)) {
		if (btrfs_test_opt(root, NODATASUM))
			BTRFS_I(inode)->flags |= BTRFS_INODE_NODATASUM;
		if (btrfs_test_opt(root, NODATACOW))
			BTRFS_I(inode)->flags |= BTRFS_INODE_NODATACOW;
	}

C
Chris Mason 已提交
3663
	insert_inode_hash(inode);
3664
	inode_tree_add(inode);
C
Chris Mason 已提交
3665
	return inode;
3666
fail:
3667 3668
	if (dir)
		BTRFS_I(dir)->index_cnt--;
3669
	btrfs_free_path(path);
3670
	iput(inode);
3671
	return ERR_PTR(ret);
C
Chris Mason 已提交
3672 3673 3674 3675 3676 3677 3678
}

static inline u8 btrfs_inode_type(struct inode *inode)
{
	return btrfs_type_by_mode[(inode->i_mode & S_IFMT) >> S_SHIFT];
}

C
Chris Mason 已提交
3679 3680 3681 3682 3683 3684
/*
 * utility function to add 'inode' into 'parent_inode' with
 * a give name and a given sequence number.
 * if 'add_backref' is true, also insert a backref from the
 * inode to the parent directory.
 */
3685 3686 3687
int btrfs_add_link(struct btrfs_trans_handle *trans,
		   struct inode *parent_inode, struct inode *inode,
		   const char *name, int name_len, int add_backref, u64 index)
C
Chris Mason 已提交
3688 3689 3690
{
	int ret;
	struct btrfs_key key;
3691
	struct btrfs_root *root = BTRFS_I(parent_inode)->root;
3692

C
Chris Mason 已提交
3693 3694 3695 3696
	key.objectid = inode->i_ino;
	btrfs_set_key_type(&key, BTRFS_INODE_ITEM_KEY);
	key.offset = 0;

3697 3698
	ret = btrfs_insert_dir_item(trans, root, name, name_len,
				    parent_inode->i_ino,
3699
				    &key, btrfs_inode_type(inode),
3700
				    index);
C
Chris Mason 已提交
3701
	if (ret == 0) {
3702 3703
		if (add_backref) {
			ret = btrfs_insert_inode_ref(trans, root,
3704 3705 3706 3707
						     name, name_len,
						     inode->i_ino,
						     parent_inode->i_ino,
						     index);
3708
		}
3709
		btrfs_i_size_write(parent_inode, parent_inode->i_size +
3710
				   name_len * 2);
3711
		parent_inode->i_mtime = parent_inode->i_ctime = CURRENT_TIME;
3712
		ret = btrfs_update_inode(trans, root, parent_inode);
C
Chris Mason 已提交
3713 3714 3715 3716 3717
	}
	return ret;
}

static int btrfs_add_nondir(struct btrfs_trans_handle *trans,
3718
			    struct dentry *dentry, struct inode *inode,
3719
			    int backref, u64 index)
C
Chris Mason 已提交
3720
{
3721 3722 3723
	int err = btrfs_add_link(trans, dentry->d_parent->d_inode,
				 inode, dentry->d_name.name,
				 dentry->d_name.len, backref, index);
C
Chris Mason 已提交
3724 3725 3726 3727 3728 3729 3730 3731 3732
	if (!err) {
		d_instantiate(dentry, inode);
		return 0;
	}
	if (err > 0)
		err = -EEXIST;
	return err;
}

J
Josef Bacik 已提交
3733 3734 3735 3736 3737
static int btrfs_mknod(struct inode *dir, struct dentry *dentry,
			int mode, dev_t rdev)
{
	struct btrfs_trans_handle *trans;
	struct btrfs_root *root = BTRFS_I(dir)->root;
3738
	struct inode *inode = NULL;
J
Josef Bacik 已提交
3739 3740 3741
	int err;
	int drop_inode = 0;
	u64 objectid;
3742
	unsigned long nr = 0;
3743
	u64 index = 0;
J
Josef Bacik 已提交
3744 3745 3746 3747

	if (!new_valid_dev(rdev))
		return -EINVAL;

J
Josef Bacik 已提交
3748
	err = btrfs_check_metadata_free_space(root);
3749 3750 3751
	if (err)
		goto fail;

J
Josef Bacik 已提交
3752 3753 3754 3755 3756 3757 3758 3759 3760
	trans = btrfs_start_transaction(root, 1);
	btrfs_set_trans_block_group(trans, dir);

	err = btrfs_find_free_objectid(trans, root, dir->i_ino, &objectid);
	if (err) {
		err = -ENOSPC;
		goto out_unlock;
	}

3761
	inode = btrfs_new_inode(trans, root, dir, dentry->d_name.name,
3762 3763
				dentry->d_name.len,
				dentry->d_parent->d_inode->i_ino, objectid,
3764
				BTRFS_I(dir)->block_group, mode, &index);
J
Josef Bacik 已提交
3765 3766 3767 3768
	err = PTR_ERR(inode);
	if (IS_ERR(inode))
		goto out_unlock;

J
Jim Owens 已提交
3769
	err = btrfs_init_inode_security(inode, dir);
J
Josef Bacik 已提交
3770 3771 3772 3773 3774
	if (err) {
		drop_inode = 1;
		goto out_unlock;
	}

J
Josef Bacik 已提交
3775
	btrfs_set_trans_block_group(trans, inode);
3776
	err = btrfs_add_nondir(trans, dentry, inode, 0, index);
J
Josef Bacik 已提交
3777 3778 3779 3780 3781
	if (err)
		drop_inode = 1;
	else {
		inode->i_op = &btrfs_special_inode_operations;
		init_special_inode(inode, inode->i_mode, rdev);
3782
		btrfs_update_inode(trans, root, inode);
J
Josef Bacik 已提交
3783 3784 3785 3786
	}
	btrfs_update_inode_block_group(trans, inode);
	btrfs_update_inode_block_group(trans, dir);
out_unlock:
3787
	nr = trans->blocks_used;
3788
	btrfs_end_transaction_throttle(trans, root);
3789
fail:
J
Josef Bacik 已提交
3790 3791 3792 3793
	if (drop_inode) {
		inode_dec_link_count(inode);
		iput(inode);
	}
3794
	btrfs_btree_balance_dirty(root, nr);
J
Josef Bacik 已提交
3795 3796 3797
	return err;
}

C
Chris Mason 已提交
3798 3799 3800 3801 3802
static int btrfs_create(struct inode *dir, struct dentry *dentry,
			int mode, struct nameidata *nd)
{
	struct btrfs_trans_handle *trans;
	struct btrfs_root *root = BTRFS_I(dir)->root;
3803
	struct inode *inode = NULL;
C
Chris Mason 已提交
3804 3805
	int err;
	int drop_inode = 0;
3806
	unsigned long nr = 0;
C
Chris Mason 已提交
3807
	u64 objectid;
3808
	u64 index = 0;
C
Chris Mason 已提交
3809

J
Josef Bacik 已提交
3810
	err = btrfs_check_metadata_free_space(root);
3811 3812
	if (err)
		goto fail;
C
Chris Mason 已提交
3813 3814 3815 3816 3817 3818 3819 3820 3821
	trans = btrfs_start_transaction(root, 1);
	btrfs_set_trans_block_group(trans, dir);

	err = btrfs_find_free_objectid(trans, root, dir->i_ino, &objectid);
	if (err) {
		err = -ENOSPC;
		goto out_unlock;
	}

3822
	inode = btrfs_new_inode(trans, root, dir, dentry->d_name.name,
3823 3824
				dentry->d_name.len,
				dentry->d_parent->d_inode->i_ino,
3825 3826
				objectid, BTRFS_I(dir)->block_group, mode,
				&index);
C
Chris Mason 已提交
3827 3828 3829 3830
	err = PTR_ERR(inode);
	if (IS_ERR(inode))
		goto out_unlock;

J
Jim Owens 已提交
3831
	err = btrfs_init_inode_security(inode, dir);
J
Josef Bacik 已提交
3832 3833 3834 3835 3836
	if (err) {
		drop_inode = 1;
		goto out_unlock;
	}

C
Chris Mason 已提交
3837
	btrfs_set_trans_block_group(trans, inode);
3838
	err = btrfs_add_nondir(trans, dentry, inode, 0, index);
C
Chris Mason 已提交
3839 3840 3841 3842
	if (err)
		drop_inode = 1;
	else {
		inode->i_mapping->a_ops = &btrfs_aops;
C
Chris Mason 已提交
3843
		inode->i_mapping->backing_dev_info = &root->fs_info->bdi;
C
Chris Mason 已提交
3844 3845
		inode->i_fop = &btrfs_file_operations;
		inode->i_op = &btrfs_file_inode_operations;
3846
		BTRFS_I(inode)->io_tree.ops = &btrfs_extent_io_ops;
C
Chris Mason 已提交
3847 3848 3849 3850
	}
	btrfs_update_inode_block_group(trans, inode);
	btrfs_update_inode_block_group(trans, dir);
out_unlock:
3851
	nr = trans->blocks_used;
3852
	btrfs_end_transaction_throttle(trans, root);
3853
fail:
C
Chris Mason 已提交
3854 3855 3856 3857
	if (drop_inode) {
		inode_dec_link_count(inode);
		iput(inode);
	}
3858
	btrfs_btree_balance_dirty(root, nr);
C
Chris Mason 已提交
3859 3860 3861 3862 3863 3864 3865 3866 3867
	return err;
}

static int btrfs_link(struct dentry *old_dentry, struct inode *dir,
		      struct dentry *dentry)
{
	struct btrfs_trans_handle *trans;
	struct btrfs_root *root = BTRFS_I(dir)->root;
	struct inode *inode = old_dentry->d_inode;
3868
	u64 index;
3869
	unsigned long nr = 0;
C
Chris Mason 已提交
3870 3871 3872 3873 3874 3875
	int err;
	int drop_inode = 0;

	if (inode->i_nlink == 0)
		return -ENOENT;

3876
	btrfs_inc_nlink(inode);
J
Josef Bacik 已提交
3877
	err = btrfs_check_metadata_free_space(root);
3878 3879
	if (err)
		goto fail;
3880
	err = btrfs_set_inode_index(dir, &index);
3881 3882 3883
	if (err)
		goto fail;

C
Chris Mason 已提交
3884
	trans = btrfs_start_transaction(root, 1);
3885

C
Chris Mason 已提交
3886 3887
	btrfs_set_trans_block_group(trans, dir);
	atomic_inc(&inode->i_count);
3888

3889
	err = btrfs_add_nondir(trans, dentry, inode, 1, index);
3890

C
Chris Mason 已提交
3891 3892
	if (err)
		drop_inode = 1;
3893

C
Chris Mason 已提交
3894
	btrfs_update_inode_block_group(trans, dir);
3895
	err = btrfs_update_inode(trans, root, inode);
3896

3897 3898
	if (err)
		drop_inode = 1;
C
Chris Mason 已提交
3899

3900
	nr = trans->blocks_used;
3901 3902

	btrfs_log_new_name(trans, inode, NULL, dentry->d_parent);
3903
	btrfs_end_transaction_throttle(trans, root);
3904
fail:
C
Chris Mason 已提交
3905 3906 3907 3908
	if (drop_inode) {
		inode_dec_link_count(inode);
		iput(inode);
	}
3909
	btrfs_btree_balance_dirty(root, nr);
C
Chris Mason 已提交
3910 3911 3912 3913 3914
	return err;
}

static int btrfs_mkdir(struct inode *dir, struct dentry *dentry, int mode)
{
3915
	struct inode *inode = NULL;
C
Chris Mason 已提交
3916 3917 3918 3919
	struct btrfs_trans_handle *trans;
	struct btrfs_root *root = BTRFS_I(dir)->root;
	int err = 0;
	int drop_on_err = 0;
3920
	u64 objectid = 0;
3921
	u64 index = 0;
3922
	unsigned long nr = 1;
C
Chris Mason 已提交
3923

J
Josef Bacik 已提交
3924
	err = btrfs_check_metadata_free_space(root);
3925 3926 3927
	if (err)
		goto out_unlock;

C
Chris Mason 已提交
3928 3929
	trans = btrfs_start_transaction(root, 1);
	btrfs_set_trans_block_group(trans, dir);
3930

C
Chris Mason 已提交
3931 3932 3933 3934 3935 3936 3937 3938 3939 3940 3941
	if (IS_ERR(trans)) {
		err = PTR_ERR(trans);
		goto out_unlock;
	}

	err = btrfs_find_free_objectid(trans, root, dir->i_ino, &objectid);
	if (err) {
		err = -ENOSPC;
		goto out_unlock;
	}

3942
	inode = btrfs_new_inode(trans, root, dir, dentry->d_name.name,
3943 3944
				dentry->d_name.len,
				dentry->d_parent->d_inode->i_ino, objectid,
3945 3946
				BTRFS_I(dir)->block_group, S_IFDIR | mode,
				&index);
C
Chris Mason 已提交
3947 3948 3949 3950
	if (IS_ERR(inode)) {
		err = PTR_ERR(inode);
		goto out_fail;
	}
3951

C
Chris Mason 已提交
3952
	drop_on_err = 1;
J
Josef Bacik 已提交
3953

J
Jim Owens 已提交
3954
	err = btrfs_init_inode_security(inode, dir);
J
Josef Bacik 已提交
3955 3956 3957
	if (err)
		goto out_fail;

C
Chris Mason 已提交
3958 3959 3960 3961
	inode->i_op = &btrfs_dir_inode_operations;
	inode->i_fop = &btrfs_dir_file_operations;
	btrfs_set_trans_block_group(trans, inode);

3962
	btrfs_i_size_write(inode, 0);
C
Chris Mason 已提交
3963 3964 3965
	err = btrfs_update_inode(trans, root, inode);
	if (err)
		goto out_fail;
3966

3967 3968 3969
	err = btrfs_add_link(trans, dentry->d_parent->d_inode,
				 inode, dentry->d_name.name,
				 dentry->d_name.len, 0, index);
C
Chris Mason 已提交
3970 3971
	if (err)
		goto out_fail;
3972

C
Chris Mason 已提交
3973 3974 3975 3976 3977 3978
	d_instantiate(dentry, inode);
	drop_on_err = 0;
	btrfs_update_inode_block_group(trans, inode);
	btrfs_update_inode_block_group(trans, dir);

out_fail:
3979
	nr = trans->blocks_used;
3980
	btrfs_end_transaction_throttle(trans, root);
3981

C
Chris Mason 已提交
3982 3983 3984
out_unlock:
	if (drop_on_err)
		iput(inode);
3985
	btrfs_btree_balance_dirty(root, nr);
C
Chris Mason 已提交
3986 3987 3988
	return err;
}

C
Chris Mason 已提交
3989 3990 3991 3992
/* helper for btfs_get_extent.  Given an existing extent in the tree,
 * and an extent that you want to insert, deal with overlap and insert
 * the new extent into the tree.
 */
3993 3994
static int merge_extent_mapping(struct extent_map_tree *em_tree,
				struct extent_map *existing,
3995 3996
				struct extent_map *em,
				u64 map_start, u64 map_len)
3997 3998 3999
{
	u64 start_diff;

4000 4001 4002 4003
	BUG_ON(map_start < em->start || map_start >= extent_map_end(em));
	start_diff = map_start - em->start;
	em->start = map_start;
	em->len = map_len;
C
Chris Mason 已提交
4004 4005
	if (em->block_start < EXTENT_MAP_LAST_BYTE &&
	    !test_bit(EXTENT_FLAG_COMPRESSED, &em->flags)) {
4006
		em->block_start += start_diff;
C
Chris Mason 已提交
4007 4008
		em->block_len -= start_diff;
	}
4009
	return add_extent_mapping(em_tree, em);
4010 4011
}

C
Chris Mason 已提交
4012 4013 4014 4015 4016 4017 4018 4019 4020 4021 4022 4023 4024 4025 4026 4027 4028 4029 4030 4031 4032
static noinline int uncompress_inline(struct btrfs_path *path,
				      struct inode *inode, struct page *page,
				      size_t pg_offset, u64 extent_offset,
				      struct btrfs_file_extent_item *item)
{
	int ret;
	struct extent_buffer *leaf = path->nodes[0];
	char *tmp;
	size_t max_size;
	unsigned long inline_size;
	unsigned long ptr;

	WARN_ON(pg_offset != 0);
	max_size = btrfs_file_extent_ram_bytes(leaf, item);
	inline_size = btrfs_file_extent_inline_item_len(leaf,
					btrfs_item_nr(leaf, path->slots[0]));
	tmp = kmalloc(inline_size, GFP_NOFS);
	ptr = btrfs_file_extent_inline_start(item);

	read_extent_buffer(leaf, tmp, ptr, inline_size);

4033
	max_size = min_t(unsigned long, PAGE_CACHE_SIZE, max_size);
C
Chris Mason 已提交
4034 4035 4036 4037 4038 4039 4040 4041 4042 4043 4044 4045 4046 4047
	ret = btrfs_zlib_decompress(tmp, page, extent_offset,
				    inline_size, max_size);
	if (ret) {
		char *kaddr = kmap_atomic(page, KM_USER0);
		unsigned long copy_size = min_t(u64,
				  PAGE_CACHE_SIZE - pg_offset,
				  max_size - extent_offset);
		memset(kaddr + pg_offset, 0, copy_size);
		kunmap_atomic(kaddr, KM_USER0);
	}
	kfree(tmp);
	return 0;
}

C
Chris Mason 已提交
4048 4049
/*
 * a bit scary, this does extent mapping from logical file offset to the disk.
C
Chris Mason 已提交
4050 4051
 * the ugly parts come from merging extents from the disk with the in-ram
 * representation.  This gets more complex because of the data=ordered code,
C
Chris Mason 已提交
4052 4053 4054 4055
 * where the in-ram extents might be locked pending data=ordered completion.
 *
 * This also copies inline extents directly into the page.
 */
C
Chris Mason 已提交
4056

4057
struct extent_map *btrfs_get_extent(struct inode *inode, struct page *page,
4058
				    size_t pg_offset, u64 start, u64 len,
4059 4060 4061 4062
				    int create)
{
	int ret;
	int err = 0;
4063
	u64 bytenr;
4064 4065 4066 4067
	u64 extent_start = 0;
	u64 extent_end = 0;
	u64 objectid = inode->i_ino;
	u32 found_type;
4068
	struct btrfs_path *path = NULL;
4069 4070
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct btrfs_file_extent_item *item;
4071 4072
	struct extent_buffer *leaf;
	struct btrfs_key found_key;
4073 4074
	struct extent_map *em = NULL;
	struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree;
4075
	struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree;
4076
	struct btrfs_trans_handle *trans = NULL;
C
Chris Mason 已提交
4077
	int compressed;
4078 4079

again:
4080
	read_lock(&em_tree->lock);
4081
	em = lookup_extent_mapping(em_tree, start, len);
4082 4083
	if (em)
		em->bdev = root->fs_info->fs_devices->latest_bdev;
4084
	read_unlock(&em_tree->lock);
4085

4086
	if (em) {
4087 4088 4089
		if (em->start > start || em->start + em->len <= start)
			free_extent_map(em);
		else if (em->block_start == EXTENT_MAP_INLINE && page)
4090 4091 4092
			free_extent_map(em);
		else
			goto out;
4093
	}
4094
	em = alloc_extent_map(GFP_NOFS);
4095
	if (!em) {
4096 4097
		err = -ENOMEM;
		goto out;
4098
	}
4099
	em->bdev = root->fs_info->fs_devices->latest_bdev;
4100
	em->start = EXTENT_MAP_HOLE;
4101
	em->orig_start = EXTENT_MAP_HOLE;
4102
	em->len = (u64)-1;
C
Chris Mason 已提交
4103
	em->block_len = (u64)-1;
4104 4105 4106 4107 4108 4109

	if (!path) {
		path = btrfs_alloc_path();
		BUG_ON(!path);
	}

4110 4111
	ret = btrfs_lookup_file_extent(trans, root, path,
				       objectid, start, trans != NULL);
4112 4113 4114 4115 4116 4117 4118 4119 4120 4121 4122
	if (ret < 0) {
		err = ret;
		goto out;
	}

	if (ret != 0) {
		if (path->slots[0] == 0)
			goto not_found;
		path->slots[0]--;
	}

4123 4124
	leaf = path->nodes[0];
	item = btrfs_item_ptr(leaf, path->slots[0],
4125 4126
			      struct btrfs_file_extent_item);
	/* are we inside the extent that was found? */
4127 4128 4129
	btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
	found_type = btrfs_key_type(&found_key);
	if (found_key.objectid != objectid ||
4130 4131 4132 4133
	    found_type != BTRFS_EXTENT_DATA_KEY) {
		goto not_found;
	}

4134 4135
	found_type = btrfs_file_extent_type(leaf, item);
	extent_start = found_key.offset;
C
Chris Mason 已提交
4136
	compressed = btrfs_file_extent_compression(leaf, item);
Y
Yan Zheng 已提交
4137 4138
	if (found_type == BTRFS_FILE_EXTENT_REG ||
	    found_type == BTRFS_FILE_EXTENT_PREALLOC) {
4139
		extent_end = extent_start +
4140
		       btrfs_file_extent_num_bytes(leaf, item);
Y
Yan Zheng 已提交
4141 4142 4143 4144 4145 4146 4147 4148 4149 4150 4151 4152 4153 4154
	} else if (found_type == BTRFS_FILE_EXTENT_INLINE) {
		size_t size;
		size = btrfs_file_extent_inline_len(leaf, item);
		extent_end = (extent_start + size + root->sectorsize - 1) &
			~((u64)root->sectorsize - 1);
	}

	if (start >= extent_end) {
		path->slots[0]++;
		if (path->slots[0] >= btrfs_header_nritems(leaf)) {
			ret = btrfs_next_leaf(root, path);
			if (ret < 0) {
				err = ret;
				goto out;
4155
			}
Y
Yan Zheng 已提交
4156 4157 4158
			if (ret > 0)
				goto not_found;
			leaf = path->nodes[0];
4159
		}
Y
Yan Zheng 已提交
4160 4161 4162 4163 4164 4165 4166 4167 4168 4169 4170
		btrfs_item_key_to_cpu(leaf, &found_key, path->slots[0]);
		if (found_key.objectid != objectid ||
		    found_key.type != BTRFS_EXTENT_DATA_KEY)
			goto not_found;
		if (start + len <= found_key.offset)
			goto not_found;
		em->start = start;
		em->len = found_key.offset - start;
		goto not_found_em;
	}

Y
Yan Zheng 已提交
4171 4172
	if (found_type == BTRFS_FILE_EXTENT_REG ||
	    found_type == BTRFS_FILE_EXTENT_PREALLOC) {
Y
Yan Zheng 已提交
4173 4174
		em->start = extent_start;
		em->len = extent_end - extent_start;
4175 4176
		em->orig_start = extent_start -
				 btrfs_file_extent_offset(leaf, item);
4177 4178
		bytenr = btrfs_file_extent_disk_bytenr(leaf, item);
		if (bytenr == 0) {
4179
			em->block_start = EXTENT_MAP_HOLE;
4180 4181
			goto insert;
		}
C
Chris Mason 已提交
4182 4183 4184 4185 4186 4187 4188 4189 4190
		if (compressed) {
			set_bit(EXTENT_FLAG_COMPRESSED, &em->flags);
			em->block_start = bytenr;
			em->block_len = btrfs_file_extent_disk_num_bytes(leaf,
									 item);
		} else {
			bytenr += btrfs_file_extent_offset(leaf, item);
			em->block_start = bytenr;
			em->block_len = em->len;
Y
Yan Zheng 已提交
4191 4192
			if (found_type == BTRFS_FILE_EXTENT_PREALLOC)
				set_bit(EXTENT_FLAG_PREALLOC, &em->flags);
C
Chris Mason 已提交
4193
		}
4194 4195
		goto insert;
	} else if (found_type == BTRFS_FILE_EXTENT_INLINE) {
4196
		unsigned long ptr;
4197
		char *map;
4198 4199 4200
		size_t size;
		size_t extent_offset;
		size_t copy_size;
4201

4202
		em->block_start = EXTENT_MAP_INLINE;
C
Chris Mason 已提交
4203
		if (!page || create) {
4204
			em->start = extent_start;
Y
Yan Zheng 已提交
4205
			em->len = extent_end - extent_start;
4206 4207
			goto out;
		}
4208

Y
Yan Zheng 已提交
4209 4210
		size = btrfs_file_extent_inline_len(leaf, item);
		extent_offset = page_offset(page) + pg_offset - extent_start;
4211
		copy_size = min_t(u64, PAGE_CACHE_SIZE - pg_offset,
4212 4213
				size - extent_offset);
		em->start = extent_start + extent_offset;
4214 4215
		em->len = (copy_size + root->sectorsize - 1) &
			~((u64)root->sectorsize - 1);
4216
		em->orig_start = EXTENT_MAP_INLINE;
C
Chris Mason 已提交
4217 4218
		if (compressed)
			set_bit(EXTENT_FLAG_COMPRESSED, &em->flags);
4219
		ptr = btrfs_file_extent_inline_start(item) + extent_offset;
4220
		if (create == 0 && !PageUptodate(page)) {
C
Chris Mason 已提交
4221 4222 4223 4224 4225 4226 4227 4228 4229 4230
			if (btrfs_file_extent_compression(leaf, item) ==
			    BTRFS_COMPRESS_ZLIB) {
				ret = uncompress_inline(path, inode, page,
							pg_offset,
							extent_offset, item);
				BUG_ON(ret);
			} else {
				map = kmap(page);
				read_extent_buffer(leaf, map + pg_offset, ptr,
						   copy_size);
4231 4232 4233 4234 4235
				if (pg_offset + copy_size < PAGE_CACHE_SIZE) {
					memset(map + pg_offset + copy_size, 0,
					       PAGE_CACHE_SIZE - pg_offset -
					       copy_size);
				}
C
Chris Mason 已提交
4236 4237
				kunmap(page);
			}
4238 4239 4240 4241 4242 4243 4244
			flush_dcache_page(page);
		} else if (create && PageUptodate(page)) {
			if (!trans) {
				kunmap(page);
				free_extent_map(em);
				em = NULL;
				btrfs_release_path(root, path);
4245
				trans = btrfs_join_transaction(root, 1);
4246 4247
				goto again;
			}
C
Chris Mason 已提交
4248
			map = kmap(page);
4249
			write_extent_buffer(leaf, map + pg_offset, ptr,
4250
					    copy_size);
C
Chris Mason 已提交
4251
			kunmap(page);
4252
			btrfs_mark_buffer_dirty(leaf);
4253
		}
4254 4255
		set_extent_uptodate(io_tree, em->start,
				    extent_map_end(em) - 1, GFP_NOFS);
4256 4257
		goto insert;
	} else {
C
Chris Mason 已提交
4258
		printk(KERN_ERR "btrfs unknown found_type %d\n", found_type);
4259 4260 4261 4262
		WARN_ON(1);
	}
not_found:
	em->start = start;
4263
	em->len = len;
4264
not_found_em:
4265
	em->block_start = EXTENT_MAP_HOLE;
Y
Yan Zheng 已提交
4266
	set_bit(EXTENT_FLAG_VACANCY, &em->flags);
4267 4268
insert:
	btrfs_release_path(root, path);
4269
	if (em->start > start || extent_map_end(em) <= start) {
C
Chris Mason 已提交
4270 4271 4272 4273 4274
		printk(KERN_ERR "Btrfs: bad extent! em: [%llu %llu] passed "
		       "[%llu %llu]\n", (unsigned long long)em->start,
		       (unsigned long long)em->len,
		       (unsigned long long)start,
		       (unsigned long long)len);
4275 4276 4277
		err = -EIO;
		goto out;
	}
4278 4279

	err = 0;
4280
	write_lock(&em_tree->lock);
4281
	ret = add_extent_mapping(em_tree, em);
4282 4283 4284 4285
	/* it is possible that someone inserted the extent into the tree
	 * while we had the lock dropped.  It is also possible that
	 * an overlapping map exists in the tree
	 */
4286
	if (ret == -EEXIST) {
4287
		struct extent_map *existing;
4288 4289 4290

		ret = 0;

4291
		existing = lookup_extent_mapping(em_tree, start, len);
4292 4293 4294 4295 4296
		if (existing && (existing->start > start ||
		    existing->start + existing->len <= start)) {
			free_extent_map(existing);
			existing = NULL;
		}
4297 4298 4299 4300 4301
		if (!existing) {
			existing = lookup_extent_mapping(em_tree, em->start,
							 em->len);
			if (existing) {
				err = merge_extent_mapping(em_tree, existing,
4302 4303
							   em, start,
							   root->sectorsize);
4304 4305 4306 4307 4308 4309 4310 4311 4312 4313 4314 4315 4316
				free_extent_map(existing);
				if (err) {
					free_extent_map(em);
					em = NULL;
				}
			} else {
				err = -EIO;
				free_extent_map(em);
				em = NULL;
			}
		} else {
			free_extent_map(em);
			em = existing;
4317
			err = 0;
4318 4319
		}
	}
4320
	write_unlock(&em_tree->lock);
4321
out:
4322 4323
	if (path)
		btrfs_free_path(path);
4324 4325
	if (trans) {
		ret = btrfs_end_transaction(trans, root);
C
Chris Mason 已提交
4326
		if (!err)
4327 4328 4329 4330 4331 4332 4333 4334 4335
			err = ret;
	}
	if (err) {
		free_extent_map(em);
		return ERR_PTR(err);
	}
	return em;
}

4336 4337 4338 4339
static ssize_t btrfs_direct_IO(int rw, struct kiocb *iocb,
			const struct iovec *iov, loff_t offset,
			unsigned long nr_segs)
{
4340
	return -EINVAL;
4341 4342
}

Y
Yehuda Sadeh 已提交
4343 4344 4345 4346 4347 4348
static int btrfs_fiemap(struct inode *inode, struct fiemap_extent_info *fieinfo,
		__u64 start, __u64 len)
{
	return extent_fiemap(inode, fieinfo, start, len, btrfs_get_extent);
}

4349
int btrfs_readpage(struct file *file, struct page *page)
C
Chris Mason 已提交
4350
{
4351 4352
	struct extent_io_tree *tree;
	tree = &BTRFS_I(page->mapping->host)->io_tree;
4353
	return extent_read_full_page(tree, page, btrfs_get_extent);
C
Chris Mason 已提交
4354
}
4355

4356
static int btrfs_writepage(struct page *page, struct writeback_control *wbc)
C
Chris Mason 已提交
4357
{
4358
	struct extent_io_tree *tree;
4359 4360 4361 4362 4363 4364 4365


	if (current->flags & PF_MEMALLOC) {
		redirty_page_for_writepage(wbc, page);
		unlock_page(page);
		return 0;
	}
4366
	tree = &BTRFS_I(page->mapping->host)->io_tree;
4367
	return extent_write_full_page(tree, page, btrfs_get_extent, wbc);
C
Chris Mason 已提交
4368 4369
}

4370 4371
int btrfs_writepages(struct address_space *mapping,
		     struct writeback_control *wbc)
C
Chris Mason 已提交
4372
{
4373
	struct extent_io_tree *tree;
4374

4375
	tree = &BTRFS_I(mapping->host)->io_tree;
C
Chris Mason 已提交
4376 4377 4378
	return extent_writepages(tree, mapping, btrfs_get_extent, wbc);
}

C
Chris Mason 已提交
4379 4380 4381 4382
static int
btrfs_readpages(struct file *file, struct address_space *mapping,
		struct list_head *pages, unsigned nr_pages)
{
4383 4384
	struct extent_io_tree *tree;
	tree = &BTRFS_I(mapping->host)->io_tree;
C
Chris Mason 已提交
4385 4386 4387
	return extent_readpages(tree, mapping, pages, nr_pages,
				btrfs_get_extent);
}
4388
static int __btrfs_releasepage(struct page *page, gfp_t gfp_flags)
C
Chris Mason 已提交
4389
{
4390 4391
	struct extent_io_tree *tree;
	struct extent_map_tree *map;
4392
	int ret;
4393

4394 4395
	tree = &BTRFS_I(page->mapping->host)->io_tree;
	map = &BTRFS_I(page->mapping->host)->extent_tree;
4396
	ret = try_release_extent_mapping(map, tree, page, gfp_flags);
4397 4398 4399 4400
	if (ret == 1) {
		ClearPagePrivate(page);
		set_page_private(page, 0);
		page_cache_release(page);
C
Chris Mason 已提交
4401
	}
4402
	return ret;
C
Chris Mason 已提交
4403 4404
}

4405 4406
static int btrfs_releasepage(struct page *page, gfp_t gfp_flags)
{
4407 4408
	if (PageWriteback(page) || PageDirty(page))
		return 0;
4409
	return __btrfs_releasepage(page, gfp_flags & GFP_NOFS);
4410 4411
}

4412
static void btrfs_invalidatepage(struct page *page, unsigned long offset)
C
Chris Mason 已提交
4413
{
4414
	struct extent_io_tree *tree;
4415 4416 4417
	struct btrfs_ordered_extent *ordered;
	u64 page_start = page_offset(page);
	u64 page_end = page_start + PAGE_CACHE_SIZE - 1;
C
Chris Mason 已提交
4418

4419 4420 4421 4422 4423 4424 4425 4426

	/*
	 * we have the page locked, so new writeback can't start,
	 * and the dirty bit won't be cleared while we are here.
	 *
	 * Wait for IO on this page so that we can safely clear
	 * the PagePrivate2 bit and do ordered accounting
	 */
4427
	wait_on_page_writeback(page);
4428

4429
	tree = &BTRFS_I(page->mapping->host)->io_tree;
4430 4431 4432 4433 4434 4435 4436 4437
	if (offset) {
		btrfs_releasepage(page, GFP_NOFS);
		return;
	}
	lock_extent(tree, page_start, page_end, GFP_NOFS);
	ordered = btrfs_lookup_ordered_extent(page->mapping->host,
					   page_offset(page));
	if (ordered) {
4438 4439 4440 4441
		/*
		 * IO on this page will never be started, so we need
		 * to account for any ordered extents now
		 */
4442 4443
		clear_extent_bit(tree, page_start, page_end,
				 EXTENT_DIRTY | EXTENT_DELALLOC |
4444
				 EXTENT_LOCKED, 1, 0, NULL, GFP_NOFS);
4445 4446 4447 4448 4449 4450 4451 4452
		/*
		 * whoever cleared the private bit is responsible
		 * for the finish_ordered_io
		 */
		if (TestClearPagePrivate2(page)) {
			btrfs_finish_ordered_io(page->mapping->host,
						page_start, page_end);
		}
4453 4454 4455 4456
		btrfs_put_ordered_extent(ordered);
		lock_extent(tree, page_start, page_end, GFP_NOFS);
	}
	clear_extent_bit(tree, page_start, page_end,
4457
		 EXTENT_LOCKED | EXTENT_DIRTY | EXTENT_DELALLOC,
4458
		 1, 1, NULL, GFP_NOFS);
4459 4460
	__btrfs_releasepage(page, GFP_NOFS);

C
Chris Mason 已提交
4461
	ClearPageChecked(page);
4462 4463 4464 4465 4466
	if (PagePrivate(page)) {
		ClearPagePrivate(page);
		set_page_private(page, 0);
		page_cache_release(page);
	}
C
Chris Mason 已提交
4467 4468
}

C
Chris Mason 已提交
4469 4470 4471 4472 4473 4474 4475 4476 4477 4478 4479 4480 4481 4482 4483
/*
 * btrfs_page_mkwrite() is not allowed to change the file size as it gets
 * called from a page fault handler when a page is first dirtied. Hence we must
 * be careful to check for EOF conditions here. We set the page up correctly
 * for a written page which means we get ENOSPC checking when writing into
 * holes and correct delalloc and unwritten extent mapping on filesystems that
 * support these features.
 *
 * We are not allowed to take the i_mutex here so we have to play games to
 * protect against truncate races as the page could now be beyond EOF.  Because
 * vmtruncate() writes the inode size before removing pages, once we have the
 * page lock we can determine safely if the page is beyond EOF. If it is not
 * beyond EOF, then the page is guaranteed safe against truncation until we
 * unlock the page.
 */
4484
int btrfs_page_mkwrite(struct vm_area_struct *vma, struct vm_fault *vmf)
C
Chris Mason 已提交
4485
{
4486
	struct page *page = vmf->page;
4487
	struct inode *inode = fdentry(vma->vm_file)->d_inode;
4488
	struct btrfs_root *root = BTRFS_I(inode)->root;
4489 4490 4491 4492
	struct extent_io_tree *io_tree = &BTRFS_I(inode)->io_tree;
	struct btrfs_ordered_extent *ordered;
	char *kaddr;
	unsigned long zero_start;
C
Chris Mason 已提交
4493
	loff_t size;
4494
	int ret;
4495
	u64 page_start;
4496
	u64 page_end;
C
Chris Mason 已提交
4497

J
Josef Bacik 已提交
4498
	ret = btrfs_check_data_free_space(root, inode, PAGE_CACHE_SIZE);
4499 4500 4501 4502 4503
	if (ret) {
		if (ret == -ENOMEM)
			ret = VM_FAULT_OOM;
		else /* -ENOSPC, -EIO, etc */
			ret = VM_FAULT_SIGBUS;
4504
		goto out;
4505
	}
4506

4507
	ret = VM_FAULT_NOPAGE; /* make the VM retry the fault */
4508
again:
C
Chris Mason 已提交
4509 4510
	lock_page(page);
	size = i_size_read(inode);
4511 4512
	page_start = page_offset(page);
	page_end = page_start + PAGE_CACHE_SIZE - 1;
4513

C
Chris Mason 已提交
4514
	if ((page->mapping != inode->i_mapping) ||
4515
	    (page_start >= size)) {
J
Josef Bacik 已提交
4516
		btrfs_free_reserved_data_space(root, inode, PAGE_CACHE_SIZE);
C
Chris Mason 已提交
4517 4518 4519
		/* page got truncated out from underneath us */
		goto out_unlock;
	}
4520 4521 4522 4523 4524
	wait_on_page_writeback(page);

	lock_extent(io_tree, page_start, page_end, GFP_NOFS);
	set_page_extent_mapped(page);

4525 4526 4527 4528
	/*
	 * we can't set the delalloc bits if there are pending ordered
	 * extents.  Drop our locks and wait for them to finish
	 */
4529 4530 4531 4532
	ordered = btrfs_lookup_ordered_extent(inode, page_start);
	if (ordered) {
		unlock_extent(io_tree, page_start, page_end, GFP_NOFS);
		unlock_page(page);
4533
		btrfs_start_ordered_extent(inode, ordered, 1);
4534 4535 4536 4537
		btrfs_put_ordered_extent(ordered);
		goto again;
	}

4538
	btrfs_set_extent_delalloc(inode, page_start, page_end);
4539
	ret = 0;
C
Chris Mason 已提交
4540 4541

	/* page is wholly or partially inside EOF */
4542
	if (page_start + PAGE_CACHE_SIZE > size)
4543
		zero_start = size & ~PAGE_CACHE_MASK;
C
Chris Mason 已提交
4544
	else
4545
		zero_start = PAGE_CACHE_SIZE;
C
Chris Mason 已提交
4546

4547 4548 4549 4550 4551 4552
	if (zero_start != PAGE_CACHE_SIZE) {
		kaddr = kmap(page);
		memset(kaddr + zero_start, 0, PAGE_CACHE_SIZE - zero_start);
		flush_dcache_page(page);
		kunmap(page);
	}
4553
	ClearPageChecked(page);
4554
	set_page_dirty(page);
4555
	SetPageUptodate(page);
4556 4557

	BTRFS_I(inode)->last_trans = root->fs_info->generation + 1;
4558
	unlock_extent(io_tree, page_start, page_end, GFP_NOFS);
C
Chris Mason 已提交
4559 4560

out_unlock:
4561 4562
	if (!ret)
		return VM_FAULT_LOCKED;
C
Chris Mason 已提交
4563
	unlock_page(page);
4564
out:
C
Chris Mason 已提交
4565 4566 4567
	return ret;
}

C
Chris Mason 已提交
4568 4569 4570 4571 4572
static void btrfs_truncate(struct inode *inode)
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	int ret;
	struct btrfs_trans_handle *trans;
4573
	unsigned long nr;
4574
	u64 mask = root->sectorsize - 1;
C
Chris Mason 已提交
4575 4576 4577 4578 4579 4580 4581

	if (!S_ISREG(inode->i_mode))
		return;
	if (IS_APPEND(inode) || IS_IMMUTABLE(inode))
		return;

	btrfs_truncate_page(inode->i_mapping, inode->i_size);
C
Chris Mason 已提交
4582
	btrfs_wait_ordered_range(inode, inode->i_size & (~mask), (u64)-1);
C
Chris Mason 已提交
4583 4584

	trans = btrfs_start_transaction(root, 1);
4585 4586 4587 4588 4589 4590 4591 4592 4593 4594 4595 4596 4597 4598 4599 4600 4601 4602 4603 4604 4605

	/*
	 * setattr is responsible for setting the ordered_data_close flag,
	 * but that is only tested during the last file release.  That
	 * could happen well after the next commit, leaving a great big
	 * window where new writes may get lost if someone chooses to write
	 * to this file after truncating to zero
	 *
	 * The inode doesn't have any dirty data here, and so if we commit
	 * this is a noop.  If someone immediately starts writing to the inode
	 * it is very likely we'll catch some of their writes in this
	 * transaction, and the commit will find this file on the ordered
	 * data list with good things to send down.
	 *
	 * This is a best effort solution, there is still a window where
	 * using truncate to replace the contents of the file will
	 * end up with a zero length file after a crash.
	 */
	if (inode->i_size == 0 && BTRFS_I(inode)->ordered_data_close)
		btrfs_add_ordered_operation(trans, root, inode);

C
Chris Mason 已提交
4606
	btrfs_set_trans_block_group(trans, inode);
4607
	btrfs_i_size_write(inode, inode->i_size);
C
Chris Mason 已提交
4608

4609 4610 4611
	ret = btrfs_orphan_add(trans, inode);
	if (ret)
		goto out;
C
Chris Mason 已提交
4612
	/* FIXME, add redo link to tree so we don't leak on crash */
4613
	ret = btrfs_truncate_inode_items(trans, root, inode, inode->i_size,
4614
				      BTRFS_EXTENT_DATA_KEY);
C
Chris Mason 已提交
4615
	btrfs_update_inode(trans, root, inode);
4616

4617 4618 4619 4620 4621
	ret = btrfs_orphan_del(trans, inode);
	BUG_ON(ret);

out:
	nr = trans->blocks_used;
4622
	ret = btrfs_end_transaction_throttle(trans, root);
C
Chris Mason 已提交
4623
	BUG_ON(ret);
4624
	btrfs_btree_balance_dirty(root, nr);
C
Chris Mason 已提交
4625 4626
}

C
Chris Mason 已提交
4627 4628 4629
/*
 * create a new subvolume directory/inode (helper for the ioctl).
 */
4630 4631 4632
int btrfs_create_subvol_root(struct btrfs_trans_handle *trans,
			     struct btrfs_root *new_root, struct dentry *dentry,
			     u64 new_dirid, u64 alloc_hint)
C
Chris Mason 已提交
4633 4634
{
	struct inode *inode;
4635
	int error;
4636
	u64 index = 0;
C
Chris Mason 已提交
4637

4638
	inode = btrfs_new_inode(trans, new_root, NULL, "..", 2, new_dirid,
4639
				new_dirid, alloc_hint, S_IFDIR | 0700, &index);
4640
	if (IS_ERR(inode))
C
Christoph Hellwig 已提交
4641
		return PTR_ERR(inode);
C
Chris Mason 已提交
4642 4643 4644 4645
	inode->i_op = &btrfs_dir_inode_operations;
	inode->i_fop = &btrfs_dir_file_operations;

	inode->i_nlink = 1;
4646
	btrfs_i_size_write(inode, 0);
4647

4648 4649 4650 4651 4652 4653
	error = btrfs_update_inode(trans, new_root, inode);
	if (error)
		return error;

	d_instantiate(dentry, inode);
	return 0;
C
Chris Mason 已提交
4654 4655
}

C
Chris Mason 已提交
4656 4657 4658
/* helper function for file defrag and space balancing.  This
 * forces readahead on a given range of bytes in an inode
 */
4659
unsigned long btrfs_force_ra(struct address_space *mapping,
4660 4661 4662
			      struct file_ra_state *ra, struct file *file,
			      pgoff_t offset, pgoff_t last_index)
{
4663
	pgoff_t req_size = last_index - offset + 1;
4664 4665 4666 4667 4668

	page_cache_sync_readahead(mapping, ra, file, offset, req_size);
	return offset + req_size;
}

C
Chris Mason 已提交
4669 4670 4671 4672 4673 4674 4675
struct inode *btrfs_alloc_inode(struct super_block *sb)
{
	struct btrfs_inode *ei;

	ei = kmem_cache_alloc(btrfs_inode_cachep, GFP_NOFS);
	if (!ei)
		return NULL;
4676
	ei->last_trans = 0;
4677
	ei->logged_trans = 0;
4678
	btrfs_ordered_inode_tree_init(&ei->ordered_tree);
4679
	INIT_LIST_HEAD(&ei->i_orphan);
4680
	INIT_LIST_HEAD(&ei->ordered_operations);
C
Chris Mason 已提交
4681 4682 4683 4684 4685
	return &ei->vfs_inode;
}

void btrfs_destroy_inode(struct inode *inode)
{
4686
	struct btrfs_ordered_extent *ordered;
4687 4688
	struct btrfs_root *root = BTRFS_I(inode)->root;

C
Chris Mason 已提交
4689 4690 4691
	WARN_ON(!list_empty(&inode->i_dentry));
	WARN_ON(inode->i_data.nrpages);

4692 4693 4694 4695 4696 4697 4698 4699 4700 4701 4702 4703
	/*
	 * Make sure we're properly removed from the ordered operation
	 * lists.
	 */
	smp_mb();
	if (!list_empty(&BTRFS_I(inode)->ordered_operations)) {
		spin_lock(&root->fs_info->ordered_extent_lock);
		list_del_init(&BTRFS_I(inode)->ordered_operations);
		spin_unlock(&root->fs_info->ordered_extent_lock);
	}

	spin_lock(&root->list_lock);
4704 4705 4706 4707 4708
	if (!list_empty(&BTRFS_I(inode)->i_orphan)) {
		printk(KERN_ERR "BTRFS: inode %lu: inode still on the orphan"
		       " list\n", inode->i_ino);
		dump_stack();
	}
4709
	spin_unlock(&root->list_lock);
4710

C
Chris Mason 已提交
4711
	while (1) {
4712 4713 4714 4715
		ordered = btrfs_lookup_first_ordered_extent(inode, (u64)-1);
		if (!ordered)
			break;
		else {
C
Chris Mason 已提交
4716 4717 4718 4719
			printk(KERN_ERR "btrfs found ordered "
			       "extent %llu %llu on inode cleanup\n",
			       (unsigned long long)ordered->file_offset,
			       (unsigned long long)ordered->len);
4720 4721 4722 4723 4724
			btrfs_remove_ordered_extent(inode, ordered);
			btrfs_put_ordered_extent(ordered);
			btrfs_put_ordered_extent(ordered);
		}
	}
4725
	inode_tree_del(inode);
4726
	btrfs_drop_extent_cache(inode, 0, (u64)-1, 0);
C
Chris Mason 已提交
4727 4728 4729
	kmem_cache_free(btrfs_inode_cachep, BTRFS_I(inode));
}

4730
static void init_once(void *foo)
C
Chris Mason 已提交
4731 4732 4733 4734 4735 4736 4737 4738 4739 4740 4741 4742 4743 4744 4745 4746 4747 4748 4749 4750
{
	struct btrfs_inode *ei = (struct btrfs_inode *) foo;

	inode_init_once(&ei->vfs_inode);
}

void btrfs_destroy_cachep(void)
{
	if (btrfs_inode_cachep)
		kmem_cache_destroy(btrfs_inode_cachep);
	if (btrfs_trans_handle_cachep)
		kmem_cache_destroy(btrfs_trans_handle_cachep);
	if (btrfs_transaction_cachep)
		kmem_cache_destroy(btrfs_transaction_cachep);
	if (btrfs_path_cachep)
		kmem_cache_destroy(btrfs_path_cachep);
}

int btrfs_init_cachep(void)
{
4751 4752 4753
	btrfs_inode_cachep = kmem_cache_create("btrfs_inode_cache",
			sizeof(struct btrfs_inode), 0,
			SLAB_RECLAIM_ACCOUNT | SLAB_MEM_SPREAD, init_once);
C
Chris Mason 已提交
4754 4755
	if (!btrfs_inode_cachep)
		goto fail;
4756 4757 4758 4759

	btrfs_trans_handle_cachep = kmem_cache_create("btrfs_trans_handle_cache",
			sizeof(struct btrfs_trans_handle), 0,
			SLAB_RECLAIM_ACCOUNT | SLAB_MEM_SPREAD, NULL);
C
Chris Mason 已提交
4760 4761
	if (!btrfs_trans_handle_cachep)
		goto fail;
4762 4763 4764 4765

	btrfs_transaction_cachep = kmem_cache_create("btrfs_transaction_cache",
			sizeof(struct btrfs_transaction), 0,
			SLAB_RECLAIM_ACCOUNT | SLAB_MEM_SPREAD, NULL);
C
Chris Mason 已提交
4766 4767
	if (!btrfs_transaction_cachep)
		goto fail;
4768 4769 4770 4771

	btrfs_path_cachep = kmem_cache_create("btrfs_path_cache",
			sizeof(struct btrfs_path), 0,
			SLAB_RECLAIM_ACCOUNT | SLAB_MEM_SPREAD, NULL);
C
Chris Mason 已提交
4772 4773
	if (!btrfs_path_cachep)
		goto fail;
4774

C
Chris Mason 已提交
4775 4776 4777 4778 4779 4780 4781 4782 4783 4784 4785
	return 0;
fail:
	btrfs_destroy_cachep();
	return -ENOMEM;
}

static int btrfs_getattr(struct vfsmount *mnt,
			 struct dentry *dentry, struct kstat *stat)
{
	struct inode *inode = dentry->d_inode;
	generic_fillattr(inode, stat);
4786
	stat->dev = BTRFS_I(inode)->root->anon_super.s_dev;
C
Chris Mason 已提交
4787
	stat->blksize = PAGE_CACHE_SIZE;
4788 4789
	stat->blocks = (inode_get_bytes(inode) +
			BTRFS_I(inode)->delalloc_bytes) >> 9;
C
Chris Mason 已提交
4790 4791 4792
	return 0;
}

C
Chris Mason 已提交
4793 4794
static int btrfs_rename(struct inode *old_dir, struct dentry *old_dentry,
			   struct inode *new_dir, struct dentry *new_dentry)
C
Chris Mason 已提交
4795 4796 4797 4798 4799 4800
{
	struct btrfs_trans_handle *trans;
	struct btrfs_root *root = BTRFS_I(old_dir)->root;
	struct inode *new_inode = new_dentry->d_inode;
	struct inode *old_inode = old_dentry->d_inode;
	struct timespec ctime = CURRENT_TIME;
4801
	u64 index = 0;
C
Chris Mason 已提交
4802 4803
	int ret;

4804 4805 4806 4807 4808
	/* we're not allowed to rename between subvolumes */
	if (BTRFS_I(old_inode)->root->root_key.objectid !=
	    BTRFS_I(new_dir)->root->root_key.objectid)
		return -EXDEV;

C
Chris Mason 已提交
4809 4810 4811 4812
	if (S_ISDIR(old_inode->i_mode) && new_inode &&
	    new_inode->i_size > BTRFS_EMPTY_DIR_SIZE) {
		return -ENOTEMPTY;
	}
4813

4814 4815 4816 4817 4818 4819
	/* to rename a snapshot or subvolume, we need to juggle the
	 * backrefs.  This isn't coded yet
	 */
	if (old_inode->i_ino == BTRFS_FIRST_FREE_OBJECTID)
		return -EXDEV;

J
Josef Bacik 已提交
4820
	ret = btrfs_check_metadata_free_space(root);
4821 4822 4823
	if (ret)
		goto out_unlock;

4824 4825 4826 4827 4828
	/*
	 * we're using rename to replace one file with another.
	 * and the replacement file is large.  Start IO on it now so
	 * we don't add too much work to the end of the transaction
	 */
4829
	if (new_inode && S_ISREG(old_inode->i_mode) && new_inode->i_size &&
4830 4831 4832
	    old_inode->i_size > BTRFS_ORDERED_OPERATIONS_FLUSH_LIMIT)
		filemap_flush(old_inode->i_mapping);

C
Chris Mason 已提交
4833
	trans = btrfs_start_transaction(root, 1);
4834

4835 4836 4837 4838 4839 4840 4841 4842 4843
	/*
	 * make sure the inode gets flushed if it is replacing
	 * something.
	 */
	if (new_inode && new_inode->i_size &&
	    old_inode && S_ISREG(old_inode->i_mode)) {
		btrfs_add_ordered_operation(trans, root, old_inode);
	}

4844 4845 4846 4847 4848 4849 4850 4851 4852
	/*
	 * this is an ugly little race, but the rename is required to make
	 * sure that if we crash, the inode is either at the old name
	 * or the new one.  pinning the log transaction lets us make sure
	 * we don't allow a log commit to come in after we unlink the
	 * name but before we add the new name back in.
	 */
	btrfs_pin_log_trans(root);

C
Chris Mason 已提交
4853 4854
	btrfs_set_trans_block_group(trans, new_dir);

4855
	btrfs_inc_nlink(old_dentry->d_inode);
C
Chris Mason 已提交
4856 4857 4858
	old_dir->i_ctime = old_dir->i_mtime = ctime;
	new_dir->i_ctime = new_dir->i_mtime = ctime;
	old_inode->i_ctime = ctime;
4859

4860 4861 4862
	if (old_dentry->d_parent != new_dentry->d_parent)
		btrfs_record_unlink_dir(trans, old_dir, old_inode, 1);

4863 4864 4865
	ret = btrfs_unlink_inode(trans, root, old_dir, old_dentry->d_inode,
				 old_dentry->d_name.name,
				 old_dentry->d_name.len);
C
Chris Mason 已提交
4866 4867 4868 4869 4870
	if (ret)
		goto out_fail;

	if (new_inode) {
		new_inode->i_ctime = CURRENT_TIME;
4871 4872 4873 4874
		ret = btrfs_unlink_inode(trans, root, new_dir,
					 new_dentry->d_inode,
					 new_dentry->d_name.name,
					 new_dentry->d_name.len);
C
Chris Mason 已提交
4875 4876
		if (ret)
			goto out_fail;
4877
		if (new_inode->i_nlink == 0) {
4878
			ret = btrfs_orphan_add(trans, new_dentry->d_inode);
4879 4880 4881
			if (ret)
				goto out_fail;
		}
4882

C
Chris Mason 已提交
4883
	}
4884
	ret = btrfs_set_inode_index(new_dir, &index);
4885 4886 4887
	if (ret)
		goto out_fail;

4888 4889 4890
	ret = btrfs_add_link(trans, new_dentry->d_parent->d_inode,
			     old_inode, new_dentry->d_name.name,
			     new_dentry->d_name.len, 1, index);
C
Chris Mason 已提交
4891 4892 4893
	if (ret)
		goto out_fail;

4894 4895
	btrfs_log_new_name(trans, old_inode, old_dir,
				       new_dentry->d_parent);
C
Chris Mason 已提交
4896
out_fail:
4897 4898 4899 4900 4901

	/* this btrfs_end_log_trans just allows the current
	 * log-sub transaction to complete
	 */
	btrfs_end_log_trans(root);
4902
	btrfs_end_transaction_throttle(trans, root);
4903
out_unlock:
C
Chris Mason 已提交
4904 4905 4906
	return ret;
}

C
Chris Mason 已提交
4907 4908 4909 4910
/*
 * some fairly slow code that needs optimization. This walks the list
 * of all the inodes with pending delalloc and forces them to disk.
 */
4911 4912 4913 4914
int btrfs_start_delalloc_inodes(struct btrfs_root *root)
{
	struct list_head *head = &root->fs_info->delalloc_inodes;
	struct btrfs_inode *binode;
4915
	struct inode *inode;
4916

Y
Yan Zheng 已提交
4917 4918 4919
	if (root->fs_info->sb->s_flags & MS_RDONLY)
		return -EROFS;

4920
	spin_lock(&root->fs_info->delalloc_lock);
C
Chris Mason 已提交
4921
	while (!list_empty(head)) {
4922 4923
		binode = list_entry(head->next, struct btrfs_inode,
				    delalloc_inodes);
4924 4925 4926
		inode = igrab(&binode->vfs_inode);
		if (!inode)
			list_del_init(&binode->delalloc_inodes);
4927
		spin_unlock(&root->fs_info->delalloc_lock);
4928
		if (inode) {
4929
			filemap_flush(inode->i_mapping);
4930 4931 4932
			iput(inode);
		}
		cond_resched();
4933
		spin_lock(&root->fs_info->delalloc_lock);
4934
	}
4935
	spin_unlock(&root->fs_info->delalloc_lock);
4936 4937 4938 4939 4940 4941

	/* the filemap_flush will queue IO into the worker threads, but
	 * we have to make sure the IO is actually started and that
	 * ordered extents get created before we return
	 */
	atomic_inc(&root->fs_info->async_submit_draining);
C
Chris Mason 已提交
4942
	while (atomic_read(&root->fs_info->nr_async_submits) ||
4943
	      atomic_read(&root->fs_info->async_delalloc_pages)) {
4944
		wait_event(root->fs_info->async_submit_wait,
4945 4946
		   (atomic_read(&root->fs_info->nr_async_submits) == 0 &&
		    atomic_read(&root->fs_info->async_delalloc_pages) == 0));
4947 4948
	}
	atomic_dec(&root->fs_info->async_submit_draining);
4949 4950 4951
	return 0;
}

C
Chris Mason 已提交
4952 4953 4954 4955 4956 4957 4958
static int btrfs_symlink(struct inode *dir, struct dentry *dentry,
			 const char *symname)
{
	struct btrfs_trans_handle *trans;
	struct btrfs_root *root = BTRFS_I(dir)->root;
	struct btrfs_path *path;
	struct btrfs_key key;
4959
	struct inode *inode = NULL;
C
Chris Mason 已提交
4960 4961 4962
	int err;
	int drop_inode = 0;
	u64 objectid;
4963
	u64 index = 0 ;
C
Chris Mason 已提交
4964 4965
	int name_len;
	int datasize;
4966
	unsigned long ptr;
C
Chris Mason 已提交
4967
	struct btrfs_file_extent_item *ei;
4968
	struct extent_buffer *leaf;
4969
	unsigned long nr = 0;
C
Chris Mason 已提交
4970 4971 4972 4973

	name_len = strlen(symname) + 1;
	if (name_len > BTRFS_MAX_INLINE_DATA_SIZE(root))
		return -ENAMETOOLONG;
4974

J
Josef Bacik 已提交
4975
	err = btrfs_check_metadata_free_space(root);
4976 4977 4978
	if (err)
		goto out_fail;

C
Chris Mason 已提交
4979 4980 4981 4982 4983 4984 4985 4986 4987
	trans = btrfs_start_transaction(root, 1);
	btrfs_set_trans_block_group(trans, dir);

	err = btrfs_find_free_objectid(trans, root, dir->i_ino, &objectid);
	if (err) {
		err = -ENOSPC;
		goto out_unlock;
	}

4988
	inode = btrfs_new_inode(trans, root, dir, dentry->d_name.name,
4989 4990
				dentry->d_name.len,
				dentry->d_parent->d_inode->i_ino, objectid,
4991 4992
				BTRFS_I(dir)->block_group, S_IFLNK|S_IRWXUGO,
				&index);
C
Chris Mason 已提交
4993 4994 4995 4996
	err = PTR_ERR(inode);
	if (IS_ERR(inode))
		goto out_unlock;

J
Jim Owens 已提交
4997
	err = btrfs_init_inode_security(inode, dir);
J
Josef Bacik 已提交
4998 4999 5000 5001 5002
	if (err) {
		drop_inode = 1;
		goto out_unlock;
	}

C
Chris Mason 已提交
5003
	btrfs_set_trans_block_group(trans, inode);
5004
	err = btrfs_add_nondir(trans, dentry, inode, 0, index);
C
Chris Mason 已提交
5005 5006 5007 5008
	if (err)
		drop_inode = 1;
	else {
		inode->i_mapping->a_ops = &btrfs_aops;
C
Chris Mason 已提交
5009
		inode->i_mapping->backing_dev_info = &root->fs_info->bdi;
C
Chris Mason 已提交
5010 5011
		inode->i_fop = &btrfs_file_operations;
		inode->i_op = &btrfs_file_inode_operations;
5012
		BTRFS_I(inode)->io_tree.ops = &btrfs_extent_io_ops;
C
Chris Mason 已提交
5013 5014 5015 5016 5017 5018 5019 5020 5021 5022 5023 5024 5025 5026
	}
	btrfs_update_inode_block_group(trans, inode);
	btrfs_update_inode_block_group(trans, dir);
	if (drop_inode)
		goto out_unlock;

	path = btrfs_alloc_path();
	BUG_ON(!path);
	key.objectid = inode->i_ino;
	key.offset = 0;
	btrfs_set_key_type(&key, BTRFS_EXTENT_DATA_KEY);
	datasize = btrfs_file_extent_calc_inline_size(name_len);
	err = btrfs_insert_empty_item(trans, root, path, &key,
				      datasize);
5027 5028 5029 5030
	if (err) {
		drop_inode = 1;
		goto out_unlock;
	}
5031 5032 5033 5034 5035
	leaf = path->nodes[0];
	ei = btrfs_item_ptr(leaf, path->slots[0],
			    struct btrfs_file_extent_item);
	btrfs_set_file_extent_generation(leaf, ei, trans->transid);
	btrfs_set_file_extent_type(leaf, ei,
C
Chris Mason 已提交
5036
				   BTRFS_FILE_EXTENT_INLINE);
C
Chris Mason 已提交
5037 5038 5039 5040 5041
	btrfs_set_file_extent_encryption(leaf, ei, 0);
	btrfs_set_file_extent_compression(leaf, ei, 0);
	btrfs_set_file_extent_other_encoding(leaf, ei, 0);
	btrfs_set_file_extent_ram_bytes(leaf, ei, name_len);

C
Chris Mason 已提交
5042
	ptr = btrfs_file_extent_inline_start(ei);
5043 5044
	write_extent_buffer(leaf, symname, ptr, name_len);
	btrfs_mark_buffer_dirty(leaf);
C
Chris Mason 已提交
5045
	btrfs_free_path(path);
5046

C
Chris Mason 已提交
5047 5048
	inode->i_op = &btrfs_symlink_inode_operations;
	inode->i_mapping->a_ops = &btrfs_symlink_aops;
C
Chris Mason 已提交
5049
	inode->i_mapping->backing_dev_info = &root->fs_info->bdi;
Y
Yan Zheng 已提交
5050
	inode_set_bytes(inode, name_len);
5051
	btrfs_i_size_write(inode, name_len - 1);
5052 5053 5054
	err = btrfs_update_inode(trans, root, inode);
	if (err)
		drop_inode = 1;
C
Chris Mason 已提交
5055 5056

out_unlock:
5057
	nr = trans->blocks_used;
5058
	btrfs_end_transaction_throttle(trans, root);
5059
out_fail:
C
Chris Mason 已提交
5060 5061 5062 5063
	if (drop_inode) {
		inode_dec_link_count(inode);
		iput(inode);
	}
5064
	btrfs_btree_balance_dirty(root, nr);
C
Chris Mason 已提交
5065 5066
	return err;
}
5067

5068 5069
static int prealloc_file_range(struct btrfs_trans_handle *trans,
			       struct inode *inode, u64 start, u64 end,
5070
			       u64 locked_end, u64 alloc_hint, int mode)
Y
Yan Zheng 已提交
5071 5072 5073 5074 5075 5076 5077 5078 5079 5080 5081 5082 5083 5084 5085 5086 5087 5088 5089 5090
{
	struct btrfs_root *root = BTRFS_I(inode)->root;
	struct btrfs_key ins;
	u64 alloc_size;
	u64 cur_offset = start;
	u64 num_bytes = end - start;
	int ret = 0;

	while (num_bytes > 0) {
		alloc_size = min(num_bytes, root->fs_info->max_extent);
		ret = btrfs_reserve_extent(trans, root, alloc_size,
					   root->sectorsize, 0, alloc_hint,
					   (u64)-1, &ins, 1);
		if (ret) {
			WARN_ON(1);
			goto out;
		}
		ret = insert_reserved_file_extent(trans, inode,
						  cur_offset, ins.objectid,
						  ins.offset, ins.offset,
5091 5092
						  ins.offset, locked_end,
						  0, 0, 0,
Y
Yan Zheng 已提交
5093 5094
						  BTRFS_FILE_EXTENT_PREALLOC);
		BUG_ON(ret);
C
Chris Mason 已提交
5095 5096
		btrfs_drop_extent_cache(inode, cur_offset,
					cur_offset + ins.offset -1, 0);
Y
Yan Zheng 已提交
5097 5098 5099 5100 5101 5102 5103
		num_bytes -= ins.offset;
		cur_offset += ins.offset;
		alloc_hint = ins.objectid + ins.offset;
	}
out:
	if (cur_offset > start) {
		inode->i_ctime = CURRENT_TIME;
5104
		BTRFS_I(inode)->flags |= BTRFS_INODE_PREALLOC;
Y
Yan Zheng 已提交
5105 5106 5107 5108 5109 5110 5111 5112 5113 5114 5115 5116 5117 5118 5119 5120 5121 5122
		if (!(mode & FALLOC_FL_KEEP_SIZE) &&
		    cur_offset > i_size_read(inode))
			btrfs_i_size_write(inode, cur_offset);
		ret = btrfs_update_inode(trans, root, inode);
		BUG_ON(ret);
	}

	return ret;
}

static long btrfs_fallocate(struct inode *inode, int mode,
			    loff_t offset, loff_t len)
{
	u64 cur_offset;
	u64 last_byte;
	u64 alloc_start;
	u64 alloc_end;
	u64 alloc_hint = 0;
5123
	u64 locked_end;
Y
Yan Zheng 已提交
5124 5125
	u64 mask = BTRFS_I(inode)->root->sectorsize - 1;
	struct extent_map *em;
5126
	struct btrfs_trans_handle *trans;
5127
	struct btrfs_root *root;
Y
Yan Zheng 已提交
5128 5129 5130 5131 5132
	int ret;

	alloc_start = offset & ~mask;
	alloc_end =  (offset + len + mask) & ~mask;

5133 5134 5135 5136 5137 5138
	/*
	 * wait for ordered IO before we have any locks.  We'll loop again
	 * below with the locks held.
	 */
	btrfs_wait_ordered_range(inode, alloc_start, alloc_end - alloc_start);

Y
Yan Zheng 已提交
5139 5140 5141 5142 5143 5144 5145
	mutex_lock(&inode->i_mutex);
	if (alloc_start > inode->i_size) {
		ret = btrfs_cont_expand(inode, alloc_start);
		if (ret)
			goto out;
	}

5146 5147 5148 5149 5150 5151 5152
	root = BTRFS_I(inode)->root;

	ret = btrfs_check_data_free_space(root, inode,
					  alloc_end - alloc_start);
	if (ret)
		goto out;

5153
	locked_end = alloc_end - 1;
Y
Yan Zheng 已提交
5154 5155
	while (1) {
		struct btrfs_ordered_extent *ordered;
5156 5157 5158 5159

		trans = btrfs_start_transaction(BTRFS_I(inode)->root, 1);
		if (!trans) {
			ret = -EIO;
5160
			goto out_free;
5161 5162 5163 5164 5165
		}

		/* the extent lock is ordered inside the running
		 * transaction
		 */
5166 5167
		lock_extent(&BTRFS_I(inode)->io_tree, alloc_start, locked_end,
			    GFP_NOFS);
Y
Yan Zheng 已提交
5168 5169 5170 5171 5172 5173 5174
		ordered = btrfs_lookup_first_ordered_extent(inode,
							    alloc_end - 1);
		if (ordered &&
		    ordered->file_offset + ordered->len > alloc_start &&
		    ordered->file_offset < alloc_end) {
			btrfs_put_ordered_extent(ordered);
			unlock_extent(&BTRFS_I(inode)->io_tree,
5175
				      alloc_start, locked_end, GFP_NOFS);
5176 5177 5178 5179 5180 5181
			btrfs_end_transaction(trans, BTRFS_I(inode)->root);

			/*
			 * we can't wait on the range with the transaction
			 * running or with the extent lock held
			 */
Y
Yan Zheng 已提交
5182 5183 5184 5185 5186 5187 5188 5189 5190 5191 5192 5193 5194 5195 5196 5197 5198
			btrfs_wait_ordered_range(inode, alloc_start,
						 alloc_end - alloc_start);
		} else {
			if (ordered)
				btrfs_put_ordered_extent(ordered);
			break;
		}
	}

	cur_offset = alloc_start;
	while (1) {
		em = btrfs_get_extent(inode, NULL, 0, cur_offset,
				      alloc_end - cur_offset, 0);
		BUG_ON(IS_ERR(em) || !em);
		last_byte = min(extent_map_end(em), alloc_end);
		last_byte = (last_byte + mask) & ~mask;
		if (em->block_start == EXTENT_MAP_HOLE) {
5199
			ret = prealloc_file_range(trans, inode, cur_offset,
5200 5201
					last_byte, locked_end + 1,
					alloc_hint, mode);
Y
Yan Zheng 已提交
5202 5203 5204 5205 5206 5207 5208 5209 5210 5211 5212 5213 5214 5215 5216
			if (ret < 0) {
				free_extent_map(em);
				break;
			}
		}
		if (em->block_start <= EXTENT_MAP_LAST_BYTE)
			alloc_hint = em->block_start;
		free_extent_map(em);

		cur_offset = last_byte;
		if (cur_offset >= alloc_end) {
			ret = 0;
			break;
		}
	}
5217
	unlock_extent(&BTRFS_I(inode)->io_tree, alloc_start, locked_end,
Y
Yan Zheng 已提交
5218
		      GFP_NOFS);
5219 5220

	btrfs_end_transaction(trans, BTRFS_I(inode)->root);
5221 5222
out_free:
	btrfs_free_reserved_data_space(root, inode, alloc_end - alloc_start);
Y
Yan Zheng 已提交
5223 5224 5225 5226 5227
out:
	mutex_unlock(&inode->i_mutex);
	return ret;
}

5228 5229 5230 5231 5232
static int btrfs_set_page_dirty(struct page *page)
{
	return __set_page_dirty_nobuffers(page);
}

5233
static int btrfs_permission(struct inode *inode, int mask)
Y
Yan 已提交
5234
{
5235
	if ((BTRFS_I(inode)->flags & BTRFS_INODE_READONLY) && (mask & MAY_WRITE))
Y
Yan 已提交
5236
		return -EACCES;
J
Josef Bacik 已提交
5237
	return generic_permission(inode, mask, btrfs_check_acl);
Y
Yan 已提交
5238
}
C
Chris Mason 已提交
5239 5240

static struct inode_operations btrfs_dir_inode_operations = {
5241
	.getattr	= btrfs_getattr,
C
Chris Mason 已提交
5242 5243 5244 5245 5246 5247 5248 5249 5250
	.lookup		= btrfs_lookup,
	.create		= btrfs_create,
	.unlink		= btrfs_unlink,
	.link		= btrfs_link,
	.mkdir		= btrfs_mkdir,
	.rmdir		= btrfs_rmdir,
	.rename		= btrfs_rename,
	.symlink	= btrfs_symlink,
	.setattr	= btrfs_setattr,
J
Josef Bacik 已提交
5251
	.mknod		= btrfs_mknod,
5252 5253
	.setxattr	= btrfs_setxattr,
	.getxattr	= btrfs_getxattr,
J
Josef Bacik 已提交
5254
	.listxattr	= btrfs_listxattr,
5255
	.removexattr	= btrfs_removexattr,
Y
Yan 已提交
5256
	.permission	= btrfs_permission,
C
Chris Mason 已提交
5257 5258 5259
};
static struct inode_operations btrfs_dir_ro_inode_operations = {
	.lookup		= btrfs_lookup,
Y
Yan 已提交
5260
	.permission	= btrfs_permission,
C
Chris Mason 已提交
5261 5262 5263 5264
};
static struct file_operations btrfs_dir_file_operations = {
	.llseek		= generic_file_llseek,
	.read		= generic_read_dir,
5265
	.readdir	= btrfs_real_readdir,
C
Christoph Hellwig 已提交
5266
	.unlocked_ioctl	= btrfs_ioctl,
C
Chris Mason 已提交
5267
#ifdef CONFIG_COMPAT
C
Christoph Hellwig 已提交
5268
	.compat_ioctl	= btrfs_ioctl,
C
Chris Mason 已提交
5269
#endif
S
Sage Weil 已提交
5270
	.release        = btrfs_release_file,
5271
	.fsync		= btrfs_sync_file,
C
Chris Mason 已提交
5272 5273
};

5274
static struct extent_io_ops btrfs_extent_io_ops = {
5275
	.fill_delalloc = run_delalloc_range,
5276
	.submit_bio_hook = btrfs_submit_bio_hook,
5277
	.merge_bio_hook = btrfs_merge_bio_hook,
5278
	.readpage_end_io_hook = btrfs_readpage_end_io_hook,
5279
	.writepage_end_io_hook = btrfs_writepage_end_io_hook,
5280
	.writepage_start_hook = btrfs_writepage_start_hook,
5281
	.readpage_io_failed_hook = btrfs_io_failed_hook,
C
Chris Mason 已提交
5282 5283
	.set_bit_hook = btrfs_set_bit_hook,
	.clear_bit_hook = btrfs_clear_bit_hook,
5284 5285
};

5286 5287 5288 5289 5290 5291 5292 5293 5294 5295 5296 5297
/*
 * btrfs doesn't support the bmap operation because swapfiles
 * use bmap to make a mapping of extents in the file.  They assume
 * these extents won't change over the life of the file and they
 * use the bmap result to do IO directly to the drive.
 *
 * the btrfs bmap call would return logical addresses that aren't
 * suitable for IO and they also will change frequently as COW
 * operations happen.  So, swapfile + btrfs == corruption.
 *
 * For now we're avoiding this by dropping bmap.
 */
C
Chris Mason 已提交
5298 5299 5300
static struct address_space_operations btrfs_aops = {
	.readpage	= btrfs_readpage,
	.writepage	= btrfs_writepage,
C
Chris Mason 已提交
5301
	.writepages	= btrfs_writepages,
C
Chris Mason 已提交
5302
	.readpages	= btrfs_readpages,
C
Chris Mason 已提交
5303
	.sync_page	= block_sync_page,
5304
	.direct_IO	= btrfs_direct_IO,
5305 5306
	.invalidatepage = btrfs_invalidatepage,
	.releasepage	= btrfs_releasepage,
5307
	.set_page_dirty	= btrfs_set_page_dirty,
C
Chris Mason 已提交
5308 5309 5310 5311 5312
};

static struct address_space_operations btrfs_symlink_aops = {
	.readpage	= btrfs_readpage,
	.writepage	= btrfs_writepage,
C
Chris Mason 已提交
5313 5314
	.invalidatepage = btrfs_invalidatepage,
	.releasepage	= btrfs_releasepage,
C
Chris Mason 已提交
5315 5316 5317 5318 5319 5320
};

static struct inode_operations btrfs_file_inode_operations = {
	.truncate	= btrfs_truncate,
	.getattr	= btrfs_getattr,
	.setattr	= btrfs_setattr,
5321 5322
	.setxattr	= btrfs_setxattr,
	.getxattr	= btrfs_getxattr,
J
Josef Bacik 已提交
5323
	.listxattr      = btrfs_listxattr,
5324
	.removexattr	= btrfs_removexattr,
Y
Yan 已提交
5325
	.permission	= btrfs_permission,
Y
Yan Zheng 已提交
5326
	.fallocate	= btrfs_fallocate,
Y
Yehuda Sadeh 已提交
5327
	.fiemap		= btrfs_fiemap,
C
Chris Mason 已提交
5328
};
J
Josef Bacik 已提交
5329 5330 5331
static struct inode_operations btrfs_special_inode_operations = {
	.getattr	= btrfs_getattr,
	.setattr	= btrfs_setattr,
Y
Yan 已提交
5332
	.permission	= btrfs_permission,
5333 5334
	.setxattr	= btrfs_setxattr,
	.getxattr	= btrfs_getxattr,
J
Josef Bacik 已提交
5335
	.listxattr	= btrfs_listxattr,
5336
	.removexattr	= btrfs_removexattr,
J
Josef Bacik 已提交
5337
};
C
Chris Mason 已提交
5338 5339 5340 5341
static struct inode_operations btrfs_symlink_inode_operations = {
	.readlink	= generic_readlink,
	.follow_link	= page_follow_link_light,
	.put_link	= page_put_link,
Y
Yan 已提交
5342
	.permission	= btrfs_permission,
J
Jim Owens 已提交
5343 5344 5345 5346
	.setxattr	= btrfs_setxattr,
	.getxattr	= btrfs_getxattr,
	.listxattr	= btrfs_listxattr,
	.removexattr	= btrfs_removexattr,
C
Chris Mason 已提交
5347
};