1 files changed, 14 insertions, 38 deletions

diff --git a/arch/s390/kernel/machine_kexec.c b/arch/s390/kernel/machine_kexec.c
index 47b168fb29c..0f8cdf1268d 100644
--- a/arch/s390/kernel/machine_kexec.c
+++ b/arch/s390/kernel/machine_kexec.c
@@ -14,6 +14,7 @@
 #include <linux/delay.h>
 #include <linux/reboot.h>
 #include <linux/ftrace.h>
+#include <linux/debug_locks.h>
 #include <asm/cio.h>
 #include <asm/setup.h>
 #include <asm/pgtable.h>
@@ -49,50 +50,21 @@ static void add_elf_notes(int cpu)
 }
 
 /*
- * Store status of next available physical CPU
- */
-static int store_status_next(int start_cpu, int this_cpu)
-{
-        struct save_area *sa = (void *) 4608 + store_prefix();
-        int cpu, rc;
-
-        for (cpu = start_cpu; cpu < 65536; cpu++) {
-                if (cpu == this_cpu)
-                        continue;
-                do {
-                        rc = raw_sigp(cpu, sigp_stop_and_store_status);
-                } while (rc == sigp_busy);
-                if (rc != sigp_order_code_accepted)
-                        continue;
-                if (sa->pref_reg)
-                        return cpu;
-        }
-        return -1;
-}
-
-/*
  * Initialize CPU ELF notes
  */
 void setup_regs(void)
 {
         unsigned long sa = S390_lowcore.prefixreg_save_area + SAVE_AREA_BASE;
-        int cpu, this_cpu, phys_cpu = 0, first = 1;
+        int cpu, this_cpu;
 
-        this_cpu = stap();
-
-        if (!S390_lowcore.prefixreg_save_area)
-                first = 0;
+        this_cpu = smp_find_processor_id(stap());
+        add_elf_notes(this_cpu);
         for_each_online_cpu(cpu) {
-                if (first) {
-                        add_elf_notes(cpu);
-                        first = 0;
+                if (cpu == this_cpu)
+                        continue;
+                if (smp_store_status(cpu))
                         continue;
-                }
-                phys_cpu = store_status_next(phys_cpu, this_cpu);
-                if (phys_cpu == -1)
-                        break;
                 add_elf_notes(cpu);
-                phys_cpu++;
         }
         /* Copy dump CPU store status info to absolute zero */
         memcpy((void *) SAVE_AREA_BASE, (void *) sa, sizeof(struct save_area));
@@ -238,10 +210,14 @@ static void __machine_kexec(void *data)
         struct kimage *image = data;
 
         pfault_fini();
-        if (image->type == KEXEC_TYPE_CRASH)
+        tracing_off();
+        debug_locks_off();
+        if (image->type == KEXEC_TYPE_CRASH) {
+                lgr_info_log();
                 s390_reset_system(__do_machine_kdump, data);
-        else
+        } else {
                 s390_reset_system(__do_machine_kexec, data);
+        }
         disabled_wait((unsigned long) __builtin_return_address(0));
 }
 
@@ -255,5 +231,5 @@ void machine_kexec(struct kimage *image)
                 return;
         tracer_disable();
         smp_send_stop();
-        smp_switch_to_ipl_cpu(__machine_kexec, image);
+        smp_call_ipl_cpu(__machine_kexec, image);
 }