1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * Copyright 2015, Cyril Bur, IBM Corp.
4  *
5  * This test attempts to see if the VMX registers change across a syscall (fork).
6  */
7 
8 #include <altivec.h>
9 #include <stdio.h>
10 #include <unistd.h>
11 #include <sys/syscall.h>
12 #include <sys/time.h>
13 #include <stdlib.h>
14 #include <sys/types.h>
15 #include <sys/wait.h>
16 #include "utils.h"
17 
18 vector int varray[] = {{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10,11,12},
19 	{13,14,15,16},{17,18,19,20},{21,22,23,24},
20 	{25,26,27,28},{29,30,31,32},{33,34,35,36},
21 	{37,38,39,40},{41,42,43,44},{45,46,47,48}};
22 
23 extern int test_vmx(vector int *varray, pid_t *pid);
24 
25 int vmx_syscall(void)
26 {
27 	pid_t fork_pid;
28 	int i;
29 	int ret;
30 	int child_ret;
31 	for (i = 0; i < 1000; i++) {
32 		/* test_vmx will fork() */
33 		ret = test_vmx(varray, &fork_pid);
34 		if (fork_pid == -1)
35 			return -1;
36 		if (fork_pid == 0)
37 			exit(ret);
38 		waitpid(fork_pid, &child_ret, 0);
39 		if (ret || child_ret)
40 			return 1;
41 	}
42 
43 	return 0;
44 }
45 
46 int test_vmx_syscall(void)
47 {
48 	/*
49 	 * Setup an environment with much context switching
50 	 */
51 	pid_t pid2;
52 	pid_t pid;
53 	int ret;
54 	int child_ret;
55 
56 	// vcmpequd used in vmx_asm.S is v2.07
57 	SKIP_IF(!have_hwcap2(PPC_FEATURE2_ARCH_2_07));
58 
59 	pid = fork();
60 	FAIL_IF(pid == -1);
61 
62 	pid2 = fork();
63 	ret = vmx_syscall();
64 	/* Can't FAIL_IF(pid2 == -1); because we've already forked */
65 	if (pid2 == -1) {
66 		/*
67 		 * Couldn't fork, ensure child_ret is set and is a fail
68 		 */
69 		ret = child_ret = 1;
70 	} else {
71 		if (pid2)
72 			waitpid(pid2, &child_ret, 0);
73 		else
74 			exit(ret);
75 	}
76 
77 	ret |= child_ret;
78 
79 	if (pid)
80 		waitpid(pid, &child_ret, 0);
81 	else
82 		exit(ret);
83 
84 	FAIL_IF(ret || child_ret);
85 	return 0;
86 }
87 
88 int main(int argc, char *argv[])
89 {
90 	return test_harness(test_vmx_syscall, "vmx_syscall");
91 
92 }
93