1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * Copyright (C) 2020-2022 Loongson Technology Corporation Limited
4 */
5 #include <linux/init.h>
6 #include <linux/export.h>
7 #include <linux/signal.h>
8 #include <linux/sched.h>
9 #include <linux/smp.h>
10 #include <linux/kernel.h>
11 #include <linux/errno.h>
12 #include <linux/string.h>
13 #include <linux/types.h>
14 #include <linux/pagemap.h>
15 #include <linux/memblock.h>
16 #include <linux/memremap.h>
17 #include <linux/mm.h>
18 #include <linux/mman.h>
19 #include <linux/highmem.h>
20 #include <linux/swap.h>
21 #include <linux/proc_fs.h>
22 #include <linux/pfn.h>
23 #include <linux/hardirq.h>
24 #include <linux/gfp.h>
25 #include <linux/hugetlb.h>
26 #include <linux/mmzone.h>
27 #include <linux/execmem.h>
28
29 #include <asm/asm-offsets.h>
30 #include <asm/bootinfo.h>
31 #include <asm/cpu.h>
32 #include <asm/dma.h>
33 #include <asm/mmu_context.h>
34 #include <asm/sections.h>
35 #include <asm/pgtable.h>
36 #include <asm/pgalloc.h>
37 #include <asm/tlb.h>
38
39 unsigned long empty_zero_page[PAGE_SIZE / sizeof(unsigned long)] __page_aligned_bss;
40 EXPORT_SYMBOL(empty_zero_page);
41
copy_user_highpage(struct page * to,struct page * from,unsigned long vaddr,struct vm_area_struct * vma)42 void copy_user_highpage(struct page *to, struct page *from,
43 unsigned long vaddr, struct vm_area_struct *vma)
44 {
45 void *vfrom, *vto;
46
47 vfrom = kmap_local_page(from);
48 vto = kmap_local_page(to);
49 copy_page(vto, vfrom);
50 kunmap_local(vfrom);
51 kunmap_local(vto);
52 /* Make sure this page is cleared on other CPU's too before using it */
53 smp_wmb();
54 }
55
page_is_ram(unsigned long pfn)56 int __ref page_is_ram(unsigned long pfn)
57 {
58 unsigned long addr = PFN_PHYS(pfn);
59
60 return memblock_is_memory(addr) && !memblock_is_reserved(addr);
61 }
62
63 #ifndef CONFIG_NUMA
paging_init(void)64 void __init paging_init(void)
65 {
66 unsigned long max_zone_pfns[MAX_NR_ZONES];
67
68 #ifdef CONFIG_ZONE_DMA32
69 max_zone_pfns[ZONE_DMA32] = MAX_DMA32_PFN;
70 #endif
71 max_zone_pfns[ZONE_NORMAL] = max_low_pfn;
72
73 free_area_init(max_zone_pfns);
74 }
75
mem_init(void)76 void __init mem_init(void)
77 {
78 max_mapnr = max_low_pfn;
79 high_memory = (void *) __va(max_low_pfn << PAGE_SHIFT);
80
81 memblock_free_all();
82 }
83 #endif /* !CONFIG_NUMA */
84
free_initmem(void)85 void __ref free_initmem(void)
86 {
87 free_initmem_default(POISON_FREE_INITMEM);
88 }
89
90 #ifdef CONFIG_MEMORY_HOTPLUG
arch_add_memory(int nid,u64 start,u64 size,struct mhp_params * params)91 int arch_add_memory(int nid, u64 start, u64 size, struct mhp_params *params)
92 {
93 unsigned long start_pfn = start >> PAGE_SHIFT;
94 unsigned long nr_pages = size >> PAGE_SHIFT;
95 int ret;
96
97 ret = __add_pages(nid, start_pfn, nr_pages, params);
98
99 if (ret)
100 pr_warn("%s: Problem encountered in __add_pages() as ret=%d\n",
101 __func__, ret);
102
103 return ret;
104 }
105
arch_remove_memory(u64 start,u64 size,struct vmem_altmap * altmap)106 void arch_remove_memory(u64 start, u64 size, struct vmem_altmap *altmap)
107 {
108 unsigned long start_pfn = start >> PAGE_SHIFT;
109 unsigned long nr_pages = size >> PAGE_SHIFT;
110 struct page *page = pfn_to_page(start_pfn);
111
112 /* With altmap the first mapped page is offset from @start */
113 if (altmap)
114 page += vmem_altmap_offset(altmap);
115 __remove_pages(start_pfn, nr_pages, altmap);
116 }
117
118 #ifdef CONFIG_NUMA
memory_add_physaddr_to_nid(u64 start)119 int memory_add_physaddr_to_nid(u64 start)
120 {
121 return pa_to_nid(start);
122 }
123 EXPORT_SYMBOL_GPL(memory_add_physaddr_to_nid);
124 #endif
125 #endif
126
127 #ifdef CONFIG_SPARSEMEM_VMEMMAP
vmemmap_set_pmd(pmd_t * pmd,void * p,int node,unsigned long addr,unsigned long next)128 void __meminit vmemmap_set_pmd(pmd_t *pmd, void *p, int node,
129 unsigned long addr, unsigned long next)
130 {
131 pmd_t entry;
132
133 entry = pfn_pmd(virt_to_pfn(p), PAGE_KERNEL);
134 pmd_val(entry) |= _PAGE_HUGE | _PAGE_HGLOBAL;
135 set_pmd_at(&init_mm, addr, pmd, entry);
136 }
137
vmemmap_check_pmd(pmd_t * pmd,int node,unsigned long addr,unsigned long next)138 int __meminit vmemmap_check_pmd(pmd_t *pmd, int node,
139 unsigned long addr, unsigned long next)
140 {
141 int huge = pmd_val(pmdp_get(pmd)) & _PAGE_HUGE;
142
143 if (huge)
144 vmemmap_verify((pte_t *)pmd, node, addr, next);
145
146 return huge;
147 }
148
vmemmap_populate(unsigned long start,unsigned long end,int node,struct vmem_altmap * altmap)149 int __meminit vmemmap_populate(unsigned long start, unsigned long end,
150 int node, struct vmem_altmap *altmap)
151 {
152 #if CONFIG_PGTABLE_LEVELS == 2
153 return vmemmap_populate_basepages(start, end, node, NULL);
154 #else
155 return vmemmap_populate_hugepages(start, end, node, NULL);
156 #endif
157 }
158
159 #ifdef CONFIG_MEMORY_HOTPLUG
vmemmap_free(unsigned long start,unsigned long end,struct vmem_altmap * altmap)160 void vmemmap_free(unsigned long start, unsigned long end, struct vmem_altmap *altmap)
161 {
162 }
163 #endif
164 #endif
165
populate_kernel_pte(unsigned long addr)166 pte_t * __init populate_kernel_pte(unsigned long addr)
167 {
168 pgd_t *pgd = pgd_offset_k(addr);
169 p4d_t *p4d = p4d_offset(pgd, addr);
170 pud_t *pud;
171 pmd_t *pmd;
172
173 if (p4d_none(p4dp_get(p4d))) {
174 pud = memblock_alloc(PAGE_SIZE, PAGE_SIZE);
175 if (!pud)
176 panic("%s: Failed to allocate memory\n", __func__);
177 p4d_populate(&init_mm, p4d, pud);
178 #ifndef __PAGETABLE_PUD_FOLDED
179 pud_init(pud);
180 #endif
181 }
182
183 pud = pud_offset(p4d, addr);
184 if (pud_none(pudp_get(pud))) {
185 pmd = memblock_alloc(PAGE_SIZE, PAGE_SIZE);
186 if (!pmd)
187 panic("%s: Failed to allocate memory\n", __func__);
188 pud_populate(&init_mm, pud, pmd);
189 #ifndef __PAGETABLE_PMD_FOLDED
190 pmd_init(pmd);
191 #endif
192 }
193
194 pmd = pmd_offset(pud, addr);
195 if (!pmd_present(pmdp_get(pmd))) {
196 pte_t *pte;
197
198 pte = memblock_alloc(PAGE_SIZE, PAGE_SIZE);
199 if (!pte)
200 panic("%s: Failed to allocate memory\n", __func__);
201
202 pmd_populate_kernel(&init_mm, pmd, pte);
203 kernel_pte_init(pte);
204 }
205
206 return pte_offset_kernel(pmd, addr);
207 }
208
__set_fixmap(enum fixed_addresses idx,phys_addr_t phys,pgprot_t flags)209 void __init __set_fixmap(enum fixed_addresses idx,
210 phys_addr_t phys, pgprot_t flags)
211 {
212 unsigned long addr = __fix_to_virt(idx);
213 pte_t *ptep;
214
215 BUG_ON(idx <= FIX_HOLE || idx >= __end_of_fixed_addresses);
216
217 ptep = populate_kernel_pte(addr);
218 if (!pte_none(ptep_get(ptep))) {
219 pte_ERROR(*ptep);
220 return;
221 }
222
223 if (pgprot_val(flags))
224 set_pte(ptep, pfn_pte(phys >> PAGE_SHIFT, flags));
225 else {
226 pte_clear(&init_mm, addr, ptep);
227 flush_tlb_kernel_range(addr, addr + PAGE_SIZE);
228 }
229 }
230
231 /*
232 * Align swapper_pg_dir in to 64K, allows its address to be loaded
233 * with a single LUI instruction in the TLB handlers. If we used
234 * __aligned(64K), its size would get rounded up to the alignment
235 * size, and waste space. So we place it in its own section and align
236 * it in the linker script.
237 */
238 pgd_t swapper_pg_dir[_PTRS_PER_PGD] __section(".bss..swapper_pg_dir");
239
240 pgd_t invalid_pg_dir[_PTRS_PER_PGD] __page_aligned_bss;
241 #ifndef __PAGETABLE_PUD_FOLDED
242 pud_t invalid_pud_table[PTRS_PER_PUD] __page_aligned_bss;
243 EXPORT_SYMBOL(invalid_pud_table);
244 #endif
245 #ifndef __PAGETABLE_PMD_FOLDED
246 pmd_t invalid_pmd_table[PTRS_PER_PMD] __page_aligned_bss;
247 EXPORT_SYMBOL(invalid_pmd_table);
248 #endif
249 pte_t invalid_pte_table[PTRS_PER_PTE] __page_aligned_bss;
250 EXPORT_SYMBOL(invalid_pte_table);
251
252 #ifdef CONFIG_EXECMEM
253 static struct execmem_info execmem_info __ro_after_init;
254
execmem_arch_setup(void)255 struct execmem_info __init *execmem_arch_setup(void)
256 {
257 execmem_info = (struct execmem_info){
258 .ranges = {
259 [EXECMEM_DEFAULT] = {
260 .start = MODULES_VADDR,
261 .end = MODULES_END,
262 .pgprot = PAGE_KERNEL,
263 .alignment = 1,
264 },
265 },
266 };
267
268 return &execmem_info;
269 }
270 #endif /* CONFIG_EXECMEM */
271