Commit 43fdcda9 authored by Junaid Shahid's avatar Junaid Shahid Committed by Paolo Bonzini
Browse files

kvm: x86: Do not release the page inside mmu_set_spte() 



Release the page at the call-site where it was originally acquired.
This makes the exit code cleaner for most call sites, since they
do not need to duplicate code between success and the failure
label.

Signed-off-by: default avatarJunaid Shahid <junaids@google.com>
Signed-off-by: default avatarPaolo Bonzini <pbonzini@redhat.com>
parent 60cec433

arch/x86/kvm/mmu.c

+7 −11
Original line number Diff line number Diff line
@@ -3095,8 +3095,6 @@ static int mmu_set_spte(struct kvm_vcpu *vcpu, u64 *sptep, unsigned pte_access, 
		}

	}



	kvm_release_pfn_clean(pfn);



	return ret;

}
@@ -3131,9 +3129,11 @@ static int direct_pte_prefetch_many(struct kvm_vcpu *vcpu, 
	if (ret <= 0)

		return -1;



	for (i = 0; i < ret; i++, gfn++, start++)

	for (i = 0; i < ret; i++, gfn++, start++) {

		mmu_set_spte(vcpu, start, access, 0, sp->role.level, gfn,

			     page_to_pfn(pages[i]), true, true);

		put_page(pages[i]);

	}



	return 0;

}
@@ -3530,6 +3530,7 @@ static int nonpaging_map(struct kvm_vcpu *vcpu, gva_t v, u32 error_code, 
	if (handle_abnormal_pfn(vcpu, v, gfn, pfn, ACC_ALL, &r))

		return r;



	r = RET_PF_RETRY;

	spin_lock(&vcpu->kvm->mmu_lock);

	if (mmu_notifier_retry(vcpu->kvm, mmu_seq))

		goto out_unlock;
@@ -3538,14 +3539,11 @@ static int nonpaging_map(struct kvm_vcpu *vcpu, gva_t v, u32 error_code, 
	if (likely(!force_pt_level))

		transparent_hugepage_adjust(vcpu, &gfn, &pfn, &level);

	r = __direct_map(vcpu, write, map_writable, level, gfn, pfn, prefault);

	spin_unlock(&vcpu->kvm->mmu_lock);



	return r;



out_unlock:

	spin_unlock(&vcpu->kvm->mmu_lock);

	kvm_release_pfn_clean(pfn);

	return RET_PF_RETRY;

	return r;

}



static void mmu_free_root_page(struct kvm *kvm, hpa_t *root_hpa,
@@ -4159,6 +4157,7 @@ static int tdp_page_fault(struct kvm_vcpu *vcpu, gva_t gpa, u32 error_code, 
	if (handle_abnormal_pfn(vcpu, 0, gfn, pfn, ACC_ALL, &r))

		return r;



	r = RET_PF_RETRY;

	spin_lock(&vcpu->kvm->mmu_lock);

	if (mmu_notifier_retry(vcpu->kvm, mmu_seq))

		goto out_unlock;
@@ -4167,14 +4166,11 @@ static int tdp_page_fault(struct kvm_vcpu *vcpu, gva_t gpa, u32 error_code, 
	if (likely(!force_pt_level))

		transparent_hugepage_adjust(vcpu, &gfn, &pfn, &level);

	r = __direct_map(vcpu, write, map_writable, level, gfn, pfn, prefault);

	spin_unlock(&vcpu->kvm->mmu_lock);



	return r;



out_unlock:

	spin_unlock(&vcpu->kvm->mmu_lock);

	kvm_release_pfn_clean(pfn);

	return RET_PF_RETRY;

	return r;

}



static void nonpaging_init_context(struct kvm_vcpu *vcpu,

arch/x86/kvm/paging_tmpl.h

+3 −5
Original line number Diff line number Diff line
@@ -543,6 +543,7 @@ FNAME(prefetch_gpte)(struct kvm_vcpu *vcpu, struct kvm_mmu_page *sp, 
	mmu_set_spte(vcpu, spte, pte_access, 0, PT_PAGE_TABLE_LEVEL, gfn, pfn,

		     true, true);



	kvm_release_pfn_clean(pfn);

	return true;

}
@@ -694,7 +695,6 @@ static int FNAME(fetch)(struct kvm_vcpu *vcpu, gva_t addr, 
	return ret;



out_gpte_changed:

	kvm_release_pfn_clean(pfn);

	return RET_PF_RETRY;

}
@@ -842,6 +842,7 @@ static int FNAME(page_fault)(struct kvm_vcpu *vcpu, gva_t addr, u32 error_code, 
			walker.pte_access &= ~ACC_EXEC_MASK;

	}



	r = RET_PF_RETRY;

	spin_lock(&vcpu->kvm->mmu_lock);

	if (mmu_notifier_retry(vcpu->kvm, mmu_seq))

		goto out_unlock;
@@ -855,14 +856,11 @@ static int FNAME(page_fault)(struct kvm_vcpu *vcpu, gva_t addr, u32 error_code, 
			 level, pfn, map_writable, prefault);

	++vcpu->stat.pf_fixed;

	kvm_mmu_audit(vcpu, AUDIT_POST_PAGE_FAULT);

	spin_unlock(&vcpu->kvm->mmu_lock);



	return r;



out_unlock:

	spin_unlock(&vcpu->kvm->mmu_lock);

	kvm_release_pfn_clean(pfn);

	return RET_PF_RETRY;

	return r;

}



static gpa_t FNAME(get_level1_sp_gpa)(struct kvm_mmu_page *sp)

CRA Git | Maintained and supported by SUSTech CRA and CCSE