Utilize the stage-2 library functions to initialize a s2_mmu, build a stage-2 page table, and turn on stage-2 translation for the nested guest. This better tests out the shadow page table code in KVM.
Signed-off-by: Wei-Lin Chang <[email protected]> --- .../selftests/kvm/arm64/shadow_stage2.c | 23 ++++++++++++++++--- 1 file changed, 20 insertions(+), 3 deletions(-) diff --git a/tools/testing/selftests/kvm/arm64/shadow_stage2.c b/tools/testing/selftests/kvm/arm64/shadow_stage2.c index 2b274b810dcf..5bce55abdea7 100644 --- a/tools/testing/selftests/kvm/arm64/shadow_stage2.c +++ b/tools/testing/selftests/kvm/arm64/shadow_stage2.c @@ -51,9 +51,11 @@ static void l2_guest_code(void) static void guest_code(void) { struct vcpu vcpu; + struct s2_mmu mmu; struct hyp_data hyp_data; int ret, i = 0; - gpa_t l2_pc, l2_stack_top; + gpa_t l2_pc, l2_stack_start, l2_stack_top, s2_pgd; + gpa_t do_hvc_gpa; struct page_pool pp; u64 mmfr0 = read_sysreg(id_aa64mmfr0_el1); @@ -68,11 +70,20 @@ static void guest_code(void) if (!has_tgran_2(mmfr0, pp.page_size)) GUEST_SYNC1(TGRAN2NOSUP); - l2_stack_top = alloc_page(&pp) + pp.page_size; + l2_stack_start = alloc_page(&pp); + l2_stack_top = l2_stack_start + pp.page_size; l2_pc = ucall_translate_to_gpa(l2_guest_code); + do_hvc_gpa = ucall_translate_to_gpa(do_hvc); + + s2_pgd = alloc_page(&pp); init_vcpu(&vcpu, l2_pc, l2_stack_top); - prepare_hyp_no_s2(); + init_s2_mmu(&mmu, 0, s2_pgd, pp.page_size, 40); + create_s2_mapping(&mmu, l2_pc, l2_pc, pp.page_size * 2, &pp); + create_s2_mapping(&mmu, do_hvc_gpa, do_hvc_gpa, pp.page_size, &pp); + create_s2_mapping(&mmu, l2_stack_start, l2_stack_start, pp.page_size, &pp); + + prepare_hyp(&mmu); while (true) { GUEST_PRINTF("L2 enter\n"); @@ -113,6 +124,12 @@ int main(void) vm_userspace_mem_region_add(vm, VM_MEM_SRC_ANONYMOUS, L2_PAGE_POOL_ADDR, L2_PAGE_POOL_MEMSLOT, L2_PAGE_POOL_NPAGES, 0); + /* + * This idmap allows L1 to traverse and build its guest stage-2, where + * it must do a PA to VA conversion in order to descend to the next + * level. + */ + virt_map(vm, L2_PAGE_POOL_ADDR, L2_PAGE_POOL_ADDR, L2_PAGE_POOL_NPAGES); while (true) { vcpu_run(vcpu); -- 2.43.0

