This patch is for kvm-coco-queue. Invoke the leaf spte change hook only when non-present => present change. Patches: 1: Add WARN_ON_ONCE() 2: Bug fix v2: - Split out WARN_ON_ONCE() into an independent patch - Removed pr_debug() v1: https://lore.kernel.org/kvm/6eecc450d0326c9bedfbb34096a0279410923c8d.1726182754.git.isaku.yamahata@xxxxxxxxx/ As the test requires the base TDX KVM patch series and the TDX KVM kselftesets, this patch series doesn't include it. [1] suggested to require the AD bits supports for TDX. Don't include the following chage because this patch series is for kvm-coco-queue. diff --git a/arch/x86/kvm/vmx/tdx.c b/arch/x86/kvm/vmx/tdx.c index 109a7440aa68..5499edb80b9a 100644 --- a/arch/x86/kvm/vmx/tdx.c +++ b/arch/x86/kvm/vmx/tdx.c @@ -3275,7 +3275,7 @@ static int __init __tdx_bringup(void) const struct tdx_sys_info_td_conf *td_conf; int r, i; - if (!tdp_mmu_enabled || !enable_mmio_caching) + if (!tdp_mmu_enabled || !enable_mmio_caching || !enable_ept_ad_bits) return -EOPNOTSUPP; if (!cpu_feature_enabled(X86_FEATURE_MOVDIR64B)) { Observation of the issue ------------------------ An issue was reported on an out-of-tree kernel with an older version of TDX support, but that uses similar code to the current existing mirror/external MMU support. It can't be triggered without future patches. In the out of tree kernel it appears as a KVM_BUG_ON() getting hit in TDX code when set_external_spte() was called on a PTE that was already present. Investigation turned up that: 1. It was due to the AD state machine trying to update the dirty bit 2. A problem in the currently queued mirror/external PTE, but that can't be hit until more TDX support is added. In more detail, the spotted issue was caused by two vcpus simultaneously faulting for same GFN: vcpu0 vcpu1 ----- ----- ept violation for GFN ept violation for GFN encounter !present mirror PTE set_external_spte() re-enter TD, and accept GFN encounter present mirror PTE trigger AD state machine transition see old/new PTEs are different call set_external_spte() again The TDX external TDP backend couldn't handle the second call to set_external_spte(). In recent TDX development branches, the TDH.MEM.PAGE.AUG() SEAMCALL() hook would fail unexpectedly with TDX_EPT_ENTRY_STATE_INCORRECT + SPTE PENDING or PRESENT. However, the callback assumes that it's called only when non-present => present leaf SPTE change. It shows as a triggering a KVM_BUG_ON() with a stacktrace like: WARNING: CPU: 4 PID: 3700 at tdx_mem_page_aug+0x16d/0x560 [kvm_intel] IP: 0010:tdx_mem_page_aug+0x16d/0x560 [kvm_intel] Call Trace: set_external_spte_present+0x244/0x7e0 tdp_mmu_map_handle_target_level+0x460/0xd60 kvm_tdp_mmu_map+0x9c6/0xdc0 kvm_tdp_mmu_page_fault+0x32b/0x3e0 kvm_mmu_do_page_fault+0x4e5/0x6a0 kvm_mmu_page_fault+0x1a0/0x5e0 vcpu_enter_guest+0x1f5f/0x3900 vcpu_run+0x133/0xb60 kvm_arch_vcpu_ioctl_run+0x8ef/0x1650 kvm_vcpu_ioctl+0x4f9/0xb70 __x64_sys_ioctl+0x132/0x1a0 do_syscall_64+0xc1/0x1d0 entry_SYSCALL_64_after_hwframe+0x77/0x7f Design of MMU mirror/external ----------------------------- In current mirror/external MMU support, the set_external_spte() allows external PTEs to be set as present, but doesn't allow configuration of any other PTE bits. remove_external_spte() allows external PTEs to be zapped. Based on the fact that external PTE callbacks are exposed to essentially configure two PTE states (present or not present), and that mirror PTEs record whether the external PTE is present or not, future patches were planned to introduce a TDX backend for mirror/external page tables to not expect set_external_spte() calls on already present external PTEs. To accommodate this future code, steps were taken in the mirror/external support in the core MMU to prevent permission bits changing on such PTEs. However, the AD state machine scenario was missed. When the multiple vCPUs fault in the same GFN simultaneously, the hook is called multiple times with some bits changed while both old/new SPTEs have the present bits. The leaf SPTE is a complex state machine in general, and the value can change with software available bits and hardware bits. Solution -------- There are several options to address this: - Tame the SPTE state machine so that SPTE change won't happen for mirrored page table. PRO: It's conceptually natural to disable D bit support because mirrored page table doesn't support AD bits. CON: Not only D bit but also other bits can be modified. setting strict kvm_page_table.role.ad_disabled = true doesn't work. It requires to change the SPTE more deeply. - Add a check to the TDP MMU so that it won't call the hook unnecessarily PRO: Direct fix that shares set_external_spte() expectations in core MMU code. CON: Hard code in the TDP MMU when the hook is needed. - Pass old and new SPTE values to the hooks, add a check to the backend PRO: The backend can determine whether the callback is needed. CON: The KVM MMU implementation details leak to the backend because The SPTE encoding is specific to the KVM MMU implementation. And it's subject to change in the future. For example, is_shadow_present_pte() needs to be exported from the KVM MMU to the backend. The first choice is out because it doesn't easily handle the problem. Choose the second option over the third choice because the current interesting change is only non-present => present because we don't support dirty page tracking by write protect and large page yet by TDX KVM for now. (TDX supports them.) They're future work for TDX KVM. Care only about the leaf SPTE because non-leaf doesn't have a state machine. [1] https://lore.kernel.org/kvm/ZuOCXarfAwPjYj19@xxxxxxxxxx/ Isaku Yamahata (2): KVM: x86/tdp_mmu: Add WARN_ON_ONCE() in tdp_mmu_map_handle_target_level() KVM: x86/tdp_mmu: Trigger the callback only when an interesting change arch/x86/kvm/mmu/spte.h | 11 +++++++++++ arch/x86/kvm/mmu/tdp_mmu.c | 24 ++++++++++++++++++++---- 2 files changed, 31 insertions(+), 4 deletions(-) base-commit: 909f9d422f59f863d7b6e4e2c6e57abb97a27d4d -- 2.45.2