xref: /openbmc/linux/fs/btrfs/file.c (revision d3741027)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright (C) 2007 Oracle.  All rights reserved.
4  */
5 
6 #include <linux/fs.h>
7 #include <linux/pagemap.h>
8 #include <linux/time.h>
9 #include <linux/init.h>
10 #include <linux/string.h>
11 #include <linux/backing-dev.h>
12 #include <linux/falloc.h>
13 #include <linux/writeback.h>
14 #include <linux/compat.h>
15 #include <linux/slab.h>
16 #include <linux/btrfs.h>
17 #include <linux/uio.h>
18 #include <linux/iversion.h>
19 #include <linux/fsverity.h>
20 #include "ctree.h"
21 #include "disk-io.h"
22 #include "transaction.h"
23 #include "btrfs_inode.h"
24 #include "print-tree.h"
25 #include "tree-log.h"
26 #include "locking.h"
27 #include "volumes.h"
28 #include "qgroup.h"
29 #include "compression.h"
30 #include "delalloc-space.h"
31 #include "reflink.h"
32 #include "subpage.h"
33 
34 static struct kmem_cache *btrfs_inode_defrag_cachep;
35 /*
36  * when auto defrag is enabled we
37  * queue up these defrag structs to remember which
38  * inodes need defragging passes
39  */
40 struct inode_defrag {
41 	struct rb_node rb_node;
42 	/* objectid */
43 	u64 ino;
44 	/*
45 	 * transid where the defrag was added, we search for
46 	 * extents newer than this
47 	 */
48 	u64 transid;
49 
50 	/* root objectid */
51 	u64 root;
52 
53 	/*
54 	 * The extent size threshold for autodefrag.
55 	 *
56 	 * This value is different for compressed/non-compressed extents,
57 	 * thus needs to be passed from higher layer.
58 	 * (aka, inode_should_defrag())
59 	 */
60 	u32 extent_thresh;
61 };
62 
63 static int __compare_inode_defrag(struct inode_defrag *defrag1,
64 				  struct inode_defrag *defrag2)
65 {
66 	if (defrag1->root > defrag2->root)
67 		return 1;
68 	else if (defrag1->root < defrag2->root)
69 		return -1;
70 	else if (defrag1->ino > defrag2->ino)
71 		return 1;
72 	else if (defrag1->ino < defrag2->ino)
73 		return -1;
74 	else
75 		return 0;
76 }
77 
78 /* pop a record for an inode into the defrag tree.  The lock
79  * must be held already
80  *
81  * If you're inserting a record for an older transid than an
82  * existing record, the transid already in the tree is lowered
83  *
84  * If an existing record is found the defrag item you
85  * pass in is freed
86  */
87 static int __btrfs_add_inode_defrag(struct btrfs_inode *inode,
88 				    struct inode_defrag *defrag)
89 {
90 	struct btrfs_fs_info *fs_info = inode->root->fs_info;
91 	struct inode_defrag *entry;
92 	struct rb_node **p;
93 	struct rb_node *parent = NULL;
94 	int ret;
95 
96 	p = &fs_info->defrag_inodes.rb_node;
97 	while (*p) {
98 		parent = *p;
99 		entry = rb_entry(parent, struct inode_defrag, rb_node);
100 
101 		ret = __compare_inode_defrag(defrag, entry);
102 		if (ret < 0)
103 			p = &parent->rb_left;
104 		else if (ret > 0)
105 			p = &parent->rb_right;
106 		else {
107 			/* if we're reinserting an entry for
108 			 * an old defrag run, make sure to
109 			 * lower the transid of our existing record
110 			 */
111 			if (defrag->transid < entry->transid)
112 				entry->transid = defrag->transid;
113 			entry->extent_thresh = min(defrag->extent_thresh,
114 						   entry->extent_thresh);
115 			return -EEXIST;
116 		}
117 	}
118 	set_bit(BTRFS_INODE_IN_DEFRAG, &inode->runtime_flags);
119 	rb_link_node(&defrag->rb_node, parent, p);
120 	rb_insert_color(&defrag->rb_node, &fs_info->defrag_inodes);
121 	return 0;
122 }
123 
124 static inline int __need_auto_defrag(struct btrfs_fs_info *fs_info)
125 {
126 	if (!btrfs_test_opt(fs_info, AUTO_DEFRAG))
127 		return 0;
128 
129 	if (btrfs_fs_closing(fs_info))
130 		return 0;
131 
132 	return 1;
133 }
134 
135 /*
136  * insert a defrag record for this inode if auto defrag is
137  * enabled
138  */
139 int btrfs_add_inode_defrag(struct btrfs_trans_handle *trans,
140 			   struct btrfs_inode *inode, u32 extent_thresh)
141 {
142 	struct btrfs_root *root = inode->root;
143 	struct btrfs_fs_info *fs_info = root->fs_info;
144 	struct inode_defrag *defrag;
145 	u64 transid;
146 	int ret;
147 
148 	if (!__need_auto_defrag(fs_info))
149 		return 0;
150 
151 	if (test_bit(BTRFS_INODE_IN_DEFRAG, &inode->runtime_flags))
152 		return 0;
153 
154 	if (trans)
155 		transid = trans->transid;
156 	else
157 		transid = inode->root->last_trans;
158 
159 	defrag = kmem_cache_zalloc(btrfs_inode_defrag_cachep, GFP_NOFS);
160 	if (!defrag)
161 		return -ENOMEM;
162 
163 	defrag->ino = btrfs_ino(inode);
164 	defrag->transid = transid;
165 	defrag->root = root->root_key.objectid;
166 	defrag->extent_thresh = extent_thresh;
167 
168 	spin_lock(&fs_info->defrag_inodes_lock);
169 	if (!test_bit(BTRFS_INODE_IN_DEFRAG, &inode->runtime_flags)) {
170 		/*
171 		 * If we set IN_DEFRAG flag and evict the inode from memory,
172 		 * and then re-read this inode, this new inode doesn't have
173 		 * IN_DEFRAG flag. At the case, we may find the existed defrag.
174 		 */
175 		ret = __btrfs_add_inode_defrag(inode, defrag);
176 		if (ret)
177 			kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
178 	} else {
179 		kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
180 	}
181 	spin_unlock(&fs_info->defrag_inodes_lock);
182 	return 0;
183 }
184 
185 /*
186  * pick the defragable inode that we want, if it doesn't exist, we will get
187  * the next one.
188  */
189 static struct inode_defrag *
190 btrfs_pick_defrag_inode(struct btrfs_fs_info *fs_info, u64 root, u64 ino)
191 {
192 	struct inode_defrag *entry = NULL;
193 	struct inode_defrag tmp;
194 	struct rb_node *p;
195 	struct rb_node *parent = NULL;
196 	int ret;
197 
198 	tmp.ino = ino;
199 	tmp.root = root;
200 
201 	spin_lock(&fs_info->defrag_inodes_lock);
202 	p = fs_info->defrag_inodes.rb_node;
203 	while (p) {
204 		parent = p;
205 		entry = rb_entry(parent, struct inode_defrag, rb_node);
206 
207 		ret = __compare_inode_defrag(&tmp, entry);
208 		if (ret < 0)
209 			p = parent->rb_left;
210 		else if (ret > 0)
211 			p = parent->rb_right;
212 		else
213 			goto out;
214 	}
215 
216 	if (parent && __compare_inode_defrag(&tmp, entry) > 0) {
217 		parent = rb_next(parent);
218 		if (parent)
219 			entry = rb_entry(parent, struct inode_defrag, rb_node);
220 		else
221 			entry = NULL;
222 	}
223 out:
224 	if (entry)
225 		rb_erase(parent, &fs_info->defrag_inodes);
226 	spin_unlock(&fs_info->defrag_inodes_lock);
227 	return entry;
228 }
229 
230 void btrfs_cleanup_defrag_inodes(struct btrfs_fs_info *fs_info)
231 {
232 	struct inode_defrag *defrag;
233 	struct rb_node *node;
234 
235 	spin_lock(&fs_info->defrag_inodes_lock);
236 	node = rb_first(&fs_info->defrag_inodes);
237 	while (node) {
238 		rb_erase(node, &fs_info->defrag_inodes);
239 		defrag = rb_entry(node, struct inode_defrag, rb_node);
240 		kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
241 
242 		cond_resched_lock(&fs_info->defrag_inodes_lock);
243 
244 		node = rb_first(&fs_info->defrag_inodes);
245 	}
246 	spin_unlock(&fs_info->defrag_inodes_lock);
247 }
248 
249 #define BTRFS_DEFRAG_BATCH	1024
250 
251 static int __btrfs_run_defrag_inode(struct btrfs_fs_info *fs_info,
252 				    struct inode_defrag *defrag)
253 {
254 	struct btrfs_root *inode_root;
255 	struct inode *inode;
256 	struct btrfs_ioctl_defrag_range_args range;
257 	int ret = 0;
258 	u64 cur = 0;
259 
260 again:
261 	if (test_bit(BTRFS_FS_STATE_REMOUNTING, &fs_info->fs_state))
262 		goto cleanup;
263 	if (!__need_auto_defrag(fs_info))
264 		goto cleanup;
265 
266 	/* get the inode */
267 	inode_root = btrfs_get_fs_root(fs_info, defrag->root, true);
268 	if (IS_ERR(inode_root)) {
269 		ret = PTR_ERR(inode_root);
270 		goto cleanup;
271 	}
272 
273 	inode = btrfs_iget(fs_info->sb, defrag->ino, inode_root);
274 	btrfs_put_root(inode_root);
275 	if (IS_ERR(inode)) {
276 		ret = PTR_ERR(inode);
277 		goto cleanup;
278 	}
279 
280 	if (cur >= i_size_read(inode)) {
281 		iput(inode);
282 		goto cleanup;
283 	}
284 
285 	/* do a chunk of defrag */
286 	clear_bit(BTRFS_INODE_IN_DEFRAG, &BTRFS_I(inode)->runtime_flags);
287 	memset(&range, 0, sizeof(range));
288 	range.len = (u64)-1;
289 	range.start = cur;
290 	range.extent_thresh = defrag->extent_thresh;
291 
292 	sb_start_write(fs_info->sb);
293 	ret = btrfs_defrag_file(inode, NULL, &range, defrag->transid,
294 				       BTRFS_DEFRAG_BATCH);
295 	sb_end_write(fs_info->sb);
296 	iput(inode);
297 
298 	if (ret < 0)
299 		goto cleanup;
300 
301 	cur = max(cur + fs_info->sectorsize, range.start);
302 	goto again;
303 
304 cleanup:
305 	kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
306 	return ret;
307 }
308 
309 /*
310  * run through the list of inodes in the FS that need
311  * defragging
312  */
313 int btrfs_run_defrag_inodes(struct btrfs_fs_info *fs_info)
314 {
315 	struct inode_defrag *defrag;
316 	u64 first_ino = 0;
317 	u64 root_objectid = 0;
318 
319 	atomic_inc(&fs_info->defrag_running);
320 	while (1) {
321 		/* Pause the auto defragger. */
322 		if (test_bit(BTRFS_FS_STATE_REMOUNTING,
323 			     &fs_info->fs_state))
324 			break;
325 
326 		if (!__need_auto_defrag(fs_info))
327 			break;
328 
329 		/* find an inode to defrag */
330 		defrag = btrfs_pick_defrag_inode(fs_info, root_objectid,
331 						 first_ino);
332 		if (!defrag) {
333 			if (root_objectid || first_ino) {
334 				root_objectid = 0;
335 				first_ino = 0;
336 				continue;
337 			} else {
338 				break;
339 			}
340 		}
341 
342 		first_ino = defrag->ino + 1;
343 		root_objectid = defrag->root;
344 
345 		__btrfs_run_defrag_inode(fs_info, defrag);
346 	}
347 	atomic_dec(&fs_info->defrag_running);
348 
349 	/*
350 	 * during unmount, we use the transaction_wait queue to
351 	 * wait for the defragger to stop
352 	 */
353 	wake_up(&fs_info->transaction_wait);
354 	return 0;
355 }
356 
357 /* simple helper to fault in pages and copy.  This should go away
358  * and be replaced with calls into generic code.
359  */
360 static noinline int btrfs_copy_from_user(loff_t pos, size_t write_bytes,
361 					 struct page **prepared_pages,
362 					 struct iov_iter *i)
363 {
364 	size_t copied = 0;
365 	size_t total_copied = 0;
366 	int pg = 0;
367 	int offset = offset_in_page(pos);
368 
369 	while (write_bytes > 0) {
370 		size_t count = min_t(size_t,
371 				     PAGE_SIZE - offset, write_bytes);
372 		struct page *page = prepared_pages[pg];
373 		/*
374 		 * Copy data from userspace to the current page
375 		 */
376 		copied = copy_page_from_iter_atomic(page, offset, count, i);
377 
378 		/* Flush processor's dcache for this page */
379 		flush_dcache_page(page);
380 
381 		/*
382 		 * if we get a partial write, we can end up with
383 		 * partially up to date pages.  These add
384 		 * a lot of complexity, so make sure they don't
385 		 * happen by forcing this copy to be retried.
386 		 *
387 		 * The rest of the btrfs_file_write code will fall
388 		 * back to page at a time copies after we return 0.
389 		 */
390 		if (unlikely(copied < count)) {
391 			if (!PageUptodate(page)) {
392 				iov_iter_revert(i, copied);
393 				copied = 0;
394 			}
395 			if (!copied)
396 				break;
397 		}
398 
399 		write_bytes -= copied;
400 		total_copied += copied;
401 		offset += copied;
402 		if (offset == PAGE_SIZE) {
403 			pg++;
404 			offset = 0;
405 		}
406 	}
407 	return total_copied;
408 }
409 
410 /*
411  * unlocks pages after btrfs_file_write is done with them
412  */
413 static void btrfs_drop_pages(struct btrfs_fs_info *fs_info,
414 			     struct page **pages, size_t num_pages,
415 			     u64 pos, u64 copied)
416 {
417 	size_t i;
418 	u64 block_start = round_down(pos, fs_info->sectorsize);
419 	u64 block_len = round_up(pos + copied, fs_info->sectorsize) - block_start;
420 
421 	ASSERT(block_len <= U32_MAX);
422 	for (i = 0; i < num_pages; i++) {
423 		/* page checked is some magic around finding pages that
424 		 * have been modified without going through btrfs_set_page_dirty
425 		 * clear it here. There should be no need to mark the pages
426 		 * accessed as prepare_pages should have marked them accessed
427 		 * in prepare_pages via find_or_create_page()
428 		 */
429 		btrfs_page_clamp_clear_checked(fs_info, pages[i], block_start,
430 					       block_len);
431 		unlock_page(pages[i]);
432 		put_page(pages[i]);
433 	}
434 }
435 
436 /*
437  * After btrfs_copy_from_user(), update the following things for delalloc:
438  * - Mark newly dirtied pages as DELALLOC in the io tree.
439  *   Used to advise which range is to be written back.
440  * - Mark modified pages as Uptodate/Dirty and not needing COW fixup
441  * - Update inode size for past EOF write
442  */
443 int btrfs_dirty_pages(struct btrfs_inode *inode, struct page **pages,
444 		      size_t num_pages, loff_t pos, size_t write_bytes,
445 		      struct extent_state **cached, bool noreserve)
446 {
447 	struct btrfs_fs_info *fs_info = inode->root->fs_info;
448 	int err = 0;
449 	int i;
450 	u64 num_bytes;
451 	u64 start_pos;
452 	u64 end_of_last_block;
453 	u64 end_pos = pos + write_bytes;
454 	loff_t isize = i_size_read(&inode->vfs_inode);
455 	unsigned int extra_bits = 0;
456 
457 	if (write_bytes == 0)
458 		return 0;
459 
460 	if (noreserve)
461 		extra_bits |= EXTENT_NORESERVE;
462 
463 	start_pos = round_down(pos, fs_info->sectorsize);
464 	num_bytes = round_up(write_bytes + pos - start_pos,
465 			     fs_info->sectorsize);
466 	ASSERT(num_bytes <= U32_MAX);
467 
468 	end_of_last_block = start_pos + num_bytes - 1;
469 
470 	/*
471 	 * The pages may have already been dirty, clear out old accounting so
472 	 * we can set things up properly
473 	 */
474 	clear_extent_bit(&inode->io_tree, start_pos, end_of_last_block,
475 			 EXTENT_DELALLOC | EXTENT_DO_ACCOUNTING | EXTENT_DEFRAG,
476 			 0, 0, cached);
477 
478 	err = btrfs_set_extent_delalloc(inode, start_pos, end_of_last_block,
479 					extra_bits, cached);
480 	if (err)
481 		return err;
482 
483 	for (i = 0; i < num_pages; i++) {
484 		struct page *p = pages[i];
485 
486 		btrfs_page_clamp_set_uptodate(fs_info, p, start_pos, num_bytes);
487 		btrfs_page_clamp_clear_checked(fs_info, p, start_pos, num_bytes);
488 		btrfs_page_clamp_set_dirty(fs_info, p, start_pos, num_bytes);
489 	}
490 
491 	/*
492 	 * we've only changed i_size in ram, and we haven't updated
493 	 * the disk i_size.  There is no need to log the inode
494 	 * at this time.
495 	 */
496 	if (end_pos > isize)
497 		i_size_write(&inode->vfs_inode, end_pos);
498 	return 0;
499 }
500 
501 /*
502  * this drops all the extents in the cache that intersect the range
503  * [start, end].  Existing extents are split as required.
504  */
505 void btrfs_drop_extent_cache(struct btrfs_inode *inode, u64 start, u64 end,
506 			     int skip_pinned)
507 {
508 	struct extent_map *em;
509 	struct extent_map *split = NULL;
510 	struct extent_map *split2 = NULL;
511 	struct extent_map_tree *em_tree = &inode->extent_tree;
512 	u64 len = end - start + 1;
513 	u64 gen;
514 	int ret;
515 	int testend = 1;
516 	unsigned long flags;
517 	int compressed = 0;
518 	bool modified;
519 
520 	WARN_ON(end < start);
521 	if (end == (u64)-1) {
522 		len = (u64)-1;
523 		testend = 0;
524 	}
525 	while (1) {
526 		int no_splits = 0;
527 
528 		modified = false;
529 		if (!split)
530 			split = alloc_extent_map();
531 		if (!split2)
532 			split2 = alloc_extent_map();
533 		if (!split || !split2)
534 			no_splits = 1;
535 
536 		write_lock(&em_tree->lock);
537 		em = lookup_extent_mapping(em_tree, start, len);
538 		if (!em) {
539 			write_unlock(&em_tree->lock);
540 			break;
541 		}
542 		flags = em->flags;
543 		gen = em->generation;
544 		if (skip_pinned && test_bit(EXTENT_FLAG_PINNED, &em->flags)) {
545 			if (testend && em->start + em->len >= start + len) {
546 				free_extent_map(em);
547 				write_unlock(&em_tree->lock);
548 				break;
549 			}
550 			start = em->start + em->len;
551 			if (testend)
552 				len = start + len - (em->start + em->len);
553 			free_extent_map(em);
554 			write_unlock(&em_tree->lock);
555 			continue;
556 		}
557 		compressed = test_bit(EXTENT_FLAG_COMPRESSED, &em->flags);
558 		clear_bit(EXTENT_FLAG_PINNED, &em->flags);
559 		clear_bit(EXTENT_FLAG_LOGGING, &flags);
560 		modified = !list_empty(&em->list);
561 		if (no_splits)
562 			goto next;
563 
564 		if (em->start < start) {
565 			split->start = em->start;
566 			split->len = start - em->start;
567 
568 			if (em->block_start < EXTENT_MAP_LAST_BYTE) {
569 				split->orig_start = em->orig_start;
570 				split->block_start = em->block_start;
571 
572 				if (compressed)
573 					split->block_len = em->block_len;
574 				else
575 					split->block_len = split->len;
576 				split->orig_block_len = max(split->block_len,
577 						em->orig_block_len);
578 				split->ram_bytes = em->ram_bytes;
579 			} else {
580 				split->orig_start = split->start;
581 				split->block_len = 0;
582 				split->block_start = em->block_start;
583 				split->orig_block_len = 0;
584 				split->ram_bytes = split->len;
585 			}
586 
587 			split->generation = gen;
588 			split->flags = flags;
589 			split->compress_type = em->compress_type;
590 			replace_extent_mapping(em_tree, em, split, modified);
591 			free_extent_map(split);
592 			split = split2;
593 			split2 = NULL;
594 		}
595 		if (testend && em->start + em->len > start + len) {
596 			u64 diff = start + len - em->start;
597 
598 			split->start = start + len;
599 			split->len = em->start + em->len - (start + len);
600 			split->flags = flags;
601 			split->compress_type = em->compress_type;
602 			split->generation = gen;
603 
604 			if (em->block_start < EXTENT_MAP_LAST_BYTE) {
605 				split->orig_block_len = max(em->block_len,
606 						    em->orig_block_len);
607 
608 				split->ram_bytes = em->ram_bytes;
609 				if (compressed) {
610 					split->block_len = em->block_len;
611 					split->block_start = em->block_start;
612 					split->orig_start = em->orig_start;
613 				} else {
614 					split->block_len = split->len;
615 					split->block_start = em->block_start
616 						+ diff;
617 					split->orig_start = em->orig_start;
618 				}
619 			} else {
620 				split->ram_bytes = split->len;
621 				split->orig_start = split->start;
622 				split->block_len = 0;
623 				split->block_start = em->block_start;
624 				split->orig_block_len = 0;
625 			}
626 
627 			if (extent_map_in_tree(em)) {
628 				replace_extent_mapping(em_tree, em, split,
629 						       modified);
630 			} else {
631 				ret = add_extent_mapping(em_tree, split,
632 							 modified);
633 				ASSERT(ret == 0); /* Logic error */
634 			}
635 			free_extent_map(split);
636 			split = NULL;
637 		}
638 next:
639 		if (extent_map_in_tree(em))
640 			remove_extent_mapping(em_tree, em);
641 		write_unlock(&em_tree->lock);
642 
643 		/* once for us */
644 		free_extent_map(em);
645 		/* once for the tree*/
646 		free_extent_map(em);
647 	}
648 	if (split)
649 		free_extent_map(split);
650 	if (split2)
651 		free_extent_map(split2);
652 }
653 
654 /*
655  * this is very complex, but the basic idea is to drop all extents
656  * in the range start - end.  hint_block is filled in with a block number
657  * that would be a good hint to the block allocator for this file.
658  *
659  * If an extent intersects the range but is not entirely inside the range
660  * it is either truncated or split.  Anything entirely inside the range
661  * is deleted from the tree.
662  *
663  * Note: the VFS' inode number of bytes is not updated, it's up to the caller
664  * to deal with that. We set the field 'bytes_found' of the arguments structure
665  * with the number of allocated bytes found in the target range, so that the
666  * caller can update the inode's number of bytes in an atomic way when
667  * replacing extents in a range to avoid races with stat(2).
668  */
669 int btrfs_drop_extents(struct btrfs_trans_handle *trans,
670 		       struct btrfs_root *root, struct btrfs_inode *inode,
671 		       struct btrfs_drop_extents_args *args)
672 {
673 	struct btrfs_fs_info *fs_info = root->fs_info;
674 	struct extent_buffer *leaf;
675 	struct btrfs_file_extent_item *fi;
676 	struct btrfs_ref ref = { 0 };
677 	struct btrfs_key key;
678 	struct btrfs_key new_key;
679 	u64 ino = btrfs_ino(inode);
680 	u64 search_start = args->start;
681 	u64 disk_bytenr = 0;
682 	u64 num_bytes = 0;
683 	u64 extent_offset = 0;
684 	u64 extent_end = 0;
685 	u64 last_end = args->start;
686 	int del_nr = 0;
687 	int del_slot = 0;
688 	int extent_type;
689 	int recow;
690 	int ret;
691 	int modify_tree = -1;
692 	int update_refs;
693 	int found = 0;
694 	struct btrfs_path *path = args->path;
695 
696 	args->bytes_found = 0;
697 	args->extent_inserted = false;
698 
699 	/* Must always have a path if ->replace_extent is true */
700 	ASSERT(!(args->replace_extent && !args->path));
701 
702 	if (!path) {
703 		path = btrfs_alloc_path();
704 		if (!path) {
705 			ret = -ENOMEM;
706 			goto out;
707 		}
708 	}
709 
710 	if (args->drop_cache)
711 		btrfs_drop_extent_cache(inode, args->start, args->end - 1, 0);
712 
713 	if (args->start >= inode->disk_i_size && !args->replace_extent)
714 		modify_tree = 0;
715 
716 	update_refs = (root->root_key.objectid != BTRFS_TREE_LOG_OBJECTID);
717 	while (1) {
718 		recow = 0;
719 		ret = btrfs_lookup_file_extent(trans, root, path, ino,
720 					       search_start, modify_tree);
721 		if (ret < 0)
722 			break;
723 		if (ret > 0 && path->slots[0] > 0 && search_start == args->start) {
724 			leaf = path->nodes[0];
725 			btrfs_item_key_to_cpu(leaf, &key, path->slots[0] - 1);
726 			if (key.objectid == ino &&
727 			    key.type == BTRFS_EXTENT_DATA_KEY)
728 				path->slots[0]--;
729 		}
730 		ret = 0;
731 next_slot:
732 		leaf = path->nodes[0];
733 		if (path->slots[0] >= btrfs_header_nritems(leaf)) {
734 			BUG_ON(del_nr > 0);
735 			ret = btrfs_next_leaf(root, path);
736 			if (ret < 0)
737 				break;
738 			if (ret > 0) {
739 				ret = 0;
740 				break;
741 			}
742 			leaf = path->nodes[0];
743 			recow = 1;
744 		}
745 
746 		btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
747 
748 		if (key.objectid > ino)
749 			break;
750 		if (WARN_ON_ONCE(key.objectid < ino) ||
751 		    key.type < BTRFS_EXTENT_DATA_KEY) {
752 			ASSERT(del_nr == 0);
753 			path->slots[0]++;
754 			goto next_slot;
755 		}
756 		if (key.type > BTRFS_EXTENT_DATA_KEY || key.offset >= args->end)
757 			break;
758 
759 		fi = btrfs_item_ptr(leaf, path->slots[0],
760 				    struct btrfs_file_extent_item);
761 		extent_type = btrfs_file_extent_type(leaf, fi);
762 
763 		if (extent_type == BTRFS_FILE_EXTENT_REG ||
764 		    extent_type == BTRFS_FILE_EXTENT_PREALLOC) {
765 			disk_bytenr = btrfs_file_extent_disk_bytenr(leaf, fi);
766 			num_bytes = btrfs_file_extent_disk_num_bytes(leaf, fi);
767 			extent_offset = btrfs_file_extent_offset(leaf, fi);
768 			extent_end = key.offset +
769 				btrfs_file_extent_num_bytes(leaf, fi);
770 		} else if (extent_type == BTRFS_FILE_EXTENT_INLINE) {
771 			extent_end = key.offset +
772 				btrfs_file_extent_ram_bytes(leaf, fi);
773 		} else {
774 			/* can't happen */
775 			BUG();
776 		}
777 
778 		/*
779 		 * Don't skip extent items representing 0 byte lengths. They
780 		 * used to be created (bug) if while punching holes we hit
781 		 * -ENOSPC condition. So if we find one here, just ensure we
782 		 * delete it, otherwise we would insert a new file extent item
783 		 * with the same key (offset) as that 0 bytes length file
784 		 * extent item in the call to setup_items_for_insert() later
785 		 * in this function.
786 		 */
787 		if (extent_end == key.offset && extent_end >= search_start) {
788 			last_end = extent_end;
789 			goto delete_extent_item;
790 		}
791 
792 		if (extent_end <= search_start) {
793 			path->slots[0]++;
794 			goto next_slot;
795 		}
796 
797 		found = 1;
798 		search_start = max(key.offset, args->start);
799 		if (recow || !modify_tree) {
800 			modify_tree = -1;
801 			btrfs_release_path(path);
802 			continue;
803 		}
804 
805 		/*
806 		 *     | - range to drop - |
807 		 *  | -------- extent -------- |
808 		 */
809 		if (args->start > key.offset && args->end < extent_end) {
810 			BUG_ON(del_nr > 0);
811 			if (extent_type == BTRFS_FILE_EXTENT_INLINE) {
812 				ret = -EOPNOTSUPP;
813 				break;
814 			}
815 
816 			memcpy(&new_key, &key, sizeof(new_key));
817 			new_key.offset = args->start;
818 			ret = btrfs_duplicate_item(trans, root, path,
819 						   &new_key);
820 			if (ret == -EAGAIN) {
821 				btrfs_release_path(path);
822 				continue;
823 			}
824 			if (ret < 0)
825 				break;
826 
827 			leaf = path->nodes[0];
828 			fi = btrfs_item_ptr(leaf, path->slots[0] - 1,
829 					    struct btrfs_file_extent_item);
830 			btrfs_set_file_extent_num_bytes(leaf, fi,
831 							args->start - key.offset);
832 
833 			fi = btrfs_item_ptr(leaf, path->slots[0],
834 					    struct btrfs_file_extent_item);
835 
836 			extent_offset += args->start - key.offset;
837 			btrfs_set_file_extent_offset(leaf, fi, extent_offset);
838 			btrfs_set_file_extent_num_bytes(leaf, fi,
839 							extent_end - args->start);
840 			btrfs_mark_buffer_dirty(leaf);
841 
842 			if (update_refs && disk_bytenr > 0) {
843 				btrfs_init_generic_ref(&ref,
844 						BTRFS_ADD_DELAYED_REF,
845 						disk_bytenr, num_bytes, 0);
846 				btrfs_init_data_ref(&ref,
847 						root->root_key.objectid,
848 						new_key.objectid,
849 						args->start - extent_offset,
850 						0, false);
851 				ret = btrfs_inc_extent_ref(trans, &ref);
852 				BUG_ON(ret); /* -ENOMEM */
853 			}
854 			key.offset = args->start;
855 		}
856 		/*
857 		 * From here on out we will have actually dropped something, so
858 		 * last_end can be updated.
859 		 */
860 		last_end = extent_end;
861 
862 		/*
863 		 *  | ---- range to drop ----- |
864 		 *      | -------- extent -------- |
865 		 */
866 		if (args->start <= key.offset && args->end < extent_end) {
867 			if (extent_type == BTRFS_FILE_EXTENT_INLINE) {
868 				ret = -EOPNOTSUPP;
869 				break;
870 			}
871 
872 			memcpy(&new_key, &key, sizeof(new_key));
873 			new_key.offset = args->end;
874 			btrfs_set_item_key_safe(fs_info, path, &new_key);
875 
876 			extent_offset += args->end - key.offset;
877 			btrfs_set_file_extent_offset(leaf, fi, extent_offset);
878 			btrfs_set_file_extent_num_bytes(leaf, fi,
879 							extent_end - args->end);
880 			btrfs_mark_buffer_dirty(leaf);
881 			if (update_refs && disk_bytenr > 0)
882 				args->bytes_found += args->end - key.offset;
883 			break;
884 		}
885 
886 		search_start = extent_end;
887 		/*
888 		 *       | ---- range to drop ----- |
889 		 *  | -------- extent -------- |
890 		 */
891 		if (args->start > key.offset && args->end >= extent_end) {
892 			BUG_ON(del_nr > 0);
893 			if (extent_type == BTRFS_FILE_EXTENT_INLINE) {
894 				ret = -EOPNOTSUPP;
895 				break;
896 			}
897 
898 			btrfs_set_file_extent_num_bytes(leaf, fi,
899 							args->start - key.offset);
900 			btrfs_mark_buffer_dirty(leaf);
901 			if (update_refs && disk_bytenr > 0)
902 				args->bytes_found += extent_end - args->start;
903 			if (args->end == extent_end)
904 				break;
905 
906 			path->slots[0]++;
907 			goto next_slot;
908 		}
909 
910 		/*
911 		 *  | ---- range to drop ----- |
912 		 *    | ------ extent ------ |
913 		 */
914 		if (args->start <= key.offset && args->end >= extent_end) {
915 delete_extent_item:
916 			if (del_nr == 0) {
917 				del_slot = path->slots[0];
918 				del_nr = 1;
919 			} else {
920 				BUG_ON(del_slot + del_nr != path->slots[0]);
921 				del_nr++;
922 			}
923 
924 			if (update_refs &&
925 			    extent_type == BTRFS_FILE_EXTENT_INLINE) {
926 				args->bytes_found += extent_end - key.offset;
927 				extent_end = ALIGN(extent_end,
928 						   fs_info->sectorsize);
929 			} else if (update_refs && disk_bytenr > 0) {
930 				btrfs_init_generic_ref(&ref,
931 						BTRFS_DROP_DELAYED_REF,
932 						disk_bytenr, num_bytes, 0);
933 				btrfs_init_data_ref(&ref,
934 						root->root_key.objectid,
935 						key.objectid,
936 						key.offset - extent_offset, 0,
937 						false);
938 				ret = btrfs_free_extent(trans, &ref);
939 				BUG_ON(ret); /* -ENOMEM */
940 				args->bytes_found += extent_end - key.offset;
941 			}
942 
943 			if (args->end == extent_end)
944 				break;
945 
946 			if (path->slots[0] + 1 < btrfs_header_nritems(leaf)) {
947 				path->slots[0]++;
948 				goto next_slot;
949 			}
950 
951 			ret = btrfs_del_items(trans, root, path, del_slot,
952 					      del_nr);
953 			if (ret) {
954 				btrfs_abort_transaction(trans, ret);
955 				break;
956 			}
957 
958 			del_nr = 0;
959 			del_slot = 0;
960 
961 			btrfs_release_path(path);
962 			continue;
963 		}
964 
965 		BUG();
966 	}
967 
968 	if (!ret && del_nr > 0) {
969 		/*
970 		 * Set path->slots[0] to first slot, so that after the delete
971 		 * if items are move off from our leaf to its immediate left or
972 		 * right neighbor leafs, we end up with a correct and adjusted
973 		 * path->slots[0] for our insertion (if args->replace_extent).
974 		 */
975 		path->slots[0] = del_slot;
976 		ret = btrfs_del_items(trans, root, path, del_slot, del_nr);
977 		if (ret)
978 			btrfs_abort_transaction(trans, ret);
979 	}
980 
981 	leaf = path->nodes[0];
982 	/*
983 	 * If btrfs_del_items() was called, it might have deleted a leaf, in
984 	 * which case it unlocked our path, so check path->locks[0] matches a
985 	 * write lock.
986 	 */
987 	if (!ret && args->replace_extent &&
988 	    path->locks[0] == BTRFS_WRITE_LOCK &&
989 	    btrfs_leaf_free_space(leaf) >=
990 	    sizeof(struct btrfs_item) + args->extent_item_size) {
991 
992 		key.objectid = ino;
993 		key.type = BTRFS_EXTENT_DATA_KEY;
994 		key.offset = args->start;
995 		if (!del_nr && path->slots[0] < btrfs_header_nritems(leaf)) {
996 			struct btrfs_key slot_key;
997 
998 			btrfs_item_key_to_cpu(leaf, &slot_key, path->slots[0]);
999 			if (btrfs_comp_cpu_keys(&key, &slot_key) > 0)
1000 				path->slots[0]++;
1001 		}
1002 		btrfs_setup_item_for_insert(root, path, &key, args->extent_item_size);
1003 		args->extent_inserted = true;
1004 	}
1005 
1006 	if (!args->path)
1007 		btrfs_free_path(path);
1008 	else if (!args->extent_inserted)
1009 		btrfs_release_path(path);
1010 out:
1011 	args->drop_end = found ? min(args->end, last_end) : args->end;
1012 
1013 	return ret;
1014 }
1015 
1016 static int extent_mergeable(struct extent_buffer *leaf, int slot,
1017 			    u64 objectid, u64 bytenr, u64 orig_offset,
1018 			    u64 *start, u64 *end)
1019 {
1020 	struct btrfs_file_extent_item *fi;
1021 	struct btrfs_key key;
1022 	u64 extent_end;
1023 
1024 	if (slot < 0 || slot >= btrfs_header_nritems(leaf))
1025 		return 0;
1026 
1027 	btrfs_item_key_to_cpu(leaf, &key, slot);
1028 	if (key.objectid != objectid || key.type != BTRFS_EXTENT_DATA_KEY)
1029 		return 0;
1030 
1031 	fi = btrfs_item_ptr(leaf, slot, struct btrfs_file_extent_item);
1032 	if (btrfs_file_extent_type(leaf, fi) != BTRFS_FILE_EXTENT_REG ||
1033 	    btrfs_file_extent_disk_bytenr(leaf, fi) != bytenr ||
1034 	    btrfs_file_extent_offset(leaf, fi) != key.offset - orig_offset ||
1035 	    btrfs_file_extent_compression(leaf, fi) ||
1036 	    btrfs_file_extent_encryption(leaf, fi) ||
1037 	    btrfs_file_extent_other_encoding(leaf, fi))
1038 		return 0;
1039 
1040 	extent_end = key.offset + btrfs_file_extent_num_bytes(leaf, fi);
1041 	if ((*start && *start != key.offset) || (*end && *end != extent_end))
1042 		return 0;
1043 
1044 	*start = key.offset;
1045 	*end = extent_end;
1046 	return 1;
1047 }
1048 
1049 /*
1050  * Mark extent in the range start - end as written.
1051  *
1052  * This changes extent type from 'pre-allocated' to 'regular'. If only
1053  * part of extent is marked as written, the extent will be split into
1054  * two or three.
1055  */
1056 int btrfs_mark_extent_written(struct btrfs_trans_handle *trans,
1057 			      struct btrfs_inode *inode, u64 start, u64 end)
1058 {
1059 	struct btrfs_fs_info *fs_info = trans->fs_info;
1060 	struct btrfs_root *root = inode->root;
1061 	struct extent_buffer *leaf;
1062 	struct btrfs_path *path;
1063 	struct btrfs_file_extent_item *fi;
1064 	struct btrfs_ref ref = { 0 };
1065 	struct btrfs_key key;
1066 	struct btrfs_key new_key;
1067 	u64 bytenr;
1068 	u64 num_bytes;
1069 	u64 extent_end;
1070 	u64 orig_offset;
1071 	u64 other_start;
1072 	u64 other_end;
1073 	u64 split;
1074 	int del_nr = 0;
1075 	int del_slot = 0;
1076 	int recow;
1077 	int ret = 0;
1078 	u64 ino = btrfs_ino(inode);
1079 
1080 	path = btrfs_alloc_path();
1081 	if (!path)
1082 		return -ENOMEM;
1083 again:
1084 	recow = 0;
1085 	split = start;
1086 	key.objectid = ino;
1087 	key.type = BTRFS_EXTENT_DATA_KEY;
1088 	key.offset = split;
1089 
1090 	ret = btrfs_search_slot(trans, root, &key, path, -1, 1);
1091 	if (ret < 0)
1092 		goto out;
1093 	if (ret > 0 && path->slots[0] > 0)
1094 		path->slots[0]--;
1095 
1096 	leaf = path->nodes[0];
1097 	btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
1098 	if (key.objectid != ino ||
1099 	    key.type != BTRFS_EXTENT_DATA_KEY) {
1100 		ret = -EINVAL;
1101 		btrfs_abort_transaction(trans, ret);
1102 		goto out;
1103 	}
1104 	fi = btrfs_item_ptr(leaf, path->slots[0],
1105 			    struct btrfs_file_extent_item);
1106 	if (btrfs_file_extent_type(leaf, fi) != BTRFS_FILE_EXTENT_PREALLOC) {
1107 		ret = -EINVAL;
1108 		btrfs_abort_transaction(trans, ret);
1109 		goto out;
1110 	}
1111 	extent_end = key.offset + btrfs_file_extent_num_bytes(leaf, fi);
1112 	if (key.offset > start || extent_end < end) {
1113 		ret = -EINVAL;
1114 		btrfs_abort_transaction(trans, ret);
1115 		goto out;
1116 	}
1117 
1118 	bytenr = btrfs_file_extent_disk_bytenr(leaf, fi);
1119 	num_bytes = btrfs_file_extent_disk_num_bytes(leaf, fi);
1120 	orig_offset = key.offset - btrfs_file_extent_offset(leaf, fi);
1121 	memcpy(&new_key, &key, sizeof(new_key));
1122 
1123 	if (start == key.offset && end < extent_end) {
1124 		other_start = 0;
1125 		other_end = start;
1126 		if (extent_mergeable(leaf, path->slots[0] - 1,
1127 				     ino, bytenr, orig_offset,
1128 				     &other_start, &other_end)) {
1129 			new_key.offset = end;
1130 			btrfs_set_item_key_safe(fs_info, path, &new_key);
1131 			fi = btrfs_item_ptr(leaf, path->slots[0],
1132 					    struct btrfs_file_extent_item);
1133 			btrfs_set_file_extent_generation(leaf, fi,
1134 							 trans->transid);
1135 			btrfs_set_file_extent_num_bytes(leaf, fi,
1136 							extent_end - end);
1137 			btrfs_set_file_extent_offset(leaf, fi,
1138 						     end - orig_offset);
1139 			fi = btrfs_item_ptr(leaf, path->slots[0] - 1,
1140 					    struct btrfs_file_extent_item);
1141 			btrfs_set_file_extent_generation(leaf, fi,
1142 							 trans->transid);
1143 			btrfs_set_file_extent_num_bytes(leaf, fi,
1144 							end - other_start);
1145 			btrfs_mark_buffer_dirty(leaf);
1146 			goto out;
1147 		}
1148 	}
1149 
1150 	if (start > key.offset && end == extent_end) {
1151 		other_start = end;
1152 		other_end = 0;
1153 		if (extent_mergeable(leaf, path->slots[0] + 1,
1154 				     ino, bytenr, orig_offset,
1155 				     &other_start, &other_end)) {
1156 			fi = btrfs_item_ptr(leaf, path->slots[0],
1157 					    struct btrfs_file_extent_item);
1158 			btrfs_set_file_extent_num_bytes(leaf, fi,
1159 							start - key.offset);
1160 			btrfs_set_file_extent_generation(leaf, fi,
1161 							 trans->transid);
1162 			path->slots[0]++;
1163 			new_key.offset = start;
1164 			btrfs_set_item_key_safe(fs_info, path, &new_key);
1165 
1166 			fi = btrfs_item_ptr(leaf, path->slots[0],
1167 					    struct btrfs_file_extent_item);
1168 			btrfs_set_file_extent_generation(leaf, fi,
1169 							 trans->transid);
1170 			btrfs_set_file_extent_num_bytes(leaf, fi,
1171 							other_end - start);
1172 			btrfs_set_file_extent_offset(leaf, fi,
1173 						     start - orig_offset);
1174 			btrfs_mark_buffer_dirty(leaf);
1175 			goto out;
1176 		}
1177 	}
1178 
1179 	while (start > key.offset || end < extent_end) {
1180 		if (key.offset == start)
1181 			split = end;
1182 
1183 		new_key.offset = split;
1184 		ret = btrfs_duplicate_item(trans, root, path, &new_key);
1185 		if (ret == -EAGAIN) {
1186 			btrfs_release_path(path);
1187 			goto again;
1188 		}
1189 		if (ret < 0) {
1190 			btrfs_abort_transaction(trans, ret);
1191 			goto out;
1192 		}
1193 
1194 		leaf = path->nodes[0];
1195 		fi = btrfs_item_ptr(leaf, path->slots[0] - 1,
1196 				    struct btrfs_file_extent_item);
1197 		btrfs_set_file_extent_generation(leaf, fi, trans->transid);
1198 		btrfs_set_file_extent_num_bytes(leaf, fi,
1199 						split - key.offset);
1200 
1201 		fi = btrfs_item_ptr(leaf, path->slots[0],
1202 				    struct btrfs_file_extent_item);
1203 
1204 		btrfs_set_file_extent_generation(leaf, fi, trans->transid);
1205 		btrfs_set_file_extent_offset(leaf, fi, split - orig_offset);
1206 		btrfs_set_file_extent_num_bytes(leaf, fi,
1207 						extent_end - split);
1208 		btrfs_mark_buffer_dirty(leaf);
1209 
1210 		btrfs_init_generic_ref(&ref, BTRFS_ADD_DELAYED_REF, bytenr,
1211 				       num_bytes, 0);
1212 		btrfs_init_data_ref(&ref, root->root_key.objectid, ino,
1213 				    orig_offset, 0, false);
1214 		ret = btrfs_inc_extent_ref(trans, &ref);
1215 		if (ret) {
1216 			btrfs_abort_transaction(trans, ret);
1217 			goto out;
1218 		}
1219 
1220 		if (split == start) {
1221 			key.offset = start;
1222 		} else {
1223 			if (start != key.offset) {
1224 				ret = -EINVAL;
1225 				btrfs_abort_transaction(trans, ret);
1226 				goto out;
1227 			}
1228 			path->slots[0]--;
1229 			extent_end = end;
1230 		}
1231 		recow = 1;
1232 	}
1233 
1234 	other_start = end;
1235 	other_end = 0;
1236 	btrfs_init_generic_ref(&ref, BTRFS_DROP_DELAYED_REF, bytenr,
1237 			       num_bytes, 0);
1238 	btrfs_init_data_ref(&ref, root->root_key.objectid, ino, orig_offset,
1239 			    0, false);
1240 	if (extent_mergeable(leaf, path->slots[0] + 1,
1241 			     ino, bytenr, orig_offset,
1242 			     &other_start, &other_end)) {
1243 		if (recow) {
1244 			btrfs_release_path(path);
1245 			goto again;
1246 		}
1247 		extent_end = other_end;
1248 		del_slot = path->slots[0] + 1;
1249 		del_nr++;
1250 		ret = btrfs_free_extent(trans, &ref);
1251 		if (ret) {
1252 			btrfs_abort_transaction(trans, ret);
1253 			goto out;
1254 		}
1255 	}
1256 	other_start = 0;
1257 	other_end = start;
1258 	if (extent_mergeable(leaf, path->slots[0] - 1,
1259 			     ino, bytenr, orig_offset,
1260 			     &other_start, &other_end)) {
1261 		if (recow) {
1262 			btrfs_release_path(path);
1263 			goto again;
1264 		}
1265 		key.offset = other_start;
1266 		del_slot = path->slots[0];
1267 		del_nr++;
1268 		ret = btrfs_free_extent(trans, &ref);
1269 		if (ret) {
1270 			btrfs_abort_transaction(trans, ret);
1271 			goto out;
1272 		}
1273 	}
1274 	if (del_nr == 0) {
1275 		fi = btrfs_item_ptr(leaf, path->slots[0],
1276 			   struct btrfs_file_extent_item);
1277 		btrfs_set_file_extent_type(leaf, fi,
1278 					   BTRFS_FILE_EXTENT_REG);
1279 		btrfs_set_file_extent_generation(leaf, fi, trans->transid);
1280 		btrfs_mark_buffer_dirty(leaf);
1281 	} else {
1282 		fi = btrfs_item_ptr(leaf, del_slot - 1,
1283 			   struct btrfs_file_extent_item);
1284 		btrfs_set_file_extent_type(leaf, fi,
1285 					   BTRFS_FILE_EXTENT_REG);
1286 		btrfs_set_file_extent_generation(leaf, fi, trans->transid);
1287 		btrfs_set_file_extent_num_bytes(leaf, fi,
1288 						extent_end - key.offset);
1289 		btrfs_mark_buffer_dirty(leaf);
1290 
1291 		ret = btrfs_del_items(trans, root, path, del_slot, del_nr);
1292 		if (ret < 0) {
1293 			btrfs_abort_transaction(trans, ret);
1294 			goto out;
1295 		}
1296 	}
1297 out:
1298 	btrfs_free_path(path);
1299 	return ret;
1300 }
1301 
1302 /*
1303  * on error we return an unlocked page and the error value
1304  * on success we return a locked page and 0
1305  */
1306 static int prepare_uptodate_page(struct inode *inode,
1307 				 struct page *page, u64 pos,
1308 				 bool force_uptodate)
1309 {
1310 	struct folio *folio = page_folio(page);
1311 	int ret = 0;
1312 
1313 	if (((pos & (PAGE_SIZE - 1)) || force_uptodate) &&
1314 	    !PageUptodate(page)) {
1315 		ret = btrfs_read_folio(NULL, folio);
1316 		if (ret)
1317 			return ret;
1318 		lock_page(page);
1319 		if (!PageUptodate(page)) {
1320 			unlock_page(page);
1321 			return -EIO;
1322 		}
1323 
1324 		/*
1325 		 * Since btrfs_read_folio() will unlock the folio before it
1326 		 * returns, there is a window where btrfs_release_folio() can be
1327 		 * called to release the page.  Here we check both inode
1328 		 * mapping and PagePrivate() to make sure the page was not
1329 		 * released.
1330 		 *
1331 		 * The private flag check is essential for subpage as we need
1332 		 * to store extra bitmap using page->private.
1333 		 */
1334 		if (page->mapping != inode->i_mapping || !PagePrivate(page)) {
1335 			unlock_page(page);
1336 			return -EAGAIN;
1337 		}
1338 	}
1339 	return 0;
1340 }
1341 
1342 /*
1343  * this just gets pages into the page cache and locks them down.
1344  */
1345 static noinline int prepare_pages(struct inode *inode, struct page **pages,
1346 				  size_t num_pages, loff_t pos,
1347 				  size_t write_bytes, bool force_uptodate)
1348 {
1349 	int i;
1350 	unsigned long index = pos >> PAGE_SHIFT;
1351 	gfp_t mask = btrfs_alloc_write_mask(inode->i_mapping);
1352 	int err = 0;
1353 	int faili;
1354 
1355 	for (i = 0; i < num_pages; i++) {
1356 again:
1357 		pages[i] = find_or_create_page(inode->i_mapping, index + i,
1358 					       mask | __GFP_WRITE);
1359 		if (!pages[i]) {
1360 			faili = i - 1;
1361 			err = -ENOMEM;
1362 			goto fail;
1363 		}
1364 
1365 		err = set_page_extent_mapped(pages[i]);
1366 		if (err < 0) {
1367 			faili = i;
1368 			goto fail;
1369 		}
1370 
1371 		if (i == 0)
1372 			err = prepare_uptodate_page(inode, pages[i], pos,
1373 						    force_uptodate);
1374 		if (!err && i == num_pages - 1)
1375 			err = prepare_uptodate_page(inode, pages[i],
1376 						    pos + write_bytes, false);
1377 		if (err) {
1378 			put_page(pages[i]);
1379 			if (err == -EAGAIN) {
1380 				err = 0;
1381 				goto again;
1382 			}
1383 			faili = i - 1;
1384 			goto fail;
1385 		}
1386 		wait_on_page_writeback(pages[i]);
1387 	}
1388 
1389 	return 0;
1390 fail:
1391 	while (faili >= 0) {
1392 		unlock_page(pages[faili]);
1393 		put_page(pages[faili]);
1394 		faili--;
1395 	}
1396 	return err;
1397 
1398 }
1399 
1400 /*
1401  * This function locks the extent and properly waits for data=ordered extents
1402  * to finish before allowing the pages to be modified if need.
1403  *
1404  * The return value:
1405  * 1 - the extent is locked
1406  * 0 - the extent is not locked, and everything is OK
1407  * -EAGAIN - need re-prepare the pages
1408  * the other < 0 number - Something wrong happens
1409  */
1410 static noinline int
1411 lock_and_cleanup_extent_if_need(struct btrfs_inode *inode, struct page **pages,
1412 				size_t num_pages, loff_t pos,
1413 				size_t write_bytes,
1414 				u64 *lockstart, u64 *lockend,
1415 				struct extent_state **cached_state)
1416 {
1417 	struct btrfs_fs_info *fs_info = inode->root->fs_info;
1418 	u64 start_pos;
1419 	u64 last_pos;
1420 	int i;
1421 	int ret = 0;
1422 
1423 	start_pos = round_down(pos, fs_info->sectorsize);
1424 	last_pos = round_up(pos + write_bytes, fs_info->sectorsize) - 1;
1425 
1426 	if (start_pos < inode->vfs_inode.i_size) {
1427 		struct btrfs_ordered_extent *ordered;
1428 
1429 		lock_extent_bits(&inode->io_tree, start_pos, last_pos,
1430 				cached_state);
1431 		ordered = btrfs_lookup_ordered_range(inode, start_pos,
1432 						     last_pos - start_pos + 1);
1433 		if (ordered &&
1434 		    ordered->file_offset + ordered->num_bytes > start_pos &&
1435 		    ordered->file_offset <= last_pos) {
1436 			unlock_extent_cached(&inode->io_tree, start_pos,
1437 					last_pos, cached_state);
1438 			for (i = 0; i < num_pages; i++) {
1439 				unlock_page(pages[i]);
1440 				put_page(pages[i]);
1441 			}
1442 			btrfs_start_ordered_extent(ordered, 1);
1443 			btrfs_put_ordered_extent(ordered);
1444 			return -EAGAIN;
1445 		}
1446 		if (ordered)
1447 			btrfs_put_ordered_extent(ordered);
1448 
1449 		*lockstart = start_pos;
1450 		*lockend = last_pos;
1451 		ret = 1;
1452 	}
1453 
1454 	/*
1455 	 * We should be called after prepare_pages() which should have locked
1456 	 * all pages in the range.
1457 	 */
1458 	for (i = 0; i < num_pages; i++)
1459 		WARN_ON(!PageLocked(pages[i]));
1460 
1461 	return ret;
1462 }
1463 
1464 /*
1465  * Check if we can do nocow write into the range [@pos, @pos + @write_bytes)
1466  *
1467  * @pos:         File offset.
1468  * @write_bytes: The length to write, will be updated to the nocow writeable
1469  *               range.
1470  *
1471  * This function will flush ordered extents in the range to ensure proper
1472  * nocow checks.
1473  *
1474  * Return:
1475  * > 0          If we can nocow, and updates @write_bytes.
1476  *  0           If we can't do a nocow write.
1477  * -EAGAIN      If we can't do a nocow write because snapshoting of the inode's
1478  *              root is in progress.
1479  * < 0          If an error happened.
1480  *
1481  * NOTE: Callers need to call btrfs_check_nocow_unlock() if we return > 0.
1482  */
1483 int btrfs_check_nocow_lock(struct btrfs_inode *inode, loff_t pos,
1484 			   size_t *write_bytes)
1485 {
1486 	struct btrfs_fs_info *fs_info = inode->root->fs_info;
1487 	struct btrfs_root *root = inode->root;
1488 	u64 lockstart, lockend;
1489 	u64 num_bytes;
1490 	int ret;
1491 
1492 	if (!(inode->flags & (BTRFS_INODE_NODATACOW | BTRFS_INODE_PREALLOC)))
1493 		return 0;
1494 
1495 	if (!btrfs_drew_try_write_lock(&root->snapshot_lock))
1496 		return -EAGAIN;
1497 
1498 	lockstart = round_down(pos, fs_info->sectorsize);
1499 	lockend = round_up(pos + *write_bytes,
1500 			   fs_info->sectorsize) - 1;
1501 	num_bytes = lockend - lockstart + 1;
1502 
1503 	btrfs_lock_and_flush_ordered_range(inode, lockstart, lockend, NULL);
1504 	ret = can_nocow_extent(&inode->vfs_inode, lockstart, &num_bytes,
1505 			NULL, NULL, NULL, false);
1506 	if (ret <= 0) {
1507 		ret = 0;
1508 		btrfs_drew_write_unlock(&root->snapshot_lock);
1509 	} else {
1510 		*write_bytes = min_t(size_t, *write_bytes ,
1511 				     num_bytes - pos + lockstart);
1512 	}
1513 	unlock_extent(&inode->io_tree, lockstart, lockend);
1514 
1515 	return ret;
1516 }
1517 
1518 void btrfs_check_nocow_unlock(struct btrfs_inode *inode)
1519 {
1520 	btrfs_drew_write_unlock(&inode->root->snapshot_lock);
1521 }
1522 
1523 static void update_time_for_write(struct inode *inode)
1524 {
1525 	struct timespec64 now;
1526 
1527 	if (IS_NOCMTIME(inode))
1528 		return;
1529 
1530 	now = current_time(inode);
1531 	if (!timespec64_equal(&inode->i_mtime, &now))
1532 		inode->i_mtime = now;
1533 
1534 	if (!timespec64_equal(&inode->i_ctime, &now))
1535 		inode->i_ctime = now;
1536 
1537 	if (IS_I_VERSION(inode))
1538 		inode_inc_iversion(inode);
1539 }
1540 
1541 static int btrfs_write_check(struct kiocb *iocb, struct iov_iter *from,
1542 			     size_t count)
1543 {
1544 	struct file *file = iocb->ki_filp;
1545 	struct inode *inode = file_inode(file);
1546 	struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
1547 	loff_t pos = iocb->ki_pos;
1548 	int ret;
1549 	loff_t oldsize;
1550 	loff_t start_pos;
1551 
1552 	/*
1553 	 * Quickly bail out on NOWAIT writes if we don't have the nodatacow or
1554 	 * prealloc flags, as without those flags we always have to COW. We will
1555 	 * later check if we can really COW into the target range (using
1556 	 * can_nocow_extent() at btrfs_get_blocks_direct_write()).
1557 	 */
1558 	if ((iocb->ki_flags & IOCB_NOWAIT) &&
1559 	    !(BTRFS_I(inode)->flags & (BTRFS_INODE_NODATACOW | BTRFS_INODE_PREALLOC)))
1560 		return -EAGAIN;
1561 
1562 	current->backing_dev_info = inode_to_bdi(inode);
1563 	ret = file_remove_privs(file);
1564 	if (ret)
1565 		return ret;
1566 
1567 	/*
1568 	 * We reserve space for updating the inode when we reserve space for the
1569 	 * extent we are going to write, so we will enospc out there.  We don't
1570 	 * need to start yet another transaction to update the inode as we will
1571 	 * update the inode when we finish writing whatever data we write.
1572 	 */
1573 	update_time_for_write(inode);
1574 
1575 	start_pos = round_down(pos, fs_info->sectorsize);
1576 	oldsize = i_size_read(inode);
1577 	if (start_pos > oldsize) {
1578 		/* Expand hole size to cover write data, preventing empty gap */
1579 		loff_t end_pos = round_up(pos + count, fs_info->sectorsize);
1580 
1581 		ret = btrfs_cont_expand(BTRFS_I(inode), oldsize, end_pos);
1582 		if (ret) {
1583 			current->backing_dev_info = NULL;
1584 			return ret;
1585 		}
1586 	}
1587 
1588 	return 0;
1589 }
1590 
1591 static noinline ssize_t btrfs_buffered_write(struct kiocb *iocb,
1592 					       struct iov_iter *i)
1593 {
1594 	struct file *file = iocb->ki_filp;
1595 	loff_t pos;
1596 	struct inode *inode = file_inode(file);
1597 	struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
1598 	struct page **pages = NULL;
1599 	struct extent_changeset *data_reserved = NULL;
1600 	u64 release_bytes = 0;
1601 	u64 lockstart;
1602 	u64 lockend;
1603 	size_t num_written = 0;
1604 	int nrptrs;
1605 	ssize_t ret;
1606 	bool only_release_metadata = false;
1607 	bool force_page_uptodate = false;
1608 	loff_t old_isize = i_size_read(inode);
1609 	unsigned int ilock_flags = 0;
1610 
1611 	if (iocb->ki_flags & IOCB_NOWAIT)
1612 		ilock_flags |= BTRFS_ILOCK_TRY;
1613 
1614 	ret = btrfs_inode_lock(inode, ilock_flags);
1615 	if (ret < 0)
1616 		return ret;
1617 
1618 	ret = generic_write_checks(iocb, i);
1619 	if (ret <= 0)
1620 		goto out;
1621 
1622 	ret = btrfs_write_check(iocb, i, ret);
1623 	if (ret < 0)
1624 		goto out;
1625 
1626 	pos = iocb->ki_pos;
1627 	nrptrs = min(DIV_ROUND_UP(iov_iter_count(i), PAGE_SIZE),
1628 			PAGE_SIZE / (sizeof(struct page *)));
1629 	nrptrs = min(nrptrs, current->nr_dirtied_pause - current->nr_dirtied);
1630 	nrptrs = max(nrptrs, 8);
1631 	pages = kmalloc_array(nrptrs, sizeof(struct page *), GFP_KERNEL);
1632 	if (!pages) {
1633 		ret = -ENOMEM;
1634 		goto out;
1635 	}
1636 
1637 	while (iov_iter_count(i) > 0) {
1638 		struct extent_state *cached_state = NULL;
1639 		size_t offset = offset_in_page(pos);
1640 		size_t sector_offset;
1641 		size_t write_bytes = min(iov_iter_count(i),
1642 					 nrptrs * (size_t)PAGE_SIZE -
1643 					 offset);
1644 		size_t num_pages;
1645 		size_t reserve_bytes;
1646 		size_t dirty_pages;
1647 		size_t copied;
1648 		size_t dirty_sectors;
1649 		size_t num_sectors;
1650 		int extents_locked;
1651 
1652 		/*
1653 		 * Fault pages before locking them in prepare_pages
1654 		 * to avoid recursive lock
1655 		 */
1656 		if (unlikely(fault_in_iov_iter_readable(i, write_bytes))) {
1657 			ret = -EFAULT;
1658 			break;
1659 		}
1660 
1661 		only_release_metadata = false;
1662 		sector_offset = pos & (fs_info->sectorsize - 1);
1663 
1664 		extent_changeset_release(data_reserved);
1665 		ret = btrfs_check_data_free_space(BTRFS_I(inode),
1666 						  &data_reserved, pos,
1667 						  write_bytes);
1668 		if (ret < 0) {
1669 			/*
1670 			 * If we don't have to COW at the offset, reserve
1671 			 * metadata only. write_bytes may get smaller than
1672 			 * requested here.
1673 			 */
1674 			if (btrfs_check_nocow_lock(BTRFS_I(inode), pos,
1675 						   &write_bytes) > 0)
1676 				only_release_metadata = true;
1677 			else
1678 				break;
1679 		}
1680 
1681 		num_pages = DIV_ROUND_UP(write_bytes + offset, PAGE_SIZE);
1682 		WARN_ON(num_pages > nrptrs);
1683 		reserve_bytes = round_up(write_bytes + sector_offset,
1684 					 fs_info->sectorsize);
1685 		WARN_ON(reserve_bytes == 0);
1686 		ret = btrfs_delalloc_reserve_metadata(BTRFS_I(inode),
1687 						      reserve_bytes,
1688 						      reserve_bytes, false);
1689 		if (ret) {
1690 			if (!only_release_metadata)
1691 				btrfs_free_reserved_data_space(BTRFS_I(inode),
1692 						data_reserved, pos,
1693 						write_bytes);
1694 			else
1695 				btrfs_check_nocow_unlock(BTRFS_I(inode));
1696 			break;
1697 		}
1698 
1699 		release_bytes = reserve_bytes;
1700 again:
1701 		/*
1702 		 * This is going to setup the pages array with the number of
1703 		 * pages we want, so we don't really need to worry about the
1704 		 * contents of pages from loop to loop
1705 		 */
1706 		ret = prepare_pages(inode, pages, num_pages,
1707 				    pos, write_bytes,
1708 				    force_page_uptodate);
1709 		if (ret) {
1710 			btrfs_delalloc_release_extents(BTRFS_I(inode),
1711 						       reserve_bytes);
1712 			break;
1713 		}
1714 
1715 		extents_locked = lock_and_cleanup_extent_if_need(
1716 				BTRFS_I(inode), pages,
1717 				num_pages, pos, write_bytes, &lockstart,
1718 				&lockend, &cached_state);
1719 		if (extents_locked < 0) {
1720 			if (extents_locked == -EAGAIN)
1721 				goto again;
1722 			btrfs_delalloc_release_extents(BTRFS_I(inode),
1723 						       reserve_bytes);
1724 			ret = extents_locked;
1725 			break;
1726 		}
1727 
1728 		copied = btrfs_copy_from_user(pos, write_bytes, pages, i);
1729 
1730 		num_sectors = BTRFS_BYTES_TO_BLKS(fs_info, reserve_bytes);
1731 		dirty_sectors = round_up(copied + sector_offset,
1732 					fs_info->sectorsize);
1733 		dirty_sectors = BTRFS_BYTES_TO_BLKS(fs_info, dirty_sectors);
1734 
1735 		/*
1736 		 * if we have trouble faulting in the pages, fall
1737 		 * back to one page at a time
1738 		 */
1739 		if (copied < write_bytes)
1740 			nrptrs = 1;
1741 
1742 		if (copied == 0) {
1743 			force_page_uptodate = true;
1744 			dirty_sectors = 0;
1745 			dirty_pages = 0;
1746 		} else {
1747 			force_page_uptodate = false;
1748 			dirty_pages = DIV_ROUND_UP(copied + offset,
1749 						   PAGE_SIZE);
1750 		}
1751 
1752 		if (num_sectors > dirty_sectors) {
1753 			/* release everything except the sectors we dirtied */
1754 			release_bytes -= dirty_sectors << fs_info->sectorsize_bits;
1755 			if (only_release_metadata) {
1756 				btrfs_delalloc_release_metadata(BTRFS_I(inode),
1757 							release_bytes, true);
1758 			} else {
1759 				u64 __pos;
1760 
1761 				__pos = round_down(pos,
1762 						   fs_info->sectorsize) +
1763 					(dirty_pages << PAGE_SHIFT);
1764 				btrfs_delalloc_release_space(BTRFS_I(inode),
1765 						data_reserved, __pos,
1766 						release_bytes, true);
1767 			}
1768 		}
1769 
1770 		release_bytes = round_up(copied + sector_offset,
1771 					fs_info->sectorsize);
1772 
1773 		ret = btrfs_dirty_pages(BTRFS_I(inode), pages,
1774 					dirty_pages, pos, copied,
1775 					&cached_state, only_release_metadata);
1776 
1777 		/*
1778 		 * If we have not locked the extent range, because the range's
1779 		 * start offset is >= i_size, we might still have a non-NULL
1780 		 * cached extent state, acquired while marking the extent range
1781 		 * as delalloc through btrfs_dirty_pages(). Therefore free any
1782 		 * possible cached extent state to avoid a memory leak.
1783 		 */
1784 		if (extents_locked)
1785 			unlock_extent_cached(&BTRFS_I(inode)->io_tree,
1786 					     lockstart, lockend, &cached_state);
1787 		else
1788 			free_extent_state(cached_state);
1789 
1790 		btrfs_delalloc_release_extents(BTRFS_I(inode), reserve_bytes);
1791 		if (ret) {
1792 			btrfs_drop_pages(fs_info, pages, num_pages, pos, copied);
1793 			break;
1794 		}
1795 
1796 		release_bytes = 0;
1797 		if (only_release_metadata)
1798 			btrfs_check_nocow_unlock(BTRFS_I(inode));
1799 
1800 		btrfs_drop_pages(fs_info, pages, num_pages, pos, copied);
1801 
1802 		cond_resched();
1803 
1804 		balance_dirty_pages_ratelimited(inode->i_mapping);
1805 
1806 		pos += copied;
1807 		num_written += copied;
1808 	}
1809 
1810 	kfree(pages);
1811 
1812 	if (release_bytes) {
1813 		if (only_release_metadata) {
1814 			btrfs_check_nocow_unlock(BTRFS_I(inode));
1815 			btrfs_delalloc_release_metadata(BTRFS_I(inode),
1816 					release_bytes, true);
1817 		} else {
1818 			btrfs_delalloc_release_space(BTRFS_I(inode),
1819 					data_reserved,
1820 					round_down(pos, fs_info->sectorsize),
1821 					release_bytes, true);
1822 		}
1823 	}
1824 
1825 	extent_changeset_free(data_reserved);
1826 	if (num_written > 0) {
1827 		pagecache_isize_extended(inode, old_isize, iocb->ki_pos);
1828 		iocb->ki_pos += num_written;
1829 	}
1830 out:
1831 	btrfs_inode_unlock(inode, ilock_flags);
1832 	return num_written ? num_written : ret;
1833 }
1834 
1835 static ssize_t check_direct_IO(struct btrfs_fs_info *fs_info,
1836 			       const struct iov_iter *iter, loff_t offset)
1837 {
1838 	const u32 blocksize_mask = fs_info->sectorsize - 1;
1839 
1840 	if (offset & blocksize_mask)
1841 		return -EINVAL;
1842 
1843 	if (iov_iter_alignment(iter) & blocksize_mask)
1844 		return -EINVAL;
1845 
1846 	return 0;
1847 }
1848 
1849 static ssize_t btrfs_direct_write(struct kiocb *iocb, struct iov_iter *from)
1850 {
1851 	const bool is_sync_write = (iocb->ki_flags & IOCB_DSYNC);
1852 	struct file *file = iocb->ki_filp;
1853 	struct inode *inode = file_inode(file);
1854 	struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
1855 	loff_t pos;
1856 	ssize_t written = 0;
1857 	ssize_t written_buffered;
1858 	size_t prev_left = 0;
1859 	loff_t endbyte;
1860 	ssize_t err;
1861 	unsigned int ilock_flags = 0;
1862 
1863 	if (iocb->ki_flags & IOCB_NOWAIT)
1864 		ilock_flags |= BTRFS_ILOCK_TRY;
1865 
1866 	/* If the write DIO is within EOF, use a shared lock */
1867 	if (iocb->ki_pos + iov_iter_count(from) <= i_size_read(inode))
1868 		ilock_flags |= BTRFS_ILOCK_SHARED;
1869 
1870 relock:
1871 	err = btrfs_inode_lock(inode, ilock_flags);
1872 	if (err < 0)
1873 		return err;
1874 
1875 	err = generic_write_checks(iocb, from);
1876 	if (err <= 0) {
1877 		btrfs_inode_unlock(inode, ilock_flags);
1878 		return err;
1879 	}
1880 
1881 	err = btrfs_write_check(iocb, from, err);
1882 	if (err < 0) {
1883 		btrfs_inode_unlock(inode, ilock_flags);
1884 		goto out;
1885 	}
1886 
1887 	pos = iocb->ki_pos;
1888 	/*
1889 	 * Re-check since file size may have changed just before taking the
1890 	 * lock or pos may have changed because of O_APPEND in generic_write_check()
1891 	 */
1892 	if ((ilock_flags & BTRFS_ILOCK_SHARED) &&
1893 	    pos + iov_iter_count(from) > i_size_read(inode)) {
1894 		btrfs_inode_unlock(inode, ilock_flags);
1895 		ilock_flags &= ~BTRFS_ILOCK_SHARED;
1896 		goto relock;
1897 	}
1898 
1899 	if (check_direct_IO(fs_info, from, pos)) {
1900 		btrfs_inode_unlock(inode, ilock_flags);
1901 		goto buffered;
1902 	}
1903 
1904 	/*
1905 	 * We remove IOCB_DSYNC so that we don't deadlock when iomap_dio_rw()
1906 	 * calls generic_write_sync() (through iomap_dio_complete()), because
1907 	 * that results in calling fsync (btrfs_sync_file()) which will try to
1908 	 * lock the inode in exclusive/write mode.
1909 	 */
1910 	if (is_sync_write)
1911 		iocb->ki_flags &= ~IOCB_DSYNC;
1912 
1913 	/*
1914 	 * The iov_iter can be mapped to the same file range we are writing to.
1915 	 * If that's the case, then we will deadlock in the iomap code, because
1916 	 * it first calls our callback btrfs_dio_iomap_begin(), which will create
1917 	 * an ordered extent, and after that it will fault in the pages that the
1918 	 * iov_iter refers to. During the fault in we end up in the readahead
1919 	 * pages code (starting at btrfs_readahead()), which will lock the range,
1920 	 * find that ordered extent and then wait for it to complete (at
1921 	 * btrfs_lock_and_flush_ordered_range()), resulting in a deadlock since
1922 	 * obviously the ordered extent can never complete as we didn't submit
1923 	 * yet the respective bio(s). This always happens when the buffer is
1924 	 * memory mapped to the same file range, since the iomap DIO code always
1925 	 * invalidates pages in the target file range (after starting and waiting
1926 	 * for any writeback).
1927 	 *
1928 	 * So here we disable page faults in the iov_iter and then retry if we
1929 	 * got -EFAULT, faulting in the pages before the retry.
1930 	 */
1931 again:
1932 	from->nofault = true;
1933 	err = btrfs_dio_rw(iocb, from, written);
1934 	from->nofault = false;
1935 
1936 	/* No increment (+=) because iomap returns a cumulative value. */
1937 	if (err > 0)
1938 		written = err;
1939 
1940 	if (iov_iter_count(from) > 0 && (err == -EFAULT || err > 0)) {
1941 		const size_t left = iov_iter_count(from);
1942 		/*
1943 		 * We have more data left to write. Try to fault in as many as
1944 		 * possible of the remainder pages and retry. We do this without
1945 		 * releasing and locking again the inode, to prevent races with
1946 		 * truncate.
1947 		 *
1948 		 * Also, in case the iov refers to pages in the file range of the
1949 		 * file we want to write to (due to a mmap), we could enter an
1950 		 * infinite loop if we retry after faulting the pages in, since
1951 		 * iomap will invalidate any pages in the range early on, before
1952 		 * it tries to fault in the pages of the iov. So we keep track of
1953 		 * how much was left of iov in the previous EFAULT and fallback
1954 		 * to buffered IO in case we haven't made any progress.
1955 		 */
1956 		if (left == prev_left) {
1957 			err = -ENOTBLK;
1958 		} else {
1959 			fault_in_iov_iter_readable(from, left);
1960 			prev_left = left;
1961 			goto again;
1962 		}
1963 	}
1964 
1965 	btrfs_inode_unlock(inode, ilock_flags);
1966 
1967 	/*
1968 	 * Add back IOCB_DSYNC. Our caller, btrfs_file_write_iter(), will do
1969 	 * the fsync (call generic_write_sync()).
1970 	 */
1971 	if (is_sync_write)
1972 		iocb->ki_flags |= IOCB_DSYNC;
1973 
1974 	/* If 'err' is -ENOTBLK then it means we must fallback to buffered IO. */
1975 	if ((err < 0 && err != -ENOTBLK) || !iov_iter_count(from))
1976 		goto out;
1977 
1978 buffered:
1979 	pos = iocb->ki_pos;
1980 	written_buffered = btrfs_buffered_write(iocb, from);
1981 	if (written_buffered < 0) {
1982 		err = written_buffered;
1983 		goto out;
1984 	}
1985 	/*
1986 	 * Ensure all data is persisted. We want the next direct IO read to be
1987 	 * able to read what was just written.
1988 	 */
1989 	endbyte = pos + written_buffered - 1;
1990 	err = btrfs_fdatawrite_range(inode, pos, endbyte);
1991 	if (err)
1992 		goto out;
1993 	err = filemap_fdatawait_range(inode->i_mapping, pos, endbyte);
1994 	if (err)
1995 		goto out;
1996 	written += written_buffered;
1997 	iocb->ki_pos = pos + written_buffered;
1998 	invalidate_mapping_pages(file->f_mapping, pos >> PAGE_SHIFT,
1999 				 endbyte >> PAGE_SHIFT);
2000 out:
2001 	return err < 0 ? err : written;
2002 }
2003 
2004 static ssize_t btrfs_encoded_write(struct kiocb *iocb, struct iov_iter *from,
2005 			const struct btrfs_ioctl_encoded_io_args *encoded)
2006 {
2007 	struct file *file = iocb->ki_filp;
2008 	struct inode *inode = file_inode(file);
2009 	loff_t count;
2010 	ssize_t ret;
2011 
2012 	btrfs_inode_lock(inode, 0);
2013 	count = encoded->len;
2014 	ret = generic_write_checks_count(iocb, &count);
2015 	if (ret == 0 && count != encoded->len) {
2016 		/*
2017 		 * The write got truncated by generic_write_checks_count(). We
2018 		 * can't do a partial encoded write.
2019 		 */
2020 		ret = -EFBIG;
2021 	}
2022 	if (ret || encoded->len == 0)
2023 		goto out;
2024 
2025 	ret = btrfs_write_check(iocb, from, encoded->len);
2026 	if (ret < 0)
2027 		goto out;
2028 
2029 	ret = btrfs_do_encoded_write(iocb, from, encoded);
2030 out:
2031 	btrfs_inode_unlock(inode, 0);
2032 	return ret;
2033 }
2034 
2035 ssize_t btrfs_do_write_iter(struct kiocb *iocb, struct iov_iter *from,
2036 			    const struct btrfs_ioctl_encoded_io_args *encoded)
2037 {
2038 	struct file *file = iocb->ki_filp;
2039 	struct btrfs_inode *inode = BTRFS_I(file_inode(file));
2040 	ssize_t num_written, num_sync;
2041 	const bool sync = iocb->ki_flags & IOCB_DSYNC;
2042 
2043 	/*
2044 	 * If the fs flips readonly due to some impossible error, although we
2045 	 * have opened a file as writable, we have to stop this write operation
2046 	 * to ensure consistency.
2047 	 */
2048 	if (BTRFS_FS_ERROR(inode->root->fs_info))
2049 		return -EROFS;
2050 
2051 	if ((iocb->ki_flags & IOCB_NOWAIT) && !(iocb->ki_flags & IOCB_DIRECT))
2052 		return -EOPNOTSUPP;
2053 
2054 	if (sync)
2055 		atomic_inc(&inode->sync_writers);
2056 
2057 	if (encoded) {
2058 		num_written = btrfs_encoded_write(iocb, from, encoded);
2059 		num_sync = encoded->len;
2060 	} else if (iocb->ki_flags & IOCB_DIRECT) {
2061 		num_written = num_sync = btrfs_direct_write(iocb, from);
2062 	} else {
2063 		num_written = num_sync = btrfs_buffered_write(iocb, from);
2064 	}
2065 
2066 	btrfs_set_inode_last_sub_trans(inode);
2067 
2068 	if (num_sync > 0) {
2069 		num_sync = generic_write_sync(iocb, num_sync);
2070 		if (num_sync < 0)
2071 			num_written = num_sync;
2072 	}
2073 
2074 	if (sync)
2075 		atomic_dec(&inode->sync_writers);
2076 
2077 	current->backing_dev_info = NULL;
2078 	return num_written;
2079 }
2080 
2081 static ssize_t btrfs_file_write_iter(struct kiocb *iocb, struct iov_iter *from)
2082 {
2083 	return btrfs_do_write_iter(iocb, from, NULL);
2084 }
2085 
2086 int btrfs_release_file(struct inode *inode, struct file *filp)
2087 {
2088 	struct btrfs_file_private *private = filp->private_data;
2089 
2090 	if (private && private->filldir_buf)
2091 		kfree(private->filldir_buf);
2092 	kfree(private);
2093 	filp->private_data = NULL;
2094 
2095 	/*
2096 	 * Set by setattr when we are about to truncate a file from a non-zero
2097 	 * size to a zero size.  This tries to flush down new bytes that may
2098 	 * have been written if the application were using truncate to replace
2099 	 * a file in place.
2100 	 */
2101 	if (test_and_clear_bit(BTRFS_INODE_FLUSH_ON_CLOSE,
2102 			       &BTRFS_I(inode)->runtime_flags))
2103 			filemap_flush(inode->i_mapping);
2104 	return 0;
2105 }
2106 
2107 static int start_ordered_ops(struct inode *inode, loff_t start, loff_t end)
2108 {
2109 	int ret;
2110 	struct blk_plug plug;
2111 
2112 	/*
2113 	 * This is only called in fsync, which would do synchronous writes, so
2114 	 * a plug can merge adjacent IOs as much as possible.  Esp. in case of
2115 	 * multiple disks using raid profile, a large IO can be split to
2116 	 * several segments of stripe length (currently 64K).
2117 	 */
2118 	blk_start_plug(&plug);
2119 	atomic_inc(&BTRFS_I(inode)->sync_writers);
2120 	ret = btrfs_fdatawrite_range(inode, start, end);
2121 	atomic_dec(&BTRFS_I(inode)->sync_writers);
2122 	blk_finish_plug(&plug);
2123 
2124 	return ret;
2125 }
2126 
2127 static inline bool skip_inode_logging(const struct btrfs_log_ctx *ctx)
2128 {
2129 	struct btrfs_inode *inode = BTRFS_I(ctx->inode);
2130 	struct btrfs_fs_info *fs_info = inode->root->fs_info;
2131 
2132 	if (btrfs_inode_in_log(inode, fs_info->generation) &&
2133 	    list_empty(&ctx->ordered_extents))
2134 		return true;
2135 
2136 	/*
2137 	 * If we are doing a fast fsync we can not bail out if the inode's
2138 	 * last_trans is <= then the last committed transaction, because we only
2139 	 * update the last_trans of the inode during ordered extent completion,
2140 	 * and for a fast fsync we don't wait for that, we only wait for the
2141 	 * writeback to complete.
2142 	 */
2143 	if (inode->last_trans <= fs_info->last_trans_committed &&
2144 	    (test_bit(BTRFS_INODE_NEEDS_FULL_SYNC, &inode->runtime_flags) ||
2145 	     list_empty(&ctx->ordered_extents)))
2146 		return true;
2147 
2148 	return false;
2149 }
2150 
2151 /*
2152  * fsync call for both files and directories.  This logs the inode into
2153  * the tree log instead of forcing full commits whenever possible.
2154  *
2155  * It needs to call filemap_fdatawait so that all ordered extent updates are
2156  * in the metadata btree are up to date for copying to the log.
2157  *
2158  * It drops the inode mutex before doing the tree log commit.  This is an
2159  * important optimization for directories because holding the mutex prevents
2160  * new operations on the dir while we write to disk.
2161  */
2162 int btrfs_sync_file(struct file *file, loff_t start, loff_t end, int datasync)
2163 {
2164 	struct dentry *dentry = file_dentry(file);
2165 	struct inode *inode = d_inode(dentry);
2166 	struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
2167 	struct btrfs_root *root = BTRFS_I(inode)->root;
2168 	struct btrfs_trans_handle *trans;
2169 	struct btrfs_log_ctx ctx;
2170 	int ret = 0, err;
2171 	u64 len;
2172 	bool full_sync;
2173 
2174 	trace_btrfs_sync_file(file, datasync);
2175 
2176 	btrfs_init_log_ctx(&ctx, inode);
2177 
2178 	/*
2179 	 * Always set the range to a full range, otherwise we can get into
2180 	 * several problems, from missing file extent items to represent holes
2181 	 * when not using the NO_HOLES feature, to log tree corruption due to
2182 	 * races between hole detection during logging and completion of ordered
2183 	 * extents outside the range, to missing checksums due to ordered extents
2184 	 * for which we flushed only a subset of their pages.
2185 	 */
2186 	start = 0;
2187 	end = LLONG_MAX;
2188 	len = (u64)LLONG_MAX + 1;
2189 
2190 	/*
2191 	 * We write the dirty pages in the range and wait until they complete
2192 	 * out of the ->i_mutex. If so, we can flush the dirty pages by
2193 	 * multi-task, and make the performance up.  See
2194 	 * btrfs_wait_ordered_range for an explanation of the ASYNC check.
2195 	 */
2196 	ret = start_ordered_ops(inode, start, end);
2197 	if (ret)
2198 		goto out;
2199 
2200 	btrfs_inode_lock(inode, BTRFS_ILOCK_MMAP);
2201 
2202 	atomic_inc(&root->log_batch);
2203 
2204 	/*
2205 	 * Always check for the full sync flag while holding the inode's lock,
2206 	 * to avoid races with other tasks. The flag must be either set all the
2207 	 * time during logging or always off all the time while logging.
2208 	 */
2209 	full_sync = test_bit(BTRFS_INODE_NEEDS_FULL_SYNC,
2210 			     &BTRFS_I(inode)->runtime_flags);
2211 
2212 	/*
2213 	 * Before we acquired the inode's lock and the mmap lock, someone may
2214 	 * have dirtied more pages in the target range. We need to make sure
2215 	 * that writeback for any such pages does not start while we are logging
2216 	 * the inode, because if it does, any of the following might happen when
2217 	 * we are not doing a full inode sync:
2218 	 *
2219 	 * 1) We log an extent after its writeback finishes but before its
2220 	 *    checksums are added to the csum tree, leading to -EIO errors
2221 	 *    when attempting to read the extent after a log replay.
2222 	 *
2223 	 * 2) We can end up logging an extent before its writeback finishes.
2224 	 *    Therefore after the log replay we will have a file extent item
2225 	 *    pointing to an unwritten extent (and no data checksums as well).
2226 	 *
2227 	 * So trigger writeback for any eventual new dirty pages and then we
2228 	 * wait for all ordered extents to complete below.
2229 	 */
2230 	ret = start_ordered_ops(inode, start, end);
2231 	if (ret) {
2232 		btrfs_inode_unlock(inode, BTRFS_ILOCK_MMAP);
2233 		goto out;
2234 	}
2235 
2236 	/*
2237 	 * We have to do this here to avoid the priority inversion of waiting on
2238 	 * IO of a lower priority task while holding a transaction open.
2239 	 *
2240 	 * For a full fsync we wait for the ordered extents to complete while
2241 	 * for a fast fsync we wait just for writeback to complete, and then
2242 	 * attach the ordered extents to the transaction so that a transaction
2243 	 * commit waits for their completion, to avoid data loss if we fsync,
2244 	 * the current transaction commits before the ordered extents complete
2245 	 * and a power failure happens right after that.
2246 	 *
2247 	 * For zoned filesystem, if a write IO uses a ZONE_APPEND command, the
2248 	 * logical address recorded in the ordered extent may change. We need
2249 	 * to wait for the IO to stabilize the logical address.
2250 	 */
2251 	if (full_sync || btrfs_is_zoned(fs_info)) {
2252 		ret = btrfs_wait_ordered_range(inode, start, len);
2253 	} else {
2254 		/*
2255 		 * Get our ordered extents as soon as possible to avoid doing
2256 		 * checksum lookups in the csum tree, and use instead the
2257 		 * checksums attached to the ordered extents.
2258 		 */
2259 		btrfs_get_ordered_extents_for_logging(BTRFS_I(inode),
2260 						      &ctx.ordered_extents);
2261 		ret = filemap_fdatawait_range(inode->i_mapping, start, end);
2262 	}
2263 
2264 	if (ret)
2265 		goto out_release_extents;
2266 
2267 	atomic_inc(&root->log_batch);
2268 
2269 	smp_mb();
2270 	if (skip_inode_logging(&ctx)) {
2271 		/*
2272 		 * We've had everything committed since the last time we were
2273 		 * modified so clear this flag in case it was set for whatever
2274 		 * reason, it's no longer relevant.
2275 		 */
2276 		clear_bit(BTRFS_INODE_NEEDS_FULL_SYNC,
2277 			  &BTRFS_I(inode)->runtime_flags);
2278 		/*
2279 		 * An ordered extent might have started before and completed
2280 		 * already with io errors, in which case the inode was not
2281 		 * updated and we end up here. So check the inode's mapping
2282 		 * for any errors that might have happened since we last
2283 		 * checked called fsync.
2284 		 */
2285 		ret = filemap_check_wb_err(inode->i_mapping, file->f_wb_err);
2286 		goto out_release_extents;
2287 	}
2288 
2289 	/*
2290 	 * We use start here because we will need to wait on the IO to complete
2291 	 * in btrfs_sync_log, which could require joining a transaction (for
2292 	 * example checking cross references in the nocow path).  If we use join
2293 	 * here we could get into a situation where we're waiting on IO to
2294 	 * happen that is blocked on a transaction trying to commit.  With start
2295 	 * we inc the extwriter counter, so we wait for all extwriters to exit
2296 	 * before we start blocking joiners.  This comment is to keep somebody
2297 	 * from thinking they are super smart and changing this to
2298 	 * btrfs_join_transaction *cough*Josef*cough*.
2299 	 */
2300 	trans = btrfs_start_transaction(root, 0);
2301 	if (IS_ERR(trans)) {
2302 		ret = PTR_ERR(trans);
2303 		goto out_release_extents;
2304 	}
2305 	trans->in_fsync = true;
2306 
2307 	ret = btrfs_log_dentry_safe(trans, dentry, &ctx);
2308 	btrfs_release_log_ctx_extents(&ctx);
2309 	if (ret < 0) {
2310 		/* Fallthrough and commit/free transaction. */
2311 		ret = 1;
2312 	}
2313 
2314 	/* we've logged all the items and now have a consistent
2315 	 * version of the file in the log.  It is possible that
2316 	 * someone will come in and modify the file, but that's
2317 	 * fine because the log is consistent on disk, and we
2318 	 * have references to all of the file's extents
2319 	 *
2320 	 * It is possible that someone will come in and log the
2321 	 * file again, but that will end up using the synchronization
2322 	 * inside btrfs_sync_log to keep things safe.
2323 	 */
2324 	btrfs_inode_unlock(inode, BTRFS_ILOCK_MMAP);
2325 
2326 	if (ret != BTRFS_NO_LOG_SYNC) {
2327 		if (!ret) {
2328 			ret = btrfs_sync_log(trans, root, &ctx);
2329 			if (!ret) {
2330 				ret = btrfs_end_transaction(trans);
2331 				goto out;
2332 			}
2333 		}
2334 		if (!full_sync) {
2335 			ret = btrfs_wait_ordered_range(inode, start, len);
2336 			if (ret) {
2337 				btrfs_end_transaction(trans);
2338 				goto out;
2339 			}
2340 		}
2341 		ret = btrfs_commit_transaction(trans);
2342 	} else {
2343 		ret = btrfs_end_transaction(trans);
2344 	}
2345 out:
2346 	ASSERT(list_empty(&ctx.list));
2347 	err = file_check_and_advance_wb_err(file);
2348 	if (!ret)
2349 		ret = err;
2350 	return ret > 0 ? -EIO : ret;
2351 
2352 out_release_extents:
2353 	btrfs_release_log_ctx_extents(&ctx);
2354 	btrfs_inode_unlock(inode, BTRFS_ILOCK_MMAP);
2355 	goto out;
2356 }
2357 
2358 static const struct vm_operations_struct btrfs_file_vm_ops = {
2359 	.fault		= filemap_fault,
2360 	.map_pages	= filemap_map_pages,
2361 	.page_mkwrite	= btrfs_page_mkwrite,
2362 };
2363 
2364 static int btrfs_file_mmap(struct file	*filp, struct vm_area_struct *vma)
2365 {
2366 	struct address_space *mapping = filp->f_mapping;
2367 
2368 	if (!mapping->a_ops->read_folio)
2369 		return -ENOEXEC;
2370 
2371 	file_accessed(filp);
2372 	vma->vm_ops = &btrfs_file_vm_ops;
2373 
2374 	return 0;
2375 }
2376 
2377 static int hole_mergeable(struct btrfs_inode *inode, struct extent_buffer *leaf,
2378 			  int slot, u64 start, u64 end)
2379 {
2380 	struct btrfs_file_extent_item *fi;
2381 	struct btrfs_key key;
2382 
2383 	if (slot < 0 || slot >= btrfs_header_nritems(leaf))
2384 		return 0;
2385 
2386 	btrfs_item_key_to_cpu(leaf, &key, slot);
2387 	if (key.objectid != btrfs_ino(inode) ||
2388 	    key.type != BTRFS_EXTENT_DATA_KEY)
2389 		return 0;
2390 
2391 	fi = btrfs_item_ptr(leaf, slot, struct btrfs_file_extent_item);
2392 
2393 	if (btrfs_file_extent_type(leaf, fi) != BTRFS_FILE_EXTENT_REG)
2394 		return 0;
2395 
2396 	if (btrfs_file_extent_disk_bytenr(leaf, fi))
2397 		return 0;
2398 
2399 	if (key.offset == end)
2400 		return 1;
2401 	if (key.offset + btrfs_file_extent_num_bytes(leaf, fi) == start)
2402 		return 1;
2403 	return 0;
2404 }
2405 
2406 static int fill_holes(struct btrfs_trans_handle *trans,
2407 		struct btrfs_inode *inode,
2408 		struct btrfs_path *path, u64 offset, u64 end)
2409 {
2410 	struct btrfs_fs_info *fs_info = trans->fs_info;
2411 	struct btrfs_root *root = inode->root;
2412 	struct extent_buffer *leaf;
2413 	struct btrfs_file_extent_item *fi;
2414 	struct extent_map *hole_em;
2415 	struct extent_map_tree *em_tree = &inode->extent_tree;
2416 	struct btrfs_key key;
2417 	int ret;
2418 
2419 	if (btrfs_fs_incompat(fs_info, NO_HOLES))
2420 		goto out;
2421 
2422 	key.objectid = btrfs_ino(inode);
2423 	key.type = BTRFS_EXTENT_DATA_KEY;
2424 	key.offset = offset;
2425 
2426 	ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
2427 	if (ret <= 0) {
2428 		/*
2429 		 * We should have dropped this offset, so if we find it then
2430 		 * something has gone horribly wrong.
2431 		 */
2432 		if (ret == 0)
2433 			ret = -EINVAL;
2434 		return ret;
2435 	}
2436 
2437 	leaf = path->nodes[0];
2438 	if (hole_mergeable(inode, leaf, path->slots[0] - 1, offset, end)) {
2439 		u64 num_bytes;
2440 
2441 		path->slots[0]--;
2442 		fi = btrfs_item_ptr(leaf, path->slots[0],
2443 				    struct btrfs_file_extent_item);
2444 		num_bytes = btrfs_file_extent_num_bytes(leaf, fi) +
2445 			end - offset;
2446 		btrfs_set_file_extent_num_bytes(leaf, fi, num_bytes);
2447 		btrfs_set_file_extent_ram_bytes(leaf, fi, num_bytes);
2448 		btrfs_set_file_extent_offset(leaf, fi, 0);
2449 		btrfs_mark_buffer_dirty(leaf);
2450 		goto out;
2451 	}
2452 
2453 	if (hole_mergeable(inode, leaf, path->slots[0], offset, end)) {
2454 		u64 num_bytes;
2455 
2456 		key.offset = offset;
2457 		btrfs_set_item_key_safe(fs_info, path, &key);
2458 		fi = btrfs_item_ptr(leaf, path->slots[0],
2459 				    struct btrfs_file_extent_item);
2460 		num_bytes = btrfs_file_extent_num_bytes(leaf, fi) + end -
2461 			offset;
2462 		btrfs_set_file_extent_num_bytes(leaf, fi, num_bytes);
2463 		btrfs_set_file_extent_ram_bytes(leaf, fi, num_bytes);
2464 		btrfs_set_file_extent_offset(leaf, fi, 0);
2465 		btrfs_mark_buffer_dirty(leaf);
2466 		goto out;
2467 	}
2468 	btrfs_release_path(path);
2469 
2470 	ret = btrfs_insert_file_extent(trans, root, btrfs_ino(inode),
2471 			offset, 0, 0, end - offset, 0, end - offset, 0, 0, 0);
2472 	if (ret)
2473 		return ret;
2474 
2475 out:
2476 	btrfs_release_path(path);
2477 
2478 	hole_em = alloc_extent_map();
2479 	if (!hole_em) {
2480 		btrfs_drop_extent_cache(inode, offset, end - 1, 0);
2481 		btrfs_set_inode_full_sync(inode);
2482 	} else {
2483 		hole_em->start = offset;
2484 		hole_em->len = end - offset;
2485 		hole_em->ram_bytes = hole_em->len;
2486 		hole_em->orig_start = offset;
2487 
2488 		hole_em->block_start = EXTENT_MAP_HOLE;
2489 		hole_em->block_len = 0;
2490 		hole_em->orig_block_len = 0;
2491 		hole_em->compress_type = BTRFS_COMPRESS_NONE;
2492 		hole_em->generation = trans->transid;
2493 
2494 		do {
2495 			btrfs_drop_extent_cache(inode, offset, end - 1, 0);
2496 			write_lock(&em_tree->lock);
2497 			ret = add_extent_mapping(em_tree, hole_em, 1);
2498 			write_unlock(&em_tree->lock);
2499 		} while (ret == -EEXIST);
2500 		free_extent_map(hole_em);
2501 		if (ret)
2502 			btrfs_set_inode_full_sync(inode);
2503 	}
2504 
2505 	return 0;
2506 }
2507 
2508 /*
2509  * Find a hole extent on given inode and change start/len to the end of hole
2510  * extent.(hole/vacuum extent whose em->start <= start &&
2511  *	   em->start + em->len > start)
2512  * When a hole extent is found, return 1 and modify start/len.
2513  */
2514 static int find_first_non_hole(struct btrfs_inode *inode, u64 *start, u64 *len)
2515 {
2516 	struct btrfs_fs_info *fs_info = inode->root->fs_info;
2517 	struct extent_map *em;
2518 	int ret = 0;
2519 
2520 	em = btrfs_get_extent(inode, NULL, 0,
2521 			      round_down(*start, fs_info->sectorsize),
2522 			      round_up(*len, fs_info->sectorsize));
2523 	if (IS_ERR(em))
2524 		return PTR_ERR(em);
2525 
2526 	/* Hole or vacuum extent(only exists in no-hole mode) */
2527 	if (em->block_start == EXTENT_MAP_HOLE) {
2528 		ret = 1;
2529 		*len = em->start + em->len > *start + *len ?
2530 		       0 : *start + *len - em->start - em->len;
2531 		*start = em->start + em->len;
2532 	}
2533 	free_extent_map(em);
2534 	return ret;
2535 }
2536 
2537 static void btrfs_punch_hole_lock_range(struct inode *inode,
2538 					const u64 lockstart,
2539 					const u64 lockend,
2540 					struct extent_state **cached_state)
2541 {
2542 	/*
2543 	 * For subpage case, if the range is not at page boundary, we could
2544 	 * have pages at the leading/tailing part of the range.
2545 	 * This could lead to dead loop since filemap_range_has_page()
2546 	 * will always return true.
2547 	 * So here we need to do extra page alignment for
2548 	 * filemap_range_has_page().
2549 	 */
2550 	const u64 page_lockstart = round_up(lockstart, PAGE_SIZE);
2551 	const u64 page_lockend = round_down(lockend + 1, PAGE_SIZE) - 1;
2552 
2553 	while (1) {
2554 		truncate_pagecache_range(inode, lockstart, lockend);
2555 
2556 		lock_extent_bits(&BTRFS_I(inode)->io_tree, lockstart, lockend,
2557 				 cached_state);
2558 		/*
2559 		 * We can't have ordered extents in the range, nor dirty/writeback
2560 		 * pages, because we have locked the inode's VFS lock in exclusive
2561 		 * mode, we have locked the inode's i_mmap_lock in exclusive mode,
2562 		 * we have flushed all delalloc in the range and we have waited
2563 		 * for any ordered extents in the range to complete.
2564 		 * We can race with anyone reading pages from this range, so after
2565 		 * locking the range check if we have pages in the range, and if
2566 		 * we do, unlock the range and retry.
2567 		 */
2568 		if (!filemap_range_has_page(inode->i_mapping, page_lockstart,
2569 					    page_lockend))
2570 			break;
2571 
2572 		unlock_extent_cached(&BTRFS_I(inode)->io_tree, lockstart,
2573 				     lockend, cached_state);
2574 	}
2575 
2576 	btrfs_assert_inode_range_clean(BTRFS_I(inode), lockstart, lockend);
2577 }
2578 
2579 static int btrfs_insert_replace_extent(struct btrfs_trans_handle *trans,
2580 				     struct btrfs_inode *inode,
2581 				     struct btrfs_path *path,
2582 				     struct btrfs_replace_extent_info *extent_info,
2583 				     const u64 replace_len,
2584 				     const u64 bytes_to_drop)
2585 {
2586 	struct btrfs_fs_info *fs_info = trans->fs_info;
2587 	struct btrfs_root *root = inode->root;
2588 	struct btrfs_file_extent_item *extent;
2589 	struct extent_buffer *leaf;
2590 	struct btrfs_key key;
2591 	int slot;
2592 	struct btrfs_ref ref = { 0 };
2593 	int ret;
2594 
2595 	if (replace_len == 0)
2596 		return 0;
2597 
2598 	if (extent_info->disk_offset == 0 &&
2599 	    btrfs_fs_incompat(fs_info, NO_HOLES)) {
2600 		btrfs_update_inode_bytes(inode, 0, bytes_to_drop);
2601 		return 0;
2602 	}
2603 
2604 	key.objectid = btrfs_ino(inode);
2605 	key.type = BTRFS_EXTENT_DATA_KEY;
2606 	key.offset = extent_info->file_offset;
2607 	ret = btrfs_insert_empty_item(trans, root, path, &key,
2608 				      sizeof(struct btrfs_file_extent_item));
2609 	if (ret)
2610 		return ret;
2611 	leaf = path->nodes[0];
2612 	slot = path->slots[0];
2613 	write_extent_buffer(leaf, extent_info->extent_buf,
2614 			    btrfs_item_ptr_offset(leaf, slot),
2615 			    sizeof(struct btrfs_file_extent_item));
2616 	extent = btrfs_item_ptr(leaf, slot, struct btrfs_file_extent_item);
2617 	ASSERT(btrfs_file_extent_type(leaf, extent) != BTRFS_FILE_EXTENT_INLINE);
2618 	btrfs_set_file_extent_offset(leaf, extent, extent_info->data_offset);
2619 	btrfs_set_file_extent_num_bytes(leaf, extent, replace_len);
2620 	if (extent_info->is_new_extent)
2621 		btrfs_set_file_extent_generation(leaf, extent, trans->transid);
2622 	btrfs_mark_buffer_dirty(leaf);
2623 	btrfs_release_path(path);
2624 
2625 	ret = btrfs_inode_set_file_extent_range(inode, extent_info->file_offset,
2626 						replace_len);
2627 	if (ret)
2628 		return ret;
2629 
2630 	/* If it's a hole, nothing more needs to be done. */
2631 	if (extent_info->disk_offset == 0) {
2632 		btrfs_update_inode_bytes(inode, 0, bytes_to_drop);
2633 		return 0;
2634 	}
2635 
2636 	btrfs_update_inode_bytes(inode, replace_len, bytes_to_drop);
2637 
2638 	if (extent_info->is_new_extent && extent_info->insertions == 0) {
2639 		key.objectid = extent_info->disk_offset;
2640 		key.type = BTRFS_EXTENT_ITEM_KEY;
2641 		key.offset = extent_info->disk_len;
2642 		ret = btrfs_alloc_reserved_file_extent(trans, root,
2643 						       btrfs_ino(inode),
2644 						       extent_info->file_offset,
2645 						       extent_info->qgroup_reserved,
2646 						       &key);
2647 	} else {
2648 		u64 ref_offset;
2649 
2650 		btrfs_init_generic_ref(&ref, BTRFS_ADD_DELAYED_REF,
2651 				       extent_info->disk_offset,
2652 				       extent_info->disk_len, 0);
2653 		ref_offset = extent_info->file_offset - extent_info->data_offset;
2654 		btrfs_init_data_ref(&ref, root->root_key.objectid,
2655 				    btrfs_ino(inode), ref_offset, 0, false);
2656 		ret = btrfs_inc_extent_ref(trans, &ref);
2657 	}
2658 
2659 	extent_info->insertions++;
2660 
2661 	return ret;
2662 }
2663 
2664 /*
2665  * The respective range must have been previously locked, as well as the inode.
2666  * The end offset is inclusive (last byte of the range).
2667  * @extent_info is NULL for fallocate's hole punching and non-NULL when replacing
2668  * the file range with an extent.
2669  * When not punching a hole, we don't want to end up in a state where we dropped
2670  * extents without inserting a new one, so we must abort the transaction to avoid
2671  * a corruption.
2672  */
2673 int btrfs_replace_file_extents(struct btrfs_inode *inode,
2674 			       struct btrfs_path *path, const u64 start,
2675 			       const u64 end,
2676 			       struct btrfs_replace_extent_info *extent_info,
2677 			       struct btrfs_trans_handle **trans_out)
2678 {
2679 	struct btrfs_drop_extents_args drop_args = { 0 };
2680 	struct btrfs_root *root = inode->root;
2681 	struct btrfs_fs_info *fs_info = root->fs_info;
2682 	u64 min_size = btrfs_calc_insert_metadata_size(fs_info, 1);
2683 	u64 ino_size = round_up(inode->vfs_inode.i_size, fs_info->sectorsize);
2684 	struct btrfs_trans_handle *trans = NULL;
2685 	struct btrfs_block_rsv *rsv;
2686 	unsigned int rsv_count;
2687 	u64 cur_offset;
2688 	u64 len = end - start;
2689 	int ret = 0;
2690 
2691 	if (end <= start)
2692 		return -EINVAL;
2693 
2694 	rsv = btrfs_alloc_block_rsv(fs_info, BTRFS_BLOCK_RSV_TEMP);
2695 	if (!rsv) {
2696 		ret = -ENOMEM;
2697 		goto out;
2698 	}
2699 	rsv->size = btrfs_calc_insert_metadata_size(fs_info, 1);
2700 	rsv->failfast = 1;
2701 
2702 	/*
2703 	 * 1 - update the inode
2704 	 * 1 - removing the extents in the range
2705 	 * 1 - adding the hole extent if no_holes isn't set or if we are
2706 	 *     replacing the range with a new extent
2707 	 */
2708 	if (!btrfs_fs_incompat(fs_info, NO_HOLES) || extent_info)
2709 		rsv_count = 3;
2710 	else
2711 		rsv_count = 2;
2712 
2713 	trans = btrfs_start_transaction(root, rsv_count);
2714 	if (IS_ERR(trans)) {
2715 		ret = PTR_ERR(trans);
2716 		trans = NULL;
2717 		goto out_free;
2718 	}
2719 
2720 	ret = btrfs_block_rsv_migrate(&fs_info->trans_block_rsv, rsv,
2721 				      min_size, false);
2722 	BUG_ON(ret);
2723 	trans->block_rsv = rsv;
2724 
2725 	cur_offset = start;
2726 	drop_args.path = path;
2727 	drop_args.end = end + 1;
2728 	drop_args.drop_cache = true;
2729 	while (cur_offset < end) {
2730 		drop_args.start = cur_offset;
2731 		ret = btrfs_drop_extents(trans, root, inode, &drop_args);
2732 		/* If we are punching a hole decrement the inode's byte count */
2733 		if (!extent_info)
2734 			btrfs_update_inode_bytes(inode, 0,
2735 						 drop_args.bytes_found);
2736 		if (ret != -ENOSPC) {
2737 			/*
2738 			 * The only time we don't want to abort is if we are
2739 			 * attempting to clone a partial inline extent, in which
2740 			 * case we'll get EOPNOTSUPP.  However if we aren't
2741 			 * clone we need to abort no matter what, because if we
2742 			 * got EOPNOTSUPP via prealloc then we messed up and
2743 			 * need to abort.
2744 			 */
2745 			if (ret &&
2746 			    (ret != -EOPNOTSUPP ||
2747 			     (extent_info && extent_info->is_new_extent)))
2748 				btrfs_abort_transaction(trans, ret);
2749 			break;
2750 		}
2751 
2752 		trans->block_rsv = &fs_info->trans_block_rsv;
2753 
2754 		if (!extent_info && cur_offset < drop_args.drop_end &&
2755 		    cur_offset < ino_size) {
2756 			ret = fill_holes(trans, inode, path, cur_offset,
2757 					 drop_args.drop_end);
2758 			if (ret) {
2759 				/*
2760 				 * If we failed then we didn't insert our hole
2761 				 * entries for the area we dropped, so now the
2762 				 * fs is corrupted, so we must abort the
2763 				 * transaction.
2764 				 */
2765 				btrfs_abort_transaction(trans, ret);
2766 				break;
2767 			}
2768 		} else if (!extent_info && cur_offset < drop_args.drop_end) {
2769 			/*
2770 			 * We are past the i_size here, but since we didn't
2771 			 * insert holes we need to clear the mapped area so we
2772 			 * know to not set disk_i_size in this area until a new
2773 			 * file extent is inserted here.
2774 			 */
2775 			ret = btrfs_inode_clear_file_extent_range(inode,
2776 					cur_offset,
2777 					drop_args.drop_end - cur_offset);
2778 			if (ret) {
2779 				/*
2780 				 * We couldn't clear our area, so we could
2781 				 * presumably adjust up and corrupt the fs, so
2782 				 * we need to abort.
2783 				 */
2784 				btrfs_abort_transaction(trans, ret);
2785 				break;
2786 			}
2787 		}
2788 
2789 		if (extent_info &&
2790 		    drop_args.drop_end > extent_info->file_offset) {
2791 			u64 replace_len = drop_args.drop_end -
2792 					  extent_info->file_offset;
2793 
2794 			ret = btrfs_insert_replace_extent(trans, inode,	path,
2795 					extent_info, replace_len,
2796 					drop_args.bytes_found);
2797 			if (ret) {
2798 				btrfs_abort_transaction(trans, ret);
2799 				break;
2800 			}
2801 			extent_info->data_len -= replace_len;
2802 			extent_info->data_offset += replace_len;
2803 			extent_info->file_offset += replace_len;
2804 		}
2805 
2806 		ret = btrfs_update_inode(trans, root, inode);
2807 		if (ret)
2808 			break;
2809 
2810 		btrfs_end_transaction(trans);
2811 		btrfs_btree_balance_dirty(fs_info);
2812 
2813 		trans = btrfs_start_transaction(root, rsv_count);
2814 		if (IS_ERR(trans)) {
2815 			ret = PTR_ERR(trans);
2816 			trans = NULL;
2817 			break;
2818 		}
2819 
2820 		ret = btrfs_block_rsv_migrate(&fs_info->trans_block_rsv,
2821 					      rsv, min_size, false);
2822 		BUG_ON(ret);	/* shouldn't happen */
2823 		trans->block_rsv = rsv;
2824 
2825 		cur_offset = drop_args.drop_end;
2826 		len = end - cur_offset;
2827 		if (!extent_info && len) {
2828 			ret = find_first_non_hole(inode, &cur_offset, &len);
2829 			if (unlikely(ret < 0))
2830 				break;
2831 			if (ret && !len) {
2832 				ret = 0;
2833 				break;
2834 			}
2835 		}
2836 	}
2837 
2838 	/*
2839 	 * If we were cloning, force the next fsync to be a full one since we
2840 	 * we replaced (or just dropped in the case of cloning holes when
2841 	 * NO_HOLES is enabled) file extent items and did not setup new extent
2842 	 * maps for the replacement extents (or holes).
2843 	 */
2844 	if (extent_info && !extent_info->is_new_extent)
2845 		btrfs_set_inode_full_sync(inode);
2846 
2847 	if (ret)
2848 		goto out_trans;
2849 
2850 	trans->block_rsv = &fs_info->trans_block_rsv;
2851 	/*
2852 	 * If we are using the NO_HOLES feature we might have had already an
2853 	 * hole that overlaps a part of the region [lockstart, lockend] and
2854 	 * ends at (or beyond) lockend. Since we have no file extent items to
2855 	 * represent holes, drop_end can be less than lockend and so we must
2856 	 * make sure we have an extent map representing the existing hole (the
2857 	 * call to __btrfs_drop_extents() might have dropped the existing extent
2858 	 * map representing the existing hole), otherwise the fast fsync path
2859 	 * will not record the existence of the hole region
2860 	 * [existing_hole_start, lockend].
2861 	 */
2862 	if (drop_args.drop_end <= end)
2863 		drop_args.drop_end = end + 1;
2864 	/*
2865 	 * Don't insert file hole extent item if it's for a range beyond eof
2866 	 * (because it's useless) or if it represents a 0 bytes range (when
2867 	 * cur_offset == drop_end).
2868 	 */
2869 	if (!extent_info && cur_offset < ino_size &&
2870 	    cur_offset < drop_args.drop_end) {
2871 		ret = fill_holes(trans, inode, path, cur_offset,
2872 				 drop_args.drop_end);
2873 		if (ret) {
2874 			/* Same comment as above. */
2875 			btrfs_abort_transaction(trans, ret);
2876 			goto out_trans;
2877 		}
2878 	} else if (!extent_info && cur_offset < drop_args.drop_end) {
2879 		/* See the comment in the loop above for the reasoning here. */
2880 		ret = btrfs_inode_clear_file_extent_range(inode, cur_offset,
2881 					drop_args.drop_end - cur_offset);
2882 		if (ret) {
2883 			btrfs_abort_transaction(trans, ret);
2884 			goto out_trans;
2885 		}
2886 
2887 	}
2888 	if (extent_info) {
2889 		ret = btrfs_insert_replace_extent(trans, inode, path,
2890 				extent_info, extent_info->data_len,
2891 				drop_args.bytes_found);
2892 		if (ret) {
2893 			btrfs_abort_transaction(trans, ret);
2894 			goto out_trans;
2895 		}
2896 	}
2897 
2898 out_trans:
2899 	if (!trans)
2900 		goto out_free;
2901 
2902 	trans->block_rsv = &fs_info->trans_block_rsv;
2903 	if (ret)
2904 		btrfs_end_transaction(trans);
2905 	else
2906 		*trans_out = trans;
2907 out_free:
2908 	btrfs_free_block_rsv(fs_info, rsv);
2909 out:
2910 	return ret;
2911 }
2912 
2913 static int btrfs_punch_hole(struct file *file, loff_t offset, loff_t len)
2914 {
2915 	struct inode *inode = file_inode(file);
2916 	struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
2917 	struct btrfs_root *root = BTRFS_I(inode)->root;
2918 	struct extent_state *cached_state = NULL;
2919 	struct btrfs_path *path;
2920 	struct btrfs_trans_handle *trans = NULL;
2921 	u64 lockstart;
2922 	u64 lockend;
2923 	u64 tail_start;
2924 	u64 tail_len;
2925 	u64 orig_start = offset;
2926 	int ret = 0;
2927 	bool same_block;
2928 	u64 ino_size;
2929 	bool truncated_block = false;
2930 	bool updated_inode = false;
2931 
2932 	btrfs_inode_lock(inode, BTRFS_ILOCK_MMAP);
2933 
2934 	ret = btrfs_wait_ordered_range(inode, offset, len);
2935 	if (ret)
2936 		goto out_only_mutex;
2937 
2938 	ino_size = round_up(inode->i_size, fs_info->sectorsize);
2939 	ret = find_first_non_hole(BTRFS_I(inode), &offset, &len);
2940 	if (ret < 0)
2941 		goto out_only_mutex;
2942 	if (ret && !len) {
2943 		/* Already in a large hole */
2944 		ret = 0;
2945 		goto out_only_mutex;
2946 	}
2947 
2948 	ret = file_modified(file);
2949 	if (ret)
2950 		goto out_only_mutex;
2951 
2952 	lockstart = round_up(offset, btrfs_inode_sectorsize(BTRFS_I(inode)));
2953 	lockend = round_down(offset + len,
2954 			     btrfs_inode_sectorsize(BTRFS_I(inode))) - 1;
2955 	same_block = (BTRFS_BYTES_TO_BLKS(fs_info, offset))
2956 		== (BTRFS_BYTES_TO_BLKS(fs_info, offset + len - 1));
2957 	/*
2958 	 * We needn't truncate any block which is beyond the end of the file
2959 	 * because we are sure there is no data there.
2960 	 */
2961 	/*
2962 	 * Only do this if we are in the same block and we aren't doing the
2963 	 * entire block.
2964 	 */
2965 	if (same_block && len < fs_info->sectorsize) {
2966 		if (offset < ino_size) {
2967 			truncated_block = true;
2968 			ret = btrfs_truncate_block(BTRFS_I(inode), offset, len,
2969 						   0);
2970 		} else {
2971 			ret = 0;
2972 		}
2973 		goto out_only_mutex;
2974 	}
2975 
2976 	/* zero back part of the first block */
2977 	if (offset < ino_size) {
2978 		truncated_block = true;
2979 		ret = btrfs_truncate_block(BTRFS_I(inode), offset, 0, 0);
2980 		if (ret) {
2981 			btrfs_inode_unlock(inode, BTRFS_ILOCK_MMAP);
2982 			return ret;
2983 		}
2984 	}
2985 
2986 	/* Check the aligned pages after the first unaligned page,
2987 	 * if offset != orig_start, which means the first unaligned page
2988 	 * including several following pages are already in holes,
2989 	 * the extra check can be skipped */
2990 	if (offset == orig_start) {
2991 		/* after truncate page, check hole again */
2992 		len = offset + len - lockstart;
2993 		offset = lockstart;
2994 		ret = find_first_non_hole(BTRFS_I(inode), &offset, &len);
2995 		if (ret < 0)
2996 			goto out_only_mutex;
2997 		if (ret && !len) {
2998 			ret = 0;
2999 			goto out_only_mutex;
3000 		}
3001 		lockstart = offset;
3002 	}
3003 
3004 	/* Check the tail unaligned part is in a hole */
3005 	tail_start = lockend + 1;
3006 	tail_len = offset + len - tail_start;
3007 	if (tail_len) {
3008 		ret = find_first_non_hole(BTRFS_I(inode), &tail_start, &tail_len);
3009 		if (unlikely(ret < 0))
3010 			goto out_only_mutex;
3011 		if (!ret) {
3012 			/* zero the front end of the last page */
3013 			if (tail_start + tail_len < ino_size) {
3014 				truncated_block = true;
3015 				ret = btrfs_truncate_block(BTRFS_I(inode),
3016 							tail_start + tail_len,
3017 							0, 1);
3018 				if (ret)
3019 					goto out_only_mutex;
3020 			}
3021 		}
3022 	}
3023 
3024 	if (lockend < lockstart) {
3025 		ret = 0;
3026 		goto out_only_mutex;
3027 	}
3028 
3029 	btrfs_punch_hole_lock_range(inode, lockstart, lockend, &cached_state);
3030 
3031 	path = btrfs_alloc_path();
3032 	if (!path) {
3033 		ret = -ENOMEM;
3034 		goto out;
3035 	}
3036 
3037 	ret = btrfs_replace_file_extents(BTRFS_I(inode), path, lockstart,
3038 					 lockend, NULL, &trans);
3039 	btrfs_free_path(path);
3040 	if (ret)
3041 		goto out;
3042 
3043 	ASSERT(trans != NULL);
3044 	inode_inc_iversion(inode);
3045 	inode->i_mtime = inode->i_ctime = current_time(inode);
3046 	ret = btrfs_update_inode(trans, root, BTRFS_I(inode));
3047 	updated_inode = true;
3048 	btrfs_end_transaction(trans);
3049 	btrfs_btree_balance_dirty(fs_info);
3050 out:
3051 	unlock_extent_cached(&BTRFS_I(inode)->io_tree, lockstart, lockend,
3052 			     &cached_state);
3053 out_only_mutex:
3054 	if (!updated_inode && truncated_block && !ret) {
3055 		/*
3056 		 * If we only end up zeroing part of a page, we still need to
3057 		 * update the inode item, so that all the time fields are
3058 		 * updated as well as the necessary btrfs inode in memory fields
3059 		 * for detecting, at fsync time, if the inode isn't yet in the
3060 		 * log tree or it's there but not up to date.
3061 		 */
3062 		struct timespec64 now = current_time(inode);
3063 
3064 		inode_inc_iversion(inode);
3065 		inode->i_mtime = now;
3066 		inode->i_ctime = now;
3067 		trans = btrfs_start_transaction(root, 1);
3068 		if (IS_ERR(trans)) {
3069 			ret = PTR_ERR(trans);
3070 		} else {
3071 			int ret2;
3072 
3073 			ret = btrfs_update_inode(trans, root, BTRFS_I(inode));
3074 			ret2 = btrfs_end_transaction(trans);
3075 			if (!ret)
3076 				ret = ret2;
3077 		}
3078 	}
3079 	btrfs_inode_unlock(inode, BTRFS_ILOCK_MMAP);
3080 	return ret;
3081 }
3082 
3083 /* Helper structure to record which range is already reserved */
3084 struct falloc_range {
3085 	struct list_head list;
3086 	u64 start;
3087 	u64 len;
3088 };
3089 
3090 /*
3091  * Helper function to add falloc range
3092  *
3093  * Caller should have locked the larger range of extent containing
3094  * [start, len)
3095  */
3096 static int add_falloc_range(struct list_head *head, u64 start, u64 len)
3097 {
3098 	struct falloc_range *range = NULL;
3099 
3100 	if (!list_empty(head)) {
3101 		/*
3102 		 * As fallocate iterates by bytenr order, we only need to check
3103 		 * the last range.
3104 		 */
3105 		range = list_last_entry(head, struct falloc_range, list);
3106 		if (range->start + range->len == start) {
3107 			range->len += len;
3108 			return 0;
3109 		}
3110 	}
3111 
3112 	range = kmalloc(sizeof(*range), GFP_KERNEL);
3113 	if (!range)
3114 		return -ENOMEM;
3115 	range->start = start;
3116 	range->len = len;
3117 	list_add_tail(&range->list, head);
3118 	return 0;
3119 }
3120 
3121 static int btrfs_fallocate_update_isize(struct inode *inode,
3122 					const u64 end,
3123 					const int mode)
3124 {
3125 	struct btrfs_trans_handle *trans;
3126 	struct btrfs_root *root = BTRFS_I(inode)->root;
3127 	int ret;
3128 	int ret2;
3129 
3130 	if (mode & FALLOC_FL_KEEP_SIZE || end <= i_size_read(inode))
3131 		return 0;
3132 
3133 	trans = btrfs_start_transaction(root, 1);
3134 	if (IS_ERR(trans))
3135 		return PTR_ERR(trans);
3136 
3137 	inode->i_ctime = current_time(inode);
3138 	i_size_write(inode, end);
3139 	btrfs_inode_safe_disk_i_size_write(BTRFS_I(inode), 0);
3140 	ret = btrfs_update_inode(trans, root, BTRFS_I(inode));
3141 	ret2 = btrfs_end_transaction(trans);
3142 
3143 	return ret ? ret : ret2;
3144 }
3145 
3146 enum {
3147 	RANGE_BOUNDARY_WRITTEN_EXTENT,
3148 	RANGE_BOUNDARY_PREALLOC_EXTENT,
3149 	RANGE_BOUNDARY_HOLE,
3150 };
3151 
3152 static int btrfs_zero_range_check_range_boundary(struct btrfs_inode *inode,
3153 						 u64 offset)
3154 {
3155 	const u64 sectorsize = btrfs_inode_sectorsize(inode);
3156 	struct extent_map *em;
3157 	int ret;
3158 
3159 	offset = round_down(offset, sectorsize);
3160 	em = btrfs_get_extent(inode, NULL, 0, offset, sectorsize);
3161 	if (IS_ERR(em))
3162 		return PTR_ERR(em);
3163 
3164 	if (em->block_start == EXTENT_MAP_HOLE)
3165 		ret = RANGE_BOUNDARY_HOLE;
3166 	else if (test_bit(EXTENT_FLAG_PREALLOC, &em->flags))
3167 		ret = RANGE_BOUNDARY_PREALLOC_EXTENT;
3168 	else
3169 		ret = RANGE_BOUNDARY_WRITTEN_EXTENT;
3170 
3171 	free_extent_map(em);
3172 	return ret;
3173 }
3174 
3175 static int btrfs_zero_range(struct inode *inode,
3176 			    loff_t offset,
3177 			    loff_t len,
3178 			    const int mode)
3179 {
3180 	struct btrfs_fs_info *fs_info = BTRFS_I(inode)->root->fs_info;
3181 	struct extent_map *em;
3182 	struct extent_changeset *data_reserved = NULL;
3183 	int ret;
3184 	u64 alloc_hint = 0;
3185 	const u64 sectorsize = btrfs_inode_sectorsize(BTRFS_I(inode));
3186 	u64 alloc_start = round_down(offset, sectorsize);
3187 	u64 alloc_end = round_up(offset + len, sectorsize);
3188 	u64 bytes_to_reserve = 0;
3189 	bool space_reserved = false;
3190 
3191 	em = btrfs_get_extent(BTRFS_I(inode), NULL, 0, alloc_start,
3192 			      alloc_end - alloc_start);
3193 	if (IS_ERR(em)) {
3194 		ret = PTR_ERR(em);
3195 		goto out;
3196 	}
3197 
3198 	/*
3199 	 * Avoid hole punching and extent allocation for some cases. More cases
3200 	 * could be considered, but these are unlikely common and we keep things
3201 	 * as simple as possible for now. Also, intentionally, if the target
3202 	 * range contains one or more prealloc extents together with regular
3203 	 * extents and holes, we drop all the existing extents and allocate a
3204 	 * new prealloc extent, so that we get a larger contiguous disk extent.
3205 	 */
3206 	if (em->start <= alloc_start &&
3207 	    test_bit(EXTENT_FLAG_PREALLOC, &em->flags)) {
3208 		const u64 em_end = em->start + em->len;
3209 
3210 		if (em_end >= offset + len) {
3211 			/*
3212 			 * The whole range is already a prealloc extent,
3213 			 * do nothing except updating the inode's i_size if
3214 			 * needed.
3215 			 */
3216 			free_extent_map(em);
3217 			ret = btrfs_fallocate_update_isize(inode, offset + len,
3218 							   mode);
3219 			goto out;
3220 		}
3221 		/*
3222 		 * Part of the range is already a prealloc extent, so operate
3223 		 * only on the remaining part of the range.
3224 		 */
3225 		alloc_start = em_end;
3226 		ASSERT(IS_ALIGNED(alloc_start, sectorsize));
3227 		len = offset + len - alloc_start;
3228 		offset = alloc_start;
3229 		alloc_hint = em->block_start + em->len;
3230 	}
3231 	free_extent_map(em);
3232 
3233 	if (BTRFS_BYTES_TO_BLKS(fs_info, offset) ==
3234 	    BTRFS_BYTES_TO_BLKS(fs_info, offset + len - 1)) {
3235 		em = btrfs_get_extent(BTRFS_I(inode), NULL, 0, alloc_start,
3236 				      sectorsize);
3237 		if (IS_ERR(em)) {
3238 			ret = PTR_ERR(em);
3239 			goto out;
3240 		}
3241 
3242 		if (test_bit(EXTENT_FLAG_PREALLOC, &em->flags)) {
3243 			free_extent_map(em);
3244 			ret = btrfs_fallocate_update_isize(inode, offset + len,
3245 							   mode);
3246 			goto out;
3247 		}
3248 		if (len < sectorsize && em->block_start != EXTENT_MAP_HOLE) {
3249 			free_extent_map(em);
3250 			ret = btrfs_truncate_block(BTRFS_I(inode), offset, len,
3251 						   0);
3252 			if (!ret)
3253 				ret = btrfs_fallocate_update_isize(inode,
3254 								   offset + len,
3255 								   mode);
3256 			return ret;
3257 		}
3258 		free_extent_map(em);
3259 		alloc_start = round_down(offset, sectorsize);
3260 		alloc_end = alloc_start + sectorsize;
3261 		goto reserve_space;
3262 	}
3263 
3264 	alloc_start = round_up(offset, sectorsize);
3265 	alloc_end = round_down(offset + len, sectorsize);
3266 
3267 	/*
3268 	 * For unaligned ranges, check the pages at the boundaries, they might
3269 	 * map to an extent, in which case we need to partially zero them, or
3270 	 * they might map to a hole, in which case we need our allocation range
3271 	 * to cover them.
3272 	 */
3273 	if (!IS_ALIGNED(offset, sectorsize)) {
3274 		ret = btrfs_zero_range_check_range_boundary(BTRFS_I(inode),
3275 							    offset);
3276 		if (ret < 0)
3277 			goto out;
3278 		if (ret == RANGE_BOUNDARY_HOLE) {
3279 			alloc_start = round_down(offset, sectorsize);
3280 			ret = 0;
3281 		} else if (ret == RANGE_BOUNDARY_WRITTEN_EXTENT) {
3282 			ret = btrfs_truncate_block(BTRFS_I(inode), offset, 0, 0);
3283 			if (ret)
3284 				goto out;
3285 		} else {
3286 			ret = 0;
3287 		}
3288 	}
3289 
3290 	if (!IS_ALIGNED(offset + len, sectorsize)) {
3291 		ret = btrfs_zero_range_check_range_boundary(BTRFS_I(inode),
3292 							    offset + len);
3293 		if (ret < 0)
3294 			goto out;
3295 		if (ret == RANGE_BOUNDARY_HOLE) {
3296 			alloc_end = round_up(offset + len, sectorsize);
3297 			ret = 0;
3298 		} else if (ret == RANGE_BOUNDARY_WRITTEN_EXTENT) {
3299 			ret = btrfs_truncate_block(BTRFS_I(inode), offset + len,
3300 						   0, 1);
3301 			if (ret)
3302 				goto out;
3303 		} else {
3304 			ret = 0;
3305 		}
3306 	}
3307 
3308 reserve_space:
3309 	if (alloc_start < alloc_end) {
3310 		struct extent_state *cached_state = NULL;
3311 		const u64 lockstart = alloc_start;
3312 		const u64 lockend = alloc_end - 1;
3313 
3314 		bytes_to_reserve = alloc_end - alloc_start;
3315 		ret = btrfs_alloc_data_chunk_ondemand(BTRFS_I(inode),
3316 						      bytes_to_reserve);
3317 		if (ret < 0)
3318 			goto out;
3319 		space_reserved = true;
3320 		btrfs_punch_hole_lock_range(inode, lockstart, lockend,
3321 					    &cached_state);
3322 		ret = btrfs_qgroup_reserve_data(BTRFS_I(inode), &data_reserved,
3323 						alloc_start, bytes_to_reserve);
3324 		if (ret) {
3325 			unlock_extent_cached(&BTRFS_I(inode)->io_tree, lockstart,
3326 					     lockend, &cached_state);
3327 			goto out;
3328 		}
3329 		ret = btrfs_prealloc_file_range(inode, mode, alloc_start,
3330 						alloc_end - alloc_start,
3331 						i_blocksize(inode),
3332 						offset + len, &alloc_hint);
3333 		unlock_extent_cached(&BTRFS_I(inode)->io_tree, lockstart,
3334 				     lockend, &cached_state);
3335 		/* btrfs_prealloc_file_range releases reserved space on error */
3336 		if (ret) {
3337 			space_reserved = false;
3338 			goto out;
3339 		}
3340 	}
3341 	ret = btrfs_fallocate_update_isize(inode, offset + len, mode);
3342  out:
3343 	if (ret && space_reserved)
3344 		btrfs_free_reserved_data_space(BTRFS_I(inode), data_reserved,
3345 					       alloc_start, bytes_to_reserve);
3346 	extent_changeset_free(data_reserved);
3347 
3348 	return ret;
3349 }
3350 
3351 static long btrfs_fallocate(struct file *file, int mode,
3352 			    loff_t offset, loff_t len)
3353 {
3354 	struct inode *inode = file_inode(file);
3355 	struct extent_state *cached_state = NULL;
3356 	struct extent_changeset *data_reserved = NULL;
3357 	struct falloc_range *range;
3358 	struct falloc_range *tmp;
3359 	struct list_head reserve_list;
3360 	u64 cur_offset;
3361 	u64 last_byte;
3362 	u64 alloc_start;
3363 	u64 alloc_end;
3364 	u64 alloc_hint = 0;
3365 	u64 locked_end;
3366 	u64 actual_end = 0;
3367 	u64 data_space_needed = 0;
3368 	u64 data_space_reserved = 0;
3369 	u64 qgroup_reserved = 0;
3370 	struct extent_map *em;
3371 	int blocksize = btrfs_inode_sectorsize(BTRFS_I(inode));
3372 	int ret;
3373 
3374 	/* Do not allow fallocate in ZONED mode */
3375 	if (btrfs_is_zoned(btrfs_sb(inode->i_sb)))
3376 		return -EOPNOTSUPP;
3377 
3378 	alloc_start = round_down(offset, blocksize);
3379 	alloc_end = round_up(offset + len, blocksize);
3380 	cur_offset = alloc_start;
3381 
3382 	/* Make sure we aren't being give some crap mode */
3383 	if (mode & ~(FALLOC_FL_KEEP_SIZE | FALLOC_FL_PUNCH_HOLE |
3384 		     FALLOC_FL_ZERO_RANGE))
3385 		return -EOPNOTSUPP;
3386 
3387 	if (mode & FALLOC_FL_PUNCH_HOLE)
3388 		return btrfs_punch_hole(file, offset, len);
3389 
3390 	btrfs_inode_lock(inode, BTRFS_ILOCK_MMAP);
3391 
3392 	if (!(mode & FALLOC_FL_KEEP_SIZE) && offset + len > inode->i_size) {
3393 		ret = inode_newsize_ok(inode, offset + len);
3394 		if (ret)
3395 			goto out;
3396 	}
3397 
3398 	ret = file_modified(file);
3399 	if (ret)
3400 		goto out;
3401 
3402 	/*
3403 	 * TODO: Move these two operations after we have checked
3404 	 * accurate reserved space, or fallocate can still fail but
3405 	 * with page truncated or size expanded.
3406 	 *
3407 	 * But that's a minor problem and won't do much harm BTW.
3408 	 */
3409 	if (alloc_start > inode->i_size) {
3410 		ret = btrfs_cont_expand(BTRFS_I(inode), i_size_read(inode),
3411 					alloc_start);
3412 		if (ret)
3413 			goto out;
3414 	} else if (offset + len > inode->i_size) {
3415 		/*
3416 		 * If we are fallocating from the end of the file onward we
3417 		 * need to zero out the end of the block if i_size lands in the
3418 		 * middle of a block.
3419 		 */
3420 		ret = btrfs_truncate_block(BTRFS_I(inode), inode->i_size, 0, 0);
3421 		if (ret)
3422 			goto out;
3423 	}
3424 
3425 	/*
3426 	 * We have locked the inode at the VFS level (in exclusive mode) and we
3427 	 * have locked the i_mmap_lock lock (in exclusive mode). Now before
3428 	 * locking the file range, flush all dealloc in the range and wait for
3429 	 * all ordered extents in the range to complete. After this we can lock
3430 	 * the file range and, due to the previous locking we did, we know there
3431 	 * can't be more delalloc or ordered extents in the range.
3432 	 */
3433 	ret = btrfs_wait_ordered_range(inode, alloc_start,
3434 				       alloc_end - alloc_start);
3435 	if (ret)
3436 		goto out;
3437 
3438 	if (mode & FALLOC_FL_ZERO_RANGE) {
3439 		ret = btrfs_zero_range(inode, offset, len, mode);
3440 		btrfs_inode_unlock(inode, BTRFS_ILOCK_MMAP);
3441 		return ret;
3442 	}
3443 
3444 	locked_end = alloc_end - 1;
3445 	lock_extent_bits(&BTRFS_I(inode)->io_tree, alloc_start, locked_end,
3446 			 &cached_state);
3447 
3448 	btrfs_assert_inode_range_clean(BTRFS_I(inode), alloc_start, locked_end);
3449 
3450 	/* First, check if we exceed the qgroup limit */
3451 	INIT_LIST_HEAD(&reserve_list);
3452 	while (cur_offset < alloc_end) {
3453 		em = btrfs_get_extent(BTRFS_I(inode), NULL, 0, cur_offset,
3454 				      alloc_end - cur_offset);
3455 		if (IS_ERR(em)) {
3456 			ret = PTR_ERR(em);
3457 			break;
3458 		}
3459 		last_byte = min(extent_map_end(em), alloc_end);
3460 		actual_end = min_t(u64, extent_map_end(em), offset + len);
3461 		last_byte = ALIGN(last_byte, blocksize);
3462 		if (em->block_start == EXTENT_MAP_HOLE ||
3463 		    (cur_offset >= inode->i_size &&
3464 		     !test_bit(EXTENT_FLAG_PREALLOC, &em->flags))) {
3465 			const u64 range_len = last_byte - cur_offset;
3466 
3467 			ret = add_falloc_range(&reserve_list, cur_offset, range_len);
3468 			if (ret < 0) {
3469 				free_extent_map(em);
3470 				break;
3471 			}
3472 			ret = btrfs_qgroup_reserve_data(BTRFS_I(inode),
3473 					&data_reserved, cur_offset, range_len);
3474 			if (ret < 0) {
3475 				free_extent_map(em);
3476 				break;
3477 			}
3478 			qgroup_reserved += range_len;
3479 			data_space_needed += range_len;
3480 		}
3481 		free_extent_map(em);
3482 		cur_offset = last_byte;
3483 	}
3484 
3485 	if (!ret && data_space_needed > 0) {
3486 		/*
3487 		 * We are safe to reserve space here as we can't have delalloc
3488 		 * in the range, see above.
3489 		 */
3490 		ret = btrfs_alloc_data_chunk_ondemand(BTRFS_I(inode),
3491 						      data_space_needed);
3492 		if (!ret)
3493 			data_space_reserved = data_space_needed;
3494 	}
3495 
3496 	/*
3497 	 * If ret is still 0, means we're OK to fallocate.
3498 	 * Or just cleanup the list and exit.
3499 	 */
3500 	list_for_each_entry_safe(range, tmp, &reserve_list, list) {
3501 		if (!ret) {
3502 			ret = btrfs_prealloc_file_range(inode, mode,
3503 					range->start,
3504 					range->len, i_blocksize(inode),
3505 					offset + len, &alloc_hint);
3506 			/*
3507 			 * btrfs_prealloc_file_range() releases space even
3508 			 * if it returns an error.
3509 			 */
3510 			data_space_reserved -= range->len;
3511 			qgroup_reserved -= range->len;
3512 		} else if (data_space_reserved > 0) {
3513 			btrfs_free_reserved_data_space(BTRFS_I(inode),
3514 					       data_reserved, range->start,
3515 					       range->len);
3516 			data_space_reserved -= range->len;
3517 			qgroup_reserved -= range->len;
3518 		} else if (qgroup_reserved > 0) {
3519 			btrfs_qgroup_free_data(BTRFS_I(inode), data_reserved,
3520 					       range->start, range->len);
3521 			qgroup_reserved -= range->len;
3522 		}
3523 		list_del(&range->list);
3524 		kfree(range);
3525 	}
3526 	if (ret < 0)
3527 		goto out_unlock;
3528 
3529 	/*
3530 	 * We didn't need to allocate any more space, but we still extended the
3531 	 * size of the file so we need to update i_size and the inode item.
3532 	 */
3533 	ret = btrfs_fallocate_update_isize(inode, actual_end, mode);
3534 out_unlock:
3535 	unlock_extent_cached(&BTRFS_I(inode)->io_tree, alloc_start, locked_end,
3536 			     &cached_state);
3537 out:
3538 	btrfs_inode_unlock(inode, BTRFS_ILOCK_MMAP);
3539 	extent_changeset_free(data_reserved);
3540 	return ret;
3541 }
3542 
3543 static loff_t find_desired_extent(struct btrfs_inode *inode, loff_t offset,
3544 				  int whence)
3545 {
3546 	struct btrfs_fs_info *fs_info = inode->root->fs_info;
3547 	struct extent_map *em = NULL;
3548 	struct extent_state *cached_state = NULL;
3549 	loff_t i_size = inode->vfs_inode.i_size;
3550 	u64 lockstart;
3551 	u64 lockend;
3552 	u64 start;
3553 	u64 len;
3554 	int ret = 0;
3555 
3556 	if (i_size == 0 || offset >= i_size)
3557 		return -ENXIO;
3558 
3559 	/*
3560 	 * offset can be negative, in this case we start finding DATA/HOLE from
3561 	 * the very start of the file.
3562 	 */
3563 	start = max_t(loff_t, 0, offset);
3564 
3565 	lockstart = round_down(start, fs_info->sectorsize);
3566 	lockend = round_up(i_size, fs_info->sectorsize);
3567 	if (lockend <= lockstart)
3568 		lockend = lockstart + fs_info->sectorsize;
3569 	lockend--;
3570 	len = lockend - lockstart + 1;
3571 
3572 	lock_extent_bits(&inode->io_tree, lockstart, lockend, &cached_state);
3573 
3574 	while (start < i_size) {
3575 		em = btrfs_get_extent_fiemap(inode, start, len);
3576 		if (IS_ERR(em)) {
3577 			ret = PTR_ERR(em);
3578 			em = NULL;
3579 			break;
3580 		}
3581 
3582 		if (whence == SEEK_HOLE &&
3583 		    (em->block_start == EXTENT_MAP_HOLE ||
3584 		     test_bit(EXTENT_FLAG_PREALLOC, &em->flags)))
3585 			break;
3586 		else if (whence == SEEK_DATA &&
3587 			   (em->block_start != EXTENT_MAP_HOLE &&
3588 			    !test_bit(EXTENT_FLAG_PREALLOC, &em->flags)))
3589 			break;
3590 
3591 		start = em->start + em->len;
3592 		free_extent_map(em);
3593 		em = NULL;
3594 		cond_resched();
3595 	}
3596 	free_extent_map(em);
3597 	unlock_extent_cached(&inode->io_tree, lockstart, lockend,
3598 			     &cached_state);
3599 	if (ret) {
3600 		offset = ret;
3601 	} else {
3602 		if (whence == SEEK_DATA && start >= i_size)
3603 			offset = -ENXIO;
3604 		else
3605 			offset = min_t(loff_t, start, i_size);
3606 	}
3607 
3608 	return offset;
3609 }
3610 
3611 static loff_t btrfs_file_llseek(struct file *file, loff_t offset, int whence)
3612 {
3613 	struct inode *inode = file->f_mapping->host;
3614 
3615 	switch (whence) {
3616 	default:
3617 		return generic_file_llseek(file, offset, whence);
3618 	case SEEK_DATA:
3619 	case SEEK_HOLE:
3620 		btrfs_inode_lock(inode, BTRFS_ILOCK_SHARED);
3621 		offset = find_desired_extent(BTRFS_I(inode), offset, whence);
3622 		btrfs_inode_unlock(inode, BTRFS_ILOCK_SHARED);
3623 		break;
3624 	}
3625 
3626 	if (offset < 0)
3627 		return offset;
3628 
3629 	return vfs_setpos(file, offset, inode->i_sb->s_maxbytes);
3630 }
3631 
3632 static int btrfs_file_open(struct inode *inode, struct file *filp)
3633 {
3634 	int ret;
3635 
3636 	filp->f_mode |= FMODE_NOWAIT | FMODE_BUF_RASYNC;
3637 
3638 	ret = fsverity_file_open(inode, filp);
3639 	if (ret)
3640 		return ret;
3641 	return generic_file_open(inode, filp);
3642 }
3643 
3644 static int check_direct_read(struct btrfs_fs_info *fs_info,
3645 			     const struct iov_iter *iter, loff_t offset)
3646 {
3647 	int ret;
3648 	int i, seg;
3649 
3650 	ret = check_direct_IO(fs_info, iter, offset);
3651 	if (ret < 0)
3652 		return ret;
3653 
3654 	if (!iter_is_iovec(iter))
3655 		return 0;
3656 
3657 	for (seg = 0; seg < iter->nr_segs; seg++)
3658 		for (i = seg + 1; i < iter->nr_segs; i++)
3659 			if (iter->iov[seg].iov_base == iter->iov[i].iov_base)
3660 				return -EINVAL;
3661 	return 0;
3662 }
3663 
3664 static ssize_t btrfs_direct_read(struct kiocb *iocb, struct iov_iter *to)
3665 {
3666 	struct inode *inode = file_inode(iocb->ki_filp);
3667 	size_t prev_left = 0;
3668 	ssize_t read = 0;
3669 	ssize_t ret;
3670 
3671 	if (fsverity_active(inode))
3672 		return 0;
3673 
3674 	if (check_direct_read(btrfs_sb(inode->i_sb), to, iocb->ki_pos))
3675 		return 0;
3676 
3677 	btrfs_inode_lock(inode, BTRFS_ILOCK_SHARED);
3678 again:
3679 	/*
3680 	 * This is similar to what we do for direct IO writes, see the comment
3681 	 * at btrfs_direct_write(), but we also disable page faults in addition
3682 	 * to disabling them only at the iov_iter level. This is because when
3683 	 * reading from a hole or prealloc extent, iomap calls iov_iter_zero(),
3684 	 * which can still trigger page fault ins despite having set ->nofault
3685 	 * to true of our 'to' iov_iter.
3686 	 *
3687 	 * The difference to direct IO writes is that we deadlock when trying
3688 	 * to lock the extent range in the inode's tree during he page reads
3689 	 * triggered by the fault in (while for writes it is due to waiting for
3690 	 * our own ordered extent). This is because for direct IO reads,
3691 	 * btrfs_dio_iomap_begin() returns with the extent range locked, which
3692 	 * is only unlocked in the endio callback (end_bio_extent_readpage()).
3693 	 */
3694 	pagefault_disable();
3695 	to->nofault = true;
3696 	ret = btrfs_dio_rw(iocb, to, read);
3697 	to->nofault = false;
3698 	pagefault_enable();
3699 
3700 	/* No increment (+=) because iomap returns a cumulative value. */
3701 	if (ret > 0)
3702 		read = ret;
3703 
3704 	if (iov_iter_count(to) > 0 && (ret == -EFAULT || ret > 0)) {
3705 		const size_t left = iov_iter_count(to);
3706 
3707 		if (left == prev_left) {
3708 			/*
3709 			 * We didn't make any progress since the last attempt,
3710 			 * fallback to a buffered read for the remainder of the
3711 			 * range. This is just to avoid any possibility of looping
3712 			 * for too long.
3713 			 */
3714 			ret = read;
3715 		} else {
3716 			/*
3717 			 * We made some progress since the last retry or this is
3718 			 * the first time we are retrying. Fault in as many pages
3719 			 * as possible and retry.
3720 			 */
3721 			fault_in_iov_iter_writeable(to, left);
3722 			prev_left = left;
3723 			goto again;
3724 		}
3725 	}
3726 	btrfs_inode_unlock(inode, BTRFS_ILOCK_SHARED);
3727 	return ret < 0 ? ret : read;
3728 }
3729 
3730 static ssize_t btrfs_file_read_iter(struct kiocb *iocb, struct iov_iter *to)
3731 {
3732 	ssize_t ret = 0;
3733 
3734 	if (iocb->ki_flags & IOCB_DIRECT) {
3735 		ret = btrfs_direct_read(iocb, to);
3736 		if (ret < 0 || !iov_iter_count(to) ||
3737 		    iocb->ki_pos >= i_size_read(file_inode(iocb->ki_filp)))
3738 			return ret;
3739 	}
3740 
3741 	return filemap_read(iocb, to, ret);
3742 }
3743 
3744 const struct file_operations btrfs_file_operations = {
3745 	.llseek		= btrfs_file_llseek,
3746 	.read_iter      = btrfs_file_read_iter,
3747 	.splice_read	= generic_file_splice_read,
3748 	.write_iter	= btrfs_file_write_iter,
3749 	.splice_write	= iter_file_splice_write,
3750 	.mmap		= btrfs_file_mmap,
3751 	.open		= btrfs_file_open,
3752 	.release	= btrfs_release_file,
3753 	.fsync		= btrfs_sync_file,
3754 	.fallocate	= btrfs_fallocate,
3755 	.unlocked_ioctl	= btrfs_ioctl,
3756 #ifdef CONFIG_COMPAT
3757 	.compat_ioctl	= btrfs_compat_ioctl,
3758 #endif
3759 	.remap_file_range = btrfs_remap_file_range,
3760 };
3761 
3762 void __cold btrfs_auto_defrag_exit(void)
3763 {
3764 	kmem_cache_destroy(btrfs_inode_defrag_cachep);
3765 }
3766 
3767 int __init btrfs_auto_defrag_init(void)
3768 {
3769 	btrfs_inode_defrag_cachep = kmem_cache_create("btrfs_inode_defrag",
3770 					sizeof(struct inode_defrag), 0,
3771 					SLAB_MEM_SPREAD,
3772 					NULL);
3773 	if (!btrfs_inode_defrag_cachep)
3774 		return -ENOMEM;
3775 
3776 	return 0;
3777 }
3778 
3779 int btrfs_fdatawrite_range(struct inode *inode, loff_t start, loff_t end)
3780 {
3781 	int ret;
3782 
3783 	/*
3784 	 * So with compression we will find and lock a dirty page and clear the
3785 	 * first one as dirty, setup an async extent, and immediately return
3786 	 * with the entire range locked but with nobody actually marked with
3787 	 * writeback.  So we can't just filemap_write_and_wait_range() and
3788 	 * expect it to work since it will just kick off a thread to do the
3789 	 * actual work.  So we need to call filemap_fdatawrite_range _again_
3790 	 * since it will wait on the page lock, which won't be unlocked until
3791 	 * after the pages have been marked as writeback and so we're good to go
3792 	 * from there.  We have to do this otherwise we'll miss the ordered
3793 	 * extents and that results in badness.  Please Josef, do not think you
3794 	 * know better and pull this out at some point in the future, it is
3795 	 * right and you are wrong.
3796 	 */
3797 	ret = filemap_fdatawrite_range(inode->i_mapping, start, end);
3798 	if (!ret && test_bit(BTRFS_INODE_HAS_ASYNC_EXTENT,
3799 			     &BTRFS_I(inode)->runtime_flags))
3800 		ret = filemap_fdatawrite_range(inode->i_mapping, start, end);
3801 
3802 	return ret;
3803 }
3804