1 // SPDX-License-Identifier: GPL-2.0
2 // Copyright (C) 2018 Hangzhou C-SKY Microsystems co.,ltd.
4 #include <linux/module.h>
5 #include <linux/highmem.h>
7 #include <linux/memblock.h>
8 #include <asm/fixmap.h>
9 #include <asm/tlbflush.h>
10 #include <asm/cacheflush.h>
12 static pte_t *kmap_pte;
14 unsigned long highstart_pfn, highend_pfn;
16 void *kmap(struct page *page)
21 if (!PageHighMem(page))
22 return page_address(page);
23 addr = kmap_high(page);
24 flush_tlb_one((unsigned long)addr);
30 void kunmap(struct page *page)
32 BUG_ON(in_interrupt());
33 if (!PageHighMem(page))
37 EXPORT_SYMBOL(kunmap);
39 void *kmap_atomic(struct page *page)
46 if (!PageHighMem(page))
47 return page_address(page);
49 type = kmap_atomic_idx_push();
50 idx = type + KM_TYPE_NR*smp_processor_id();
51 vaddr = __fix_to_virt(FIX_KMAP_BEGIN + idx);
52 #ifdef CONFIG_DEBUG_HIGHMEM
53 BUG_ON(!pte_none(*(kmap_pte - idx)));
55 set_pte(kmap_pte-idx, mk_pte(page, PAGE_KERNEL));
56 flush_tlb_one((unsigned long)vaddr);
60 EXPORT_SYMBOL(kmap_atomic);
62 void __kunmap_atomic(void *kvaddr)
64 unsigned long vaddr = (unsigned long) kvaddr & PAGE_MASK;
67 if (vaddr < FIXADDR_START)
70 #ifdef CONFIG_DEBUG_HIGHMEM
71 idx = KM_TYPE_NR*smp_processor_id() + kmap_atomic_idx();
73 BUG_ON(vaddr != __fix_to_virt(FIX_KMAP_BEGIN + idx));
75 pte_clear(&init_mm, vaddr, kmap_pte - idx);
78 (void) idx; /* to kill a warning */
80 kmap_atomic_idx_pop();
85 EXPORT_SYMBOL(__kunmap_atomic);
88 * This is the same as kmap_atomic() but can map memory that doesn't
89 * have a struct page associated with it.
91 void *kmap_atomic_pfn(unsigned long pfn)
98 type = kmap_atomic_idx_push();
99 idx = type + KM_TYPE_NR*smp_processor_id();
100 vaddr = __fix_to_virt(FIX_KMAP_BEGIN + idx);
101 set_pte(kmap_pte-idx, pfn_pte(pfn, PAGE_KERNEL));
102 flush_tlb_one(vaddr);
104 return (void *) vaddr;
107 struct page *kmap_atomic_to_page(void *ptr)
109 unsigned long idx, vaddr = (unsigned long)ptr;
112 if (vaddr < FIXADDR_START)
113 return virt_to_page(ptr);
115 idx = virt_to_fix(vaddr);
116 pte = kmap_pte - (idx - FIX_KMAP_BEGIN);
117 return pte_page(*pte);
120 static void __init fixrange_init(unsigned long start, unsigned long end,
123 #ifdef CONFIG_HIGHMEM
132 i = __pgd_offset(vaddr);
133 j = __pud_offset(vaddr);
134 k = __pmd_offset(vaddr);
137 for ( ; (i < PTRS_PER_PGD) && (vaddr != end); pgd++, i++) {
139 for ( ; (j < PTRS_PER_PUD) && (vaddr != end); pud++, j++) {
141 for (; (k < PTRS_PER_PMD) && (vaddr != end); pmd++, k++) {
142 if (pmd_none(*pmd)) {
143 pte = (pte_t *) memblock_alloc_low(PAGE_SIZE, PAGE_SIZE);
144 set_pmd(pmd, __pmd(__pa(pte)));
145 BUG_ON(pte != pte_offset_kernel(pmd, 0));
156 void __init fixaddr_kmap_pages_init(void)
160 #ifdef CONFIG_HIGHMEM
166 pgd_base = swapper_pg_dir;
171 vaddr = __fix_to_virt(__end_of_fixed_addresses - 1) & PMD_MASK;
172 fixrange_init(vaddr, 0, pgd_base);
174 #ifdef CONFIG_HIGHMEM
179 fixrange_init(vaddr, vaddr + PAGE_SIZE*LAST_PKMAP, pgd_base);
181 pgd = swapper_pg_dir + __pgd_offset(vaddr);
183 pmd = pmd_offset(pud, vaddr);
184 pte = pte_offset_kernel(pmd, vaddr);
185 pkmap_page_table = pte;
189 void __init kmap_init(void)
193 fixaddr_kmap_pages_init();
195 vaddr = __fix_to_virt(FIX_KMAP_BEGIN);
197 kmap_pte = pte_offset_kernel((pmd_t *)pgd_offset_k(vaddr), vaddr);