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