1// RUN: %clangxx -O0 %s -o %t && %run %t
2
3#include <assert.h>
4#include <signal.h>
5#include <stdio.h>
6#include <sys/ptrace.h>
7#include <sys/types.h>
8#include <sys/user.h>
9#include <sys/wait.h>
10#include <sys/uio.h>
11#include <unistd.h>
12#include <elf.h>
13#if __mips64 || __arm__
14 #include <asm/ptrace.h>
15 #include <sys/procfs.h>
16#endif
17#ifdef __aarch64__
18// GLIBC 2.20+ sys/user does not include asm/ptrace.h
19 #include <asm/ptrace.h>
20#endif
21
22int main(void) {
23  pid_t pid;
24  pid = fork();
25  if (pid == 0) { // child
26    ptrace(PTRACE_TRACEME, 0, NULL, NULL);
27    execl("/bin/true", "true", NULL);
28  } else {
29    wait(NULL);
30    int res;
31
32#if __x86_64__
33    user_regs_struct regs;
34    res = ptrace(PTRACE_GETREGS, pid, NULL, &regs);
35    assert(!res);
36    if (regs.rip)
37      printf("%zx\n", regs.rip);
38
39    user_fpregs_struct fpregs;
40    res = ptrace(PTRACE_GETFPREGS, pid, NULL, &fpregs);
41    assert(!res);
42    if (fpregs.mxcsr)
43      printf("%x\n", fpregs.mxcsr);
44#endif // __x86_64__
45
46#if (__powerpc64__ || __mips64 || __arm__)
47    struct pt_regs regs;
48    res = ptrace((enum __ptrace_request)PTRACE_GETREGS, pid, NULL, &regs);
49    assert(!res);
50#if (__powerpc64__)
51    if (regs.nip)
52      printf("%lx\n", regs.nip);
53#elif (__mips64)
54    if (regs.cp0_epc)
55    printf("%lx\n", regs.cp0_epc);
56#elif (__arm__)
57    if (regs.ARM_pc)
58    printf("%lx\n", regs.ARM_pc);
59#endif
60#if (__powerpc64 || __mips64)
61    elf_fpregset_t fpregs;
62    res = ptrace((enum __ptrace_request)PTRACE_GETFPREGS, pid, NULL, &fpregs);
63    assert(!res);
64    if ((elf_greg_t)fpregs[32]) // fpscr
65      printf("%lx\n", (elf_greg_t)fpregs[32]);
66#elif (__arm__)
67    char regbuf[ARM_VFPREGS_SIZE];
68    res = ptrace((enum __ptrace_request)PTRACE_GETVFPREGS, pid, 0, regbuf);
69    assert(!res);
70    unsigned fpscr = *(unsigned*)(regbuf + (32 * 8));
71    printf ("%x\n", fpscr);
72#endif
73#endif // (__powerpc64__ || __mips64 || __arm__)
74
75#if (__aarch64__)
76    struct iovec regset_io;
77
78    struct user_pt_regs regs;
79    regset_io.iov_base = &regs;
80    regset_io.iov_len = sizeof(regs);
81    res = ptrace(PTRACE_GETREGSET, pid, (void*)NT_PRSTATUS, (void*)&regset_io);
82    assert(!res);
83    if (regs.pc)
84      printf("%llx\n", regs.pc);
85
86    struct user_fpsimd_state fpregs;
87    regset_io.iov_base = &fpregs;
88    regset_io.iov_len = sizeof(fpregs);
89    res = ptrace(PTRACE_GETREGSET, pid, (void*)NT_FPREGSET, (void*)&regset_io);
90    assert(!res);
91    if (fpregs.fpsr)
92      printf("%x\n", fpregs.fpsr);
93#endif // (__aarch64__)
94
95    siginfo_t siginfo;
96    res = ptrace(PTRACE_GETSIGINFO, pid, NULL, &siginfo);
97    assert(!res);
98    assert(siginfo.si_pid == pid);
99
100    ptrace(PTRACE_CONT, pid, NULL, NULL);
101
102    wait(NULL);
103  }
104  return 0;
105}
106