/*
 * Copyright (C) 2017 XRADIO TECHNOLOGY CO., LTD. All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions
 *  are met:
 *    1. Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *    2. Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the
 *       distribution.
 *    3. Neither the name of XRADIO TECHNOLOGY CO., LTD. nor the names of
 *       its contributors may be used to endorse or promote products derived
 *       from this software without specific prior written permission.
 *
 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#ifndef _SYS_INTERRUPT_H_
#define _SYS_INTERRUPT_H_

#include "compiler.h"

#if defined(__CC_ARM) /* ARM Compiler */

/*
 * CPU interrupt mask handling.
 */

#define IRQMASK_REG_NAME_R primask
#define IRQMASK_REG_NAME_W primask

/* Save the current interrupt enable state and disable IRQ */
static __always_inline unsigned long arch_irq_save(void)
{
	unsigned long flags;

	__asm {
		mrs	flags, IRQMASK_REG_NAME_R
		cpsid	i
	}
	return flags;
}

/* Restore saved IRQ state */
static __always_inline void arch_irq_restore(unsigned long flags)
{
	__asm { msr	IRQMASK_REG_NAME_W, flags }
}

/* Get the current interrupt enable state */
static __always_inline unsigned long arch_irq_get_flags(void)
{
	unsigned long flags;

	__asm { mrs	flags, IRQMASK_REG_NAME_R }
	return flags;
}

/* Disable IRQ, non-nested version */
#define arch_irq_disable()	__disable_irq()

/* Enable IRQ, non-nested version */
#define arch_irq_enable()	__enable_irq()

/* Disable FIQ, non-nested version */
#define arch_fiq_disable()	__disable_fiq()

/* Enable FIQ, non-nested version */
#define arch_fiq_enable()	__enable_fiq()

#elif defined(__GNUC__) /* GNU Compiler */

/*
 * CPU interrupt mask handling.
 */

#define IRQMASK_REG_NAME_R "primask"
#define IRQMASK_REG_NAME_W "primask"

/* Save the current interrupt enable state and disable IRQ */
static __always_inline unsigned long arch_irq_save(void)
{
	unsigned long flags;

	__asm volatile(
		"mrs	%0, " IRQMASK_REG_NAME_R "\n"
		"cpsid	i"
		: "=r" (flags) : : "memory", "cc");
	return flags;
}

/* Restore saved IRQ state */
static __always_inline void arch_irq_restore(unsigned long flags)
{
	__asm volatile(
		"msr	" IRQMASK_REG_NAME_W ", %0"
		:
		: "r" (flags)
		: "memory", "cc");
}

/* Get the current interrupt enable state */
static __always_inline unsigned long arch_irq_get_flags(void)
{
	unsigned long flags;

	__asm volatile(
		"mrs	%0, " IRQMASK_REG_NAME_R "\n"
		: "=r" (flags) : : "memory", "cc");
	return flags;

}

/* Disable IRQ, non-nested version */
#define arch_irq_disable()	__asm volatile("cpsid i" : : : "memory", "cc")

/* Enable IRQ, non-nested version */
#define arch_irq_enable()	__asm volatile("cpsie i" : : : "memory", "cc")

/* Disable FIQ, non-nested version */
#define arch_fiq_disable()	__asm volatile("cpsid f" : : : "memory", "cc")

/* Enable FIQ, non-nested version */
#define arch_fiq_enable()	__asm volatile("cpsie f" : : : "memory", "cc")

#else
#error "Compiler not supported."
#endif

/* Disable IRQ, nested version */
void arch_irq_disable_nested(void);

/* Enable IRQ, nested version */
void arch_irq_enable_nested(void);

/* Disable FIQ, nested version */
void arch_fiq_disable_nested(void);

/* Enable FIQ, nested version */
void arch_fiq_enable_nested(void);

/* Obsoleted macros, for compatibility only */
#define xr_irq_save arch_irq_save
#define xr_irq_restore arch_irq_restore
#define arch_local_save_flags arch_irq_get_flags
#define xr_irq_disable arch_irq_disable
#define xr_irq_enable arch_irq_enable
#define xr_fiq_disable arch_fiq_disable
#define xr_fiq_enable arch_fiq_enable

#endif /* _SYS_INTERRUPT_H_ */