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 #ifdef __aarch64__
21 // GLIBC 2.20+ sys/user does not include asm/ptrace.h
22  #include <asm/ptrace.h>
23 #endif
24 
25 int main(void) {
26   pid_t pid;
27   pid = fork();
28   if (pid == 0) { // child
29     ptrace(PTRACE_TRACEME, 0, NULL, NULL);
30     execl("/bin/true", "true", NULL);
31   } else {
32     wait(NULL);
33     int res;
34 
35 #if __x86_64__
36     user_regs_struct regs;
37     res = ptrace(PTRACE_GETREGS, pid, NULL, &regs);
38     assert(!res);
39     if (regs.rip)
40       printf("%zx\n", regs.rip);
41 
42     user_fpregs_struct fpregs;
43     res = ptrace(PTRACE_GETFPREGS, pid, NULL, &fpregs);
44     assert(!res);
45     if (fpregs.mxcsr)
46       printf("%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 (__s390__)
104     struct iovec regset_io;
105 
106     struct _user_regs_struct regs;
107     regset_io.iov_base = &regs;
108     regset_io.iov_len = sizeof(regs);
109     res = ptrace(PTRACE_GETREGSET, pid, (void*)NT_PRSTATUS, (void*)&regset_io);
110     assert(!res);
111     if (regs.psw.addr)
112       printf("%lx\n", regs.psw.addr);
113 
114     struct _user_fpregs_struct fpregs;
115     regset_io.iov_base = &fpregs;
116     regset_io.iov_len = sizeof(fpregs);
117     res = ptrace(PTRACE_GETREGSET, pid, (void*)NT_FPREGSET, (void*)&regset_io);
118     assert(!res);
119     if (fpregs.fpc)
120       printf("%x\n", fpregs.fpc);
121 #endif // (__s390__)
122 
123     siginfo_t siginfo;
124     res = ptrace(PTRACE_GETSIGINFO, pid, NULL, &siginfo);
125     assert(!res);
126     assert(siginfo.si_pid == pid);
127 
128     ptrace(PTRACE_CONT, pid, NULL, NULL);
129 
130     wait(NULL);
131   }
132   return 0;
133 }
134