Apply d->memslot->partial_roe_bitmap to shadow page table entries
too.
Signed-off-by: Ahmed Abd El Mawgood <ahmedsoliman@xxxxxxxxxxx>
---
arch/x86/kvm/roe.c | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/arch/x86/kvm/roe.c b/arch/x86/kvm/roe.c
index f787106be8..700f69823b 100644
--- a/arch/x86/kvm/roe.c
+++ b/arch/x86/kvm/roe.c
@@ -25,11 +25,14 @@ static bool __rmap_write_protect_roe(struct kvm *kvm,
struct rmap_iterator iter;
bool prot;
bool flush = false;
+ void *full_bmp = memslot->roe_bitmap;
+ void *part_bmp = memslot->partial_roe_bitmap;
for_each_rmap_spte(rmap_head, &iter, sptep) {
int idx = spte_to_gfn(sptep) - memslot->base_gfn;
- prot = !test_bit(idx, memslot->roe_bitmap) && pt_protect;
+ prot = !(test_bit(idx, full_bmp) || test_bit(idx, part_bmp));
+ prot = prot && pt_protect;
flush |= spte_write_protect(sptep, prot);
}
return flush;
--
2.19.2
