/* SPDX-License-Identifier: GPL-2.0-only */
/*
 * Copyright (C) 2012 ARM Ltd.
 */
#ifndef __ASM_IRQFLAGS_H
#define __ASM_IRQFLAGS_H

#include <asm/barrier.h>
#include <asm/ptrace.h>
#include <asm/sysreg.h>

/*
 * Aarch64 has flags for masking: Debug, Asynchronous (serror), Interrupts and
 * FIQ exceptions, in the 'daif' register. We mask and unmask them in 'daif'
 * order:
 * Masking debug exceptions causes all other exceptions to be masked too/
 * Masking SError masks IRQ/FIQ, but not debug exceptions. IRQ and FIQ are
 * always masked and unmasked together, and have no side effects for other
 * flags. Keeping to this order makes it easier for entry.S to know which
 * exceptions should be unmasked.
 */

static __always_inline void __daif_local_irq_enable(void)
{
 barrier();
 asm volatile("msr daifclr, #3");
 barrier();
}

static __always_inline void __pmr_local_irq_enable(void)
{
 if (IS_ENABLED(CONFIG_ARM64_DEBUG_PRIORITY_MASKING)) {
  u32 pmr = read_sysreg_s(SYS_ICC_PMR_EL1);
  WARN_ON_ONCE(pmr != GIC_PRIO_IRQON && pmr != GIC_PRIO_IRQOFF);
 }

 barrier();
 write_sysreg_s(GIC_PRIO_IRQON, SYS_ICC_PMR_EL1);
 pmr_sync();
 barrier();
}

static inline void arch_local_irq_enable(void)
{
 if (system_uses_irq_prio_masking()) {
  __pmr_local_irq_enable();
 } else {
  __daif_local_irq_enable();
 }
}

static __always_inline void __daif_local_irq_disable(void)
{
 barrier();
 asm volatile("msr daifset, #3");
 barrier();
}

static __always_inline void __pmr_local_irq_disable(void)
{
 if (IS_ENABLED(CONFIG_ARM64_DEBUG_PRIORITY_MASKING)) {
  u32 pmr = read_sysreg_s(SYS_ICC_PMR_EL1);
  WARN_ON_ONCE(pmr != GIC_PRIO_IRQON && pmr != GIC_PRIO_IRQOFF);
 }

 barrier();
 write_sysreg_s(GIC_PRIO_IRQOFF, SYS_ICC_PMR_EL1);
 barrier();
}

static inline void arch_local_irq_disable(void)
{
 if (system_uses_irq_prio_masking()) {
  __pmr_local_irq_disable();
 } else {
  __daif_local_irq_disable();
 }
}

static __always_inline unsigned long __daif_local_save_flags(void)
{
 return read_sysreg(daif);
}

static __always_inline unsigned long __pmr_local_save_flags(void)
{
 return read_sysreg_s(SYS_ICC_PMR_EL1);
}

/*
 * Save the current interrupt enable state.
 */
static inline unsigned long arch_local_save_flags(void)
{
 if (system_uses_irq_prio_masking()) {
  return __pmr_local_save_flags();
 } else {
  return __daif_local_save_flags();
 }
}

static __always_inline bool __daif_irqs_disabled_flags(unsigned long flags)
{
 return flags & PSR_I_BIT;
}

static __always_inline bool __pmr_irqs_disabled_flags(unsigned long flags)
{
 return flags != GIC_PRIO_IRQON;
}

static inline bool arch_irqs_disabled_flags(unsigned long flags)
{
 if (system_uses_irq_prio_masking()) {
  return __pmr_irqs_disabled_flags(flags);
 } else {
  return __daif_irqs_disabled_flags(flags);
 }
}

static __always_inline bool __daif_irqs_disabled(void)
{
 return __daif_irqs_disabled_flags(__daif_local_save_flags());
}

static __always_inline bool __pmr_irqs_disabled(void)
{
 return __pmr_irqs_disabled_flags(__pmr_local_save_flags());
}

static inline bool arch_irqs_disabled(void)
{
 if (system_uses_irq_prio_masking()) {
  return __pmr_irqs_disabled();
 } else {
  return __daif_irqs_disabled();
 }
}

static __always_inline unsigned long __daif_local_irq_save(void)
{
 unsigned long flags = __daif_local_save_flags();

 __daif_local_irq_disable();

 return flags;
}

static __always_inline unsigned long __pmr_local_irq_save(void)
{
 unsigned long flags = __pmr_local_save_flags();

 /*
 * There are too many states with IRQs disabled, just keep the current
 * state if interrupts are already disabled/masked.
 */
 if (!__pmr_irqs_disabled_flags(flags))
  __pmr_local_irq_disable();

 return flags;
}

static inline unsigned long arch_local_irq_save(void)
{
 if (system_uses_irq_prio_masking()) {
  return __pmr_local_irq_save();
 } else {
  return __daif_local_irq_save();
 }
}

static __always_inline void __daif_local_irq_restore(unsigned long flags)
{
 barrier();
 write_sysreg(flags, daif);
 barrier();
}

static __always_inline void __pmr_local_irq_restore(unsigned long flags)
{
 barrier();
 write_sysreg_s(flags, SYS_ICC_PMR_EL1);
 pmr_sync();
 barrier();
}

/*
 * restore saved IRQ state
 */
static inline void arch_local_irq_restore(unsigned long flags)
{
 if (system_uses_irq_prio_masking()) {
  __pmr_local_irq_restore(flags);
 } else {
  __daif_local_irq_restore(flags);
 }
}

#endif /* __ASM_IRQFLAGS_H */