1106 lines · 31.3 KB · C
RawDownload
// SPDX-License-Identifier: GPL-2.0-only/* * This kernel test validates architecture page table helpers and * accessors and helps in verifying their continued compliance with * expected generic MM semantics. * * Copyright (C) 2019 ARM Ltd. * * Author: Anshuman Khandual <anshuman.khandual@arm.com> */#define pr_fmt(fmt) "debug_vm_pgtable: [%-25s]: " fmt, __func__ #include <linux/gfp.h>#include <linux/highmem.h>#include <linux/hugetlb.h>#include <linux/kernel.h>#include <linux/kconfig.h>#include <linux/mm.h>#include <linux/mman.h>#include <linux/mm_types.h>#include <linux/module.h>#include <linux/pfn_t.h>#include <linux/printk.h>#include <linux/pgtable.h>#include <linux/random.h>#include <linux/spinlock.h>#include <linux/swap.h>#include <linux/swapops.h>#include <linux/start_kernel.h>#include <linux/sched/mm.h>#include <linux/io.h>#include <asm/pgalloc.h>#include <asm/tlbflush.h> /* * Please refer Documentation/vm/arch_pgtable_helpers.rst for the semantics * expectations that are being validated here. All future changes in here * or the documentation need to be in sync. */ #define VMFLAGS	(VM_READ|VM_WRITE|VM_EXEC) /* * On s390 platform, the lower 4 bits are used to identify given page table * entry type. But these bits might affect the ability to clear entries with * pxx_clear() because of how dynamic page table folding works on s390. So * while loading up the entries do not change the lower 4 bits. It does not * have affect any other platform. Also avoid the 62nd bit on ppc64 that is * used to mark a pte entry. */#define S390_SKIP_MASK		GENMASK(3, 0)#if __BITS_PER_LONG == 64#define PPC64_SKIP_MASK		GENMASK(62, 62)#else#define PPC64_SKIP_MASK		0x0#endif#define ARCH_SKIP_MASK (S390_SKIP_MASK | PPC64_SKIP_MASK)#define RANDOM_ORVALUE (GENMASK(BITS_PER_LONG - 1, 0) & ~ARCH_SKIP_MASK)#define RANDOM_NZVALUE	GENMASK(7, 0) static void __init pte_basic_tests(unsigned long pfn, int idx){	pgprot_t prot = protection_map[idx];	pte_t pte = pfn_pte(pfn, prot);	unsigned long val = idx, *ptr = &val; 	pr_debug("Validating PTE basic (%pGv)\n", ptr); 	/*	 * This test needs to be executed after the given page table entry	 * is created with pfn_pte() to make sure that protection_map[idx]	 * does not have the dirty bit enabled from the beginning. This is	 * important for platforms like arm64 where (!PTE_RDONLY) indicate	 * dirty bit being set.	 */	WARN_ON(pte_dirty(pte_wrprotect(pte))); 	WARN_ON(!pte_same(pte, pte));	WARN_ON(!pte_young(pte_mkyoung(pte_mkold(pte))));	WARN_ON(!pte_dirty(pte_mkdirty(pte_mkclean(pte))));	WARN_ON(!pte_write(pte_mkwrite(pte_wrprotect(pte))));	WARN_ON(pte_young(pte_mkold(pte_mkyoung(pte))));	WARN_ON(pte_dirty(pte_mkclean(pte_mkdirty(pte))));	WARN_ON(pte_write(pte_wrprotect(pte_mkwrite(pte))));	WARN_ON(pte_dirty(pte_wrprotect(pte_mkclean(pte))));	WARN_ON(!pte_dirty(pte_wrprotect(pte_mkdirty(pte))));} static void __init pte_advanced_tests(struct mm_struct *mm,				      struct vm_area_struct *vma, pte_t *ptep,				      unsigned long pfn, unsigned long vaddr,				      pgprot_t prot){	pte_t pte = pfn_pte(pfn, prot); 	/*	 * Architectures optimize set_pte_at by avoiding TLB flush.	 * This requires set_pte_at to be not used to update an	 * existing pte entry. Clear pte before we do set_pte_at	 */ 	pr_debug("Validating PTE advanced\n");	pte = pfn_pte(pfn, prot);	set_pte_at(mm, vaddr, ptep, pte);	ptep_set_wrprotect(mm, vaddr, ptep);	pte = ptep_get(ptep);	WARN_ON(pte_write(pte));	ptep_get_and_clear(mm, vaddr, ptep);	pte = ptep_get(ptep);	WARN_ON(!pte_none(pte)); 	pte = pfn_pte(pfn, prot);	pte = pte_wrprotect(pte);	pte = pte_mkclean(pte);	set_pte_at(mm, vaddr, ptep, pte);	pte = pte_mkwrite(pte);	pte = pte_mkdirty(pte);	ptep_set_access_flags(vma, vaddr, ptep, pte, 1);	pte = ptep_get(ptep);	WARN_ON(!(pte_write(pte) && pte_dirty(pte)));	ptep_get_and_clear_full(mm, vaddr, ptep, 1);	pte = ptep_get(ptep);	WARN_ON(!pte_none(pte)); 	pte = pfn_pte(pfn, prot);	pte = pte_mkyoung(pte);	set_pte_at(mm, vaddr, ptep, pte);	ptep_test_and_clear_young(vma, vaddr, ptep);	pte = ptep_get(ptep);	WARN_ON(pte_young(pte));} static void __init pte_savedwrite_tests(unsigned long pfn, pgprot_t prot){	pte_t pte = pfn_pte(pfn, prot); 	if (!IS_ENABLED(CONFIG_NUMA_BALANCING))		return; 	pr_debug("Validating PTE saved write\n");	WARN_ON(!pte_savedwrite(pte_mk_savedwrite(pte_clear_savedwrite(pte))));	WARN_ON(pte_savedwrite(pte_clear_savedwrite(pte_mk_savedwrite(pte))));} #ifdef CONFIG_TRANSPARENT_HUGEPAGEstatic void __init pmd_basic_tests(unsigned long pfn, int idx){	pgprot_t prot = protection_map[idx];	pmd_t pmd = pfn_pmd(pfn, prot);	unsigned long val = idx, *ptr = &val; 	if (!has_transparent_hugepage())		return; 	pr_debug("Validating PMD basic (%pGv)\n", ptr); 	/*	 * This test needs to be executed after the given page table entry	 * is created with pfn_pmd() to make sure that protection_map[idx]	 * does not have the dirty bit enabled from the beginning. This is	 * important for platforms like arm64 where (!PTE_RDONLY) indicate	 * dirty bit being set.	 */	WARN_ON(pmd_dirty(pmd_wrprotect(pmd)));  	WARN_ON(!pmd_same(pmd, pmd));	WARN_ON(!pmd_young(pmd_mkyoung(pmd_mkold(pmd))));	WARN_ON(!pmd_dirty(pmd_mkdirty(pmd_mkclean(pmd))));	WARN_ON(!pmd_write(pmd_mkwrite(pmd_wrprotect(pmd))));	WARN_ON(pmd_young(pmd_mkold(pmd_mkyoung(pmd))));	WARN_ON(pmd_dirty(pmd_mkclean(pmd_mkdirty(pmd))));	WARN_ON(pmd_write(pmd_wrprotect(pmd_mkwrite(pmd))));	WARN_ON(pmd_dirty(pmd_wrprotect(pmd_mkclean(pmd))));	WARN_ON(!pmd_dirty(pmd_wrprotect(pmd_mkdirty(pmd))));	/*	 * A huge page does not point to next level page table	 * entry. Hence this must qualify as pmd_bad().	 */	WARN_ON(!pmd_bad(pmd_mkhuge(pmd)));} static void __init pmd_advanced_tests(struct mm_struct *mm,				      struct vm_area_struct *vma, pmd_t *pmdp,				      unsigned long pfn, unsigned long vaddr,				      pgprot_t prot, pgtable_t pgtable){	pmd_t pmd = pfn_pmd(pfn, prot); 	if (!has_transparent_hugepage())		return; 	pr_debug("Validating PMD advanced\n");	/* Align the address wrt HPAGE_PMD_SIZE */	vaddr = (vaddr & HPAGE_PMD_MASK) + HPAGE_PMD_SIZE; 	pgtable_trans_huge_deposit(mm, pmdp, pgtable); 	pmd = pfn_pmd(pfn, prot);	set_pmd_at(mm, vaddr, pmdp, pmd);	pmdp_set_wrprotect(mm, vaddr, pmdp);	pmd = READ_ONCE(*pmdp);	WARN_ON(pmd_write(pmd));	pmdp_huge_get_and_clear(mm, vaddr, pmdp);	pmd = READ_ONCE(*pmdp);	WARN_ON(!pmd_none(pmd)); 	pmd = pfn_pmd(pfn, prot);	pmd = pmd_wrprotect(pmd);	pmd = pmd_mkclean(pmd);	set_pmd_at(mm, vaddr, pmdp, pmd);	pmd = pmd_mkwrite(pmd);	pmd = pmd_mkdirty(pmd);	pmdp_set_access_flags(vma, vaddr, pmdp, pmd, 1);	pmd = READ_ONCE(*pmdp);	WARN_ON(!(pmd_write(pmd) && pmd_dirty(pmd)));	pmdp_huge_get_and_clear_full(vma, vaddr, pmdp, 1);	pmd = READ_ONCE(*pmdp);	WARN_ON(!pmd_none(pmd)); 	pmd = pmd_mkhuge(pfn_pmd(pfn, prot));	pmd = pmd_mkyoung(pmd);	set_pmd_at(mm, vaddr, pmdp, pmd);	pmdp_test_and_clear_young(vma, vaddr, pmdp);	pmd = READ_ONCE(*pmdp);	WARN_ON(pmd_young(pmd)); 	/*  Clear the pte entries  */	pmdp_huge_get_and_clear(mm, vaddr, pmdp);	pgtable = pgtable_trans_huge_withdraw(mm, pmdp);} static void __init pmd_leaf_tests(unsigned long pfn, pgprot_t prot){	pmd_t pmd = pfn_pmd(pfn, prot); 	pr_debug("Validating PMD leaf\n");	/*	 * PMD based THP is a leaf entry.	 */	pmd = pmd_mkhuge(pmd);	WARN_ON(!pmd_leaf(pmd));} #ifdef CONFIG_HAVE_ARCH_HUGE_VMAPstatic void __init pmd_huge_tests(pmd_t *pmdp, unsigned long pfn, pgprot_t prot){	pmd_t pmd; 	if (!arch_ioremap_pmd_supported())		return; 	pr_debug("Validating PMD huge\n");	/*	 * X86 defined pmd_set_huge() verifies that the given	 * PMD is not a populated non-leaf entry.	 */	WRITE_ONCE(*pmdp, __pmd(0));	WARN_ON(!pmd_set_huge(pmdp, __pfn_to_phys(pfn), prot));	WARN_ON(!pmd_clear_huge(pmdp));	pmd = READ_ONCE(*pmdp);	WARN_ON(!pmd_none(pmd));}#else /* CONFIG_HAVE_ARCH_HUGE_VMAP */static void __init pmd_huge_tests(pmd_t *pmdp, unsigned long pfn, pgprot_t prot) { }#endif /* CONFIG_HAVE_ARCH_HUGE_VMAP */ static void __init pmd_savedwrite_tests(unsigned long pfn, pgprot_t prot){	pmd_t pmd = pfn_pmd(pfn, prot); 	if (!IS_ENABLED(CONFIG_NUMA_BALANCING))		return; 	pr_debug("Validating PMD saved write\n");	WARN_ON(!pmd_savedwrite(pmd_mk_savedwrite(pmd_clear_savedwrite(pmd))));	WARN_ON(pmd_savedwrite(pmd_clear_savedwrite(pmd_mk_savedwrite(pmd))));} #ifdef CONFIG_HAVE_ARCH_TRANSPARENT_HUGEPAGE_PUDstatic void __init pud_basic_tests(struct mm_struct *mm, unsigned long pfn, int idx){	pgprot_t prot = protection_map[idx];	pud_t pud = pfn_pud(pfn, prot);	unsigned long val = idx, *ptr = &val; 	if (!has_transparent_hugepage())		return; 	pr_debug("Validating PUD basic (%pGv)\n", ptr); 	/*	 * This test needs to be executed after the given page table entry	 * is created with pfn_pud() to make sure that protection_map[idx]	 * does not have the dirty bit enabled from the beginning. This is	 * important for platforms like arm64 where (!PTE_RDONLY) indicate	 * dirty bit being set.	 */	WARN_ON(pud_dirty(pud_wrprotect(pud))); 	WARN_ON(!pud_same(pud, pud));	WARN_ON(!pud_young(pud_mkyoung(pud_mkold(pud))));	WARN_ON(!pud_dirty(pud_mkdirty(pud_mkclean(pud))));	WARN_ON(pud_dirty(pud_mkclean(pud_mkdirty(pud))));	WARN_ON(!pud_write(pud_mkwrite(pud_wrprotect(pud))));	WARN_ON(pud_write(pud_wrprotect(pud_mkwrite(pud))));	WARN_ON(pud_young(pud_mkold(pud_mkyoung(pud))));	WARN_ON(pud_dirty(pud_wrprotect(pud_mkclean(pud))));	WARN_ON(!pud_dirty(pud_wrprotect(pud_mkdirty(pud)))); 	if (mm_pmd_folded(mm))		return; 	/*	 * A huge page does not point to next level page table	 * entry. Hence this must qualify as pud_bad().	 */	WARN_ON(!pud_bad(pud_mkhuge(pud)));} static void __init pud_advanced_tests(struct mm_struct *mm,				      struct vm_area_struct *vma, pud_t *pudp,				      unsigned long pfn, unsigned long vaddr,				      pgprot_t prot){	pud_t pud = pfn_pud(pfn, prot); 	if (!has_transparent_hugepage())		return; 	pr_debug("Validating PUD advanced\n");	/* Align the address wrt HPAGE_PUD_SIZE */	vaddr = (vaddr & HPAGE_PUD_MASK) + HPAGE_PUD_SIZE; 	set_pud_at(mm, vaddr, pudp, pud);	pudp_set_wrprotect(mm, vaddr, pudp);	pud = READ_ONCE(*pudp);	WARN_ON(pud_write(pud)); #ifndef __PAGETABLE_PMD_FOLDED	pudp_huge_get_and_clear(mm, vaddr, pudp);	pud = READ_ONCE(*pudp);	WARN_ON(!pud_none(pud));#endif /* __PAGETABLE_PMD_FOLDED */	pud = pfn_pud(pfn, prot);	pud = pud_wrprotect(pud);	pud = pud_mkclean(pud);	set_pud_at(mm, vaddr, pudp, pud);	pud = pud_mkwrite(pud);	pud = pud_mkdirty(pud);	pudp_set_access_flags(vma, vaddr, pudp, pud, 1);	pud = READ_ONCE(*pudp);	WARN_ON(!(pud_write(pud) && pud_dirty(pud))); #ifndef __PAGETABLE_PMD_FOLDED	pudp_huge_get_and_clear_full(mm, vaddr, pudp, 1);	pud = READ_ONCE(*pudp);	WARN_ON(!pud_none(pud));#endif /* __PAGETABLE_PMD_FOLDED */ 	pud = pfn_pud(pfn, prot);	pud = pud_mkyoung(pud);	set_pud_at(mm, vaddr, pudp, pud);	pudp_test_and_clear_young(vma, vaddr, pudp);	pud = READ_ONCE(*pudp);	WARN_ON(pud_young(pud)); 	pudp_huge_get_and_clear(mm, vaddr, pudp);} static void __init pud_leaf_tests(unsigned long pfn, pgprot_t prot){	pud_t pud = pfn_pud(pfn, prot); 	pr_debug("Validating PUD leaf\n");	/*	 * PUD based THP is a leaf entry.	 */	pud = pud_mkhuge(pud);	WARN_ON(!pud_leaf(pud));} #ifdef CONFIG_HAVE_ARCH_HUGE_VMAPstatic void __init pud_huge_tests(pud_t *pudp, unsigned long pfn, pgprot_t prot){	pud_t pud; 	if (!arch_ioremap_pud_supported())		return; 	pr_debug("Validating PUD huge\n");	/*	 * X86 defined pud_set_huge() verifies that the given	 * PUD is not a populated non-leaf entry.	 */	WRITE_ONCE(*pudp, __pud(0));	WARN_ON(!pud_set_huge(pudp, __pfn_to_phys(pfn), prot));	WARN_ON(!pud_clear_huge(pudp));	pud = READ_ONCE(*pudp);	WARN_ON(!pud_none(pud));}#else /* !CONFIG_HAVE_ARCH_HUGE_VMAP */static void __init pud_huge_tests(pud_t *pudp, unsigned long pfn, pgprot_t prot) { }#endif /* !CONFIG_HAVE_ARCH_HUGE_VMAP */ #else  /* !CONFIG_HAVE_ARCH_TRANSPARENT_HUGEPAGE_PUD */static void __init pud_basic_tests(struct mm_struct *mm, unsigned long pfn, int idx) { }static void __init pud_advanced_tests(struct mm_struct *mm,				      struct vm_area_struct *vma, pud_t *pudp,				      unsigned long pfn, unsigned long vaddr,				      pgprot_t prot){}static void __init pud_leaf_tests(unsigned long pfn, pgprot_t prot) { }static void __init pud_huge_tests(pud_t *pudp, unsigned long pfn, pgprot_t prot){}#endif /* CONFIG_HAVE_ARCH_TRANSPARENT_HUGEPAGE_PUD */#else  /* !CONFIG_TRANSPARENT_HUGEPAGE */static void __init pmd_basic_tests(unsigned long pfn, int idx) { }static void __init pud_basic_tests(struct mm_struct *mm, unsigned long pfn, int idx) { }static void __init pmd_advanced_tests(struct mm_struct *mm,				      struct vm_area_struct *vma, pmd_t *pmdp,				      unsigned long pfn, unsigned long vaddr,				      pgprot_t prot, pgtable_t pgtable){}static void __init pud_advanced_tests(struct mm_struct *mm,				      struct vm_area_struct *vma, pud_t *pudp,				      unsigned long pfn, unsigned long vaddr,				      pgprot_t prot){}static void __init pmd_leaf_tests(unsigned long pfn, pgprot_t prot) { }static void __init pud_leaf_tests(unsigned long pfn, pgprot_t prot) { }static void __init pmd_huge_tests(pmd_t *pmdp, unsigned long pfn, pgprot_t prot){}static void __init pud_huge_tests(pud_t *pudp, unsigned long pfn, pgprot_t prot){}static void __init pmd_savedwrite_tests(unsigned long pfn, pgprot_t prot) { }#endif /* CONFIG_TRANSPARENT_HUGEPAGE */ static void __init p4d_basic_tests(unsigned long pfn, pgprot_t prot){	p4d_t p4d; 	pr_debug("Validating P4D basic\n");	memset(&p4d, RANDOM_NZVALUE, sizeof(p4d_t));	WARN_ON(!p4d_same(p4d, p4d));} static void __init pgd_basic_tests(unsigned long pfn, pgprot_t prot){	pgd_t pgd; 	pr_debug("Validating PGD basic\n");	memset(&pgd, RANDOM_NZVALUE, sizeof(pgd_t));	WARN_ON(!pgd_same(pgd, pgd));} #ifndef __PAGETABLE_PUD_FOLDEDstatic void __init pud_clear_tests(struct mm_struct *mm, pud_t *pudp){	pud_t pud = READ_ONCE(*pudp); 	if (mm_pmd_folded(mm))		return; 	pr_debug("Validating PUD clear\n");	pud = __pud(pud_val(pud) | RANDOM_ORVALUE);	WRITE_ONCE(*pudp, pud);	pud_clear(pudp);	pud = READ_ONCE(*pudp);	WARN_ON(!pud_none(pud));} static void __init pud_populate_tests(struct mm_struct *mm, pud_t *pudp,				      pmd_t *pmdp){	pud_t pud; 	if (mm_pmd_folded(mm))		return; 	pr_debug("Validating PUD populate\n");	/*	 * This entry points to next level page table page.	 * Hence this must not qualify as pud_bad().	 */	pud_populate(mm, pudp, pmdp);	pud = READ_ONCE(*pudp);	WARN_ON(pud_bad(pud));}#else  /* !__PAGETABLE_PUD_FOLDED */static void __init pud_clear_tests(struct mm_struct *mm, pud_t *pudp) { }static void __init pud_populate_tests(struct mm_struct *mm, pud_t *pudp,				      pmd_t *pmdp){

Lines 1–500 of 1106.