1#ifndef _ASM_POWERPC_PTE_WALK_H
2#define _ASM_POWERPC_PTE_WALK_H
3
4#include <linux/sched.h>
5
6/* Don't use this directly */
7extern pte_t *__find_linux_pte(pgd_t *pgdir, unsigned long ea,
8			       bool *is_thp, unsigned *hshift);
9
10static inline pte_t *find_linux_pte(pgd_t *pgdir, unsigned long ea,
11				    bool *is_thp, unsigned *hshift)
12{
13	pte_t *pte;
14
15	VM_WARN(!arch_irqs_disabled(), "%s called with irq enabled\n", __func__);
16	pte = __find_linux_pte(pgdir, ea, is_thp, hshift);
17
18#if defined(CONFIG_DEBUG_VM) &&						\
19	!(defined(CONFIG_HUGETLB_PAGE) || defined(CONFIG_TRANSPARENT_HUGEPAGE))
20	/*
21	 * We should not find huge page if these configs are not enabled.
22	 */
23	if (hshift)
24		WARN_ON(*hshift);
25#endif
26	return pte;
27}
28
29static inline pte_t *find_init_mm_pte(unsigned long ea, unsigned *hshift)
30{
31	pgd_t *pgdir = init_mm.pgd;
32	return __find_linux_pte(pgdir, ea, NULL, hshift);
33}
34/*
35 * This is what we should always use. Any other lockless page table lookup needs
36 * careful audit against THP split.
37 */
38static inline pte_t *find_current_mm_pte(pgd_t *pgdir, unsigned long ea,
39					 bool *is_thp, unsigned *hshift)
40{
41	pte_t *pte;
42
43	VM_WARN(!arch_irqs_disabled(), "%s called with irq enabled\n", __func__);
44	VM_WARN(pgdir != current->mm->pgd,
45		"%s lock less page table lookup called on wrong mm\n", __func__);
46	pte = __find_linux_pte(pgdir, ea, is_thp, hshift);
47
48#if defined(CONFIG_DEBUG_VM) &&						\
49	!(defined(CONFIG_HUGETLB_PAGE) || defined(CONFIG_TRANSPARENT_HUGEPAGE))
50	/*
51	 * We should not find huge page if these configs are not enabled.
52	 */
53	if (hshift)
54		WARN_ON(*hshift);
55#endif
56	return pte;
57}
58
59#endif /* _ASM_POWERPC_PTE_WALK_H */
60