1 files changed, 8 insertions, 0 deletions

diff --git a/arch/x86/kvm/mmu.c b/arch/x86/kvm/mmu.c
index a5872b3c466d..dd4b95b3896b 100644
--- a/arch/x86/kvm/mmu.c
+++ b/arch/x86/kvm/mmu.c
@@ -626,6 +626,14 @@ static void rmap_write_protect(struct kvm *kvm, u64 gfn)
                 }
                 spte = rmap_next(kvm, rmapp, spte);
         }
+        if (write_protected) {
+                struct page *page;
+
+                spte = rmap_next(kvm, rmapp, NULL);
+                page = pfn_to_page((*spte & PT64_BASE_ADDR_MASK) >> PAGE_SHIFT);
+                SetPageDirty(page);
+        }
+
         /* check for huge page mappings */
         rmapp = gfn_to_rmap(kvm, gfn, 1);
         spte = rmap_next(kvm, rmapp, NULL);