From: Nicholas Piggin <[email protected]>

virt_arch_pgd_alloc all do the same test and set of pgd_created. Move
this into common code.

Signed-off-by: Nicholas Piggin <[email protected]>
[Rebased to latest mainline tree]
Signed-off-by: Ritesh Harjani (IBM) <[email protected]>
---
 tools/testing/selftests/kvm/include/kvm_util.h        | 5 +++++
 tools/testing/selftests/kvm/lib/arm64/processor.c     | 4 ----
 tools/testing/selftests/kvm/lib/loongarch/processor.c | 4 ----
 tools/testing/selftests/kvm/lib/riscv/processor.c     | 4 ----
 tools/testing/selftests/kvm/lib/s390/processor.c      | 4 ----
 tools/testing/selftests/kvm/lib/x86/processor.c       | 9 +++------
 6 files changed, 8 insertions(+), 22 deletions(-)

diff --git a/tools/testing/selftests/kvm/include/kvm_util.h 
b/tools/testing/selftests/kvm/include/kvm_util.h
index 2ecaaa0e9965..3666a8530f31 100644
--- a/tools/testing/selftests/kvm/include/kvm_util.h
+++ b/tools/testing/selftests/kvm/include/kvm_util.h
@@ -1197,7 +1197,12 @@ void virt_arch_pgd_alloc(struct kvm_vm *vm);
 
 static inline void virt_pgd_alloc(struct kvm_vm *vm)
 {
+       if (vm->mmu.pgd_created)
+               return;
+
        virt_arch_pgd_alloc(vm);
+
+       vm->mmu.pgd_created = true;
 }
 
 /*
diff --git a/tools/testing/selftests/kvm/lib/arm64/processor.c 
b/tools/testing/selftests/kvm/lib/arm64/processor.c
index 01325bf4d36f..498fbcb0ea16 100644
--- a/tools/testing/selftests/kvm/lib/arm64/processor.c
+++ b/tools/testing/selftests/kvm/lib/arm64/processor.c
@@ -112,13 +112,9 @@ void virt_arch_pgd_alloc(struct kvm_vm *vm)
 {
        size_t nr_pages = vm_page_align(vm, ptrs_per_pgd(vm) * 8) / 
vm->page_size;
 
-       if (vm->mmu.pgd_created)
-               return;
-
        vm->mmu.pgd = vm_phy_pages_alloc(vm, nr_pages,
                                         KVM_GUEST_PAGE_TABLE_MIN_PADDR,
                                         vm->memslots[MEM_REGION_PT]);
-       vm->mmu.pgd_created = true;
 }
 
 static void _virt_pg_map(struct kvm_vm *vm, gva_t gva, gpa_t gpa,
diff --git a/tools/testing/selftests/kvm/lib/loongarch/processor.c 
b/tools/testing/selftests/kvm/lib/loongarch/processor.c
index 64d91fb76522..207055db5f5d 100644
--- a/tools/testing/selftests/kvm/lib/loongarch/processor.c
+++ b/tools/testing/selftests/kvm/lib/loongarch/processor.c
@@ -51,9 +51,6 @@ void virt_arch_pgd_alloc(struct kvm_vm *vm)
        int i;
        gpa_t child, table;
 
-       if (vm->mmu.pgd_created)
-               return;
-
        child = table = 0;
        for (i = 0; i < vm->mmu.pgtable_levels; i++) {
                invalid_pgtable[i] = child;
@@ -64,7 +61,6 @@ void virt_arch_pgd_alloc(struct kvm_vm *vm)
                child = table;
        }
        vm->mmu.pgd = table;
-       vm->mmu.pgd_created = true;
 }
 
 static int virt_pte_none(u64 *ptep, int level)
diff --git a/tools/testing/selftests/kvm/lib/riscv/processor.c 
b/tools/testing/selftests/kvm/lib/riscv/processor.c
index ded5429f3448..75a5d4c46001 100644
--- a/tools/testing/selftests/kvm/lib/riscv/processor.c
+++ b/tools/testing/selftests/kvm/lib/riscv/processor.c
@@ -66,13 +66,9 @@ void virt_arch_pgd_alloc(struct kvm_vm *vm)
 {
        size_t nr_pages = vm_page_align(vm, ptrs_per_pte(vm) * 8) / 
vm->page_size;
 
-       if (vm->mmu.pgd_created)
-               return;
-
        vm->mmu.pgd = vm_phy_pages_alloc(vm, nr_pages,
                                         KVM_GUEST_PAGE_TABLE_MIN_PADDR,
                                         vm->memslots[MEM_REGION_PT]);
-       vm->mmu.pgd_created = true;
 }
 
 void virt_arch_pg_map(struct kvm_vm *vm, gva_t gva, gpa_t gpa)
diff --git a/tools/testing/selftests/kvm/lib/s390/processor.c 
b/tools/testing/selftests/kvm/lib/s390/processor.c
index a9adb3782b35..342b7c92463e 100644
--- a/tools/testing/selftests/kvm/lib/s390/processor.c
+++ b/tools/testing/selftests/kvm/lib/s390/processor.c
@@ -17,16 +17,12 @@ void virt_arch_pgd_alloc(struct kvm_vm *vm)
        TEST_ASSERT(vm->page_size == PAGE_SIZE, "Unsupported page size: 0x%x",
                    vm->page_size);
 
-       if (vm->mmu.pgd_created)
-               return;
-
        gpa = vm_phy_pages_alloc(vm, PAGES_PER_REGION,
                                   KVM_GUEST_PAGE_TABLE_MIN_PADDR,
                                   vm->memslots[MEM_REGION_PT]);
        memset(addr_gpa2hva(vm, gpa), 0xff, PAGES_PER_REGION * vm->page_size);
 
        vm->mmu.pgd = gpa;
-       vm->mmu.pgd_created = true;
 }
 
 /*
diff --git a/tools/testing/selftests/kvm/lib/x86/processor.c 
b/tools/testing/selftests/kvm/lib/x86/processor.c
index b51467d70f6e..e420afdfbcfb 100644
--- a/tools/testing/selftests/kvm/lib/x86/processor.c
+++ b/tools/testing/selftests/kvm/lib/x86/processor.c
@@ -164,12 +164,9 @@ bool kvm_is_tdp_enabled(void)
 static void virt_mmu_init(struct kvm_vm *vm, struct kvm_mmu *mmu,
                          struct pte_masks *pte_masks)
 {
-       /* If needed, create the top-level page table. */
-       if (!mmu->pgd_created) {
-               mmu->pgd = vm_alloc_page_table(vm);
-               mmu->pgd_created = true;
-               mmu->arch.pte_masks = *pte_masks;
-       }
+       /* Create the top-level page table. */
+       mmu->pgd = vm_alloc_page_table(vm);
+       mmu->arch.pte_masks = *pte_masks;
 
        TEST_ASSERT(mmu->pgtable_levels == 4 || mmu->pgtable_levels == 5,
                    "Selftests MMU only supports 4-level and 5-level paging, 
not %u-level paging",
-- 
2.39.5


Reply via email to