1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24#include <linux/highmem.h>
25#include <linux/module.h>
26
27
28
29
30
31
32
33void *kmap_atomic_prot(struct page *page, pgprot_t prot)
34{
35 unsigned long vaddr;
36 int idx, type;
37
38 preempt_disable();
39 pagefault_disable();
40 if (!PageHighMem(page))
41 return page_address(page);
42
43 type = kmap_atomic_idx_push();
44 idx = type + KM_TYPE_NR*smp_processor_id();
45 vaddr = __fix_to_virt(FIX_KMAP_BEGIN + idx);
46#ifdef CONFIG_DEBUG_HIGHMEM
47 BUG_ON(!pte_none(*(kmap_pte-idx)));
48#endif
49 __set_pte_at(&init_mm, vaddr, kmap_pte-idx, mk_pte(page, prot), 1);
50 local_flush_tlb_page(NULL, vaddr);
51
52 return (void*) vaddr;
53}
54EXPORT_SYMBOL(kmap_atomic_prot);
55
56void __kunmap_atomic(void *kvaddr)
57{
58 unsigned long vaddr = (unsigned long) kvaddr & PAGE_MASK;
59 int type __maybe_unused;
60
61 if (vaddr < __fix_to_virt(FIX_KMAP_END)) {
62 pagefault_enable();
63 preempt_enable();
64 return;
65 }
66
67 type = kmap_atomic_idx();
68
69#ifdef CONFIG_DEBUG_HIGHMEM
70 {
71 unsigned int idx;
72
73 idx = type + KM_TYPE_NR * smp_processor_id();
74 BUG_ON(vaddr != __fix_to_virt(FIX_KMAP_BEGIN + idx));
75
76
77
78
79
80 pte_clear(&init_mm, vaddr, kmap_pte-idx);
81 local_flush_tlb_page(NULL, vaddr);
82 }
83#endif
84
85 kmap_atomic_idx_pop();
86 pagefault_enable();
87 preempt_enable();
88}
89EXPORT_SYMBOL(__kunmap_atomic);
90