We're going to be getting fscrypt_info from the extent maps, update the
helpers to take an fscrypt_info argument and use that to set the
encryption type on the ordered extent.

Signed-off-by: Josef Bacik <jo...@toxicpanda.com>
---
 fs/btrfs/inode.c        | 20 +++++++++++---------
 fs/btrfs/ordered-data.c | 32 ++++++++++++++++++++------------
 fs/btrfs/ordered-data.h |  9 +++++----
 3 files changed, 36 insertions(+), 25 deletions(-)

diff --git a/fs/btrfs/inode.c b/fs/btrfs/inode.c
index 19087fd68cfe..a1fa5b6f3790 100644
--- a/fs/btrfs/inode.c
+++ b/fs/btrfs/inode.c
@@ -1162,7 +1162,8 @@ static void submit_one_async_extent(struct async_chunk 
*async_chunk,
        }
        free_extent_map(em);
 
-       ordered = btrfs_alloc_ordered_extent(inode, start,      /* file_offset 
*/
+       ordered = btrfs_alloc_ordered_extent(inode, NULL,
+                                      start,                   /* file_offset 
*/
                                       async_extent->ram_size,  /* num_bytes */
                                       async_extent->ram_size,  /* ram_bytes */
                                       ins.objectid,            /* disk_bytenr 
*/
@@ -1425,9 +1426,10 @@ static noinline int cow_file_range(struct btrfs_inode 
*inode,
                }
                free_extent_map(em);
 
-               ordered = btrfs_alloc_ordered_extent(inode, start, ram_size,
-                                       ram_size, ins.objectid, cur_alloc_size,
-                                       0, 1 << BTRFS_ORDERED_REGULAR,
+               ordered = btrfs_alloc_ordered_extent(inode, NULL,
+                                       start, ram_size, ram_size, ins.objectid,
+                                       cur_alloc_size, 0,
+                                       1 << BTRFS_ORDERED_REGULAR,
                                        BTRFS_COMPRESS_NONE);
                if (IS_ERR(ordered)) {
                        ret = PTR_ERR(ordered);
@@ -2158,7 +2160,7 @@ static noinline int run_delalloc_nocow(struct btrfs_inode 
*inode,
                        free_extent_map(em);
                }
 
-               ordered = btrfs_alloc_ordered_extent(inode, cur_offset,
+               ordered = btrfs_alloc_ordered_extent(inode, NULL, cur_offset,
                                nocow_args.num_bytes, nocow_args.num_bytes,
                                nocow_args.disk_bytenr, nocow_args.num_bytes, 0,
                                is_prealloc
@@ -7040,7 +7042,7 @@ static struct extent_map *btrfs_create_dio_extent(struct 
btrfs_inode *inode,
                if (IS_ERR(em))
                        goto out;
        }
-       ordered = btrfs_alloc_ordered_extent(inode, start, len, len,
+       ordered = btrfs_alloc_ordered_extent(inode, NULL, start, len, len,
                                             block_start, block_len, 0,
                                             (1 << type) |
                                             (1 << BTRFS_ORDERED_DIRECT),
@@ -10512,9 +10514,9 @@ ssize_t btrfs_do_encoded_write(struct kiocb *iocb, 
struct iov_iter *from,
        }
        free_extent_map(em);
 
-       ordered = btrfs_alloc_ordered_extent(inode, start, num_bytes, ram_bytes,
-                                      ins.objectid, ins.offset,
-                                      encoded->unencoded_offset,
+       ordered = btrfs_alloc_ordered_extent(inode, NULL, start,
+                                      num_bytes, ram_bytes, ins.objectid,
+                                      ins.offset, encoded->unencoded_offset,
                                       (1 << BTRFS_ORDERED_ENCODED) |
                                       (1 << BTRFS_ORDERED_COMPRESSED),
                                       compression);
diff --git a/fs/btrfs/ordered-data.c b/fs/btrfs/ordered-data.c
index 27350dd50828..ee3138a6d11e 100644
--- a/fs/btrfs/ordered-data.c
+++ b/fs/btrfs/ordered-data.c
@@ -146,9 +146,11 @@ static inline struct rb_node *ordered_tree_search(struct 
btrfs_inode *inode,
 }
 
 static struct btrfs_ordered_extent *alloc_ordered_extent(
-                       struct btrfs_inode *inode, u64 file_offset, u64 
num_bytes,
-                       u64 ram_bytes, u64 disk_bytenr, u64 disk_num_bytes,
-                       u64 offset, unsigned long flags, int compress_type)
+                       struct btrfs_inode *inode,
+                       struct fscrypt_extent_info *fscrypt_info,
+                       u64 file_offset, u64 num_bytes, u64 ram_bytes,
+                       u64 disk_bytenr, u64 disk_num_bytes, u64 offset,
+                       unsigned long flags, int compress_type)
 {
        struct btrfs_ordered_extent *entry;
        int ret;
@@ -181,10 +183,12 @@ static struct btrfs_ordered_extent *alloc_ordered_extent(
        entry->bytes_left = num_bytes;
        entry->inode = igrab(&inode->vfs_inode);
        entry->compress_type = compress_type;
-       entry->encryption_type = BTRFS_ENCRYPTION_NONE;
        entry->truncated_len = (u64)-1;
        entry->qgroup_rsv = ret;
        entry->flags = flags;
+       entry->fscrypt_info = fscrypt_get_extent_info(fscrypt_info);
+       entry->encryption_type = entry->fscrypt_info ?
+               BTRFS_ENCRYPTION_FSCRYPT : BTRFS_ENCRYPTION_NONE;
        refcount_set(&entry->refs, 1);
        init_waitqueue_head(&entry->wait);
        INIT_LIST_HEAD(&entry->list);
@@ -247,6 +251,7 @@ static void insert_ordered_extent(struct 
btrfs_ordered_extent *entry)
  * Add an ordered extent to the per-inode tree.
  *
  * @inode:           Inode that this extent is for.
+ * @fscrypt_info:    The fscrypt_extent_info for this extent, if necessary.
  * @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.
@@ -263,17 +268,19 @@ static void insert_ordered_extent(struct 
btrfs_ordered_extent *entry)
  * Return: the new ordered extent or error pointer.
  */
 struct btrfs_ordered_extent *btrfs_alloc_ordered_extent(
-                       struct btrfs_inode *inode, u64 file_offset,
-                       u64 num_bytes, u64 ram_bytes, u64 disk_bytenr,
-                       u64 disk_num_bytes, u64 offset, unsigned long flags,
-                       int compress_type)
+                       struct btrfs_inode *inode,
+                       struct fscrypt_extent_info *fscrypt_info,
+                       u64 file_offset, u64 num_bytes, u64 ram_bytes,
+                       u64 disk_bytenr, u64 disk_num_bytes, u64 offset,
+                       unsigned long flags, int compress_type)
 {
        struct btrfs_ordered_extent *entry;
 
        ASSERT((flags & ~BTRFS_ORDERED_TYPE_FLAGS) == 0);
 
-       entry = alloc_ordered_extent(inode, file_offset, num_bytes, ram_bytes,
-                                    disk_bytenr, disk_num_bytes, offset, flags,
+       entry = alloc_ordered_extent(inode, fscrypt_info, file_offset,
+                                    num_bytes, ram_bytes, disk_bytenr,
+                                    disk_num_bytes, offset, flags,
                                     compress_type);
        if (!IS_ERR(entry))
                insert_ordered_extent(entry);
@@ -1166,8 +1173,9 @@ struct btrfs_ordered_extent *btrfs_split_ordered_extent(
        if (WARN_ON_ONCE(ordered->disk_num_bytes != ordered->num_bytes))
                return ERR_PTR(-EINVAL);
 
-       new = alloc_ordered_extent(inode, file_offset, len, len, disk_bytenr,
-                                  len, 0, flags, ordered->compress_type);
+       new = alloc_ordered_extent(inode, ordered->fscrypt_info, file_offset,
+                                  len, len, disk_bytenr, len, 0, flags,
+                                  ordered->compress_type);
        if (IS_ERR(new))
                return new;
 
diff --git a/fs/btrfs/ordered-data.h b/fs/btrfs/ordered-data.h
index cc422bdb5363..767efcd74bee 100644
--- a/fs/btrfs/ordered-data.h
+++ b/fs/btrfs/ordered-data.h
@@ -170,10 +170,11 @@ bool btrfs_dec_test_ordered_pending(struct btrfs_inode 
*inode,
                                    struct btrfs_ordered_extent **cached,
                                    u64 file_offset, u64 io_size);
 struct btrfs_ordered_extent *btrfs_alloc_ordered_extent(
-                       struct btrfs_inode *inode, u64 file_offset,
-                       u64 num_bytes, u64 ram_bytes, u64 disk_bytenr,
-                       u64 disk_num_bytes, u64 offset, unsigned long flags,
-                       int compress_type);
+                       struct btrfs_inode *inode,
+                       struct fscrypt_extent_info *fscrypt_info,
+                       u64 file_offset, u64 num_bytes, u64 ram_bytes,
+                       u64 disk_bytenr, u64 disk_num_bytes, u64 offset,
+                       unsigned long 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.41.0

Reply via email to