177 		kvm_call_hyp_nvhe(__pkvm_tlb_flush_vmid, kvm->arch.pkvm.handle);  in kvm_arch_flush_remote_tlbs()
296 	return kvm_call_hyp_nvhe(__pkvm_host_unmap_guest, kvm->arch.pkvm.handle,  in __pkvm_unmap_guest_call()
313 	mtree_erase(&kvm->arch.pkvm.pinned_pages, ppage->ipa);  in pkvm_unmap_guest()
327 	mt_for_each(&kvm->arch.pkvm.pinned_pages, entry, index, end - 1) {  in pkvm_unmap_range()
394 	if (is_protected_kvm_enabled() && kvm->arch.pkvm.enabled)  in __unmap_stage2_range()
425 	mt_for_each(&kvm->arch.pkvm.pinned_pages, entry, index, ULONG_MAX) {  in pkvm_stage2_flush()
1010 	mt_init(&kvm->arch.pkvm.pinned_pages);  in kvm_init_stage2_mmu()
1256 	return kvm_call_hyp_nvhe(__pkvm_wrprotect, kvm->arch.pkvm.handle, pfn,  in __pkvm_wrprotect_call()
1265 	mt_for_each(&kvm->arch.pkvm.pinned_pages, entry, index, end - 1) {  in pkvm_wp_range()
1601 	mt_for_each(&kvm->arch.pkvm.pinned_pages, entry, index, ULONG_MAX)  in find_ppage_or_above()
1613 	return mtree_insert_range(&kvm->arch.pkvm.pinned_pages, start, end,  in insert_ppage()
1621 	return mt_find(&kvm->arch.pkvm.pinned_pages, &index, ipa + PAGE_SIZE - 1);  in find_ppage()
1705 	} else if (kvm->arch.pkvm.enabled && !PageSwapBacked(page)) {  in pkvm_mem_abort()
1747 	    mt_find(&kvm->arch.pkvm.pinned_pages, &index, index + page_size - 1)) {  in pkvm_mem_abort()
1810 			ppage = mt_next(&vcpu->kvm->arch.pkvm.pinned_pages,  in pkvm_mem_abort_range()
2486 		    kvm->arch.pkvm.handle && kvm->arch.pkvm.enabled) {  in kvm_arch_prepare_memory_region()
2490 		if (new && kvm->arch.pkvm.enabled &&  in kvm_arch_prepare_memory_region()