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