1 files changed, 8 insertions, 0 deletions

diff --git a/fs/iomap/direct-io.c b/fs/iomap/direct-io.c
index 933f234d5bec..70488904a778 100644
--- a/fs/iomap/direct-io.c
+++ b/fs/iomap/direct-io.c
@@ -6,6 +6,7 @@
 #include <linux/module.h>
 #include <linux/compiler.h>
 #include <linux/fs.h>
+#include <linux/fscrypt.h>
 #include <linux/iomap.h>
 #include <linux/backing-dev.h>
 #include <linux/uio.h>
@@ -185,11 +186,14 @@ static void
 iomap_dio_zero(struct iomap_dio *dio, struct iomap *iomap, loff_t pos,
                 unsigned len)
 {
+        struct inode *inode = file_inode(dio->iocb->ki_filp);
         struct page *page = ZERO_PAGE(0);
         int flags = REQ_SYNC | REQ_IDLE;
         struct bio *bio;
 
         bio = bio_alloc(GFP_KERNEL, 1);
+        fscrypt_set_bio_crypt_ctx(bio, inode, pos >> inode->i_blkbits,
+                                  GFP_KERNEL);
         bio_set_dev(bio, iomap->bdev);
         bio->bi_iter.bi_sector = iomap_sector(iomap, pos);
         bio->bi_private = dio;
@@ -255,6 +259,7 @@ iomap_dio_bio_actor(struct inode *inode, loff_t pos, loff_t length,
                 ret = nr_pages;
                 goto out;
         }
+        nr_pages = fscrypt_limit_dio_pages(inode, pos, nr_pages);
 
         if (need_zeroout) {
                 /* zero out from the start of the block to the write offset */
@@ -272,6 +277,8 @@ iomap_dio_bio_actor(struct inode *inode, loff_t pos, loff_t length,
                 }
 
                 bio = bio_alloc(GFP_KERNEL, nr_pages);
+                fscrypt_set_bio_crypt_ctx(bio, inode, pos >> inode->i_blkbits,
+                                          GFP_KERNEL);
                 bio_set_dev(bio, iomap->bdev);
                 bio->bi_iter.bi_sector = iomap_sector(iomap, pos);
                 bio->bi_write_hint = dio->iocb->ki_hint;
@@ -309,6 +316,7 @@ iomap_dio_bio_actor(struct inode *inode, loff_t pos, loff_t length,
                 copied += n;
 
                 nr_pages = iov_iter_npages(dio->submit.iter, BIO_MAX_PAGES);
+                nr_pages = fscrypt_limit_dio_pages(inode, pos, nr_pages);
                 iomap_dio_submit_bio(dio, iomap, bio, pos);
                 pos += n;
         } while (nr_pages);