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