KVM: Allocate cpus_hardware_enabled after arch hardware setup

Allocate cpus_hardware_enabled after arch hardware setup so that arch
"init" and "hardware setup" are called back-to-back and thus can be
combined in a future patch.  cpus_hardware_enabled is never used before
kvm_create_vm(), i.e. doesn't have a dependency with hardware setup and
only needs to be allocated before /dev/kvm is exposed to userspace.

Free the object before the arch hooks are invoked to maintain symmetry,
and so that arch code can move away from the hooks without having to
worry about ordering changes.

Signed-off-by: Sean Christopherson <seanjc@google.com>
Reviewed-by: Yuan Yao <yuan.yao@intel.com>
Message-Id: <20221130230934.1014142-4-seanjc@google.com>
Signed-off-by: Paolo Bonzini <pbonzini@redhat.com>

diff --git a/virt/kvm/kvm_main.c b/virt/kvm/kvm_main.c
index 6793a7e3525addf76e7ffc99846be7530566134e..63983588c93b61cf9c7d4aca5775fb453f625c72 100644 (file)
--- a/virt/kvm/kvm_main.c
+++ b/virt/kvm/kvm_main.c
@@ -5931,15 +5931,15 @@ int kvm_init(void *opaque, unsigned vcpu_size, unsigned vcpu_align,
        if (r)
                return r;
 
+       r = kvm_arch_hardware_setup(opaque);
+       if (r < 0)
+               goto err_hw_setup;
+
        if (!zalloc_cpumask_var(&cpus_hardware_enabled, GFP_KERNEL)) {
                r = -ENOMEM;
                goto err_hw_enabled;
        }
 
-       r = kvm_arch_hardware_setup(opaque);
-       if (r < 0)
-               goto out_free_1;
-
        c.ret = &r;
        c.opaque = opaque;
        for_each_online_cpu(cpu) {
@@ -6025,10 +6025,10 @@ out_free_3:
        unregister_reboot_notifier(&kvm_reboot_notifier);
        cpuhp_remove_state_nocalls(CPUHP_AP_KVM_STARTING);
 out_free_2:
-       kvm_arch_hardware_unsetup();
-out_free_1:
        free_cpumask_var(cpus_hardware_enabled);
 err_hw_enabled:
+       kvm_arch_hardware_unsetup();
+err_hw_setup:
        kvm_arch_exit();
        return r;
 }
@@ -6055,9 +6055,9 @@ void kvm_exit(void)
        cpuhp_remove_state_nocalls(CPUHP_AP_KVM_STARTING);
        on_each_cpu(hardware_disable_nolock, NULL, 1);
        kvm_irqfd_exit();
+       free_cpumask_var(cpus_hardware_enabled);
        kvm_arch_hardware_unsetup();
        kvm_arch_exit();
-       free_cpumask_var(cpus_hardware_enabled);
        kvm_vfio_ops_exit();
 }
 EXPORT_SYMBOL_GPL(kvm_exit);