[PATCHv3,1/5] arm64/mm: introduce pgtable allocator for idmap_pg_dir and init_pg_dir

Commit Message

Pingfan Liu May 31, 2021, 8:45 a.m. UTC

When building a pgtable, __create_pgd_mapping() resorts to pgtable
allocator to alloc mem.

By introducing an allocator, __create_pgd_mapping() can consider both
idmap_pg_dir and init_pg_dir as a memory pool, and get memory from them
during the building of pgtable.

Signed-off-by: Pingfan Liu <[e-mail]>
Cc: Catalin Marinas <[e-mail]>
Cc: Will Deacon <[e-mail]>
Cc: Ard Biesheuvel <[e-mail]>
Cc: Marc Zyngier <[e-mail]>
Cc: Kristina Martsenko <[e-mail]>
Cc: James Morse <[e-mail]>
Cc: Steven Price <[e-mail]>
Cc: Jonathan Cameron <[e-mail]>
Cc: Pavel Tatashin <[e-mail]>
Cc: Anshuman Khandual <[e-mail]>
Cc: Atish Patra <[e-mail]>
Cc: Mike Rapoport <[e-mail]>
Cc: Logan Gunthorpe <[e-mail]>
Cc: Mark Brown <[e-mail]>
To: [e-mail]
---
 arch/arm64/mm/mmu.c | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

Patch

diff --git a/arch/arm64/mm/mmu.c b/arch/arm64/mm/mmu.c
index 6dd9369e3ea0..f4fc905718ca 100644
--- a/arch/arm64/mm/mmu.c
+++ b/arch/arm64/mm/mmu.c
@@ -90,6 +90,28 @@  pgprot_t phys_mem_access_prot(struct file *file, unsigned long pfn,
 }
 EXPORT_SYMBOL(phys_mem_access_prot);
 
+struct headpool {
+	phys_addr_t start;
+	unsigned long size;
+	unsigned long next_idx;
+};
+
+static struct headpool cur_pool __initdata;
+
+void __init set_cur_headpool(phys_addr_t start, unsigned long size)
+{
+	cur_pool.start = start;
+	cur_pool.size = size;
+	cur_pool.next_idx = 0;
+}
+
+phys_addr_t __init head_pgtable_alloc(unsigned long unused_a)
+{
+	unsigned long idx = cur_pool.next_idx++;
+
+	return cur_pool.start + (idx << PAGE_SHIFT);
+}
+
 static phys_addr_t __init early_pgtable_alloc(int shift)
 {
 	phys_addr_t phys;