]> git.proxmox.com Git - mirror_ubuntu-bionic-kernel.git/blame - fs/btrfs/file.c
Merge branch 'WIP.x86-pti.base.prep-for-linus' of git://git.kernel.org/pub/scm/linux...
[mirror_ubuntu-bionic-kernel.git] / fs / btrfs / file.c
CommitLineData
6cbd5570
CM
1/*
2 * Copyright (C) 2007 Oracle. All rights reserved.
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public
6 * License v2 as published by the Free Software Foundation.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 * General Public License for more details.
12 *
13 * You should have received a copy of the GNU General Public
14 * License along with this program; if not, write to the
15 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
16 * Boston, MA 021110-1307, USA.
17 */
18
39279cc3
CM
19#include <linux/fs.h>
20#include <linux/pagemap.h>
21#include <linux/highmem.h>
22#include <linux/time.h>
23#include <linux/init.h>
24#include <linux/string.h>
39279cc3
CM
25#include <linux/backing-dev.h>
26#include <linux/mpage.h>
2fe17c10 27#include <linux/falloc.h>
39279cc3
CM
28#include <linux/swap.h>
29#include <linux/writeback.h>
39279cc3 30#include <linux/compat.h>
5a0e3ad6 31#include <linux/slab.h>
55e301fd 32#include <linux/btrfs.h>
e2e40f2c 33#include <linux/uio.h>
39279cc3
CM
34#include "ctree.h"
35#include "disk-io.h"
36#include "transaction.h"
37#include "btrfs_inode.h"
39279cc3 38#include "print-tree.h"
e02119d5
CM
39#include "tree-log.h"
40#include "locking.h"
2aaa6655 41#include "volumes.h"
fcebe456 42#include "qgroup.h"
ebb8765b 43#include "compression.h"
39279cc3 44
9247f317 45static struct kmem_cache *btrfs_inode_defrag_cachep;
4cb5300b
CM
46/*
47 * when auto defrag is enabled we
48 * queue up these defrag structs to remember which
49 * inodes need defragging passes
50 */
51struct inode_defrag {
52 struct rb_node rb_node;
53 /* objectid */
54 u64 ino;
55 /*
56 * transid where the defrag was added, we search for
57 * extents newer than this
58 */
59 u64 transid;
60
61 /* root objectid */
62 u64 root;
63
64 /* last offset we were able to defrag */
65 u64 last_offset;
66
67 /* if we've wrapped around back to zero once already */
68 int cycled;
69};
70
762f2263
MX
71static int __compare_inode_defrag(struct inode_defrag *defrag1,
72 struct inode_defrag *defrag2)
73{
74 if (defrag1->root > defrag2->root)
75 return 1;
76 else if (defrag1->root < defrag2->root)
77 return -1;
78 else if (defrag1->ino > defrag2->ino)
79 return 1;
80 else if (defrag1->ino < defrag2->ino)
81 return -1;
82 else
83 return 0;
84}
85
4cb5300b
CM
86/* pop a record for an inode into the defrag tree. The lock
87 * must be held already
88 *
89 * If you're inserting a record for an older transid than an
90 * existing record, the transid already in the tree is lowered
91 *
92 * If an existing record is found the defrag item you
93 * pass in is freed
94 */
6158e1ce 95static int __btrfs_add_inode_defrag(struct btrfs_inode *inode,
4cb5300b
CM
96 struct inode_defrag *defrag)
97{
6158e1ce 98 struct btrfs_fs_info *fs_info = btrfs_sb(inode->vfs_inode.i_sb);
4cb5300b
CM
99 struct inode_defrag *entry;
100 struct rb_node **p;
101 struct rb_node *parent = NULL;
762f2263 102 int ret;
4cb5300b 103
0b246afa 104 p = &fs_info->defrag_inodes.rb_node;
4cb5300b
CM
105 while (*p) {
106 parent = *p;
107 entry = rb_entry(parent, struct inode_defrag, rb_node);
108
762f2263
MX
109 ret = __compare_inode_defrag(defrag, entry);
110 if (ret < 0)
4cb5300b 111 p = &parent->rb_left;
762f2263 112 else if (ret > 0)
4cb5300b
CM
113 p = &parent->rb_right;
114 else {
115 /* if we're reinserting an entry for
116 * an old defrag run, make sure to
117 * lower the transid of our existing record
118 */
119 if (defrag->transid < entry->transid)
120 entry->transid = defrag->transid;
121 if (defrag->last_offset > entry->last_offset)
122 entry->last_offset = defrag->last_offset;
8ddc4734 123 return -EEXIST;
4cb5300b
CM
124 }
125 }
6158e1ce 126 set_bit(BTRFS_INODE_IN_DEFRAG, &inode->runtime_flags);
4cb5300b 127 rb_link_node(&defrag->rb_node, parent, p);
0b246afa 128 rb_insert_color(&defrag->rb_node, &fs_info->defrag_inodes);
8ddc4734
MX
129 return 0;
130}
4cb5300b 131
2ff7e61e 132static inline int __need_auto_defrag(struct btrfs_fs_info *fs_info)
8ddc4734 133{
0b246afa 134 if (!btrfs_test_opt(fs_info, AUTO_DEFRAG))
8ddc4734
MX
135 return 0;
136
0b246afa 137 if (btrfs_fs_closing(fs_info))
8ddc4734 138 return 0;
4cb5300b 139
8ddc4734 140 return 1;
4cb5300b
CM
141}
142
143/*
144 * insert a defrag record for this inode if auto defrag is
145 * enabled
146 */
147int btrfs_add_inode_defrag(struct btrfs_trans_handle *trans,
6158e1ce 148 struct btrfs_inode *inode)
4cb5300b 149{
6158e1ce
NB
150 struct btrfs_fs_info *fs_info = btrfs_sb(inode->vfs_inode.i_sb);
151 struct btrfs_root *root = inode->root;
4cb5300b 152 struct inode_defrag *defrag;
4cb5300b 153 u64 transid;
8ddc4734 154 int ret;
4cb5300b 155
2ff7e61e 156 if (!__need_auto_defrag(fs_info))
4cb5300b
CM
157 return 0;
158
6158e1ce 159 if (test_bit(BTRFS_INODE_IN_DEFRAG, &inode->runtime_flags))
4cb5300b
CM
160 return 0;
161
162 if (trans)
163 transid = trans->transid;
164 else
6158e1ce 165 transid = inode->root->last_trans;
4cb5300b 166
9247f317 167 defrag = kmem_cache_zalloc(btrfs_inode_defrag_cachep, GFP_NOFS);
4cb5300b
CM
168 if (!defrag)
169 return -ENOMEM;
170
6158e1ce 171 defrag->ino = btrfs_ino(inode);
4cb5300b
CM
172 defrag->transid = transid;
173 defrag->root = root->root_key.objectid;
174
0b246afa 175 spin_lock(&fs_info->defrag_inodes_lock);
6158e1ce 176 if (!test_bit(BTRFS_INODE_IN_DEFRAG, &inode->runtime_flags)) {
8ddc4734
MX
177 /*
178 * If we set IN_DEFRAG flag and evict the inode from memory,
179 * and then re-read this inode, this new inode doesn't have
180 * IN_DEFRAG flag. At the case, we may find the existed defrag.
181 */
182 ret = __btrfs_add_inode_defrag(inode, defrag);
183 if (ret)
184 kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
185 } else {
9247f317 186 kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
8ddc4734 187 }
0b246afa 188 spin_unlock(&fs_info->defrag_inodes_lock);
a0f98dde 189 return 0;
4cb5300b
CM
190}
191
192/*
8ddc4734
MX
193 * Requeue the defrag object. If there is a defrag object that points to
194 * the same inode in the tree, we will merge them together (by
195 * __btrfs_add_inode_defrag()) and free the one that we want to requeue.
4cb5300b 196 */
46e59791 197static void btrfs_requeue_inode_defrag(struct btrfs_inode *inode,
48a3b636 198 struct inode_defrag *defrag)
8ddc4734 199{
46e59791 200 struct btrfs_fs_info *fs_info = btrfs_sb(inode->vfs_inode.i_sb);
8ddc4734
MX
201 int ret;
202
2ff7e61e 203 if (!__need_auto_defrag(fs_info))
8ddc4734
MX
204 goto out;
205
206 /*
207 * Here we don't check the IN_DEFRAG flag, because we need merge
208 * them together.
209 */
0b246afa 210 spin_lock(&fs_info->defrag_inodes_lock);
8ddc4734 211 ret = __btrfs_add_inode_defrag(inode, defrag);
0b246afa 212 spin_unlock(&fs_info->defrag_inodes_lock);
8ddc4734
MX
213 if (ret)
214 goto out;
215 return;
216out:
217 kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
218}
219
4cb5300b 220/*
26176e7c
MX
221 * pick the defragable inode that we want, if it doesn't exist, we will get
222 * the next one.
4cb5300b 223 */
26176e7c
MX
224static struct inode_defrag *
225btrfs_pick_defrag_inode(struct btrfs_fs_info *fs_info, u64 root, u64 ino)
4cb5300b
CM
226{
227 struct inode_defrag *entry = NULL;
762f2263 228 struct inode_defrag tmp;
4cb5300b
CM
229 struct rb_node *p;
230 struct rb_node *parent = NULL;
762f2263
MX
231 int ret;
232
233 tmp.ino = ino;
234 tmp.root = root;
4cb5300b 235
26176e7c
MX
236 spin_lock(&fs_info->defrag_inodes_lock);
237 p = fs_info->defrag_inodes.rb_node;
4cb5300b
CM
238 while (p) {
239 parent = p;
240 entry = rb_entry(parent, struct inode_defrag, rb_node);
241
762f2263
MX
242 ret = __compare_inode_defrag(&tmp, entry);
243 if (ret < 0)
4cb5300b 244 p = parent->rb_left;
762f2263 245 else if (ret > 0)
4cb5300b
CM
246 p = parent->rb_right;
247 else
26176e7c 248 goto out;
4cb5300b
CM
249 }
250
26176e7c
MX
251 if (parent && __compare_inode_defrag(&tmp, entry) > 0) {
252 parent = rb_next(parent);
253 if (parent)
4cb5300b 254 entry = rb_entry(parent, struct inode_defrag, rb_node);
26176e7c
MX
255 else
256 entry = NULL;
4cb5300b 257 }
26176e7c
MX
258out:
259 if (entry)
260 rb_erase(parent, &fs_info->defrag_inodes);
261 spin_unlock(&fs_info->defrag_inodes_lock);
262 return entry;
4cb5300b
CM
263}
264
26176e7c 265void btrfs_cleanup_defrag_inodes(struct btrfs_fs_info *fs_info)
4cb5300b
CM
266{
267 struct inode_defrag *defrag;
26176e7c
MX
268 struct rb_node *node;
269
270 spin_lock(&fs_info->defrag_inodes_lock);
271 node = rb_first(&fs_info->defrag_inodes);
272 while (node) {
273 rb_erase(node, &fs_info->defrag_inodes);
274 defrag = rb_entry(node, struct inode_defrag, rb_node);
275 kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
276
351810c1 277 cond_resched_lock(&fs_info->defrag_inodes_lock);
26176e7c
MX
278
279 node = rb_first(&fs_info->defrag_inodes);
280 }
281 spin_unlock(&fs_info->defrag_inodes_lock);
282}
283
284#define BTRFS_DEFRAG_BATCH 1024
285
286static int __btrfs_run_defrag_inode(struct btrfs_fs_info *fs_info,
287 struct inode_defrag *defrag)
288{
4cb5300b
CM
289 struct btrfs_root *inode_root;
290 struct inode *inode;
4cb5300b
CM
291 struct btrfs_key key;
292 struct btrfs_ioctl_defrag_range_args range;
4cb5300b 293 int num_defrag;
6f1c3605
LB
294 int index;
295 int ret;
4cb5300b 296
26176e7c
MX
297 /* get the inode */
298 key.objectid = defrag->root;
962a298f 299 key.type = BTRFS_ROOT_ITEM_KEY;
26176e7c 300 key.offset = (u64)-1;
6f1c3605
LB
301
302 index = srcu_read_lock(&fs_info->subvol_srcu);
303
26176e7c
MX
304 inode_root = btrfs_read_fs_root_no_name(fs_info, &key);
305 if (IS_ERR(inode_root)) {
6f1c3605
LB
306 ret = PTR_ERR(inode_root);
307 goto cleanup;
308 }
26176e7c
MX
309
310 key.objectid = defrag->ino;
962a298f 311 key.type = BTRFS_INODE_ITEM_KEY;
26176e7c
MX
312 key.offset = 0;
313 inode = btrfs_iget(fs_info->sb, &key, inode_root, NULL);
314 if (IS_ERR(inode)) {
6f1c3605
LB
315 ret = PTR_ERR(inode);
316 goto cleanup;
26176e7c 317 }
6f1c3605 318 srcu_read_unlock(&fs_info->subvol_srcu, index);
26176e7c
MX
319
320 /* do a chunk of defrag */
321 clear_bit(BTRFS_INODE_IN_DEFRAG, &BTRFS_I(inode)->runtime_flags);
4cb5300b
CM
322 memset(&range, 0, sizeof(range));
323 range.len = (u64)-1;
26176e7c 324 range.start = defrag->last_offset;
b66f00da
MX
325
326 sb_start_write(fs_info->sb);
26176e7c
MX
327 num_defrag = btrfs_defrag_file(inode, NULL, &range, defrag->transid,
328 BTRFS_DEFRAG_BATCH);
b66f00da 329 sb_end_write(fs_info->sb);
26176e7c
MX
330 /*
331 * if we filled the whole defrag batch, there
332 * must be more work to do. Queue this defrag
333 * again
334 */
335 if (num_defrag == BTRFS_DEFRAG_BATCH) {
336 defrag->last_offset = range.start;
46e59791 337 btrfs_requeue_inode_defrag(BTRFS_I(inode), defrag);
26176e7c
MX
338 } else if (defrag->last_offset && !defrag->cycled) {
339 /*
340 * we didn't fill our defrag batch, but
341 * we didn't start at zero. Make sure we loop
342 * around to the start of the file.
343 */
344 defrag->last_offset = 0;
345 defrag->cycled = 1;
46e59791 346 btrfs_requeue_inode_defrag(BTRFS_I(inode), defrag);
26176e7c
MX
347 } else {
348 kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
349 }
350
351 iput(inode);
352 return 0;
6f1c3605
LB
353cleanup:
354 srcu_read_unlock(&fs_info->subvol_srcu, index);
355 kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
356 return ret;
26176e7c
MX
357}
358
359/*
360 * run through the list of inodes in the FS that need
361 * defragging
362 */
363int btrfs_run_defrag_inodes(struct btrfs_fs_info *fs_info)
364{
365 struct inode_defrag *defrag;
366 u64 first_ino = 0;
367 u64 root_objectid = 0;
4cb5300b
CM
368
369 atomic_inc(&fs_info->defrag_running);
67871254 370 while (1) {
dc81cdc5
MX
371 /* Pause the auto defragger. */
372 if (test_bit(BTRFS_FS_STATE_REMOUNTING,
373 &fs_info->fs_state))
374 break;
375
2ff7e61e 376 if (!__need_auto_defrag(fs_info))
26176e7c 377 break;
4cb5300b
CM
378
379 /* find an inode to defrag */
26176e7c
MX
380 defrag = btrfs_pick_defrag_inode(fs_info, root_objectid,
381 first_ino);
4cb5300b 382 if (!defrag) {
26176e7c 383 if (root_objectid || first_ino) {
762f2263 384 root_objectid = 0;
4cb5300b
CM
385 first_ino = 0;
386 continue;
387 } else {
388 break;
389 }
390 }
391
4cb5300b 392 first_ino = defrag->ino + 1;
762f2263 393 root_objectid = defrag->root;
4cb5300b 394
26176e7c 395 __btrfs_run_defrag_inode(fs_info, defrag);
4cb5300b 396 }
4cb5300b
CM
397 atomic_dec(&fs_info->defrag_running);
398
399 /*
400 * during unmount, we use the transaction_wait queue to
401 * wait for the defragger to stop
402 */
403 wake_up(&fs_info->transaction_wait);
404 return 0;
405}
39279cc3 406
d352ac68
CM
407/* simple helper to fault in pages and copy. This should go away
408 * and be replaced with calls into generic code.
409 */
ee22f0c4 410static noinline int btrfs_copy_from_user(loff_t pos, size_t write_bytes,
a1b32a59 411 struct page **prepared_pages,
11c65dcc 412 struct iov_iter *i)
39279cc3 413{
914ee295 414 size_t copied = 0;
d0215f3e 415 size_t total_copied = 0;
11c65dcc 416 int pg = 0;
09cbfeaf 417 int offset = pos & (PAGE_SIZE - 1);
39279cc3 418
11c65dcc 419 while (write_bytes > 0) {
39279cc3 420 size_t count = min_t(size_t,
09cbfeaf 421 PAGE_SIZE - offset, write_bytes);
11c65dcc 422 struct page *page = prepared_pages[pg];
914ee295
XZ
423 /*
424 * Copy data from userspace to the current page
914ee295 425 */
914ee295 426 copied = iov_iter_copy_from_user_atomic(page, i, offset, count);
11c65dcc 427
39279cc3
CM
428 /* Flush processor's dcache for this page */
429 flush_dcache_page(page);
31339acd
CM
430
431 /*
432 * if we get a partial write, we can end up with
433 * partially up to date pages. These add
434 * a lot of complexity, so make sure they don't
435 * happen by forcing this copy to be retried.
436 *
437 * The rest of the btrfs_file_write code will fall
438 * back to page at a time copies after we return 0.
439 */
440 if (!PageUptodate(page) && copied < count)
441 copied = 0;
442
11c65dcc
JB
443 iov_iter_advance(i, copied);
444 write_bytes -= copied;
914ee295 445 total_copied += copied;
39279cc3 446
b30ac0fc 447 /* Return to btrfs_file_write_iter to fault page */
9f570b8d 448 if (unlikely(copied == 0))
914ee295 449 break;
11c65dcc 450
09cbfeaf 451 if (copied < PAGE_SIZE - offset) {
11c65dcc
JB
452 offset += copied;
453 } else {
454 pg++;
455 offset = 0;
456 }
39279cc3 457 }
914ee295 458 return total_copied;
39279cc3
CM
459}
460
d352ac68
CM
461/*
462 * unlocks pages after btrfs_file_write is done with them
463 */
48a3b636 464static void btrfs_drop_pages(struct page **pages, size_t num_pages)
39279cc3
CM
465{
466 size_t i;
467 for (i = 0; i < num_pages; i++) {
d352ac68
CM
468 /* page checked is some magic around finding pages that
469 * have been modified without going through btrfs_set_page_dirty
2457aec6
MG
470 * clear it here. There should be no need to mark the pages
471 * accessed as prepare_pages should have marked them accessed
472 * in prepare_pages via find_or_create_page()
d352ac68 473 */
4a096752 474 ClearPageChecked(pages[i]);
39279cc3 475 unlock_page(pages[i]);
09cbfeaf 476 put_page(pages[i]);
39279cc3
CM
477 }
478}
479
f48bf66b
FM
480static int btrfs_find_new_delalloc_bytes(struct btrfs_inode *inode,
481 const u64 start,
482 const u64 len,
483 struct extent_state **cached_state)
484{
485 u64 search_start = start;
486 const u64 end = start + len - 1;
487
488 while (search_start < end) {
489 const u64 search_len = end - search_start + 1;
490 struct extent_map *em;
491 u64 em_len;
492 int ret = 0;
493
494 em = btrfs_get_extent(inode, NULL, 0, search_start,
495 search_len, 0);
496 if (IS_ERR(em))
497 return PTR_ERR(em);
498
499 if (em->block_start != EXTENT_MAP_HOLE)
500 goto next;
501
502 em_len = em->len;
503 if (em->start < search_start)
504 em_len -= search_start - em->start;
505 if (em_len > search_len)
506 em_len = search_len;
507
508 ret = set_extent_bit(&inode->io_tree, search_start,
509 search_start + em_len - 1,
510 EXTENT_DELALLOC_NEW,
511 NULL, cached_state, GFP_NOFS);
512next:
513 search_start = extent_map_end(em);
514 free_extent_map(em);
515 if (ret)
516 return ret;
517 }
518 return 0;
519}
520
d352ac68
CM
521/*
522 * after copy_from_user, pages need to be dirtied and we need to make
523 * sure holes are created between the current EOF and the start of
524 * any next extents (if required).
525 *
526 * this also makes the decision about creating an inline extent vs
527 * doing real data extents, marking pages dirty and delalloc as required.
528 */
2ff7e61e
JM
529int btrfs_dirty_pages(struct inode *inode, struct page **pages,
530 size_t num_pages, loff_t pos, size_t write_bytes,
531 struct extent_state **cached)
39279cc3 532{
0b246afa 533 struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
39279cc3 534 int err = 0;
a52d9a80 535 int i;
db94535d 536 u64 num_bytes;
a52d9a80
CM
537 u64 start_pos;
538 u64 end_of_last_block;
539 u64 end_pos = pos + write_bytes;
540 loff_t isize = i_size_read(inode);
e3b8a485 541 unsigned int extra_bits = 0;
39279cc3 542
0b246afa 543 start_pos = pos & ~((u64) fs_info->sectorsize - 1);
da17066c 544 num_bytes = round_up(write_bytes + pos - start_pos,
0b246afa 545 fs_info->sectorsize);
39279cc3 546
db94535d 547 end_of_last_block = start_pos + num_bytes - 1;
e3b8a485
FM
548
549 if (!btrfs_is_free_space_inode(BTRFS_I(inode))) {
550 if (start_pos >= isize &&
551 !(BTRFS_I(inode)->flags & BTRFS_INODE_PREALLOC)) {
552 /*
553 * There can't be any extents following eof in this case
554 * so just set the delalloc new bit for the range
555 * directly.
556 */
557 extra_bits |= EXTENT_DELALLOC_NEW;
558 } else {
559 err = btrfs_find_new_delalloc_bytes(BTRFS_I(inode),
560 start_pos,
561 num_bytes, cached);
562 if (err)
563 return err;
564 }
565 }
566
2ac55d41 567 err = btrfs_set_extent_delalloc(inode, start_pos, end_of_last_block,
e3b8a485 568 extra_bits, cached, 0);
d0215f3e
JB
569 if (err)
570 return err;
9ed74f2d 571
c8b97818
CM
572 for (i = 0; i < num_pages; i++) {
573 struct page *p = pages[i];
574 SetPageUptodate(p);
575 ClearPageChecked(p);
576 set_page_dirty(p);
a52d9a80 577 }
9f570b8d
JB
578
579 /*
580 * we've only changed i_size in ram, and we haven't updated
581 * the disk i_size. There is no need to log the inode
582 * at this time.
583 */
584 if (end_pos > isize)
a52d9a80 585 i_size_write(inode, end_pos);
a22285a6 586 return 0;
39279cc3
CM
587}
588
d352ac68
CM
589/*
590 * this drops all the extents in the cache that intersect the range
591 * [start, end]. Existing extents are split as required.
592 */
dcdbc059 593void btrfs_drop_extent_cache(struct btrfs_inode *inode, u64 start, u64 end,
7014cdb4 594 int skip_pinned)
a52d9a80
CM
595{
596 struct extent_map *em;
3b951516
CM
597 struct extent_map *split = NULL;
598 struct extent_map *split2 = NULL;
dcdbc059 599 struct extent_map_tree *em_tree = &inode->extent_tree;
39b5637f 600 u64 len = end - start + 1;
5dc562c5 601 u64 gen;
3b951516
CM
602 int ret;
603 int testend = 1;
5b21f2ed 604 unsigned long flags;
c8b97818 605 int compressed = 0;
09a2a8f9 606 bool modified;
a52d9a80 607
e6dcd2dc 608 WARN_ON(end < start);
3b951516 609 if (end == (u64)-1) {
39b5637f 610 len = (u64)-1;
3b951516
CM
611 testend = 0;
612 }
d397712b 613 while (1) {
7014cdb4
JB
614 int no_splits = 0;
615
09a2a8f9 616 modified = false;
3b951516 617 if (!split)
172ddd60 618 split = alloc_extent_map();
3b951516 619 if (!split2)
172ddd60 620 split2 = alloc_extent_map();
7014cdb4
JB
621 if (!split || !split2)
622 no_splits = 1;
3b951516 623
890871be 624 write_lock(&em_tree->lock);
39b5637f 625 em = lookup_extent_mapping(em_tree, start, len);
d1310b2e 626 if (!em) {
890871be 627 write_unlock(&em_tree->lock);
a52d9a80 628 break;
d1310b2e 629 }
5b21f2ed 630 flags = em->flags;
5dc562c5 631 gen = em->generation;
5b21f2ed 632 if (skip_pinned && test_bit(EXTENT_FLAG_PINNED, &em->flags)) {
55ef6899 633 if (testend && em->start + em->len >= start + len) {
5b21f2ed 634 free_extent_map(em);
a1ed835e 635 write_unlock(&em_tree->lock);
5b21f2ed
ZY
636 break;
637 }
55ef6899
YZ
638 start = em->start + em->len;
639 if (testend)
5b21f2ed 640 len = start + len - (em->start + em->len);
5b21f2ed 641 free_extent_map(em);
a1ed835e 642 write_unlock(&em_tree->lock);
5b21f2ed
ZY
643 continue;
644 }
c8b97818 645 compressed = test_bit(EXTENT_FLAG_COMPRESSED, &em->flags);
3ce7e67a 646 clear_bit(EXTENT_FLAG_PINNED, &em->flags);
3b277594 647 clear_bit(EXTENT_FLAG_LOGGING, &flags);
09a2a8f9 648 modified = !list_empty(&em->list);
7014cdb4
JB
649 if (no_splits)
650 goto next;
3b951516 651
ee20a983 652 if (em->start < start) {
3b951516
CM
653 split->start = em->start;
654 split->len = start - em->start;
ee20a983
JB
655
656 if (em->block_start < EXTENT_MAP_LAST_BYTE) {
657 split->orig_start = em->orig_start;
658 split->block_start = em->block_start;
659
660 if (compressed)
661 split->block_len = em->block_len;
662 else
663 split->block_len = split->len;
664 split->orig_block_len = max(split->block_len,
665 em->orig_block_len);
666 split->ram_bytes = em->ram_bytes;
667 } else {
668 split->orig_start = split->start;
669 split->block_len = 0;
670 split->block_start = em->block_start;
671 split->orig_block_len = 0;
672 split->ram_bytes = split->len;
673 }
674
5dc562c5 675 split->generation = gen;
3b951516 676 split->bdev = em->bdev;
5b21f2ed 677 split->flags = flags;
261507a0 678 split->compress_type = em->compress_type;
176840b3 679 replace_extent_mapping(em_tree, em, split, modified);
3b951516
CM
680 free_extent_map(split);
681 split = split2;
682 split2 = NULL;
683 }
ee20a983 684 if (testend && em->start + em->len > start + len) {
3b951516
CM
685 u64 diff = start + len - em->start;
686
687 split->start = start + len;
688 split->len = em->start + em->len - (start + len);
689 split->bdev = em->bdev;
5b21f2ed 690 split->flags = flags;
261507a0 691 split->compress_type = em->compress_type;
5dc562c5 692 split->generation = gen;
ee20a983
JB
693
694 if (em->block_start < EXTENT_MAP_LAST_BYTE) {
695 split->orig_block_len = max(em->block_len,
b4939680 696 em->orig_block_len);
3b951516 697
ee20a983
JB
698 split->ram_bytes = em->ram_bytes;
699 if (compressed) {
700 split->block_len = em->block_len;
701 split->block_start = em->block_start;
702 split->orig_start = em->orig_start;
703 } else {
704 split->block_len = split->len;
705 split->block_start = em->block_start
706 + diff;
707 split->orig_start = em->orig_start;
708 }
c8b97818 709 } else {
ee20a983
JB
710 split->ram_bytes = split->len;
711 split->orig_start = split->start;
712 split->block_len = 0;
713 split->block_start = em->block_start;
714 split->orig_block_len = 0;
c8b97818 715 }
3b951516 716
176840b3
FM
717 if (extent_map_in_tree(em)) {
718 replace_extent_mapping(em_tree, em, split,
719 modified);
720 } else {
721 ret = add_extent_mapping(em_tree, split,
722 modified);
723 ASSERT(ret == 0); /* Logic error */
724 }
3b951516
CM
725 free_extent_map(split);
726 split = NULL;
727 }
7014cdb4 728next:
176840b3
FM
729 if (extent_map_in_tree(em))
730 remove_extent_mapping(em_tree, em);
890871be 731 write_unlock(&em_tree->lock);
d1310b2e 732
a52d9a80
CM
733 /* once for us */
734 free_extent_map(em);
735 /* once for the tree*/
736 free_extent_map(em);
737 }
3b951516
CM
738 if (split)
739 free_extent_map(split);
740 if (split2)
741 free_extent_map(split2);
a52d9a80
CM
742}
743
39279cc3
CM
744/*
745 * this is very complex, but the basic idea is to drop all extents
746 * in the range start - end. hint_block is filled in with a block number
747 * that would be a good hint to the block allocator for this file.
748 *
749 * If an extent intersects the range but is not entirely inside the range
750 * it is either truncated or split. Anything entirely inside the range
751 * is deleted from the tree.
752 */
5dc562c5
JB
753int __btrfs_drop_extents(struct btrfs_trans_handle *trans,
754 struct btrfs_root *root, struct inode *inode,
755 struct btrfs_path *path, u64 start, u64 end,
1acae57b
FDBM
756 u64 *drop_end, int drop_cache,
757 int replace_extent,
758 u32 extent_item_size,
759 int *key_inserted)
39279cc3 760{
0b246afa 761 struct btrfs_fs_info *fs_info = root->fs_info;
5f39d397 762 struct extent_buffer *leaf;
920bbbfb 763 struct btrfs_file_extent_item *fi;
00f5c795 764 struct btrfs_key key;
920bbbfb 765 struct btrfs_key new_key;
4a0cc7ca 766 u64 ino = btrfs_ino(BTRFS_I(inode));
920bbbfb
YZ
767 u64 search_start = start;
768 u64 disk_bytenr = 0;
769 u64 num_bytes = 0;
770 u64 extent_offset = 0;
771 u64 extent_end = 0;
62fe51c1 772 u64 last_end = start;
920bbbfb
YZ
773 int del_nr = 0;
774 int del_slot = 0;
775 int extent_type;
ccd467d6 776 int recow;
00f5c795 777 int ret;
dc7fdde3 778 int modify_tree = -1;
27cdeb70 779 int update_refs;
c3308f84 780 int found = 0;
1acae57b 781 int leafs_visited = 0;
39279cc3 782
a1ed835e 783 if (drop_cache)
dcdbc059 784 btrfs_drop_extent_cache(BTRFS_I(inode), start, end - 1, 0);
a52d9a80 785
d5f37527 786 if (start >= BTRFS_I(inode)->disk_i_size && !replace_extent)
dc7fdde3
CM
787 modify_tree = 0;
788
27cdeb70 789 update_refs = (test_bit(BTRFS_ROOT_REF_COWS, &root->state) ||
0b246afa 790 root == fs_info->tree_root);
d397712b 791 while (1) {
ccd467d6 792 recow = 0;
33345d01 793 ret = btrfs_lookup_file_extent(trans, root, path, ino,
dc7fdde3 794 search_start, modify_tree);
39279cc3 795 if (ret < 0)
920bbbfb
YZ
796 break;
797 if (ret > 0 && path->slots[0] > 0 && search_start == start) {
798 leaf = path->nodes[0];
799 btrfs_item_key_to_cpu(leaf, &key, path->slots[0] - 1);
33345d01 800 if (key.objectid == ino &&
920bbbfb
YZ
801 key.type == BTRFS_EXTENT_DATA_KEY)
802 path->slots[0]--;
39279cc3 803 }
920bbbfb 804 ret = 0;
1acae57b 805 leafs_visited++;
8c2383c3 806next_slot:
5f39d397 807 leaf = path->nodes[0];
920bbbfb
YZ
808 if (path->slots[0] >= btrfs_header_nritems(leaf)) {
809 BUG_ON(del_nr > 0);
810 ret = btrfs_next_leaf(root, path);
811 if (ret < 0)
812 break;
813 if (ret > 0) {
814 ret = 0;
815 break;
8c2383c3 816 }
1acae57b 817 leafs_visited++;
920bbbfb
YZ
818 leaf = path->nodes[0];
819 recow = 1;
820 }
821
822 btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
aeafbf84
FM
823
824 if (key.objectid > ino)
825 break;
826 if (WARN_ON_ONCE(key.objectid < ino) ||
827 key.type < BTRFS_EXTENT_DATA_KEY) {
828 ASSERT(del_nr == 0);
829 path->slots[0]++;
830 goto next_slot;
831 }
832 if (key.type > BTRFS_EXTENT_DATA_KEY || key.offset >= end)
920bbbfb
YZ
833 break;
834
835 fi = btrfs_item_ptr(leaf, path->slots[0],
836 struct btrfs_file_extent_item);
837 extent_type = btrfs_file_extent_type(leaf, fi);
838
839 if (extent_type == BTRFS_FILE_EXTENT_REG ||
840 extent_type == BTRFS_FILE_EXTENT_PREALLOC) {
841 disk_bytenr = btrfs_file_extent_disk_bytenr(leaf, fi);
842 num_bytes = btrfs_file_extent_disk_num_bytes(leaf, fi);
843 extent_offset = btrfs_file_extent_offset(leaf, fi);
844 extent_end = key.offset +
845 btrfs_file_extent_num_bytes(leaf, fi);
846 } else if (extent_type == BTRFS_FILE_EXTENT_INLINE) {
847 extent_end = key.offset +
514ac8ad
CM
848 btrfs_file_extent_inline_len(leaf,
849 path->slots[0], fi);
8c2383c3 850 } else {
aeafbf84
FM
851 /* can't happen */
852 BUG();
39279cc3
CM
853 }
854
fc19c5e7
FM
855 /*
856 * Don't skip extent items representing 0 byte lengths. They
857 * used to be created (bug) if while punching holes we hit
858 * -ENOSPC condition. So if we find one here, just ensure we
859 * delete it, otherwise we would insert a new file extent item
860 * with the same key (offset) as that 0 bytes length file
861 * extent item in the call to setup_items_for_insert() later
862 * in this function.
863 */
62fe51c1
JB
864 if (extent_end == key.offset && extent_end >= search_start) {
865 last_end = extent_end;
fc19c5e7 866 goto delete_extent_item;
62fe51c1 867 }
fc19c5e7 868
920bbbfb
YZ
869 if (extent_end <= search_start) {
870 path->slots[0]++;
8c2383c3 871 goto next_slot;
39279cc3
CM
872 }
873
c3308f84 874 found = 1;
920bbbfb 875 search_start = max(key.offset, start);
dc7fdde3
CM
876 if (recow || !modify_tree) {
877 modify_tree = -1;
b3b4aa74 878 btrfs_release_path(path);
920bbbfb 879 continue;
39279cc3 880 }
6643558d 881
920bbbfb
YZ
882 /*
883 * | - range to drop - |
884 * | -------- extent -------- |
885 */
886 if (start > key.offset && end < extent_end) {
887 BUG_ON(del_nr > 0);
00fdf13a 888 if (extent_type == BTRFS_FILE_EXTENT_INLINE) {
3f9e3df8 889 ret = -EOPNOTSUPP;
00fdf13a
LB
890 break;
891 }
920bbbfb
YZ
892
893 memcpy(&new_key, &key, sizeof(new_key));
894 new_key.offset = start;
895 ret = btrfs_duplicate_item(trans, root, path,
896 &new_key);
897 if (ret == -EAGAIN) {
b3b4aa74 898 btrfs_release_path(path);
920bbbfb 899 continue;
6643558d 900 }
920bbbfb
YZ
901 if (ret < 0)
902 break;
903
904 leaf = path->nodes[0];
905 fi = btrfs_item_ptr(leaf, path->slots[0] - 1,
906 struct btrfs_file_extent_item);
907 btrfs_set_file_extent_num_bytes(leaf, fi,
908 start - key.offset);
909
910 fi = btrfs_item_ptr(leaf, path->slots[0],
911 struct btrfs_file_extent_item);
912
913 extent_offset += start - key.offset;
914 btrfs_set_file_extent_offset(leaf, fi, extent_offset);
915 btrfs_set_file_extent_num_bytes(leaf, fi,
916 extent_end - start);
917 btrfs_mark_buffer_dirty(leaf);
918
5dc562c5 919 if (update_refs && disk_bytenr > 0) {
84f7d8e6 920 ret = btrfs_inc_extent_ref(trans, root,
920bbbfb
YZ
921 disk_bytenr, num_bytes, 0,
922 root->root_key.objectid,
923 new_key.objectid,
b06c4bf5 924 start - extent_offset);
79787eaa 925 BUG_ON(ret); /* -ENOMEM */
771ed689 926 }
920bbbfb 927 key.offset = start;
6643558d 928 }
62fe51c1
JB
929 /*
930 * From here on out we will have actually dropped something, so
931 * last_end can be updated.
932 */
933 last_end = extent_end;
934
920bbbfb
YZ
935 /*
936 * | ---- range to drop ----- |
937 * | -------- extent -------- |
938 */
939 if (start <= key.offset && end < extent_end) {
00fdf13a 940 if (extent_type == BTRFS_FILE_EXTENT_INLINE) {
3f9e3df8 941 ret = -EOPNOTSUPP;
00fdf13a
LB
942 break;
943 }
6643558d 944
920bbbfb
YZ
945 memcpy(&new_key, &key, sizeof(new_key));
946 new_key.offset = end;
0b246afa 947 btrfs_set_item_key_safe(fs_info, path, &new_key);
6643558d 948
920bbbfb
YZ
949 extent_offset += end - key.offset;
950 btrfs_set_file_extent_offset(leaf, fi, extent_offset);
951 btrfs_set_file_extent_num_bytes(leaf, fi,
952 extent_end - end);
953 btrfs_mark_buffer_dirty(leaf);
2671485d 954 if (update_refs && disk_bytenr > 0)
920bbbfb 955 inode_sub_bytes(inode, end - key.offset);
920bbbfb 956 break;
39279cc3 957 }
771ed689 958
920bbbfb
YZ
959 search_start = extent_end;
960 /*
961 * | ---- range to drop ----- |
962 * | -------- extent -------- |
963 */
964 if (start > key.offset && end >= extent_end) {
965 BUG_ON(del_nr > 0);
00fdf13a 966 if (extent_type == BTRFS_FILE_EXTENT_INLINE) {
3f9e3df8 967 ret = -EOPNOTSUPP;
00fdf13a
LB
968 break;
969 }
8c2383c3 970
920bbbfb
YZ
971 btrfs_set_file_extent_num_bytes(leaf, fi,
972 start - key.offset);
973 btrfs_mark_buffer_dirty(leaf);
2671485d 974 if (update_refs && disk_bytenr > 0)
920bbbfb 975 inode_sub_bytes(inode, extent_end - start);
920bbbfb
YZ
976 if (end == extent_end)
977 break;
c8b97818 978
920bbbfb
YZ
979 path->slots[0]++;
980 goto next_slot;
31840ae1
ZY
981 }
982
920bbbfb
YZ
983 /*
984 * | ---- range to drop ----- |
985 * | ------ extent ------ |
986 */
987 if (start <= key.offset && end >= extent_end) {
fc19c5e7 988delete_extent_item:
920bbbfb
YZ
989 if (del_nr == 0) {
990 del_slot = path->slots[0];
991 del_nr = 1;
992 } else {
993 BUG_ON(del_slot + del_nr != path->slots[0]);
994 del_nr++;
995 }
31840ae1 996
5dc562c5
JB
997 if (update_refs &&
998 extent_type == BTRFS_FILE_EXTENT_INLINE) {
a76a3cd4 999 inode_sub_bytes(inode,
920bbbfb
YZ
1000 extent_end - key.offset);
1001 extent_end = ALIGN(extent_end,
0b246afa 1002 fs_info->sectorsize);
5dc562c5 1003 } else if (update_refs && disk_bytenr > 0) {
84f7d8e6 1004 ret = btrfs_free_extent(trans, root,
920bbbfb
YZ
1005 disk_bytenr, num_bytes, 0,
1006 root->root_key.objectid,
5d4f98a2 1007 key.objectid, key.offset -
b06c4bf5 1008 extent_offset);
79787eaa 1009 BUG_ON(ret); /* -ENOMEM */
920bbbfb
YZ
1010 inode_sub_bytes(inode,
1011 extent_end - key.offset);
31840ae1 1012 }
31840ae1 1013
920bbbfb
YZ
1014 if (end == extent_end)
1015 break;
1016
1017 if (path->slots[0] + 1 < btrfs_header_nritems(leaf)) {
1018 path->slots[0]++;
1019 goto next_slot;
1020 }
1021
1022 ret = btrfs_del_items(trans, root, path, del_slot,
1023 del_nr);
79787eaa 1024 if (ret) {
66642832 1025 btrfs_abort_transaction(trans, ret);
5dc562c5 1026 break;
79787eaa 1027 }
920bbbfb
YZ
1028
1029 del_nr = 0;
1030 del_slot = 0;
1031
b3b4aa74 1032 btrfs_release_path(path);
920bbbfb 1033 continue;
39279cc3 1034 }
920bbbfb
YZ
1035
1036 BUG_ON(1);
39279cc3 1037 }
920bbbfb 1038
79787eaa 1039 if (!ret && del_nr > 0) {
1acae57b
FDBM
1040 /*
1041 * Set path->slots[0] to first slot, so that after the delete
1042 * if items are move off from our leaf to its immediate left or
1043 * right neighbor leafs, we end up with a correct and adjusted
d5f37527 1044 * path->slots[0] for our insertion (if replace_extent != 0).
1acae57b
FDBM
1045 */
1046 path->slots[0] = del_slot;
920bbbfb 1047 ret = btrfs_del_items(trans, root, path, del_slot, del_nr);
79787eaa 1048 if (ret)
66642832 1049 btrfs_abort_transaction(trans, ret);
d5f37527 1050 }
1acae57b 1051
d5f37527
FDBM
1052 leaf = path->nodes[0];
1053 /*
1054 * If btrfs_del_items() was called, it might have deleted a leaf, in
1055 * which case it unlocked our path, so check path->locks[0] matches a
1056 * write lock.
1057 */
1058 if (!ret && replace_extent && leafs_visited == 1 &&
1059 (path->locks[0] == BTRFS_WRITE_LOCK_BLOCKING ||
1060 path->locks[0] == BTRFS_WRITE_LOCK) &&
2ff7e61e 1061 btrfs_leaf_free_space(fs_info, leaf) >=
d5f37527
FDBM
1062 sizeof(struct btrfs_item) + extent_item_size) {
1063
1064 key.objectid = ino;
1065 key.type = BTRFS_EXTENT_DATA_KEY;
1066 key.offset = start;
1067 if (!del_nr && path->slots[0] < btrfs_header_nritems(leaf)) {
1068 struct btrfs_key slot_key;
1069
1070 btrfs_item_key_to_cpu(leaf, &slot_key, path->slots[0]);
1071 if (btrfs_comp_cpu_keys(&key, &slot_key) > 0)
1072 path->slots[0]++;
1acae57b 1073 }
d5f37527
FDBM
1074 setup_items_for_insert(root, path, &key,
1075 &extent_item_size,
1076 extent_item_size,
1077 sizeof(struct btrfs_item) +
1078 extent_item_size, 1);
1079 *key_inserted = 1;
6643558d 1080 }
920bbbfb 1081
1acae57b
FDBM
1082 if (!replace_extent || !(*key_inserted))
1083 btrfs_release_path(path);
2aaa6655 1084 if (drop_end)
62fe51c1 1085 *drop_end = found ? min(end, last_end) : end;
5dc562c5
JB
1086 return ret;
1087}
1088
1089int btrfs_drop_extents(struct btrfs_trans_handle *trans,
1090 struct btrfs_root *root, struct inode *inode, u64 start,
2671485d 1091 u64 end, int drop_cache)
5dc562c5
JB
1092{
1093 struct btrfs_path *path;
1094 int ret;
1095
1096 path = btrfs_alloc_path();
1097 if (!path)
1098 return -ENOMEM;
2aaa6655 1099 ret = __btrfs_drop_extents(trans, root, inode, path, start, end, NULL,
1acae57b 1100 drop_cache, 0, 0, NULL);
920bbbfb 1101 btrfs_free_path(path);
39279cc3
CM
1102 return ret;
1103}
1104
d899e052 1105static int extent_mergeable(struct extent_buffer *leaf, int slot,
6c7d54ac
YZ
1106 u64 objectid, u64 bytenr, u64 orig_offset,
1107 u64 *start, u64 *end)
d899e052
YZ
1108{
1109 struct btrfs_file_extent_item *fi;
1110 struct btrfs_key key;
1111 u64 extent_end;
1112
1113 if (slot < 0 || slot >= btrfs_header_nritems(leaf))
1114 return 0;
1115
1116 btrfs_item_key_to_cpu(leaf, &key, slot);
1117 if (key.objectid != objectid || key.type != BTRFS_EXTENT_DATA_KEY)
1118 return 0;
1119
1120 fi = btrfs_item_ptr(leaf, slot, struct btrfs_file_extent_item);
1121 if (btrfs_file_extent_type(leaf, fi) != BTRFS_FILE_EXTENT_REG ||
1122 btrfs_file_extent_disk_bytenr(leaf, fi) != bytenr ||
6c7d54ac 1123 btrfs_file_extent_offset(leaf, fi) != key.offset - orig_offset ||
d899e052
YZ
1124 btrfs_file_extent_compression(leaf, fi) ||
1125 btrfs_file_extent_encryption(leaf, fi) ||
1126 btrfs_file_extent_other_encoding(leaf, fi))
1127 return 0;
1128
1129 extent_end = key.offset + btrfs_file_extent_num_bytes(leaf, fi);
1130 if ((*start && *start != key.offset) || (*end && *end != extent_end))
1131 return 0;
1132
1133 *start = key.offset;
1134 *end = extent_end;
1135 return 1;
1136}
1137
1138/*
1139 * Mark extent in the range start - end as written.
1140 *
1141 * This changes extent type from 'pre-allocated' to 'regular'. If only
1142 * part of extent is marked as written, the extent will be split into
1143 * two or three.
1144 */
1145int btrfs_mark_extent_written(struct btrfs_trans_handle *trans,
7a6d7067 1146 struct btrfs_inode *inode, u64 start, u64 end)
d899e052 1147{
7a6d7067
NB
1148 struct btrfs_fs_info *fs_info = btrfs_sb(inode->vfs_inode.i_sb);
1149 struct btrfs_root *root = inode->root;
d899e052
YZ
1150 struct extent_buffer *leaf;
1151 struct btrfs_path *path;
1152 struct btrfs_file_extent_item *fi;
1153 struct btrfs_key key;
920bbbfb 1154 struct btrfs_key new_key;
d899e052
YZ
1155 u64 bytenr;
1156 u64 num_bytes;
1157 u64 extent_end;
5d4f98a2 1158 u64 orig_offset;
d899e052
YZ
1159 u64 other_start;
1160 u64 other_end;
920bbbfb
YZ
1161 u64 split;
1162 int del_nr = 0;
1163 int del_slot = 0;
6c7d54ac 1164 int recow;
d899e052 1165 int ret;
7a6d7067 1166 u64 ino = btrfs_ino(inode);
d899e052 1167
d899e052 1168 path = btrfs_alloc_path();
d8926bb3
MF
1169 if (!path)
1170 return -ENOMEM;
d899e052 1171again:
6c7d54ac 1172 recow = 0;
920bbbfb 1173 split = start;
33345d01 1174 key.objectid = ino;
d899e052 1175 key.type = BTRFS_EXTENT_DATA_KEY;
920bbbfb 1176 key.offset = split;
d899e052
YZ
1177
1178 ret = btrfs_search_slot(trans, root, &key, path, -1, 1);
41415730
JB
1179 if (ret < 0)
1180 goto out;
d899e052
YZ
1181 if (ret > 0 && path->slots[0] > 0)
1182 path->slots[0]--;
1183
1184 leaf = path->nodes[0];
1185 btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
9c8e63db
JB
1186 if (key.objectid != ino ||
1187 key.type != BTRFS_EXTENT_DATA_KEY) {
1188 ret = -EINVAL;
1189 btrfs_abort_transaction(trans, ret);
1190 goto out;
1191 }
d899e052
YZ
1192 fi = btrfs_item_ptr(leaf, path->slots[0],
1193 struct btrfs_file_extent_item);
9c8e63db
JB
1194 if (btrfs_file_extent_type(leaf, fi) != BTRFS_FILE_EXTENT_PREALLOC) {
1195 ret = -EINVAL;
1196 btrfs_abort_transaction(trans, ret);
1197 goto out;
1198 }
d899e052 1199 extent_end = key.offset + btrfs_file_extent_num_bytes(leaf, fi);
9c8e63db
JB
1200 if (key.offset > start || extent_end < end) {
1201 ret = -EINVAL;
1202 btrfs_abort_transaction(trans, ret);
1203 goto out;
1204 }
d899e052
YZ
1205
1206 bytenr = btrfs_file_extent_disk_bytenr(leaf, fi);
1207 num_bytes = btrfs_file_extent_disk_num_bytes(leaf, fi);
5d4f98a2 1208 orig_offset = key.offset - btrfs_file_extent_offset(leaf, fi);
6c7d54ac
YZ
1209 memcpy(&new_key, &key, sizeof(new_key));
1210
1211 if (start == key.offset && end < extent_end) {
1212 other_start = 0;
1213 other_end = start;
1214 if (extent_mergeable(leaf, path->slots[0] - 1,
33345d01 1215 ino, bytenr, orig_offset,
6c7d54ac
YZ
1216 &other_start, &other_end)) {
1217 new_key.offset = end;
0b246afa 1218 btrfs_set_item_key_safe(fs_info, path, &new_key);
6c7d54ac
YZ
1219 fi = btrfs_item_ptr(leaf, path->slots[0],
1220 struct btrfs_file_extent_item);
224ecce5
JB
1221 btrfs_set_file_extent_generation(leaf, fi,
1222 trans->transid);
6c7d54ac
YZ
1223 btrfs_set_file_extent_num_bytes(leaf, fi,
1224 extent_end - end);
1225 btrfs_set_file_extent_offset(leaf, fi,
1226 end - orig_offset);
1227 fi = btrfs_item_ptr(leaf, path->slots[0] - 1,
1228 struct btrfs_file_extent_item);
224ecce5
JB
1229 btrfs_set_file_extent_generation(leaf, fi,
1230 trans->transid);
6c7d54ac
YZ
1231 btrfs_set_file_extent_num_bytes(leaf, fi,
1232 end - other_start);
1233 btrfs_mark_buffer_dirty(leaf);
1234 goto out;
1235 }
1236 }
1237
1238 if (start > key.offset && end == extent_end) {
1239 other_start = end;
1240 other_end = 0;
1241 if (extent_mergeable(leaf, path->slots[0] + 1,
33345d01 1242 ino, bytenr, orig_offset,
6c7d54ac
YZ
1243 &other_start, &other_end)) {
1244 fi = btrfs_item_ptr(leaf, path->slots[0],
1245 struct btrfs_file_extent_item);
1246 btrfs_set_file_extent_num_bytes(leaf, fi,
1247 start - key.offset);
224ecce5
JB
1248 btrfs_set_file_extent_generation(leaf, fi,
1249 trans->transid);
6c7d54ac
YZ
1250 path->slots[0]++;
1251 new_key.offset = start;
0b246afa 1252 btrfs_set_item_key_safe(fs_info, path, &new_key);
6c7d54ac
YZ
1253
1254 fi = btrfs_item_ptr(leaf, path->slots[0],
1255 struct btrfs_file_extent_item);
224ecce5
JB
1256 btrfs_set_file_extent_generation(leaf, fi,
1257 trans->transid);
6c7d54ac
YZ
1258 btrfs_set_file_extent_num_bytes(leaf, fi,
1259 other_end - start);
1260 btrfs_set_file_extent_offset(leaf, fi,
1261 start - orig_offset);
1262 btrfs_mark_buffer_dirty(leaf);
1263 goto out;
1264 }
1265 }
d899e052 1266
920bbbfb
YZ
1267 while (start > key.offset || end < extent_end) {
1268 if (key.offset == start)
1269 split = end;
1270
920bbbfb
YZ
1271 new_key.offset = split;
1272 ret = btrfs_duplicate_item(trans, root, path, &new_key);
1273 if (ret == -EAGAIN) {
b3b4aa74 1274 btrfs_release_path(path);
920bbbfb 1275 goto again;
d899e052 1276 }
79787eaa 1277 if (ret < 0) {
66642832 1278 btrfs_abort_transaction(trans, ret);
79787eaa
JM
1279 goto out;
1280 }
d899e052 1281
920bbbfb
YZ
1282 leaf = path->nodes[0];
1283 fi = btrfs_item_ptr(leaf, path->slots[0] - 1,
d899e052 1284 struct btrfs_file_extent_item);
224ecce5 1285 btrfs_set_file_extent_generation(leaf, fi, trans->transid);
d899e052 1286 btrfs_set_file_extent_num_bytes(leaf, fi,
920bbbfb
YZ
1287 split - key.offset);
1288
1289 fi = btrfs_item_ptr(leaf, path->slots[0],
1290 struct btrfs_file_extent_item);
1291
224ecce5 1292 btrfs_set_file_extent_generation(leaf, fi, trans->transid);
920bbbfb
YZ
1293 btrfs_set_file_extent_offset(leaf, fi, split - orig_offset);
1294 btrfs_set_file_extent_num_bytes(leaf, fi,
1295 extent_end - split);
d899e052
YZ
1296 btrfs_mark_buffer_dirty(leaf);
1297
84f7d8e6 1298 ret = btrfs_inc_extent_ref(trans, root, bytenr, num_bytes,
2ff7e61e 1299 0, root->root_key.objectid,
b06c4bf5 1300 ino, orig_offset);
9c8e63db
JB
1301 if (ret) {
1302 btrfs_abort_transaction(trans, ret);
1303 goto out;
1304 }
d899e052 1305
920bbbfb
YZ
1306 if (split == start) {
1307 key.offset = start;
1308 } else {
9c8e63db
JB
1309 if (start != key.offset) {
1310 ret = -EINVAL;
1311 btrfs_abort_transaction(trans, ret);
1312 goto out;
1313 }
d899e052 1314 path->slots[0]--;
920bbbfb 1315 extent_end = end;
d899e052 1316 }
6c7d54ac 1317 recow = 1;
d899e052
YZ
1318 }
1319
920bbbfb
YZ
1320 other_start = end;
1321 other_end = 0;
6c7d54ac 1322 if (extent_mergeable(leaf, path->slots[0] + 1,
33345d01 1323 ino, bytenr, orig_offset,
6c7d54ac
YZ
1324 &other_start, &other_end)) {
1325 if (recow) {
b3b4aa74 1326 btrfs_release_path(path);
6c7d54ac
YZ
1327 goto again;
1328 }
920bbbfb
YZ
1329 extent_end = other_end;
1330 del_slot = path->slots[0] + 1;
1331 del_nr++;
84f7d8e6 1332 ret = btrfs_free_extent(trans, root, bytenr, num_bytes,
920bbbfb 1333 0, root->root_key.objectid,
b06c4bf5 1334 ino, orig_offset);
9c8e63db
JB
1335 if (ret) {
1336 btrfs_abort_transaction(trans, ret);
1337 goto out;
1338 }
d899e052 1339 }
920bbbfb
YZ
1340 other_start = 0;
1341 other_end = start;
6c7d54ac 1342 if (extent_mergeable(leaf, path->slots[0] - 1,
33345d01 1343 ino, bytenr, orig_offset,
6c7d54ac
YZ
1344 &other_start, &other_end)) {
1345 if (recow) {
b3b4aa74 1346 btrfs_release_path(path);
6c7d54ac
YZ
1347 goto again;
1348 }
920bbbfb
YZ
1349 key.offset = other_start;
1350 del_slot = path->slots[0];
1351 del_nr++;
84f7d8e6 1352 ret = btrfs_free_extent(trans, root, bytenr, num_bytes,
920bbbfb 1353 0, root->root_key.objectid,
b06c4bf5 1354 ino, orig_offset);
9c8e63db
JB
1355 if (ret) {
1356 btrfs_abort_transaction(trans, ret);
1357 goto out;
1358 }
920bbbfb
YZ
1359 }
1360 if (del_nr == 0) {
3f6fae95
SL
1361 fi = btrfs_item_ptr(leaf, path->slots[0],
1362 struct btrfs_file_extent_item);
920bbbfb
YZ
1363 btrfs_set_file_extent_type(leaf, fi,
1364 BTRFS_FILE_EXTENT_REG);
224ecce5 1365 btrfs_set_file_extent_generation(leaf, fi, trans->transid);
920bbbfb 1366 btrfs_mark_buffer_dirty(leaf);
6c7d54ac 1367 } else {
3f6fae95
SL
1368 fi = btrfs_item_ptr(leaf, del_slot - 1,
1369 struct btrfs_file_extent_item);
6c7d54ac
YZ
1370 btrfs_set_file_extent_type(leaf, fi,
1371 BTRFS_FILE_EXTENT_REG);
224ecce5 1372 btrfs_set_file_extent_generation(leaf, fi, trans->transid);
6c7d54ac
YZ
1373 btrfs_set_file_extent_num_bytes(leaf, fi,
1374 extent_end - key.offset);
1375 btrfs_mark_buffer_dirty(leaf);
920bbbfb 1376
6c7d54ac 1377 ret = btrfs_del_items(trans, root, path, del_slot, del_nr);
79787eaa 1378 if (ret < 0) {
66642832 1379 btrfs_abort_transaction(trans, ret);
79787eaa
JM
1380 goto out;
1381 }
6c7d54ac 1382 }
920bbbfb 1383out:
d899e052
YZ
1384 btrfs_free_path(path);
1385 return 0;
1386}
1387
b1bf862e
CM
1388/*
1389 * on error we return an unlocked page and the error value
1390 * on success we return a locked page and 0
1391 */
bb1591b4
CM
1392static int prepare_uptodate_page(struct inode *inode,
1393 struct page *page, u64 pos,
b6316429 1394 bool force_uptodate)
b1bf862e
CM
1395{
1396 int ret = 0;
1397
09cbfeaf 1398 if (((pos & (PAGE_SIZE - 1)) || force_uptodate) &&
b6316429 1399 !PageUptodate(page)) {
b1bf862e
CM
1400 ret = btrfs_readpage(NULL, page);
1401 if (ret)
1402 return ret;
1403 lock_page(page);
1404 if (!PageUptodate(page)) {
1405 unlock_page(page);
1406 return -EIO;
1407 }
bb1591b4
CM
1408 if (page->mapping != inode->i_mapping) {
1409 unlock_page(page);
1410 return -EAGAIN;
1411 }
b1bf862e
CM
1412 }
1413 return 0;
1414}
1415
39279cc3 1416/*
376cc685 1417 * this just gets pages into the page cache and locks them down.
39279cc3 1418 */
b37392ea
MX
1419static noinline int prepare_pages(struct inode *inode, struct page **pages,
1420 size_t num_pages, loff_t pos,
1421 size_t write_bytes, bool force_uptodate)
39279cc3
CM
1422{
1423 int i;
09cbfeaf 1424 unsigned long index = pos >> PAGE_SHIFT;
3b16a4e3 1425 gfp_t mask = btrfs_alloc_write_mask(inode->i_mapping);
fc28b62d 1426 int err = 0;
376cc685 1427 int faili;
8c2383c3 1428
39279cc3 1429 for (i = 0; i < num_pages; i++) {
bb1591b4 1430again:
a94733d0 1431 pages[i] = find_or_create_page(inode->i_mapping, index + i,
e3a41a5b 1432 mask | __GFP_WRITE);
39279cc3 1433 if (!pages[i]) {
b1bf862e
CM
1434 faili = i - 1;
1435 err = -ENOMEM;
1436 goto fail;
1437 }
1438
1439 if (i == 0)
bb1591b4 1440 err = prepare_uptodate_page(inode, pages[i], pos,
b6316429 1441 force_uptodate);
bb1591b4
CM
1442 if (!err && i == num_pages - 1)
1443 err = prepare_uptodate_page(inode, pages[i],
b6316429 1444 pos + write_bytes, false);
b1bf862e 1445 if (err) {
09cbfeaf 1446 put_page(pages[i]);
bb1591b4
CM
1447 if (err == -EAGAIN) {
1448 err = 0;
1449 goto again;
1450 }
b1bf862e
CM
1451 faili = i - 1;
1452 goto fail;
39279cc3 1453 }
ccd467d6 1454 wait_on_page_writeback(pages[i]);
39279cc3 1455 }
376cc685
MX
1456
1457 return 0;
1458fail:
1459 while (faili >= 0) {
1460 unlock_page(pages[faili]);
09cbfeaf 1461 put_page(pages[faili]);
376cc685
MX
1462 faili--;
1463 }
1464 return err;
1465
1466}
1467
1468/*
1469 * This function locks the extent and properly waits for data=ordered extents
1470 * to finish before allowing the pages to be modified if need.
1471 *
1472 * The return value:
1473 * 1 - the extent is locked
1474 * 0 - the extent is not locked, and everything is OK
1475 * -EAGAIN - need re-prepare the pages
1476 * the other < 0 number - Something wrong happens
1477 */
1478static noinline int
2cff578c 1479lock_and_cleanup_extent_if_need(struct btrfs_inode *inode, struct page **pages,
376cc685 1480 size_t num_pages, loff_t pos,
2e78c927 1481 size_t write_bytes,
376cc685
MX
1482 u64 *lockstart, u64 *lockend,
1483 struct extent_state **cached_state)
1484{
2cff578c 1485 struct btrfs_fs_info *fs_info = btrfs_sb(inode->vfs_inode.i_sb);
376cc685
MX
1486 u64 start_pos;
1487 u64 last_pos;
1488 int i;
1489 int ret = 0;
1490
0b246afa 1491 start_pos = round_down(pos, fs_info->sectorsize);
2e78c927 1492 last_pos = start_pos
da17066c 1493 + round_up(pos + write_bytes - start_pos,
0b246afa 1494 fs_info->sectorsize) - 1;
376cc685 1495
e3b8a485 1496 if (start_pos < inode->vfs_inode.i_size) {
e6dcd2dc 1497 struct btrfs_ordered_extent *ordered;
a7e3b975 1498
2cff578c
NB
1499 lock_extent_bits(&inode->io_tree, start_pos, last_pos,
1500 cached_state);
b88935bf
MX
1501 ordered = btrfs_lookup_ordered_range(inode, start_pos,
1502 last_pos - start_pos + 1);
e6dcd2dc
CM
1503 if (ordered &&
1504 ordered->file_offset + ordered->len > start_pos &&
376cc685 1505 ordered->file_offset <= last_pos) {
2cff578c
NB
1506 unlock_extent_cached(&inode->io_tree, start_pos,
1507 last_pos, cached_state, GFP_NOFS);
e6dcd2dc
CM
1508 for (i = 0; i < num_pages; i++) {
1509 unlock_page(pages[i]);
09cbfeaf 1510 put_page(pages[i]);
e6dcd2dc 1511 }
2cff578c
NB
1512 btrfs_start_ordered_extent(&inode->vfs_inode,
1513 ordered, 1);
b88935bf
MX
1514 btrfs_put_ordered_extent(ordered);
1515 return -EAGAIN;
e6dcd2dc
CM
1516 }
1517 if (ordered)
1518 btrfs_put_ordered_extent(ordered);
e3b8a485
FM
1519 clear_extent_bit(&inode->io_tree, start_pos, last_pos,
1520 EXTENT_DIRTY | EXTENT_DELALLOC |
1521 EXTENT_DO_ACCOUNTING | EXTENT_DEFRAG,
1522 0, 0, cached_state, GFP_NOFS);
376cc685
MX
1523 *lockstart = start_pos;
1524 *lockend = last_pos;
1525 ret = 1;
0762704b 1526 }
376cc685 1527
e6dcd2dc 1528 for (i = 0; i < num_pages; i++) {
32c7f202
WF
1529 if (clear_page_dirty_for_io(pages[i]))
1530 account_page_redirty(pages[i]);
e6dcd2dc
CM
1531 set_page_extent_mapped(pages[i]);
1532 WARN_ON(!PageLocked(pages[i]));
1533 }
b1bf862e 1534
376cc685 1535 return ret;
39279cc3
CM
1536}
1537
85b7ab67 1538static noinline int check_can_nocow(struct btrfs_inode *inode, loff_t pos,
7ee9e440
JB
1539 size_t *write_bytes)
1540{
85b7ab67
NB
1541 struct btrfs_fs_info *fs_info = btrfs_sb(inode->vfs_inode.i_sb);
1542 struct btrfs_root *root = inode->root;
7ee9e440
JB
1543 struct btrfs_ordered_extent *ordered;
1544 u64 lockstart, lockend;
1545 u64 num_bytes;
1546 int ret;
1547
ea14b57f 1548 ret = btrfs_start_write_no_snapshotting(root);
8257b2dc
MX
1549 if (!ret)
1550 return -ENOSPC;
1551
0b246afa 1552 lockstart = round_down(pos, fs_info->sectorsize);
da17066c 1553 lockend = round_up(pos + *write_bytes,
0b246afa 1554 fs_info->sectorsize) - 1;
7ee9e440
JB
1555
1556 while (1) {
85b7ab67 1557 lock_extent(&inode->io_tree, lockstart, lockend);
7ee9e440
JB
1558 ordered = btrfs_lookup_ordered_range(inode, lockstart,
1559 lockend - lockstart + 1);
1560 if (!ordered) {
1561 break;
1562 }
85b7ab67
NB
1563 unlock_extent(&inode->io_tree, lockstart, lockend);
1564 btrfs_start_ordered_extent(&inode->vfs_inode, ordered, 1);
7ee9e440
JB
1565 btrfs_put_ordered_extent(ordered);
1566 }
1567
7ee9e440 1568 num_bytes = lockend - lockstart + 1;
85b7ab67
NB
1569 ret = can_nocow_extent(&inode->vfs_inode, lockstart, &num_bytes,
1570 NULL, NULL, NULL);
7ee9e440
JB
1571 if (ret <= 0) {
1572 ret = 0;
ea14b57f 1573 btrfs_end_write_no_snapshotting(root);
7ee9e440 1574 } else {
c933956d
MX
1575 *write_bytes = min_t(size_t, *write_bytes ,
1576 num_bytes - pos + lockstart);
7ee9e440
JB
1577 }
1578
85b7ab67 1579 unlock_extent(&inode->io_tree, lockstart, lockend);
7ee9e440
JB
1580
1581 return ret;
1582}
1583
d0215f3e
JB
1584static noinline ssize_t __btrfs_buffered_write(struct file *file,
1585 struct iov_iter *i,
1586 loff_t pos)
4b46fce2 1587{
496ad9aa 1588 struct inode *inode = file_inode(file);
0b246afa 1589 struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
11c65dcc 1590 struct btrfs_root *root = BTRFS_I(inode)->root;
11c65dcc 1591 struct page **pages = NULL;
376cc685 1592 struct extent_state *cached_state = NULL;
364ecf36 1593 struct extent_changeset *data_reserved = NULL;
7ee9e440 1594 u64 release_bytes = 0;
376cc685
MX
1595 u64 lockstart;
1596 u64 lockend;
d0215f3e
JB
1597 size_t num_written = 0;
1598 int nrptrs;
c9149235 1599 int ret = 0;
7ee9e440 1600 bool only_release_metadata = false;
b6316429 1601 bool force_page_uptodate = false;
4b46fce2 1602
09cbfeaf
KS
1603 nrptrs = min(DIV_ROUND_UP(iov_iter_count(i), PAGE_SIZE),
1604 PAGE_SIZE / (sizeof(struct page *)));
142349f5
WF
1605 nrptrs = min(nrptrs, current->nr_dirtied_pause - current->nr_dirtied);
1606 nrptrs = max(nrptrs, 8);
31e818fe 1607 pages = kmalloc_array(nrptrs, sizeof(struct page *), GFP_KERNEL);
d0215f3e
JB
1608 if (!pages)
1609 return -ENOMEM;
ab93dbec 1610
d0215f3e 1611 while (iov_iter_count(i) > 0) {
09cbfeaf 1612 size_t offset = pos & (PAGE_SIZE - 1);
2e78c927 1613 size_t sector_offset;
d0215f3e 1614 size_t write_bytes = min(iov_iter_count(i),
09cbfeaf 1615 nrptrs * (size_t)PAGE_SIZE -
8c2383c3 1616 offset);
ed6078f7 1617 size_t num_pages = DIV_ROUND_UP(write_bytes + offset,
09cbfeaf 1618 PAGE_SIZE);
7ee9e440 1619 size_t reserve_bytes;
d0215f3e
JB
1620 size_t dirty_pages;
1621 size_t copied;
2e78c927
CR
1622 size_t dirty_sectors;
1623 size_t num_sectors;
79f015f2 1624 int extents_locked;
39279cc3 1625
8c2383c3 1626 WARN_ON(num_pages > nrptrs);
1832a6d5 1627
914ee295
XZ
1628 /*
1629 * Fault pages before locking them in prepare_pages
1630 * to avoid recursive lock
1631 */
d0215f3e 1632 if (unlikely(iov_iter_fault_in_readable(i, write_bytes))) {
914ee295 1633 ret = -EFAULT;
d0215f3e 1634 break;
914ee295
XZ
1635 }
1636
da17066c 1637 sector_offset = pos & (fs_info->sectorsize - 1);
2e78c927 1638 reserve_bytes = round_up(write_bytes + sector_offset,
da17066c 1639 fs_info->sectorsize);
d9d8b2a5 1640
364ecf36
QW
1641 extent_changeset_release(data_reserved);
1642 ret = btrfs_check_data_free_space(inode, &data_reserved, pos,
1643 write_bytes);
c6887cd1
JB
1644 if (ret < 0) {
1645 if ((BTRFS_I(inode)->flags & (BTRFS_INODE_NODATACOW |
1646 BTRFS_INODE_PREALLOC)) &&
85b7ab67
NB
1647 check_can_nocow(BTRFS_I(inode), pos,
1648 &write_bytes) > 0) {
c6887cd1
JB
1649 /*
1650 * For nodata cow case, no need to reserve
1651 * data space.
1652 */
1653 only_release_metadata = true;
1654 /*
1655 * our prealloc extent may be smaller than
1656 * write_bytes, so scale down.
1657 */
1658 num_pages = DIV_ROUND_UP(write_bytes + offset,
1659 PAGE_SIZE);
1660 reserve_bytes = round_up(write_bytes +
1661 sector_offset,
da17066c 1662 fs_info->sectorsize);
c6887cd1
JB
1663 } else {
1664 break;
1665 }
1666 }
1832a6d5 1667
8b62f87b 1668 WARN_ON(reserve_bytes == 0);
9f3db423
NB
1669 ret = btrfs_delalloc_reserve_metadata(BTRFS_I(inode),
1670 reserve_bytes);
7ee9e440
JB
1671 if (ret) {
1672 if (!only_release_metadata)
bc42bda2
QW
1673 btrfs_free_reserved_data_space(inode,
1674 data_reserved, pos,
1675 write_bytes);
8257b2dc 1676 else
ea14b57f 1677 btrfs_end_write_no_snapshotting(root);
7ee9e440
JB
1678 break;
1679 }
1680
1681 release_bytes = reserve_bytes;
376cc685 1682again:
4a64001f
JB
1683 /*
1684 * This is going to setup the pages array with the number of
1685 * pages we want, so we don't really need to worry about the
1686 * contents of pages from loop to loop
1687 */
b37392ea
MX
1688 ret = prepare_pages(inode, pages, num_pages,
1689 pos, write_bytes,
b6316429 1690 force_page_uptodate);
8b62f87b
JB
1691 if (ret) {
1692 btrfs_delalloc_release_extents(BTRFS_I(inode),
1693 reserve_bytes);
d0215f3e 1694 break;
8b62f87b 1695 }
39279cc3 1696
79f015f2
GR
1697 extents_locked = lock_and_cleanup_extent_if_need(
1698 BTRFS_I(inode), pages,
2cff578c
NB
1699 num_pages, pos, write_bytes, &lockstart,
1700 &lockend, &cached_state);
79f015f2
GR
1701 if (extents_locked < 0) {
1702 if (extents_locked == -EAGAIN)
376cc685 1703 goto again;
8b62f87b
JB
1704 btrfs_delalloc_release_extents(BTRFS_I(inode),
1705 reserve_bytes);
79f015f2 1706 ret = extents_locked;
376cc685 1707 break;
376cc685
MX
1708 }
1709
ee22f0c4 1710 copied = btrfs_copy_from_user(pos, write_bytes, pages, i);
b1bf862e 1711
0b246afa 1712 num_sectors = BTRFS_BYTES_TO_BLKS(fs_info, reserve_bytes);
56244ef1 1713 dirty_sectors = round_up(copied + sector_offset,
0b246afa
JM
1714 fs_info->sectorsize);
1715 dirty_sectors = BTRFS_BYTES_TO_BLKS(fs_info, dirty_sectors);
56244ef1 1716
b1bf862e
CM
1717 /*
1718 * if we have trouble faulting in the pages, fall
1719 * back to one page at a time
1720 */
1721 if (copied < write_bytes)
1722 nrptrs = 1;
1723
b6316429
JB
1724 if (copied == 0) {
1725 force_page_uptodate = true;
56244ef1 1726 dirty_sectors = 0;
b1bf862e 1727 dirty_pages = 0;
b6316429
JB
1728 } else {
1729 force_page_uptodate = false;
ed6078f7 1730 dirty_pages = DIV_ROUND_UP(copied + offset,
09cbfeaf 1731 PAGE_SIZE);
b6316429 1732 }
914ee295 1733
2e78c927 1734 if (num_sectors > dirty_sectors) {
8b8b08cb
CM
1735 /* release everything except the sectors we dirtied */
1736 release_bytes -= dirty_sectors <<
0b246afa 1737 fs_info->sb->s_blocksize_bits;
485290a7 1738 if (only_release_metadata) {
691fa059 1739 btrfs_delalloc_release_metadata(BTRFS_I(inode),
7ee9e440 1740 release_bytes);
485290a7
QW
1741 } else {
1742 u64 __pos;
1743
da17066c 1744 __pos = round_down(pos,
0b246afa 1745 fs_info->sectorsize) +
09cbfeaf 1746 (dirty_pages << PAGE_SHIFT);
bc42bda2
QW
1747 btrfs_delalloc_release_space(inode,
1748 data_reserved, __pos,
1749 release_bytes);
485290a7 1750 }
914ee295
XZ
1751 }
1752
2e78c927 1753 release_bytes = round_up(copied + sector_offset,
0b246afa 1754 fs_info->sectorsize);
376cc685
MX
1755
1756 if (copied > 0)
2ff7e61e
JM
1757 ret = btrfs_dirty_pages(inode, pages, dirty_pages,
1758 pos, copied, NULL);
79f015f2 1759 if (extents_locked)
376cc685
MX
1760 unlock_extent_cached(&BTRFS_I(inode)->io_tree,
1761 lockstart, lockend, &cached_state,
1762 GFP_NOFS);
8b62f87b 1763 btrfs_delalloc_release_extents(BTRFS_I(inode), reserve_bytes);
f1de9683
MX
1764 if (ret) {
1765 btrfs_drop_pages(pages, num_pages);
376cc685 1766 break;
f1de9683 1767 }
39279cc3 1768
376cc685 1769 release_bytes = 0;
8257b2dc 1770 if (only_release_metadata)
ea14b57f 1771 btrfs_end_write_no_snapshotting(root);
8257b2dc 1772
7ee9e440 1773 if (only_release_metadata && copied > 0) {
da17066c 1774 lockstart = round_down(pos,
0b246afa 1775 fs_info->sectorsize);
da17066c 1776 lockend = round_up(pos + copied,
0b246afa 1777 fs_info->sectorsize) - 1;
7ee9e440
JB
1778
1779 set_extent_bit(&BTRFS_I(inode)->io_tree, lockstart,
1780 lockend, EXTENT_NORESERVE, NULL,
1781 NULL, GFP_NOFS);
1782 only_release_metadata = false;
1783 }
1784
f1de9683
MX
1785 btrfs_drop_pages(pages, num_pages);
1786
d0215f3e
JB
1787 cond_resched();
1788
d0e1d66b 1789 balance_dirty_pages_ratelimited(inode->i_mapping);
0b246afa 1790 if (dirty_pages < (fs_info->nodesize >> PAGE_SHIFT) + 1)
2ff7e61e 1791 btrfs_btree_balance_dirty(fs_info);
cb843a6f 1792
914ee295
XZ
1793 pos += copied;
1794 num_written += copied;
d0215f3e 1795 }
39279cc3 1796
d0215f3e
JB
1797 kfree(pages);
1798
7ee9e440 1799 if (release_bytes) {
8257b2dc 1800 if (only_release_metadata) {
ea14b57f 1801 btrfs_end_write_no_snapshotting(root);
691fa059
NB
1802 btrfs_delalloc_release_metadata(BTRFS_I(inode),
1803 release_bytes);
8257b2dc 1804 } else {
bc42bda2
QW
1805 btrfs_delalloc_release_space(inode, data_reserved,
1806 round_down(pos, fs_info->sectorsize),
1807 release_bytes);
8257b2dc 1808 }
7ee9e440
JB
1809 }
1810
364ecf36 1811 extent_changeset_free(data_reserved);
d0215f3e
JB
1812 return num_written ? num_written : ret;
1813}
1814
1af5bb49 1815static ssize_t __btrfs_direct_write(struct kiocb *iocb, struct iov_iter *from)
d0215f3e
JB
1816{
1817 struct file *file = iocb->ki_filp;
728404da 1818 struct inode *inode = file_inode(file);
1af5bb49 1819 loff_t pos = iocb->ki_pos;
d0215f3e
JB
1820 ssize_t written;
1821 ssize_t written_buffered;
1822 loff_t endbyte;
1823 int err;
1824
1af5bb49 1825 written = generic_file_direct_write(iocb, from);
d0215f3e 1826
0c949334 1827 if (written < 0 || !iov_iter_count(from))
d0215f3e
JB
1828 return written;
1829
1830 pos += written;
0ae5e4d3 1831 written_buffered = __btrfs_buffered_write(file, from, pos);
d0215f3e
JB
1832 if (written_buffered < 0) {
1833 err = written_buffered;
1834 goto out;
39279cc3 1835 }
075bdbdb
FM
1836 /*
1837 * Ensure all data is persisted. We want the next direct IO read to be
1838 * able to read what was just written.
1839 */
d0215f3e 1840 endbyte = pos + written_buffered - 1;
728404da 1841 err = btrfs_fdatawrite_range(inode, pos, endbyte);
075bdbdb
FM
1842 if (err)
1843 goto out;
728404da 1844 err = filemap_fdatawait_range(inode->i_mapping, pos, endbyte);
d0215f3e
JB
1845 if (err)
1846 goto out;
1847 written += written_buffered;
867c4f93 1848 iocb->ki_pos = pos + written_buffered;
09cbfeaf
KS
1849 invalidate_mapping_pages(file->f_mapping, pos >> PAGE_SHIFT,
1850 endbyte >> PAGE_SHIFT);
39279cc3 1851out:
d0215f3e
JB
1852 return written ? written : err;
1853}
5b92ee72 1854
6c760c07
JB
1855static void update_time_for_write(struct inode *inode)
1856{
1857 struct timespec now;
1858
1859 if (IS_NOCMTIME(inode))
1860 return;
1861
c2050a45 1862 now = current_time(inode);
6c760c07
JB
1863 if (!timespec_equal(&inode->i_mtime, &now))
1864 inode->i_mtime = now;
1865
1866 if (!timespec_equal(&inode->i_ctime, &now))
1867 inode->i_ctime = now;
1868
1869 if (IS_I_VERSION(inode))
1870 inode_inc_iversion(inode);
1871}
1872
b30ac0fc
AV
1873static ssize_t btrfs_file_write_iter(struct kiocb *iocb,
1874 struct iov_iter *from)
d0215f3e
JB
1875{
1876 struct file *file = iocb->ki_filp;
496ad9aa 1877 struct inode *inode = file_inode(file);
0b246afa 1878 struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
d0215f3e 1879 struct btrfs_root *root = BTRFS_I(inode)->root;
0c1a98c8 1880 u64 start_pos;
3ac0d7b9 1881 u64 end_pos;
d0215f3e 1882 ssize_t num_written = 0;
b812ce28 1883 bool sync = (file->f_flags & O_DSYNC) || IS_SYNC(file->f_mapping->host);
3309dd04 1884 ssize_t err;
ff0fa732 1885 loff_t pos;
edf064e7 1886 size_t count = iov_iter_count(from);
27772b68
CR
1887 loff_t oldsize;
1888 int clean_page = 0;
d0215f3e 1889
91f9943e
CH
1890 if (!(iocb->ki_flags & IOCB_DIRECT) &&
1891 (iocb->ki_flags & IOCB_NOWAIT))
1892 return -EOPNOTSUPP;
1893
ff0fa732
GR
1894 if (!inode_trylock(inode)) {
1895 if (iocb->ki_flags & IOCB_NOWAIT)
edf064e7 1896 return -EAGAIN;
ff0fa732
GR
1897 inode_lock(inode);
1898 }
1899
1900 err = generic_write_checks(iocb, from);
1901 if (err <= 0) {
1902 inode_unlock(inode);
1903 return err;
1904 }
1905
1906 pos = iocb->ki_pos;
1907 if (iocb->ki_flags & IOCB_NOWAIT) {
edf064e7
GR
1908 /*
1909 * We will allocate space in case nodatacow is not set,
1910 * so bail
1911 */
1912 if (!(BTRFS_I(inode)->flags & (BTRFS_INODE_NODATACOW |
1913 BTRFS_INODE_PREALLOC)) ||
1914 check_can_nocow(BTRFS_I(inode), pos, &count) <= 0) {
1915 inode_unlock(inode);
1916 return -EAGAIN;
1917 }
d0215f3e
JB
1918 }
1919
3309dd04 1920 current->backing_dev_info = inode_to_bdi(inode);
5fa8e0a1 1921 err = file_remove_privs(file);
d0215f3e 1922 if (err) {
5955102c 1923 inode_unlock(inode);
d0215f3e
JB
1924 goto out;
1925 }
1926
1927 /*
1928 * If BTRFS flips readonly due to some impossible error
1929 * (fs_info->fs_state now has BTRFS_SUPER_FLAG_ERROR),
1930 * although we have opened a file as writable, we have
1931 * to stop this write operation to ensure FS consistency.
1932 */
0b246afa 1933 if (test_bit(BTRFS_FS_STATE_ERROR, &fs_info->fs_state)) {
5955102c 1934 inode_unlock(inode);
d0215f3e
JB
1935 err = -EROFS;
1936 goto out;
1937 }
1938
6c760c07
JB
1939 /*
1940 * We reserve space for updating the inode when we reserve space for the
1941 * extent we are going to write, so we will enospc out there. We don't
1942 * need to start yet another transaction to update the inode as we will
1943 * update the inode when we finish writing whatever data we write.
1944 */
1945 update_time_for_write(inode);
d0215f3e 1946
0b246afa 1947 start_pos = round_down(pos, fs_info->sectorsize);
27772b68
CR
1948 oldsize = i_size_read(inode);
1949 if (start_pos > oldsize) {
3ac0d7b9 1950 /* Expand hole size to cover write data, preventing empty gap */
da17066c 1951 end_pos = round_up(pos + count,
0b246afa 1952 fs_info->sectorsize);
27772b68 1953 err = btrfs_cont_expand(inode, oldsize, end_pos);
0c1a98c8 1954 if (err) {
5955102c 1955 inode_unlock(inode);
0c1a98c8
MX
1956 goto out;
1957 }
0b246afa 1958 if (start_pos > round_up(oldsize, fs_info->sectorsize))
27772b68 1959 clean_page = 1;
0c1a98c8
MX
1960 }
1961
b812ce28
JB
1962 if (sync)
1963 atomic_inc(&BTRFS_I(inode)->sync_writers);
1964
2ba48ce5 1965 if (iocb->ki_flags & IOCB_DIRECT) {
1af5bb49 1966 num_written = __btrfs_direct_write(iocb, from);
d0215f3e 1967 } else {
b30ac0fc 1968 num_written = __btrfs_buffered_write(file, from, pos);
d0215f3e 1969 if (num_written > 0)
867c4f93 1970 iocb->ki_pos = pos + num_written;
27772b68
CR
1971 if (clean_page)
1972 pagecache_isize_extended(inode, oldsize,
1973 i_size_read(inode));
d0215f3e
JB
1974 }
1975
5955102c 1976 inode_unlock(inode);
2ff3e9b6 1977
5a3f23d5 1978 /*
6c760c07
JB
1979 * We also have to set last_sub_trans to the current log transid,
1980 * otherwise subsequent syncs to a file that's been synced in this
bb7ab3b9 1981 * transaction will appear to have already occurred.
5a3f23d5 1982 */
2f2ff0ee 1983 spin_lock(&BTRFS_I(inode)->lock);
6c760c07 1984 BTRFS_I(inode)->last_sub_trans = root->log_transid;
2f2ff0ee 1985 spin_unlock(&BTRFS_I(inode)->lock);
e2592217
CH
1986 if (num_written > 0)
1987 num_written = generic_write_sync(iocb, num_written);
0a3404dc 1988
b812ce28
JB
1989 if (sync)
1990 atomic_dec(&BTRFS_I(inode)->sync_writers);
0a3404dc 1991out:
39279cc3 1992 current->backing_dev_info = NULL;
39279cc3
CM
1993 return num_written ? num_written : err;
1994}
1995
d397712b 1996int btrfs_release_file(struct inode *inode, struct file *filp)
e1b81e67 1997{
23b5ec74
JB
1998 struct btrfs_file_private *private = filp->private_data;
1999
2000 if (private && private->trans)
6bf13c0c 2001 btrfs_ioctl_trans_end(filp);
23b5ec74
JB
2002 if (private && private->filldir_buf)
2003 kfree(private->filldir_buf);
2004 kfree(private);
2005 filp->private_data = NULL;
2006
f6dc45c7
CM
2007 /*
2008 * ordered_data_close is set by settattr when we are about to truncate
2009 * a file from a non-zero size to a zero size. This tries to
2010 * flush down new bytes that may have been written if the
2011 * application were using truncate to replace a file in place.
2012 */
2013 if (test_and_clear_bit(BTRFS_INODE_ORDERED_DATA_CLOSE,
2014 &BTRFS_I(inode)->runtime_flags))
2015 filemap_flush(inode->i_mapping);
e1b81e67
M
2016 return 0;
2017}
2018
669249ee
FM
2019static int start_ordered_ops(struct inode *inode, loff_t start, loff_t end)
2020{
2021 int ret;
2022
2023 atomic_inc(&BTRFS_I(inode)->sync_writers);
728404da 2024 ret = btrfs_fdatawrite_range(inode, start, end);
669249ee
FM
2025 atomic_dec(&BTRFS_I(inode)->sync_writers);
2026
2027 return ret;
2028}
2029
d352ac68
CM
2030/*
2031 * fsync call for both files and directories. This logs the inode into
2032 * the tree log instead of forcing full commits whenever possible.
2033 *
2034 * It needs to call filemap_fdatawait so that all ordered extent updates are
2035 * in the metadata btree are up to date for copying to the log.
2036 *
2037 * It drops the inode mutex before doing the tree log commit. This is an
2038 * important optimization for directories because holding the mutex prevents
2039 * new operations on the dir while we write to disk.
2040 */
02c24a82 2041int btrfs_sync_file(struct file *file, loff_t start, loff_t end, int datasync)
39279cc3 2042{
de17e793 2043 struct dentry *dentry = file_dentry(file);
2b0143b5 2044 struct inode *inode = d_inode(dentry);
0b246afa 2045 struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
39279cc3 2046 struct btrfs_root *root = BTRFS_I(inode)->root;
39279cc3 2047 struct btrfs_trans_handle *trans;
8b050d35 2048 struct btrfs_log_ctx ctx;
333427a5 2049 int ret = 0, err;
897ca819 2050 bool full_sync = false;
9dcbeed4 2051 u64 len;
39279cc3 2052
9dcbeed4
DS
2053 /*
2054 * The range length can be represented by u64, we have to do the typecasts
2055 * to avoid signed overflow if it's [0, LLONG_MAX] eg. from fsync()
2056 */
2057 len = (u64)end - (u64)start + 1;
1abe9b8a 2058 trace_btrfs_sync_file(file, datasync);
257c62e1 2059
ebb70442
LB
2060 btrfs_init_log_ctx(&ctx, inode);
2061
90abccf2
MX
2062 /*
2063 * We write the dirty pages in the range and wait until they complete
2064 * out of the ->i_mutex. If so, we can flush the dirty pages by
2ab28f32
JB
2065 * multi-task, and make the performance up. See
2066 * btrfs_wait_ordered_range for an explanation of the ASYNC check.
90abccf2 2067 */
669249ee 2068 ret = start_ordered_ops(inode, start, end);
90abccf2 2069 if (ret)
333427a5 2070 goto out;
90abccf2 2071
5955102c 2072 inode_lock(inode);
2ecb7923 2073 atomic_inc(&root->log_batch);
2ab28f32
JB
2074 full_sync = test_bit(BTRFS_INODE_NEEDS_FULL_SYNC,
2075 &BTRFS_I(inode)->runtime_flags);
669249ee
FM
2076 /*
2077 * We might have have had more pages made dirty after calling
2078 * start_ordered_ops and before acquiring the inode's i_mutex.
2079 */
0ef8b726 2080 if (full_sync) {
669249ee
FM
2081 /*
2082 * For a full sync, we need to make sure any ordered operations
2083 * start and finish before we start logging the inode, so that
2084 * all extents are persisted and the respective file extent
2085 * items are in the fs/subvol btree.
2086 */
b659ef02 2087 ret = btrfs_wait_ordered_range(inode, start, len);
669249ee
FM
2088 } else {
2089 /*
2090 * Start any new ordered operations before starting to log the
2091 * inode. We will wait for them to finish in btrfs_sync_log().
2092 *
2093 * Right before acquiring the inode's mutex, we might have new
2094 * writes dirtying pages, which won't immediately start the
2095 * respective ordered operations - that is done through the
2096 * fill_delalloc callbacks invoked from the writepage and
2097 * writepages address space operations. So make sure we start
2098 * all ordered operations before starting to log our inode. Not
2099 * doing this means that while logging the inode, writeback
2100 * could start and invoke writepage/writepages, which would call
2101 * the fill_delalloc callbacks (cow_file_range,
2102 * submit_compressed_extents). These callbacks add first an
2103 * extent map to the modified list of extents and then create
2104 * the respective ordered operation, which means in
2105 * tree-log.c:btrfs_log_inode() we might capture all existing
2106 * ordered operations (with btrfs_get_logged_extents()) before
2107 * the fill_delalloc callback adds its ordered operation, and by
2108 * the time we visit the modified list of extent maps (with
2109 * btrfs_log_changed_extents()), we see and process the extent
2110 * map they created. We then use the extent map to construct a
2111 * file extent item for logging without waiting for the
2112 * respective ordered operation to finish - this file extent
2113 * item points to a disk location that might not have yet been
2114 * written to, containing random data - so after a crash a log
2115 * replay will make our inode have file extent items that point
2116 * to disk locations containing invalid data, as we returned
2117 * success to userspace without waiting for the respective
2118 * ordered operation to finish, because it wasn't captured by
2119 * btrfs_get_logged_extents().
2120 */
2121 ret = start_ordered_ops(inode, start, end);
2122 }
2123 if (ret) {
5955102c 2124 inode_unlock(inode);
669249ee 2125 goto out;
0ef8b726 2126 }
2ecb7923 2127 atomic_inc(&root->log_batch);
257c62e1 2128
39279cc3 2129 /*
3a8b36f3
FM
2130 * If the last transaction that changed this file was before the current
2131 * transaction and we have the full sync flag set in our inode, we can
2132 * bail out now without any syncing.
2133 *
2134 * Note that we can't bail out if the full sync flag isn't set. This is
2135 * because when the full sync flag is set we start all ordered extents
2136 * and wait for them to fully complete - when they complete they update
2137 * the inode's last_trans field through:
2138 *
2139 * btrfs_finish_ordered_io() ->
2140 * btrfs_update_inode_fallback() ->
2141 * btrfs_update_inode() ->
2142 * btrfs_set_inode_last_trans()
2143 *
2144 * So we are sure that last_trans is up to date and can do this check to
2145 * bail out safely. For the fast path, when the full sync flag is not
2146 * set in our inode, we can not do it because we start only our ordered
2147 * extents and don't wait for them to complete (that is when
2148 * btrfs_finish_ordered_io runs), so here at this point their last_trans
2149 * value might be less than or equals to fs_info->last_trans_committed,
2150 * and setting a speculative last_trans for an inode when a buffered
2151 * write is made (such as fs_info->generation + 1 for example) would not
2152 * be reliable since after setting the value and before fsync is called
2153 * any number of transactions can start and commit (transaction kthread
2154 * commits the current transaction periodically), and a transaction
2155 * commit does not start nor waits for ordered extents to complete.
257c62e1 2156 */
a4abeea4 2157 smp_mb();
0f8939b8 2158 if (btrfs_inode_in_log(BTRFS_I(inode), fs_info->generation) ||
affc0ff9 2159 (full_sync && BTRFS_I(inode)->last_trans <=
0b246afa 2160 fs_info->last_trans_committed) ||
affc0ff9
FM
2161 (!btrfs_have_ordered_extents_in_range(inode, start, len) &&
2162 BTRFS_I(inode)->last_trans
0b246afa 2163 <= fs_info->last_trans_committed)) {
5dc562c5 2164 /*
01327610 2165 * We've had everything committed since the last time we were
5dc562c5
JB
2166 * modified so clear this flag in case it was set for whatever
2167 * reason, it's no longer relevant.
2168 */
2169 clear_bit(BTRFS_INODE_NEEDS_FULL_SYNC,
2170 &BTRFS_I(inode)->runtime_flags);
0596a904
FM
2171 /*
2172 * An ordered extent might have started before and completed
2173 * already with io errors, in which case the inode was not
2174 * updated and we end up here. So check the inode's mapping
333427a5
JL
2175 * for any errors that might have happened since we last
2176 * checked called fsync.
0596a904 2177 */
333427a5 2178 ret = filemap_check_wb_err(inode->i_mapping, file->f_wb_err);
5955102c 2179 inode_unlock(inode);
15ee9bc7
JB
2180 goto out;
2181 }
15ee9bc7
JB
2182
2183 /*
a52d9a80
CM
2184 * ok we haven't committed the transaction yet, lets do a commit
2185 */
6f902af4 2186 if (file->private_data)
6bf13c0c
SW
2187 btrfs_ioctl_trans_end(file);
2188
5039eddc
JB
2189 /*
2190 * We use start here because we will need to wait on the IO to complete
2191 * in btrfs_sync_log, which could require joining a transaction (for
2192 * example checking cross references in the nocow path). If we use join
2193 * here we could get into a situation where we're waiting on IO to
2194 * happen that is blocked on a transaction trying to commit. With start
2195 * we inc the extwriter counter, so we wait for all extwriters to exit
2196 * before we start blocking join'ers. This comment is to keep somebody
2197 * from thinking they are super smart and changing this to
2198 * btrfs_join_transaction *cough*Josef*cough*.
2199 */
a22285a6
YZ
2200 trans = btrfs_start_transaction(root, 0);
2201 if (IS_ERR(trans)) {
2202 ret = PTR_ERR(trans);
5955102c 2203 inode_unlock(inode);
39279cc3
CM
2204 goto out;
2205 }
5039eddc 2206 trans->sync = true;
e02119d5 2207
49dae1bc 2208 ret = btrfs_log_dentry_safe(trans, root, dentry, start, end, &ctx);
02c24a82 2209 if (ret < 0) {
a0634be5
FDBM
2210 /* Fallthrough and commit/free transaction. */
2211 ret = 1;
02c24a82 2212 }
49eb7e46
CM
2213
2214 /* we've logged all the items and now have a consistent
2215 * version of the file in the log. It is possible that
2216 * someone will come in and modify the file, but that's
2217 * fine because the log is consistent on disk, and we
2218 * have references to all of the file's extents
2219 *
2220 * It is possible that someone will come in and log the
2221 * file again, but that will end up using the synchronization
2222 * inside btrfs_sync_log to keep things safe.
2223 */
5955102c 2224 inode_unlock(inode);
49eb7e46 2225
8407f553
FM
2226 /*
2227 * If any of the ordered extents had an error, just return it to user
2228 * space, so that the application knows some writes didn't succeed and
2229 * can take proper action (retry for e.g.). Blindly committing the
2230 * transaction in this case, would fool userspace that everything was
2231 * successful. And we also want to make sure our log doesn't contain
2232 * file extent items pointing to extents that weren't fully written to -
2233 * just like in the non fast fsync path, where we check for the ordered
2234 * operation's error flag before writing to the log tree and return -EIO
2235 * if any of them had this flag set (btrfs_wait_ordered_range) -
2236 * therefore we need to check for errors in the ordered operations,
2237 * which are indicated by ctx.io_err.
2238 */
2239 if (ctx.io_err) {
3a45bb20 2240 btrfs_end_transaction(trans);
8407f553
FM
2241 ret = ctx.io_err;
2242 goto out;
2243 }
2244
257c62e1 2245 if (ret != BTRFS_NO_LOG_SYNC) {
0ef8b726 2246 if (!ret) {
8b050d35 2247 ret = btrfs_sync_log(trans, root, &ctx);
0ef8b726 2248 if (!ret) {
3a45bb20 2249 ret = btrfs_end_transaction(trans);
0ef8b726 2250 goto out;
2ab28f32 2251 }
257c62e1 2252 }
0ef8b726 2253 if (!full_sync) {
9dcbeed4 2254 ret = btrfs_wait_ordered_range(inode, start, len);
b05fd874 2255 if (ret) {
3a45bb20 2256 btrfs_end_transaction(trans);
0ef8b726 2257 goto out;
b05fd874 2258 }
0ef8b726 2259 }
3a45bb20 2260 ret = btrfs_commit_transaction(trans);
257c62e1 2261 } else {
3a45bb20 2262 ret = btrfs_end_transaction(trans);
e02119d5 2263 }
39279cc3 2264out:
ebb70442 2265 ASSERT(list_empty(&ctx.list));
333427a5
JL
2266 err = file_check_and_advance_wb_err(file);
2267 if (!ret)
2268 ret = err;
014e4ac4 2269 return ret > 0 ? -EIO : ret;
39279cc3
CM
2270}
2271
f0f37e2f 2272static const struct vm_operations_struct btrfs_file_vm_ops = {
92fee66d 2273 .fault = filemap_fault,
f1820361 2274 .map_pages = filemap_map_pages,
9ebefb18
CM
2275 .page_mkwrite = btrfs_page_mkwrite,
2276};
2277
2278static int btrfs_file_mmap(struct file *filp, struct vm_area_struct *vma)
2279{
058a457e
MX
2280 struct address_space *mapping = filp->f_mapping;
2281
2282 if (!mapping->a_ops->readpage)
2283 return -ENOEXEC;
2284
9ebefb18 2285 file_accessed(filp);
058a457e 2286 vma->vm_ops = &btrfs_file_vm_ops;
058a457e 2287
9ebefb18
CM
2288 return 0;
2289}
2290
35339c24 2291static int hole_mergeable(struct btrfs_inode *inode, struct extent_buffer *leaf,
2aaa6655
JB
2292 int slot, u64 start, u64 end)
2293{
2294 struct btrfs_file_extent_item *fi;
2295 struct btrfs_key key;
2296
2297 if (slot < 0 || slot >= btrfs_header_nritems(leaf))
2298 return 0;
2299
2300 btrfs_item_key_to_cpu(leaf, &key, slot);
35339c24 2301 if (key.objectid != btrfs_ino(inode) ||
2aaa6655
JB
2302 key.type != BTRFS_EXTENT_DATA_KEY)
2303 return 0;
2304
2305 fi = btrfs_item_ptr(leaf, slot, struct btrfs_file_extent_item);
2306
2307 if (btrfs_file_extent_type(leaf, fi) != BTRFS_FILE_EXTENT_REG)
2308 return 0;
2309
2310 if (btrfs_file_extent_disk_bytenr(leaf, fi))
2311 return 0;
2312
2313 if (key.offset == end)
2314 return 1;
2315 if (key.offset + btrfs_file_extent_num_bytes(leaf, fi) == start)
2316 return 1;
2317 return 0;
2318}
2319
a012a74e
NB
2320static int fill_holes(struct btrfs_trans_handle *trans,
2321 struct btrfs_inode *inode,
2322 struct btrfs_path *path, u64 offset, u64 end)
2aaa6655 2323{
a012a74e
NB
2324 struct btrfs_fs_info *fs_info = btrfs_sb(inode->vfs_inode.i_sb);
2325 struct btrfs_root *root = inode->root;
2aaa6655
JB
2326 struct extent_buffer *leaf;
2327 struct btrfs_file_extent_item *fi;
2328 struct extent_map *hole_em;
a012a74e 2329 struct extent_map_tree *em_tree = &inode->extent_tree;
2aaa6655
JB
2330 struct btrfs_key key;
2331 int ret;
2332
0b246afa 2333 if (btrfs_fs_incompat(fs_info, NO_HOLES))
16e7549f
JB
2334 goto out;
2335
a012a74e 2336 key.objectid = btrfs_ino(inode);
2aaa6655
JB
2337 key.type = BTRFS_EXTENT_DATA_KEY;
2338 key.offset = offset;
2339
2aaa6655 2340 ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
f94480bd
JB
2341 if (ret <= 0) {
2342 /*
2343 * We should have dropped this offset, so if we find it then
2344 * something has gone horribly wrong.
2345 */
2346 if (ret == 0)
2347 ret = -EINVAL;
2aaa6655 2348 return ret;
f94480bd 2349 }
2aaa6655
JB
2350
2351 leaf = path->nodes[0];
a012a74e 2352 if (hole_mergeable(inode, leaf, path->slots[0] - 1, offset, end)) {
2aaa6655
JB
2353 u64 num_bytes;
2354
2355 path->slots[0]--;
2356 fi = btrfs_item_ptr(leaf, path->slots[0],
2357 struct btrfs_file_extent_item);
2358 num_bytes = btrfs_file_extent_num_bytes(leaf, fi) +
2359 end - offset;
2360 btrfs_set_file_extent_num_bytes(leaf, fi, num_bytes);
2361 btrfs_set_file_extent_ram_bytes(leaf, fi, num_bytes);
2362 btrfs_set_file_extent_offset(leaf, fi, 0);
2363 btrfs_mark_buffer_dirty(leaf);
2364 goto out;
2365 }
2366
1707e26d 2367 if (hole_mergeable(inode, leaf, path->slots[0], offset, end)) {
2aaa6655
JB
2368 u64 num_bytes;
2369
2aaa6655 2370 key.offset = offset;
0b246afa 2371 btrfs_set_item_key_safe(fs_info, path, &key);
2aaa6655
JB
2372 fi = btrfs_item_ptr(leaf, path->slots[0],
2373 struct btrfs_file_extent_item);
2374 num_bytes = btrfs_file_extent_num_bytes(leaf, fi) + end -
2375 offset;
2376 btrfs_set_file_extent_num_bytes(leaf, fi, num_bytes);
2377 btrfs_set_file_extent_ram_bytes(leaf, fi, num_bytes);
2378 btrfs_set_file_extent_offset(leaf, fi, 0);
2379 btrfs_mark_buffer_dirty(leaf);
2380 goto out;
2381 }
2382 btrfs_release_path(path);
2383
a012a74e 2384 ret = btrfs_insert_file_extent(trans, root, btrfs_ino(inode),
f85b7379 2385 offset, 0, 0, end - offset, 0, end - offset, 0, 0, 0);
2aaa6655
JB
2386 if (ret)
2387 return ret;
2388
2389out:
2390 btrfs_release_path(path);
2391
2392 hole_em = alloc_extent_map();
2393 if (!hole_em) {
2394 btrfs_drop_extent_cache(inode, offset, end - 1, 0);
a012a74e 2395 set_bit(BTRFS_INODE_NEEDS_FULL_SYNC, &inode->runtime_flags);
2aaa6655
JB
2396 } else {
2397 hole_em->start = offset;
2398 hole_em->len = end - offset;
cc95bef6 2399 hole_em->ram_bytes = hole_em->len;
2aaa6655
JB
2400 hole_em->orig_start = offset;
2401
2402 hole_em->block_start = EXTENT_MAP_HOLE;
2403 hole_em->block_len = 0;
b4939680 2404 hole_em->orig_block_len = 0;
0b246afa 2405 hole_em->bdev = fs_info->fs_devices->latest_bdev;
2aaa6655
JB
2406 hole_em->compress_type = BTRFS_COMPRESS_NONE;
2407 hole_em->generation = trans->transid;
2408
2409 do {
2410 btrfs_drop_extent_cache(inode, offset, end - 1, 0);
2411 write_lock(&em_tree->lock);
09a2a8f9 2412 ret = add_extent_mapping(em_tree, hole_em, 1);
2aaa6655
JB
2413 write_unlock(&em_tree->lock);
2414 } while (ret == -EEXIST);
2415 free_extent_map(hole_em);
2416 if (ret)
2417 set_bit(BTRFS_INODE_NEEDS_FULL_SYNC,
a012a74e 2418 &inode->runtime_flags);
2aaa6655
JB
2419 }
2420
2421 return 0;
2422}
2423
d7781546
QW
2424/*
2425 * Find a hole extent on given inode and change start/len to the end of hole
2426 * extent.(hole/vacuum extent whose em->start <= start &&
2427 * em->start + em->len > start)
2428 * When a hole extent is found, return 1 and modify start/len.
2429 */
2430static int find_first_non_hole(struct inode *inode, u64 *start, u64 *len)
2431{
609805d8 2432 struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
d7781546
QW
2433 struct extent_map *em;
2434 int ret = 0;
2435
609805d8
FM
2436 em = btrfs_get_extent(BTRFS_I(inode), NULL, 0,
2437 round_down(*start, fs_info->sectorsize),
2438 round_up(*len, fs_info->sectorsize), 0);
9986277e
DC
2439 if (IS_ERR(em))
2440 return PTR_ERR(em);
d7781546
QW
2441
2442 /* Hole or vacuum extent(only exists in no-hole mode) */
2443 if (em->block_start == EXTENT_MAP_HOLE) {
2444 ret = 1;
2445 *len = em->start + em->len > *start + *len ?
2446 0 : *start + *len - em->start - em->len;
2447 *start = em->start + em->len;
2448 }
2449 free_extent_map(em);
2450 return ret;
2451}
2452
2aaa6655
JB
2453static int btrfs_punch_hole(struct inode *inode, loff_t offset, loff_t len)
2454{
0b246afa 2455 struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
2aaa6655
JB
2456 struct btrfs_root *root = BTRFS_I(inode)->root;
2457 struct extent_state *cached_state = NULL;
2458 struct btrfs_path *path;
2459 struct btrfs_block_rsv *rsv;
2460 struct btrfs_trans_handle *trans;
d7781546
QW
2461 u64 lockstart;
2462 u64 lockend;
2463 u64 tail_start;
2464 u64 tail_len;
2465 u64 orig_start = offset;
2466 u64 cur_offset;
5f52a2c5 2467 u64 min_size = btrfs_calc_trans_metadata_size(fs_info, 1);
2aaa6655 2468 u64 drop_end;
2aaa6655
JB
2469 int ret = 0;
2470 int err = 0;
6e4d6fa1 2471 unsigned int rsv_count;
9703fefe 2472 bool same_block;
0b246afa 2473 bool no_holes = btrfs_fs_incompat(fs_info, NO_HOLES);
a1a50f60 2474 u64 ino_size;
9703fefe 2475 bool truncated_block = false;
e8c1c76e 2476 bool updated_inode = false;
2aaa6655 2477
0ef8b726
JB
2478 ret = btrfs_wait_ordered_range(inode, offset, len);
2479 if (ret)
2480 return ret;
2aaa6655 2481
5955102c 2482 inode_lock(inode);
0b246afa 2483 ino_size = round_up(inode->i_size, fs_info->sectorsize);
d7781546
QW
2484 ret = find_first_non_hole(inode, &offset, &len);
2485 if (ret < 0)
2486 goto out_only_mutex;
2487 if (ret && !len) {
2488 /* Already in a large hole */
2489 ret = 0;
2490 goto out_only_mutex;
2491 }
2492
da17066c 2493 lockstart = round_up(offset, btrfs_inode_sectorsize(inode));
d7781546 2494 lockend = round_down(offset + len,
da17066c 2495 btrfs_inode_sectorsize(inode)) - 1;
0b246afa
JM
2496 same_block = (BTRFS_BYTES_TO_BLKS(fs_info, offset))
2497 == (BTRFS_BYTES_TO_BLKS(fs_info, offset + len - 1));
7426cc04 2498 /*
9703fefe 2499 * We needn't truncate any block which is beyond the end of the file
7426cc04
MX
2500 * because we are sure there is no data there.
2501 */
2aaa6655 2502 /*
9703fefe
CR
2503 * Only do this if we are in the same block and we aren't doing the
2504 * entire block.
2aaa6655 2505 */
0b246afa 2506 if (same_block && len < fs_info->sectorsize) {
e8c1c76e 2507 if (offset < ino_size) {
9703fefe
CR
2508 truncated_block = true;
2509 ret = btrfs_truncate_block(inode, offset, len, 0);
e8c1c76e
FM
2510 } else {
2511 ret = 0;
2512 }
d7781546 2513 goto out_only_mutex;
2aaa6655
JB
2514 }
2515
9703fefe 2516 /* zero back part of the first block */
12870f1c 2517 if (offset < ino_size) {
9703fefe
CR
2518 truncated_block = true;
2519 ret = btrfs_truncate_block(inode, offset, 0, 0);
7426cc04 2520 if (ret) {
5955102c 2521 inode_unlock(inode);
7426cc04
MX
2522 return ret;
2523 }
2aaa6655
JB
2524 }
2525
d7781546
QW
2526 /* Check the aligned pages after the first unaligned page,
2527 * if offset != orig_start, which means the first unaligned page
01327610 2528 * including several following pages are already in holes,
d7781546
QW
2529 * the extra check can be skipped */
2530 if (offset == orig_start) {
2531 /* after truncate page, check hole again */
2532 len = offset + len - lockstart;
2533 offset = lockstart;
2534 ret = find_first_non_hole(inode, &offset, &len);
2535 if (ret < 0)
2536 goto out_only_mutex;
2537 if (ret && !len) {
2538 ret = 0;
2539 goto out_only_mutex;
2540 }
2541 lockstart = offset;
2542 }
2543
2544 /* Check the tail unaligned part is in a hole */
2545 tail_start = lockend + 1;
2546 tail_len = offset + len - tail_start;
2547 if (tail_len) {
2548 ret = find_first_non_hole(inode, &tail_start, &tail_len);
2549 if (unlikely(ret < 0))
2550 goto out_only_mutex;
2551 if (!ret) {
2552 /* zero the front end of the last page */
2553 if (tail_start + tail_len < ino_size) {
9703fefe
CR
2554 truncated_block = true;
2555 ret = btrfs_truncate_block(inode,
2556 tail_start + tail_len,
2557 0, 1);
d7781546
QW
2558 if (ret)
2559 goto out_only_mutex;
51f395ad 2560 }
0061280d 2561 }
2aaa6655
JB
2562 }
2563
2564 if (lockend < lockstart) {
e8c1c76e
FM
2565 ret = 0;
2566 goto out_only_mutex;
2aaa6655
JB
2567 }
2568
2569 while (1) {
2570 struct btrfs_ordered_extent *ordered;
2571
2572 truncate_pagecache_range(inode, lockstart, lockend);
2573
2574 lock_extent_bits(&BTRFS_I(inode)->io_tree, lockstart, lockend,
ff13db41 2575 &cached_state);
2aaa6655
JB
2576 ordered = btrfs_lookup_first_ordered_extent(inode, lockend);
2577
2578 /*
2579 * We need to make sure we have no ordered extents in this range
2580 * and nobody raced in and read a page in this range, if we did
2581 * we need to try again.
2582 */
2583 if ((!ordered ||
6126e3ca 2584 (ordered->file_offset + ordered->len <= lockstart ||
2aaa6655 2585 ordered->file_offset > lockend)) &&
fc4adbff 2586 !btrfs_page_exists_in_range(inode, lockstart, lockend)) {
2aaa6655
JB
2587 if (ordered)
2588 btrfs_put_ordered_extent(ordered);
2589 break;
2590 }
2591 if (ordered)
2592 btrfs_put_ordered_extent(ordered);
2593 unlock_extent_cached(&BTRFS_I(inode)->io_tree, lockstart,
2594 lockend, &cached_state, GFP_NOFS);
0ef8b726
JB
2595 ret = btrfs_wait_ordered_range(inode, lockstart,
2596 lockend - lockstart + 1);
2597 if (ret) {
5955102c 2598 inode_unlock(inode);
0ef8b726
JB
2599 return ret;
2600 }
2aaa6655
JB
2601 }
2602
2603 path = btrfs_alloc_path();
2604 if (!path) {
2605 ret = -ENOMEM;
2606 goto out;
2607 }
2608
2ff7e61e 2609 rsv = btrfs_alloc_block_rsv(fs_info, BTRFS_BLOCK_RSV_TEMP);
2aaa6655
JB
2610 if (!rsv) {
2611 ret = -ENOMEM;
2612 goto out_free;
2613 }
5f52a2c5 2614 rsv->size = btrfs_calc_trans_metadata_size(fs_info, 1);
2aaa6655
JB
2615 rsv->failfast = 1;
2616
2617 /*
2618 * 1 - update the inode
2619 * 1 - removing the extents in the range
16e7549f 2620 * 1 - adding the hole extent if no_holes isn't set
2aaa6655 2621 */
16e7549f
JB
2622 rsv_count = no_holes ? 2 : 3;
2623 trans = btrfs_start_transaction(root, rsv_count);
2aaa6655
JB
2624 if (IS_ERR(trans)) {
2625 err = PTR_ERR(trans);
2626 goto out_free;
2627 }
2628
0b246afa 2629 ret = btrfs_block_rsv_migrate(&fs_info->trans_block_rsv, rsv,
25d609f8 2630 min_size, 0);
2aaa6655
JB
2631 BUG_ON(ret);
2632 trans->block_rsv = rsv;
2633
d7781546
QW
2634 cur_offset = lockstart;
2635 len = lockend - cur_offset;
2aaa6655
JB
2636 while (cur_offset < lockend) {
2637 ret = __btrfs_drop_extents(trans, root, inode, path,
2638 cur_offset, lockend + 1,
1acae57b 2639 &drop_end, 1, 0, 0, NULL);
2aaa6655
JB
2640 if (ret != -ENOSPC)
2641 break;
2642
0b246afa 2643 trans->block_rsv = &fs_info->trans_block_rsv;
2aaa6655 2644
62fe51c1 2645 if (cur_offset < drop_end && cur_offset < ino_size) {
a012a74e
NB
2646 ret = fill_holes(trans, BTRFS_I(inode), path,
2647 cur_offset, drop_end);
12870f1c 2648 if (ret) {
f94480bd
JB
2649 /*
2650 * If we failed then we didn't insert our hole
2651 * entries for the area we dropped, so now the
2652 * fs is corrupted, so we must abort the
2653 * transaction.
2654 */
2655 btrfs_abort_transaction(trans, ret);
12870f1c
FM
2656 err = ret;
2657 break;
2658 }
2aaa6655
JB
2659 }
2660
2661 cur_offset = drop_end;
2662
2663 ret = btrfs_update_inode(trans, root, inode);
2664 if (ret) {
2665 err = ret;
2666 break;
2667 }
2668
3a45bb20 2669 btrfs_end_transaction(trans);
2ff7e61e 2670 btrfs_btree_balance_dirty(fs_info);
2aaa6655 2671
16e7549f 2672 trans = btrfs_start_transaction(root, rsv_count);
2aaa6655
JB
2673 if (IS_ERR(trans)) {
2674 ret = PTR_ERR(trans);
2675 trans = NULL;
2676 break;
2677 }
2678
0b246afa 2679 ret = btrfs_block_rsv_migrate(&fs_info->trans_block_rsv,
25d609f8 2680 rsv, min_size, 0);
2aaa6655
JB
2681 BUG_ON(ret); /* shouldn't happen */
2682 trans->block_rsv = rsv;
d7781546
QW
2683
2684 ret = find_first_non_hole(inode, &cur_offset, &len);
2685 if (unlikely(ret < 0))
2686 break;
2687 if (ret && !len) {
2688 ret = 0;
2689 break;
2690 }
2aaa6655
JB
2691 }
2692
2693 if (ret) {
2694 err = ret;
2695 goto out_trans;
2696 }
2697
0b246afa 2698 trans->block_rsv = &fs_info->trans_block_rsv;
2959a32a
FM
2699 /*
2700 * If we are using the NO_HOLES feature we might have had already an
2701 * hole that overlaps a part of the region [lockstart, lockend] and
2702 * ends at (or beyond) lockend. Since we have no file extent items to
2703 * represent holes, drop_end can be less than lockend and so we must
2704 * make sure we have an extent map representing the existing hole (the
2705 * call to __btrfs_drop_extents() might have dropped the existing extent
2706 * map representing the existing hole), otherwise the fast fsync path
2707 * will not record the existence of the hole region
2708 * [existing_hole_start, lockend].
2709 */
2710 if (drop_end <= lockend)
2711 drop_end = lockend + 1;
fc19c5e7
FM
2712 /*
2713 * Don't insert file hole extent item if it's for a range beyond eof
2714 * (because it's useless) or if it represents a 0 bytes range (when
2715 * cur_offset == drop_end).
2716 */
2717 if (cur_offset < ino_size && cur_offset < drop_end) {
a012a74e
NB
2718 ret = fill_holes(trans, BTRFS_I(inode), path,
2719 cur_offset, drop_end);
12870f1c 2720 if (ret) {
f94480bd
JB
2721 /* Same comment as above. */
2722 btrfs_abort_transaction(trans, ret);
12870f1c
FM
2723 err = ret;
2724 goto out_trans;
2725 }
2aaa6655
JB
2726 }
2727
2728out_trans:
2729 if (!trans)
2730 goto out_free;
2731
e1f5790e 2732 inode_inc_iversion(inode);
c2050a45 2733 inode->i_mtime = inode->i_ctime = current_time(inode);
e1f5790e 2734
0b246afa 2735 trans->block_rsv = &fs_info->trans_block_rsv;
2aaa6655 2736 ret = btrfs_update_inode(trans, root, inode);
e8c1c76e 2737 updated_inode = true;
3a45bb20 2738 btrfs_end_transaction(trans);
2ff7e61e 2739 btrfs_btree_balance_dirty(fs_info);
2aaa6655
JB
2740out_free:
2741 btrfs_free_path(path);
2ff7e61e 2742 btrfs_free_block_rsv(fs_info, rsv);
2aaa6655
JB
2743out:
2744 unlock_extent_cached(&BTRFS_I(inode)->io_tree, lockstart, lockend,
2745 &cached_state, GFP_NOFS);
d7781546 2746out_only_mutex:
9703fefe 2747 if (!updated_inode && truncated_block && !ret && !err) {
e8c1c76e
FM
2748 /*
2749 * If we only end up zeroing part of a page, we still need to
2750 * update the inode item, so that all the time fields are
2751 * updated as well as the necessary btrfs inode in memory fields
2752 * for detecting, at fsync time, if the inode isn't yet in the
2753 * log tree or it's there but not up to date.
2754 */
2755 trans = btrfs_start_transaction(root, 1);
2756 if (IS_ERR(trans)) {
2757 err = PTR_ERR(trans);
2758 } else {
2759 err = btrfs_update_inode(trans, root, inode);
3a45bb20 2760 ret = btrfs_end_transaction(trans);
e8c1c76e
FM
2761 }
2762 }
5955102c 2763 inode_unlock(inode);
2aaa6655
JB
2764 if (ret && !err)
2765 err = ret;
2766 return err;
2767}
2768
14524a84
QW
2769/* Helper structure to record which range is already reserved */
2770struct falloc_range {
2771 struct list_head list;
2772 u64 start;
2773 u64 len;
2774};
2775
2776/*
2777 * Helper function to add falloc range
2778 *
2779 * Caller should have locked the larger range of extent containing
2780 * [start, len)
2781 */
2782static int add_falloc_range(struct list_head *head, u64 start, u64 len)
2783{
2784 struct falloc_range *prev = NULL;
2785 struct falloc_range *range = NULL;
2786
2787 if (list_empty(head))
2788 goto insert;
2789
2790 /*
2791 * As fallocate iterate by bytenr order, we only need to check
2792 * the last range.
2793 */
2794 prev = list_entry(head->prev, struct falloc_range, list);
2795 if (prev->start + prev->len == start) {
2796 prev->len += len;
2797 return 0;
2798 }
2799insert:
32fc932e 2800 range = kmalloc(sizeof(*range), GFP_KERNEL);
14524a84
QW
2801 if (!range)
2802 return -ENOMEM;
2803 range->start = start;
2804 range->len = len;
2805 list_add_tail(&range->list, head);
2806 return 0;
2807}
2808
2fe17c10
CH
2809static long btrfs_fallocate(struct file *file, int mode,
2810 loff_t offset, loff_t len)
2811{
496ad9aa 2812 struct inode *inode = file_inode(file);
2fe17c10 2813 struct extent_state *cached_state = NULL;
364ecf36 2814 struct extent_changeset *data_reserved = NULL;
14524a84
QW
2815 struct falloc_range *range;
2816 struct falloc_range *tmp;
2817 struct list_head reserve_list;
2fe17c10
CH
2818 u64 cur_offset;
2819 u64 last_byte;
2820 u64 alloc_start;
2821 u64 alloc_end;
2822 u64 alloc_hint = 0;
2823 u64 locked_end;
14524a84 2824 u64 actual_end = 0;
2fe17c10 2825 struct extent_map *em;
da17066c 2826 int blocksize = btrfs_inode_sectorsize(inode);
2fe17c10
CH
2827 int ret;
2828
797f4277
MX
2829 alloc_start = round_down(offset, blocksize);
2830 alloc_end = round_up(offset + len, blocksize);
18513091 2831 cur_offset = alloc_start;
2fe17c10 2832
2aaa6655
JB
2833 /* Make sure we aren't being give some crap mode */
2834 if (mode & ~(FALLOC_FL_KEEP_SIZE | FALLOC_FL_PUNCH_HOLE))
2fe17c10
CH
2835 return -EOPNOTSUPP;
2836
2aaa6655
JB
2837 if (mode & FALLOC_FL_PUNCH_HOLE)
2838 return btrfs_punch_hole(inode, offset, len);
2839
d98456fc 2840 /*
14524a84
QW
2841 * Only trigger disk allocation, don't trigger qgroup reserve
2842 *
2843 * For qgroup space, it will be checked later.
d98456fc 2844 */
04f4f916
NB
2845 ret = btrfs_alloc_data_chunk_ondemand(BTRFS_I(inode),
2846 alloc_end - alloc_start);
14524a84 2847 if (ret < 0)
d98456fc
CM
2848 return ret;
2849
5955102c 2850 inode_lock(inode);
2a162ce9
DI
2851
2852 if (!(mode & FALLOC_FL_KEEP_SIZE) && offset + len > inode->i_size) {
2853 ret = inode_newsize_ok(inode, offset + len);
2854 if (ret)
2855 goto out;
2856 }
2fe17c10 2857
14524a84
QW
2858 /*
2859 * TODO: Move these two operations after we have checked
2860 * accurate reserved space, or fallocate can still fail but
2861 * with page truncated or size expanded.
2862 *
2863 * But that's a minor problem and won't do much harm BTW.
2864 */
2fe17c10 2865 if (alloc_start > inode->i_size) {
a41ad394
JB
2866 ret = btrfs_cont_expand(inode, i_size_read(inode),
2867 alloc_start);
2fe17c10
CH
2868 if (ret)
2869 goto out;
0f6925fa 2870 } else if (offset + len > inode->i_size) {
a71754fc
JB
2871 /*
2872 * If we are fallocating from the end of the file onward we
9703fefe
CR
2873 * need to zero out the end of the block if i_size lands in the
2874 * middle of a block.
a71754fc 2875 */
9703fefe 2876 ret = btrfs_truncate_block(inode, inode->i_size, 0, 0);
a71754fc
JB
2877 if (ret)
2878 goto out;
2fe17c10
CH
2879 }
2880
a71754fc
JB
2881 /*
2882 * wait for ordered IO before we have any locks. We'll loop again
2883 * below with the locks held.
2884 */
0ef8b726
JB
2885 ret = btrfs_wait_ordered_range(inode, alloc_start,
2886 alloc_end - alloc_start);
2887 if (ret)
2888 goto out;
a71754fc 2889
2fe17c10
CH
2890 locked_end = alloc_end - 1;
2891 while (1) {
2892 struct btrfs_ordered_extent *ordered;
2893
2894 /* the extent lock is ordered inside the running
2895 * transaction
2896 */
2897 lock_extent_bits(&BTRFS_I(inode)->io_tree, alloc_start,
ff13db41 2898 locked_end, &cached_state);
2fe17c10
CH
2899 ordered = btrfs_lookup_first_ordered_extent(inode,
2900 alloc_end - 1);
2901 if (ordered &&
2902 ordered->file_offset + ordered->len > alloc_start &&
2903 ordered->file_offset < alloc_end) {
2904 btrfs_put_ordered_extent(ordered);
2905 unlock_extent_cached(&BTRFS_I(inode)->io_tree,
2906 alloc_start, locked_end,
32fc932e 2907 &cached_state, GFP_KERNEL);
2fe17c10
CH
2908 /*
2909 * we can't wait on the range with the transaction
2910 * running or with the extent lock held
2911 */
0ef8b726
JB
2912 ret = btrfs_wait_ordered_range(inode, alloc_start,
2913 alloc_end - alloc_start);
2914 if (ret)
2915 goto out;
2fe17c10
CH
2916 } else {
2917 if (ordered)
2918 btrfs_put_ordered_extent(ordered);
2919 break;
2920 }
2921 }
2922
14524a84
QW
2923 /* First, check if we exceed the qgroup limit */
2924 INIT_LIST_HEAD(&reserve_list);
2fe17c10 2925 while (1) {
fc4f21b1 2926 em = btrfs_get_extent(BTRFS_I(inode), NULL, 0, cur_offset,
2fe17c10 2927 alloc_end - cur_offset, 0);
9986277e
DC
2928 if (IS_ERR(em)) {
2929 ret = PTR_ERR(em);
79787eaa
JM
2930 break;
2931 }
2fe17c10 2932 last_byte = min(extent_map_end(em), alloc_end);
f1e490a7 2933 actual_end = min_t(u64, extent_map_end(em), offset + len);
797f4277 2934 last_byte = ALIGN(last_byte, blocksize);
2fe17c10
CH
2935 if (em->block_start == EXTENT_MAP_HOLE ||
2936 (cur_offset >= inode->i_size &&
2937 !test_bit(EXTENT_FLAG_PREALLOC, &em->flags))) {
14524a84
QW
2938 ret = add_falloc_range(&reserve_list, cur_offset,
2939 last_byte - cur_offset);
2940 if (ret < 0) {
2941 free_extent_map(em);
2942 break;
3d850dd4 2943 }
364ecf36
QW
2944 ret = btrfs_qgroup_reserve_data(inode, &data_reserved,
2945 cur_offset, last_byte - cur_offset);
be2d253c
FM
2946 if (ret < 0) {
2947 free_extent_map(em);
14524a84 2948 break;
be2d253c 2949 }
18513091
WX
2950 } else {
2951 /*
2952 * Do not need to reserve unwritten extent for this
2953 * range, free reserved data space first, otherwise
2954 * it'll result in false ENOSPC error.
2955 */
bc42bda2
QW
2956 btrfs_free_reserved_data_space(inode, data_reserved,
2957 cur_offset, last_byte - cur_offset);
2fe17c10
CH
2958 }
2959 free_extent_map(em);
2fe17c10 2960 cur_offset = last_byte;
14524a84 2961 if (cur_offset >= alloc_end)
2fe17c10 2962 break;
14524a84
QW
2963 }
2964
2965 /*
2966 * If ret is still 0, means we're OK to fallocate.
2967 * Or just cleanup the list and exit.
2968 */
2969 list_for_each_entry_safe(range, tmp, &reserve_list, list) {
2970 if (!ret)
2971 ret = btrfs_prealloc_file_range(inode, mode,
2972 range->start,
93407472 2973 range->len, i_blocksize(inode),
14524a84 2974 offset + len, &alloc_hint);
18513091 2975 else
bc42bda2
QW
2976 btrfs_free_reserved_data_space(inode,
2977 data_reserved, range->start,
2978 range->len);
14524a84
QW
2979 list_del(&range->list);
2980 kfree(range);
2981 }
2982 if (ret < 0)
2983 goto out_unlock;
2984
2985 if (actual_end > inode->i_size &&
2986 !(mode & FALLOC_FL_KEEP_SIZE)) {
2987 struct btrfs_trans_handle *trans;
2988 struct btrfs_root *root = BTRFS_I(inode)->root;
2989
2990 /*
2991 * We didn't need to allocate any more space, but we
2992 * still extended the size of the file so we need to
2993 * update i_size and the inode item.
2994 */
2995 trans = btrfs_start_transaction(root, 1);
2996 if (IS_ERR(trans)) {
2997 ret = PTR_ERR(trans);
2998 } else {
c2050a45 2999 inode->i_ctime = current_time(inode);
14524a84
QW
3000 i_size_write(inode, actual_end);
3001 btrfs_ordered_update_i_size(inode, actual_end, NULL);
3002 ret = btrfs_update_inode(trans, root, inode);
3003 if (ret)
3a45bb20 3004 btrfs_end_transaction(trans);
14524a84 3005 else
3a45bb20 3006 ret = btrfs_end_transaction(trans);
2fe17c10
CH
3007 }
3008 }
14524a84 3009out_unlock:
2fe17c10 3010 unlock_extent_cached(&BTRFS_I(inode)->io_tree, alloc_start, locked_end,
32fc932e 3011 &cached_state, GFP_KERNEL);
2fe17c10 3012out:
5955102c 3013 inode_unlock(inode);
d98456fc 3014 /* Let go of our reservation. */
18513091 3015 if (ret != 0)
bc42bda2
QW
3016 btrfs_free_reserved_data_space(inode, data_reserved,
3017 alloc_start, alloc_end - cur_offset);
364ecf36 3018 extent_changeset_free(data_reserved);
2fe17c10
CH
3019 return ret;
3020}
3021
965c8e59 3022static int find_desired_extent(struct inode *inode, loff_t *offset, int whence)
b2675157 3023{
0b246afa 3024 struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
7f4ca37c 3025 struct extent_map *em = NULL;
b2675157 3026 struct extent_state *cached_state = NULL;
4d1a40c6
LB
3027 u64 lockstart;
3028 u64 lockend;
3029 u64 start;
3030 u64 len;
b2675157
JB
3031 int ret = 0;
3032
4d1a40c6
LB
3033 if (inode->i_size == 0)
3034 return -ENXIO;
3035
3036 /*
3037 * *offset can be negative, in this case we start finding DATA/HOLE from
3038 * the very start of the file.
3039 */
3040 start = max_t(loff_t, 0, *offset);
3041
0b246afa 3042 lockstart = round_down(start, fs_info->sectorsize);
da17066c 3043 lockend = round_up(i_size_read(inode),
0b246afa 3044 fs_info->sectorsize);
b2675157 3045 if (lockend <= lockstart)
0b246afa 3046 lockend = lockstart + fs_info->sectorsize;
1214b53f 3047 lockend--;
b2675157
JB
3048 len = lockend - lockstart + 1;
3049
ff13db41 3050 lock_extent_bits(&BTRFS_I(inode)->io_tree, lockstart, lockend,
d0082371 3051 &cached_state);
b2675157 3052
7f4ca37c 3053 while (start < inode->i_size) {
fc4f21b1
NB
3054 em = btrfs_get_extent_fiemap(BTRFS_I(inode), NULL, 0,
3055 start, len, 0);
b2675157 3056 if (IS_ERR(em)) {
6af021d8 3057 ret = PTR_ERR(em);
7f4ca37c 3058 em = NULL;
b2675157
JB
3059 break;
3060 }
3061
7f4ca37c
JB
3062 if (whence == SEEK_HOLE &&
3063 (em->block_start == EXTENT_MAP_HOLE ||
3064 test_bit(EXTENT_FLAG_PREALLOC, &em->flags)))
3065 break;
3066 else if (whence == SEEK_DATA &&
3067 (em->block_start != EXTENT_MAP_HOLE &&
3068 !test_bit(EXTENT_FLAG_PREALLOC, &em->flags)))
3069 break;
b2675157
JB
3070
3071 start = em->start + em->len;
b2675157 3072 free_extent_map(em);
7f4ca37c 3073 em = NULL;
b2675157
JB
3074 cond_resched();
3075 }
7f4ca37c
JB
3076 free_extent_map(em);
3077 if (!ret) {
3078 if (whence == SEEK_DATA && start >= inode->i_size)
3079 ret = -ENXIO;
3080 else
3081 *offset = min_t(loff_t, start, inode->i_size);
3082 }
b2675157
JB
3083 unlock_extent_cached(&BTRFS_I(inode)->io_tree, lockstart, lockend,
3084 &cached_state, GFP_NOFS);
3085 return ret;
3086}
3087
965c8e59 3088static loff_t btrfs_file_llseek(struct file *file, loff_t offset, int whence)
b2675157
JB
3089{
3090 struct inode *inode = file->f_mapping->host;
3091 int ret;
3092
5955102c 3093 inode_lock(inode);
965c8e59 3094 switch (whence) {
b2675157
JB
3095 case SEEK_END:
3096 case SEEK_CUR:
965c8e59 3097 offset = generic_file_llseek(file, offset, whence);
b2675157
JB
3098 goto out;
3099 case SEEK_DATA:
3100 case SEEK_HOLE:
48802c8a 3101 if (offset >= i_size_read(inode)) {
5955102c 3102 inode_unlock(inode);
48802c8a
JL
3103 return -ENXIO;
3104 }
3105
965c8e59 3106 ret = find_desired_extent(inode, &offset, whence);
b2675157 3107 if (ret) {
5955102c 3108 inode_unlock(inode);
b2675157
JB
3109 return ret;
3110 }
3111 }
3112
46a1c2c7 3113 offset = vfs_setpos(file, offset, inode->i_sb->s_maxbytes);
b2675157 3114out:
5955102c 3115 inode_unlock(inode);
b2675157
JB
3116 return offset;
3117}
3118
edf064e7
GR
3119static int btrfs_file_open(struct inode *inode, struct file *filp)
3120{
91f9943e 3121 filp->f_mode |= FMODE_NOWAIT;
edf064e7
GR
3122 return generic_file_open(inode, filp);
3123}
3124
828c0950 3125const struct file_operations btrfs_file_operations = {
b2675157 3126 .llseek = btrfs_file_llseek,
aad4f8bb 3127 .read_iter = generic_file_read_iter,
e9906a98 3128 .splice_read = generic_file_splice_read,
b30ac0fc 3129 .write_iter = btrfs_file_write_iter,
9ebefb18 3130 .mmap = btrfs_file_mmap,
edf064e7 3131 .open = btrfs_file_open,
e1b81e67 3132 .release = btrfs_release_file,
39279cc3 3133 .fsync = btrfs_sync_file,
2fe17c10 3134 .fallocate = btrfs_fallocate,
34287aa3 3135 .unlocked_ioctl = btrfs_ioctl,
39279cc3 3136#ifdef CONFIG_COMPAT
4c63c245 3137 .compat_ioctl = btrfs_compat_ioctl,
39279cc3 3138#endif
04b38d60 3139 .clone_file_range = btrfs_clone_file_range,
2b3909f8 3140 .dedupe_file_range = btrfs_dedupe_file_range,
39279cc3 3141};
9247f317
MX
3142
3143void btrfs_auto_defrag_exit(void)
3144{
5598e900 3145 kmem_cache_destroy(btrfs_inode_defrag_cachep);
9247f317
MX
3146}
3147
3148int btrfs_auto_defrag_init(void)
3149{
3150 btrfs_inode_defrag_cachep = kmem_cache_create("btrfs_inode_defrag",
3151 sizeof(struct inode_defrag), 0,
fba4b697 3152 SLAB_MEM_SPREAD,
9247f317
MX
3153 NULL);
3154 if (!btrfs_inode_defrag_cachep)
3155 return -ENOMEM;
3156
3157 return 0;
3158}
728404da
FM
3159
3160int btrfs_fdatawrite_range(struct inode *inode, loff_t start, loff_t end)
3161{
3162 int ret;
3163
3164 /*
3165 * So with compression we will find and lock a dirty page and clear the
3166 * first one as dirty, setup an async extent, and immediately return
3167 * with the entire range locked but with nobody actually marked with
3168 * writeback. So we can't just filemap_write_and_wait_range() and
3169 * expect it to work since it will just kick off a thread to do the
3170 * actual work. So we need to call filemap_fdatawrite_range _again_
3171 * since it will wait on the page lock, which won't be unlocked until
3172 * after the pages have been marked as writeback and so we're good to go
3173 * from there. We have to do this otherwise we'll miss the ordered
3174 * extents and that results in badness. Please Josef, do not think you
3175 * know better and pull this out at some point in the future, it is
3176 * right and you are wrong.
3177 */
3178 ret = filemap_fdatawrite_range(inode->i_mapping, start, end);
3179 if (!ret && test_bit(BTRFS_INODE_HAS_ASYNC_EXTENT,
3180 &BTRFS_I(inode)->runtime_flags))
3181 ret = filemap_fdatawrite_range(inode->i_mapping, start, end);
3182
3183 return ret;
3184}