arch/loongarch/mm/init.c

   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/initrd.h>
  26 #include <linux/mmzone.h>
  27
  28 #include <asm/asm-offsets.h>
  29 #include <asm/bootinfo.h>
  30 #include <asm/cpu.h>
  31 #include <asm/dma.h>
  32 #include <asm/mmu_context.h>
  33 #include <asm/sections.h>
  34 #include <asm/pgtable.h>
  35 #include <asm/pgalloc.h>
  36 #include <asm/tlb.h>
  37
  38 /*
  39  * We have up to 8 empty zeroed pages so we can map one of the right colour
  40  * when needed.  Since page is never written to after the initialization we
  41  * don't have to care about aliases on other CPUs.
  42  */
  43 unsigned long empty_zero_page, zero_page_mask;
  44 EXPORT_SYMBOL_GPL(empty_zero_page);
  45 EXPORT_SYMBOL(zero_page_mask);
  46
  47 void setup_zero_pages(void)
  48 {
  49         unsigned int order, i;
  50         struct page *page;
  51
  52         order = 0;
  53
  54         empty_zero_page = __get_free_pages(GFP_KERNEL | __GFP_ZERO, order);
  55         if (!empty_zero_page)
  56                 panic("Oh boy, that early out of memory?");
  57
  58         page = virt_to_page((void *)empty_zero_page);
  59         split_page(page, order);
  60         for (i = 0; i < (1 << order); i++, page++)
  61                 mark_page_reserved(page);
  62
  63         zero_page_mask = ((PAGE_SIZE << order) - 1) & PAGE_MASK;
  64 }
  65
  66 void copy_user_highpage(struct page *to, struct page *from,
  67         unsigned long vaddr, struct vm_area_struct *vma)
  68 {
  69         void *vfrom, *vto;
  70
  71         vto = kmap_atomic(to);
  72         vfrom = kmap_atomic(from);
  73         copy_page(vto, vfrom);
  74         kunmap_atomic(vfrom);
  75         kunmap_atomic(vto);
  76         /* Make sure this page is cleared on other CPU's too before using it */
  77         smp_wmb();
  78 }
  79
  80 int __ref page_is_ram(unsigned long pfn)
  81 {
  82         unsigned long addr = PFN_PHYS(pfn);
  83
  84         return memblock_is_memory(addr) && !memblock_is_reserved(addr);
  85 }
  86
  87 #ifndef CONFIG_NUMA
  88 void __init paging_init(void)
  89 {
  90         unsigned long max_zone_pfns[MAX_NR_ZONES];
  91
  92 #ifdef CONFIG_ZONE_DMA
  93         max_zone_pfns[ZONE_DMA] = MAX_DMA_PFN;
  94 #endif
  95 #ifdef CONFIG_ZONE_DMA32
  96         max_zone_pfns[ZONE_DMA32] = MAX_DMA32_PFN;
  97 #endif
  98         max_zone_pfns[ZONE_NORMAL] = max_low_pfn;
  99
 100         free_area_init(max_zone_pfns);
 101 }
 102
 103 void __init mem_init(void)
 104 {
 105         max_mapnr = max_low_pfn;
 106         high_memory = (void *) __va(max_low_pfn << PAGE_SHIFT);
 107
 108         memblock_free_all();
 109         setup_zero_pages();     /* Setup zeroed pages.  */
 110 }
 111 #endif /* !CONFIG_NUMA */
 112
 113 void __ref free_initmem(void)
 114 {
 115         free_initmem_default(POISON_FREE_INITMEM);
 116 }
 117
 118 #ifdef CONFIG_MEMORY_HOTPLUG
 119 int arch_add_memory(int nid, u64 start, u64 size, struct mhp_params *params)
 120 {
 121         unsigned long start_pfn = start >> PAGE_SHIFT;
 122         unsigned long nr_pages = size >> PAGE_SHIFT;
 123         int ret;
 124
 125         ret = __add_pages(nid, start_pfn, nr_pages, params);
 126
 127         if (ret)
 128                 pr_warn("%s: Problem encountered in __add_pages() as ret=%d\n",
 129                                 __func__,  ret);
 130
 131         return ret;
 132 }
 133
 134 void arch_remove_memory(u64 start, u64 size, struct vmem_altmap *altmap)
 135 {
 136         unsigned long start_pfn = start >> PAGE_SHIFT;
 137         unsigned long nr_pages = size >> PAGE_SHIFT;
 138         struct page *page = pfn_to_page(start_pfn);
 139
 140         /* With altmap the first mapped page is offset from @start */
 141         if (altmap)
 142                 page += vmem_altmap_offset(altmap);
 143         __remove_pages(start_pfn, nr_pages, altmap);
 144 }
 145
 146 #ifdef CONFIG_NUMA
 147 int memory_add_physaddr_to_nid(u64 start)
 148 {
 149         return pa_to_nid(start);
 150 }
 151 EXPORT_SYMBOL_GPL(memory_add_physaddr_to_nid);
 152 #endif
 153 #endif
 154
 155 static pte_t *fixmap_pte(unsigned long addr)
 156 {
 157         pgd_t *pgd;
 158         p4d_t *p4d;
 159         pud_t *pud;
 160         pmd_t *pmd;
 161
 162         pgd = pgd_offset_k(addr);
 163         p4d = p4d_offset(pgd, addr);
 164
 165         if (pgd_none(*pgd)) {
 166                 pud_t *new __maybe_unused;
 167
 168                 new = memblock_alloc_low(PAGE_SIZE, PAGE_SIZE);
 169                 pgd_populate(&init_mm, pgd, new);
 170 #ifndef __PAGETABLE_PUD_FOLDED
 171                 pud_init((unsigned long)new, (unsigned long)invalid_pmd_table);
 172 #endif
 173         }
 174
 175         pud = pud_offset(p4d, addr);
 176         if (pud_none(*pud)) {
 177                 pmd_t *new __maybe_unused;
 178
 179                 new = memblock_alloc_low(PAGE_SIZE, PAGE_SIZE);
 180                 pud_populate(&init_mm, pud, new);
 181 #ifndef __PAGETABLE_PMD_FOLDED
 182                 pmd_init((unsigned long)new, (unsigned long)invalid_pte_table);
 183 #endif
 184         }
 185
 186         pmd = pmd_offset(pud, addr);
 187         if (pmd_none(*pmd)) {
 188                 pte_t *new __maybe_unused;
 189
 190                 new = memblock_alloc_low(PAGE_SIZE, PAGE_SIZE);
 191                 pmd_populate_kernel(&init_mm, pmd, new);
 192         }
 193
 194         return pte_offset_kernel(pmd, addr);
 195 }
 196
 197 void __init __set_fixmap(enum fixed_addresses idx,
 198                                phys_addr_t phys, pgprot_t flags)
 199 {
 200         unsigned long addr = __fix_to_virt(idx);
 201         pte_t *ptep;
 202
 203         BUG_ON(idx <= FIX_HOLE || idx >= __end_of_fixed_addresses);
 204
 205         ptep = fixmap_pte(addr);
 206         if (!pte_none(*ptep)) {
 207                 pte_ERROR(*ptep);
 208                 return;
 209         }
 210
 211         if (pgprot_val(flags))
 212                 set_pte(ptep, pfn_pte(phys >> PAGE_SHIFT, flags));
 213         else {
 214                 pte_clear(&init_mm, addr, ptep);
 215                 flush_tlb_kernel_range(addr, addr + PAGE_SIZE);
 216         }
 217 }
 218
 219 /*
 220  * Align swapper_pg_dir in to 64K, allows its address to be loaded
 221  * with a single LUI instruction in the TLB handlers.  If we used
 222  * __aligned(64K), its size would get rounded up to the alignment
 223  * size, and waste space.  So we place it in its own section and align
 224  * it in the linker script.
 225  */
 226 pgd_t swapper_pg_dir[_PTRS_PER_PGD] __section(".bss..swapper_pg_dir");
 227
 228 pgd_t invalid_pg_dir[_PTRS_PER_PGD] __page_aligned_bss;
 229 #ifndef __PAGETABLE_PUD_FOLDED
 230 pud_t invalid_pud_table[PTRS_PER_PUD] __page_aligned_bss;
 231 #endif
 232 #ifndef __PAGETABLE_PMD_FOLDED
 233 pmd_t invalid_pmd_table[PTRS_PER_PMD] __page_aligned_bss;
 234 EXPORT_SYMBOL_GPL(invalid_pmd_table);
 235 #endif
 236 pte_t invalid_pte_table[PTRS_PER_PTE] __page_aligned_bss;
 237 EXPORT_SYMBOL(invalid_pte_table);