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

#ifndef SELFTEST_RISCV_CFI_H
#define SELFTEST_RISCV_CFI_H
#include <stddef.h>
#include <sys/types.h>
#include "shadowstack.h"

#define CHILD_EXIT_CODE_SSWRITE		10
#define CHILD_EXIT_CODE_SIG_TEST	11

#define my_syscall5(num, arg1, arg2, arg3, arg4, arg5)			\
({									\
	register long _num  __asm__ ("a7") = (num);			\
	register long _arg1 __asm__ ("a0") = (long)(arg1);		\
	register long _arg2 __asm__ ("a1") = (long)(arg2);		\
	register long _arg3 __asm__ ("a2") = (long)(arg3);		\
	register long _arg4 __asm__ ("a3") = (long)(arg4);		\
	register long _arg5 __asm__ ("a4") = (long)(arg5);		\
									\
	__asm__ volatile(						\
		"ecall\n"						\
		: "+r"							\
		(_arg1)							\
		: "r"(_arg2), "r"(_arg3), "r"(_arg4), "r"(_arg5),	\
		  "r"(_num)						\
		: "memory", "cc"					\
	);								\
	_arg1;								\
})

#define my_syscall3(num, arg1, arg2, arg3)				\
({									\
	register long _num  __asm__ ("a7") = (num);			\
	register long _arg1 __asm__ ("a0") = (long)(arg1);		\
	register long _arg2 __asm__ ("a1") = (long)(arg2);		\
	register long _arg3 __asm__ ("a2") = (long)(arg3);		\
									\
	__asm__ volatile(						\
		"ecall\n"						\
		: "+r" (_arg1)						\
		: "r"(_arg2), "r"(_arg3),				\
		  "r"(_num)						\
		: "memory", "cc"					\
	);								\
	_arg1;								\
})

#ifndef __NR_prctl
#define __NR_prctl 167
#endif

#ifndef __NR_map_shadow_stack
#define __NR_map_shadow_stack 453
#endif

#define CSR_SSP 0x011

#ifdef __ASSEMBLY__
#define __ASM_STR(x)    x
#else
#define __ASM_STR(x)    #x
#endif

#define csr_read(csr)							\
({									\
	register unsigned long __v;					\
	__asm__ __volatile__ ("csrr %0, " __ASM_STR(csr)		\
				: "=r" (__v) :				\
				: "memory");				\
	__v;								\
})

#define csr_write(csr, val)						\
({									\
	unsigned long __v = (unsigned long)(val);			\
	__asm__ __volatile__ ("csrw " __ASM_STR(csr) ", %0"		\
				: : "rK" (__v)				\
				: "memory");				\
})

#endif