1 /*
2 * Ptrace test for GPR/FPR registers in TM context
3 *
4 * Copyright (C) 2015 Anshuman Khandual, IBM Corporation.
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version
9 * 2 of the License, or (at your option) any later version.
10 */
11 #include "ptrace.h"
12 #include "ptrace-gpr.h"
13 #include "tm.h"
14
15 /* Tracer and Tracee Shared Data */
16 int shm_id;
17 unsigned long *cptr, *pptr;
18
19 float a = FPR_1;
20 float b = FPR_2;
21 float c = FPR_3;
22
tm_gpr(void)23 void tm_gpr(void)
24 {
25 unsigned long gpr_buf[18];
26 unsigned long result, texasr;
27 float fpr_buf[32];
28
29 printf("Starting the child\n");
30 cptr = (unsigned long *)shmat(shm_id, NULL, 0);
31
32 trans:
33 cptr[1] = 0;
34 asm __volatile__(
35 ASM_LOAD_GPR_IMMED(gpr_1)
36 ASM_LOAD_FPR_SINGLE_PRECISION(flt_1)
37 "1: ;"
38 "tbegin.;"
39 "beq 2f;"
40 ASM_LOAD_GPR_IMMED(gpr_2)
41 ASM_LOAD_FPR_SINGLE_PRECISION(flt_2)
42 "tsuspend.;"
43 "li 7, 1;"
44 "stw 7, 0(%[cptr1]);"
45 "tresume.;"
46 "b .;"
47
48 "tend.;"
49 "li 0, 0;"
50 "ori %[res], 0, 0;"
51 "b 3f;"
52
53 /* Transaction abort handler */
54 "2: ;"
55 "li 0, 1;"
56 "ori %[res], 0, 0;"
57 "mfspr %[texasr], %[sprn_texasr];"
58
59 "3: ;"
60 : [res] "=r" (result), [texasr] "=r" (texasr)
61 : [gpr_1]"i"(GPR_1), [gpr_2]"i"(GPR_2),
62 [sprn_texasr] "i" (SPRN_TEXASR), [flt_1] "r" (&a),
63 [flt_2] "r" (&b), [cptr1] "r" (&cptr[1])
64 : "memory", "r7", "r8", "r9", "r10",
65 "r11", "r12", "r13", "r14", "r15", "r16",
66 "r17", "r18", "r19", "r20", "r21", "r22",
67 "r23", "r24", "r25", "r26", "r27", "r28",
68 "r29", "r30", "r31"
69 );
70
71 if (result) {
72 if (!cptr[0])
73 goto trans;
74
75 shmdt((void *)cptr);
76 store_gpr(gpr_buf);
77 store_fpr_single_precision(fpr_buf);
78
79 if (validate_gpr(gpr_buf, GPR_3))
80 exit(1);
81
82 if (validate_fpr_float(fpr_buf, c))
83 exit(1);
84
85 exit(0);
86 }
87 shmdt((void *)cptr);
88 exit(1);
89 }
90
trace_tm_gpr(pid_t child)91 int trace_tm_gpr(pid_t child)
92 {
93 unsigned long gpr[18];
94 unsigned long fpr[32];
95
96 FAIL_IF(start_trace(child));
97 FAIL_IF(show_gpr(child, gpr));
98 FAIL_IF(validate_gpr(gpr, GPR_2));
99 FAIL_IF(show_fpr(child, fpr));
100 FAIL_IF(validate_fpr(fpr, FPR_2_REP));
101 FAIL_IF(show_ckpt_fpr(child, fpr));
102 FAIL_IF(validate_fpr(fpr, FPR_1_REP));
103 FAIL_IF(show_ckpt_gpr(child, gpr));
104 FAIL_IF(validate_gpr(gpr, GPR_1));
105 FAIL_IF(write_ckpt_gpr(child, GPR_3));
106 FAIL_IF(write_ckpt_fpr(child, FPR_3_REP));
107
108 pptr[0] = 1;
109 FAIL_IF(stop_trace(child));
110
111 return TEST_PASS;
112 }
113
ptrace_tm_gpr(void)114 int ptrace_tm_gpr(void)
115 {
116 pid_t pid;
117 int ret, status;
118
119 SKIP_IF(!have_htm());
120 shm_id = shmget(IPC_PRIVATE, sizeof(int) * 2, 0777|IPC_CREAT);
121 pid = fork();
122 if (pid < 0) {
123 perror("fork() failed");
124 return TEST_FAIL;
125 }
126 if (pid == 0)
127 tm_gpr();
128
129 if (pid) {
130 pptr = (unsigned long *)shmat(shm_id, NULL, 0);
131
132 while (!pptr[1])
133 asm volatile("" : : : "memory");
134 ret = trace_tm_gpr(pid);
135 if (ret) {
136 kill(pid, SIGTERM);
137 return TEST_FAIL;
138 }
139
140 shmdt((void *)pptr);
141
142 ret = wait(&status);
143 shmctl(shm_id, IPC_RMID, NULL);
144 if (ret != pid) {
145 printf("Child's exit status not captured\n");
146 return TEST_FAIL;
147 }
148
149 return (WIFEXITED(status) && WEXITSTATUS(status)) ? TEST_FAIL :
150 TEST_PASS;
151 }
152 return TEST_PASS;
153 }
154
main(int argc,char * argv[])155 int main(int argc, char *argv[])
156 {
157 return test_harness(ptrace_tm_gpr, "ptrace_tm_gpr");
158 }
159