1// RUN: %clangxx -O0 %s -o %t && %run %t
2
3// UNSUPPORTED: android
4
5#include <assert.h>
6#include <elf.h>
7#include <signal.h>
8#include <stdio.h>
9#include <string.h>
10#include <sys/ptrace.h>
11#include <sys/types.h>
12#include <sys/uio.h>
13#include <sys/user.h>
14#include <sys/wait.h>
15#include <unistd.h>
16#if __mips64 || __arm__
17 #include <asm/ptrace.h>
18 #include <sys/procfs.h>
19#endif
20#if defined(__aarch64__) || defined(__loongarch__)
21// GLIBC 2.20+ sys/user does not include asm/ptrace.h
22 #include <asm/ptrace.h>
23#endif
24
25int main(void) {
26 pid_t pid;
27 pid = fork();
28 if (pid == 0) { // child
29 ptrace(request: PTRACE_TRACEME, 0, NULL, NULL);
30 execl(path: "/bin/true", arg: "true", NULL);
31 } else {
32 wait(NULL);
33 int res;
34
35#if __x86_64__
36 user_regs_struct regs;
37 res = ptrace(request: PTRACE_GETREGS, pid, NULL, &regs);
38 assert(!res);
39 if (regs.rip)
40 printf(format: "%zx\n", regs.rip);
41
42 user_fpregs_struct fpregs;
43 res = ptrace(request: PTRACE_GETFPREGS, pid, NULL, &fpregs);
44 assert(!res);
45 if (fpregs.mxcsr)
46 printf(format: "%x\n", fpregs.mxcsr);
47#endif // __x86_64__
48
49#if (__powerpc64__ || __mips64 || __arm__)
50 // Check that nothing writes out-of-bounds.
51 struct pt_regs regs_buf[4];
52 memset(&regs_buf, 0xcd, sizeof(regs_buf));
53 struct pt_regs &regs = regs_buf[1];
54 res = ptrace((enum __ptrace_request)PTRACE_GETREGS, pid, NULL, &regs);
55 assert(!res);
56 assert(memcmp(&regs_buf[0], &regs_buf[3], sizeof(regs_buf[3])) == 0);
57 assert(memcmp(&regs_buf[2], &regs_buf[3], sizeof(regs_buf[3])) == 0);
58#if (__powerpc64__)
59 if (regs.nip)
60 printf("%lx\n", regs.nip);
61#elif (__mips64)
62 if (regs.cp0_epc)
63 printf("%lx\n", regs.cp0_epc);
64#elif (__arm__)
65 if (regs.ARM_pc)
66 printf("%lx\n", regs.ARM_pc);
67#endif
68#if (__powerpc64 || __mips64)
69 elf_fpregset_t fpregs;
70 res = ptrace((enum __ptrace_request)PTRACE_GETFPREGS, pid, NULL, &fpregs);
71 assert(!res);
72 if ((elf_greg_t)fpregs[32]) // fpscr
73 printf("%lx\n", (elf_greg_t)fpregs[32]);
74#elif (__arm__)
75 char regbuf[ARM_VFPREGS_SIZE];
76 res = ptrace((enum __ptrace_request)PTRACE_GETVFPREGS, pid, 0, regbuf);
77 assert(!res);
78 unsigned fpscr = *(unsigned*)(regbuf + (32 * 8));
79 printf ("%x\n", fpscr);
80#endif
81#endif // (__powerpc64__ || __mips64 || __arm__)
82
83#if (__aarch64__)
84 struct iovec regset_io;
85
86 struct user_pt_regs regs;
87 regset_io.iov_base = &regs;
88 regset_io.iov_len = sizeof(regs);
89 res = ptrace(PTRACE_GETREGSET, pid, (void*)NT_PRSTATUS, (void*)&regset_io);
90 assert(!res);
91 if (regs.pc)
92 printf("%llx\n", regs.pc);
93
94 struct user_fpsimd_state fpregs;
95 regset_io.iov_base = &fpregs;
96 regset_io.iov_len = sizeof(fpregs);
97 res = ptrace(PTRACE_GETREGSET, pid, (void*)NT_FPREGSET, (void*)&regset_io);
98 assert(!res);
99 if (fpregs.fpsr)
100 printf("%x\n", fpregs.fpsr);
101#endif // (__aarch64__)
102
103#if (__loongarch__)
104 struct iovec regset_io;
105
106 struct user_pt_regs regs;
107 regset_io.iov_base = &regs;
108 regset_io.iov_len = sizeof(regs);
109 res =
110 ptrace(PTRACE_GETREGSET, pid, (void *)NT_PRSTATUS, (void *)&regset_io);
111 assert(!res);
112 if (regs.csr_era)
113 printf("%lx\n", regs.csr_era);
114
115 struct user_fp_state fpregs;
116 regset_io.iov_base = &fpregs;
117 regset_io.iov_len = sizeof(fpregs);
118 res =
119 ptrace(PTRACE_GETREGSET, pid, (void *)NT_FPREGSET, (void *)&regset_io);
120 assert(!res);
121 if (fpregs.fcsr)
122 printf("%x\n", fpregs.fcsr);
123#endif // (__loongarch__)
124
125#if (__s390__)
126 struct iovec regset_io;
127
128 struct _user_regs_struct regs;
129 regset_io.iov_base = &regs;
130 regset_io.iov_len = sizeof(regs);
131 res = ptrace(PTRACE_GETREGSET, pid, (void*)NT_PRSTATUS, (void*)&regset_io);
132 assert(!res);
133 if (regs.psw.addr)
134 printf("%lx\n", regs.psw.addr);
135
136 struct _user_fpregs_struct fpregs;
137 regset_io.iov_base = &fpregs;
138 regset_io.iov_len = sizeof(fpregs);
139 res = ptrace(PTRACE_GETREGSET, pid, (void*)NT_FPREGSET, (void*)&regset_io);
140 assert(!res);
141 if (fpregs.fpc)
142 printf("%x\n", fpregs.fpc);
143#endif // (__s390__)
144
145 siginfo_t siginfo;
146 res = ptrace(request: PTRACE_GETSIGINFO, pid, NULL, &siginfo);
147 assert(!res);
148 assert(siginfo.si_pid == pid);
149
150 ptrace(request: PTRACE_CONT, pid, NULL, NULL);
151
152 wait(NULL);
153 }
154 return 0;
155}
156

source code of compiler-rt/test/sanitizer_common/TestCases/Linux/ptrace.cpp