Merge branch 'next-general' of git://git.kernel.org/pub/scm/linux/kernel/git/jmorris...
[sfrench/cifs-2.6.git] / arch / x86 / include / asm / pgtable.h
1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef _ASM_X86_PGTABLE_H
3 #define _ASM_X86_PGTABLE_H
4
5 #include <linux/mem_encrypt.h>
6 #include <asm/page.h>
7 #include <asm/pgtable_types.h>
8
9 /*
10  * Macro to mark a page protection value as UC-
11  */
12 #define pgprot_noncached(prot)                                          \
13         ((boot_cpu_data.x86 > 3)                                        \
14          ? (__pgprot(pgprot_val(prot) |                                 \
15                      cachemode2protval(_PAGE_CACHE_MODE_UC_MINUS)))     \
16          : (prot))
17
18 /*
19  * Macros to add or remove encryption attribute
20  */
21 #define pgprot_encrypted(prot)  __pgprot(__sme_set(pgprot_val(prot)))
22 #define pgprot_decrypted(prot)  __pgprot(__sme_clr(pgprot_val(prot)))
23
24 #ifndef __ASSEMBLY__
25 #include <asm/x86_init.h>
26
27 extern pgd_t early_top_pgt[PTRS_PER_PGD];
28 int __init __early_make_pgtable(unsigned long address, pmdval_t pmd);
29
30 void ptdump_walk_pgd_level(struct seq_file *m, pgd_t *pgd);
31 void ptdump_walk_pgd_level_debugfs(struct seq_file *m, pgd_t *pgd, bool user);
32 void ptdump_walk_pgd_level_checkwx(void);
33 void ptdump_walk_user_pgd_level_checkwx(void);
34
35 #ifdef CONFIG_DEBUG_WX
36 #define debug_checkwx()         ptdump_walk_pgd_level_checkwx()
37 #define debug_checkwx_user()    ptdump_walk_user_pgd_level_checkwx()
38 #else
39 #define debug_checkwx()         do { } while (0)
40 #define debug_checkwx_user()    do { } while (0)
41 #endif
42
43 /*
44  * ZERO_PAGE is a global shared page that is always zero: used
45  * for zero-mapped memory areas etc..
46  */
47 extern unsigned long empty_zero_page[PAGE_SIZE / sizeof(unsigned long)]
48         __visible;
49 #define ZERO_PAGE(vaddr) ((void)(vaddr),virt_to_page(empty_zero_page))
50
51 extern spinlock_t pgd_lock;
52 extern struct list_head pgd_list;
53
54 extern struct mm_struct *pgd_page_get_mm(struct page *page);
55
56 extern pmdval_t early_pmd_flags;
57
58 #ifdef CONFIG_PARAVIRT_XXL
59 #include <asm/paravirt.h>
60 #else  /* !CONFIG_PARAVIRT_XXL */
61 #define set_pte(ptep, pte)              native_set_pte(ptep, pte)
62 #define set_pte_at(mm, addr, ptep, pte) native_set_pte_at(mm, addr, ptep, pte)
63
64 #define set_pte_atomic(ptep, pte)                                       \
65         native_set_pte_atomic(ptep, pte)
66
67 #define set_pmd(pmdp, pmd)              native_set_pmd(pmdp, pmd)
68
69 #ifndef __PAGETABLE_P4D_FOLDED
70 #define set_pgd(pgdp, pgd)              native_set_pgd(pgdp, pgd)
71 #define pgd_clear(pgd)                  (pgtable_l5_enabled() ? native_pgd_clear(pgd) : 0)
72 #endif
73
74 #ifndef set_p4d
75 # define set_p4d(p4dp, p4d)             native_set_p4d(p4dp, p4d)
76 #endif
77
78 #ifndef __PAGETABLE_PUD_FOLDED
79 #define p4d_clear(p4d)                  native_p4d_clear(p4d)
80 #endif
81
82 #ifndef set_pud
83 # define set_pud(pudp, pud)             native_set_pud(pudp, pud)
84 #endif
85
86 #ifndef __PAGETABLE_PUD_FOLDED
87 #define pud_clear(pud)                  native_pud_clear(pud)
88 #endif
89
90 #define pte_clear(mm, addr, ptep)       native_pte_clear(mm, addr, ptep)
91 #define pmd_clear(pmd)                  native_pmd_clear(pmd)
92
93 #define pgd_val(x)      native_pgd_val(x)
94 #define __pgd(x)        native_make_pgd(x)
95
96 #ifndef __PAGETABLE_P4D_FOLDED
97 #define p4d_val(x)      native_p4d_val(x)
98 #define __p4d(x)        native_make_p4d(x)
99 #endif
100
101 #ifndef __PAGETABLE_PUD_FOLDED
102 #define pud_val(x)      native_pud_val(x)
103 #define __pud(x)        native_make_pud(x)
104 #endif
105
106 #ifndef __PAGETABLE_PMD_FOLDED
107 #define pmd_val(x)      native_pmd_val(x)
108 #define __pmd(x)        native_make_pmd(x)
109 #endif
110
111 #define pte_val(x)      native_pte_val(x)
112 #define __pte(x)        native_make_pte(x)
113
114 #define arch_end_context_switch(prev)   do {} while(0)
115 #endif  /* CONFIG_PARAVIRT_XXL */
116
117 /*
118  * The following only work if pte_present() is true.
119  * Undefined behaviour if not..
120  */
121 static inline int pte_dirty(pte_t pte)
122 {
123         return pte_flags(pte) & _PAGE_DIRTY;
124 }
125
126
127 static inline u32 read_pkru(void)
128 {
129         if (boot_cpu_has(X86_FEATURE_OSPKE))
130                 return __read_pkru();
131         return 0;
132 }
133
134 static inline void write_pkru(u32 pkru)
135 {
136         if (boot_cpu_has(X86_FEATURE_OSPKE))
137                 __write_pkru(pkru);
138 }
139
140 static inline int pte_young(pte_t pte)
141 {
142         return pte_flags(pte) & _PAGE_ACCESSED;
143 }
144
145 static inline int pmd_dirty(pmd_t pmd)
146 {
147         return pmd_flags(pmd) & _PAGE_DIRTY;
148 }
149
150 static inline int pmd_young(pmd_t pmd)
151 {
152         return pmd_flags(pmd) & _PAGE_ACCESSED;
153 }
154
155 static inline int pud_dirty(pud_t pud)
156 {
157         return pud_flags(pud) & _PAGE_DIRTY;
158 }
159
160 static inline int pud_young(pud_t pud)
161 {
162         return pud_flags(pud) & _PAGE_ACCESSED;
163 }
164
165 static inline int pte_write(pte_t pte)
166 {
167         return pte_flags(pte) & _PAGE_RW;
168 }
169
170 static inline int pte_huge(pte_t pte)
171 {
172         return pte_flags(pte) & _PAGE_PSE;
173 }
174
175 static inline int pte_global(pte_t pte)
176 {
177         return pte_flags(pte) & _PAGE_GLOBAL;
178 }
179
180 static inline int pte_exec(pte_t pte)
181 {
182         return !(pte_flags(pte) & _PAGE_NX);
183 }
184
185 static inline int pte_special(pte_t pte)
186 {
187         return pte_flags(pte) & _PAGE_SPECIAL;
188 }
189
190 /* Entries that were set to PROT_NONE are inverted */
191
192 static inline u64 protnone_mask(u64 val);
193
194 static inline unsigned long pte_pfn(pte_t pte)
195 {
196         phys_addr_t pfn = pte_val(pte);
197         pfn ^= protnone_mask(pfn);
198         return (pfn & PTE_PFN_MASK) >> PAGE_SHIFT;
199 }
200
201 static inline unsigned long pmd_pfn(pmd_t pmd)
202 {
203         phys_addr_t pfn = pmd_val(pmd);
204         pfn ^= protnone_mask(pfn);
205         return (pfn & pmd_pfn_mask(pmd)) >> PAGE_SHIFT;
206 }
207
208 static inline unsigned long pud_pfn(pud_t pud)
209 {
210         phys_addr_t pfn = pud_val(pud);
211         pfn ^= protnone_mask(pfn);
212         return (pfn & pud_pfn_mask(pud)) >> PAGE_SHIFT;
213 }
214
215 static inline unsigned long p4d_pfn(p4d_t p4d)
216 {
217         return (p4d_val(p4d) & p4d_pfn_mask(p4d)) >> PAGE_SHIFT;
218 }
219
220 static inline unsigned long pgd_pfn(pgd_t pgd)
221 {
222         return (pgd_val(pgd) & PTE_PFN_MASK) >> PAGE_SHIFT;
223 }
224
225 static inline int p4d_large(p4d_t p4d)
226 {
227         /* No 512 GiB pages yet */
228         return 0;
229 }
230
231 #define pte_page(pte)   pfn_to_page(pte_pfn(pte))
232
233 static inline int pmd_large(pmd_t pte)
234 {
235         return pmd_flags(pte) & _PAGE_PSE;
236 }
237
238 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
239 static inline int pmd_trans_huge(pmd_t pmd)
240 {
241         return (pmd_val(pmd) & (_PAGE_PSE|_PAGE_DEVMAP)) == _PAGE_PSE;
242 }
243
244 #ifdef CONFIG_HAVE_ARCH_TRANSPARENT_HUGEPAGE_PUD
245 static inline int pud_trans_huge(pud_t pud)
246 {
247         return (pud_val(pud) & (_PAGE_PSE|_PAGE_DEVMAP)) == _PAGE_PSE;
248 }
249 #endif
250
251 #define has_transparent_hugepage has_transparent_hugepage
252 static inline int has_transparent_hugepage(void)
253 {
254         return boot_cpu_has(X86_FEATURE_PSE);
255 }
256
257 #ifdef __HAVE_ARCH_PTE_DEVMAP
258 static inline int pmd_devmap(pmd_t pmd)
259 {
260         return !!(pmd_val(pmd) & _PAGE_DEVMAP);
261 }
262
263 #ifdef CONFIG_HAVE_ARCH_TRANSPARENT_HUGEPAGE_PUD
264 static inline int pud_devmap(pud_t pud)
265 {
266         return !!(pud_val(pud) & _PAGE_DEVMAP);
267 }
268 #else
269 static inline int pud_devmap(pud_t pud)
270 {
271         return 0;
272 }
273 #endif
274
275 static inline int pgd_devmap(pgd_t pgd)
276 {
277         return 0;
278 }
279 #endif
280 #endif /* CONFIG_TRANSPARENT_HUGEPAGE */
281
282 static inline pte_t pte_set_flags(pte_t pte, pteval_t set)
283 {
284         pteval_t v = native_pte_val(pte);
285
286         return native_make_pte(v | set);
287 }
288
289 static inline pte_t pte_clear_flags(pte_t pte, pteval_t clear)
290 {
291         pteval_t v = native_pte_val(pte);
292
293         return native_make_pte(v & ~clear);
294 }
295
296 static inline pte_t pte_mkclean(pte_t pte)
297 {
298         return pte_clear_flags(pte, _PAGE_DIRTY);
299 }
300
301 static inline pte_t pte_mkold(pte_t pte)
302 {
303         return pte_clear_flags(pte, _PAGE_ACCESSED);
304 }
305
306 static inline pte_t pte_wrprotect(pte_t pte)
307 {
308         return pte_clear_flags(pte, _PAGE_RW);
309 }
310
311 static inline pte_t pte_mkexec(pte_t pte)
312 {
313         return pte_clear_flags(pte, _PAGE_NX);
314 }
315
316 static inline pte_t pte_mkdirty(pte_t pte)
317 {
318         return pte_set_flags(pte, _PAGE_DIRTY | _PAGE_SOFT_DIRTY);
319 }
320
321 static inline pte_t pte_mkyoung(pte_t pte)
322 {
323         return pte_set_flags(pte, _PAGE_ACCESSED);
324 }
325
326 static inline pte_t pte_mkwrite(pte_t pte)
327 {
328         return pte_set_flags(pte, _PAGE_RW);
329 }
330
331 static inline pte_t pte_mkhuge(pte_t pte)
332 {
333         return pte_set_flags(pte, _PAGE_PSE);
334 }
335
336 static inline pte_t pte_clrhuge(pte_t pte)
337 {
338         return pte_clear_flags(pte, _PAGE_PSE);
339 }
340
341 static inline pte_t pte_mkglobal(pte_t pte)
342 {
343         return pte_set_flags(pte, _PAGE_GLOBAL);
344 }
345
346 static inline pte_t pte_clrglobal(pte_t pte)
347 {
348         return pte_clear_flags(pte, _PAGE_GLOBAL);
349 }
350
351 static inline pte_t pte_mkspecial(pte_t pte)
352 {
353         return pte_set_flags(pte, _PAGE_SPECIAL);
354 }
355
356 static inline pte_t pte_mkdevmap(pte_t pte)
357 {
358         return pte_set_flags(pte, _PAGE_SPECIAL|_PAGE_DEVMAP);
359 }
360
361 static inline pmd_t pmd_set_flags(pmd_t pmd, pmdval_t set)
362 {
363         pmdval_t v = native_pmd_val(pmd);
364
365         return native_make_pmd(v | set);
366 }
367
368 static inline pmd_t pmd_clear_flags(pmd_t pmd, pmdval_t clear)
369 {
370         pmdval_t v = native_pmd_val(pmd);
371
372         return native_make_pmd(v & ~clear);
373 }
374
375 static inline pmd_t pmd_mkold(pmd_t pmd)
376 {
377         return pmd_clear_flags(pmd, _PAGE_ACCESSED);
378 }
379
380 static inline pmd_t pmd_mkclean(pmd_t pmd)
381 {
382         return pmd_clear_flags(pmd, _PAGE_DIRTY);
383 }
384
385 static inline pmd_t pmd_wrprotect(pmd_t pmd)
386 {
387         return pmd_clear_flags(pmd, _PAGE_RW);
388 }
389
390 static inline pmd_t pmd_mkdirty(pmd_t pmd)
391 {
392         return pmd_set_flags(pmd, _PAGE_DIRTY | _PAGE_SOFT_DIRTY);
393 }
394
395 static inline pmd_t pmd_mkdevmap(pmd_t pmd)
396 {
397         return pmd_set_flags(pmd, _PAGE_DEVMAP);
398 }
399
400 static inline pmd_t pmd_mkhuge(pmd_t pmd)
401 {
402         return pmd_set_flags(pmd, _PAGE_PSE);
403 }
404
405 static inline pmd_t pmd_mkyoung(pmd_t pmd)
406 {
407         return pmd_set_flags(pmd, _PAGE_ACCESSED);
408 }
409
410 static inline pmd_t pmd_mkwrite(pmd_t pmd)
411 {
412         return pmd_set_flags(pmd, _PAGE_RW);
413 }
414
415 static inline pud_t pud_set_flags(pud_t pud, pudval_t set)
416 {
417         pudval_t v = native_pud_val(pud);
418
419         return native_make_pud(v | set);
420 }
421
422 static inline pud_t pud_clear_flags(pud_t pud, pudval_t clear)
423 {
424         pudval_t v = native_pud_val(pud);
425
426         return native_make_pud(v & ~clear);
427 }
428
429 static inline pud_t pud_mkold(pud_t pud)
430 {
431         return pud_clear_flags(pud, _PAGE_ACCESSED);
432 }
433
434 static inline pud_t pud_mkclean(pud_t pud)
435 {
436         return pud_clear_flags(pud, _PAGE_DIRTY);
437 }
438
439 static inline pud_t pud_wrprotect(pud_t pud)
440 {
441         return pud_clear_flags(pud, _PAGE_RW);
442 }
443
444 static inline pud_t pud_mkdirty(pud_t pud)
445 {
446         return pud_set_flags(pud, _PAGE_DIRTY | _PAGE_SOFT_DIRTY);
447 }
448
449 static inline pud_t pud_mkdevmap(pud_t pud)
450 {
451         return pud_set_flags(pud, _PAGE_DEVMAP);
452 }
453
454 static inline pud_t pud_mkhuge(pud_t pud)
455 {
456         return pud_set_flags(pud, _PAGE_PSE);
457 }
458
459 static inline pud_t pud_mkyoung(pud_t pud)
460 {
461         return pud_set_flags(pud, _PAGE_ACCESSED);
462 }
463
464 static inline pud_t pud_mkwrite(pud_t pud)
465 {
466         return pud_set_flags(pud, _PAGE_RW);
467 }
468
469 #ifdef CONFIG_HAVE_ARCH_SOFT_DIRTY
470 static inline int pte_soft_dirty(pte_t pte)
471 {
472         return pte_flags(pte) & _PAGE_SOFT_DIRTY;
473 }
474
475 static inline int pmd_soft_dirty(pmd_t pmd)
476 {
477         return pmd_flags(pmd) & _PAGE_SOFT_DIRTY;
478 }
479
480 static inline int pud_soft_dirty(pud_t pud)
481 {
482         return pud_flags(pud) & _PAGE_SOFT_DIRTY;
483 }
484
485 static inline pte_t pte_mksoft_dirty(pte_t pte)
486 {
487         return pte_set_flags(pte, _PAGE_SOFT_DIRTY);
488 }
489
490 static inline pmd_t pmd_mksoft_dirty(pmd_t pmd)
491 {
492         return pmd_set_flags(pmd, _PAGE_SOFT_DIRTY);
493 }
494
495 static inline pud_t pud_mksoft_dirty(pud_t pud)
496 {
497         return pud_set_flags(pud, _PAGE_SOFT_DIRTY);
498 }
499
500 static inline pte_t pte_clear_soft_dirty(pte_t pte)
501 {
502         return pte_clear_flags(pte, _PAGE_SOFT_DIRTY);
503 }
504
505 static inline pmd_t pmd_clear_soft_dirty(pmd_t pmd)
506 {
507         return pmd_clear_flags(pmd, _PAGE_SOFT_DIRTY);
508 }
509
510 static inline pud_t pud_clear_soft_dirty(pud_t pud)
511 {
512         return pud_clear_flags(pud, _PAGE_SOFT_DIRTY);
513 }
514
515 #endif /* CONFIG_HAVE_ARCH_SOFT_DIRTY */
516
517 /*
518  * Mask out unsupported bits in a present pgprot.  Non-present pgprots
519  * can use those bits for other purposes, so leave them be.
520  */
521 static inline pgprotval_t massage_pgprot(pgprot_t pgprot)
522 {
523         pgprotval_t protval = pgprot_val(pgprot);
524
525         if (protval & _PAGE_PRESENT)
526                 protval &= __supported_pte_mask;
527
528         return protval;
529 }
530
531 static inline pgprotval_t check_pgprot(pgprot_t pgprot)
532 {
533         pgprotval_t massaged_val = massage_pgprot(pgprot);
534
535         /* mmdebug.h can not be included here because of dependencies */
536 #ifdef CONFIG_DEBUG_VM
537         WARN_ONCE(pgprot_val(pgprot) != massaged_val,
538                   "attempted to set unsupported pgprot: %016llx "
539                   "bits: %016llx supported: %016llx\n",
540                   (u64)pgprot_val(pgprot),
541                   (u64)pgprot_val(pgprot) ^ massaged_val,
542                   (u64)__supported_pte_mask);
543 #endif
544
545         return massaged_val;
546 }
547
548 static inline pte_t pfn_pte(unsigned long page_nr, pgprot_t pgprot)
549 {
550         phys_addr_t pfn = (phys_addr_t)page_nr << PAGE_SHIFT;
551         pfn ^= protnone_mask(pgprot_val(pgprot));
552         pfn &= PTE_PFN_MASK;
553         return __pte(pfn | check_pgprot(pgprot));
554 }
555
556 static inline pmd_t pfn_pmd(unsigned long page_nr, pgprot_t pgprot)
557 {
558         phys_addr_t pfn = (phys_addr_t)page_nr << PAGE_SHIFT;
559         pfn ^= protnone_mask(pgprot_val(pgprot));
560         pfn &= PHYSICAL_PMD_PAGE_MASK;
561         return __pmd(pfn | check_pgprot(pgprot));
562 }
563
564 static inline pud_t pfn_pud(unsigned long page_nr, pgprot_t pgprot)
565 {
566         phys_addr_t pfn = (phys_addr_t)page_nr << PAGE_SHIFT;
567         pfn ^= protnone_mask(pgprot_val(pgprot));
568         pfn &= PHYSICAL_PUD_PAGE_MASK;
569         return __pud(pfn | check_pgprot(pgprot));
570 }
571
572 static inline pmd_t pmd_mknotpresent(pmd_t pmd)
573 {
574         return pfn_pmd(pmd_pfn(pmd),
575                       __pgprot(pmd_flags(pmd) & ~(_PAGE_PRESENT|_PAGE_PROTNONE)));
576 }
577
578 static inline pud_t pud_mknotpresent(pud_t pud)
579 {
580         return pfn_pud(pud_pfn(pud),
581               __pgprot(pud_flags(pud) & ~(_PAGE_PRESENT|_PAGE_PROTNONE)));
582 }
583
584 static inline u64 flip_protnone_guard(u64 oldval, u64 val, u64 mask);
585
586 static inline pte_t pte_modify(pte_t pte, pgprot_t newprot)
587 {
588         pteval_t val = pte_val(pte), oldval = val;
589
590         /*
591          * Chop off the NX bit (if present), and add the NX portion of
592          * the newprot (if present):
593          */
594         val &= _PAGE_CHG_MASK;
595         val |= check_pgprot(newprot) & ~_PAGE_CHG_MASK;
596         val = flip_protnone_guard(oldval, val, PTE_PFN_MASK);
597         return __pte(val);
598 }
599
600 static inline pmd_t pmd_modify(pmd_t pmd, pgprot_t newprot)
601 {
602         pmdval_t val = pmd_val(pmd), oldval = val;
603
604         val &= _HPAGE_CHG_MASK;
605         val |= check_pgprot(newprot) & ~_HPAGE_CHG_MASK;
606         val = flip_protnone_guard(oldval, val, PHYSICAL_PMD_PAGE_MASK);
607         return __pmd(val);
608 }
609
610 /* mprotect needs to preserve PAT bits when updating vm_page_prot */
611 #define pgprot_modify pgprot_modify
612 static inline pgprot_t pgprot_modify(pgprot_t oldprot, pgprot_t newprot)
613 {
614         pgprotval_t preservebits = pgprot_val(oldprot) & _PAGE_CHG_MASK;
615         pgprotval_t addbits = pgprot_val(newprot);
616         return __pgprot(preservebits | addbits);
617 }
618
619 #define pte_pgprot(x) __pgprot(pte_flags(x))
620 #define pmd_pgprot(x) __pgprot(pmd_flags(x))
621 #define pud_pgprot(x) __pgprot(pud_flags(x))
622 #define p4d_pgprot(x) __pgprot(p4d_flags(x))
623
624 #define canon_pgprot(p) __pgprot(massage_pgprot(p))
625
626 static inline pgprot_t arch_filter_pgprot(pgprot_t prot)
627 {
628         return canon_pgprot(prot);
629 }
630
631 static inline int is_new_memtype_allowed(u64 paddr, unsigned long size,
632                                          enum page_cache_mode pcm,
633                                          enum page_cache_mode new_pcm)
634 {
635         /*
636          * PAT type is always WB for untracked ranges, so no need to check.
637          */
638         if (x86_platform.is_untracked_pat_range(paddr, paddr + size))
639                 return 1;
640
641         /*
642          * Certain new memtypes are not allowed with certain
643          * requested memtype:
644          * - request is uncached, return cannot be write-back
645          * - request is write-combine, return cannot be write-back
646          * - request is write-through, return cannot be write-back
647          * - request is write-through, return cannot be write-combine
648          */
649         if ((pcm == _PAGE_CACHE_MODE_UC_MINUS &&
650              new_pcm == _PAGE_CACHE_MODE_WB) ||
651             (pcm == _PAGE_CACHE_MODE_WC &&
652              new_pcm == _PAGE_CACHE_MODE_WB) ||
653             (pcm == _PAGE_CACHE_MODE_WT &&
654              new_pcm == _PAGE_CACHE_MODE_WB) ||
655             (pcm == _PAGE_CACHE_MODE_WT &&
656              new_pcm == _PAGE_CACHE_MODE_WC)) {
657                 return 0;
658         }
659
660         return 1;
661 }
662
663 pmd_t *populate_extra_pmd(unsigned long vaddr);
664 pte_t *populate_extra_pte(unsigned long vaddr);
665
666 #ifdef CONFIG_PAGE_TABLE_ISOLATION
667 pgd_t __pti_set_user_pgtbl(pgd_t *pgdp, pgd_t pgd);
668
669 /*
670  * Take a PGD location (pgdp) and a pgd value that needs to be set there.
671  * Populates the user and returns the resulting PGD that must be set in
672  * the kernel copy of the page tables.
673  */
674 static inline pgd_t pti_set_user_pgtbl(pgd_t *pgdp, pgd_t pgd)
675 {
676         if (!static_cpu_has(X86_FEATURE_PTI))
677                 return pgd;
678         return __pti_set_user_pgtbl(pgdp, pgd);
679 }
680 #else   /* CONFIG_PAGE_TABLE_ISOLATION */
681 static inline pgd_t pti_set_user_pgtbl(pgd_t *pgdp, pgd_t pgd)
682 {
683         return pgd;
684 }
685 #endif  /* CONFIG_PAGE_TABLE_ISOLATION */
686
687 #endif  /* __ASSEMBLY__ */
688
689
690 #ifdef CONFIG_X86_32
691 # include <asm/pgtable_32.h>
692 #else
693 # include <asm/pgtable_64.h>
694 #endif
695
696 #ifndef __ASSEMBLY__
697 #include <linux/mm_types.h>
698 #include <linux/mmdebug.h>
699 #include <linux/log2.h>
700 #include <asm/fixmap.h>
701
702 static inline int pte_none(pte_t pte)
703 {
704         return !(pte.pte & ~(_PAGE_KNL_ERRATUM_MASK));
705 }
706
707 #define __HAVE_ARCH_PTE_SAME
708 static inline int pte_same(pte_t a, pte_t b)
709 {
710         return a.pte == b.pte;
711 }
712
713 static inline int pte_present(pte_t a)
714 {
715         return pte_flags(a) & (_PAGE_PRESENT | _PAGE_PROTNONE);
716 }
717
718 #ifdef __HAVE_ARCH_PTE_DEVMAP
719 static inline int pte_devmap(pte_t a)
720 {
721         return (pte_flags(a) & _PAGE_DEVMAP) == _PAGE_DEVMAP;
722 }
723 #endif
724
725 #define pte_accessible pte_accessible
726 static inline bool pte_accessible(struct mm_struct *mm, pte_t a)
727 {
728         if (pte_flags(a) & _PAGE_PRESENT)
729                 return true;
730
731         if ((pte_flags(a) & _PAGE_PROTNONE) &&
732                         mm_tlb_flush_pending(mm))
733                 return true;
734
735         return false;
736 }
737
738 static inline int pmd_present(pmd_t pmd)
739 {
740         /*
741          * Checking for _PAGE_PSE is needed too because
742          * split_huge_page will temporarily clear the present bit (but
743          * the _PAGE_PSE flag will remain set at all times while the
744          * _PAGE_PRESENT bit is clear).
745          */
746         return pmd_flags(pmd) & (_PAGE_PRESENT | _PAGE_PROTNONE | _PAGE_PSE);
747 }
748
749 #ifdef CONFIG_NUMA_BALANCING
750 /*
751  * These work without NUMA balancing but the kernel does not care. See the
752  * comment in include/asm-generic/pgtable.h
753  */
754 static inline int pte_protnone(pte_t pte)
755 {
756         return (pte_flags(pte) & (_PAGE_PROTNONE | _PAGE_PRESENT))
757                 == _PAGE_PROTNONE;
758 }
759
760 static inline int pmd_protnone(pmd_t pmd)
761 {
762         return (pmd_flags(pmd) & (_PAGE_PROTNONE | _PAGE_PRESENT))
763                 == _PAGE_PROTNONE;
764 }
765 #endif /* CONFIG_NUMA_BALANCING */
766
767 static inline int pmd_none(pmd_t pmd)
768 {
769         /* Only check low word on 32-bit platforms, since it might be
770            out of sync with upper half. */
771         unsigned long val = native_pmd_val(pmd);
772         return (val & ~_PAGE_KNL_ERRATUM_MASK) == 0;
773 }
774
775 static inline unsigned long pmd_page_vaddr(pmd_t pmd)
776 {
777         return (unsigned long)__va(pmd_val(pmd) & pmd_pfn_mask(pmd));
778 }
779
780 /*
781  * Currently stuck as a macro due to indirect forward reference to
782  * linux/mmzone.h's __section_mem_map_addr() definition:
783  */
784 #define pmd_page(pmd)   pfn_to_page(pmd_pfn(pmd))
785
786 /*
787  * the pmd page can be thought of an array like this: pmd_t[PTRS_PER_PMD]
788  *
789  * this macro returns the index of the entry in the pmd page which would
790  * control the given virtual address
791  */
792 static inline unsigned long pmd_index(unsigned long address)
793 {
794         return (address >> PMD_SHIFT) & (PTRS_PER_PMD - 1);
795 }
796
797 /*
798  * Conversion functions: convert a page and protection to a page entry,
799  * and a page entry and page directory to the page they refer to.
800  *
801  * (Currently stuck as a macro because of indirect forward reference
802  * to linux/mm.h:page_to_nid())
803  */
804 #define mk_pte(page, pgprot)   pfn_pte(page_to_pfn(page), (pgprot))
805
806 /*
807  * the pte page can be thought of an array like this: pte_t[PTRS_PER_PTE]
808  *
809  * this function returns the index of the entry in the pte page which would
810  * control the given virtual address
811  */
812 static inline unsigned long pte_index(unsigned long address)
813 {
814         return (address >> PAGE_SHIFT) & (PTRS_PER_PTE - 1);
815 }
816
817 static inline pte_t *pte_offset_kernel(pmd_t *pmd, unsigned long address)
818 {
819         return (pte_t *)pmd_page_vaddr(*pmd) + pte_index(address);
820 }
821
822 static inline int pmd_bad(pmd_t pmd)
823 {
824         return (pmd_flags(pmd) & ~_PAGE_USER) != _KERNPG_TABLE;
825 }
826
827 static inline unsigned long pages_to_mb(unsigned long npg)
828 {
829         return npg >> (20 - PAGE_SHIFT);
830 }
831
832 #if CONFIG_PGTABLE_LEVELS > 2
833 static inline int pud_none(pud_t pud)
834 {
835         return (native_pud_val(pud) & ~(_PAGE_KNL_ERRATUM_MASK)) == 0;
836 }
837
838 static inline int pud_present(pud_t pud)
839 {
840         return pud_flags(pud) & _PAGE_PRESENT;
841 }
842
843 static inline unsigned long pud_page_vaddr(pud_t pud)
844 {
845         return (unsigned long)__va(pud_val(pud) & pud_pfn_mask(pud));
846 }
847
848 /*
849  * Currently stuck as a macro due to indirect forward reference to
850  * linux/mmzone.h's __section_mem_map_addr() definition:
851  */
852 #define pud_page(pud)   pfn_to_page(pud_pfn(pud))
853
854 /* Find an entry in the second-level page table.. */
855 static inline pmd_t *pmd_offset(pud_t *pud, unsigned long address)
856 {
857         return (pmd_t *)pud_page_vaddr(*pud) + pmd_index(address);
858 }
859
860 static inline int pud_large(pud_t pud)
861 {
862         return (pud_val(pud) & (_PAGE_PSE | _PAGE_PRESENT)) ==
863                 (_PAGE_PSE | _PAGE_PRESENT);
864 }
865
866 static inline int pud_bad(pud_t pud)
867 {
868         return (pud_flags(pud) & ~(_KERNPG_TABLE | _PAGE_USER)) != 0;
869 }
870 #else
871 static inline int pud_large(pud_t pud)
872 {
873         return 0;
874 }
875 #endif  /* CONFIG_PGTABLE_LEVELS > 2 */
876
877 static inline unsigned long pud_index(unsigned long address)
878 {
879         return (address >> PUD_SHIFT) & (PTRS_PER_PUD - 1);
880 }
881
882 #if CONFIG_PGTABLE_LEVELS > 3
883 static inline int p4d_none(p4d_t p4d)
884 {
885         return (native_p4d_val(p4d) & ~(_PAGE_KNL_ERRATUM_MASK)) == 0;
886 }
887
888 static inline int p4d_present(p4d_t p4d)
889 {
890         return p4d_flags(p4d) & _PAGE_PRESENT;
891 }
892
893 static inline unsigned long p4d_page_vaddr(p4d_t p4d)
894 {
895         return (unsigned long)__va(p4d_val(p4d) & p4d_pfn_mask(p4d));
896 }
897
898 /*
899  * Currently stuck as a macro due to indirect forward reference to
900  * linux/mmzone.h's __section_mem_map_addr() definition:
901  */
902 #define p4d_page(p4d)   pfn_to_page(p4d_pfn(p4d))
903
904 /* Find an entry in the third-level page table.. */
905 static inline pud_t *pud_offset(p4d_t *p4d, unsigned long address)
906 {
907         return (pud_t *)p4d_page_vaddr(*p4d) + pud_index(address);
908 }
909
910 static inline int p4d_bad(p4d_t p4d)
911 {
912         unsigned long ignore_flags = _KERNPG_TABLE | _PAGE_USER;
913
914         if (IS_ENABLED(CONFIG_PAGE_TABLE_ISOLATION))
915                 ignore_flags |= _PAGE_NX;
916
917         return (p4d_flags(p4d) & ~ignore_flags) != 0;
918 }
919 #endif  /* CONFIG_PGTABLE_LEVELS > 3 */
920
921 static inline unsigned long p4d_index(unsigned long address)
922 {
923         return (address >> P4D_SHIFT) & (PTRS_PER_P4D - 1);
924 }
925
926 #if CONFIG_PGTABLE_LEVELS > 4
927 static inline int pgd_present(pgd_t pgd)
928 {
929         if (!pgtable_l5_enabled())
930                 return 1;
931         return pgd_flags(pgd) & _PAGE_PRESENT;
932 }
933
934 static inline unsigned long pgd_page_vaddr(pgd_t pgd)
935 {
936         return (unsigned long)__va((unsigned long)pgd_val(pgd) & PTE_PFN_MASK);
937 }
938
939 /*
940  * Currently stuck as a macro due to indirect forward reference to
941  * linux/mmzone.h's __section_mem_map_addr() definition:
942  */
943 #define pgd_page(pgd)   pfn_to_page(pgd_pfn(pgd))
944
945 /* to find an entry in a page-table-directory. */
946 static inline p4d_t *p4d_offset(pgd_t *pgd, unsigned long address)
947 {
948         if (!pgtable_l5_enabled())
949                 return (p4d_t *)pgd;
950         return (p4d_t *)pgd_page_vaddr(*pgd) + p4d_index(address);
951 }
952
953 static inline int pgd_bad(pgd_t pgd)
954 {
955         unsigned long ignore_flags = _PAGE_USER;
956
957         if (!pgtable_l5_enabled())
958                 return 0;
959
960         if (IS_ENABLED(CONFIG_PAGE_TABLE_ISOLATION))
961                 ignore_flags |= _PAGE_NX;
962
963         return (pgd_flags(pgd) & ~ignore_flags) != _KERNPG_TABLE;
964 }
965
966 static inline int pgd_none(pgd_t pgd)
967 {
968         if (!pgtable_l5_enabled())
969                 return 0;
970         /*
971          * There is no need to do a workaround for the KNL stray
972          * A/D bit erratum here.  PGDs only point to page tables
973          * except on 32-bit non-PAE which is not supported on
974          * KNL.
975          */
976         return !native_pgd_val(pgd);
977 }
978 #endif  /* CONFIG_PGTABLE_LEVELS > 4 */
979
980 #endif  /* __ASSEMBLY__ */
981
982 /*
983  * the pgd page can be thought of an array like this: pgd_t[PTRS_PER_PGD]
984  *
985  * this macro returns the index of the entry in the pgd page which would
986  * control the given virtual address
987  */
988 #define pgd_index(address) (((address) >> PGDIR_SHIFT) & (PTRS_PER_PGD - 1))
989
990 /*
991  * pgd_offset() returns a (pgd_t *)
992  * pgd_index() is used get the offset into the pgd page's array of pgd_t's;
993  */
994 #define pgd_offset_pgd(pgd, address) (pgd + pgd_index((address)))
995 /*
996  * a shortcut to get a pgd_t in a given mm
997  */
998 #define pgd_offset(mm, address) pgd_offset_pgd((mm)->pgd, (address))
999 /*
1000  * a shortcut which implies the use of the kernel's pgd, instead
1001  * of a process's
1002  */
1003 #define pgd_offset_k(address) pgd_offset(&init_mm, (address))
1004
1005
1006 #define KERNEL_PGD_BOUNDARY     pgd_index(PAGE_OFFSET)
1007 #define KERNEL_PGD_PTRS         (PTRS_PER_PGD - KERNEL_PGD_BOUNDARY)
1008
1009 #ifndef __ASSEMBLY__
1010
1011 extern int direct_gbpages;
1012 void init_mem_mapping(void);
1013 void early_alloc_pgt_buf(void);
1014 extern void memblock_find_dma_reserve(void);
1015
1016 #ifdef CONFIG_X86_64
1017 /* Realmode trampoline initialization. */
1018 extern pgd_t trampoline_pgd_entry;
1019 static inline void __meminit init_trampoline_default(void)
1020 {
1021         /* Default trampoline pgd value */
1022         trampoline_pgd_entry = init_top_pgt[pgd_index(__PAGE_OFFSET)];
1023 }
1024
1025 void __init poking_init(void);
1026
1027 # ifdef CONFIG_RANDOMIZE_MEMORY
1028 void __meminit init_trampoline(void);
1029 # else
1030 #  define init_trampoline init_trampoline_default
1031 # endif
1032 #else
1033 static inline void init_trampoline(void) { }
1034 #endif
1035
1036 /* local pte updates need not use xchg for locking */
1037 static inline pte_t native_local_ptep_get_and_clear(pte_t *ptep)
1038 {
1039         pte_t res = *ptep;
1040
1041         /* Pure native function needs no input for mm, addr */
1042         native_pte_clear(NULL, 0, ptep);
1043         return res;
1044 }
1045
1046 static inline pmd_t native_local_pmdp_get_and_clear(pmd_t *pmdp)
1047 {
1048         pmd_t res = *pmdp;
1049
1050         native_pmd_clear(pmdp);
1051         return res;
1052 }
1053
1054 static inline pud_t native_local_pudp_get_and_clear(pud_t *pudp)
1055 {
1056         pud_t res = *pudp;
1057
1058         native_pud_clear(pudp);
1059         return res;
1060 }
1061
1062 static inline void native_set_pte_at(struct mm_struct *mm, unsigned long addr,
1063                                      pte_t *ptep , pte_t pte)
1064 {
1065         native_set_pte(ptep, pte);
1066 }
1067
1068 static inline void set_pmd_at(struct mm_struct *mm, unsigned long addr,
1069                               pmd_t *pmdp, pmd_t pmd)
1070 {
1071         set_pmd(pmdp, pmd);
1072 }
1073
1074 static inline void set_pud_at(struct mm_struct *mm, unsigned long addr,
1075                               pud_t *pudp, pud_t pud)
1076 {
1077         native_set_pud(pudp, pud);
1078 }
1079
1080 /*
1081  * We only update the dirty/accessed state if we set
1082  * the dirty bit by hand in the kernel, since the hardware
1083  * will do the accessed bit for us, and we don't want to
1084  * race with other CPU's that might be updating the dirty
1085  * bit at the same time.
1086  */
1087 struct vm_area_struct;
1088
1089 #define  __HAVE_ARCH_PTEP_SET_ACCESS_FLAGS
1090 extern int ptep_set_access_flags(struct vm_area_struct *vma,
1091                                  unsigned long address, pte_t *ptep,
1092                                  pte_t entry, int dirty);
1093
1094 #define __HAVE_ARCH_PTEP_TEST_AND_CLEAR_YOUNG
1095 extern int ptep_test_and_clear_young(struct vm_area_struct *vma,
1096                                      unsigned long addr, pte_t *ptep);
1097
1098 #define __HAVE_ARCH_PTEP_CLEAR_YOUNG_FLUSH
1099 extern int ptep_clear_flush_young(struct vm_area_struct *vma,
1100                                   unsigned long address, pte_t *ptep);
1101
1102 #define __HAVE_ARCH_PTEP_GET_AND_CLEAR
1103 static inline pte_t ptep_get_and_clear(struct mm_struct *mm, unsigned long addr,
1104                                        pte_t *ptep)
1105 {
1106         pte_t pte = native_ptep_get_and_clear(ptep);
1107         return pte;
1108 }
1109
1110 #define __HAVE_ARCH_PTEP_GET_AND_CLEAR_FULL
1111 static inline pte_t ptep_get_and_clear_full(struct mm_struct *mm,
1112                                             unsigned long addr, pte_t *ptep,
1113                                             int full)
1114 {
1115         pte_t pte;
1116         if (full) {
1117                 /*
1118                  * Full address destruction in progress; paravirt does not
1119                  * care about updates and native needs no locking
1120                  */
1121                 pte = native_local_ptep_get_and_clear(ptep);
1122         } else {
1123                 pte = ptep_get_and_clear(mm, addr, ptep);
1124         }
1125         return pte;
1126 }
1127
1128 #define __HAVE_ARCH_PTEP_SET_WRPROTECT
1129 static inline void ptep_set_wrprotect(struct mm_struct *mm,
1130                                       unsigned long addr, pte_t *ptep)
1131 {
1132         clear_bit(_PAGE_BIT_RW, (unsigned long *)&ptep->pte);
1133 }
1134
1135 #define flush_tlb_fix_spurious_fault(vma, address) do { } while (0)
1136
1137 #define mk_pmd(page, pgprot)   pfn_pmd(page_to_pfn(page), (pgprot))
1138
1139 #define  __HAVE_ARCH_PMDP_SET_ACCESS_FLAGS
1140 extern int pmdp_set_access_flags(struct vm_area_struct *vma,
1141                                  unsigned long address, pmd_t *pmdp,
1142                                  pmd_t entry, int dirty);
1143 extern int pudp_set_access_flags(struct vm_area_struct *vma,
1144                                  unsigned long address, pud_t *pudp,
1145                                  pud_t entry, int dirty);
1146
1147 #define __HAVE_ARCH_PMDP_TEST_AND_CLEAR_YOUNG
1148 extern int pmdp_test_and_clear_young(struct vm_area_struct *vma,
1149                                      unsigned long addr, pmd_t *pmdp);
1150 extern int pudp_test_and_clear_young(struct vm_area_struct *vma,
1151                                      unsigned long addr, pud_t *pudp);
1152
1153 #define __HAVE_ARCH_PMDP_CLEAR_YOUNG_FLUSH
1154 extern int pmdp_clear_flush_young(struct vm_area_struct *vma,
1155                                   unsigned long address, pmd_t *pmdp);
1156
1157
1158 #define pmd_write pmd_write
1159 static inline int pmd_write(pmd_t pmd)
1160 {
1161         return pmd_flags(pmd) & _PAGE_RW;
1162 }
1163
1164 #define __HAVE_ARCH_PMDP_HUGE_GET_AND_CLEAR
1165 static inline pmd_t pmdp_huge_get_and_clear(struct mm_struct *mm, unsigned long addr,
1166                                        pmd_t *pmdp)
1167 {
1168         return native_pmdp_get_and_clear(pmdp);
1169 }
1170
1171 #define __HAVE_ARCH_PUDP_HUGE_GET_AND_CLEAR
1172 static inline pud_t pudp_huge_get_and_clear(struct mm_struct *mm,
1173                                         unsigned long addr, pud_t *pudp)
1174 {
1175         return native_pudp_get_and_clear(pudp);
1176 }
1177
1178 #define __HAVE_ARCH_PMDP_SET_WRPROTECT
1179 static inline void pmdp_set_wrprotect(struct mm_struct *mm,
1180                                       unsigned long addr, pmd_t *pmdp)
1181 {
1182         clear_bit(_PAGE_BIT_RW, (unsigned long *)pmdp);
1183 }
1184
1185 #define pud_write pud_write
1186 static inline int pud_write(pud_t pud)
1187 {
1188         return pud_flags(pud) & _PAGE_RW;
1189 }
1190
1191 #ifndef pmdp_establish
1192 #define pmdp_establish pmdp_establish
1193 static inline pmd_t pmdp_establish(struct vm_area_struct *vma,
1194                 unsigned long address, pmd_t *pmdp, pmd_t pmd)
1195 {
1196         if (IS_ENABLED(CONFIG_SMP)) {
1197                 return xchg(pmdp, pmd);
1198         } else {
1199                 pmd_t old = *pmdp;
1200                 WRITE_ONCE(*pmdp, pmd);
1201                 return old;
1202         }
1203 }
1204 #endif
1205 /*
1206  * Page table pages are page-aligned.  The lower half of the top
1207  * level is used for userspace and the top half for the kernel.
1208  *
1209  * Returns true for parts of the PGD that map userspace and
1210  * false for the parts that map the kernel.
1211  */
1212 static inline bool pgdp_maps_userspace(void *__ptr)
1213 {
1214         unsigned long ptr = (unsigned long)__ptr;
1215
1216         return (((ptr & ~PAGE_MASK) / sizeof(pgd_t)) < PGD_KERNEL_START);
1217 }
1218
1219 static inline int pgd_large(pgd_t pgd) { return 0; }
1220
1221 #ifdef CONFIG_PAGE_TABLE_ISOLATION
1222 /*
1223  * All top-level PAGE_TABLE_ISOLATION page tables are order-1 pages
1224  * (8k-aligned and 8k in size).  The kernel one is at the beginning 4k and
1225  * the user one is in the last 4k.  To switch between them, you
1226  * just need to flip the 12th bit in their addresses.
1227  */
1228 #define PTI_PGTABLE_SWITCH_BIT  PAGE_SHIFT
1229
1230 /*
1231  * This generates better code than the inline assembly in
1232  * __set_bit().
1233  */
1234 static inline void *ptr_set_bit(void *ptr, int bit)
1235 {
1236         unsigned long __ptr = (unsigned long)ptr;
1237
1238         __ptr |= BIT(bit);
1239         return (void *)__ptr;
1240 }
1241 static inline void *ptr_clear_bit(void *ptr, int bit)
1242 {
1243         unsigned long __ptr = (unsigned long)ptr;
1244
1245         __ptr &= ~BIT(bit);
1246         return (void *)__ptr;
1247 }
1248
1249 static inline pgd_t *kernel_to_user_pgdp(pgd_t *pgdp)
1250 {
1251         return ptr_set_bit(pgdp, PTI_PGTABLE_SWITCH_BIT);
1252 }
1253
1254 static inline pgd_t *user_to_kernel_pgdp(pgd_t *pgdp)
1255 {
1256         return ptr_clear_bit(pgdp, PTI_PGTABLE_SWITCH_BIT);
1257 }
1258
1259 static inline p4d_t *kernel_to_user_p4dp(p4d_t *p4dp)
1260 {
1261         return ptr_set_bit(p4dp, PTI_PGTABLE_SWITCH_BIT);
1262 }
1263
1264 static inline p4d_t *user_to_kernel_p4dp(p4d_t *p4dp)
1265 {
1266         return ptr_clear_bit(p4dp, PTI_PGTABLE_SWITCH_BIT);
1267 }
1268 #endif /* CONFIG_PAGE_TABLE_ISOLATION */
1269
1270 /*
1271  * clone_pgd_range(pgd_t *dst, pgd_t *src, int count);
1272  *
1273  *  dst - pointer to pgd range anwhere on a pgd page
1274  *  src - ""
1275  *  count - the number of pgds to copy.
1276  *
1277  * dst and src can be on the same page, but the range must not overlap,
1278  * and must not cross a page boundary.
1279  */
1280 static inline void clone_pgd_range(pgd_t *dst, pgd_t *src, int count)
1281 {
1282         memcpy(dst, src, count * sizeof(pgd_t));
1283 #ifdef CONFIG_PAGE_TABLE_ISOLATION
1284         if (!static_cpu_has(X86_FEATURE_PTI))
1285                 return;
1286         /* Clone the user space pgd as well */
1287         memcpy(kernel_to_user_pgdp(dst), kernel_to_user_pgdp(src),
1288                count * sizeof(pgd_t));
1289 #endif
1290 }
1291
1292 #define PTE_SHIFT ilog2(PTRS_PER_PTE)
1293 static inline int page_level_shift(enum pg_level level)
1294 {
1295         return (PAGE_SHIFT - PTE_SHIFT) + level * PTE_SHIFT;
1296 }
1297 static inline unsigned long page_level_size(enum pg_level level)
1298 {
1299         return 1UL << page_level_shift(level);
1300 }
1301 static inline unsigned long page_level_mask(enum pg_level level)
1302 {
1303         return ~(page_level_size(level) - 1);
1304 }
1305
1306 /*
1307  * The x86 doesn't have any external MMU info: the kernel page
1308  * tables contain all the necessary information.
1309  */
1310 static inline void update_mmu_cache(struct vm_area_struct *vma,
1311                 unsigned long addr, pte_t *ptep)
1312 {
1313 }
1314 static inline void update_mmu_cache_pmd(struct vm_area_struct *vma,
1315                 unsigned long addr, pmd_t *pmd)
1316 {
1317 }
1318 static inline void update_mmu_cache_pud(struct vm_area_struct *vma,
1319                 unsigned long addr, pud_t *pud)
1320 {
1321 }
1322
1323 #ifdef CONFIG_HAVE_ARCH_SOFT_DIRTY
1324 static inline pte_t pte_swp_mksoft_dirty(pte_t pte)
1325 {
1326         return pte_set_flags(pte, _PAGE_SWP_SOFT_DIRTY);
1327 }
1328
1329 static inline int pte_swp_soft_dirty(pte_t pte)
1330 {
1331         return pte_flags(pte) & _PAGE_SWP_SOFT_DIRTY;
1332 }
1333
1334 static inline pte_t pte_swp_clear_soft_dirty(pte_t pte)
1335 {
1336         return pte_clear_flags(pte, _PAGE_SWP_SOFT_DIRTY);
1337 }
1338
1339 #ifdef CONFIG_ARCH_ENABLE_THP_MIGRATION
1340 static inline pmd_t pmd_swp_mksoft_dirty(pmd_t pmd)
1341 {
1342         return pmd_set_flags(pmd, _PAGE_SWP_SOFT_DIRTY);
1343 }
1344
1345 static inline int pmd_swp_soft_dirty(pmd_t pmd)
1346 {
1347         return pmd_flags(pmd) & _PAGE_SWP_SOFT_DIRTY;
1348 }
1349
1350 static inline pmd_t pmd_swp_clear_soft_dirty(pmd_t pmd)
1351 {
1352         return pmd_clear_flags(pmd, _PAGE_SWP_SOFT_DIRTY);
1353 }
1354 #endif
1355 #endif
1356
1357 #define PKRU_AD_BIT 0x1
1358 #define PKRU_WD_BIT 0x2
1359 #define PKRU_BITS_PER_PKEY 2
1360
1361 static inline bool __pkru_allows_read(u32 pkru, u16 pkey)
1362 {
1363         int pkru_pkey_bits = pkey * PKRU_BITS_PER_PKEY;
1364         return !(pkru & (PKRU_AD_BIT << pkru_pkey_bits));
1365 }
1366
1367 static inline bool __pkru_allows_write(u32 pkru, u16 pkey)
1368 {
1369         int pkru_pkey_bits = pkey * PKRU_BITS_PER_PKEY;
1370         /*
1371          * Access-disable disables writes too so we need to check
1372          * both bits here.
1373          */
1374         return !(pkru & ((PKRU_AD_BIT|PKRU_WD_BIT) << pkru_pkey_bits));
1375 }
1376
1377 static inline u16 pte_flags_pkey(unsigned long pte_flags)
1378 {
1379 #ifdef CONFIG_X86_INTEL_MEMORY_PROTECTION_KEYS
1380         /* ifdef to avoid doing 59-bit shift on 32-bit values */
1381         return (pte_flags & _PAGE_PKEY_MASK) >> _PAGE_BIT_PKEY_BIT0;
1382 #else
1383         return 0;
1384 #endif
1385 }
1386
1387 static inline bool __pkru_allows_pkey(u16 pkey, bool write)
1388 {
1389         u32 pkru = read_pkru();
1390
1391         if (!__pkru_allows_read(pkru, pkey))
1392                 return false;
1393         if (write && !__pkru_allows_write(pkru, pkey))
1394                 return false;
1395
1396         return true;
1397 }
1398
1399 /*
1400  * 'pteval' can come from a PTE, PMD or PUD.  We only check
1401  * _PAGE_PRESENT, _PAGE_USER, and _PAGE_RW in here which are the
1402  * same value on all 3 types.
1403  */
1404 static inline bool __pte_access_permitted(unsigned long pteval, bool write)
1405 {
1406         unsigned long need_pte_bits = _PAGE_PRESENT|_PAGE_USER;
1407
1408         if (write)
1409                 need_pte_bits |= _PAGE_RW;
1410
1411         if ((pteval & need_pte_bits) != need_pte_bits)
1412                 return 0;
1413
1414         return __pkru_allows_pkey(pte_flags_pkey(pteval), write);
1415 }
1416
1417 #define pte_access_permitted pte_access_permitted
1418 static inline bool pte_access_permitted(pte_t pte, bool write)
1419 {
1420         return __pte_access_permitted(pte_val(pte), write);
1421 }
1422
1423 #define pmd_access_permitted pmd_access_permitted
1424 static inline bool pmd_access_permitted(pmd_t pmd, bool write)
1425 {
1426         return __pte_access_permitted(pmd_val(pmd), write);
1427 }
1428
1429 #define pud_access_permitted pud_access_permitted
1430 static inline bool pud_access_permitted(pud_t pud, bool write)
1431 {
1432         return __pte_access_permitted(pud_val(pud), write);
1433 }
1434
1435 #define __HAVE_ARCH_PFN_MODIFY_ALLOWED 1
1436 extern bool pfn_modify_allowed(unsigned long pfn, pgprot_t prot);
1437
1438 static inline bool arch_has_pfn_modify_check(void)
1439 {
1440         return boot_cpu_has_bug(X86_BUG_L1TF);
1441 }
1442
1443 #include <asm-generic/pgtable.h>
1444 #endif  /* __ASSEMBLY__ */
1445
1446 #endif /* _ASM_X86_PGTABLE_H */