/* SPDX-License-Identifier: GPL-2.0+ */

#include <linux/linkage.h>
#include <asm/asm.h>
#include "test-kprobes.h"

SYM_FUNC_START(test_kprobes_add)
	li a1, KPROBE_TEST_MAGIC_UPPER
	li a2, KPROBE_TEST_MAGIC_LOWER
test_kprobes_add_addr1:
	add a1, a1, a2
test_kprobes_add_addr2:
	add a0, a1, x0
	ret
SYM_FUNC_END(test_kprobes_add)

SYM_FUNC_START(test_kprobes_jal)
	li a0, 0
	mv a1, ra
	.option push
	.option norvc
test_kprobes_jal_addr1:
	jal x0, 2f
	ret
	.option pop
1:	li a0, KPROBE_TEST_MAGIC_UPPER
	ret
	.option push
	.option norvc
test_kprobes_jal_addr2:
2:	jal 1b
	.option pop
	li a2, KPROBE_TEST_MAGIC_LOWER
	add a0, a0, a2
	jr a1
SYM_FUNC_END(test_kprobes_jal)

SYM_FUNC_START(test_kprobes_jalr)
	la a0, 1f
	mv a1, ra
	.option push
	.option norvc
test_kprobes_jalr_addr:
	jalr a0
	.option pop
	li t0, KPROBE_TEST_MAGIC_UPPER
	add a0, a0, t0
	jr a1
1:	li a0, KPROBE_TEST_MAGIC_LOWER
	ret
SYM_FUNC_END(test_kprobes_jalr)

SYM_FUNC_START(test_kprobes_auipc)
test_kprobes_auipc_addr:
	auipc a0, KPROBE_TEST_MAGIC_LOWER
	la a1, test_kprobes_auipc_addr
	sub a0, a0, a1
	srli a0, a0, 12
	li a1, KPROBE_TEST_MAGIC_UPPER
	add a0, a0, a1
	ret
SYM_FUNC_END(test_kprobes_auipc)

SYM_FUNC_START(test_kprobes_branch)
	.option push
	.option norvc
	li a0, 0
	li a1, 1
	li a2, 2
test_kprobes_branch_addr1:
	beqz a0, 1f
	ret
1:
test_kprobes_branch_addr2:
	beqz a1, 3f
test_kprobes_branch_addr3:
	bnez a0, 3f
test_kprobes_branch_addr4:
	bnez a2, 1f
	ret
1:
test_kprobes_branch_addr5:
	bge a1, a2, 3f
test_kprobes_branch_addr6:
	bge a2, a1, 2f
	ret
1:
	li t0, KPROBE_TEST_MAGIC_UPPER
	add a0, a0, t0
	ret
2:
test_kprobes_branch_addr7:
	blt a2, a1, 3f
	li a0, KPROBE_TEST_MAGIC_LOWER
test_kprobes_branch_addr8:
	blt a1, a2, 1b
3:
	li a0, 0
	ret
	.option pop
SYM_FUNC_END(test_kprobes_branch)

#ifdef CONFIG_RISCV_ISA_C

SYM_FUNC_START(test_kprobes_c_j)
	li a0, 0
test_kprobes_branch_c_j_addr1:
	c.j 2f
1:
	li a1, KPROBE_TEST_MAGIC_UPPER
	add a0, a0, a1
	ret
2:	li a0, KPROBE_TEST_MAGIC_LOWER
test_kprobes_branch_c_j_addr2:
	c.j 1b
SYM_FUNC_END(test_kprobes_c_j)

SYM_FUNC_START(test_kprobes_c_jr)
	la a0, 2f
test_kprobes_c_jr_addr1:
	c.jr a0
	ret
1:	li a1, KPROBE_TEST_MAGIC_LOWER
	add a0, a0, a1
	ret
2:
	li a0, KPROBE_TEST_MAGIC_UPPER
	la a1, 1b
test_kprobes_c_jr_addr2:
	c.jr a1
SYM_FUNC_END(test_kprobes_c_jr)

SYM_FUNC_START(test_kprobes_c_jalr)
	mv a1, ra
	la a0, 1f
test_kprobes_c_jalr_addr:
	c.jalr a0
	li a2, KPROBE_TEST_MAGIC_UPPER
	add a0, a0, a2
	jr a1
1:	li a0, KPROBE_TEST_MAGIC_LOWER
	ret
SYM_FUNC_END(test_kprobes_c_jalr)

SYM_FUNC_START(test_kprobes_c_beqz)
	li a0, 0
	li a1, 1
test_kprobes_c_beqz_addr1:
	c.beqz a0, 2f
	ret
1:	li a1, KPROBE_TEST_MAGIC_UPPER
	add a0, a0, a1
	ret
test_kprobes_c_beqz_addr2:
2:	c.beqz a1, 3f
	li a0, KPROBE_TEST_MAGIC_LOWER
	mv a1, x0
test_kprobes_c_beqz_addr3:
	c.beqz a1, 1b
3:	li a0, 0
	ret
SYM_FUNC_END(test_kprobes_c_beqz)

SYM_FUNC_START(test_kprobes_c_bnez)
	li a0, 0
	li a1, 1
test_kprobes_c_bnez_addr1:
	c.bnez a1, 2f
	ret
1:	li a1, KPROBE_TEST_MAGIC_UPPER
	add a0, a0, a1
	ret
test_kprobes_c_bnez_addr2:
2:	c.bnez a0, 3f
	li a0, KPROBE_TEST_MAGIC_LOWER
test_kprobes_c_bnez_addr3:
	c.bnez a0, 1b
3:	li a0, 0
	ret
SYM_FUNC_END(test_kprobes_c_bnez)

#endif /* CONFIG_RISCV_ISA_C */

.section .rodata
SYM_DATA_START(test_kprobes_addresses)
	RISCV_PTR test_kprobes_add_addr1
	RISCV_PTR test_kprobes_add_addr2
	RISCV_PTR test_kprobes_jal_addr1
	RISCV_PTR test_kprobes_jal_addr2
	RISCV_PTR test_kprobes_jalr_addr
	RISCV_PTR test_kprobes_auipc_addr
	RISCV_PTR test_kprobes_branch_addr1
	RISCV_PTR test_kprobes_branch_addr2
	RISCV_PTR test_kprobes_branch_addr3
	RISCV_PTR test_kprobes_branch_addr4
	RISCV_PTR test_kprobes_branch_addr5
	RISCV_PTR test_kprobes_branch_addr6
	RISCV_PTR test_kprobes_branch_addr7
	RISCV_PTR test_kprobes_branch_addr8
#ifdef CONFIG_RISCV_ISA_C
	RISCV_PTR test_kprobes_branch_c_j_addr1
	RISCV_PTR test_kprobes_branch_c_j_addr2
	RISCV_PTR test_kprobes_c_jr_addr1
	RISCV_PTR test_kprobes_c_jr_addr2
	RISCV_PTR test_kprobes_c_jalr_addr
	RISCV_PTR test_kprobes_c_beqz_addr1
	RISCV_PTR test_kprobes_c_beqz_addr2
	RISCV_PTR test_kprobes_c_beqz_addr3
	RISCV_PTR test_kprobes_c_bnez_addr1
	RISCV_PTR test_kprobes_c_bnez_addr2
	RISCV_PTR test_kprobes_c_bnez_addr3
#endif /* CONFIG_RISCV_ISA_C */
	RISCV_PTR 0
SYM_DATA_END(test_kprobes_addresses)

.section .rodata
SYM_DATA_START(test_kprobes_functions)
	RISCV_PTR test_kprobes_add
	RISCV_PTR test_kprobes_jal
	RISCV_PTR test_kprobes_jalr
	RISCV_PTR test_kprobes_auipc
	RISCV_PTR test_kprobes_branch
#ifdef CONFIG_RISCV_ISA_C
	RISCV_PTR test_kprobes_c_j
	RISCV_PTR test_kprobes_c_jr
	RISCV_PTR test_kprobes_c_jalr
	RISCV_PTR test_kprobes_c_beqz
	RISCV_PTR test_kprobes_c_bnez
#endif /* CONFIG_RISCV_ISA_C */
	RISCV_PTR 0
SYM_DATA_END(test_kprobes_functions)