/* SPDX-License-Identifier: GPL-2.0 */
/*
 * Copyright (C) 2020-2022 Loongson Technology Corporation Limited
 */

#ifndef __ASM_HUGETLB_H
#define __ASM_HUGETLB_H

#include <asm/page.h>

uint64_t pmd_to_entrylo(unsigned long pmd_val);

#define __HAVE_ARCH_HUGE_PTE_CLEAR
static inline void huge_pte_clear(struct mm_struct *mm, unsigned long addr,
                                  pte_t *ptep, unsigned long sz)
{
        pte_t clear;

        pte_val(clear) = (unsigned long)invalid_pte_table;
        set_pte_at(mm, addr, ptep, clear);
}

#define __HAVE_ARCH_HUGE_PTEP_GET_AND_CLEAR
static inline pte_t huge_ptep_get_and_clear(struct mm_struct *mm,
                                            unsigned long addr, pte_t *ptep,
                                            unsigned long sz)
{
        pte_t clear;
        pte_t pte = ptep_get(ptep);

        pte_val(clear) = (unsigned long)invalid_pte_table;
        set_pte_at(mm, addr, ptep, clear);
        return pte;
}

#define __HAVE_ARCH_HUGE_PTEP_CLEAR_FLUSH
static inline pte_t huge_ptep_clear_flush(struct vm_area_struct *vma,
                                          unsigned long addr, pte_t *ptep)
{
        pte_t pte;
        unsigned long sz = huge_page_size(hstate_vma(vma));

        pte = huge_ptep_get_and_clear(vma->vm_mm, addr, ptep, sz);
        flush_tlb_page(vma, addr);
        return pte;
}

#define __HAVE_ARCH_HUGE_PTE_NONE
static inline int huge_pte_none(pte_t pte)
{
        unsigned long val = pte_val(pte) & ~_PAGE_GLOBAL;
        return !val || (val == (unsigned long)invalid_pte_table);
}

#define __HAVE_ARCH_HUGE_PTEP_SET_ACCESS_FLAGS
static inline int huge_ptep_set_access_flags(struct vm_area_struct *vma,
                                             unsigned long addr,
                                             pte_t *ptep, pte_t pte,
                                             int dirty)
{
        int changed = !pte_same(ptep_get(ptep), pte);

        if (changed) {
                set_pte_at(vma->vm_mm, addr, ptep, pte);
                /*
                 * There could be some standard sized pages in there,
                 * get them all.
                 */
                flush_tlb_range(vma, addr, addr + HPAGE_SIZE);
        }
        return changed;
}

#include <asm-generic/hugetlb.h>

#endif /* __ASM_HUGETLB_H */