1
2
3
4
5
6
7
8
9
10
11#include <stdio.h>
12#include <unistd.h>
13#include <sys/syscall.h>
14#include <sys/time.h>
15#include <sys/types.h>
16#include <sys/wait.h>
17#include <stdlib.h>
18#include <string.h>
19#include <pthread.h>
20#include <altivec.h>
21
22#include "utils.h"
23
24
25#define ITERATIONS 10
26
27
28
29
30#define THREAD_FACTOR 8
31
32__thread vector int varray[] = {{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10,11,12},
33 {13,14,15,16},{17,18,19,20},{21,22,23,24},
34 {25,26,27,28},{29,30,31,32},{33,34,35,36},
35 {37,38,39,40},{41,42,43,44},{45,46,47,48}};
36
37bool bad_context;
38int running;
39int threads_starting;
40
41extern int preempt_vmx(vector int *varray, int *threads_starting, int *sentinal);
42
43void signal_vmx_sig(int sig, siginfo_t *info, void *context)
44{
45 int i;
46 ucontext_t *uc = context;
47 mcontext_t *mc = &uc->uc_mcontext;
48
49
50 for (i = 20; i < 32; i++) {
51 if (memcmp(mc->v_regs->vrregs[i], &varray[i - 20], 16)) {
52 int j;
53
54
55
56
57
58
59 printf("VMX mismatch at reg %d!\n", i);
60 printf("Reg | Actual | Expected\n");
61 for (j = 20; j < 32; j++) {
62 printf("%d | 0x%04x%04x%04x%04x | 0x%04x%04x%04x%04x\n", j, mc->v_regs->vrregs[j][0],
63 mc->v_regs->vrregs[j][1], mc->v_regs->vrregs[j][2], mc->v_regs->vrregs[j][3],
64 varray[j - 20][0], varray[j - 20][1], varray[j - 20][2], varray[j - 20][3]);
65 }
66 bad_context = true;
67 break;
68 }
69 }
70}
71
72void *signal_vmx_c(void *p)
73{
74 int i, j;
75 long rc;
76 struct sigaction act;
77 act.sa_sigaction = signal_vmx_sig;
78 act.sa_flags = SA_SIGINFO;
79 rc = sigaction(SIGUSR1, &act, NULL);
80 if (rc)
81 return p;
82
83 srand(pthread_self());
84 for (i = 0; i < 12; i++)
85 for (j = 0; j < 4; j++)
86 varray[i][j] = rand();
87
88 rc = preempt_vmx(varray, &threads_starting, &running);
89
90 return (void *) rc;
91}
92
93int test_signal_vmx(void)
94{
95 int i, j, rc, threads;
96 void *rc_p;
97 pthread_t *tids;
98
99 threads = sysconf(_SC_NPROCESSORS_ONLN) * THREAD_FACTOR;
100 tids = malloc(threads * sizeof(pthread_t));
101 FAIL_IF(!tids);
102
103 running = true;
104 threads_starting = threads;
105 for (i = 0; i < threads; i++) {
106 rc = pthread_create(&tids[i], NULL, signal_vmx_c, NULL);
107 FAIL_IF(rc);
108 }
109
110 setbuf(stdout, NULL);
111 printf("\tWaiting for %d workers to start... %d", threads, threads_starting);
112 while (threads_starting) {
113 asm volatile("": : :"memory");
114 usleep(1000);
115 printf(", %d", threads_starting);
116 }
117 printf(" ...done\n");
118
119 printf("\tSending signals to all threads %d times...", ITERATIONS);
120 for (i = 0; i < ITERATIONS; i++) {
121 for (j = 0; j < threads; j++) {
122 pthread_kill(tids[j], SIGUSR1);
123 }
124 sleep(1);
125 }
126 printf("done\n");
127
128 printf("\tKilling workers...");
129 running = 0;
130 for (i = 0; i < threads; i++) {
131 pthread_join(tids[i], &rc_p);
132
133
134
135
136
137 if ((long) rc_p || bad_context)
138 printf("oops\n");
139 if (bad_context)
140 fprintf(stderr, "\t!! bad_context is true\n");
141 FAIL_IF((long) rc_p || bad_context);
142 }
143 printf("done\n");
144
145 free(tids);
146 return 0;
147}
148
149int main(int argc, char *argv[])
150{
151 return test_harness(test_signal_vmx, "vmx_signal");
152}
153