12874c5fdSThomas Gleixner // SPDX-License-Identifier: GPL-2.0-or-later
2f666ad41SAnshuman Khandual /*
3f666ad41SAnshuman Khandual * Ptrace test for GPR/FPR registers
4f666ad41SAnshuman Khandual *
5f666ad41SAnshuman Khandual * Copyright (C) 2015 Anshuman Khandual, IBM Corporation.
6f666ad41SAnshuman Khandual */
7f666ad41SAnshuman Khandual #include "ptrace.h"
8f666ad41SAnshuman Khandual #include "ptrace-gpr.h"
9f666ad41SAnshuman Khandual #include "reg.h"
10c5a814ccSMichael Ellerman #include <time.h>
11f666ad41SAnshuman Khandual
12f666ad41SAnshuman Khandual /* Tracer and Tracee Shared Data */
13f666ad41SAnshuman Khandual int shm_id;
14f666ad41SAnshuman Khandual int *cptr, *pptr;
15f666ad41SAnshuman Khandual
16611e3850SMichael Ellerman extern void gpr_child_loop(int *read_flag, int *write_flag,
17611e3850SMichael Ellerman unsigned long *gpr_buf, double *fpr_buf);
18611e3850SMichael Ellerman
19c5a814ccSMichael Ellerman unsigned long child_gpr_val, parent_gpr_val;
20c5a814ccSMichael Ellerman double child_fpr_val, parent_fpr_val;
21c5a814ccSMichael Ellerman
child(void)227b1513d0SMichael Ellerman static int child(void)
23f666ad41SAnshuman Khandual {
24611e3850SMichael Ellerman unsigned long gpr_buf[32];
2553fa86e7SMichael Ellerman double fpr_buf[32];
26611e3850SMichael Ellerman int i;
27f666ad41SAnshuman Khandual
28f666ad41SAnshuman Khandual cptr = (int *)shmat(shm_id, NULL, 0);
29611e3850SMichael Ellerman memset(gpr_buf, 0, sizeof(gpr_buf));
30611e3850SMichael Ellerman memset(fpr_buf, 0, sizeof(fpr_buf));
31f666ad41SAnshuman Khandual
32611e3850SMichael Ellerman for (i = 0; i < 32; i++) {
33c5a814ccSMichael Ellerman gpr_buf[i] = child_gpr_val;
34c5a814ccSMichael Ellerman fpr_buf[i] = child_fpr_val;
35611e3850SMichael Ellerman }
36f666ad41SAnshuman Khandual
37611e3850SMichael Ellerman gpr_child_loop(&cptr[0], &cptr[1], gpr_buf, fpr_buf);
38f666ad41SAnshuman Khandual
39f666ad41SAnshuman Khandual shmdt((void *)cptr);
40f666ad41SAnshuman Khandual
41c5a814ccSMichael Ellerman FAIL_IF(validate_gpr(gpr_buf, parent_gpr_val));
42c5a814ccSMichael Ellerman FAIL_IF(validate_fpr_double(fpr_buf, parent_fpr_val));
43f666ad41SAnshuman Khandual
447b1513d0SMichael Ellerman return 0;
45f666ad41SAnshuman Khandual }
46f666ad41SAnshuman Khandual
trace_gpr(pid_t child)47f666ad41SAnshuman Khandual int trace_gpr(pid_t child)
48f666ad41SAnshuman Khandual {
49*6c9c7d8fSMichael Ellerman __u64 tmp, fpr[32], *peeked_fprs;
50f666ad41SAnshuman Khandual unsigned long gpr[18];
51f666ad41SAnshuman Khandual
52f666ad41SAnshuman Khandual FAIL_IF(start_trace(child));
53*6c9c7d8fSMichael Ellerman
54*6c9c7d8fSMichael Ellerman // Check child GPRs match what we expect using GETREGS
55f666ad41SAnshuman Khandual FAIL_IF(show_gpr(child, gpr));
56c5a814ccSMichael Ellerman FAIL_IF(validate_gpr(gpr, child_gpr_val));
57c5a814ccSMichael Ellerman
58*6c9c7d8fSMichael Ellerman // Check child FPRs match what we expect using GETFPREGS
59*6c9c7d8fSMichael Ellerman FAIL_IF(show_fpr(child, fpr));
60c5a814ccSMichael Ellerman memcpy(&tmp, &child_fpr_val, sizeof(tmp));
61c5a814ccSMichael Ellerman FAIL_IF(validate_fpr(fpr, tmp));
62c5a814ccSMichael Ellerman
63*6c9c7d8fSMichael Ellerman // Check child FPRs match what we expect using PEEKUSR
64*6c9c7d8fSMichael Ellerman peeked_fprs = peek_fprs(child);
65*6c9c7d8fSMichael Ellerman FAIL_IF(!peeked_fprs);
66*6c9c7d8fSMichael Ellerman FAIL_IF(validate_fpr(peeked_fprs, tmp));
67*6c9c7d8fSMichael Ellerman free(peeked_fprs);
68*6c9c7d8fSMichael Ellerman
69*6c9c7d8fSMichael Ellerman // Write child GPRs using SETREGS
70c5a814ccSMichael Ellerman FAIL_IF(write_gpr(child, parent_gpr_val));
71c5a814ccSMichael Ellerman
72*6c9c7d8fSMichael Ellerman // Write child FPRs using SETFPREGS
73c5a814ccSMichael Ellerman memcpy(&tmp, &parent_fpr_val, sizeof(tmp));
74c5a814ccSMichael Ellerman FAIL_IF(write_fpr(child, tmp));
75c5a814ccSMichael Ellerman
76*6c9c7d8fSMichael Ellerman // Check child FPRs match what we just set, using PEEKUSR
77*6c9c7d8fSMichael Ellerman peeked_fprs = peek_fprs(child);
78*6c9c7d8fSMichael Ellerman FAIL_IF(!peeked_fprs);
79*6c9c7d8fSMichael Ellerman FAIL_IF(validate_fpr(peeked_fprs, tmp));
80*6c9c7d8fSMichael Ellerman
81*6c9c7d8fSMichael Ellerman // Write child FPRs using POKEUSR
82*6c9c7d8fSMichael Ellerman FAIL_IF(poke_fprs(child, (unsigned long *)peeked_fprs));
83*6c9c7d8fSMichael Ellerman
84*6c9c7d8fSMichael Ellerman // Child will check its FPRs match before exiting
85f666ad41SAnshuman Khandual FAIL_IF(stop_trace(child));
86f666ad41SAnshuman Khandual
87f666ad41SAnshuman Khandual return TEST_PASS;
88f666ad41SAnshuman Khandual }
89f666ad41SAnshuman Khandual
90c5a814ccSMichael Ellerman #ifndef __LONG_WIDTH__
91c5a814ccSMichael Ellerman #define __LONG_WIDTH__ (sizeof(long) * 8)
92c5a814ccSMichael Ellerman #endif
93c5a814ccSMichael Ellerman
rand_reg(void)94c5a814ccSMichael Ellerman static uint64_t rand_reg(void)
95c5a814ccSMichael Ellerman {
96c5a814ccSMichael Ellerman uint64_t result;
97c5a814ccSMichael Ellerman long r;
98c5a814ccSMichael Ellerman
99c5a814ccSMichael Ellerman r = random();
100c5a814ccSMichael Ellerman
101c5a814ccSMichael Ellerman // Small values are typical
102c5a814ccSMichael Ellerman result = r & 0xffff;
103c5a814ccSMichael Ellerman if (r & 0x10000)
104c5a814ccSMichael Ellerman return result;
105c5a814ccSMichael Ellerman
106c5a814ccSMichael Ellerman // Pointers tend to have high bits set
107c5a814ccSMichael Ellerman result |= random() << (__LONG_WIDTH__ - 31);
108c5a814ccSMichael Ellerman if (r & 0x100000)
109c5a814ccSMichael Ellerman return result;
110c5a814ccSMichael Ellerman
111c5a814ccSMichael Ellerman // And sometimes we want a full 64-bit value
112c5a814ccSMichael Ellerman result ^= random() << 16;
113c5a814ccSMichael Ellerman
114c5a814ccSMichael Ellerman return result;
115c5a814ccSMichael Ellerman }
116c5a814ccSMichael Ellerman
ptrace_gpr(void)117f666ad41SAnshuman Khandual int ptrace_gpr(void)
118f666ad41SAnshuman Khandual {
119c5a814ccSMichael Ellerman unsigned long seed;
120f666ad41SAnshuman Khandual int ret, status;
121c5a814ccSMichael Ellerman pid_t pid;
122c5a814ccSMichael Ellerman
123c5a814ccSMichael Ellerman seed = getpid() ^ time(NULL);
124c5a814ccSMichael Ellerman printf("srand(%lu)\n", seed);
125c5a814ccSMichael Ellerman srand(seed);
126c5a814ccSMichael Ellerman
127c5a814ccSMichael Ellerman child_gpr_val = rand_reg();
128c5a814ccSMichael Ellerman child_fpr_val = rand_reg();
129c5a814ccSMichael Ellerman parent_gpr_val = rand_reg();
130c5a814ccSMichael Ellerman parent_fpr_val = rand_reg();
131f666ad41SAnshuman Khandual
132f666ad41SAnshuman Khandual shm_id = shmget(IPC_PRIVATE, sizeof(int) * 2, 0777|IPC_CREAT);
133f666ad41SAnshuman Khandual pid = fork();
134f666ad41SAnshuman Khandual if (pid < 0) {
135f666ad41SAnshuman Khandual perror("fork() failed");
136f666ad41SAnshuman Khandual return TEST_FAIL;
137f666ad41SAnshuman Khandual }
138f666ad41SAnshuman Khandual if (pid == 0)
1397b1513d0SMichael Ellerman exit(child());
140f666ad41SAnshuman Khandual
141f666ad41SAnshuman Khandual if (pid) {
142f666ad41SAnshuman Khandual pptr = (int *)shmat(shm_id, NULL, 0);
143f666ad41SAnshuman Khandual while (!pptr[1])
144f666ad41SAnshuman Khandual asm volatile("" : : : "memory");
145f666ad41SAnshuman Khandual
146f666ad41SAnshuman Khandual ret = trace_gpr(pid);
147f666ad41SAnshuman Khandual if (ret) {
148f666ad41SAnshuman Khandual kill(pid, SIGTERM);
149f666ad41SAnshuman Khandual shmdt((void *)pptr);
150f666ad41SAnshuman Khandual shmctl(shm_id, IPC_RMID, NULL);
151f666ad41SAnshuman Khandual return TEST_FAIL;
152f666ad41SAnshuman Khandual }
153f666ad41SAnshuman Khandual
154f666ad41SAnshuman Khandual pptr[0] = 1;
155f666ad41SAnshuman Khandual shmdt((void *)pptr);
156f666ad41SAnshuman Khandual
157f666ad41SAnshuman Khandual ret = wait(&status);
158f666ad41SAnshuman Khandual shmctl(shm_id, IPC_RMID, NULL);
159f666ad41SAnshuman Khandual if (ret != pid) {
160f666ad41SAnshuman Khandual printf("Child's exit status not captured\n");
161f666ad41SAnshuman Khandual return TEST_FAIL;
162f666ad41SAnshuman Khandual }
163f666ad41SAnshuman Khandual
164f666ad41SAnshuman Khandual return (WIFEXITED(status) && WEXITSTATUS(status)) ? TEST_FAIL :
165f666ad41SAnshuman Khandual TEST_PASS;
166f666ad41SAnshuman Khandual }
167f666ad41SAnshuman Khandual
168f666ad41SAnshuman Khandual return TEST_PASS;
169f666ad41SAnshuman Khandual }
170f666ad41SAnshuman Khandual
main(int argc,char * argv[])171f666ad41SAnshuman Khandual int main(int argc, char *argv[])
172f666ad41SAnshuman Khandual {
173f666ad41SAnshuman Khandual return test_harness(ptrace_gpr, "ptrace_gpr");
174f666ad41SAnshuman Khandual }
175