[sanitizer] Add ptrace interceptor support for LoongArch

[tangyouling]

Summary

Add ptrace interceptor support for LoongArch, ptrace.cpp has been
tested and passed.

[tangyouling]

compiler-rt/test/asan/TestCases/Linux/ptrace.cpp

36

aarch64 should be printed with %llx here , but this is not related to our change anyway.

[SixWeining]

compiler-rt/lib/sanitizer_common/sanitizer_platform_interceptors.h

276

__loongarch__ or __loongarch_grlen == 64? I'm not sure. Same as below.

[tangyouling]

compiler-rt/lib/sanitizer_common/sanitizer_platform_interceptors.h

276

__loongarch__ or __loongarch_grlen == 64? I'm not sure. Same as below.

This part of the kernel ptrace treats 32 and 64 bits the same, so I used __loongarch__ for better compatibility with 32-bit implementations in the future

[tangyouling]

Add missing processing in sanitizer_common_syscalls.inc.

[tangyouling]

$ ./test/asan/LOONGARCH64LinuxConfig/TestCases/Linux/Output/ptrace.cpp.tmp
=================================================================
==3270112==ERROR: AddressSanitizer: stack-buffer-overflow on address 0x7ffff14fd188 at pc 0x5555575d8674 bp 0x7ffffbccdee0 sp 0x7ffffbccd670
WRITE of size 360 at 0x7ffff14fd188 thread T0
    #0 0x5555575d8670  (/home/loongson/llvm-work/llvm-project-test2/llvm-project/build_crt/test/asan/LOONGARCH64LinuxConfig/TestCases/Linux/Output/ptrace.cpp.tmp+0x68670)
    #1 0x55555767b8d4 in main /home/loongson/llvm-work/llvm-project-test2/llvm-project/compiler-rt/test/asan/TestCases/Linux/ptrace.cpp:115:11
    #2 0x7ffff2f71678  (/usr/lib64/libc.so.6+0x25678)
    #3 0x7ffff2f71764 in __libc_start_main (/usr/lib64/libc.so.6+0x25764)
    #4 0x555557596d84 in _start /opt/mylaos/build/glibc-2.36/csu/../sysdeps/loongarch/start.S:61

Address 0x7ffff14fd188 is located in stack of thread T0 at offset 392 in frame
    #0 0x55555767b3f0 in main /home/loongson/llvm-work/llvm-project-test2/llvm-project/compiler-rt/test/asan/TestCases/Linux/ptrace.cpp:87

  This frame has 4 object(s):
    [32, 392) 'regs' (line 95)
    [464, 472) 'pregs' (line 96) <== Memory access at offset 392 partially underflows this variable
    [496, 512) 'regset_io' (line 98) <== Memory access at offset 392 partially underflows this variable
    [528, 800) 'fpregs' (line 121) <== Memory access at offset 392 partially underflows this variable
HINT: this may be a false positive if your program uses some custom stack unwind mechanism, swapcontext or vfork
      (longjmp and C++ exceptions *are* supported)
SUMMARY: AddressSanitizer: stack-buffer-overflow (/home/loongson/llvm-work/llvm-project-test2/llvm-project/build_crt/test/asan/LOONGARCH64LinuxConfig/TestCases/Linux/Output/ptrace.cpp.tmp+0x68670) 
Shadow bytes around the buggy address:
  0x7ffff14fcf00: 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
  0x7ffff14fcf80: 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
  0x7ffff14fd000: f1 f1 f1 f1 00 00 00 00 00 00 00 00 00 00 00 00
  0x7ffff14fd080: 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
  0x7ffff14fd100: 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
=>0x7ffff14fd180: 00[f2]f2 f2 f2 f2 f2 f2 f2 f2 00 f2 f2 f2 00 00
  0x7ffff14fd200: f2 f2 f8 f8 f8 f8 f8 f8 f8 f8 f8 f8 f8 f8 f8 f8
  0x7ffff14fd280: f8 f8 f8 f8 f8 f8 f8 f8 f8 f8 f8 f8 f8 f8 f8 f8
  0x7ffff14fd300: f8 f8 f8 f8 f3 f3 f3 f3 f3 f3 f3 f3 00 00 00 00
  0x7ffff14fd380: 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
  0x7ffff14fd400: 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
Shadow byte legend (one shadow byte represents 8 application bytes):
  Addressable:           00
  Partially addressable: 01 02 03 04 05 06 07 
  Heap left redzone:       fa
  Freed heap region:       fd
  Stack left redzone:      f1
  Stack mid redzone:       f2
  Stack right redzone:     f3
  Stack after return:      f5
  Stack use after scope:   f8
  Global redzone:          f9
  Global init order:       f6
  Poisoned by user:        f7
  Container overflow:      fc
  Array cookie:            ac
  Intra object redzone:    bb
  ASan internal:           fe
  Left alloca redzone:     ca
  Right alloca redzone:    cb
==3270112==ABORTING

This is not related to our change anyway.

The current stack display has some problems, the location of the function definition cannot be found at level #0.
in LoongArch:

#0 0x5555575d8670 (/home/loongson/llvm-work/llvm-project-test2/llvm-project/build_crt/test/asan/LOONGARCH64LinuxConfig/TestCases/ Linux/Output/ptrace.cpp.tmp+0x68670)
#1 0x55555767b8d4 in main /home/loongson/llvm-work/llvm-project-test2/llvm-project/compiler-rt/test/asan/TestCases/Linux/ptrace.cpp: 115:11

in x86_64:

#0 0x561c13d22b18 in __interceptor_ptrace /home/loongson/tang/llvm-work/llvm-project/compiler-rt/lib/asan/../sanitizer_common/sanitizer_common_interceptors.inc
#1 0x561c13dbdc9a in main /home/loongson/tang/llvm-work/llvm-project/compiler-rt/test/asan/TestCases/Linux/ptrace.cpp:108:11

This phenomenon will cause some other test cases to fail.

[tangyouling]

ping?

[SixWeining]

Does the test still fail now?

[tangyouling]

Does the test still fail now?

ptrace.cpp test passed.

The current stack display has some problems, the location of the function definition cannot be found at level #0.

Calltrace display problem has been fixed by D137384 .

compiler-rt/lib/sanitizer_common/sanitizer_common_syscalls.inc

 PRE_SYSCALL(ni_syscall)() {}
 
 POST_SYSCALL(ni_syscall)(long res) {}
 
 PRE_SYSCALL(ptrace)(long request, long pid, long addr, long data) {
 #  if !SANITIZER_ANDROID &&                                                   \
       (defined(__i386) || defined(__x86_64) || defined(__mips64) ||           \
        defined(__powerpc64__) || defined(__aarch64__) || defined(__s390__) || \
-       SANITIZER_RISCV64)
+       defined(__loongarch__) || SANITIZER_RISCV64)
   if (data) {
     if (request == ptrace_setregs) {
       PRE_READ((void *)data, struct_user_regs_struct_sz);
     } else if (request == ptrace_setfpregs) {
       PRE_READ((void *)data, struct_user_fpregs_struct_sz);
     } else if (request == ptrace_setfpxregs) {
       PRE_READ((void *)data, struct_user_fpxregs_struct_sz);
     } else if (request == ptrace_setsiginfo) {
       PRE_READ((void *)data, siginfo_t_sz);
     } else if (request == ptrace_setregset) {
       __sanitizer_iovec *iov = (__sanitizer_iovec *)data;
       PRE_READ(iov->iov_base, iov->iov_len);
     }
   }
 #  endif
 }
 
 POST_SYSCALL(ptrace)(long res, long request, long pid, long addr, long data) {
 #  if !SANITIZER_ANDROID &&                                                   \
       (defined(__i386) || defined(__x86_64) || defined(__mips64) ||           \
        defined(__powerpc64__) || defined(__aarch64__) || defined(__s390__) || \
-       SANITIZER_RISCV64)
+       defined(__loongarch__) || SANITIZER_RISCV64)
   if (res >= 0 && data) {
     // Note that this is different from the interceptor in
     // sanitizer_common_interceptors.inc.
     // PEEK* requests return resulting values through data pointer.
     if (request == ptrace_getregs) {
       POST_WRITE((void *)data, struct_user_regs_struct_sz);
     } else if (request == ptrace_getfpregs) {
       POST_WRITE((void *)data, struct_user_fpregs_struct_sz);

compiler-rt/lib/sanitizer_common/sanitizer_platform_interceptors.h

 #define SANITIZER_INTERCEPT_IOCTL SI_POSIX
 #define SANITIZER_INTERCEPT_INET_ATON SI_POSIX
 #define SANITIZER_INTERCEPT_SYSINFO SI_LINUX
 #define SANITIZER_INTERCEPT_READDIR SI_POSIX
 #define SANITIZER_INTERCEPT_READDIR64 SI_LINUX_NOT_ANDROID || SI_SOLARIS32
 #if SI_LINUX_NOT_ANDROID &&                                                \
     (defined(__i386) || defined(__x86_64) || defined(__mips64) ||          \
      defined(__powerpc64__) || defined(__aarch64__) || defined(__arm__) || \
-     defined(__s390__) || SANITIZER_RISCV64)
+     defined(__s390__) || defined(__loongarch__) || SANITIZER_RISCV64)
 #define SANITIZER_INTERCEPT_PTRACE 1
 #else
 #define SANITIZER_INTERCEPT_PTRACE 0
 #endif
 #define SANITIZER_INTERCEPT_SETLOCALE SI_POSIX
 #define SANITIZER_INTERCEPT_GETCWD SI_POSIX
 #define SANITIZER_INTERCEPT_GET_CURRENT_DIR_NAME SI_LINUX_NOT_ANDROID
 #define SANITIZER_INTERCEPT_STRTOIMAX SI_POSIX

compiler-rt/lib/sanitizer_common/sanitizer_platform_limits_posix.h

 #else
 typedef void __sanitizer_FILE;
 # define SANITIZER_HAS_STRUCT_FILE 0
 #endif
 
 #if SANITIZER_LINUX && !SANITIZER_ANDROID &&                               \
     (defined(__i386) || defined(__x86_64) || defined(__mips64) ||          \
      defined(__powerpc64__) || defined(__aarch64__) || defined(__arm__) || \
-     defined(__s390__) || SANITIZER_RISCV64)
+     defined(__s390__) || defined(__loongarch__) || SANITIZER_RISCV64)
 extern unsigned struct_user_regs_struct_sz;
 extern unsigned struct_user_fpregs_struct_sz;
 extern unsigned struct_user_fpxregs_struct_sz;
 extern unsigned struct_user_vfpregs_struct_sz;
 
 extern int ptrace_peektext;
 extern int ptrace_peekdata;
 extern int ptrace_peekuser;

compiler-rt/lib/sanitizer_common/sanitizer_platform_limits_posix.cpp

 #if SANITIZER_IOS
 #undef IOC_DIRMASK
 #endif
 
 #if SANITIZER_LINUX
 # include <utime.h>
 # include <sys/ptrace.h>
 #    if defined(__mips64) || defined(__aarch64__) || defined(__arm__) || \
-        defined(__hexagon__) || SANITIZER_RISCV64
+        defined(__hexagon__) || defined(__loongarch__) ||SANITIZER_RISCV64
 #      include <asm/ptrace.h>
 #      ifdef __arm__
 typedef struct user_fpregs elf_fpregset_t;
 #   define ARM_VFPREGS_SIZE_ASAN (32 * 8 /*fpregs*/ + 4 /*fpscr*/)
 #   if !defined(ARM_VFPREGS_SIZE)
 #     define ARM_VFPREGS_SIZE ARM_VFPREGS_SIZE_ASAN
 #   endif
 #  endif

 
 #  if !SANITIZER_ANDROID
   const int wordexp_wrde_dooffs = WRDE_DOOFFS;
 #  endif  // !SANITIZER_ANDROID
 
 #if SANITIZER_LINUX && !SANITIZER_ANDROID &&                               \
     (defined(__i386) || defined(__x86_64) || defined(__mips64) ||          \
      defined(__powerpc64__) || defined(__aarch64__) || defined(__arm__) || \
-     defined(__s390__) || SANITIZER_RISCV64)
+     defined(__s390__) || defined(__loongarch__)|| SANITIZER_RISCV64)
 #if defined(__mips64) || defined(__powerpc64__) || defined(__arm__)
   unsigned struct_user_regs_struct_sz = sizeof(struct pt_regs);
   unsigned struct_user_fpregs_struct_sz = sizeof(elf_fpregset_t);
 #elif SANITIZER_RISCV64
   unsigned struct_user_regs_struct_sz = sizeof(struct user_regs_struct);
   unsigned struct_user_fpregs_struct_sz = sizeof(struct __riscv_q_ext_state);
 #elif defined(__aarch64__)
   unsigned struct_user_regs_struct_sz = sizeof(struct user_pt_regs);
   unsigned struct_user_fpregs_struct_sz = sizeof(struct user_fpsimd_state);
+#elif defined(__loongarch__)
+  unsigned struct_user_regs_struct_sz = sizeof(struct user_pt_regs);
+  unsigned struct_user_fpregs_struct_sz = sizeof(struct user_fp_state);
 #elif defined(__s390__)
   unsigned struct_user_regs_struct_sz = sizeof(struct _user_regs_struct);
   unsigned struct_user_fpregs_struct_sz = sizeof(struct _user_fpregs_struct);
 #else
   unsigned struct_user_regs_struct_sz = sizeof(struct user_regs_struct);
   unsigned struct_user_fpregs_struct_sz = sizeof(struct user_fpregs_struct);
-#endif // __mips64 || __powerpc64__ || __aarch64__
+#endif // __mips64 || __powerpc64__ || __aarch64__ || __loongarch__
 #if defined(__x86_64) || defined(__mips64) || defined(__powerpc64__) || \
     defined(__aarch64__) || defined(__arm__) || defined(__s390__) ||    \
-    SANITIZER_RISCV64
+    defined(__loongarch__) || SANITIZER_RISCV64
   unsigned struct_user_fpxregs_struct_sz = 0;
 #else
   unsigned struct_user_fpxregs_struct_sz = sizeof(struct user_fpxregs_struct);
 #endif // __x86_64 || __mips64 || __powerpc64__ || __aarch64__ || __arm__
-// || __s390__
+// || __s390__ || __loongarch__
 #ifdef __arm__
   unsigned struct_user_vfpregs_struct_sz = ARM_VFPREGS_SIZE;
 #else
   unsigned struct_user_vfpregs_struct_sz = 0;
 #endif
 
   int ptrace_peektext = PTRACE_PEEKTEXT;
   int ptrace_peekdata = PTRACE_PEEKDATA;

compiler-rt/test/asan/TestCases/Linux/ptrace.cpp

 // FIXME: https://code.google.com/p/address-sanitizer/issues/detail?id=316
 // XFAIL: android
 // XFAIL: mips
 //
 // RUN: %clangxx_asan -O0 %s -o %t && %run %t
 // RUN: %clangxx_asan -DPOSITIVE -O0 %s -o %t && not %run %t 2>&1 | FileCheck %s
 
 #include <assert.h>
 #include <stdio.h>
 #include <sys/ptrace.h>
 #include <sys/types.h>
 #include <sys/user.h>
 #include <sys/wait.h>
 #include <unistd.h>
 #include <sys/uio.h> // for iovec
 #include <elf.h> // for NT_PRSTATUS
-#ifdef __aarch64__
-# include <asm/ptrace.h>
+#if defined(__aarch64__) || defined(__loongarch__)
+#  include <asm/ptrace.h>
 #endif
 
 #if defined(__i386__) || defined(__x86_64__)
 typedef user_regs_struct   regs_struct;
 typedef user_fpregs_struct fpregs_struct;
 #if defined(__i386__)
 #define REG_IP  eip
 #else
 #define REG_IP  rip
 #endif
 #define PRINT_REG_PC(__regs)    printf ("%lx\n", (unsigned long) (__regs.REG_IP))
 #define PRINT_REG_FP(__fpregs)  printf ("%lx\n", (unsigned long) (__fpregs.cwd))
 #define __PTRACE_FPREQUEST PTRACE_GETFPREGS
 
 #elif defined(__aarch64__)
 typedef struct user_pt_regs      regs_struct;
 typedef struct user_fpsimd_state fpregs_struct;
 #define PRINT_REG_PC(__regs)    printf ("%x\n", (unsigned) (__regs.pc))
 #define PRINT_REG_FP(__fpregs)  printf ("%x\n", (unsigned) (__fpregs.fpsr))
 #define ARCH_IOVEC_FOR_GETREGSET
 
+#elif defined(__loongarch__)
+typedef struct user_pt_regs regs_struct;
+typedef struct user_fp_state fpregs_struct;
+#  define PRINT_REG_PC(__regs) printf("%lx\n", (unsigned long)(__regs.csr_era))
+#  define PRINT_REG_FP(__fpregs) printf("%x\n", (unsigned)(__fpregs.fcsr))
+#  define ARCH_IOVEC_FOR_GETREGSET
+
 #elif defined(__powerpc64__)
 typedef struct pt_regs regs_struct;
 typedef elf_fpregset_t fpregs_struct;
 #define PRINT_REG_PC(__regs)    printf ("%lx\n", (unsigned long) (__regs.nip))
 #define PRINT_REG_FP(__fpregs)  printf ("%lx\n", (elf_greg_t)fpregs[32])
 #define ARCH_IOVEC_FOR_GETREGSET
 
 #elif defined(__mips__)

compiler-rt/test/sanitizer_common/TestCases/Linux/ptrace.cpp

 #include <sys/uio.h>
 #include <sys/user.h>
 #include <sys/wait.h>
 #include <unistd.h>
 #if __mips64 || __arm__
  #include <asm/ptrace.h>
  #include <sys/procfs.h>
 #endif
-#ifdef __aarch64__
+#if defined(__aarch64__) || defined(__loongarch__)
 // GLIBC 2.20+ sys/user does not include asm/ptrace.h
  #include <asm/ptrace.h>
 #endif
 
 int main(void) {
   pid_t pid;
   pid = fork();
   if (pid == 0) { // child
#if (__aarch64__)
     regset_io.iov_base = &fpregs;
     regset_io.iov_len = sizeof(fpregs);
     res = ptrace(PTRACE_GETREGSET, pid, (void*)NT_FPREGSET, (void*)&regset_io);
     assert(!res);
     if (fpregs.fpsr)
       printf("%x\n", fpregs.fpsr);
 #endif // (__aarch64__)
 
+#if (__loongarch__)
+    struct iovec regset_io;
+
+    struct user_pt_regs regs;
+    regset_io.iov_base = &regs;
+    regset_io.iov_len = sizeof(regs);
+    res =
+        ptrace(PTRACE_GETREGSET, pid, (void *)NT_PRSTATUS, (void *)&regset_io);
+    assert(!res);
+    if (regs.csr_era)
+      printf("%lx\n", regs.csr_era);
+
+    struct user_fp_state fpregs;
+    regset_io.iov_base = &fpregs;
+    regset_io.iov_len = sizeof(fpregs);
+    res =
+        ptrace(PTRACE_GETREGSET, pid, (void *)NT_FPREGSET, (void *)&regset_io);
+    assert(!res);
+    if (fpregs.fcsr)
+      printf("%x\n", fpregs.fcsr);
+#endif // (__loongarch__)
+
 #if (__s390__)
     struct iovec regset_io;
 
     struct _user_regs_struct regs;
     regset_io.iov_base = &regs;
     regset_io.iov_len = sizeof(regs);
     res = ptrace(PTRACE_GETREGSET, pid, (void*)NT_PRSTATUS, (void*)&regset_io);
     assert(!res);