1
2
3
4
5
6
7
8
9#define KMSG_COMPONENT "hugetlb"
10#define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
11
12#include <linux/mm.h>
13#include <linux/hugetlb.h>
14
15
16
17
18
19#define move_set_bit(x, a, b) (((x) & (a)) >> ilog2(a) << ilog2(b))
20
21static inline unsigned long __pte_to_rste(pte_t pte)
22{
23 unsigned long rste;
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45 if (pte_present(pte)) {
46 rste = pte_val(pte) & PAGE_MASK;
47 rste |= move_set_bit(pte_val(pte), _PAGE_READ,
48 _SEGMENT_ENTRY_READ);
49 rste |= move_set_bit(pte_val(pte), _PAGE_WRITE,
50 _SEGMENT_ENTRY_WRITE);
51 rste |= move_set_bit(pte_val(pte), _PAGE_INVALID,
52 _SEGMENT_ENTRY_INVALID);
53 rste |= move_set_bit(pte_val(pte), _PAGE_PROTECT,
54 _SEGMENT_ENTRY_PROTECT);
55 rste |= move_set_bit(pte_val(pte), _PAGE_DIRTY,
56 _SEGMENT_ENTRY_DIRTY);
57 rste |= move_set_bit(pte_val(pte), _PAGE_YOUNG,
58 _SEGMENT_ENTRY_YOUNG);
59#ifdef CONFIG_MEM_SOFT_DIRTY
60 rste |= move_set_bit(pte_val(pte), _PAGE_SOFT_DIRTY,
61 _SEGMENT_ENTRY_SOFT_DIRTY);
62#endif
63 rste |= move_set_bit(pte_val(pte), _PAGE_NOEXEC,
64 _SEGMENT_ENTRY_NOEXEC);
65 } else
66 rste = _SEGMENT_ENTRY_EMPTY;
67 return rste;
68}
69
70static inline pte_t __rste_to_pte(unsigned long rste)
71{
72 int present;
73 pte_t pte;
74
75 if ((rste & _REGION_ENTRY_TYPE_MASK) == _REGION_ENTRY_TYPE_R3)
76 present = pud_present(__pud(rste));
77 else
78 present = pmd_present(__pmd(rste));
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100 if (present) {
101 pte_val(pte) = rste & _SEGMENT_ENTRY_ORIGIN_LARGE;
102 pte_val(pte) |= _PAGE_LARGE | _PAGE_PRESENT;
103 pte_val(pte) |= move_set_bit(rste, _SEGMENT_ENTRY_READ,
104 _PAGE_READ);
105 pte_val(pte) |= move_set_bit(rste, _SEGMENT_ENTRY_WRITE,
106 _PAGE_WRITE);
107 pte_val(pte) |= move_set_bit(rste, _SEGMENT_ENTRY_INVALID,
108 _PAGE_INVALID);
109 pte_val(pte) |= move_set_bit(rste, _SEGMENT_ENTRY_PROTECT,
110 _PAGE_PROTECT);
111 pte_val(pte) |= move_set_bit(rste, _SEGMENT_ENTRY_DIRTY,
112 _PAGE_DIRTY);
113 pte_val(pte) |= move_set_bit(rste, _SEGMENT_ENTRY_YOUNG,
114 _PAGE_YOUNG);
115#ifdef CONFIG_MEM_SOFT_DIRTY
116 pte_val(pte) |= move_set_bit(rste, _SEGMENT_ENTRY_SOFT_DIRTY,
117 _PAGE_DIRTY);
118#endif
119 pte_val(pte) |= move_set_bit(rste, _SEGMENT_ENTRY_NOEXEC,
120 _PAGE_NOEXEC);
121 } else
122 pte_val(pte) = _PAGE_INVALID;
123 return pte;
124}
125
126static void clear_huge_pte_skeys(struct mm_struct *mm, unsigned long rste)
127{
128 struct page *page;
129 unsigned long size, paddr;
130
131 if (!mm_uses_skeys(mm) ||
132 rste & _SEGMENT_ENTRY_INVALID)
133 return;
134
135 if ((rste & _REGION_ENTRY_TYPE_MASK) == _REGION_ENTRY_TYPE_R3) {
136 page = pud_page(__pud(rste));
137 size = PUD_SIZE;
138 paddr = rste & PUD_MASK;
139 } else {
140 page = pmd_page(__pmd(rste));
141 size = PMD_SIZE;
142 paddr = rste & PMD_MASK;
143 }
144
145 if (!test_and_set_bit(PG_arch_1, &page->flags))
146 __storage_key_init_range(paddr, paddr + size - 1);
147}
148
149void set_huge_pte_at(struct mm_struct *mm, unsigned long addr,
150 pte_t *ptep, pte_t pte)
151{
152 unsigned long rste;
153
154 rste = __pte_to_rste(pte);
155 if (!MACHINE_HAS_NX)
156 rste &= ~_SEGMENT_ENTRY_NOEXEC;
157
158
159 if ((pte_val(*ptep) & _REGION_ENTRY_TYPE_MASK) == _REGION_ENTRY_TYPE_R3)
160 rste |= _REGION_ENTRY_TYPE_R3 | _REGION3_ENTRY_LARGE;
161 else
162 rste |= _SEGMENT_ENTRY_LARGE;
163 clear_huge_pte_skeys(mm, rste);
164 pte_val(*ptep) = rste;
165}
166
167pte_t huge_ptep_get(pte_t *ptep)
168{
169 return __rste_to_pte(pte_val(*ptep));
170}
171
172pte_t huge_ptep_get_and_clear(struct mm_struct *mm,
173 unsigned long addr, pte_t *ptep)
174{
175 pte_t pte = huge_ptep_get(ptep);
176 pmd_t *pmdp = (pmd_t *) ptep;
177 pud_t *pudp = (pud_t *) ptep;
178
179 if ((pte_val(*ptep) & _REGION_ENTRY_TYPE_MASK) == _REGION_ENTRY_TYPE_R3)
180 pudp_xchg_direct(mm, addr, pudp, __pud(_REGION3_ENTRY_EMPTY));
181 else
182 pmdp_xchg_direct(mm, addr, pmdp, __pmd(_SEGMENT_ENTRY_EMPTY));
183 return pte;
184}
185
186pte_t *huge_pte_alloc(struct mm_struct *mm,
187 unsigned long addr, unsigned long sz)
188{
189 pgd_t *pgdp;
190 p4d_t *p4dp;
191 pud_t *pudp;
192 pmd_t *pmdp = NULL;
193
194 pgdp = pgd_offset(mm, addr);
195 p4dp = p4d_alloc(mm, pgdp, addr);
196 if (p4dp) {
197 pudp = pud_alloc(mm, p4dp, addr);
198 if (pudp) {
199 if (sz == PUD_SIZE)
200 return (pte_t *) pudp;
201 else if (sz == PMD_SIZE)
202 pmdp = pmd_alloc(mm, pudp, addr);
203 }
204 }
205 return (pte_t *) pmdp;
206}
207
208pte_t *huge_pte_offset(struct mm_struct *mm,
209 unsigned long addr, unsigned long sz)
210{
211 pgd_t *pgdp;
212 p4d_t *p4dp;
213 pud_t *pudp;
214 pmd_t *pmdp = NULL;
215
216 pgdp = pgd_offset(mm, addr);
217 if (pgd_present(*pgdp)) {
218 p4dp = p4d_offset(pgdp, addr);
219 if (p4d_present(*p4dp)) {
220 pudp = pud_offset(p4dp, addr);
221 if (pud_present(*pudp)) {
222 if (pud_large(*pudp))
223 return (pte_t *) pudp;
224 pmdp = pmd_offset(pudp, addr);
225 }
226 }
227 }
228 return (pte_t *) pmdp;
229}
230
231int pmd_huge(pmd_t pmd)
232{
233 return pmd_large(pmd);
234}
235
236int pud_huge(pud_t pud)
237{
238 return pud_large(pud);
239}
240
241struct page *
242follow_huge_pud(struct mm_struct *mm, unsigned long address,
243 pud_t *pud, int flags)
244{
245 if (flags & FOLL_GET)
246 return NULL;
247
248 return pud_page(*pud) + ((address & ~PUD_MASK) >> PAGE_SHIFT);
249}
250
251static __init int setup_hugepagesz(char *opt)
252{
253 unsigned long size;
254 char *string = opt;
255
256 size = memparse(opt, &opt);
257 if (MACHINE_HAS_EDAT1 && size == PMD_SIZE) {
258 hugetlb_add_hstate(PMD_SHIFT - PAGE_SHIFT);
259 } else if (MACHINE_HAS_EDAT2 && size == PUD_SIZE) {
260 hugetlb_add_hstate(PUD_SHIFT - PAGE_SHIFT);
261 } else {
262 hugetlb_bad_size();
263 pr_err("hugepagesz= specifies an unsupported page size %s\n",
264 string);
265 return 0;
266 }
267 return 1;
268}
269__setup("hugepagesz=", setup_hugepagesz);
270
