83 lines
1.9 KiB
C
83 lines
1.9 KiB
C
/* 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
|