The current allocator of __create_pgd_mapping_locked() takes only one
parameter, which reflects the current level of page table.

But kexec_page_alloc() should extra control information from the
parameter kimage. In order to make kexec_page_alloc() as an allocator,
re-shape the prototype of the allocator and hence of the
__create_pgd_mapping_locked().

Signed-off-by: Pingfan Liu <pi...@redhat.com>
Cc: Ard Biesheuvel <a...@kernel.org>
Cc: Mark Rutland <mark.rutl...@arm.com>
Cc: Will Deacon <w...@kernel.org>
Cc: Catalin Marinas <catalin.mari...@arm.com>
To: linux-arm-ker...@lists.infradead.org
---
 arch/arm64/include/asm/mmu.h |  6 ++++
 arch/arm64/mm/mmu.c          | 67 ++++++++++++++++++++----------------
 2 files changed, 44 insertions(+), 29 deletions(-)

diff --git a/arch/arm64/include/asm/mmu.h b/arch/arm64/include/asm/mmu.h
index 65977c7783c58..4b908e0650105 100644
--- a/arch/arm64/include/asm/mmu.h
+++ b/arch/arm64/include/asm/mmu.h
@@ -64,6 +64,12 @@ extern void arm64_memblock_init(void);
 extern void paging_init(void);
 extern void bootmem_init(void);
 extern void __iomem *early_io_map(phys_addr_t phys, unsigned long virt);
+extern void __create_pgd_mapping_locked(pgd_t *pgdir, phys_addr_t phys,
+                                       unsigned long virt, phys_addr_t size,
+                                       pgprot_t prot,
+                                       phys_addr_t (*pgtable_alloc)(int, void 
*),
+                                       void *arg,
+                                       int flags);
 extern void create_mapping_noalloc(phys_addr_t phys, unsigned long virt,
                                   phys_addr_t size, pgprot_t prot);
 extern void create_pgd_mapping(struct mm_struct *mm, phys_addr_t phys,
diff --git a/arch/arm64/mm/mmu.c b/arch/arm64/mm/mmu.c
index c927e9312f102..875e2bc0cfb39 100644
--- a/arch/arm64/mm/mmu.c
+++ b/arch/arm64/mm/mmu.c
@@ -106,7 +106,7 @@ pgprot_t phys_mem_access_prot(struct file *file, unsigned 
long pfn,
 }
 EXPORT_SYMBOL(phys_mem_access_prot);
 
-static phys_addr_t __init early_pgtable_alloc(int shift)
+static phys_addr_t __init early_pgtable_alloc(int shift, void *unused)
 {
        phys_addr_t phys;
 
@@ -190,7 +190,8 @@ static void init_pte(pte_t *ptep, unsigned long addr, 
unsigned long end,
 static void alloc_init_cont_pte(pmd_t *pmdp, unsigned long addr,
                                unsigned long end, phys_addr_t phys,
                                pgprot_t prot,
-                               phys_addr_t (*pgtable_alloc)(int),
+                               phys_addr_t (*pgtable_alloc)(int, void *arg),
+                               void *arg,
                                int flags)
 {
        unsigned long next;
@@ -205,7 +206,7 @@ static void alloc_init_cont_pte(pmd_t *pmdp, unsigned long 
addr,
                if (flags & NO_EXEC_MAPPINGS)
                        pmdval |= PMD_TABLE_PXN;
                BUG_ON(!pgtable_alloc);
-               pte_phys = pgtable_alloc(PAGE_SHIFT);
+               pte_phys = pgtable_alloc(PAGE_SHIFT, arg);
                ptep = pte_set_fixmap(pte_phys);
                init_clear_pgtable(ptep);
                ptep += pte_index(addr);
@@ -241,7 +242,9 @@ static void alloc_init_cont_pte(pmd_t *pmdp, unsigned long 
addr,
 
 static void init_pmd(pmd_t *pmdp, unsigned long addr, unsigned long end,
                     phys_addr_t phys, pgprot_t prot,
-                    phys_addr_t (*pgtable_alloc)(int), int flags)
+                    phys_addr_t (*pgtable_alloc)(int, void *),
+                    void *arg,
+                    int flags)
 {
        unsigned long next;
 
@@ -263,7 +266,7 @@ static void init_pmd(pmd_t *pmdp, unsigned long addr, 
unsigned long end,
                                                      
READ_ONCE(pmd_val(*pmdp))));
                } else {
                        alloc_init_cont_pte(pmdp, addr, next, phys, prot,
-                                           pgtable_alloc, flags);
+                                           pgtable_alloc, arg, flags);
 
                        BUG_ON(pmd_val(old_pmd) != 0 &&
                               pmd_val(old_pmd) != READ_ONCE(pmd_val(*pmdp)));
@@ -275,7 +278,9 @@ static void init_pmd(pmd_t *pmdp, unsigned long addr, 
unsigned long end,
 static void alloc_init_cont_pmd(pud_t *pudp, unsigned long addr,
                                unsigned long end, phys_addr_t phys,
                                pgprot_t prot,
-                               phys_addr_t (*pgtable_alloc)(int), int flags)
+                               phys_addr_t (*pgtable_alloc)(int, void *),
+                               void * arg,
+                               int flags)
 {
        unsigned long next;
        pud_t pud = READ_ONCE(*pudp);
@@ -292,7 +297,7 @@ static void alloc_init_cont_pmd(pud_t *pudp, unsigned long 
addr,
                if (flags & NO_EXEC_MAPPINGS)
                        pudval |= PUD_TABLE_PXN;
                BUG_ON(!pgtable_alloc);
-               pmd_phys = pgtable_alloc(PMD_SHIFT);
+               pmd_phys = pgtable_alloc(PMD_SHIFT, arg);
                pmdp = pmd_set_fixmap(pmd_phys);
                init_clear_pgtable(pmdp);
                pmdp += pmd_index(addr);
@@ -312,7 +317,7 @@ static void alloc_init_cont_pmd(pud_t *pudp, unsigned long 
addr,
                    (flags & NO_CONT_MAPPINGS) == 0)
                        __prot = __pgprot(pgprot_val(prot) | PTE_CONT);
 
-               init_pmd(pmdp, addr, next, phys, __prot, pgtable_alloc, flags);
+               init_pmd(pmdp, addr, next, phys, __prot, pgtable_alloc, arg, 
flags);
 
                pmdp += pmd_index(next) - pmd_index(addr);
                phys += next - addr;
@@ -323,7 +328,8 @@ static void alloc_init_cont_pmd(pud_t *pudp, unsigned long 
addr,
 
 static void alloc_init_pud(p4d_t *p4dp, unsigned long addr, unsigned long end,
                           phys_addr_t phys, pgprot_t prot,
-                          phys_addr_t (*pgtable_alloc)(int),
+                          phys_addr_t (*pgtable_alloc)(int, void *),
+                          void *arg,
                           int flags)
 {
        unsigned long next;
@@ -337,7 +343,7 @@ static void alloc_init_pud(p4d_t *p4dp, unsigned long addr, 
unsigned long end,
                if (flags & NO_EXEC_MAPPINGS)
                        p4dval |= P4D_TABLE_PXN;
                BUG_ON(!pgtable_alloc);
-               pud_phys = pgtable_alloc(PUD_SHIFT);
+               pud_phys = pgtable_alloc(PUD_SHIFT, arg);
                pudp = pud_set_fixmap(pud_phys);
                init_clear_pgtable(pudp);
                pudp += pud_index(addr);
@@ -368,7 +374,7 @@ static void alloc_init_pud(p4d_t *p4dp, unsigned long addr, 
unsigned long end,
                                                      
READ_ONCE(pud_val(*pudp))));
                } else {
                        alloc_init_cont_pmd(pudp, addr, next, phys, prot,
-                                           pgtable_alloc, flags);
+                                           pgtable_alloc, arg, flags);
 
                        BUG_ON(pud_val(old_pud) != 0 &&
                               pud_val(old_pud) != READ_ONCE(pud_val(*pudp)));
@@ -381,7 +387,8 @@ static void alloc_init_pud(p4d_t *p4dp, unsigned long addr, 
unsigned long end,
 
 static void alloc_init_p4d(pgd_t *pgdp, unsigned long addr, unsigned long end,
                           phys_addr_t phys, pgprot_t prot,
-                          phys_addr_t (*pgtable_alloc)(int),
+                          phys_addr_t (*pgtable_alloc)(int, void *),
+                          void *arg,
                           int flags)
 {
        unsigned long next;
@@ -395,7 +402,7 @@ static void alloc_init_p4d(pgd_t *pgdp, unsigned long addr, 
unsigned long end,
                if (flags & NO_EXEC_MAPPINGS)
                        pgdval |= PGD_TABLE_PXN;
                BUG_ON(!pgtable_alloc);
-               p4d_phys = pgtable_alloc(P4D_SHIFT);
+               p4d_phys = pgtable_alloc(P4D_SHIFT, arg);
                p4dp = p4d_set_fixmap(p4d_phys);
                init_clear_pgtable(p4dp);
                p4dp += p4d_index(addr);
@@ -411,7 +418,7 @@ static void alloc_init_p4d(pgd_t *pgdp, unsigned long addr, 
unsigned long end,
                next = p4d_addr_end(addr, end);
 
                alloc_init_pud(p4dp, addr, next, phys, prot,
-                              pgtable_alloc, flags);
+                              pgtable_alloc, arg, flags);
 
                BUG_ON(p4d_val(old_p4d) != 0 &&
                       p4d_val(old_p4d) != READ_ONCE(p4d_val(*p4dp)));
@@ -422,10 +429,11 @@ static void alloc_init_p4d(pgd_t *pgdp, unsigned long 
addr, unsigned long end,
        p4d_clear_fixmap();
 }
 
-static void __create_pgd_mapping_locked(pgd_t *pgdir, phys_addr_t phys,
+void __create_pgd_mapping_locked(pgd_t *pgdir, phys_addr_t phys,
                                        unsigned long virt, phys_addr_t size,
                                        pgprot_t prot,
-                                       phys_addr_t (*pgtable_alloc)(int),
+                                       phys_addr_t (*pgtable_alloc)(int, void 
*),
+                                       void *arg,
                                        int flags)
 {
        unsigned long addr, end, next;
@@ -445,7 +453,7 @@ static void __create_pgd_mapping_locked(pgd_t *pgdir, 
phys_addr_t phys,
        do {
                next = pgd_addr_end(addr, end);
                alloc_init_p4d(pgdp, addr, next, phys, prot, pgtable_alloc,
-                              flags);
+                              arg, flags);
                phys += next - addr;
        } while (pgdp++, addr = next, addr != end);
 }
@@ -453,12 +461,13 @@ static void __create_pgd_mapping_locked(pgd_t *pgdir, 
phys_addr_t phys,
 static void __create_pgd_mapping(pgd_t *pgdir, phys_addr_t phys,
                                 unsigned long virt, phys_addr_t size,
                                 pgprot_t prot,
-                                phys_addr_t (*pgtable_alloc)(int),
+                                phys_addr_t (*pgtable_alloc)(int, void *),
+                                void *arg,
                                 int flags)
 {
        mutex_lock(&fixmap_lock);
        __create_pgd_mapping_locked(pgdir, phys, virt, size, prot,
-                                   pgtable_alloc, flags);
+                                   pgtable_alloc, arg, flags);
        mutex_unlock(&fixmap_lock);
 }
 
@@ -466,10 +475,10 @@ static void __create_pgd_mapping(pgd_t *pgdir, 
phys_addr_t phys,
 extern __alias(__create_pgd_mapping_locked)
 void create_kpti_ng_temp_pgd(pgd_t *pgdir, phys_addr_t phys, unsigned long 
virt,
                             phys_addr_t size, pgprot_t prot,
-                            phys_addr_t (*pgtable_alloc)(int), int flags);
+                            phys_addr_t (*pgtable_alloc)(int, void *), void 
*arg, int flags);
 #endif
 
-static phys_addr_t __pgd_pgtable_alloc(int shift)
+static phys_addr_t __pgd_pgtable_alloc(int shift, void *unused)
 {
        /* Page is zeroed by init_clear_pgtable() so don't duplicate effort. */
        void *ptr = (void *)__get_free_page(GFP_PGTABLE_KERNEL & ~__GFP_ZERO);
@@ -478,9 +487,9 @@ static phys_addr_t __pgd_pgtable_alloc(int shift)
        return __pa(ptr);
 }
 
-static phys_addr_t pgd_pgtable_alloc(int shift)
+static phys_addr_t pgd_pgtable_alloc(int shift, void *unused)
 {
-       phys_addr_t pa = __pgd_pgtable_alloc(shift);
+       phys_addr_t pa = __pgd_pgtable_alloc(shift, unused);
        struct ptdesc *ptdesc = page_ptdesc(phys_to_page(pa));
 
        /*
@@ -512,7 +521,7 @@ void __init create_mapping_noalloc(phys_addr_t phys, 
unsigned long virt,
                        &phys, virt);
                return;
        }
-       __create_pgd_mapping(init_mm.pgd, phys, virt, size, prot, NULL,
+       __create_pgd_mapping(init_mm.pgd, phys, virt, size, prot, NULL, NULL,
                             NO_CONT_MAPPINGS);
 }
 
@@ -528,7 +537,7 @@ void __init create_pgd_mapping(struct mm_struct *mm, 
phys_addr_t phys,
                flags = NO_BLOCK_MAPPINGS | NO_CONT_MAPPINGS;
 
        __create_pgd_mapping(mm->pgd, phys, virt, size, prot,
-                            pgd_pgtable_alloc, flags);
+                            pgd_pgtable_alloc, NULL, flags);
 }
 
 static void update_mapping_prot(phys_addr_t phys, unsigned long virt,
@@ -540,7 +549,7 @@ static void update_mapping_prot(phys_addr_t phys, unsigned 
long virt,
                return;
        }
 
-       __create_pgd_mapping(init_mm.pgd, phys, virt, size, prot, NULL,
+       __create_pgd_mapping(init_mm.pgd, phys, virt, size, prot, NULL, NULL,
                             NO_CONT_MAPPINGS);
 
        /* flush the TLBs after updating live kernel mappings */
@@ -551,7 +560,7 @@ static void __init __map_memblock(pgd_t *pgdp, phys_addr_t 
start,
                                  phys_addr_t end, pgprot_t prot, int flags)
 {
        __create_pgd_mapping(pgdp, start, __phys_to_virt(start), end - start,
-                            prot, early_pgtable_alloc, flags);
+                            prot, early_pgtable_alloc, NULL, flags);
 }
 
 void __init mark_linear_text_alias_ro(void)
@@ -742,7 +751,7 @@ static int __init map_entry_trampoline(void)
        memset(tramp_pg_dir, 0, PGD_SIZE);
        __create_pgd_mapping(tramp_pg_dir, pa_start, TRAMP_VALIAS,
                             entry_tramp_text_size(), prot,
-                            __pgd_pgtable_alloc, NO_BLOCK_MAPPINGS);
+                            __pgd_pgtable_alloc, NULL, NO_BLOCK_MAPPINGS);
 
        /* Map both the text and data into the kernel page table */
        for (i = 0; i < DIV_ROUND_UP(entry_tramp_text_size(), PAGE_SIZE); i++)
@@ -1344,7 +1353,7 @@ int arch_add_memory(int nid, u64 start, u64 size,
                flags |= NO_BLOCK_MAPPINGS | NO_CONT_MAPPINGS;
 
        __create_pgd_mapping(swapper_pg_dir, start, __phys_to_virt(start),
-                            size, params->pgprot, __pgd_pgtable_alloc,
+                            size, params->pgprot, __pgd_pgtable_alloc, NULL,
                             flags);
 
        memblock_clear_nomap(start, size);
-- 
2.41.0


_______________________________________________
kexec mailing list
kexec@lists.infradead.org
http://lists.infradead.org/mailman/listinfo/kexec

Reply via email to