/* SPDX-License-Identifier: GPL-2.0 */
/*
 * Copyright (C) 2006 Atmark Techno, Inc.
 */

#ifndef _ASM_MICROBLAZE_IRQFLAGS_H
#define _ASM_MICROBLAZE_IRQFLAGS_H

#include <linux/types.h>
#include <asm/registers.h>

#if CONFIG_XILINX_MICROBLAZE0_USE_MSR_INSTR

static inline notrace unsigned long arch_local_irq_save(void)
{
        unsigned long flags;
        asm volatile("  msrclr %0, %1   \n"
                     "  nop             \n"
                     : "=r"(flags)
                     : "i"(MSR_IE)
                     : "memory");
        return flags;
}

static inline notrace void arch_local_irq_disable(void)
{
        /* this uses r0 without declaring it - is that correct? */
        asm volatile("  msrclr r0, %0   \n"
                     "  nop             \n"
                     :
                     : "i"(MSR_IE)
                     : "memory");
}

static inline notrace void arch_local_irq_enable(void)
{
        /* this uses r0 without declaring it - is that correct? */
        asm volatile("  msrset  r0, %0  \n"
                     "  nop             \n"
                     :
                     : "i"(MSR_IE)
                     : "memory");
}

#else /* !CONFIG_XILINX_MICROBLAZE0_USE_MSR_INSTR */

static inline notrace unsigned long arch_local_irq_save(void)
{
        unsigned long flags, tmp;
        asm volatile (" mfs     %0, rmsr        \n"
                      " nop                     \n"
                      " andi    %1, %0, %2      \n"
                      " mts     rmsr, %1        \n"
                      " nop                     \n"
                      : "=r"(flags), "=r"(tmp)
                      : "i"(~MSR_IE)
                      : "memory");
        return flags;
}

static inline notrace void arch_local_irq_disable(void)
{
        unsigned long tmp;
        asm volatile("  mfs     %0, rmsr        \n"
                     "  nop                     \n"
                     "  andi    %0, %0, %1      \n"
                     "  mts     rmsr, %0        \n"
                     "  nop                     \n"
                     : "=r"(tmp)
                     : "i"(~MSR_IE)
                     : "memory");
}

static inline notrace void arch_local_irq_enable(void)
{
        unsigned long tmp;
        asm volatile("  mfs     %0, rmsr        \n"
                     "  nop                     \n"
                     "  ori     %0, %0, %1      \n"
                     "  mts     rmsr, %0        \n"
                     "  nop                     \n"
                     : "=r"(tmp)
                     : "i"(MSR_IE)
                     : "memory");
}

#endif /* CONFIG_XILINX_MICROBLAZE0_USE_MSR_INSTR */

static inline notrace unsigned long arch_local_save_flags(void)
{
        unsigned long flags;
        asm volatile("  mfs     %0, rmsr        \n"
                     "  nop                     \n"
                     : "=r"(flags)
                     :
                     : "memory");
        return flags;
}

static inline notrace void arch_local_irq_restore(unsigned long flags)
{
        asm volatile("  mts     rmsr, %0        \n"
                     "  nop                     \n"
                     :
                     : "r"(flags)
                     : "memory");
}

static inline notrace bool arch_irqs_disabled_flags(unsigned long flags)
{
        return (flags & MSR_IE) == 0;
}

static inline notrace bool arch_irqs_disabled(void)
{
        return arch_irqs_disabled_flags(arch_local_save_flags());
}

#endif /* _ASM_MICROBLAZE_IRQFLAGS_H */