/* SPDX-License-Identifier: GPL-2.0 */
#ifndef __ASM_CPUIDLE_H
#define __ASM_CPUIDLE_H

#include <asm/proc-fns.h>

#ifdef CONFIG_ARM64_PSEUDO_NMI
#include <asm/arch_gicv3.h>

struct arm_cpuidle_irq_context {
    unsigned long pmr;
    unsigned long daif_bits;
};

#define arm_cpuidle_save_irq_context(__c)                \
    do {                                \
        struct arm_cpuidle_irq_context *c = __c;        \
        if (system_uses_irq_prio_masking()) {            \
            c->daif_bits = read_sysreg(daif);        \
            write_sysreg(c->daif_bits | PSR_I_BIT | PSR_F_BIT, \
                     daif);                \
            c->pmr = gic_read_pmr();            \
            gic_write_pmr(GIC_PRIO_IRQON | GIC_PRIO_PSR_I_SET); \
        }                            \
    } while (0)

#define arm_cpuidle_restore_irq_context(__c)                \
    do {                                \
        struct arm_cpuidle_irq_context *c = __c;        \
        if (system_uses_irq_prio_masking()) {            \
            gic_write_pmr(c->pmr);                \
            write_sysreg(c->daif_bits, daif);        \
        }                            \
    } while (0)
#else
struct arm_cpuidle_irq_context { };

#define arm_cpuidle_save_irq_context(c)        (void)c
#define arm_cpuidle_restore_irq_context(c)    (void)c
#endif
#endif