1 // FIXME: https://code.google.com/p/address-sanitizer/issues/detail?id=316
2 // XFAIL: android
3 // XFAIL: mips
4 //
5 // RUN: %clangxx_asan -O0 %s -o %t && %run %t
6 // RUN: %clangxx_asan -DPOSITIVE -O0 %s -o %t && not %run %t 2>&1 | FileCheck %s
7
8 #include <assert.h>
9 #include <stdio.h>
10 #include <sys/ptrace.h>
11 #include <sys/types.h>
12 #include <sys/user.h>
13 #include <sys/wait.h>
14 #include <unistd.h>
15 #include <sys/uio.h> // for iovec
16 #include <elf.h> // for NT_PRSTATUS
17 #ifdef __aarch64__
18 # include <asm/ptrace.h>
19 #endif
20
21 #if defined(__i386__) || defined(__x86_64__)
22 typedef user_regs_struct regs_struct;
23 typedef user_fpregs_struct fpregs_struct;
24 #if defined(__i386__)
25 #define REG_IP eip
26 #else
27 #define REG_IP rip
28 #endif
29 #define PRINT_REG_PC(__regs) printf ("%lx\n", (unsigned long) (__regs.REG_IP))
30 #define PRINT_REG_FP(__fpregs) printf ("%lx\n", (unsigned long) (__fpregs.cwd))
31 #define __PTRACE_FPREQUEST PTRACE_GETFPREGS
32
33 #elif defined(__aarch64__)
34 typedef struct user_pt_regs regs_struct;
35 typedef struct user_fpsimd_state fpregs_struct;
36 #define PRINT_REG_PC(__regs) printf ("%x\n", (unsigned) (__regs.pc))
37 #define PRINT_REG_FP(__fpregs) printf ("%x\n", (unsigned) (__fpregs.fpsr))
38 #define ARCH_IOVEC_FOR_GETREGSET
39
40 #elif defined(__powerpc64__)
41 typedef struct pt_regs regs_struct;
42 typedef elf_fpregset_t fpregs_struct;
43 #define PRINT_REG_PC(__regs) printf ("%lx\n", (unsigned long) (__regs.nip))
44 #define PRINT_REG_FP(__fpregs) printf ("%lx\n", (elf_greg_t)fpregs[32])
45 #define ARCH_IOVEC_FOR_GETREGSET
46
47 #elif defined(__mips__)
48 typedef struct pt_regs regs_struct;
49 typedef elf_fpregset_t fpregs_struct;
50 #define PRINT_REG_PC(__regs) printf ("%lx\n", (unsigned long) (__regs.cp0_epc))
51 #define PRINT_REG_FP(__fpregs) printf ("%lx\n", (elf_greg_t) (__fpregs[32]))
52 #define __PTRACE_FPREQUEST PTRACE_GETFPREGS
53
54 #elif defined(__arm__)
55 # include <asm/ptrace.h>
56 # include <sys/procfs.h>
57 typedef struct pt_regs regs_struct;
58 typedef char fpregs_struct[ARM_VFPREGS_SIZE];
59 #define PRINT_REG_PC(__regs) printf ("%x\n", (unsigned) (__regs.ARM_pc))
60 #define PRINT_REG_FP(__fpregs) printf ("%x\n", (unsigned) (__fpregs + 32 * 8))
61 #define __PTRACE_FPREQUEST PTRACE_GETVFPREGS
62
63 #elif defined(__s390__)
64 typedef _user_regs_struct regs_struct;
65 typedef _user_fpregs_struct fpregs_struct;
66 #define PRINT_REG_PC(__regs) printf ("%lx\n", (unsigned long) (__regs.psw.addr))
67 #define PRINT_REG_FP(__fpregs) printf ("%lx\n", (unsigned long) (__fpregs.fpc))
68 #define ARCH_IOVEC_FOR_GETREGSET
69
70 #elif defined(__riscv) && (__riscv_xlen == 64)
71 #include <asm/ptrace.h>
72 typedef user_regs_struct regs_struct;
73 typedef __riscv_q_ext_state fpregs_struct;
74 #define PRINT_REG_PC(__regs) printf("%lx\n", (unsigned long)(__regs.pc))
75 #define PRINT_REG_FP(__fpregs) printf("%lx\n", (unsigned long)(__fpregs.fcsr))
76 #define ARCH_IOVEC_FOR_GETREGSET
77 #endif
78
79
main(void)80 int main(void) {
81 pid_t pid;
82 pid = fork();
83 if (pid == 0) { // child
84 ptrace(PTRACE_TRACEME, 0, NULL, NULL);
85 execl("/bin/true", "true", NULL);
86 } else {
87 wait(NULL);
88 regs_struct regs;
89 regs_struct* volatile pregs = ®s;
90 #ifdef ARCH_IOVEC_FOR_GETREGSET
91 struct iovec regset_io;
92 #endif
93 int res;
94
95 #ifdef POSITIVE
96 ++pregs;
97 #endif
98
99 #ifdef ARCH_IOVEC_FOR_GETREGSET
100 # define __PTRACE_REQUEST PTRACE_GETREGSET
101 # define __PTRACE_ARGS (void*)NT_PRSTATUS, (void*)®set_io
102 regset_io.iov_base = pregs;
103 regset_io.iov_len = sizeof(regs_struct);
104 #else
105 # define __PTRACE_REQUEST PTRACE_GETREGS
106 # define __PTRACE_ARGS NULL, pregs
107 #endif
108 res = ptrace((enum __ptrace_request)__PTRACE_REQUEST, pid, __PTRACE_ARGS);
109 // CHECK: AddressSanitizer: stack-buffer-overflow
110 // CHECK: {{.*ptrace.cpp:}}[[@LINE-2]]
111 assert(!res);
112 PRINT_REG_PC(regs);
113
114 fpregs_struct fpregs;
115 #ifdef ARCH_IOVEC_FOR_GETREGSET
116 # define __PTRACE_FPREQUEST PTRACE_GETREGSET
117 # define __PTRACE_FPARGS (void*)NT_PRSTATUS, (void*)®set_io
118 regset_io.iov_base = &fpregs;
119 regset_io.iov_len = sizeof(fpregs_struct);
120 res = ptrace((enum __ptrace_request)PTRACE_GETREGSET, pid, (void*)NT_FPREGSET,
121 (void*)®set_io);
122 #else
123 # define __PTRACE_FPARGS NULL, &fpregs
124 #endif
125 res = ptrace((enum __ptrace_request)__PTRACE_FPREQUEST, pid, __PTRACE_FPARGS);
126 assert(!res);
127 PRINT_REG_FP(fpregs);
128
129 #ifdef __i386__
130 user_fpxregs_struct fpxregs;
131 res = ptrace(PTRACE_GETFPXREGS, pid, NULL, &fpxregs);
132 assert(!res);
133 printf("%lx\n", (unsigned long)fpxregs.mxcsr);
134 #endif
135
136 ptrace(PTRACE_CONT, pid, NULL, NULL);
137 wait(NULL);
138 }
139 return 0;
140 }
141