162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-or-later
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * This file contains the routines setting up the linux page tables.
462306a36Sopenharmony_ci *  -- paulus
562306a36Sopenharmony_ci *
662306a36Sopenharmony_ci *  Derived from arch/ppc/mm/init.c:
762306a36Sopenharmony_ci *    Copyright (C) 1995-1996 Gary Thomas (gdt@linuxppc.org)
862306a36Sopenharmony_ci *
962306a36Sopenharmony_ci *  Modifications by Paul Mackerras (PowerMac) (paulus@cs.anu.edu.au)
1062306a36Sopenharmony_ci *  and Cort Dougan (PReP) (cort@cs.nmt.edu)
1162306a36Sopenharmony_ci *    Copyright (C) 1996 Paul Mackerras
1262306a36Sopenharmony_ci *
1362306a36Sopenharmony_ci *  Derived from "arch/i386/mm/init.c"
1462306a36Sopenharmony_ci *    Copyright (C) 1991, 1992, 1993, 1994  Linus Torvalds
1562306a36Sopenharmony_ci */
1662306a36Sopenharmony_ci
1762306a36Sopenharmony_ci#include <linux/kernel.h>
1862306a36Sopenharmony_ci#include <linux/module.h>
1962306a36Sopenharmony_ci#include <linux/types.h>
2062306a36Sopenharmony_ci#include <linux/mm.h>
2162306a36Sopenharmony_ci#include <linux/vmalloc.h>
2262306a36Sopenharmony_ci#include <linux/init.h>
2362306a36Sopenharmony_ci#include <linux/highmem.h>
2462306a36Sopenharmony_ci#include <linux/memblock.h>
2562306a36Sopenharmony_ci#include <linux/slab.h>
2662306a36Sopenharmony_ci#include <linux/set_memory.h>
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_ci#include <asm/pgalloc.h>
2962306a36Sopenharmony_ci#include <asm/fixmap.h>
3062306a36Sopenharmony_ci#include <asm/setup.h>
3162306a36Sopenharmony_ci#include <asm/sections.h>
3262306a36Sopenharmony_ci#include <asm/early_ioremap.h>
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_ci#include <mm/mmu_decl.h>
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_cistatic u8 early_fixmap_pagetable[FIXMAP_PTE_SIZE] __page_aligned_data;
3762306a36Sopenharmony_ci
3862306a36Sopenharmony_cinotrace void __init early_ioremap_init(void)
3962306a36Sopenharmony_ci{
4062306a36Sopenharmony_ci	unsigned long addr = ALIGN_DOWN(FIXADDR_START, PGDIR_SIZE);
4162306a36Sopenharmony_ci	pte_t *ptep = (pte_t *)early_fixmap_pagetable;
4262306a36Sopenharmony_ci	pmd_t *pmdp = pmd_off_k(addr);
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_ci	for (; (s32)(FIXADDR_TOP - addr) > 0;
4562306a36Sopenharmony_ci	     addr += PGDIR_SIZE, ptep += PTRS_PER_PTE, pmdp++)
4662306a36Sopenharmony_ci		pmd_populate_kernel(&init_mm, pmdp, ptep);
4762306a36Sopenharmony_ci
4862306a36Sopenharmony_ci	early_ioremap_setup();
4962306a36Sopenharmony_ci}
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_cistatic void __init *early_alloc_pgtable(unsigned long size)
5262306a36Sopenharmony_ci{
5362306a36Sopenharmony_ci	void *ptr = memblock_alloc(size, size);
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci	if (!ptr)
5662306a36Sopenharmony_ci		panic("%s: Failed to allocate %lu bytes align=0x%lx\n",
5762306a36Sopenharmony_ci		      __func__, size, size);
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_ci	return ptr;
6062306a36Sopenharmony_ci}
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_cipte_t __init *early_pte_alloc_kernel(pmd_t *pmdp, unsigned long va)
6362306a36Sopenharmony_ci{
6462306a36Sopenharmony_ci	if (pmd_none(*pmdp)) {
6562306a36Sopenharmony_ci		pte_t *ptep = early_alloc_pgtable(PTE_FRAG_SIZE);
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_ci		pmd_populate_kernel(&init_mm, pmdp, ptep);
6862306a36Sopenharmony_ci	}
6962306a36Sopenharmony_ci	return pte_offset_kernel(pmdp, va);
7062306a36Sopenharmony_ci}
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_ciint __ref map_kernel_page(unsigned long va, phys_addr_t pa, pgprot_t prot)
7462306a36Sopenharmony_ci{
7562306a36Sopenharmony_ci	pmd_t *pd;
7662306a36Sopenharmony_ci	pte_t *pg;
7762306a36Sopenharmony_ci	int err = -ENOMEM;
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_ci	/* Use upper 10 bits of VA to index the first level map */
8062306a36Sopenharmony_ci	pd = pmd_off_k(va);
8162306a36Sopenharmony_ci	/* Use middle 10 bits of VA to index the second-level map */
8262306a36Sopenharmony_ci	if (likely(slab_is_available()))
8362306a36Sopenharmony_ci		pg = pte_alloc_kernel(pd, va);
8462306a36Sopenharmony_ci	else
8562306a36Sopenharmony_ci		pg = early_pte_alloc_kernel(pd, va);
8662306a36Sopenharmony_ci	if (pg) {
8762306a36Sopenharmony_ci		err = 0;
8862306a36Sopenharmony_ci		/* The PTE should never be already set nor present in the
8962306a36Sopenharmony_ci		 * hash table
9062306a36Sopenharmony_ci		 */
9162306a36Sopenharmony_ci		BUG_ON((pte_present(*pg) | pte_hashpte(*pg)) && pgprot_val(prot));
9262306a36Sopenharmony_ci		set_pte_at(&init_mm, va, pg, pfn_pte(pa >> PAGE_SHIFT, prot));
9362306a36Sopenharmony_ci	}
9462306a36Sopenharmony_ci	smp_wmb();
9562306a36Sopenharmony_ci	return err;
9662306a36Sopenharmony_ci}
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_ci/*
9962306a36Sopenharmony_ci * Map in a chunk of physical memory starting at start.
10062306a36Sopenharmony_ci */
10162306a36Sopenharmony_cistatic void __init __mapin_ram_chunk(unsigned long offset, unsigned long top)
10262306a36Sopenharmony_ci{
10362306a36Sopenharmony_ci	unsigned long v, s;
10462306a36Sopenharmony_ci	phys_addr_t p;
10562306a36Sopenharmony_ci	bool ktext;
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_ci	s = offset;
10862306a36Sopenharmony_ci	v = PAGE_OFFSET + s;
10962306a36Sopenharmony_ci	p = memstart_addr + s;
11062306a36Sopenharmony_ci	for (; s < top; s += PAGE_SIZE) {
11162306a36Sopenharmony_ci		ktext = core_kernel_text(v);
11262306a36Sopenharmony_ci		map_kernel_page(v, p, ktext ? PAGE_KERNEL_TEXT : PAGE_KERNEL);
11362306a36Sopenharmony_ci		v += PAGE_SIZE;
11462306a36Sopenharmony_ci		p += PAGE_SIZE;
11562306a36Sopenharmony_ci	}
11662306a36Sopenharmony_ci}
11762306a36Sopenharmony_ci
11862306a36Sopenharmony_civoid __init mapin_ram(void)
11962306a36Sopenharmony_ci{
12062306a36Sopenharmony_ci	phys_addr_t base, end;
12162306a36Sopenharmony_ci	u64 i;
12262306a36Sopenharmony_ci
12362306a36Sopenharmony_ci	for_each_mem_range(i, &base, &end) {
12462306a36Sopenharmony_ci		phys_addr_t top = min(end, total_lowmem);
12562306a36Sopenharmony_ci
12662306a36Sopenharmony_ci		if (base >= top)
12762306a36Sopenharmony_ci			continue;
12862306a36Sopenharmony_ci		base = mmu_mapin_ram(base, top);
12962306a36Sopenharmony_ci		__mapin_ram_chunk(base, top);
13062306a36Sopenharmony_ci	}
13162306a36Sopenharmony_ci}
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_civoid mark_initmem_nx(void)
13462306a36Sopenharmony_ci{
13562306a36Sopenharmony_ci	unsigned long numpages = PFN_UP((unsigned long)_einittext) -
13662306a36Sopenharmony_ci				 PFN_DOWN((unsigned long)_sinittext);
13762306a36Sopenharmony_ci
13862306a36Sopenharmony_ci	mmu_mark_initmem_nx();
13962306a36Sopenharmony_ci
14062306a36Sopenharmony_ci	if (!v_block_mapped((unsigned long)_sinittext)) {
14162306a36Sopenharmony_ci		set_memory_nx((unsigned long)_sinittext, numpages);
14262306a36Sopenharmony_ci		set_memory_rw((unsigned long)_sinittext, numpages);
14362306a36Sopenharmony_ci	}
14462306a36Sopenharmony_ci}
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_ci#ifdef CONFIG_STRICT_KERNEL_RWX
14762306a36Sopenharmony_civoid mark_rodata_ro(void)
14862306a36Sopenharmony_ci{
14962306a36Sopenharmony_ci	unsigned long numpages;
15062306a36Sopenharmony_ci
15162306a36Sopenharmony_ci	if (IS_ENABLED(CONFIG_STRICT_MODULE_RWX) && mmu_has_feature(MMU_FTR_HPTE_TABLE))
15262306a36Sopenharmony_ci		pr_warn("This platform has HASH MMU, STRICT_MODULE_RWX won't work\n");
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_ci	if (v_block_mapped((unsigned long)_stext + 1)) {
15562306a36Sopenharmony_ci		mmu_mark_rodata_ro();
15662306a36Sopenharmony_ci		ptdump_check_wx();
15762306a36Sopenharmony_ci		return;
15862306a36Sopenharmony_ci	}
15962306a36Sopenharmony_ci
16062306a36Sopenharmony_ci	/*
16162306a36Sopenharmony_ci	 * mark text and rodata as read only. __end_rodata is set by
16262306a36Sopenharmony_ci	 * powerpc's linker script and includes tables and data
16362306a36Sopenharmony_ci	 * requiring relocation which are not put in RO_DATA.
16462306a36Sopenharmony_ci	 */
16562306a36Sopenharmony_ci	numpages = PFN_UP((unsigned long)__end_rodata) -
16662306a36Sopenharmony_ci		   PFN_DOWN((unsigned long)_stext);
16762306a36Sopenharmony_ci
16862306a36Sopenharmony_ci	set_memory_ro((unsigned long)_stext, numpages);
16962306a36Sopenharmony_ci
17062306a36Sopenharmony_ci	// mark_initmem_nx() should have already run by now
17162306a36Sopenharmony_ci	ptdump_check_wx();
17262306a36Sopenharmony_ci}
17362306a36Sopenharmony_ci#endif
17462306a36Sopenharmony_ci
17562306a36Sopenharmony_ci#if defined(CONFIG_ARCH_SUPPORTS_DEBUG_PAGEALLOC) && defined(CONFIG_DEBUG_PAGEALLOC)
17662306a36Sopenharmony_civoid __kernel_map_pages(struct page *page, int numpages, int enable)
17762306a36Sopenharmony_ci{
17862306a36Sopenharmony_ci	unsigned long addr = (unsigned long)page_address(page);
17962306a36Sopenharmony_ci
18062306a36Sopenharmony_ci	if (PageHighMem(page))
18162306a36Sopenharmony_ci		return;
18262306a36Sopenharmony_ci
18362306a36Sopenharmony_ci	if (enable)
18462306a36Sopenharmony_ci		set_memory_p(addr, numpages);
18562306a36Sopenharmony_ci	else
18662306a36Sopenharmony_ci		set_memory_np(addr, numpages);
18762306a36Sopenharmony_ci}
18862306a36Sopenharmony_ci#endif /* CONFIG_DEBUG_PAGEALLOC */
189