Home | History | Annotate | Download | only in Linux 
      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 #endif
     70 
     71 
     72 int main(void) {
     73   pid_t pid;
     74   pid = fork();
     75   if (pid == 0) { // child
     76     ptrace(PTRACE_TRACEME, 0, NULL, NULL);
     77     execl("/bin/true", "true", NULL);
     78   } else {
     79     wait(NULL);
     80     regs_struct regs;
     81     regs_struct* volatile pregs = &regs;
     82 #ifdef ARCH_IOVEC_FOR_GETREGSET
     83     struct iovec regset_io;
     84 #endif
     85     int res;
     86 
     87 #ifdef POSITIVE
     88     ++pregs;
     89 #endif
     90 
     91 #ifdef ARCH_IOVEC_FOR_GETREGSET
     92 # define __PTRACE_REQUEST  PTRACE_GETREGSET
     93 # define __PTRACE_ARGS     (void*)NT_PRSTATUS, (void*)&regset_io
     94     regset_io.iov_base = pregs;
     95     regset_io.iov_len = sizeof(regs_struct);
     96 #else
     97 # define __PTRACE_REQUEST  PTRACE_GETREGS
     98 # define __PTRACE_ARGS     NULL, pregs
     99 #endif
    100     res = ptrace((enum __ptrace_request)__PTRACE_REQUEST, pid, __PTRACE_ARGS);
    101     // CHECK: AddressSanitizer: stack-buffer-overflow
    102     // CHECK: {{.*ptrace.cc:}}[[@LINE-2]]
    103     assert(!res);
    104     PRINT_REG_PC(regs);
    105 
    106     fpregs_struct fpregs;
    107 #ifdef ARCH_IOVEC_FOR_GETREGSET
    108 # define __PTRACE_FPREQUEST  PTRACE_GETREGSET
    109 # define __PTRACE_FPARGS     (void*)NT_PRSTATUS, (void*)&regset_io
    110     regset_io.iov_base = &fpregs;
    111     regset_io.iov_len = sizeof(fpregs_struct);
    112     res = ptrace((enum __ptrace_request)PTRACE_GETREGSET, pid, (void*)NT_FPREGSET,
    113                  (void*)&regset_io);
    114 #else
    115 # define __PTRACE_FPARGS     NULL, &fpregs
    116 #endif
    117     res = ptrace((enum __ptrace_request)__PTRACE_FPREQUEST, pid, __PTRACE_FPARGS);
    118     assert(!res);
    119     PRINT_REG_FP(fpregs);
    120 
    121 #ifdef __i386__
    122     user_fpxregs_struct fpxregs;
    123     res = ptrace(PTRACE_GETFPXREGS, pid, NULL, &fpxregs);
    124     assert(!res);
    125     printf("%lx\n", (unsigned long)fpxregs.mxcsr);
    126 #endif
    127 
    128     ptrace(PTRACE_CONT, pid, NULL, NULL);
    129     wait(NULL);
    130   }
    131   return 0;
    132 }
    133