xen/vcpu: Fix vcpu restore path.

The runstate of vcpu should be restored for all possible cpus, as well as the
vcpu info placement.

Acked-by: Jan Beulich <jbeulich@suse.com>
Signed-off-by: Wei Liu <wei.liu2@citrix.com>
Signed-off-by: Konrad Rzeszutek Wilk <konrad.wilk@oracle.com>

1 files changed, 4 insertions, 3 deletions

diff --git a/arch/x86/xen/enlighten.c b/arch/x86/xen/enlighten.c
index 3aeaa933b527..138e5667409a 100644
--- a/arch/x86/xen/enlighten.c
+++ b/arch/x86/xen/enlighten.c
@@ -193,10 +193,11 @@ void xen_vcpu_restore(void)
 {
         int cpu;
 
-        for_each_online_cpu(cpu) {
+        for_each_possible_cpu(cpu) {
                 bool other_cpu = (cpu != smp_processor_id());
+                bool is_up = HYPERVISOR_vcpu_op(VCPUOP_is_up, cpu, NULL);
 
-                if (other_cpu &&
+                if (other_cpu && is_up &&
                     HYPERVISOR_vcpu_op(VCPUOP_down, cpu, NULL))
                         BUG();
 
@@ -205,7 +206,7 @@ void xen_vcpu_restore(void)
                 if (have_vcpu_info_placement)
                         xen_vcpu_setup(cpu);
 
-                if (other_cpu &&
+                if (other_cpu && is_up &&
                     HYPERVISOR_vcpu_op(VCPUOP_up, cpu, NULL))
                         BUG();
         }