#ifndef _ASM_POWERPC_PTE_WALK_H
#define _ASM_POWERPC_PTE_WALK_H

#include <linux/sched.h>

/* Don't use this directly */
extern pte_t *__find_linux_pte(pgd_t *pgdir, unsigned long ea,
                               bool *is_thp, unsigned *hshift);

static inline pte_t *find_linux_pte(pgd_t *pgdir, unsigned long ea,
                                    bool *is_thp, unsigned *hshift)
{
        pte_t *pte;

        VM_WARN(!arch_irqs_disabled(), "%s called with irq enabled\n", __func__);
        pte = __find_linux_pte(pgdir, ea, is_thp, hshift);

#if defined(CONFIG_DEBUG_VM) &&                                         \
        !(defined(CONFIG_HUGETLB_PAGE) || defined(CONFIG_TRANSPARENT_HUGEPAGE))
        /*
         * We should not find huge page if these configs are not enabled.
         */
        if (hshift)
                WARN_ON(*hshift);
#endif
        return pte;
}

static inline pte_t *find_init_mm_pte(unsigned long ea, unsigned *hshift)
{
        pgd_t *pgdir = init_mm.pgd;
        return __find_linux_pte(pgdir, ea, NULL, hshift);
}

/*
 * Convert a kernel vmap virtual address (vmalloc or ioremap space) to a
 * physical address, without taking locks. This can be used in real-mode.
 */
static inline phys_addr_t ppc_find_vmap_phys(unsigned long addr)
{
        pte_t *ptep;
        phys_addr_t pa;
        int hugepage_shift;

        /*
         * init_mm does not free page tables, and does not do THP. It may
         * have huge pages from huge vmalloc / ioremap etc.
         */
        ptep = find_init_mm_pte(addr, &hugepage_shift);
        if (WARN_ON(!ptep))
                return 0;

        pa = PFN_PHYS(pte_pfn(*ptep));

        if (!hugepage_shift)
                hugepage_shift = PAGE_SHIFT;

        pa |= addr & ((1ul << hugepage_shift) - 1);

        return pa;
}

#endif /* _ASM_POWERPC_PTE_WALK_H */