[patch 6/6] x86/smp: Put CPUs into INIT on shutdown if possible

From: Thomas Gleixner
Date: Sat Jun 03 2023 - 16:09:00 EST


Parking CPUs in a HLT loop is not completely safe vs. kexec() as HLT can
resume execution due to NMI, SMI and MCE, which has the same issue as the
MWAIT loop.

Kicking the secondary CPUs into INIT makes this safe against NMI and SMI.

A broadcast MCE will take the machine down, but a broadcast MCE which makes
HLT resume and execute overwritten text, pagetables or data will end up in
a disaster too.

So chose the lesser of two evils and kick the secondary CPUs into INIT
unless the system has installed special wakeup mechanisms which are not
using INIT.

Signed-off-by: Thomas Gleixner <tglx@xxxxxxxxxxxxx>
---
arch/x86/include/asm/smp.h | 2 ++
arch/x86/kernel/smp.c | 38 +++++++++++++++++++++++++++++---------
arch/x86/kernel/smpboot.c | 19 +++++++++++++++++++
3 files changed, 50 insertions(+), 9 deletions(-)

--- a/arch/x86/include/asm/smp.h
+++ b/arch/x86/include/asm/smp.h
@@ -139,6 +139,8 @@ void native_send_call_func_ipi(const str
void native_send_call_func_single_ipi(int cpu);
void x86_idle_thread_init(unsigned int cpu, struct task_struct *idle);

+bool smp_park_nonboot_cpus_in_init(void);
+
void smp_store_boot_cpu_info(void);
void smp_store_cpu_info(int id);

--- a/arch/x86/kernel/smp.c
+++ b/arch/x86/kernel/smp.c
@@ -130,7 +130,7 @@ static int smp_stop_nmi_callback(unsigne
}

/*
- * this function calls the 'stop' function on all other CPUs in the system.
+ * Disable virtualization, APIC etc. and park the CPU in a HLT loop
*/
DEFINE_IDTENTRY_SYSVEC(sysvec_reboot)
{
@@ -147,8 +147,7 @@ static int register_stop_handler(void)

static void native_stop_other_cpus(int wait)
{
- unsigned long flags;
- unsigned long timeout;
+ unsigned long flags, timeout;

if (reboot_force)
return;
@@ -164,10 +163,10 @@ static void native_stop_other_cpus(int w
/*
* Start by using the REBOOT_VECTOR. That acts as a sync point to
* allow critical regions of code on other cpus to leave their
- * critical regions. Jumping straight to an NMI might accidentally
- * cause deadlocks with further shutdown code. This gives the CPUs
- * up to one second to finish their work before forcing them off
- * with the NMI.
+ * critical regions. Jumping straight to NMI or INIT might
+ * accidentally cause deadlocks with further shutdown code. This
+ * gives the CPUs up to one second to finish their work before
+ * forcing them off with the NMI or INIT.
*/
if (num_online_cpus() > 1) {
apic_send_IPI_allbutself(REBOOT_VECTOR);
@@ -175,7 +174,7 @@ static void native_stop_other_cpus(int w
/*
* Don't wait longer than a second for IPI completion. The
* wait request is not checked here because that would
- * prevent an NMI shutdown attempt in case that not all
+ * prevent an NMI/INIT shutdown in case that not all
* CPUs reach shutdown state.
*/
timeout = USEC_PER_SEC;
@@ -183,7 +182,27 @@ static void native_stop_other_cpus(int w
udelay(1);
}

- /* if the REBOOT_VECTOR didn't work, try with the NMI */
+ /*
+ * Park all nonboot CPUs in INIT including offline CPUs, if
+ * possible. That's a safe place where they can't resume execution
+ * of HLT and then execute the HLT loop from overwritten text or
+ * page tables.
+ *
+ * The only downside is a broadcast MCE, but up to the point where
+ * the kexec() kernel brought all APs online again an MCE will just
+ * make HLT resume and handle the MCE. The machine crashs and burns
+ * due to overwritten text, page tables and data. So there is a
+ * choice between fire and frying pan. The result is pretty much
+ * the same. Chose frying pan until x86 provides a sane mechanism
+ * to park a CPU.
+ */
+ if (smp_park_nonboot_cpus_in_init())
+ goto done;
+
+ /*
+ * If park with INIT was not possible and the REBOOT_VECTOR didn't
+ * take all secondary CPUs offline, try with the NMI.
+ */
if (num_online_cpus() > 1) {
/*
* If NMI IPI is enabled, try to register the stop handler
@@ -208,6 +227,7 @@ static void native_stop_other_cpus(int w
udelay(1);
}

+done:
local_irq_save(flags);
disable_local_APIC();
mcheck_cpu_clear(this_cpu_ptr(&cpu_info));
--- a/arch/x86/kernel/smpboot.c
+++ b/arch/x86/kernel/smpboot.c
@@ -1467,6 +1467,25 @@ void arch_thaw_secondary_cpus_end(void)
cache_aps_init();
}

+bool smp_park_nonboot_cpus_in_init(void)
+{
+ unsigned int cpu, this_cpu = smp_processor_id();
+ unsigned int apicid;
+
+ if (apic->wakeup_secondary_cpu_64 || apic->wakeup_secondary_cpu)
+ return false;
+
+ for_each_present_cpu(cpu) {
+ if (cpu == this_cpu)
+ continue;
+ apicid = apic->cpu_present_to_apicid(cpu);
+ if (apicid == BAD_APICID)
+ continue;
+ send_init_sequence(apicid);
+ }
+ return true;
+}
+
/*
* Early setup to make printk work.
*/