[PATCH v4 00/10] Rework the DAIF mask, unmask and track API

From: Liao Chang
Date: Thu Jun 13 2024 - 23:52:36 EST


This patch series reworks the DAIF mask, unmask, and track API for the
upcoming FEAT_NMI extension added in Armv8.8.

As platform and virtualization[1] supports for FEAT_NMI is emerging, and
Mark Brown's FEAT_NMI patch series[2] highlighted the need for clean up
the existing hacking style approach about DAIF management code before
adding NMI functionality, furthermore, we discover some subtle bugs
during 'perf' and 'ipi_backtrace' transition from PSEUDO_NMI to
FEAT_NMI, in summary, all of these emphasize the importance of rework.

This series of reworking patches follows the suggestion from Mark
Rutland mentioned in Mark Brown's patchset. In summary, he think the
better way for DAIF manangement look likes as following:

(a) Adding entry-specific helpers to manipulate abstract exception masks
covering DAIF + PMR + ALLINT. Those need unmask-at-entry and
mask-at-exit behaviour, and today only need to manage DAIF + PMR.

It should be possible to do this ahead of ALLINT / NMI support.

(b) Adding new "logical exception mask" helpers that treat DAIF + PMR +
ALLINT as separate elements.

This patches cherry-pick a part of Mark Brown' FEAT_NMI series, in order
to pass compilation and basic testing, includes perf and ipi_backtrace.

v4->v3:
General Enhancements
--------------------
Commit messages of [PATCH 04/05/06] have been enriched to outline the
implementation details, motivations and potential effects. This might
improve develper understanding and review efficiency.

Specific Changes
----------------
1. [PATCH 01] new utilize the existing helper maco in sysregs.h to
generate the "MSR ALLLINT, #Imm1" instruction. Additionally, helper
names have been renamed to start with msr_pstate_ for better
discoverability (as suggested by Mark Brown).

2. For [PATCH 04], due to the barrier side-effect of writing to PSTATE
fields, it is unnecessary to call pmr_sync() in
__pmr_local_allint_restore(). Add a table in comments to depict the
relationship between the type of interrupt masking and hardware
register configuration.

3. For [PATCH 05/06], function names have been revised to better reflect
their purpose:

local_errint_enable() -> local_irq_serror_enable()
local_errint_disable() -> local_nmi_serror_disable()
local_allint_mark_enabled() -> local_irq_mark_enabled()
local_allint_disable() -> local_nmi_disable()
local_errnmi_enable() -> local_nmi_serror_enable()

4. For [PATCH 07], A bug in local_nmi_enable() has been fixed. The v3
version is overly complex and included an unnecessary write operation
to PSTATE.DAIF.

5. [PATCH 09] introduce a slight optimization for NMI handling. Since
the intermediate step of marking IRQ TO-BE enabled is no longer
required, dropping PMR before acknowledge PSEUDO_NMI is also
unnecessary.

6. [PATCH 10] migrates CPU idle contex save/restore operation to the
newly introduced logical interrupt masking helper functions.

v3->v2:
1. Squash two commits that address two minor issues into Mark Brown's
origin patch for detecting FEAT_NMI.
2. Add one patch resolves the kprobe reenter panic while testing
FEAT_NMI on QEMU.

v2->v1:
Add SoB tags following the origin author's SoBs.

[1] https://lore.kernel.org/all/20240407081733.3231820-1-ruanjinjie@xxxxxxxxxx/
[2] https://lore.kernel.org/linux-arm-kernel/Y4sH5qX5bK9xfEBp@lpieralisi/

Liao Chang (8):
arm64/sysreg: Add definitions for immediate versions of MSR ALLINT
arm64: daifflags: Introduce logical interrupt masking
arm64: Sipmlify exception masking during exception entry and exit
arm64: Deprecate old local_daif_{mask,save,restore} helper functions
irqchip/gic-v3: Improve the maintainability of NMI masking in GIC
driver
arm64: kprobe: Keep NMI maskabled while kprobe is stepping xol
arm64: irqchip/gic-v3: Simplify NMI handling in IRQs disabled context
arm64: Migrate idle context save/restore to logical interrupt masking

Mark Brown (2):
arm64/cpufeature: Detect PE support for FEAT_NMI
arm64/nmi: Add Kconfig for NMI

arch/arm64/Kconfig | 17 ++
arch/arm64/include/asm/cpufeature.h | 6 +
arch/arm64/include/asm/cpuidle.h | 24 +-
arch/arm64/include/asm/daifflags.h | 376 +++++++++++++++++++++------
arch/arm64/include/asm/mte-kasan.h | 4 +-
arch/arm64/include/asm/mte.h | 2 +-
arch/arm64/include/asm/sysreg.h | 27 +-
arch/arm64/include/asm/uaccess.h | 4 +-
arch/arm64/include/uapi/asm/ptrace.h | 1 +
arch/arm64/kernel/acpi.c | 10 +-
arch/arm64/kernel/cpufeature.c | 61 ++++-
arch/arm64/kernel/debug-monitors.c | 6 +-
arch/arm64/kernel/entry-common.c | 94 +++----
arch/arm64/kernel/entry.S | 4 +-
arch/arm64/kernel/hibernate.c | 6 +-
arch/arm64/kernel/idle.c | 2 +-
arch/arm64/kernel/irq.c | 2 +-
arch/arm64/kernel/machine_kexec.c | 2 +-
arch/arm64/kernel/probes/kprobes.c | 4 +-
arch/arm64/kernel/proton-pack.c | 4 +-
arch/arm64/kernel/setup.c | 2 +-
arch/arm64/kernel/smp.c | 6 +-
arch/arm64/kernel/suspend.c | 10 +-
arch/arm64/kvm/hyp/entry.S | 2 +-
arch/arm64/kvm/hyp/vgic-v3-sr.c | 6 +-
arch/arm64/kvm/hyp/vhe/switch.c | 4 +-
arch/arm64/mm/mmu.c | 6 +-
arch/arm64/tools/cpucaps | 2 +
drivers/firmware/psci/psci.c | 2 +-
drivers/irqchip/irq-gic-v3.c | 29 +--
30 files changed, 490 insertions(+), 235 deletions(-)

--
2.34.1