1 // SPDX-License-Identifier: GPL-2.0-only 2 3 #include <linux/wait.h> 4 5 #define THIS_PROGRAM "./vstate_exec_nolibc" 6 7 int main(int argc, char **argv) 8 { 9 int rc, pid, status, test_inherit = 0, xtheadvector = 0; 10 long ctrl, ctrl_c; 11 char *exec_argv[2], *exec_envp[2]; 12 13 if (argc > 1 && strcmp(argv[1], "x")) 14 test_inherit = 1; 15 16 if (argc > 2 && strcmp(argv[2], "x")) 17 xtheadvector = 1; 18 19 ctrl = my_syscall1(__NR_prctl, PR_RISCV_V_GET_CONTROL); 20 if (ctrl < 0) { 21 puts("PR_RISCV_V_GET_CONTROL is not supported\n"); 22 return ctrl; 23 } 24 25 if (test_inherit) { 26 pid = fork(); 27 if (pid == -1) { 28 puts("fork failed\n"); 29 exit(-1); 30 } 31 32 /* child */ 33 if (!pid) { 34 exec_argv[0] = THIS_PROGRAM; 35 exec_argv[1] = NULL; 36 exec_envp[0] = NULL; 37 exec_envp[1] = NULL; 38 /* launch the program again to check inherit */ 39 rc = execve(THIS_PROGRAM, exec_argv, exec_envp); 40 if (rc) { 41 puts("child execve failed\n"); 42 exit(-1); 43 } 44 } 45 46 } else { 47 pid = fork(); 48 if (pid == -1) { 49 puts("fork failed\n"); 50 exit(-1); 51 } 52 53 if (!pid) { 54 rc = my_syscall1(__NR_prctl, PR_RISCV_V_GET_CONTROL); 55 if (rc != ctrl) { 56 puts("child's vstate_ctrl not equal to parent's\n"); 57 exit(-1); 58 } 59 if (xtheadvector) 60 asm volatile (".4byte 0x00007ed7"); 61 else 62 asm volatile (".option push\n\t" 63 ".option arch, +v\n\t" 64 "vsetvli x0, x0, e32, m8, ta, ma\n\t" 65 ".option pop\n\t" 66 ); 67 exit(ctrl); 68 } 69 } 70 71 rc = waitpid(-1, &status, 0); 72 73 if (WIFEXITED(status) && WEXITSTATUS(status) == -1) { 74 puts("child exited abnormally\n"); 75 exit(-1); 76 } 77 78 if (WIFSIGNALED(status)) { 79 if (WTERMSIG(status) != SIGILL) { 80 puts("child was terminated by unexpected signal\n"); 81 exit(-1); 82 } 83 84 if ((ctrl & PR_RISCV_V_VSTATE_CTRL_CUR_MASK) != PR_RISCV_V_VSTATE_CTRL_OFF) { 85 puts("child signaled by illegal V access but vstate_ctrl is not off\n"); 86 exit(-1); 87 } 88 89 /* child terminated, and its vstate_ctrl is off */ 90 exit(ctrl); 91 } 92 93 ctrl_c = WEXITSTATUS(status); 94 if (test_inherit) { 95 if (ctrl & PR_RISCV_V_VSTATE_CTRL_INHERIT) { 96 if (!(ctrl_c & PR_RISCV_V_VSTATE_CTRL_INHERIT)) { 97 puts("parent has inherit bit, but child has not\n"); 98 exit(-1); 99 } 100 } 101 rc = (ctrl & PR_RISCV_V_VSTATE_CTRL_NEXT_MASK) >> 2; 102 if (rc != PR_RISCV_V_VSTATE_CTRL_DEFAULT) { 103 if (rc != (ctrl_c & PR_RISCV_V_VSTATE_CTRL_CUR_MASK)) { 104 puts("parent's next setting does not equal to child's\n"); 105 exit(-1); 106 } 107 108 if (!(ctrl & PR_RISCV_V_VSTATE_CTRL_INHERIT)) { 109 if ((ctrl_c & PR_RISCV_V_VSTATE_CTRL_NEXT_MASK) != 110 PR_RISCV_V_VSTATE_CTRL_DEFAULT) { 111 puts("must clear child's next vstate_ctrl if !inherit\n"); 112 exit(-1); 113 } 114 } 115 } 116 } 117 return ctrl; 118 } 119