1 // SPDX-License-Identifier: GPL-2.0
2 #include <linux/compiler.h>
3 #include <linux/init.h>
4 #include <linux/export.h>
5 #include <linux/highmem.h>
6 #include <linux/sched.h>
8 #include <asm/fixmap.h>
9 #include <asm/tlbflush.h>
11 static pte_t *kmap_pte;
13 unsigned long highstart_pfn, highend_pfn;
15 void *kmap(struct page *page)
20 if (!PageHighMem(page))
21 return page_address(page);
22 addr = kmap_high(page);
23 flush_tlb_one((unsigned long)addr);
29 void kunmap(struct page *page)
31 BUG_ON(in_interrupt());
32 if (!PageHighMem(page))
36 EXPORT_SYMBOL(kunmap);
39 * kmap_atomic/kunmap_atomic is significantly faster than kmap/kunmap because
40 * no global lock is needed and because the kmap code must perform a global TLB
41 * invalidation when the kmap pool wraps.
43 * However when holding an atomic kmap is is not legal to sleep, so atomic
44 * kmaps are appropriate for short, tight code paths only.
47 void *kmap_atomic(struct page *page)
54 if (!PageHighMem(page))
55 return page_address(page);
57 type = kmap_atomic_idx_push();
58 idx = type + KM_TYPE_NR*smp_processor_id();
59 vaddr = __fix_to_virt(FIX_KMAP_BEGIN + idx);
60 #ifdef CONFIG_DEBUG_HIGHMEM
61 BUG_ON(!pte_none(*(kmap_pte - idx)));
63 set_pte(kmap_pte-idx, mk_pte(page, PAGE_KERNEL));
64 local_flush_tlb_one((unsigned long)vaddr);
68 EXPORT_SYMBOL(kmap_atomic);
70 void __kunmap_atomic(void *kvaddr)
72 unsigned long vaddr = (unsigned long) kvaddr & PAGE_MASK;
73 int type __maybe_unused;
75 if (vaddr < FIXADDR_START) { // FIXME
81 type = kmap_atomic_idx();
82 #ifdef CONFIG_DEBUG_HIGHMEM
84 int idx = type + KM_TYPE_NR * smp_processor_id();
86 BUG_ON(vaddr != __fix_to_virt(FIX_KMAP_BEGIN + idx));
89 * force other mappings to Oops if they'll try to access
90 * this pte without first remap it
92 pte_clear(&init_mm, vaddr, kmap_pte-idx);
93 local_flush_tlb_one(vaddr);
96 kmap_atomic_idx_pop();
100 EXPORT_SYMBOL(__kunmap_atomic);
103 * This is the same as kmap_atomic() but can map memory that doesn't
104 * have a struct page associated with it.
106 void *kmap_atomic_pfn(unsigned long pfn)
114 type = kmap_atomic_idx_push();
115 idx = type + KM_TYPE_NR*smp_processor_id();
116 vaddr = __fix_to_virt(FIX_KMAP_BEGIN + idx);
117 set_pte(kmap_pte-idx, pfn_pte(pfn, PAGE_KERNEL));
118 flush_tlb_one(vaddr);
120 return (void*) vaddr;
123 void __init kmap_init(void)
125 unsigned long kmap_vstart;
127 /* cache the first kmap pte */
128 kmap_vstart = __fix_to_virt(FIX_KMAP_BEGIN);
129 kmap_pte = kmap_get_fixmap_pte(kmap_vstart);