21 static uint64_t page_align(struct kvm_vm *vm, uint64_t v)  in page_align()  argument
23 	return (v + vm->page_size) & ~(vm->page_size - 1);  in page_align()
26 static uint64_t pgd_index(struct kvm_vm *vm, vm_vaddr_t gva)  in pgd_index()  argument
28 	unsigned int shift = (vm->pgtable_levels - 1) * (vm->page_shift - 3) + vm->page_shift;  in pgd_index()
29 	uint64_t mask = (1UL << (vm->va_bits - shift)) - 1;  in pgd_index()
34 static uint64_t pud_index(struct kvm_vm *vm, vm_vaddr_t gva)  in pud_index()  argument
36 	unsigned int shift = 2 * (vm->page_shift - 3) + vm->page_shift;  in pud_index()
37 	uint64_t mask = (1UL << (vm->page_shift - 3)) - 1;  in pud_index()
39 	TEST_ASSERT(vm->pgtable_levels == 4,  in pud_index()
40 		"Mode %d does not have 4 page table levels", vm->mode);  in pud_index()
45 static uint64_t pmd_index(struct kvm_vm *vm, vm_vaddr_t gva)  in pmd_index()  argument
47 	unsigned int shift = (vm->page_shift - 3) + vm->page_shift;  in pmd_index()
48 	uint64_t mask = (1UL << (vm->page_shift - 3)) - 1;  in pmd_index()
50 	TEST_ASSERT(vm->pgtable_levels >= 3,  in pmd_index()
51 		"Mode %d does not have >= 3 page table levels", vm->mode);  in pmd_index()
56 static uint64_t pte_index(struct kvm_vm *vm, vm_vaddr_t gva)  in pte_index()  argument
58 	uint64_t mask = (1UL << (vm->page_shift - 3)) - 1;  in pte_index()
59 	return (gva >> vm->page_shift) & mask;  in pte_index()
62 static inline bool use_lpa2_pte_format(struct kvm_vm *vm)  in use_lpa2_pte_format()  argument
64 	return (vm->page_size == SZ_4K || vm->page_size == SZ_16K) &&  in use_lpa2_pte_format()
65 	    (vm->pa_bits > 48 || vm->va_bits > 48);  in use_lpa2_pte_format()
68 static uint64_t addr_pte(struct kvm_vm *vm, uint64_t pa, uint64_t attrs)  in addr_pte()  argument
72 	if (use_lpa2_pte_format(vm)) {  in addr_pte()
73 		pte = pa & GENMASK(49, vm->page_shift);  in addr_pte()
77 		pte = pa & GENMASK(47, vm->page_shift);  in addr_pte()
78 		if (vm->page_shift == 16)  in addr_pte()
86 static uint64_t pte_addr(struct kvm_vm *vm, uint64_t pte)  in pte_addr()  argument
90 	if (use_lpa2_pte_format(vm)) {  in pte_addr()
91 		pa = pte & GENMASK(49, vm->page_shift);  in pte_addr()
94 		pa = pte & GENMASK(47, vm->page_shift);  in pte_addr()
95 		if (vm->page_shift == 16)  in pte_addr()
102 static uint64_t ptrs_per_pgd(struct kvm_vm *vm)  in ptrs_per_pgd()  argument
104 	unsigned int shift = (vm->pgtable_levels - 1) * (vm->page_shift - 3) + vm->page_shift;  in ptrs_per_pgd()
105 	return 1 << (vm->va_bits - shift);  in ptrs_per_pgd()
108 static uint64_t __maybe_unused ptrs_per_pte(struct kvm_vm *vm)  in ptrs_per_pte()  argument
110 	return 1 << (vm->page_shift - 3);  in ptrs_per_pte()
113 void virt_arch_pgd_alloc(struct kvm_vm *vm)  in virt_arch_pgd_alloc()  argument
115 	size_t nr_pages = page_align(vm, ptrs_per_pgd(vm) * 8) / vm->page_size;  in virt_arch_pgd_alloc()
117 	if (vm->pgd_created)  in virt_arch_pgd_alloc()
120 	vm->pgd = vm_phy_pages_alloc(vm, nr_pages,  in virt_arch_pgd_alloc()
122 				     vm->memslots[MEM_REGION_PT]);  in virt_arch_pgd_alloc()
123 	vm->pgd_created = true;  in virt_arch_pgd_alloc()
126 static void _virt_pg_map(struct kvm_vm *vm, uint64_t vaddr, uint64_t paddr,  in _virt_pg_map()  argument
132 	TEST_ASSERT((vaddr % vm->page_size) == 0,  in _virt_pg_map()
134 		"  vaddr: 0x%lx vm->page_size: 0x%x", vaddr, vm->page_size);  in _virt_pg_map()
135 	TEST_ASSERT(sparsebit_is_set(vm->vpages_valid,  in _virt_pg_map()
136 		(vaddr >> vm->page_shift)),  in _virt_pg_map()
138 	TEST_ASSERT((paddr % vm->page_size) == 0,  in _virt_pg_map()
140 		"  paddr: 0x%lx vm->page_size: 0x%x", paddr, vm->page_size);  in _virt_pg_map()
141 	TEST_ASSERT((paddr >> vm->page_shift) <= vm->max_gfn,  in _virt_pg_map()
143 		"  paddr: 0x%lx vm->max_gfn: 0x%lx vm->page_size: 0x%x",  in _virt_pg_map()
144 		paddr, vm->max_gfn, vm->page_size);  in _virt_pg_map()
146 	ptep = addr_gpa2hva(vm, vm->pgd) + pgd_index(vm, vaddr) * 8;  in _virt_pg_map()
148 		*ptep = addr_pte(vm, vm_alloc_page_table(vm), 3);  in _virt_pg_map()
150 	switch (vm->pgtable_levels) {  in _virt_pg_map()
152 		ptep = addr_gpa2hva(vm, pte_addr(vm, *ptep)) + pud_index(vm, vaddr) * 8;  in _virt_pg_map()
154 			*ptep = addr_pte(vm, vm_alloc_page_table(vm), 3);  in _virt_pg_map()
157 		ptep = addr_gpa2hva(vm, pte_addr(vm, *ptep)) + pmd_index(vm, vaddr) * 8;  in _virt_pg_map()
159 			*ptep = addr_pte(vm, vm_alloc_page_table(vm), 3);  in _virt_pg_map()
162 		ptep = addr_gpa2hva(vm, pte_addr(vm, *ptep)) + pte_index(vm, vaddr) * 8;  in _virt_pg_map()
168 	*ptep = addr_pte(vm, paddr, (attr_idx << 2) | (1 << 10) | 3);  /* AF */  in _virt_pg_map()
171 void virt_arch_pg_map(struct kvm_vm *vm, uint64_t vaddr, uint64_t paddr)  in virt_arch_pg_map()  argument
175 	_virt_pg_map(vm, vaddr, paddr, attr_idx);  in virt_arch_pg_map()
178 uint64_t *virt_get_pte_hva(struct kvm_vm *vm, vm_vaddr_t gva)  in virt_get_pte_hva()  argument
182 	if (!vm->pgd_created)  in virt_get_pte_hva()
185 	ptep = addr_gpa2hva(vm, vm->pgd) + pgd_index(vm, gva) * 8;  in virt_get_pte_hva()
189 	switch (vm->pgtable_levels) {  in virt_get_pte_hva()
191 		ptep = addr_gpa2hva(vm, pte_addr(vm, *ptep)) + pud_index(vm, gva) * 8;  in virt_get_pte_hva()
196 		ptep = addr_gpa2hva(vm, pte_addr(vm, *ptep)) + pmd_index(vm, gva) * 8;  in virt_get_pte_hva()
201 		ptep = addr_gpa2hva(vm, pte_addr(vm, *ptep)) + pte_index(vm, gva) * 8;  in virt_get_pte_hva()
212 	TEST_FAIL("No mapping for vm virtual address, gva: 0x%lx", gva);  in virt_get_pte_hva()
216 vm_paddr_t addr_arch_gva2gpa(struct kvm_vm *vm, vm_vaddr_t gva)  in addr_arch_gva2gpa()  argument
218 	uint64_t *ptep = virt_get_pte_hva(vm, gva);  in addr_arch_gva2gpa()
220 	return pte_addr(vm, *ptep) + (gva & (vm->page_size - 1));  in addr_arch_gva2gpa()
223 static void pte_dump(FILE *stream, struct kvm_vm *vm, uint8_t indent, uint64_t page, int level)  in pte_dump()  argument
232 	for (pte = page; pte < page + ptrs_per_pte(vm) * 8; pte += 8) {  in pte_dump()
233 		ptep = addr_gpa2hva(vm, pte);  in pte_dump()
237 		pte_dump(stream, vm, indent + 1, pte_addr(vm, *ptep), level + 1);  in pte_dump()
242 void virt_arch_dump(FILE *stream, struct kvm_vm *vm, uint8_t indent)  in virt_arch_dump()  argument
244 	int level = 4 - (vm->pgtable_levels - 1);  in virt_arch_dump()
247 	if (!vm->pgd_created)  in virt_arch_dump()
250 	for (pgd = vm->pgd; pgd < vm->pgd + ptrs_per_pgd(vm) * 8; pgd += 8) {  in virt_arch_dump()
251 		ptep = addr_gpa2hva(vm, pgd);  in virt_arch_dump()
255 		pte_dump(stream, vm, indent + 1, pte_addr(vm, *ptep), level);  in virt_arch_dump()
262 	struct kvm_vm *vm = vcpu->vm;  in aarch64_vcpu_setup()  local
270 		vm_ioctl(vm, KVM_ARM_PREFERRED_TARGET, &preferred);  in aarch64_vcpu_setup()
286 	switch (vm->mode) {  in aarch64_vcpu_setup()
310 		TEST_FAIL("Unknown guest mode, mode: 0x%x", vm->mode);  in aarch64_vcpu_setup()
313 	ttbr0_el1 = vm->pgd & GENMASK(47, vm->page_shift);  in aarch64_vcpu_setup()
316 	switch (vm->mode) {  in aarch64_vcpu_setup()
321 		ttbr0_el1 |= FIELD_GET(GENMASK(51, 48), vm->pgd) << 2;  in aarch64_vcpu_setup()
340 		TEST_FAIL("Unknown guest mode, mode: 0x%x", vm->mode);  in aarch64_vcpu_setup()
346 	tcr_el1 |= (64 - vm->va_bits) /* T0SZ */;  in aarch64_vcpu_setup()
347 	if (use_lpa2_pte_format(vm))  in aarch64_vcpu_setup()
368 struct kvm_vcpu *aarch64_vcpu_add(struct kvm_vm *vm, uint32_t vcpu_id,  in aarch64_vcpu_add()  argument
373 	struct kvm_vcpu *vcpu = __vm_vcpu_add(vm, vcpu_id);  in aarch64_vcpu_add()
375 	stack_size = vm->page_size == 4096 ? DEFAULT_STACK_PGS * vm->page_size :  in aarch64_vcpu_add()
376 					     vm->page_size;  in aarch64_vcpu_add()
377 	stack_vaddr = __vm_vaddr_alloc(vm, stack_size,  in aarch64_vcpu_add()
389 struct kvm_vcpu *vm_arch_vcpu_add(struct kvm_vm *vm, uint32_t vcpu_id,  in vm_arch_vcpu_add()  argument
392 	return aarch64_vcpu_add(vm, vcpu_id, NULL, guest_code);  in vm_arch_vcpu_add()
482 void vm_init_descriptor_tables(struct kvm_vm *vm)  in vm_init_descriptor_tables()  argument
484 	vm->handlers = __vm_vaddr_alloc(vm, sizeof(struct handlers),  in vm_init_descriptor_tables()
485 					vm->page_size, MEM_REGION_DATA);  in vm_init_descriptor_tables()
487 	*(vm_vaddr_t *)addr_gva2hva(vm, (vm_vaddr_t)(&exception_handlers)) = vm->handlers;  in vm_init_descriptor_tables()
490 void vm_install_sync_handler(struct kvm_vm *vm, int vector, int ec,  in vm_install_sync_handler()  argument
493 	struct handlers *handlers = addr_gva2hva(vm, vm->handlers);  in vm_install_sync_handler()
501 void vm_install_exception_handler(struct kvm_vm *vm, int vector,  in vm_install_exception_handler()  argument
504 	struct handlers *handlers = addr_gva2hva(vm, vm->handlers);  in vm_install_exception_handler()
619 void vm_vaddr_populate_bitmap(struct kvm_vm *vm)  in vm_vaddr_populate_bitmap()  argument
625 	sparsebit_set_num(vm->vpages_valid, 0,  in vm_vaddr_populate_bitmap()
626 			  (1ULL << vm->va_bits) >> vm->page_shift);  in vm_vaddr_populate_bitmap()