1 // SPDX-License-Identifier: GPL-2.0
2
3 #include <linux/blkdev.h>
4 #include <linux/iversion.h>
5 #include "ctree.h"
6 #include "fs.h"
7 #include "messages.h"
8 #include "compression.h"
9 #include "delalloc-space.h"
10 #include "disk-io.h"
11 #include "reflink.h"
12 #include "transaction.h"
13 #include "subpage.h"
14 #include "accessors.h"
15 #include "file-item.h"
16 #include "file.h"
17 #include "super.h"
18
19 #define BTRFS_MAX_DEDUPE_LEN SZ_16M
20
clone_finish_inode_update(struct btrfs_trans_handle * trans,struct inode * inode,u64 endoff,const u64 destoff,const u64 olen,int no_time_update)21 static int clone_finish_inode_update(struct btrfs_trans_handle *trans,
22 struct inode *inode,
23 u64 endoff,
24 const u64 destoff,
25 const u64 olen,
26 int no_time_update)
27 {
28 int ret;
29
30 inode_inc_iversion(inode);
31 if (!no_time_update) {
32 inode_set_mtime_to_ts(inode, inode_set_ctime_current(inode));
33 }
34 /*
35 * We round up to the block size at eof when determining which
36 * extents to clone above, but shouldn't round up the file size.
37 */
38 if (endoff > destoff + olen)
39 endoff = destoff + olen;
40 if (endoff > inode->i_size) {
41 i_size_write(inode, endoff);
42 btrfs_inode_safe_disk_i_size_write(BTRFS_I(inode), 0);
43 }
44
45 ret = btrfs_update_inode(trans, BTRFS_I(inode));
46 if (ret) {
47 btrfs_abort_transaction(trans, ret);
48 btrfs_end_transaction(trans);
49 goto out;
50 }
51 ret = btrfs_end_transaction(trans);
52 out:
53 return ret;
54 }
55
copy_inline_to_page(struct btrfs_inode * inode,const u64 file_offset,char * inline_data,const u64 size,const u64 datal,const u8 comp_type)56 static int copy_inline_to_page(struct btrfs_inode *inode,
57 const u64 file_offset,
58 char *inline_data,
59 const u64 size,
60 const u64 datal,
61 const u8 comp_type)
62 {
63 struct btrfs_fs_info *fs_info = inode->root->fs_info;
64 const u32 block_size = fs_info->sectorsize;
65 const u64 range_end = file_offset + block_size - 1;
66 const size_t inline_size = size - btrfs_file_extent_calc_inline_size(0);
67 char *data_start = inline_data + btrfs_file_extent_calc_inline_size(0);
68 struct extent_changeset *data_reserved = NULL;
69 struct folio *folio = NULL;
70 struct address_space *mapping = inode->vfs_inode.i_mapping;
71 int ret;
72
73 ASSERT(IS_ALIGNED(file_offset, block_size));
74
75 /*
76 * We have flushed and locked the ranges of the source and destination
77 * inodes, we also have locked the inodes, so we are safe to do a
78 * reservation here. Also we must not do the reservation while holding
79 * a transaction open, otherwise we would deadlock.
80 */
81 ret = btrfs_delalloc_reserve_space(inode, &data_reserved, file_offset,
82 block_size);
83 if (ret)
84 goto out;
85
86 folio = __filemap_get_folio(mapping, file_offset >> PAGE_SHIFT,
87 FGP_LOCK | FGP_ACCESSED | FGP_CREAT,
88 btrfs_alloc_write_mask(mapping));
89 if (IS_ERR(folio)) {
90 ret = -ENOMEM;
91 goto out_unlock;
92 }
93
94 ret = set_folio_extent_mapped(folio);
95 if (ret < 0)
96 goto out_unlock;
97
98 clear_extent_bit(&inode->io_tree, file_offset, range_end,
99 EXTENT_DELALLOC | EXTENT_DO_ACCOUNTING | EXTENT_DEFRAG,
100 NULL);
101 ret = btrfs_set_extent_delalloc(inode, file_offset, range_end, 0, NULL);
102 if (ret)
103 goto out_unlock;
104
105 /*
106 * After dirtying the page our caller will need to start a transaction,
107 * and if we are low on metadata free space, that can cause flushing of
108 * delalloc for all inodes in order to get metadata space released.
109 * However we are holding the range locked for the whole duration of
110 * the clone/dedupe operation, so we may deadlock if that happens and no
111 * other task releases enough space. So mark this inode as not being
112 * possible to flush to avoid such deadlock. We will clear that flag
113 * when we finish cloning all extents, since a transaction is started
114 * after finding each extent to clone.
115 */
116 set_bit(BTRFS_INODE_NO_DELALLOC_FLUSH, &inode->runtime_flags);
117
118 if (comp_type == BTRFS_COMPRESS_NONE) {
119 memcpy_to_folio(folio, offset_in_folio(folio, file_offset), data_start,
120 datal);
121 } else {
122 ret = btrfs_decompress(comp_type, data_start, folio,
123 offset_in_folio(folio, file_offset),
124 inline_size, datal);
125 if (ret)
126 goto out_unlock;
127 flush_dcache_folio(folio);
128 }
129
130 /*
131 * If our inline data is smaller then the block/page size, then the
132 * remaining of the block/page is equivalent to zeroes. We had something
133 * like the following done:
134 *
135 * $ xfs_io -f -c "pwrite -S 0xab 0 500" file
136 * $ sync # (or fsync)
137 * $ xfs_io -c "falloc 0 4K" file
138 * $ xfs_io -c "pwrite -S 0xcd 4K 4K"
139 *
140 * So what's in the range [500, 4095] corresponds to zeroes.
141 */
142 if (datal < block_size)
143 folio_zero_range(folio, datal, block_size - datal);
144
145 btrfs_folio_set_uptodate(fs_info, folio, file_offset, block_size);
146 btrfs_folio_clear_checked(fs_info, folio, file_offset, block_size);
147 btrfs_folio_set_dirty(fs_info, folio, file_offset, block_size);
148 out_unlock:
149 if (!IS_ERR(folio)) {
150 folio_unlock(folio);
151 folio_put(folio);
152 }
153 if (ret)
154 btrfs_delalloc_release_space(inode, data_reserved, file_offset,
155 block_size, true);
156 btrfs_delalloc_release_extents(inode, block_size);
157 out:
158 extent_changeset_free(data_reserved);
159
160 return ret;
161 }
162
163 /*
164 * Deal with cloning of inline extents. We try to copy the inline extent from
165 * the source inode to destination inode when possible. When not possible we
166 * copy the inline extent's data into the respective page of the inode.
167 */
clone_copy_inline_extent(struct btrfs_inode * inode,struct btrfs_path * path,struct btrfs_key * new_key,const u64 drop_start,const u64 datal,const u64 size,const u8 comp_type,char * inline_data,struct btrfs_trans_handle ** trans_out)168 static int clone_copy_inline_extent(struct btrfs_inode *inode,
169 struct btrfs_path *path,
170 struct btrfs_key *new_key,
171 const u64 drop_start,
172 const u64 datal,
173 const u64 size,
174 const u8 comp_type,
175 char *inline_data,
176 struct btrfs_trans_handle **trans_out)
177 {
178 struct btrfs_root *root = inode->root;
179 struct btrfs_fs_info *fs_info = root->fs_info;
180 const u64 aligned_end = ALIGN(new_key->offset + datal,
181 fs_info->sectorsize);
182 struct btrfs_trans_handle *trans = NULL;
183 struct btrfs_drop_extents_args drop_args = { 0 };
184 int ret;
185 struct btrfs_key key;
186
187 if (new_key->offset > 0) {
188 ret = copy_inline_to_page(inode, new_key->offset,
189 inline_data, size, datal, comp_type);
190 goto out;
191 }
192
193 key.objectid = btrfs_ino(inode);
194 key.type = BTRFS_EXTENT_DATA_KEY;
195 key.offset = 0;
196 ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
197 if (ret < 0) {
198 return ret;
199 } else if (ret > 0) {
200 if (path->slots[0] >= btrfs_header_nritems(path->nodes[0])) {
201 ret = btrfs_next_leaf(root, path);
202 if (ret < 0)
203 return ret;
204 else if (ret > 0)
205 goto copy_inline_extent;
206 }
207 btrfs_item_key_to_cpu(path->nodes[0], &key, path->slots[0]);
208 if (key.objectid == btrfs_ino(inode) &&
209 key.type == BTRFS_EXTENT_DATA_KEY) {
210 /*
211 * There's an implicit hole at file offset 0, copy the
212 * inline extent's data to the page.
213 */
214 ASSERT(key.offset > 0);
215 goto copy_to_page;
216 }
217 } else if (i_size_read(&inode->vfs_inode) <= datal) {
218 struct btrfs_file_extent_item *ei;
219
220 ei = btrfs_item_ptr(path->nodes[0], path->slots[0],
221 struct btrfs_file_extent_item);
222 /*
223 * If it's an inline extent replace it with the source inline
224 * extent, otherwise copy the source inline extent data into
225 * the respective page at the destination inode.
226 */
227 if (btrfs_file_extent_type(path->nodes[0], ei) ==
228 BTRFS_FILE_EXTENT_INLINE)
229 goto copy_inline_extent;
230
231 goto copy_to_page;
232 }
233
234 copy_inline_extent:
235 /*
236 * We have no extent items, or we have an extent at offset 0 which may
237 * or may not be inlined. All these cases are dealt the same way.
238 */
239 if (i_size_read(&inode->vfs_inode) > datal) {
240 /*
241 * At the destination offset 0 we have either a hole, a regular
242 * extent or an inline extent larger then the one we want to
243 * clone. Deal with all these cases by copying the inline extent
244 * data into the respective page at the destination inode.
245 */
246 goto copy_to_page;
247 }
248
249 /*
250 * Release path before starting a new transaction so we don't hold locks
251 * that would confuse lockdep.
252 */
253 btrfs_release_path(path);
254 /*
255 * If we end up here it means were copy the inline extent into a leaf
256 * of the destination inode. We know we will drop or adjust at most one
257 * extent item in the destination root.
258 *
259 * 1 unit - adjusting old extent (we may have to split it)
260 * 1 unit - add new extent
261 * 1 unit - inode update
262 */
263 trans = btrfs_start_transaction(root, 3);
264 if (IS_ERR(trans)) {
265 ret = PTR_ERR(trans);
266 trans = NULL;
267 goto out;
268 }
269 drop_args.path = path;
270 drop_args.start = drop_start;
271 drop_args.end = aligned_end;
272 drop_args.drop_cache = true;
273 ret = btrfs_drop_extents(trans, root, inode, &drop_args);
274 if (ret)
275 goto out;
276 ret = btrfs_insert_empty_item(trans, root, path, new_key, size);
277 if (ret)
278 goto out;
279
280 write_extent_buffer(path->nodes[0], inline_data,
281 btrfs_item_ptr_offset(path->nodes[0],
282 path->slots[0]),
283 size);
284 btrfs_update_inode_bytes(inode, datal, drop_args.bytes_found);
285 btrfs_set_inode_full_sync(inode);
286 ret = btrfs_inode_set_file_extent_range(inode, 0, aligned_end);
287 out:
288 if (!ret && !trans) {
289 /*
290 * No transaction here means we copied the inline extent into a
291 * page of the destination inode.
292 *
293 * 1 unit to update inode item
294 */
295 trans = btrfs_start_transaction(root, 1);
296 if (IS_ERR(trans)) {
297 ret = PTR_ERR(trans);
298 trans = NULL;
299 }
300 }
301 if (ret && trans) {
302 btrfs_abort_transaction(trans, ret);
303 btrfs_end_transaction(trans);
304 }
305 if (!ret)
306 *trans_out = trans;
307
308 return ret;
309
310 copy_to_page:
311 /*
312 * Release our path because we don't need it anymore and also because
313 * copy_inline_to_page() needs to reserve data and metadata, which may
314 * need to flush delalloc when we are low on available space and
315 * therefore cause a deadlock if writeback of an inline extent needs to
316 * write to the same leaf or an ordered extent completion needs to write
317 * to the same leaf.
318 */
319 btrfs_release_path(path);
320
321 ret = copy_inline_to_page(inode, new_key->offset,
322 inline_data, size, datal, comp_type);
323 goto out;
324 }
325
326 /*
327 * Clone a range from inode file to another.
328 *
329 * @src: Inode to clone from
330 * @inode: Inode to clone to
331 * @off: Offset within source to start clone from
332 * @olen: Original length, passed by user, of range to clone
333 * @olen_aligned: Block-aligned value of olen
334 * @destoff: Offset within @inode to start clone
335 * @no_time_update: Whether to update mtime/ctime on the target inode
336 */
btrfs_clone(struct inode * src,struct inode * inode,const u64 off,const u64 olen,const u64 olen_aligned,const u64 destoff,int no_time_update)337 static int btrfs_clone(struct inode *src, struct inode *inode,
338 const u64 off, const u64 olen, const u64 olen_aligned,
339 const u64 destoff, int no_time_update)
340 {
341 struct btrfs_fs_info *fs_info = inode_to_fs_info(inode);
342 struct btrfs_path *path = NULL;
343 struct extent_buffer *leaf;
344 struct btrfs_trans_handle *trans;
345 char *buf = NULL;
346 struct btrfs_key key;
347 u32 nritems;
348 int slot;
349 int ret;
350 const u64 len = olen_aligned;
351 u64 last_dest_end = destoff;
352 u64 prev_extent_end = off;
353
354 ret = -ENOMEM;
355 buf = kvmalloc(fs_info->nodesize, GFP_KERNEL);
356 if (!buf)
357 return ret;
358
359 path = btrfs_alloc_path();
360 if (!path) {
361 kvfree(buf);
362 return ret;
363 }
364
365 path->reada = READA_FORWARD;
366 /* Clone data */
367 key.objectid = btrfs_ino(BTRFS_I(src));
368 key.type = BTRFS_EXTENT_DATA_KEY;
369 key.offset = off;
370
371 while (1) {
372 struct btrfs_file_extent_item *extent;
373 u64 extent_gen;
374 int type;
375 u32 size;
376 struct btrfs_key new_key;
377 u64 disko = 0, diskl = 0;
378 u64 datao = 0, datal = 0;
379 u8 comp;
380 u64 drop_start;
381
382 /* Note the key will change type as we walk through the tree */
383 ret = btrfs_search_slot(NULL, BTRFS_I(src)->root, &key, path,
384 0, 0);
385 if (ret < 0)
386 goto out;
387 /*
388 * First search, if no extent item that starts at offset off was
389 * found but the previous item is an extent item, it's possible
390 * it might overlap our target range, therefore process it.
391 */
392 if (key.offset == off && ret > 0 && path->slots[0] > 0) {
393 btrfs_item_key_to_cpu(path->nodes[0], &key,
394 path->slots[0] - 1);
395 if (key.type == BTRFS_EXTENT_DATA_KEY)
396 path->slots[0]--;
397 }
398
399 nritems = btrfs_header_nritems(path->nodes[0]);
400 process_slot:
401 if (path->slots[0] >= nritems) {
402 ret = btrfs_next_leaf(BTRFS_I(src)->root, path);
403 if (ret < 0)
404 goto out;
405 if (ret > 0)
406 break;
407 nritems = btrfs_header_nritems(path->nodes[0]);
408 }
409 leaf = path->nodes[0];
410 slot = path->slots[0];
411
412 btrfs_item_key_to_cpu(leaf, &key, slot);
413 if (key.type > BTRFS_EXTENT_DATA_KEY ||
414 key.objectid != btrfs_ino(BTRFS_I(src)))
415 break;
416
417 ASSERT(key.type == BTRFS_EXTENT_DATA_KEY);
418
419 extent = btrfs_item_ptr(leaf, slot,
420 struct btrfs_file_extent_item);
421 extent_gen = btrfs_file_extent_generation(leaf, extent);
422 comp = btrfs_file_extent_compression(leaf, extent);
423 type = btrfs_file_extent_type(leaf, extent);
424 if (type == BTRFS_FILE_EXTENT_REG ||
425 type == BTRFS_FILE_EXTENT_PREALLOC) {
426 disko = btrfs_file_extent_disk_bytenr(leaf, extent);
427 diskl = btrfs_file_extent_disk_num_bytes(leaf, extent);
428 datao = btrfs_file_extent_offset(leaf, extent);
429 datal = btrfs_file_extent_num_bytes(leaf, extent);
430 } else if (type == BTRFS_FILE_EXTENT_INLINE) {
431 /* Take upper bound, may be compressed */
432 datal = btrfs_file_extent_ram_bytes(leaf, extent);
433 }
434
435 /*
436 * The first search might have left us at an extent item that
437 * ends before our target range's start, can happen if we have
438 * holes and NO_HOLES feature enabled.
439 *
440 * Subsequent searches may leave us on a file range we have
441 * processed before - this happens due to a race with ordered
442 * extent completion for a file range that is outside our source
443 * range, but that range was part of a file extent item that
444 * also covered a leading part of our source range.
445 */
446 if (key.offset + datal <= prev_extent_end) {
447 path->slots[0]++;
448 goto process_slot;
449 } else if (key.offset >= off + len) {
450 break;
451 }
452
453 prev_extent_end = key.offset + datal;
454 size = btrfs_item_size(leaf, slot);
455 read_extent_buffer(leaf, buf, btrfs_item_ptr_offset(leaf, slot),
456 size);
457
458 btrfs_release_path(path);
459
460 memcpy(&new_key, &key, sizeof(new_key));
461 new_key.objectid = btrfs_ino(BTRFS_I(inode));
462 if (off <= key.offset)
463 new_key.offset = key.offset + destoff - off;
464 else
465 new_key.offset = destoff;
466
467 /*
468 * Deal with a hole that doesn't have an extent item that
469 * represents it (NO_HOLES feature enabled).
470 * This hole is either in the middle of the cloning range or at
471 * the beginning (fully overlaps it or partially overlaps it).
472 */
473 if (new_key.offset != last_dest_end)
474 drop_start = last_dest_end;
475 else
476 drop_start = new_key.offset;
477
478 if (type == BTRFS_FILE_EXTENT_REG ||
479 type == BTRFS_FILE_EXTENT_PREALLOC) {
480 struct btrfs_replace_extent_info clone_info;
481
482 /*
483 * a | --- range to clone ---| b
484 * | ------------- extent ------------- |
485 */
486
487 /* Subtract range b */
488 if (key.offset + datal > off + len)
489 datal = off + len - key.offset;
490
491 /* Subtract range a */
492 if (off > key.offset) {
493 datao += off - key.offset;
494 datal -= off - key.offset;
495 }
496
497 clone_info.disk_offset = disko;
498 clone_info.disk_len = diskl;
499 clone_info.data_offset = datao;
500 clone_info.data_len = datal;
501 clone_info.file_offset = new_key.offset;
502 clone_info.extent_buf = buf;
503 clone_info.is_new_extent = false;
504 clone_info.update_times = !no_time_update;
505 ret = btrfs_replace_file_extents(BTRFS_I(inode), path,
506 drop_start, new_key.offset + datal - 1,
507 &clone_info, &trans);
508 if (ret)
509 goto out;
510 } else {
511 ASSERT(type == BTRFS_FILE_EXTENT_INLINE);
512 /*
513 * Inline extents always have to start at file offset 0
514 * and can never be bigger then the sector size. We can
515 * never clone only parts of an inline extent, since all
516 * reflink operations must start at a sector size aligned
517 * offset, and the length must be aligned too or end at
518 * the i_size (which implies the whole inlined data).
519 */
520 ASSERT(key.offset == 0);
521 ASSERT(datal <= fs_info->sectorsize);
522 if (WARN_ON(type != BTRFS_FILE_EXTENT_INLINE) ||
523 WARN_ON(key.offset != 0) ||
524 WARN_ON(datal > fs_info->sectorsize)) {
525 ret = -EUCLEAN;
526 goto out;
527 }
528
529 ret = clone_copy_inline_extent(BTRFS_I(inode), path, &new_key,
530 drop_start, datal, size,
531 comp, buf, &trans);
532 if (ret)
533 goto out;
534 }
535
536 btrfs_release_path(path);
537
538 /*
539 * Whenever we share an extent we update the last_reflink_trans
540 * of each inode to the current transaction. This is needed to
541 * make sure fsync does not log multiple checksum items with
542 * overlapping ranges (because some extent items might refer
543 * only to sections of the original extent). For the destination
544 * inode we do this regardless of the generation of the extents
545 * or even if they are inline extents or explicit holes, to make
546 * sure a full fsync does not skip them. For the source inode,
547 * we only need to update last_reflink_trans in case it's a new
548 * extent that is not a hole or an inline extent, to deal with
549 * the checksums problem on fsync.
550 */
551 if (extent_gen == trans->transid && disko > 0)
552 BTRFS_I(src)->last_reflink_trans = trans->transid;
553
554 BTRFS_I(inode)->last_reflink_trans = trans->transid;
555
556 last_dest_end = ALIGN(new_key.offset + datal,
557 fs_info->sectorsize);
558 ret = clone_finish_inode_update(trans, inode, last_dest_end,
559 destoff, olen, no_time_update);
560 if (ret)
561 goto out;
562 if (new_key.offset + datal >= destoff + len)
563 break;
564
565 btrfs_release_path(path);
566 key.offset = prev_extent_end;
567
568 if (fatal_signal_pending(current)) {
569 ret = -EINTR;
570 goto out;
571 }
572
573 cond_resched();
574 }
575 ret = 0;
576
577 if (last_dest_end < destoff + len) {
578 /*
579 * We have an implicit hole that fully or partially overlaps our
580 * cloning range at its end. This means that we either have the
581 * NO_HOLES feature enabled or the implicit hole happened due to
582 * mixing buffered and direct IO writes against this file.
583 */
584 btrfs_release_path(path);
585
586 /*
587 * When using NO_HOLES and we are cloning a range that covers
588 * only a hole (no extents) into a range beyond the current
589 * i_size, punching a hole in the target range will not create
590 * an extent map defining a hole, because the range starts at or
591 * beyond current i_size. If the file previously had an i_size
592 * greater than the new i_size set by this clone operation, we
593 * need to make sure the next fsync is a full fsync, so that it
594 * detects and logs a hole covering a range from the current
595 * i_size to the new i_size. If the clone range covers extents,
596 * besides a hole, then we know the full sync flag was already
597 * set by previous calls to btrfs_replace_file_extents() that
598 * replaced file extent items.
599 */
600 if (last_dest_end >= i_size_read(inode))
601 btrfs_set_inode_full_sync(BTRFS_I(inode));
602
603 ret = btrfs_replace_file_extents(BTRFS_I(inode), path,
604 last_dest_end, destoff + len - 1, NULL, &trans);
605 if (ret)
606 goto out;
607
608 ret = clone_finish_inode_update(trans, inode, destoff + len,
609 destoff, olen, no_time_update);
610 }
611
612 out:
613 btrfs_free_path(path);
614 kvfree(buf);
615 clear_bit(BTRFS_INODE_NO_DELALLOC_FLUSH, &BTRFS_I(inode)->runtime_flags);
616
617 return ret;
618 }
619
btrfs_double_mmap_lock(struct btrfs_inode * inode1,struct btrfs_inode * inode2)620 static void btrfs_double_mmap_lock(struct btrfs_inode *inode1, struct btrfs_inode *inode2)
621 {
622 if (inode1 < inode2)
623 swap(inode1, inode2);
624 down_write(&inode1->i_mmap_lock);
625 down_write_nested(&inode2->i_mmap_lock, SINGLE_DEPTH_NESTING);
626 }
627
btrfs_double_mmap_unlock(struct btrfs_inode * inode1,struct btrfs_inode * inode2)628 static void btrfs_double_mmap_unlock(struct btrfs_inode *inode1, struct btrfs_inode *inode2)
629 {
630 up_write(&inode1->i_mmap_lock);
631 up_write(&inode2->i_mmap_lock);
632 }
633
btrfs_extent_same_range(struct btrfs_inode * src,u64 loff,u64 len,struct btrfs_inode * dst,u64 dst_loff)634 static int btrfs_extent_same_range(struct btrfs_inode *src, u64 loff, u64 len,
635 struct btrfs_inode *dst, u64 dst_loff)
636 {
637 const u64 end = dst_loff + len - 1;
638 struct extent_state *cached_state = NULL;
639 struct btrfs_fs_info *fs_info = src->root->fs_info;
640 const u64 bs = fs_info->sectorsize;
641 int ret;
642
643 /*
644 * Lock destination range to serialize with concurrent readahead(), and
645 * we are safe from concurrency with relocation of source extents
646 * because we have already locked the inode's i_mmap_lock in exclusive
647 * mode.
648 */
649 lock_extent(&dst->io_tree, dst_loff, end, &cached_state);
650 ret = btrfs_clone(&src->vfs_inode, &dst->vfs_inode, loff, len,
651 ALIGN(len, bs), dst_loff, 1);
652 unlock_extent(&dst->io_tree, dst_loff, end, &cached_state);
653
654 btrfs_btree_balance_dirty(fs_info);
655
656 return ret;
657 }
658
btrfs_extent_same(struct inode * src,u64 loff,u64 olen,struct inode * dst,u64 dst_loff)659 static int btrfs_extent_same(struct inode *src, u64 loff, u64 olen,
660 struct inode *dst, u64 dst_loff)
661 {
662 int ret = 0;
663 u64 i, tail_len, chunk_count;
664 struct btrfs_root *root_dst = BTRFS_I(dst)->root;
665
666 spin_lock(&root_dst->root_item_lock);
667 if (root_dst->send_in_progress) {
668 btrfs_warn_rl(root_dst->fs_info,
669 "cannot deduplicate to root %llu while send operations are using it (%d in progress)",
670 btrfs_root_id(root_dst),
671 root_dst->send_in_progress);
672 spin_unlock(&root_dst->root_item_lock);
673 return -EAGAIN;
674 }
675 root_dst->dedupe_in_progress++;
676 spin_unlock(&root_dst->root_item_lock);
677
678 tail_len = olen % BTRFS_MAX_DEDUPE_LEN;
679 chunk_count = div_u64(olen, BTRFS_MAX_DEDUPE_LEN);
680
681 for (i = 0; i < chunk_count; i++) {
682 ret = btrfs_extent_same_range(BTRFS_I(src), loff, BTRFS_MAX_DEDUPE_LEN,
683 BTRFS_I(dst), dst_loff);
684 if (ret)
685 goto out;
686
687 loff += BTRFS_MAX_DEDUPE_LEN;
688 dst_loff += BTRFS_MAX_DEDUPE_LEN;
689 }
690
691 if (tail_len > 0)
692 ret = btrfs_extent_same_range(BTRFS_I(src), loff, tail_len,
693 BTRFS_I(dst), dst_loff);
694 out:
695 spin_lock(&root_dst->root_item_lock);
696 root_dst->dedupe_in_progress--;
697 spin_unlock(&root_dst->root_item_lock);
698
699 return ret;
700 }
701
btrfs_clone_files(struct file * file,struct file * file_src,u64 off,u64 olen,u64 destoff)702 static noinline int btrfs_clone_files(struct file *file, struct file *file_src,
703 u64 off, u64 olen, u64 destoff)
704 {
705 struct extent_state *cached_state = NULL;
706 struct inode *inode = file_inode(file);
707 struct inode *src = file_inode(file_src);
708 struct btrfs_fs_info *fs_info = inode_to_fs_info(inode);
709 int ret;
710 int wb_ret;
711 u64 len = olen;
712 u64 bs = fs_info->sectorsize;
713 u64 end;
714
715 /*
716 * VFS's generic_remap_file_range_prep() protects us from cloning the
717 * eof block into the middle of a file, which would result in corruption
718 * if the file size is not blocksize aligned. So we don't need to check
719 * for that case here.
720 */
721 if (off + len == src->i_size)
722 len = ALIGN(src->i_size, bs) - off;
723
724 if (destoff > inode->i_size) {
725 const u64 wb_start = ALIGN_DOWN(inode->i_size, bs);
726
727 ret = btrfs_cont_expand(BTRFS_I(inode), inode->i_size, destoff);
728 if (ret)
729 return ret;
730 /*
731 * We may have truncated the last block if the inode's size is
732 * not sector size aligned, so we need to wait for writeback to
733 * complete before proceeding further, otherwise we can race
734 * with cloning and attempt to increment a reference to an
735 * extent that no longer exists (writeback completed right after
736 * we found the previous extent covering eof and before we
737 * attempted to increment its reference count).
738 */
739 ret = btrfs_wait_ordered_range(BTRFS_I(inode), wb_start,
740 destoff - wb_start);
741 if (ret)
742 return ret;
743 }
744
745 /*
746 * Lock destination range to serialize with concurrent readahead(), and
747 * we are safe from concurrency with relocation of source extents
748 * because we have already locked the inode's i_mmap_lock in exclusive
749 * mode.
750 */
751 end = destoff + len - 1;
752 lock_extent(&BTRFS_I(inode)->io_tree, destoff, end, &cached_state);
753 ret = btrfs_clone(src, inode, off, olen, len, destoff, 0);
754 unlock_extent(&BTRFS_I(inode)->io_tree, destoff, end, &cached_state);
755
756 /*
757 * We may have copied an inline extent into a page of the destination
758 * range, so wait for writeback to complete before truncating pages
759 * from the page cache. This is a rare case.
760 */
761 wb_ret = btrfs_wait_ordered_range(BTRFS_I(inode), destoff, len);
762 ret = ret ? ret : wb_ret;
763 /*
764 * Truncate page cache pages so that future reads will see the cloned
765 * data immediately and not the previous data.
766 */
767 truncate_inode_pages_range(&inode->i_data,
768 round_down(destoff, PAGE_SIZE),
769 round_up(destoff + len, PAGE_SIZE) - 1);
770
771 btrfs_btree_balance_dirty(fs_info);
772
773 return ret;
774 }
775
btrfs_remap_file_range_prep(struct file * file_in,loff_t pos_in,struct file * file_out,loff_t pos_out,loff_t * len,unsigned int remap_flags)776 static int btrfs_remap_file_range_prep(struct file *file_in, loff_t pos_in,
777 struct file *file_out, loff_t pos_out,
778 loff_t *len, unsigned int remap_flags)
779 {
780 struct btrfs_inode *inode_in = BTRFS_I(file_inode(file_in));
781 struct btrfs_inode *inode_out = BTRFS_I(file_inode(file_out));
782 u64 bs = inode_out->root->fs_info->sectorsize;
783 u64 wb_len;
784 int ret;
785
786 if (!(remap_flags & REMAP_FILE_DEDUP)) {
787 struct btrfs_root *root_out = inode_out->root;
788
789 if (btrfs_root_readonly(root_out))
790 return -EROFS;
791
792 ASSERT(inode_in->vfs_inode.i_sb == inode_out->vfs_inode.i_sb);
793 }
794
795 /* Don't make the dst file partly checksummed */
796 if ((inode_in->flags & BTRFS_INODE_NODATASUM) !=
797 (inode_out->flags & BTRFS_INODE_NODATASUM)) {
798 return -EINVAL;
799 }
800
801 /*
802 * Now that the inodes are locked, we need to start writeback ourselves
803 * and can not rely on the writeback from the VFS's generic helper
804 * generic_remap_file_range_prep() because:
805 *
806 * 1) For compression we must call filemap_fdatawrite_range() range
807 * twice (btrfs_fdatawrite_range() does it for us), and the generic
808 * helper only calls it once;
809 *
810 * 2) filemap_fdatawrite_range(), called by the generic helper only
811 * waits for the writeback to complete, i.e. for IO to be done, and
812 * not for the ordered extents to complete. We need to wait for them
813 * to complete so that new file extent items are in the fs tree.
814 */
815 if (*len == 0 && !(remap_flags & REMAP_FILE_DEDUP))
816 wb_len = ALIGN(inode_in->vfs_inode.i_size, bs) - ALIGN_DOWN(pos_in, bs);
817 else
818 wb_len = ALIGN(*len, bs);
819
820 /*
821 * Workaround to make sure NOCOW buffered write reach disk as NOCOW.
822 *
823 * Btrfs' back references do not have a block level granularity, they
824 * work at the whole extent level.
825 * NOCOW buffered write without data space reserved may not be able
826 * to fall back to CoW due to lack of data space, thus could cause
827 * data loss.
828 *
829 * Here we take a shortcut by flushing the whole inode, so that all
830 * nocow write should reach disk as nocow before we increase the
831 * reference of the extent. We could do better by only flushing NOCOW
832 * data, but that needs extra accounting.
833 *
834 * Also we don't need to check ASYNC_EXTENT, as async extent will be
835 * CoWed anyway, not affecting nocow part.
836 */
837 ret = filemap_flush(inode_in->vfs_inode.i_mapping);
838 if (ret < 0)
839 return ret;
840
841 ret = btrfs_wait_ordered_range(inode_in, ALIGN_DOWN(pos_in, bs), wb_len);
842 if (ret < 0)
843 return ret;
844 ret = btrfs_wait_ordered_range(inode_out, ALIGN_DOWN(pos_out, bs), wb_len);
845 if (ret < 0)
846 return ret;
847
848 return generic_remap_file_range_prep(file_in, pos_in, file_out, pos_out,
849 len, remap_flags);
850 }
851
file_sync_write(const struct file * file)852 static bool file_sync_write(const struct file *file)
853 {
854 if (file->f_flags & (__O_SYNC | O_DSYNC))
855 return true;
856 if (IS_SYNC(file_inode(file)))
857 return true;
858
859 return false;
860 }
861
btrfs_remap_file_range(struct file * src_file,loff_t off,struct file * dst_file,loff_t destoff,loff_t len,unsigned int remap_flags)862 loff_t btrfs_remap_file_range(struct file *src_file, loff_t off,
863 struct file *dst_file, loff_t destoff, loff_t len,
864 unsigned int remap_flags)
865 {
866 struct btrfs_inode *src_inode = BTRFS_I(file_inode(src_file));
867 struct btrfs_inode *dst_inode = BTRFS_I(file_inode(dst_file));
868 bool same_inode = dst_inode == src_inode;
869 int ret;
870
871 if (remap_flags & ~(REMAP_FILE_DEDUP | REMAP_FILE_ADVISORY))
872 return -EINVAL;
873
874 if (same_inode) {
875 btrfs_inode_lock(src_inode, BTRFS_ILOCK_MMAP);
876 } else {
877 lock_two_nondirectories(&src_inode->vfs_inode, &dst_inode->vfs_inode);
878 btrfs_double_mmap_lock(src_inode, dst_inode);
879 }
880
881 ret = btrfs_remap_file_range_prep(src_file, off, dst_file, destoff,
882 &len, remap_flags);
883 if (ret < 0 || len == 0)
884 goto out_unlock;
885
886 if (remap_flags & REMAP_FILE_DEDUP)
887 ret = btrfs_extent_same(&src_inode->vfs_inode, off, len,
888 &dst_inode->vfs_inode, destoff);
889 else
890 ret = btrfs_clone_files(dst_file, src_file, off, len, destoff);
891
892 out_unlock:
893 if (same_inode) {
894 btrfs_inode_unlock(src_inode, BTRFS_ILOCK_MMAP);
895 } else {
896 btrfs_double_mmap_unlock(src_inode, dst_inode);
897 unlock_two_nondirectories(&src_inode->vfs_inode,
898 &dst_inode->vfs_inode);
899 }
900
901 /*
902 * If either the source or the destination file was opened with O_SYNC,
903 * O_DSYNC or has the S_SYNC attribute, fsync both the destination and
904 * source files/ranges, so that after a successful return (0) followed
905 * by a power failure results in the reflinked data to be readable from
906 * both files/ranges.
907 */
908 if (ret == 0 && len > 0 &&
909 (file_sync_write(src_file) || file_sync_write(dst_file))) {
910 ret = btrfs_sync_file(src_file, off, off + len - 1, 0);
911 if (ret == 0)
912 ret = btrfs_sync_file(dst_file, destoff,
913 destoff + len - 1, 0);
914 }
915
916 return ret < 0 ? ret : len;
917 }
918