This unit test covers: 1. CR3 LAM bits toggles. 2. Memory/MMIO access with user mode address containing LAM metadata. Signed-off-by: Binbin Wu <binbin.wu@xxxxxxxxxxxxxxx> Reviewed-by: Chao Gao <chao.gao@xxxxxxxxx> --- v7: - Rename lam_{u48,u57}_active() to is_lam_{u48,u57}_enabled(), and move them to processor.h (Sean) - Get lam status based on the address and vCPU state. (Sean) - Test LAM userspace address in kernel mode directly to simplify the interface of test_ptr() since LAM identifies a address as kernel or user only based on the address itself. - Add comments about the virtualization hole of CR3 LAM bits. --- lib/x86/processor.h | 12 ++++++++ x86/lam.c | 69 ++++++++++++++++++++++++++++++++++++++++++++- 2 files changed, 80 insertions(+), 1 deletion(-) diff --git a/lib/x86/processor.h b/lib/x86/processor.h index a38f87ed..c2cafb01 100644 --- a/lib/x86/processor.h +++ b/lib/x86/processor.h @@ -77,7 +77,9 @@ static inline u64 set_la_non_canonical(u64 src, u64 mask) #define X86_CR3_PCID_MASK GENMASK(11, 0) #define X86_CR3_LAM_U57_BIT (61) +#define X86_CR3_LAM_U57 BIT_ULL(X86_CR3_LAM_U57_BIT) #define X86_CR3_LAM_U48_BIT (62) +#define X86_CR3_LAM_U48 BIT_ULL(X86_CR3_LAM_U48_BIT) #define X86_CR4_VME_BIT (0) #define X86_CR4_VME BIT(X86_CR4_VME_BIT) @@ -988,4 +990,14 @@ static inline bool is_lam_sup_enabled(void) return !!(read_cr4() & X86_CR4_LAM_SUP); } +static inline bool is_lam_u48_enabled(void) +{ + return (read_cr3() & (X86_CR3_LAM_U48 | X86_CR3_LAM_U57)) == X86_CR3_LAM_U48; +} + +static inline bool is_lam_u57_enabled(void) +{ + return !!(read_cr3() & X86_CR3_LAM_U57); +} + #endif diff --git a/x86/lam.c b/x86/lam.c index 2f95b6c9..40b8ecdd 100644 --- a/x86/lam.c +++ b/x86/lam.c @@ -67,7 +67,14 @@ static bool get_lam_mask(u64 address, u64* lam_mask) return true; } - /* TODO: Get LAM mask for userspace address. */ + if(is_lam_u48_enabled()) { + *lam_mask = LAM48_MASK; + return true; + } + + if(is_lam_u57_enabled()) + return true; + return false; } @@ -88,6 +95,17 @@ static void test_ptr(u64* ptr, bool is_mmio) report(fault != lam_active, "Expected access to tagged address for %s %s LAM to %s", is_mmio ? "MMIO" : "memory", lam_active ? "with" : "without", lam_active ? "succeed" : "#GP"); + + /* + * This test case is only triggered when LAM_U57 is active and 4-level + * paging is used. For the case, bit[56:47] aren't all 0 triggers #GP. + */ + if (lam_active && (lam_mask == LAM57_MASK) && !is_la57_enabled()) { + ptr = (u64 *)set_la_non_canonical((u64)ptr, LAM48_MASK); + fault = test_for_exception(GP_VECTOR, do_mov, ptr); + report(fault, "Expected access to non-LAM-canonical address for %s to #GP", + is_mmio ? "MMIO" : "memory"); + } } /* invlpg with tagged address is same as NOP, no #GP expected. */ @@ -199,6 +217,54 @@ static void test_lam_sup(void) "use kvm.force_emulation_prefix=1 to enable\n"); } +static void test_lam_user(void) +{ + void* vaddr; + int vector; + unsigned long cr3 = read_cr3() & ~(X86_CR3_LAM_U48 | X86_CR3_LAM_U57); + bool has_lam = this_cpu_has(X86_FEATURE_LAM); + + /* + * The physical address of AREA_NORMAL is within 36 bits, so that using + * identical mapping, the linear address will be considered as user mode + * address from the view of LAM, and the metadata bits are not used as + * address for both LAM48 and LAM57. + */ + vaddr = alloc_pages_flags(0, AREA_NORMAL); + _Static_assert((AREA_NORMAL_PFN & GENMASK(63, 47)) == 0UL, + "Identical mapping range check"); + + /* + * Note, LAM doesn't have a global control bit to turn on/off LAM + * completely, but purely depends on hardware's CPUID to determine it + * can be enabled or not. That means, when EPT is on, even when KVM + * doesn't expose LAM to guest, the guest can still set LAM control bits + * in CR3 w/o causing problem. This is an unfortunate virtualization + * hole. KVM doesn't choose to intercept CR3 in this case for + * performance. + * Only enable LAM CR3 bits when LAM feature is exposed. + */ + if (has_lam) { + vector = write_cr3_safe(cr3 | X86_CR3_LAM_U48); + report(!vector && is_lam_u48_enabled(), "Expected CR3.LAM_U48=1 to succeed"); + } + /* + * Physical memory & MMIO have already been identical mapped in + * setup_mmu(). + */ + test_ptr(vaddr, false); + test_ptr(phys_to_virt(IORAM_BASE_PHYS), true); + + if (has_lam) { + vector = write_cr3_safe(cr3 | X86_CR3_LAM_U57); + report(!vector && is_lam_u57_enabled(), "Expected CR3.LAM_U57=1 to succeed"); + + /* If !has_lam, it has been tested above, no need to test again. */ + test_ptr(vaddr, false); + test_ptr(phys_to_virt(IORAM_BASE_PHYS), true); + } +} + int main(int ac, char **av) { setup_vm(); @@ -209,6 +275,7 @@ int main(int ac, char **av) report_info("This CPU supports LAM feature\n"); test_lam_sup(); + test_lam_user(); return report_summary(); } -- 2.43.2