3 files changed, 12 insertions, 5 deletions

diff --git a/fs/ext4/ext4.h b/fs/ext4/ext4.h
index 1d08a1b51bdd..aeda5061a59a 100644
--- a/fs/ext4/ext4.h
+++ b/fs/ext4/ext4.h
@@ -2772,7 +2772,8 @@ extern void ext4_io_submit(struct ext4_io_submit *io);
 extern int ext4_bio_write_page(struct ext4_io_submit *io,
                                struct page *page,
                                int len,
-                               struct writeback_control *wbc);
+                               struct writeback_control *wbc,
+                               bool keep_towrite);
 
 /* mmp.c */
 extern int ext4_multi_mount_protect(struct super_block *, ext4_fsblk_t);
diff --git a/fs/ext4/inode.c b/fs/ext4/inode.c
index 9b3c52fbe86d..04dd2de10796 100644
--- a/fs/ext4/inode.c
+++ b/fs/ext4/inode.c
@@ -1846,6 +1846,7 @@ static int ext4_writepage(struct page *page,
         struct buffer_head *page_bufs = NULL;
         struct inode *inode = page->mapping->host;
         struct ext4_io_submit io_submit;
+        bool keep_towrite = false;
 
         trace_ext4_writepage(page);
         size = i_size_read(inode);
@@ -1876,6 +1877,7 @@ static int ext4_writepage(struct page *page,
                         unlock_page(page);
                         return 0;
                 }
+                keep_towrite = true;
         }
 
         if (PageChecked(page) && ext4_should_journal_data(inode))
@@ -1892,7 +1894,7 @@ static int ext4_writepage(struct page *page,
                 unlock_page(page);
                 return -ENOMEM;
         }
-        ret = ext4_bio_write_page(&io_submit, page, len, wbc);
+        ret = ext4_bio_write_page(&io_submit, page, len, wbc, keep_towrite);
         ext4_io_submit(&io_submit);
         /* Drop io_end reference we got from init */
         ext4_put_io_end_defer(io_submit.io_end);
@@ -1911,7 +1913,7 @@ static int mpage_submit_page(struct mpage_da_data *mpd, struct page *page)
         else
                 len = PAGE_CACHE_SIZE;
         clear_page_dirty_for_io(page);
-        err = ext4_bio_write_page(&mpd->io_submit, page, len, mpd->wbc);
+        err = ext4_bio_write_page(&mpd->io_submit, page, len, mpd->wbc, false);
         if (!err)
                 mpd->wbc->nr_to_write--;
         mpd->first_page++;
diff --git a/fs/ext4/page-io.c b/fs/ext4/page-io.c
index c18d95b50540..4cb2743cb2e3 100644
--- a/fs/ext4/page-io.c
+++ b/fs/ext4/page-io.c
@@ -401,7 +401,8 @@ submit_and_retry:
 int ext4_bio_write_page(struct ext4_io_submit *io,
                         struct page *page,
                         int len,
-                        struct writeback_control *wbc)
+                        struct writeback_control *wbc,
+                        bool keep_towrite)
 {
         struct inode *inode = page->mapping->host;
         unsigned block_start, blocksize;
@@ -414,7 +415,10 @@ int ext4_bio_write_page(struct ext4_io_submit *io,
         BUG_ON(!PageLocked(page));
         BUG_ON(PageWriteback(page));
 
-        set_page_writeback(page);
+        if (keep_towrite)
+                set_page_writeback_keepwrite(page);
+        else
+                set_page_writeback(page);
         ClearPageError(page);
 
         /*