They are duplicated codes to create vcpu.arch.{user,guest}_fpu in VMX
and SVM. Make them common functions and delay it a little bit later
after .create_vcpu.

No functional change intended.

Signed-off-by: Xiaoyao Li <xiaoyao...@intel.com>
---
 arch/x86/kvm/svm.c     | 18 ------------------
 arch/x86/kvm/vmx/vmx.c | 18 ------------------
 arch/x86/kvm/x86.c     | 24 ++++++++++++++++++++++++
 3 files changed, 24 insertions(+), 36 deletions(-)

diff --git a/arch/x86/kvm/svm.c b/arch/x86/kvm/svm.c
index d35ef5456462..cbb5f5b9a9e7 100644
--- a/arch/x86/kvm/svm.c
+++ b/arch/x86/kvm/svm.c
@@ -2181,20 +2181,6 @@ static struct kvm_vcpu *svm_create_vcpu(struct kvm *kvm, 
unsigned int id)
        if (!svm)
                goto out;
 
-       svm->vcpu.arch.user_fpu = kmem_cache_zalloc(x86_fpu_cache,
-                                                    GFP_KERNEL_ACCOUNT);
-       if (!svm->vcpu.arch.user_fpu) {
-               printk(KERN_ERR "kvm: failed to allocate kvm userspace's 
fpu\n");
-               goto free_partial_svm;
-       }
-
-       svm->vcpu.arch.guest_fpu = kmem_cache_zalloc(x86_fpu_cache,
-                                                    GFP_KERNEL_ACCOUNT);
-       if (!svm->vcpu.arch.guest_fpu) {
-               printk(KERN_ERR "kvm: failed to allocate vcpu's fpu\n");
-               goto free_user_fpu;
-       }
-
        page = alloc_page(GFP_KERNEL_ACCOUNT);
        if (!page)
                goto free_svm;
@@ -2228,10 +2214,6 @@ static struct kvm_vcpu *svm_create_vcpu(struct kvm *kvm, 
unsigned int id)
 free_page1:
        __free_page(page);
 free_svm:
-       kmem_cache_free(x86_fpu_cache, svm->vcpu.arch.guest_fpu);
-free_user_fpu:
-       kmem_cache_free(x86_fpu_cache, svm->vcpu.arch.user_fpu);
-free_partial_svm:
        kmem_cache_free(kvm_vcpu_cache, svm);
 out:
        return ERR_PTR(err);
diff --git a/arch/x86/kvm/vmx/vmx.c b/arch/x86/kvm/vmx/vmx.c
index 2f54a3bcb6a5..183112604f04 100644
--- a/arch/x86/kvm/vmx/vmx.c
+++ b/arch/x86/kvm/vmx/vmx.c
@@ -6773,20 +6773,6 @@ static struct kvm_vcpu *vmx_create_vcpu(struct kvm *kvm, 
unsigned int id)
        if (!vmx)
                goto out;
 
-       vmx->vcpu.arch.user_fpu = kmem_cache_zalloc(x86_fpu_cache,
-                       GFP_KERNEL_ACCOUNT);
-       if (!vmx->vcpu.arch.user_fpu) {
-               printk(KERN_ERR "kvm: failed to allocate kvm userspace's 
fpu\n");
-               goto free_partial_vcpu;
-       }
-
-       vmx->vcpu.arch.guest_fpu = kmem_cache_zalloc(x86_fpu_cache,
-                       GFP_KERNEL_ACCOUNT);
-       if (!vmx->vcpu.arch.guest_fpu) {
-               printk(KERN_ERR "kvm: failed to allocate vcpu's fpu\n");
-               goto free_user_fpu;
-       }
-
        vmx->vpid = allocate_vpid();
 
        /*
@@ -6822,10 +6808,6 @@ static struct kvm_vcpu *vmx_create_vcpu(struct kvm *kvm, 
unsigned int id)
        vmx_destroy_pml_buffer(vmx);
 free_vcpu:
        free_vpid(vmx->vpid);
-       kmem_cache_free(x86_fpu_cache, vmx->vcpu.arch.guest_fpu);
-free_user_fpu:
-       kmem_cache_free(x86_fpu_cache, vmx->vcpu.arch.user_fpu);
-free_partial_vcpu:
        kmem_cache_free(kvm_vcpu_cache, vmx);
 out:
        return ERR_PTR(err);
diff --git a/arch/x86/kvm/x86.c b/arch/x86/kvm/x86.c
index fd9f1a1f0f01..c4b477a9c7f6 100644
--- a/arch/x86/kvm/x86.c
+++ b/arch/x86/kvm/x86.c
@@ -9021,6 +9021,26 @@ void kvm_arch_vcpu_free(struct kvm_vcpu *vcpu)
        free_cpumask_var(wbinvd_dirty_mask);
 }
 
+static int kvm_vcpu_create_fpu(struct kvm_vcpu *vcpu)
+{
+       vcpu->arch.user_fpu = kmem_cache_zalloc(x86_fpu_cache,
+                       GFP_KERNEL_ACCOUNT);
+       if (vcpu->arch.user_fpu) {
+               printk(KERN_ERR "kvm: failed to allocate kvm userspace's 
fpu\n");
+               return -ENOMEM;
+       }
+
+       vcpu->arch.guest_fpu = kmem_cache_zalloc(x86_fpu_cache,
+                       GFP_KERNEL_ACCOUNT);
+       if (vcpu->arch.guest_fpu) {
+               printk(KERN_ERR "kvm: failed to allocate vcpu's fpu\n");
+               kmem_cache_free(x86_fpu_cache, vcpu->arch.user_fpu);
+               return -ENOMEM;
+       }
+
+       return 0;
+}
+
 struct kvm_vcpu *kvm_arch_vcpu_create(struct kvm *kvm,
                                                unsigned int id)
 {
@@ -9036,6 +9056,10 @@ struct kvm_vcpu *kvm_arch_vcpu_create(struct kvm *kvm,
        if (IS_ERR(vcpu))
                return vcpu;
 
+       err = kvm_vcpu_create_fpu(vcpu);
+       if (err)
+               return ERR_PTR(err);
+
        err = kvm_vcpu_init(vcpu, kvm, id);
        if (err) {
                kvm_x86_ops->vcpu_free(vcpu);
-- 
2.19.1

Reply via email to