1 // SPDX-License-Identifier: GPL-2.0
2 #include <linux/module.h>
3 #include <linux/kernel.h>
4 #include <linux/slab.h>
5 #include <linux/mm_types.h>
6 #include <linux/pgtable.h>
7
8 #include <asm/cputype.h>
9 #include <asm/idmap.h>
10 #include <asm/hwcap.h>
11 #include <asm/pgalloc.h>
12 #include <asm/sections.h>
13 #include <asm/system_info.h>
14
15 /*
16 * Note: accesses outside of the kernel image and the identity map area
17 * are not supported on any CPU using the idmap tables as its current
18 * page tables.
19 */
20 pgd_t *idmap_pgd __ro_after_init;
21 long long arch_phys_to_idmap_offset __ro_after_init;
22
23 #ifdef CONFIG_ARM_LPAE
idmap_add_pmd(pud_t * pud,unsigned long addr,unsigned long end,unsigned long prot)24 static void idmap_add_pmd(pud_t *pud, unsigned long addr, unsigned long end,
25 unsigned long prot)
26 {
27 pmd_t *pmd;
28 unsigned long next;
29
30 if (pud_none_or_clear_bad(pud) || (pud_val(*pud) & L_PGD_SWAPPER)) {
31 pmd = pmd_alloc_one(&init_mm, addr);
32 if (!pmd) {
33 pr_warn("Failed to allocate identity pmd.\n");
34 return;
35 }
36 /*
37 * Copy the original PMD to ensure that the PMD entries for
38 * the kernel image are preserved.
39 */
40 if (!pud_none(*pud))
41 memcpy(pmd, pmd_offset(pud, 0),
42 PTRS_PER_PMD * sizeof(pmd_t));
43 pud_populate(&init_mm, pud, pmd);
44 pmd += pmd_index(addr);
45 } else
46 pmd = pmd_offset(pud, addr);
47
48 do {
49 next = pmd_addr_end(addr, end);
50 *pmd = __pmd((addr & PMD_MASK) | prot);
51 flush_pmd_entry(pmd);
52 } while (pmd++, addr = next, addr != end);
53 }
54 #else /* !CONFIG_ARM_LPAE */
idmap_add_pmd(pud_t * pud,unsigned long addr,unsigned long end,unsigned long prot)55 static void idmap_add_pmd(pud_t *pud, unsigned long addr, unsigned long end,
56 unsigned long prot)
57 {
58 pmd_t *pmd = pmd_offset(pud, addr);
59
60 addr = (addr & PMD_MASK) | prot;
61 pmd[0] = __pmd(addr);
62 addr += SECTION_SIZE;
63 pmd[1] = __pmd(addr);
64 flush_pmd_entry(pmd);
65 }
66 #endif /* CONFIG_ARM_LPAE */
67
idmap_add_pud(pgd_t * pgd,unsigned long addr,unsigned long end,unsigned long prot)68 static void idmap_add_pud(pgd_t *pgd, unsigned long addr, unsigned long end,
69 unsigned long prot)
70 {
71 p4d_t *p4d = p4d_offset(pgd, addr);
72 pud_t *pud = pud_offset(p4d, addr);
73 unsigned long next;
74
75 do {
76 next = pud_addr_end(addr, end);
77 idmap_add_pmd(pud, addr, next, prot);
78 } while (pud++, addr = next, addr != end);
79 }
80
identity_mapping_add(pgd_t * pgd,const char * text_start,const char * text_end,unsigned long prot)81 static void identity_mapping_add(pgd_t *pgd, const char *text_start,
82 const char *text_end, unsigned long prot)
83 {
84 unsigned long addr, end;
85 unsigned long next;
86
87 #ifdef CONFIG_XIP_KERNEL
88 addr = (phys_addr_t)(text_start) - XIP_VIRT_ADDR(CONFIG_XIP_PHYS_ADDR)
89 + CONFIG_XIP_PHYS_ADDR;
90 end = (phys_addr_t)(text_end) - XIP_VIRT_ADDR(CONFIG_XIP_PHYS_ADDR)
91 + CONFIG_XIP_PHYS_ADDR;
92 #else
93 addr = virt_to_idmap(text_start);
94 end = virt_to_idmap(text_end);
95 #endif
96 pr_info("Setting up static identity map for 0x%lx - 0x%lx\n", addr, end);
97
98 prot |= PMD_TYPE_SECT | PMD_SECT_AP_WRITE | PMD_SECT_AF;
99
100 if (cpu_architecture() <= CPU_ARCH_ARMv5TEJ && !cpu_is_xscale_family())
101 prot |= PMD_BIT4;
102
103 pgd += pgd_index(addr);
104 do {
105 next = pgd_addr_end(addr, end);
106 idmap_add_pud(pgd, addr, next, prot);
107 } while (pgd++, addr = next, addr != end);
108 }
109
110 extern char __idmap_text_start[], __idmap_text_end[];
111
init_static_idmap(void)112 static int __init init_static_idmap(void)
113 {
114 idmap_pgd = pgd_alloc(&init_mm);
115 if (!idmap_pgd)
116 return -ENOMEM;
117
118 identity_mapping_add(idmap_pgd, __idmap_text_start,
119 __idmap_text_end, 0);
120
121 /* Flush L1 for the hardware to see this page table content */
122 if (!(elf_hwcap & HWCAP_LPAE))
123 flush_cache_louis();
124
125 return 0;
126 }
127 early_initcall(init_static_idmap);
128
129 /*
130 * In order to soft-boot, we need to switch to a 1:1 mapping for the
131 * cpu_reset functions. This will then ensure that we have predictable
132 * results when turning off the mmu.
133 */
setup_mm_for_reboot(void)134 void setup_mm_for_reboot(void)
135 {
136 /* Switch to the identity mapping. */
137 cpu_switch_mm(idmap_pgd, &init_mm);
138 local_flush_bp_all();
139
140 #ifdef CONFIG_CPU_HAS_ASID
141 /*
142 * We don't have a clean ASID for the identity mapping, which
143 * may clash with virtual addresses of the previous page tables
144 * and therefore potentially in the TLB.
145 */
146 local_flush_tlb_all();
147 #endif
148 }
149