diff --git a/arch/arm64/kvm/hyp/nvhe/mem_protect.c b/arch/arm64/kvm/hyp/nvhe/mem_protect.c index 95349b97e56e..bc1f8cb3faf3 100644 --- a/arch/arm64/kvm/hyp/nvhe/mem_protect.c +++ b/arch/arm64/kvm/hyp/nvhe/mem_protect.c @@ -389,7 +389,7 @@ static int relinquish_walker(const struct kvm_pgtable_visit_ctx *ctx, /* We don't support splitting non-leaf mappings */ if (ctx->level != (KVM_PGTABLE_MAX_LEVELS - 1)) - return 0; + return -E2BIG; state = pkvm_getstate(kvm_pgtable_stage2_pte_prot(pte)); if (state != data->expected_state) diff --git a/arch/arm64/kvm/hyp/nvhe/pkvm.c b/arch/arm64/kvm/hyp/nvhe/pkvm.c index fa4c40b0a172..98b0a173408f 100644 --- a/arch/arm64/kvm/hyp/nvhe/pkvm.c +++ b/arch/arm64/kvm/hyp/nvhe/pkvm.c @@ -1588,9 +1588,19 @@ static bool pkvm_memrelinquish_call(struct pkvm_hyp_vcpu *hyp_vcpu, goto out_guest_err; ret = __pkvm_guest_relinquish_to_host(hyp_vcpu, ipa, &pa); - if (ret == -ENOMEM) { - if (pkvm_handle_empty_memcache(hyp_vcpu, exit_code)) + if (ret == -E2BIG) { + struct kvm_hyp_req *req = pkvm_hyp_req_reserve(hyp_vcpu, KVM_HYP_REQ_TYPE_SPLIT); + + if (!req) { + ret = -ENOMEM; goto out_guest_err; + } + + req->split.guest_ipa = ALIGN_DOWN(ipa, PMD_SIZE); + req->split.size = PMD_SIZE; + + write_sysreg_el2(read_sysreg_el2(SYS_ELR) - 4, SYS_ELR); + *exit_code = ARM_EXCEPTION_HYP_REQ; return false; } else if (ret) {