#ifndef _ASM_L_SWITCH_TO_H
#define _ASM_L_SWITCH_TO_H

#ifdef __KERNEL__

#include <asm/mmu_context.h>
#include <asm/monitors.h>
#include <asm/regs_state.h>

extern void preempt_schedule_irq(void);

extern long __ret_from_fork(struct task_struct *prev);

static inline struct task_struct *
native_ret_from_fork_get_prev_task(struct task_struct *prev)
{
	return prev;
}

static inline int
native_ret_from_fork_prepare_hv_stacks(struct pt_regs *regs)
{
	return 0;	/* nothing to do */
}

#ifdef	CONFIG_KVM_GUEST_KERNEL
/* it is pure guest kernel (not paravirtualized based on pv_ops) */
#include <asm/kvm/guest/switch_to.h>
#else
/* it is native kernel without any virtualization */
/* or it is host kernel with virtualization support */
/* or paravirtualized host and guest kernel */

static inline struct task_struct *
ret_from_fork_get_prev_task(struct task_struct *prev)
{
	return native_ret_from_fork_get_prev_task(prev);
}

static inline int
ret_from_fork_prepare_hv_stacks(struct pt_regs *regs)
{
	return native_ret_from_fork_prepare_hv_stacks(regs);
}
#endif	/* CONFIG_KVM_GUEST_KERNEL */

extern struct task_struct *__switch_to(struct task_struct *prev,
				       struct task_struct *next);

#define native_switch_to(prev, next, last)	\
do {						\
	last = __switch_to(prev, next);		\
	e2k_finish_switch(last);		\
} while (0)

#define prepare_arch_switch(next)		\
do {						\
	prefetch_nospec_range(&next->thread.sw_regs,	\
			offsetof(struct sw_regs, cs_lo)); \
        /* It works under CONFIG_MCST_RT */     \
        SAVE_CURR_TIME_SWITCH_TO;               \
	prepare_monitor_regs(next);		\
} while (0)

#define e2k_finish_switch(prev) \
do { \
	CALCULATE_TIME_SWITCH_TO; \
	finish_monitor_regs(prev); \
} while (0)

#ifdef CONFIG_MONITORS
#define prepare_monitor_regs(next)		\
do {						\
	if (MONITORING_IS_ACTIVE)		\
		store_monitors_delta(current);	\
} while (0)
#define finish_monitor_regs(prev)					\
do {									\
	if (MONITORING_IS_ACTIVE) {					\
		prev->thread.sw_regs.ddmcr = NATIVE_READ_DDMCR_REG();	\
		prev->thread.sw_regs.dimcr = NATIVE_READ_DIMCR_REG();	\
		process_monitors(current);				\
	}								\
} while (0)
#else /* !CONFIG_MONITORS */
#define prepare_monitor_regs(next)
#define finish_monitor_regs(next)
#endif /* CONFIG_MONITORS */

#if	defined(CONFIG_PARAVIRT_GUEST)
/* it is paravirtualized host and guest kernel */
#include <asm/paravirt/switch_to.h>
#elif	defined(CONFIG_KVM_GUEST_KERNEL)
/* it is pure guest kernel (not paravirtualized based on pv_ops) */
#include <asm/kvm/guest/switch_to.h>
#else	/* ! CONFIG_PARAVIRT_GUEST && ! CONFIG_KVM_GUEST_KERNEL */
/* it is native kernel without virtualization support */
/* or native kernel with virtualization support */

#define	switch_to(prev, next, last)	native_switch_to(prev, next, last)

#endif	/* CONFIG_PARAVIRT_GUEST */

#endif /* __KERNEL__ */

#endif /* _ASM_L_SWITCH_TO_H */