18c2ecf20Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
28c2ecf20Sopenharmony_ci/*
38c2ecf20Sopenharmony_ci * Copyright (C) 2020 Loongson Technology Corporation Limited
48c2ecf20Sopenharmony_ci */
58c2ecf20Sopenharmony_ci#include <linux/export.h>
68c2ecf20Sopenharmony_ci#include <linux/init.h>
78c2ecf20Sopenharmony_ci#include <linux/mm.h>
88c2ecf20Sopenharmony_ci#include <asm/fixmap.h>
98c2ecf20Sopenharmony_ci#include <asm/pgtable.h>
108c2ecf20Sopenharmony_ci#include <asm/pgalloc.h>
118c2ecf20Sopenharmony_ci#include <asm/tlbflush.h>
128c2ecf20Sopenharmony_ci
138c2ecf20Sopenharmony_civoid pgd_init(unsigned long page)
148c2ecf20Sopenharmony_ci{
158c2ecf20Sopenharmony_ci	unsigned long *p, *end;
168c2ecf20Sopenharmony_ci	unsigned long entry;
178c2ecf20Sopenharmony_ci
188c2ecf20Sopenharmony_ci#if !defined(__PAGETABLE_PUD_FOLDED)
198c2ecf20Sopenharmony_ci	entry = (unsigned long)invalid_pud_table;
208c2ecf20Sopenharmony_ci#elif !defined(__PAGETABLE_PMD_FOLDED)
218c2ecf20Sopenharmony_ci	entry = (unsigned long)invalid_pmd_table;
228c2ecf20Sopenharmony_ci#else
238c2ecf20Sopenharmony_ci	entry = (unsigned long)invalid_pte_table;
248c2ecf20Sopenharmony_ci#endif
258c2ecf20Sopenharmony_ci
268c2ecf20Sopenharmony_ci	p = (unsigned long *) page;
278c2ecf20Sopenharmony_ci	end = p + PTRS_PER_PGD;
288c2ecf20Sopenharmony_ci
298c2ecf20Sopenharmony_ci	do {
308c2ecf20Sopenharmony_ci		p[0] = entry;
318c2ecf20Sopenharmony_ci		p[1] = entry;
328c2ecf20Sopenharmony_ci		p[2] = entry;
338c2ecf20Sopenharmony_ci		p[3] = entry;
348c2ecf20Sopenharmony_ci		p[4] = entry;
358c2ecf20Sopenharmony_ci		p += 8;
368c2ecf20Sopenharmony_ci		p[-3] = entry;
378c2ecf20Sopenharmony_ci		p[-2] = entry;
388c2ecf20Sopenharmony_ci		p[-1] = entry;
398c2ecf20Sopenharmony_ci	} while (p != end);
408c2ecf20Sopenharmony_ci}
418c2ecf20Sopenharmony_ciEXPORT_SYMBOL_GPL(pgd_init);
428c2ecf20Sopenharmony_ci
438c2ecf20Sopenharmony_ci#ifndef __PAGETABLE_PMD_FOLDED
448c2ecf20Sopenharmony_civoid pmd_init(unsigned long addr, unsigned long pagetable)
458c2ecf20Sopenharmony_ci{
468c2ecf20Sopenharmony_ci	unsigned long *p, *end;
478c2ecf20Sopenharmony_ci
488c2ecf20Sopenharmony_ci	p = (unsigned long *) addr;
498c2ecf20Sopenharmony_ci	end = p + PTRS_PER_PMD;
508c2ecf20Sopenharmony_ci
518c2ecf20Sopenharmony_ci	do {
528c2ecf20Sopenharmony_ci		p[0] = pagetable;
538c2ecf20Sopenharmony_ci		p[1] = pagetable;
548c2ecf20Sopenharmony_ci		p[2] = pagetable;
558c2ecf20Sopenharmony_ci		p[3] = pagetable;
568c2ecf20Sopenharmony_ci		p[4] = pagetable;
578c2ecf20Sopenharmony_ci		p += 8;
588c2ecf20Sopenharmony_ci		p[-3] = pagetable;
598c2ecf20Sopenharmony_ci		p[-2] = pagetable;
608c2ecf20Sopenharmony_ci		p[-1] = pagetable;
618c2ecf20Sopenharmony_ci	} while (p != end);
628c2ecf20Sopenharmony_ci}
638c2ecf20Sopenharmony_ciEXPORT_SYMBOL_GPL(pmd_init);
648c2ecf20Sopenharmony_ci#endif
658c2ecf20Sopenharmony_ci
668c2ecf20Sopenharmony_ci#ifndef __PAGETABLE_PUD_FOLDED
678c2ecf20Sopenharmony_civoid pud_init(unsigned long addr, unsigned long pagetable)
688c2ecf20Sopenharmony_ci{
698c2ecf20Sopenharmony_ci	unsigned long *p, *end;
708c2ecf20Sopenharmony_ci
718c2ecf20Sopenharmony_ci	p = (unsigned long *)addr;
728c2ecf20Sopenharmony_ci	end = p + PTRS_PER_PUD;
738c2ecf20Sopenharmony_ci
748c2ecf20Sopenharmony_ci	do {
758c2ecf20Sopenharmony_ci		p[0] = pagetable;
768c2ecf20Sopenharmony_ci		p[1] = pagetable;
778c2ecf20Sopenharmony_ci		p[2] = pagetable;
788c2ecf20Sopenharmony_ci		p[3] = pagetable;
798c2ecf20Sopenharmony_ci		p[4] = pagetable;
808c2ecf20Sopenharmony_ci		p += 8;
818c2ecf20Sopenharmony_ci		p[-3] = pagetable;
828c2ecf20Sopenharmony_ci		p[-2] = pagetable;
838c2ecf20Sopenharmony_ci		p[-1] = pagetable;
848c2ecf20Sopenharmony_ci	} while (p != end);
858c2ecf20Sopenharmony_ci}
868c2ecf20Sopenharmony_ci#endif
878c2ecf20Sopenharmony_ci
888c2ecf20Sopenharmony_cipmd_t mk_pmd(struct page *page, pgprot_t prot)
898c2ecf20Sopenharmony_ci{
908c2ecf20Sopenharmony_ci	pmd_t pmd;
918c2ecf20Sopenharmony_ci
928c2ecf20Sopenharmony_ci	pmd_val(pmd) = (page_to_pfn(page) << _PFN_SHIFT) | pgprot_val(prot);
938c2ecf20Sopenharmony_ci
948c2ecf20Sopenharmony_ci	return pmd;
958c2ecf20Sopenharmony_ci}
968c2ecf20Sopenharmony_ci
978c2ecf20Sopenharmony_civoid set_pmd_at(struct mm_struct *mm, unsigned long addr,
988c2ecf20Sopenharmony_ci		pmd_t *pmdp, pmd_t pmd)
998c2ecf20Sopenharmony_ci{
1008c2ecf20Sopenharmony_ci	*pmdp = pmd;
1018c2ecf20Sopenharmony_ci	flush_tlb_all();
1028c2ecf20Sopenharmony_ci}
1038c2ecf20Sopenharmony_ci
1048c2ecf20Sopenharmony_civoid __init pagetable_init(void)
1058c2ecf20Sopenharmony_ci{
1068c2ecf20Sopenharmony_ci	unsigned long vaddr;
1078c2ecf20Sopenharmony_ci
1088c2ecf20Sopenharmony_ci	/* Initialize the entire pgd.  */
1098c2ecf20Sopenharmony_ci	pgd_init((unsigned long)swapper_pg_dir);
1108c2ecf20Sopenharmony_ci	pgd_init((unsigned long)invalid_pg_dir);
1118c2ecf20Sopenharmony_ci#ifndef __PAGETABLE_PUD_FOLDED
1128c2ecf20Sopenharmony_ci	pud_init((unsigned long)invalid_pud_table, (unsigned long)invalid_pmd_table);
1138c2ecf20Sopenharmony_ci#endif
1148c2ecf20Sopenharmony_ci#ifndef __PAGETABLE_PMD_FOLDED
1158c2ecf20Sopenharmony_ci	pmd_init((unsigned long)invalid_pmd_table, (unsigned long)invalid_pte_table);
1168c2ecf20Sopenharmony_ci#endif
1178c2ecf20Sopenharmony_ci	/*
1188c2ecf20Sopenharmony_ci	 * Fixed mappings:
1198c2ecf20Sopenharmony_ci	 */
1208c2ecf20Sopenharmony_ci	vaddr = __fix_to_virt(__end_of_fixed_addresses - 1) & PMD_MASK;
1218c2ecf20Sopenharmony_ci}
122