tools/testing/selftests/powerpc/math/vmx_syscall.c

   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
  22 vector 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
  27 extern int test_vmx(vector int *varray, pid_t *pid);
  28
  29 int 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
  50 int 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
  87 int main(int argc, char *argv[])
  88 {
  89         return test_harness(test_vmx_syscall, "vmx_syscall");
  90
  91 }