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

#include <linux/irqflags.h>

/*
 * To get proper branch prediction for the main line, we must branch
 * forward to code at the end of this object's .text section, then
 * branch back to restart the operation.
 */

#define ATOMIC_OP(op, c_op)                                             \
static inline void arch_atomic_##op(int i, atomic_t *v)                 \
{                                                                       \
        unsigned long flags;                                            \
                                                                        \
        raw_local_irq_save(flags);                                      \
        v->counter c_op i;                                              \
        raw_local_irq_restore(flags);                                   \
}

#define ATOMIC_OP_RETURN(op, c_op)                                      \
static inline int arch_atomic_##op##_return(int i, atomic_t *v)         \
{                                                                       \
        unsigned long temp, flags;                                      \
                                                                        \
        raw_local_irq_save(flags);                                      \
        temp = v->counter;                                              \
        temp c_op i;                                                    \
        v->counter = temp;                                              \
        raw_local_irq_restore(flags);                                   \
                                                                        \
        return temp;                                                    \
}

#define ATOMIC_FETCH_OP(op, c_op)                                       \
static inline int arch_atomic_fetch_##op(int i, atomic_t *v)            \
{                                                                       \
        unsigned long temp, flags;                                      \
                                                                        \
        raw_local_irq_save(flags);                                      \
        temp = v->counter;                                              \
        v->counter c_op i;                                              \
        raw_local_irq_restore(flags);                                   \
                                                                        \
        return temp;                                                    \
}

#define ATOMIC_OPS(op, c_op)                                            \
        ATOMIC_OP(op, c_op)                                             \
        ATOMIC_OP_RETURN(op, c_op)                                      \
        ATOMIC_FETCH_OP(op, c_op)

ATOMIC_OPS(add, +=)
ATOMIC_OPS(sub, -=)

#define arch_atomic_add_return  arch_atomic_add_return
#define arch_atomic_sub_return  arch_atomic_sub_return
#define arch_atomic_fetch_add   arch_atomic_fetch_add
#define arch_atomic_fetch_sub   arch_atomic_fetch_sub

#undef ATOMIC_OPS
#define ATOMIC_OPS(op, c_op)                                            \
        ATOMIC_OP(op, c_op)                                             \
        ATOMIC_FETCH_OP(op, c_op)

ATOMIC_OPS(and, &=)
ATOMIC_OPS(or, |=)
ATOMIC_OPS(xor, ^=)

#define arch_atomic_fetch_and   arch_atomic_fetch_and
#define arch_atomic_fetch_or    arch_atomic_fetch_or
#define arch_atomic_fetch_xor   arch_atomic_fetch_xor

#undef ATOMIC_OPS
#undef ATOMIC_FETCH_OP
#undef ATOMIC_OP_RETURN
#undef ATOMIC_OP

#endif /* __ASM_SH_ATOMIC_IRQ_H */