/* SPDX-License-Identifier: GPL-2.0 */
/*
 * Copyright (C) 2020-2022 Loongson Technology Corporation Limited
 */
#include <asm/asm.h>
#include <asm/asmmacro.h>
#include <asm/asm-offsets.h>
#include <asm/loongarch.h>
#include <asm/regdef.h>
#include <asm/stackframe.h>
#include <asm/thread_info.h>

/*
 * task_struct *__switch_to(task_struct *prev, task_struct *next,
 *			    struct thread_info *next_ti, void *sched_ra, void *sched_cfa)
 */
	.align	5
SYM_FUNC_START(__switch_to)
#ifdef CONFIG_32BIT
	PTR_ADDI	a0, a0, TASK_STRUCT_OFFSET
	PTR_ADDI	a1, a1, TASK_STRUCT_OFFSET
#endif
	csrrd		t1, LOONGARCH_CSR_PRMD
	LONG_SPTR	t1, a0, (THREAD_CSRPRMD - TASK_STRUCT_OFFSET)

	cpu_save_nonscratch a0
	LONG_SPTR	a3, a0, (THREAD_SCHED_RA - TASK_STRUCT_OFFSET)
	LONG_SPTR	a4, a0, (THREAD_SCHED_CFA - TASK_STRUCT_OFFSET)

#if defined(CONFIG_STACKPROTECTOR) && !defined(CONFIG_SMP)
	la		t7, __stack_chk_guard
	LONG_LPTR	t8, a1, (TASK_STACK_CANARY - TASK_STRUCT_OFFSET)
	LONG_SPTR	t8, t7, 0
#endif

	move	tp, a2
	cpu_restore_nonscratch a1

	li.w		t0, _THREAD_SIZE
	PTR_ADD		t0, t0, tp
	set_saved_sp	t0, t1, t2

	LONG_LPTR	t1, a1, (THREAD_CSRPRMD - TASK_STRUCT_OFFSET)
	csrwr		t1, LOONGARCH_CSR_PRMD

#ifdef CONFIG_32BIT
	PTR_ADDI	a0, a0, -TASK_STRUCT_OFFSET
#endif
	jr	ra
SYM_FUNC_END(__switch_to)