From: Omar Sandoval <osan...@fb.com>

Currently, we only create ordered extents when ram_bytes == num_bytes
and offset == 0. However, RWF_ENCODED writes may create extents which
only refer to a subset of the full unencoded extent, so we need to plumb
these fields through the ordered extent infrastructure and pass them
down to insert_reserved_file_extent().

Since we're changing the btrfs_add_ordered_extent* signature, let's get
rid of the trivial wrappers and add a kernel-doc.

Reviewed-by: Nikolay Borisov <nbori...@suse.com>
Reviewed-by: Josef Bacik <jo...@toxicpanda.com>
Signed-off-by: Omar Sandoval <osan...@fb.com>
---
 fs/btrfs/inode.c        | 56 ++++++++++++++++++---------------
 fs/btrfs/ordered-data.c | 68 ++++++++++++++++-------------------------
 fs/btrfs/ordered-data.h | 16 ++++------
 3 files changed, 64 insertions(+), 76 deletions(-)

diff --git a/fs/btrfs/inode.c b/fs/btrfs/inode.c
index b92c4a1d6d2f..139d690f171e 100644
--- a/fs/btrfs/inode.c
+++ b/fs/btrfs/inode.c
@@ -912,13 +912,12 @@ static noinline void submit_compressed_extents(struct 
async_chunk *async_chunk)
                        goto out_free_reserve;
                free_extent_map(em);
 
-               ret = btrfs_add_ordered_extent_compress(inode,
-                                               async_extent->start,
-                                               ins.objectid,
-                                               async_extent->ram_size,
-                                               ins.offset,
-                                               BTRFS_ORDERED_COMPRESSED,
-                                               async_extent->compress_type);
+               ret = btrfs_add_ordered_extent(inode, async_extent->start,
+                                              async_extent->ram_size,
+                                              async_extent->ram_size,
+                                              ins.objectid, ins.offset, 0,
+                                              1 << BTRFS_ORDERED_COMPRESSED,
+                                              async_extent->compress_type);
                if (ret) {
                        btrfs_drop_extent_cache(inode, async_extent->start,
                                                async_extent->start +
@@ -1126,8 +1125,9 @@ static noinline int cow_file_range(struct btrfs_inode 
*inode,
                }
                free_extent_map(em);
 
-               ret = btrfs_add_ordered_extent(inode, start, ins.objectid,
-                                              ram_size, cur_alloc_size, 0);
+               ret = btrfs_add_ordered_extent(inode, start, ram_size, ram_size,
+                                              ins.objectid, cur_alloc_size, 0,
+                                              0, BTRFS_COMPRESS_NONE);
                if (ret)
                        goto out_drop_extent_cache;
 
@@ -1768,10 +1768,11 @@ static noinline int run_delalloc_nocow(struct 
btrfs_inode *inode,
                                goto error;
                        }
                        free_extent_map(em);
-                       ret = btrfs_add_ordered_extent(inode, cur_offset,
-                                                      disk_bytenr, num_bytes,
-                                                      num_bytes,
-                                                      BTRFS_ORDERED_PREALLOC);
+                       ret = btrfs_add_ordered_extent(inode,
+                                       cur_offset, num_bytes, num_bytes,
+                                       disk_bytenr, num_bytes, 0,
+                                       1 << BTRFS_ORDERED_PREALLOC,
+                                       BTRFS_COMPRESS_NONE);
                        if (ret) {
                                btrfs_drop_extent_cache(inode, cur_offset,
                                                        cur_offset + num_bytes 
- 1,
@@ -1780,9 +1781,11 @@ static noinline int run_delalloc_nocow(struct 
btrfs_inode *inode,
                        }
                } else {
                        ret = btrfs_add_ordered_extent(inode, cur_offset,
+                                                      num_bytes, num_bytes,
                                                       disk_bytenr, num_bytes,
-                                                      num_bytes,
-                                                      BTRFS_ORDERED_NOCOW);
+                                                      0,
+                                                      1 << BTRFS_ORDERED_NOCOW,
+                                                      BTRFS_COMPRESS_NONE);
                        if (ret)
                                goto error;
                }
@@ -2586,6 +2589,7 @@ static int insert_reserved_file_extent(struct 
btrfs_trans_handle *trans,
        struct btrfs_key ins;
        u64 disk_num_bytes = btrfs_stack_file_extent_disk_num_bytes(stack_fi);
        u64 disk_bytenr = btrfs_stack_file_extent_disk_bytenr(stack_fi);
+       u64 offset = btrfs_stack_file_extent_offset(stack_fi);
        u64 num_bytes = btrfs_stack_file_extent_num_bytes(stack_fi);
        u64 ram_bytes = btrfs_stack_file_extent_ram_bytes(stack_fi);
        struct btrfs_drop_extents_args drop_args = { 0 };
@@ -2660,7 +2664,8 @@ static int insert_reserved_file_extent(struct 
btrfs_trans_handle *trans,
                goto out;
 
        ret = btrfs_alloc_reserved_file_extent(trans, root, btrfs_ino(inode),
-                                              file_pos, qgroup_reserved, &ins);
+                                              file_pos - offset,
+                                              qgroup_reserved, &ins);
 out:
        btrfs_free_path(path);
 
@@ -2686,20 +2691,20 @@ static int insert_ordered_extent_file_extent(struct 
btrfs_trans_handle *trans,
                                             struct btrfs_ordered_extent *oe)
 {
        struct btrfs_file_extent_item stack_fi;
-       u64 logical_len;
        bool update_inode_bytes;
+       u64 num_bytes = oe->num_bytes;
+       u64 ram_bytes = oe->ram_bytes;
 
        memset(&stack_fi, 0, sizeof(stack_fi));
        btrfs_set_stack_file_extent_type(&stack_fi, BTRFS_FILE_EXTENT_REG);
        btrfs_set_stack_file_extent_disk_bytenr(&stack_fi, oe->disk_bytenr);
        btrfs_set_stack_file_extent_disk_num_bytes(&stack_fi,
                                                   oe->disk_num_bytes);
+       btrfs_set_stack_file_extent_offset(&stack_fi, oe->offset);
        if (test_bit(BTRFS_ORDERED_TRUNCATED, &oe->flags))
-               logical_len = oe->truncated_len;
-       else
-               logical_len = oe->num_bytes;
-       btrfs_set_stack_file_extent_num_bytes(&stack_fi, logical_len);
-       btrfs_set_stack_file_extent_ram_bytes(&stack_fi, logical_len);
+               num_bytes = ram_bytes = oe->truncated_len;
+       btrfs_set_stack_file_extent_num_bytes(&stack_fi, num_bytes);
+       btrfs_set_stack_file_extent_ram_bytes(&stack_fi, ram_bytes);
        btrfs_set_stack_file_extent_compression(&stack_fi, oe->compress_type);
        /* Encryption and other encoding is reserved and all 0 */
 
@@ -7053,8 +7058,11 @@ static struct extent_map *btrfs_create_dio_extent(struct 
btrfs_inode *inode,
                if (IS_ERR(em))
                        goto out;
        }
-       ret = btrfs_add_ordered_extent_dio(inode, start, block_start, len,
-                                          block_len, type);
+       ret = btrfs_add_ordered_extent(inode, start, len, len, block_start,
+                                      block_len, 0,
+                                      (1 << type) |
+                                      (1 << BTRFS_ORDERED_DIRECT),
+                                      BTRFS_COMPRESS_NONE);
        if (ret) {
                if (em) {
                        free_extent_map(em);
diff --git a/fs/btrfs/ordered-data.c b/fs/btrfs/ordered-data.c
index d5d326c674b1..e11a517fab10 100644
--- a/fs/btrfs/ordered-data.c
+++ b/fs/btrfs/ordered-data.c
@@ -153,16 +153,27 @@ static inline struct rb_node *tree_search(struct 
btrfs_ordered_inode_tree *tree,
        return ret;
 }
 
-/*
- * Allocate and add a new ordered_extent into the per-inode tree.
+/**
+ * btrfs_add_ordered_extent - Add an ordered extent to the per-inode tree.
+ * @inode: inode that this extent is for.
+ * @file_offset: Logical offset in file where the extent starts.
+ * @num_bytes: Logical length of extent in file.
+ * @ram_bytes: Full length of unencoded data.
+ * @disk_bytenr: Offset of extent on disk.
+ * @disk_num_bytes: Size of extent on disk.
+ * @offset: Offset into unencoded data where file data starts.
+ * @flags: Flags specifying type of extent (1 << BTRFS_ORDERED_*).
+ * @compress_type: Compression algorithm used for data.
  *
- * The tree is given a single reference on the ordered extent that was
- * inserted.
+ * Most of these parameters correspond to &struct btrfs_file_extent_item. The
+ * tree is given a single reference on the ordered extent that was inserted.
+ *
+ * Return: 0 or -ENOMEM.
  */
-static int __btrfs_add_ordered_extent(struct btrfs_inode *inode, u64 
file_offset,
-                                     u64 disk_bytenr, u64 num_bytes,
-                                     u64 disk_num_bytes, int type, int dio,
-                                     int compress_type)
+int btrfs_add_ordered_extent(struct btrfs_inode *inode, u64 file_offset,
+                            u64 num_bytes, u64 ram_bytes, u64 disk_bytenr,
+                            u64 disk_num_bytes, u64 offset, int flags,
+                            int compress_type)
 {
        struct btrfs_root *root = inode->root;
        struct btrfs_fs_info *fs_info = root->fs_info;
@@ -171,7 +182,8 @@ static int __btrfs_add_ordered_extent(struct btrfs_inode 
*inode, u64 file_offset
        struct btrfs_ordered_extent *entry;
        int ret;
 
-       if (type == BTRFS_ORDERED_NOCOW || type == BTRFS_ORDERED_PREALLOC) {
+       if (flags &
+           ((1 << BTRFS_ORDERED_NOCOW) | (1 << BTRFS_ORDERED_PREALLOC))) {
                /* For nocow write, we can release the qgroup rsv right now */
                ret = btrfs_qgroup_free_data(inode, NULL, file_offset, 
num_bytes);
                if (ret < 0)
@@ -191,21 +203,21 @@ static int __btrfs_add_ordered_extent(struct btrfs_inode 
*inode, u64 file_offset
                return -ENOMEM;
 
        entry->file_offset = file_offset;
-       entry->disk_bytenr = disk_bytenr;
        entry->num_bytes = num_bytes;
+       entry->ram_bytes = ram_bytes;
+       entry->disk_bytenr = disk_bytenr;
        entry->disk_num_bytes = disk_num_bytes;
+       entry->offset = offset;
        entry->bytes_left = num_bytes;
        entry->inode = igrab(&inode->vfs_inode);
        entry->compress_type = compress_type;
        entry->truncated_len = (u64)-1;
        entry->qgroup_rsv = ret;
-       if (type != BTRFS_ORDERED_IO_DONE && type != BTRFS_ORDERED_COMPLETE)
-               set_bit(type, &entry->flags);
 
-       if (dio) {
+       entry->flags = flags;
+       if (flags & (1 << BTRFS_ORDERED_DIRECT)) {
                percpu_counter_add_batch(&fs_info->dio_bytes, num_bytes,
                                         fs_info->delalloc_batch);
-               set_bit(BTRFS_ORDERED_DIRECT, &entry->flags);
        }
 
        /* one ref for the tree */
@@ -252,34 +264,6 @@ static int __btrfs_add_ordered_extent(struct btrfs_inode 
*inode, u64 file_offset
        return 0;
 }
 
-int btrfs_add_ordered_extent(struct btrfs_inode *inode, u64 file_offset,
-                            u64 disk_bytenr, u64 num_bytes, u64 disk_num_bytes,
-                            int type)
-{
-       return __btrfs_add_ordered_extent(inode, file_offset, disk_bytenr,
-                                         num_bytes, disk_num_bytes, type, 0,
-                                         BTRFS_COMPRESS_NONE);
-}
-
-int btrfs_add_ordered_extent_dio(struct btrfs_inode *inode, u64 file_offset,
-                                u64 disk_bytenr, u64 num_bytes,
-                                u64 disk_num_bytes, int type)
-{
-       return __btrfs_add_ordered_extent(inode, file_offset, disk_bytenr,
-                                         num_bytes, disk_num_bytes, type, 1,
-                                         BTRFS_COMPRESS_NONE);
-}
-
-int btrfs_add_ordered_extent_compress(struct btrfs_inode *inode, u64 
file_offset,
-                                     u64 disk_bytenr, u64 num_bytes,
-                                     u64 disk_num_bytes, int type,
-                                     int compress_type)
-{
-       return __btrfs_add_ordered_extent(inode, file_offset, disk_bytenr,
-                                         num_bytes, disk_num_bytes, type, 0,
-                                         compress_type);
-}
-
 /*
  * Add a struct btrfs_ordered_sum into the list of checksums to be inserted
  * when an ordered extent is finished.  If the list covers more than one
diff --git a/fs/btrfs/ordered-data.h b/fs/btrfs/ordered-data.h
index 46194c2c05d4..8c8d6ed7a350 100644
--- a/fs/btrfs/ordered-data.h
+++ b/fs/btrfs/ordered-data.h
@@ -72,9 +72,11 @@ struct btrfs_ordered_extent {
         * These fields directly correspond to the same fields in
         * btrfs_file_extent_item.
         */
-       u64 disk_bytenr;
        u64 num_bytes;
+       u64 ram_bytes;
+       u64 disk_bytenr;
        u64 disk_num_bytes;
+       u64 offset;
 
        /* number of bytes that still need writing */
        u64 bytes_left;
@@ -160,15 +162,9 @@ bool btrfs_dec_test_first_ordered_pending(struct 
btrfs_inode *inode,
                                   u64 *file_offset, u64 io_size,
                                   int uptodate);
 int btrfs_add_ordered_extent(struct btrfs_inode *inode, u64 file_offset,
-                            u64 disk_bytenr, u64 num_bytes, u64 disk_num_bytes,
-                            int type);
-int btrfs_add_ordered_extent_dio(struct btrfs_inode *inode, u64 file_offset,
-                                u64 disk_bytenr, u64 num_bytes,
-                                u64 disk_num_bytes, int type);
-int btrfs_add_ordered_extent_compress(struct btrfs_inode *inode, u64 
file_offset,
-                                     u64 disk_bytenr, u64 num_bytes,
-                                     u64 disk_num_bytes, int type,
-                                     int compress_type);
+                            u64 num_bytes, u64 ram_bytes, u64 disk_bytenr,
+                            u64 disk_num_bytes, u64 offset, int flags,
+                            int compress_type);
 void btrfs_add_ordered_sum(struct btrfs_ordered_extent *entry,
                           struct btrfs_ordered_sum *sum);
 struct btrfs_ordered_extent *btrfs_lookup_ordered_extent(struct btrfs_inode 
*inode,
-- 
2.30.2

Reply via email to