contrib/compiler-rt/lib/sanitizer_common/sanitizer_linux.cc

   1 //===-- sanitizer_linux.cc ------------------------------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file is shared between AddressSanitizer and ThreadSanitizer
  11 // run-time libraries and implements linux-specific functions from
  12 // sanitizer_libc.h.
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "sanitizer_platform.h"
  16
  17 #if SANITIZER_FREEBSD || SANITIZER_LINUX
  18
  19 #include "sanitizer_common.h"
  20 #include "sanitizer_flags.h"
  21 #include "sanitizer_internal_defs.h"
  22 #include "sanitizer_libc.h"
  23 #include "sanitizer_linux.h"
  24 #include "sanitizer_mutex.h"
  25 #include "sanitizer_placement_new.h"
  26 #include "sanitizer_procmaps.h"
  27 #include "sanitizer_stacktrace.h"
  28 #include "sanitizer_symbolizer.h"
  29
  30 #if !SANITIZER_FREEBSD
  31 #include <asm/param.h>
  32 #endif
  33
  34 // For mips64, syscall(__NR_stat) fills the buffer in the 'struct kernel_stat'
  35 // format. Struct kernel_stat is defined as 'struct stat' in asm/stat.h. To
  36 // access stat from asm/stat.h, without conflicting with definition in
  37 // sys/stat.h, we use this trick.
  38 #if defined(__mips64)
  39 #include <asm/unistd.h>
  40 #include <sys/types.h>
  41 #define stat kernel_stat
  42 #include <asm/stat.h>
  43 #undef stat
  44 #endif
  45
  46 #include <dlfcn.h>
  47 #include <errno.h>
  48 #include <fcntl.h>
  49 #include <link.h>
  50 #include <pthread.h>
  51 #include <sched.h>
  52 #include <sys/mman.h>
  53 #include <sys/ptrace.h>
  54 #include <sys/resource.h>
  55 #include <sys/stat.h>
  56 #include <sys/syscall.h>
  57 #include <sys/time.h>
  58 #include <sys/types.h>
  59 #include <ucontext.h>
  60 #include <unistd.h>
  61
  62 #if SANITIZER_FREEBSD
  63 #include <sys/exec.h>
  64 #include <sys/sysctl.h>
  65 #include <machine/atomic.h>
  66 extern "C" {
  67 // <sys/umtx.h> must be included after <errno.h> and <sys/types.h> on
  68 // FreeBSD 9.2 and 10.0.
  69 #include <sys/umtx.h>
  70 }
  71 extern char **environ;  // provided by crt1
  72 #endif  // SANITIZER_FREEBSD
  73
  74 #if !SANITIZER_ANDROID
  75 #include <sys/signal.h>
  76 #endif
  77
  78 #if SANITIZER_USE_GETAUXVAL
  79 #include <sys/auxv.h>
  80 #endif
  81
  82 #if SANITIZER_LINUX
  83 // <linux/time.h>
  84 struct kernel_timeval {
  85   long tv_sec;
  86   long tv_usec;
  87 };
  88
  89 // <linux/futex.h> is broken on some linux distributions.
  90 const int FUTEX_WAIT = 0;
  91 const int FUTEX_WAKE = 1;
  92 #endif  // SANITIZER_LINUX
  93
  94 // Are we using 32-bit or 64-bit Linux syscalls?
  95 // x32 (which defines __x86_64__) has SANITIZER_WORDSIZE == 32
  96 // but it still needs to use 64-bit syscalls.
  97 #if SANITIZER_LINUX && (defined(__x86_64__) || defined(__powerpc64__) || \
  98     SANITIZER_WORDSIZE == 64)
  99 # define SANITIZER_LINUX_USES_64BIT_SYSCALLS 1
 100 #else
 101 # define SANITIZER_LINUX_USES_64BIT_SYSCALLS 0
 102 #endif
 103
 104 #if defined(__x86_64__) || SANITIZER_MIPS64
 105 extern "C" {
 106 extern void internal_sigreturn();
 107 }
 108 #endif
 109
 110 namespace __sanitizer {
 111
 112 #if SANITIZER_LINUX && defined(__x86_64__)
 113 #include "sanitizer_syscall_linux_x86_64.inc"
 114 #elif SANITIZER_LINUX && defined(__aarch64__)
 115 #include "sanitizer_syscall_linux_aarch64.inc"
 116 #else
 117 #include "sanitizer_syscall_generic.inc"
 118 #endif
 119
 120 // --------------- sanitizer_libc.h
 121 #if !SANITIZER_S390
 122 uptr internal_mmap(void *addr, uptr length, int prot, int flags, int fd,
 123                    OFF_T offset) {
 124 #if SANITIZER_FREEBSD || SANITIZER_LINUX_USES_64BIT_SYSCALLS
 125   return internal_syscall(SYSCALL(mmap), (uptr)addr, length, prot, flags, fd,
 126                           offset);
 127 #else
 128   // mmap2 specifies file offset in 4096-byte units.
 129   CHECK(IsAligned(offset, 4096));
 130   return internal_syscall(SYSCALL(mmap2), addr, length, prot, flags, fd,
 131                           offset / 4096);
 132 #endif
 133 }
 134 #endif // !SANITIZER_S390
 135
 136 uptr internal_munmap(void *addr, uptr length) {
 137   return internal_syscall(SYSCALL(munmap), (uptr)addr, length);
 138 }
 139
 140 int internal_mprotect(void *addr, uptr length, int prot) {
 141   return internal_syscall(SYSCALL(mprotect), (uptr)addr, length, prot);
 142 }
 143
 144 uptr internal_close(fd_t fd) {
 145   return internal_syscall(SYSCALL(close), fd);
 146 }
 147
 148 uptr internal_open(const char *filename, int flags) {
 149 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
 150   return internal_syscall(SYSCALL(openat), AT_FDCWD, (uptr)filename, flags);
 151 #else
 152   return internal_syscall(SYSCALL(open), (uptr)filename, flags);
 153 #endif
 154 }
 155
 156 uptr internal_open(const char *filename, int flags, u32 mode) {
 157 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
 158   return internal_syscall(SYSCALL(openat), AT_FDCWD, (uptr)filename, flags,
 159                           mode);
 160 #else
 161   return internal_syscall(SYSCALL(open), (uptr)filename, flags, mode);
 162 #endif
 163 }
 164
 165 uptr internal_read(fd_t fd, void *buf, uptr count) {
 166   sptr res;
 167   HANDLE_EINTR(res, (sptr)internal_syscall(SYSCALL(read), fd, (uptr)buf,
 168                count));
 169   return res;
 170 }
 171
 172 uptr internal_write(fd_t fd, const void *buf, uptr count) {
 173   sptr res;
 174   HANDLE_EINTR(res, (sptr)internal_syscall(SYSCALL(write), fd, (uptr)buf,
 175                count));
 176   return res;
 177 }
 178
 179 uptr internal_ftruncate(fd_t fd, uptr size) {
 180   sptr res;
 181   HANDLE_EINTR(res, (sptr)internal_syscall(SYSCALL(ftruncate), fd,
 182                (OFF_T)size));
 183   return res;
 184 }
 185
 186 #if !SANITIZER_LINUX_USES_64BIT_SYSCALLS && !SANITIZER_FREEBSD
 187 static void stat64_to_stat(struct stat64 *in, struct stat *out) {
 188   internal_memset(out, 0, sizeof(*out));
 189   out->st_dev = in->st_dev;
 190   out->st_ino = in->st_ino;
 191   out->st_mode = in->st_mode;
 192   out->st_nlink = in->st_nlink;
 193   out->st_uid = in->st_uid;
 194   out->st_gid = in->st_gid;
 195   out->st_rdev = in->st_rdev;
 196   out->st_size = in->st_size;
 197   out->st_blksize = in->st_blksize;
 198   out->st_blocks = in->st_blocks;
 199   out->st_atime = in->st_atime;
 200   out->st_mtime = in->st_mtime;
 201   out->st_ctime = in->st_ctime;
 202   out->st_ino = in->st_ino;
 203 }
 204 #endif
 205
 206 #if defined(__mips64)
 207 static void kernel_stat_to_stat(struct kernel_stat *in, struct stat *out) {
 208   internal_memset(out, 0, sizeof(*out));
 209   out->st_dev = in->st_dev;
 210   out->st_ino = in->st_ino;
 211   out->st_mode = in->st_mode;
 212   out->st_nlink = in->st_nlink;
 213   out->st_uid = in->st_uid;
 214   out->st_gid = in->st_gid;
 215   out->st_rdev = in->st_rdev;
 216   out->st_size = in->st_size;
 217   out->st_blksize = in->st_blksize;
 218   out->st_blocks = in->st_blocks;
 219   out->st_atime = in->st_atime_nsec;
 220   out->st_mtime = in->st_mtime_nsec;
 221   out->st_ctime = in->st_ctime_nsec;
 222   out->st_ino = in->st_ino;
 223 }
 224 #endif
 225
 226 uptr internal_stat(const char *path, void *buf) {
 227 #if SANITIZER_FREEBSD
 228   return internal_syscall(SYSCALL(fstatat), AT_FDCWD, (uptr)path,
 229                           (uptr)buf, 0);
 230 #elif SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
 231   return internal_syscall(SYSCALL(newfstatat), AT_FDCWD, (uptr)path,
 232                           (uptr)buf, 0);
 233 #elif SANITIZER_LINUX_USES_64BIT_SYSCALLS
 234 # if defined(__mips64)
 235   // For mips64, stat syscall fills buffer in the format of kernel_stat
 236   struct kernel_stat kbuf;
 237   int res = internal_syscall(SYSCALL(stat), path, &kbuf);
 238   kernel_stat_to_stat(&kbuf, (struct stat *)buf);
 239   return res;
 240 # else
 241   return internal_syscall(SYSCALL(stat), (uptr)path, (uptr)buf);
 242 # endif
 243 #else
 244   struct stat64 buf64;
 245   int res = internal_syscall(SYSCALL(stat64), path, &buf64);
 246   stat64_to_stat(&buf64, (struct stat *)buf);
 247   return res;
 248 #endif
 249 }
 250
 251 uptr internal_lstat(const char *path, void *buf) {
 252 #if SANITIZER_FREEBSD
 253   return internal_syscall(SYSCALL(fstatat), AT_FDCWD, (uptr)path,
 254                           (uptr)buf, AT_SYMLINK_NOFOLLOW);
 255 #elif SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
 256   return internal_syscall(SYSCALL(newfstatat), AT_FDCWD, (uptr)path,
 257                          (uptr)buf, AT_SYMLINK_NOFOLLOW);
 258 #elif SANITIZER_LINUX_USES_64BIT_SYSCALLS
 259 # if SANITIZER_MIPS64
 260   // For mips64, lstat syscall fills buffer in the format of kernel_stat
 261   struct kernel_stat kbuf;
 262   int res = internal_syscall(SYSCALL(lstat), path, &kbuf);
 263   kernel_stat_to_stat(&kbuf, (struct stat *)buf);
 264   return res;
 265 # else
 266   return internal_syscall(SYSCALL(lstat), (uptr)path, (uptr)buf);
 267 # endif
 268 #else
 269   struct stat64 buf64;
 270   int res = internal_syscall(SYSCALL(lstat64), path, &buf64);
 271   stat64_to_stat(&buf64, (struct stat *)buf);
 272   return res;
 273 #endif
 274 }
 275
 276 uptr internal_fstat(fd_t fd, void *buf) {
 277 #if SANITIZER_FREEBSD || SANITIZER_LINUX_USES_64BIT_SYSCALLS
 278 # if SANITIZER_MIPS64
 279   // For mips64, fstat syscall fills buffer in the format of kernel_stat
 280   struct kernel_stat kbuf;
 281   int res = internal_syscall(SYSCALL(fstat), fd, &kbuf);
 282   kernel_stat_to_stat(&kbuf, (struct stat *)buf);
 283   return res;
 284 # else
 285   return internal_syscall(SYSCALL(fstat), fd, (uptr)buf);
 286 # endif
 287 #else
 288   struct stat64 buf64;
 289   int res = internal_syscall(SYSCALL(fstat64), fd, &buf64);
 290   stat64_to_stat(&buf64, (struct stat *)buf);
 291   return res;
 292 #endif
 293 }
 294
 295 uptr internal_filesize(fd_t fd) {
 296   struct stat st;
 297   if (internal_fstat(fd, &st))
 298     return -1;
 299   return (uptr)st.st_size;
 300 }
 301
 302 uptr internal_dup2(int oldfd, int newfd) {
 303 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
 304   return internal_syscall(SYSCALL(dup3), oldfd, newfd, 0);
 305 #else
 306   return internal_syscall(SYSCALL(dup2), oldfd, newfd);
 307 #endif
 308 }
 309
 310 uptr internal_readlink(const char *path, char *buf, uptr bufsize) {
 311 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
 312   return internal_syscall(SYSCALL(readlinkat), AT_FDCWD,
 313                           (uptr)path, (uptr)buf, bufsize);
 314 #else
 315   return internal_syscall(SYSCALL(readlink), (uptr)path, (uptr)buf, bufsize);
 316 #endif
 317 }
 318
 319 uptr internal_unlink(const char *path) {
 320 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
 321   return internal_syscall(SYSCALL(unlinkat), AT_FDCWD, (uptr)path, 0);
 322 #else
 323   return internal_syscall(SYSCALL(unlink), (uptr)path);
 324 #endif
 325 }
 326
 327 uptr internal_rename(const char *oldpath, const char *newpath) {
 328 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
 329   return internal_syscall(SYSCALL(renameat), AT_FDCWD, (uptr)oldpath, AT_FDCWD,
 330                           (uptr)newpath);
 331 #else
 332   return internal_syscall(SYSCALL(rename), (uptr)oldpath, (uptr)newpath);
 333 #endif
 334 }
 335
 336 uptr internal_sched_yield() {
 337   return internal_syscall(SYSCALL(sched_yield));
 338 }
 339
 340 void internal__exit(int exitcode) {
 341 #if SANITIZER_FREEBSD
 342   internal_syscall(SYSCALL(exit), exitcode);
 343 #else
 344   internal_syscall(SYSCALL(exit_group), exitcode);
 345 #endif
 346   Die();  // Unreachable.
 347 }
 348
 349 unsigned int internal_sleep(unsigned int seconds) {
 350   struct timespec ts;
 351   ts.tv_sec = 1;
 352   ts.tv_nsec = 0;
 353   int res = internal_syscall(SYSCALL(nanosleep), &ts, &ts);
 354   if (res) return ts.tv_sec;
 355   return 0;
 356 }
 357
 358 uptr internal_execve(const char *filename, char *const argv[],
 359                      char *const envp[]) {
 360   return internal_syscall(SYSCALL(execve), (uptr)filename, (uptr)argv,
 361                           (uptr)envp);
 362 }
 363
 364 // ----------------- sanitizer_common.h
 365 bool FileExists(const char *filename) {
 366   struct stat st;
 367 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
 368   if (internal_syscall(SYSCALL(newfstatat), AT_FDCWD, filename, &st, 0))
 369 #else
 370   if (internal_stat(filename, &st))
 371 #endif
 372     return false;
 373   // Sanity check: filename is a regular file.
 374   return S_ISREG(st.st_mode);
 375 }
 376
 377 tid_t GetTid() {
 378 #if SANITIZER_FREEBSD
 379   return (uptr)pthread_self();
 380 #else
 381   return internal_syscall(SYSCALL(gettid));
 382 #endif
 383 }
 384
 385 u64 NanoTime() {
 386 #if SANITIZER_FREEBSD
 387   timeval tv;
 388 #else
 389   kernel_timeval tv;
 390 #endif
 391   internal_memset(&tv, 0, sizeof(tv));
 392   internal_syscall(SYSCALL(gettimeofday), (uptr)&tv, 0);
 393   return (u64)tv.tv_sec * 1000*1000*1000 + tv.tv_usec * 1000;
 394 }
 395
 396 // Like getenv, but reads env directly from /proc (on Linux) or parses the
 397 // 'environ' array (on FreeBSD) and does not use libc. This function should be
 398 // called first inside __asan_init.
 399 const char *GetEnv(const char *name) {
 400 #if SANITIZER_FREEBSD
 401   if (::environ != 0) {
 402     uptr NameLen = internal_strlen(name);
 403     for (char **Env = ::environ; *Env != 0; Env++) {
 404       if (internal_strncmp(*Env, name, NameLen) == 0 && (*Env)[NameLen] == '=')
 405         return (*Env) + NameLen + 1;
 406     }
 407   }
 408   return 0;  // Not found.
 409 #elif SANITIZER_LINUX
 410   static char *environ;
 411   static uptr len;
 412   static bool inited;
 413   if (!inited) {
 414     inited = true;
 415     uptr environ_size;
 416     if (!ReadFileToBuffer("/proc/self/environ", &environ, &environ_size, &len))
 417       environ = nullptr;
 418   }
 419   if (!environ || len == 0) return nullptr;
 420   uptr namelen = internal_strlen(name);
 421   const char *p = environ;
 422   while (*p != '\0') {  // will happen at the \0\0 that terminates the buffer
 423     // proc file has the format NAME=value\0NAME=value\0NAME=value\0...
 424     const char* endp =
 425         (char*)internal_memchr(p, '\0', len - (p - environ));
 426     if (!endp)  // this entry isn't NUL terminated
 427       return nullptr;
 428     else if (!internal_memcmp(p, name, namelen) && p[namelen] == '=')  // Match.
 429       return p + namelen + 1;  // point after =
 430     p = endp + 1;
 431   }
 432   return nullptr;  // Not found.
 433 #else
 434 #error "Unsupported platform"
 435 #endif
 436 }
 437
 438 #if !SANITIZER_FREEBSD
 439 extern "C" {
 440   SANITIZER_WEAK_ATTRIBUTE extern void *__libc_stack_end;
 441 }
 442 #endif
 443
 444 #if !SANITIZER_GO && !SANITIZER_FREEBSD
 445 static void ReadNullSepFileToArray(const char *path, char ***arr,
 446                                    int arr_size) {
 447   char *buff;
 448   uptr buff_size;
 449   uptr buff_len;
 450   *arr = (char **)MmapOrDie(arr_size * sizeof(char *), "NullSepFileArray");
 451   if (!ReadFileToBuffer(path, &buff, &buff_size, &buff_len, 1024 * 1024)) {
 452     (*arr)[0] = nullptr;
 453     return;
 454   }
 455   (*arr)[0] = buff;
 456   int count, i;
 457   for (count = 1, i = 1; ; i++) {
 458     if (buff[i] == 0) {
 459       if (buff[i+1] == 0) break;
 460       (*arr)[count] = &buff[i+1];
 461       CHECK_LE(count, arr_size - 1);  // FIXME: make this more flexible.
 462       count++;
 463     }
 464   }
 465   (*arr)[count] = nullptr;
 466 }
 467 #endif
 468
 469 static void GetArgsAndEnv(char ***argv, char ***envp) {
 470 #if !SANITIZER_FREEBSD
 471 #if !SANITIZER_GO
 472   if (&__libc_stack_end) {
 473 #endif
 474     uptr* stack_end = (uptr*)__libc_stack_end;
 475     int argc = *stack_end;
 476     *argv = (char**)(stack_end + 1);
 477     *envp = (char**)(stack_end + argc + 2);
 478 #if !SANITIZER_GO
 479   } else {
 480     static const int kMaxArgv = 2000, kMaxEnvp = 2000;
 481     ReadNullSepFileToArray("/proc/self/cmdline", argv, kMaxArgv);
 482     ReadNullSepFileToArray("/proc/self/environ", envp, kMaxEnvp);
 483   }
 484 #endif
 485 #else
 486   // On FreeBSD, retrieving the argument and environment arrays is done via the
 487   // kern.ps_strings sysctl, which returns a pointer to a structure containing
 488   // this information. See also <sys/exec.h>.
 489   ps_strings *pss;
 490   size_t sz = sizeof(pss);
 491   if (sysctlbyname("kern.ps_strings", &pss, &sz, NULL, 0) == -1) {
 492     Printf("sysctl kern.ps_strings failed\n");
 493     Die();
 494   }
 495   *argv = pss->ps_argvstr;
 496   *envp = pss->ps_envstr;
 497 #endif
 498 }
 499
 500 char **GetArgv() {
 501   char **argv, **envp;
 502   GetArgsAndEnv(&argv, &envp);
 503   return argv;
 504 }
 505
 506 void ReExec() {
 507   char **argv, **envp;
 508   GetArgsAndEnv(&argv, &envp);
 509   uptr rv = internal_execve("/proc/self/exe", argv, envp);
 510   int rverrno;
 511   CHECK_EQ(internal_iserror(rv, &rverrno), true);
 512   Printf("execve failed, errno %d\n", rverrno);
 513   Die();
 514 }
 515
 516 enum MutexState {
 517   MtxUnlocked = 0,
 518   MtxLocked = 1,
 519   MtxSleeping = 2
 520 };
 521
 522 BlockingMutex::BlockingMutex() {
 523   internal_memset(this, 0, sizeof(*this));
 524 }
 525
 526 void BlockingMutex::Lock() {
 527   CHECK_EQ(owner_, 0);
 528   atomic_uint32_t *m = reinterpret_cast<atomic_uint32_t *>(&opaque_storage_);
 529   if (atomic_exchange(m, MtxLocked, memory_order_acquire) == MtxUnlocked)
 530     return;
 531   while (atomic_exchange(m, MtxSleeping, memory_order_acquire) != MtxUnlocked) {
 532 #if SANITIZER_FREEBSD
 533     _umtx_op(m, UMTX_OP_WAIT_UINT, MtxSleeping, 0, 0);
 534 #else
 535     internal_syscall(SYSCALL(futex), (uptr)m, FUTEX_WAIT, MtxSleeping, 0, 0, 0);
 536 #endif
 537   }
 538 }
 539
 540 void BlockingMutex::Unlock() {
 541   atomic_uint32_t *m = reinterpret_cast<atomic_uint32_t *>(&opaque_storage_);
 542   u32 v = atomic_exchange(m, MtxUnlocked, memory_order_release);
 543   CHECK_NE(v, MtxUnlocked);
 544   if (v == MtxSleeping) {
 545 #if SANITIZER_FREEBSD
 546     _umtx_op(m, UMTX_OP_WAKE, 1, 0, 0);
 547 #else
 548     internal_syscall(SYSCALL(futex), (uptr)m, FUTEX_WAKE, 1, 0, 0, 0);
 549 #endif
 550   }
 551 }
 552
 553 void BlockingMutex::CheckLocked() {
 554   atomic_uint32_t *m = reinterpret_cast<atomic_uint32_t *>(&opaque_storage_);
 555   CHECK_NE(MtxUnlocked, atomic_load(m, memory_order_relaxed));
 556 }
 557
 558 // ----------------- sanitizer_linux.h
 559 // The actual size of this structure is specified by d_reclen.
 560 // Note that getdents64 uses a different structure format. We only provide the
 561 // 32-bit syscall here.
 562 struct linux_dirent {
 563 #if SANITIZER_X32 || defined(__aarch64__)
 564   u64 d_ino;
 565   u64 d_off;
 566 #else
 567   unsigned long      d_ino;
 568   unsigned long      d_off;
 569 #endif
 570   unsigned short     d_reclen;
 571 #ifdef __aarch64__
 572   unsigned char      d_type;
 573 #endif
 574   char               d_name[256];
 575 };
 576
 577 // Syscall wrappers.
 578 uptr internal_ptrace(int request, int pid, void *addr, void *data) {
 579   return internal_syscall(SYSCALL(ptrace), request, pid, (uptr)addr,
 580                           (uptr)data);
 581 }
 582
 583 uptr internal_waitpid(int pid, int *status, int options) {
 584   return internal_syscall(SYSCALL(wait4), pid, (uptr)status, options,
 585                           0 /* rusage */);
 586 }
 587
 588 uptr internal_getpid() {
 589   return internal_syscall(SYSCALL(getpid));
 590 }
 591
 592 uptr internal_getppid() {
 593   return internal_syscall(SYSCALL(getppid));
 594 }
 595
 596 uptr internal_getdents(fd_t fd, struct linux_dirent *dirp, unsigned int count) {
 597 #if SANITIZER_FREEBSD
 598   return internal_syscall(SYSCALL(getdirentries), fd, (uptr)dirp, count, NULL);
 599 #elif SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
 600   return internal_syscall(SYSCALL(getdents64), fd, (uptr)dirp, count);
 601 #else
 602   return internal_syscall(SYSCALL(getdents), fd, (uptr)dirp, count);
 603 #endif
 604 }
 605
 606 uptr internal_lseek(fd_t fd, OFF_T offset, int whence) {
 607   return internal_syscall(SYSCALL(lseek), fd, offset, whence);
 608 }
 609
 610 #if SANITIZER_LINUX
 611 uptr internal_prctl(int option, uptr arg2, uptr arg3, uptr arg4, uptr arg5) {
 612   return internal_syscall(SYSCALL(prctl), option, arg2, arg3, arg4, arg5);
 613 }
 614 #endif
 615
 616 uptr internal_sigaltstack(const struct sigaltstack *ss,
 617                          struct sigaltstack *oss) {
 618   return internal_syscall(SYSCALL(sigaltstack), (uptr)ss, (uptr)oss);
 619 }
 620
 621 int internal_fork() {
 622 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
 623   return internal_syscall(SYSCALL(clone), SIGCHLD, 0);
 624 #else
 625   return internal_syscall(SYSCALL(fork));
 626 #endif
 627 }
 628
 629 #if SANITIZER_LINUX
 630 #define SA_RESTORER 0x04000000
 631 // Doesn't set sa_restorer if the caller did not set it, so use with caution
 632 //(see below).
 633 int internal_sigaction_norestorer(int signum, const void *act, void *oldact) {
 634   __sanitizer_kernel_sigaction_t k_act, k_oldact;
 635   internal_memset(&k_act, 0, sizeof(__sanitizer_kernel_sigaction_t));
 636   internal_memset(&k_oldact, 0, sizeof(__sanitizer_kernel_sigaction_t));
 637   const __sanitizer_sigaction *u_act = (const __sanitizer_sigaction *)act;
 638   __sanitizer_sigaction *u_oldact = (__sanitizer_sigaction *)oldact;
 639   if (u_act) {
 640     k_act.handler = u_act->handler;
 641     k_act.sigaction = u_act->sigaction;
 642     internal_memcpy(&k_act.sa_mask, &u_act->sa_mask,
 643                     sizeof(__sanitizer_kernel_sigset_t));
 644     // Without SA_RESTORER kernel ignores the calls (probably returns EINVAL).
 645     k_act.sa_flags = u_act->sa_flags | SA_RESTORER;
 646     // FIXME: most often sa_restorer is unset, however the kernel requires it
 647     // to point to a valid signal restorer that calls the rt_sigreturn syscall.
 648     // If sa_restorer passed to the kernel is NULL, the program may crash upon
 649     // signal delivery or fail to unwind the stack in the signal handler.
 650     // libc implementation of sigaction() passes its own restorer to
 651     // rt_sigaction, so we need to do the same (we'll need to reimplement the
 652     // restorers; for x86_64 the restorer address can be obtained from
 653     // oldact->sa_restorer upon a call to sigaction(xxx, NULL, oldact).
 654 #if !SANITIZER_ANDROID || !SANITIZER_MIPS32
 655     k_act.sa_restorer = u_act->sa_restorer;
 656 #endif
 657   }
 658
 659   uptr result = internal_syscall(SYSCALL(rt_sigaction), (uptr)signum,
 660       (uptr)(u_act ? &k_act : nullptr),
 661       (uptr)(u_oldact ? &k_oldact : nullptr),
 662       (uptr)sizeof(__sanitizer_kernel_sigset_t));
 663
 664   if ((result == 0) && u_oldact) {
 665     u_oldact->handler = k_oldact.handler;
 666     u_oldact->sigaction = k_oldact.sigaction;
 667     internal_memcpy(&u_oldact->sa_mask, &k_oldact.sa_mask,
 668                     sizeof(__sanitizer_kernel_sigset_t));
 669     u_oldact->sa_flags = k_oldact.sa_flags;
 670 #if !SANITIZER_ANDROID || !SANITIZER_MIPS32
 671     u_oldact->sa_restorer = k_oldact.sa_restorer;
 672 #endif
 673   }
 674   return result;
 675 }
 676
 677 // Invokes sigaction via a raw syscall with a restorer, but does not support
 678 // all platforms yet.
 679 // We disable for Go simply because we have not yet added to buildgo.sh.
 680 #if (defined(__x86_64__) || SANITIZER_MIPS64) && !SANITIZER_GO
 681 int internal_sigaction_syscall(int signum, const void *act, void *oldact) {
 682   if (act == nullptr)
 683     return internal_sigaction_norestorer(signum, act, oldact);
 684   __sanitizer_sigaction u_adjust;
 685   internal_memcpy(&u_adjust, act, sizeof(u_adjust));
 686 #if !SANITIZER_ANDROID || !SANITIZER_MIPS32
 687     if (u_adjust.sa_restorer == nullptr) {
 688       u_adjust.sa_restorer = internal_sigreturn;
 689     }
 690 #endif
 691     return internal_sigaction_norestorer(signum, (const void *)&u_adjust,
 692                                          oldact);
 693 }
 694 #endif // defined(__x86_64__) && !SANITIZER_GO
 695 #endif  // SANITIZER_LINUX
 696
 697 uptr internal_sigprocmask(int how, __sanitizer_sigset_t *set,
 698     __sanitizer_sigset_t *oldset) {
 699 #if SANITIZER_FREEBSD
 700   return internal_syscall(SYSCALL(sigprocmask), how, set, oldset);
 701 #else
 702   __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
 703   __sanitizer_kernel_sigset_t *k_oldset = (__sanitizer_kernel_sigset_t *)oldset;
 704   return internal_syscall(SYSCALL(rt_sigprocmask), (uptr)how,
 705                           (uptr)&k_set->sig[0], (uptr)&k_oldset->sig[0],
 706                           sizeof(__sanitizer_kernel_sigset_t));
 707 #endif
 708 }
 709
 710 void internal_sigfillset(__sanitizer_sigset_t *set) {
 711   internal_memset(set, 0xff, sizeof(*set));
 712 }
 713
 714 void internal_sigemptyset(__sanitizer_sigset_t *set) {
 715   internal_memset(set, 0, sizeof(*set));
 716 }
 717
 718 #if SANITIZER_LINUX
 719 void internal_sigdelset(__sanitizer_sigset_t *set, int signum) {
 720   signum -= 1;
 721   CHECK_GE(signum, 0);
 722   CHECK_LT(signum, sizeof(*set) * 8);
 723   __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
 724   const uptr idx = signum / (sizeof(k_set->sig[0]) * 8);
 725   const uptr bit = signum % (sizeof(k_set->sig[0]) * 8);
 726   k_set->sig[idx] &= ~(1 << bit);
 727 }
 728
 729 bool internal_sigismember(__sanitizer_sigset_t *set, int signum) {
 730   signum -= 1;
 731   CHECK_GE(signum, 0);
 732   CHECK_LT(signum, sizeof(*set) * 8);
 733   __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
 734   const uptr idx = signum / (sizeof(k_set->sig[0]) * 8);
 735   const uptr bit = signum % (sizeof(k_set->sig[0]) * 8);
 736   return k_set->sig[idx] & (1 << bit);
 737 }
 738 #endif  // SANITIZER_LINUX
 739
 740 // ThreadLister implementation.
 741 ThreadLister::ThreadLister(int pid)
 742   : pid_(pid),
 743     descriptor_(-1),
 744     buffer_(4096),
 745     error_(true),
 746     entry_((struct linux_dirent *)buffer_.data()),
 747     bytes_read_(0) {
 748   char task_directory_path[80];
 749   internal_snprintf(task_directory_path, sizeof(task_directory_path),
 750                     "/proc/%d/task/", pid);
 751   uptr openrv = internal_open(task_directory_path, O_RDONLY | O_DIRECTORY);
 752   if (internal_iserror(openrv)) {
 753     error_ = true;
 754     Report("Can't open /proc/%d/task for reading.\n", pid);
 755   } else {
 756     error_ = false;
 757     descriptor_ = openrv;
 758   }
 759 }
 760
 761 int ThreadLister::GetNextTID() {
 762   int tid = -1;
 763   do {
 764     if (error_)
 765       return -1;
 766     if ((char *)entry_ >= &buffer_[bytes_read_] && !GetDirectoryEntries())
 767       return -1;
 768     if (entry_->d_ino != 0 && entry_->d_name[0] >= '0' &&
 769         entry_->d_name[0] <= '9') {
 770       // Found a valid tid.
 771       tid = (int)internal_atoll(entry_->d_name);
 772     }
 773     entry_ = (struct linux_dirent *)(((char *)entry_) + entry_->d_reclen);
 774   } while (tid < 0);
 775   return tid;
 776 }
 777
 778 void ThreadLister::Reset() {
 779   if (error_ || descriptor_ < 0)
 780     return;
 781   internal_lseek(descriptor_, 0, SEEK_SET);
 782 }
 783
 784 ThreadLister::~ThreadLister() {
 785   if (descriptor_ >= 0)
 786     internal_close(descriptor_);
 787 }
 788
 789 bool ThreadLister::error() { return error_; }
 790
 791 bool ThreadLister::GetDirectoryEntries() {
 792   CHECK_GE(descriptor_, 0);
 793   CHECK_NE(error_, true);
 794   bytes_read_ = internal_getdents(descriptor_,
 795                                   (struct linux_dirent *)buffer_.data(),
 796                                   buffer_.size());
 797   if (internal_iserror(bytes_read_)) {
 798     Report("Can't read directory entries from /proc/%d/task.\n", pid_);
 799     error_ = true;
 800     return false;
 801   } else if (bytes_read_ == 0) {
 802     return false;
 803   }
 804   entry_ = (struct linux_dirent *)buffer_.data();
 805   return true;
 806 }
 807
 808 uptr GetPageSize() {
 809 // Android post-M sysconf(_SC_PAGESIZE) crashes if called from .preinit_array.
 810 #if SANITIZER_ANDROID
 811   return 4096;
 812 #elif SANITIZER_LINUX && (defined(__x86_64__) || defined(__i386__))
 813   return EXEC_PAGESIZE;
 814 #elif SANITIZER_USE_GETAUXVAL
 815   return getauxval(AT_PAGESZ);
 816 #else
 817   return sysconf(_SC_PAGESIZE);  // EXEC_PAGESIZE may not be trustworthy.
 818 #endif
 819 }
 820
 821 uptr ReadBinaryName(/*out*/char *buf, uptr buf_len) {
 822 #if SANITIZER_FREEBSD
 823   const int Mib[] = { CTL_KERN, KERN_PROC, KERN_PROC_PATHNAME, -1 };
 824   const char *default_module_name = "kern.proc.pathname";
 825   size_t Size = buf_len;
 826   bool IsErr = (sysctl(Mib, ARRAY_SIZE(Mib), buf, &Size, NULL, 0) != 0);
 827   int readlink_error = IsErr ? errno : 0;
 828   uptr module_name_len = Size;
 829 #else
 830   const char *default_module_name = "/proc/self/exe";
 831   uptr module_name_len = internal_readlink(
 832       default_module_name, buf, buf_len);
 833   int readlink_error;
 834   bool IsErr = internal_iserror(module_name_len, &readlink_error);
 835 #endif
 836   if (IsErr) {
 837     // We can't read binary name for some reason, assume it's unknown.
 838     Report("WARNING: reading executable name failed with errno %d, "
 839            "some stack frames may not be symbolized\n", readlink_error);
 840     module_name_len = internal_snprintf(buf, buf_len, "%s",
 841                                         default_module_name);
 842     CHECK_LT(module_name_len, buf_len);
 843   }
 844   return module_name_len;
 845 }
 846
 847 uptr ReadLongProcessName(/*out*/ char *buf, uptr buf_len) {
 848 #if SANITIZER_LINUX
 849   char *tmpbuf;
 850   uptr tmpsize;
 851   uptr tmplen;
 852   if (ReadFileToBuffer("/proc/self/cmdline", &tmpbuf, &tmpsize, &tmplen,
 853                        1024 * 1024)) {
 854     internal_strncpy(buf, tmpbuf, buf_len);
 855     UnmapOrDie(tmpbuf, tmpsize);
 856     return internal_strlen(buf);
 857   }
 858 #endif
 859   return ReadBinaryName(buf, buf_len);
 860 }
 861
 862 // Match full names of the form /path/to/base_name{-,.}*
 863 bool LibraryNameIs(const char *full_name, const char *base_name) {
 864   const char *name = full_name;
 865   // Strip path.
 866   while (*name != '\0') name++;
 867   while (name > full_name && *name != '/') name--;
 868   if (*name == '/') name++;
 869   uptr base_name_length = internal_strlen(base_name);
 870   if (internal_strncmp(name, base_name, base_name_length)) return false;
 871   return (name[base_name_length] == '-' || name[base_name_length] == '.');
 872 }
 873
 874 #if !SANITIZER_ANDROID
 875 // Call cb for each region mapped by map.
 876 void ForEachMappedRegion(link_map *map, void (*cb)(const void *, uptr)) {
 877   CHECK_NE(map, nullptr);
 878 #if !SANITIZER_FREEBSD
 879   typedef ElfW(Phdr) Elf_Phdr;
 880   typedef ElfW(Ehdr) Elf_Ehdr;
 881 #endif  // !SANITIZER_FREEBSD
 882   char *base = (char *)map->l_addr;
 883   Elf_Ehdr *ehdr = (Elf_Ehdr *)base;
 884   char *phdrs = base + ehdr->e_phoff;
 885   char *phdrs_end = phdrs + ehdr->e_phnum * ehdr->e_phentsize;
 886
 887   // Find the segment with the minimum base so we can "relocate" the p_vaddr
 888   // fields.  Typically ET_DYN objects (DSOs) have base of zero and ET_EXEC
 889   // objects have a non-zero base.
 890   uptr preferred_base = (uptr)-1;
 891   for (char *iter = phdrs; iter != phdrs_end; iter += ehdr->e_phentsize) {
 892     Elf_Phdr *phdr = (Elf_Phdr *)iter;
 893     if (phdr->p_type == PT_LOAD && preferred_base > (uptr)phdr->p_vaddr)
 894       preferred_base = (uptr)phdr->p_vaddr;
 895   }
 896
 897   // Compute the delta from the real base to get a relocation delta.
 898   sptr delta = (uptr)base - preferred_base;
 899   // Now we can figure out what the loader really mapped.
 900   for (char *iter = phdrs; iter != phdrs_end; iter += ehdr->e_phentsize) {
 901     Elf_Phdr *phdr = (Elf_Phdr *)iter;
 902     if (phdr->p_type == PT_LOAD) {
 903       uptr seg_start = phdr->p_vaddr + delta;
 904       uptr seg_end = seg_start + phdr->p_memsz;
 905       // None of these values are aligned.  We consider the ragged edges of the
 906       // load command as defined, since they are mapped from the file.
 907       seg_start = RoundDownTo(seg_start, GetPageSizeCached());
 908       seg_end = RoundUpTo(seg_end, GetPageSizeCached());
 909       cb((void *)seg_start, seg_end - seg_start);
 910     }
 911   }
 912 }
 913 #endif
 914
 915 #if defined(__x86_64__) && SANITIZER_LINUX
 916 // We cannot use glibc's clone wrapper, because it messes with the child
 917 // task's TLS. It writes the PID and TID of the child task to its thread
 918 // descriptor, but in our case the child task shares the thread descriptor with
 919 // the parent (because we don't know how to allocate a new thread
 920 // descriptor to keep glibc happy). So the stock version of clone(), when
 921 // used with CLONE_VM, would end up corrupting the parent's thread descriptor.
 922 uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
 923                     int *parent_tidptr, void *newtls, int *child_tidptr) {
 924   long long res;
 925   if (!fn || !child_stack)
 926     return -EINVAL;
 927   CHECK_EQ(0, (uptr)child_stack % 16);
 928   child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
 929   ((unsigned long long *)child_stack)[0] = (uptr)fn;
 930   ((unsigned long long *)child_stack)[1] = (uptr)arg;
 931   register void *r8 __asm__("r8") = newtls;
 932   register int *r10 __asm__("r10") = child_tidptr;
 933   __asm__ __volatile__(
 934                        /* %rax = syscall(%rax = SYSCALL(clone),
 935                         *                %rdi = flags,
 936                         *                %rsi = child_stack,
 937                         *                %rdx = parent_tidptr,
 938                         *                %r8  = new_tls,
 939                         *                %r10 = child_tidptr)
 940                         */
 941                        "syscall\n"
 942
 943                        /* if (%rax != 0)
 944                         *   return;
 945                         */
 946                        "testq  %%rax,%%rax\n"
 947                        "jnz    1f\n"
 948
 949                        /* In the child. Terminate unwind chain. */
 950                        // XXX: We should also terminate the CFI unwind chain
 951                        // here. Unfortunately clang 3.2 doesn't support the
 952                        // necessary CFI directives, so we skip that part.
 953                        "xorq   %%rbp,%%rbp\n"
 954
 955                        /* Call "fn(arg)". */
 956                        "popq   %%rax\n"
 957                        "popq   %%rdi\n"
 958                        "call   *%%rax\n"
 959
 960                        /* Call _exit(%rax). */
 961                        "movq   %%rax,%%rdi\n"
 962                        "movq   %2,%%rax\n"
 963                        "syscall\n"
 964
 965                        /* Return to parent. */
 966                      "1:\n"
 967                        : "=a" (res)
 968                        : "a"(SYSCALL(clone)), "i"(SYSCALL(exit)),
 969                          "S"(child_stack),
 970                          "D"(flags),
 971                          "d"(parent_tidptr),
 972                          "r"(r8),
 973                          "r"(r10)
 974                        : "rsp", "memory", "r11", "rcx");
 975   return res;
 976 }
 977 #elif defined(__mips__)
 978 uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
 979                     int *parent_tidptr, void *newtls, int *child_tidptr) {
 980   long long res;
 981   if (!fn || !child_stack)
 982     return -EINVAL;
 983   CHECK_EQ(0, (uptr)child_stack % 16);
 984   child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
 985   ((unsigned long long *)child_stack)[0] = (uptr)fn;
 986   ((unsigned long long *)child_stack)[1] = (uptr)arg;
 987   register void *a3 __asm__("$7") = newtls;
 988   register int *a4 __asm__("$8") = child_tidptr;
 989   // We don't have proper CFI directives here because it requires alot of code
 990   // for very marginal benefits.
 991   __asm__ __volatile__(
 992                        /* $v0 = syscall($v0 = __NR_clone,
 993                         * $a0 = flags,
 994                         * $a1 = child_stack,
 995                         * $a2 = parent_tidptr,
 996                         * $a3 = new_tls,
 997                         * $a4 = child_tidptr)
 998                         */
 999                        ".cprestore 16;\n"
1000                        "move $4,%1;\n"
1001                        "move $5,%2;\n"
1002                        "move $6,%3;\n"
1003                        "move $7,%4;\n"
1004                        /* Store the fifth argument on stack
1005                         * if we are using 32-bit abi.
1006                         */
1007 #if SANITIZER_WORDSIZE == 32
1008                        "lw %5,16($29);\n"
1009 #else
1010                        "move $8,%5;\n"
1011 #endif
1012                        "li $2,%6;\n"
1013                        "syscall;\n"
1014
1015                        /* if ($v0 != 0)
1016                         * return;
1017                         */
1018                        "bnez $2,1f;\n"
1019
1020                        /* Call "fn(arg)". */
1021 #if SANITIZER_WORDSIZE == 32
1022 #ifdef __BIG_ENDIAN__
1023                        "lw $25,4($29);\n"
1024                        "lw $4,12($29);\n"
1025 #else
1026                        "lw $25,0($29);\n"
1027                        "lw $4,8($29);\n"
1028 #endif
1029 #else
1030                        "ld $25,0($29);\n"
1031                        "ld $4,8($29);\n"
1032 #endif
1033                        "jal $25;\n"
1034
1035                        /* Call _exit($v0). */
1036                        "move $4,$2;\n"
1037                        "li $2,%7;\n"
1038                        "syscall;\n"
1039
1040                        /* Return to parent. */
1041                      "1:\n"
1042                        : "=r" (res)
1043                        : "r"(flags),
1044                          "r"(child_stack),
1045                          "r"(parent_tidptr),
1046                          "r"(a3),
1047                          "r"(a4),
1048                          "i"(__NR_clone),
1049                          "i"(__NR_exit)
1050                        : "memory", "$29" );
1051   return res;
1052 }
1053 #elif defined(__aarch64__)
1054 uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1055                     int *parent_tidptr, void *newtls, int *child_tidptr) {
1056   long long res;
1057   if (!fn || !child_stack)
1058     return -EINVAL;
1059   CHECK_EQ(0, (uptr)child_stack % 16);
1060   child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
1061   ((unsigned long long *)child_stack)[0] = (uptr)fn;
1062   ((unsigned long long *)child_stack)[1] = (uptr)arg;
1063
1064   register int (*__fn)(void *)  __asm__("x0") = fn;
1065   register void *__stack __asm__("x1") = child_stack;
1066   register int   __flags __asm__("x2") = flags;
1067   register void *__arg   __asm__("x3") = arg;
1068   register int  *__ptid  __asm__("x4") = parent_tidptr;
1069   register void *__tls   __asm__("x5") = newtls;
1070   register int  *__ctid  __asm__("x6") = child_tidptr;
1071
1072   __asm__ __volatile__(
1073                        "mov x0,x2\n" /* flags  */
1074                        "mov x2,x4\n" /* ptid  */
1075                        "mov x3,x5\n" /* tls  */
1076                        "mov x4,x6\n" /* ctid  */
1077                        "mov x8,%9\n" /* clone  */
1078
1079                        "svc 0x0\n"
1080
1081                        /* if (%r0 != 0)
1082                         *   return %r0;
1083                         */
1084                        "cmp x0, #0\n"
1085                        "bne 1f\n"
1086
1087                        /* In the child, now. Call "fn(arg)". */
1088                        "ldp x1, x0, [sp], #16\n"
1089                        "blr x1\n"
1090
1091                        /* Call _exit(%r0).  */
1092                        "mov x8, %10\n"
1093                        "svc 0x0\n"
1094                      "1:\n"
1095
1096                        : "=r" (res)
1097                        : "i"(-EINVAL),
1098                          "r"(__fn), "r"(__stack), "r"(__flags), "r"(__arg),
1099                          "r"(__ptid), "r"(__tls), "r"(__ctid),
1100                          "i"(__NR_clone), "i"(__NR_exit)
1101                        : "x30", "memory");
1102   return res;
1103 }
1104 #elif defined(__powerpc64__)
1105 uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1106                    int *parent_tidptr, void *newtls, int *child_tidptr) {
1107   long long res;
1108 // Stack frame structure.
1109 #if SANITIZER_PPC64V1
1110 //   Back chain == 0        (SP + 112)
1111 // Frame (112 bytes):
1112 //   Parameter save area    (SP + 48), 8 doublewords
1113 //   TOC save area          (SP + 40)
1114 //   Link editor doubleword (SP + 32)
1115 //   Compiler doubleword    (SP + 24)
1116 //   LR save area           (SP + 16)
1117 //   CR save area           (SP + 8)
1118 //   Back chain             (SP + 0)
1119 # define FRAME_SIZE 112
1120 # define FRAME_TOC_SAVE_OFFSET 40
1121 #elif SANITIZER_PPC64V2
1122 //   Back chain == 0        (SP + 32)
1123 // Frame (32 bytes):
1124 //   TOC save area          (SP + 24)
1125 //   LR save area           (SP + 16)
1126 //   CR save area           (SP + 8)
1127 //   Back chain             (SP + 0)
1128 # define FRAME_SIZE 32
1129 # define FRAME_TOC_SAVE_OFFSET 24
1130 #else
1131 # error "Unsupported PPC64 ABI"
1132 #endif
1133   if (!fn || !child_stack)
1134     return -EINVAL;
1135   CHECK_EQ(0, (uptr)child_stack % 16);
1136
1137   register int (*__fn)(void *) __asm__("r3") = fn;
1138   register void *__cstack      __asm__("r4") = child_stack;
1139   register int __flags         __asm__("r5") = flags;
1140   register void *__arg         __asm__("r6") = arg;
1141   register int *__ptidptr      __asm__("r7") = parent_tidptr;
1142   register void *__newtls      __asm__("r8") = newtls;
1143   register int *__ctidptr      __asm__("r9") = child_tidptr;
1144
1145  __asm__ __volatile__(
1146            /* fn and arg are saved across the syscall */
1147            "mr 28, %5\n\t"
1148            "mr 27, %8\n\t"
1149
1150            /* syscall
1151              r0 == __NR_clone
1152              r3 == flags
1153              r4 == child_stack
1154              r5 == parent_tidptr
1155              r6 == newtls
1156              r7 == child_tidptr */
1157            "mr 3, %7\n\t"
1158            "mr 5, %9\n\t"
1159            "mr 6, %10\n\t"
1160            "mr 7, %11\n\t"
1161            "li 0, %3\n\t"
1162            "sc\n\t"
1163
1164            /* Test if syscall was successful */
1165            "cmpdi  cr1, 3, 0\n\t"
1166            "crandc cr1*4+eq, cr1*4+eq, cr0*4+so\n\t"
1167            "bne-   cr1, 1f\n\t"
1168
1169            /* Set up stack frame */
1170            "li    29, 0\n\t"
1171            "stdu  29, -8(1)\n\t"
1172            "stdu  1, -%12(1)\n\t"
1173            /* Do the function call */
1174            "std   2, %13(1)\n\t"
1175 #if SANITIZER_PPC64V1
1176            "ld    0, 0(28)\n\t"
1177            "ld    2, 8(28)\n\t"
1178            "mtctr 0\n\t"
1179 #elif SANITIZER_PPC64V2
1180            "mr    12, 28\n\t"
1181            "mtctr 12\n\t"
1182 #else
1183 # error "Unsupported PPC64 ABI"
1184 #endif
1185            "mr    3, 27\n\t"
1186            "bctrl\n\t"
1187            "ld    2, %13(1)\n\t"
1188
1189            /* Call _exit(r3) */
1190            "li 0, %4\n\t"
1191            "sc\n\t"
1192
1193            /* Return to parent */
1194            "1:\n\t"
1195            "mr %0, 3\n\t"
1196              : "=r" (res)
1197              : "0" (-1),
1198                "i" (EINVAL),
1199                "i" (__NR_clone),
1200                "i" (__NR_exit),
1201                "r" (__fn),
1202                "r" (__cstack),
1203                "r" (__flags),
1204                "r" (__arg),
1205                "r" (__ptidptr),
1206                "r" (__newtls),
1207                "r" (__ctidptr),
1208                "i" (FRAME_SIZE),
1209                "i" (FRAME_TOC_SAVE_OFFSET)
1210              : "cr0", "cr1", "memory", "ctr", "r0", "r27", "r28", "r29");
1211   return res;
1212 }
1213 #elif defined(__i386__) && SANITIZER_LINUX
1214 uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1215                     int *parent_tidptr, void *newtls, int *child_tidptr) {
1216   int res;
1217   if (!fn || !child_stack)
1218     return -EINVAL;
1219   CHECK_EQ(0, (uptr)child_stack % 16);
1220   child_stack = (char *)child_stack - 7 * sizeof(unsigned int);
1221   ((unsigned int *)child_stack)[0] = (uptr)flags;
1222   ((unsigned int *)child_stack)[1] = (uptr)0;
1223   ((unsigned int *)child_stack)[2] = (uptr)fn;
1224   ((unsigned int *)child_stack)[3] = (uptr)arg;
1225   __asm__ __volatile__(
1226                        /* %eax = syscall(%eax = SYSCALL(clone),
1227                         *                %ebx = flags,
1228                         *                %ecx = child_stack,
1229                         *                %edx = parent_tidptr,
1230                         *                %esi  = new_tls,
1231                         *                %edi = child_tidptr)
1232                         */
1233
1234                         /* Obtain flags */
1235                         "movl    (%%ecx), %%ebx\n"
1236                         /* Do the system call */
1237                         "pushl   %%ebx\n"
1238                         "pushl   %%esi\n"
1239                         "pushl   %%edi\n"
1240                         /* Remember the flag value.  */
1241                         "movl    %%ebx, (%%ecx)\n"
1242                         "int     $0x80\n"
1243                         "popl    %%edi\n"
1244                         "popl    %%esi\n"
1245                         "popl    %%ebx\n"
1246
1247                         /* if (%eax != 0)
1248                          *   return;
1249                          */
1250
1251                         "test    %%eax,%%eax\n"
1252                         "jnz    1f\n"
1253
1254                         /* terminate the stack frame */
1255                         "xorl   %%ebp,%%ebp\n"
1256                         /* Call FN. */
1257                         "call    *%%ebx\n"
1258 #ifdef PIC
1259                         "call    here\n"
1260                         "here:\n"
1261                         "popl    %%ebx\n"
1262                         "addl    $_GLOBAL_OFFSET_TABLE_+[.-here], %%ebx\n"
1263 #endif
1264                         /* Call exit */
1265                         "movl    %%eax, %%ebx\n"
1266                         "movl    %2, %%eax\n"
1267                         "int     $0x80\n"
1268                         "1:\n"
1269                        : "=a" (res)
1270                        : "a"(SYSCALL(clone)), "i"(SYSCALL(exit)),
1271                          "c"(child_stack),
1272                          "d"(parent_tidptr),
1273                          "S"(newtls),
1274                          "D"(child_tidptr)
1275                        : "memory");
1276   return res;
1277 }
1278 #elif defined(__arm__) && SANITIZER_LINUX
1279 uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1280                     int *parent_tidptr, void *newtls, int *child_tidptr) {
1281   unsigned int res;
1282   if (!fn || !child_stack)
1283     return -EINVAL;
1284   child_stack = (char *)child_stack - 2 * sizeof(unsigned int);
1285   ((unsigned int *)child_stack)[0] = (uptr)fn;
1286   ((unsigned int *)child_stack)[1] = (uptr)arg;
1287   register int r0 __asm__("r0") = flags;
1288   register void *r1 __asm__("r1") = child_stack;
1289   register int *r2 __asm__("r2") = parent_tidptr;
1290   register void *r3 __asm__("r3") = newtls;
1291   register int *r4 __asm__("r4") = child_tidptr;
1292   register int r7 __asm__("r7") = __NR_clone;
1293
1294 #if __ARM_ARCH > 4 || defined (__ARM_ARCH_4T__)
1295 # define ARCH_HAS_BX
1296 #endif
1297 #if __ARM_ARCH > 4
1298 # define ARCH_HAS_BLX
1299 #endif
1300
1301 #ifdef ARCH_HAS_BX
1302 # ifdef ARCH_HAS_BLX
1303 #  define BLX(R) "blx "  #R "\n"
1304 # else
1305 #  define BLX(R) "mov lr, pc; bx " #R "\n"
1306 # endif
1307 #else
1308 # define BLX(R)  "mov lr, pc; mov pc," #R "\n"
1309 #endif
1310
1311   __asm__ __volatile__(
1312                        /* %r0 = syscall(%r7 = SYSCALL(clone),
1313                         *               %r0 = flags,
1314                         *               %r1 = child_stack,
1315                         *               %r2 = parent_tidptr,
1316                         *               %r3  = new_tls,
1317                         *               %r4 = child_tidptr)
1318                         */
1319
1320                        /* Do the system call */
1321                        "swi 0x0\n"
1322
1323                        /* if (%r0 != 0)
1324                         *   return %r0;
1325                         */
1326                        "cmp r0, #0\n"
1327                        "bne 1f\n"
1328
1329                        /* In the child, now. Call "fn(arg)". */
1330                        "ldr r0, [sp, #4]\n"
1331                        "ldr ip, [sp], #8\n"
1332                        BLX(ip)
1333                        /* Call _exit(%r0). */
1334                        "mov r7, %7\n"
1335                        "swi 0x0\n"
1336                        "1:\n"
1337                        "mov %0, r0\n"
1338                        : "=r"(res)
1339                        : "r"(r0), "r"(r1), "r"(r2), "r"(r3), "r"(r4), "r"(r7),
1340                          "i"(__NR_exit)
1341                        : "memory");
1342   return res;
1343 }
1344 #endif  // defined(__x86_64__) && SANITIZER_LINUX
1345
1346 #if SANITIZER_ANDROID
1347 #if __ANDROID_API__ < 21
1348 extern "C" __attribute__((weak)) int dl_iterate_phdr(
1349     int (*)(struct dl_phdr_info *, size_t, void *), void *);
1350 #endif
1351
1352 static int dl_iterate_phdr_test_cb(struct dl_phdr_info *info, size_t size,
1353                                    void *data) {
1354   // Any name starting with "lib" indicates a bug in L where library base names
1355   // are returned instead of paths.
1356   if (info->dlpi_name && info->dlpi_name[0] == 'l' &&
1357       info->dlpi_name[1] == 'i' && info->dlpi_name[2] == 'b') {
1358     *(bool *)data = true;
1359     return 1;
1360   }
1361   return 0;
1362 }
1363
1364 static atomic_uint32_t android_api_level;
1365
1366 static AndroidApiLevel AndroidDetectApiLevel() {
1367   if (!&dl_iterate_phdr)
1368     return ANDROID_KITKAT; // K or lower
1369   bool base_name_seen = false;
1370   dl_iterate_phdr(dl_iterate_phdr_test_cb, &base_name_seen);
1371   if (base_name_seen)
1372     return ANDROID_LOLLIPOP_MR1; // L MR1
1373   return ANDROID_POST_LOLLIPOP;   // post-L
1374   // Plain L (API level 21) is completely broken wrt ASan and not very
1375   // interesting to detect.
1376 }
1377
1378 AndroidApiLevel AndroidGetApiLevel() {
1379   AndroidApiLevel level =
1380       (AndroidApiLevel)atomic_load(&android_api_level, memory_order_relaxed);
1381   if (level) return level;
1382   level = AndroidDetectApiLevel();
1383   atomic_store(&android_api_level, level, memory_order_relaxed);
1384   return level;
1385 }
1386
1387 #endif
1388
1389 HandleSignalMode GetHandleSignalMode(int signum) {
1390   switch (signum) {
1391     case SIGABRT:
1392       return common_flags()->handle_abort;
1393     case SIGILL:
1394       return common_flags()->handle_sigill;
1395     case SIGFPE:
1396       return common_flags()->handle_sigfpe;
1397     case SIGSEGV:
1398       return common_flags()->handle_segv;
1399     case SIGBUS:
1400       return common_flags()->handle_sigbus;
1401   }
1402   return kHandleSignalNo;
1403 }
1404
1405 #if !SANITIZER_GO
1406 void *internal_start_thread(void(*func)(void *arg), void *arg) {
1407   // Start the thread with signals blocked, otherwise it can steal user signals.
1408   __sanitizer_sigset_t set, old;
1409   internal_sigfillset(&set);
1410 #if SANITIZER_LINUX && !SANITIZER_ANDROID
1411   // Glibc uses SIGSETXID signal during setuid call. If this signal is blocked
1412   // on any thread, setuid call hangs (see test/tsan/setuid.c).
1413   internal_sigdelset(&set, 33);
1414 #endif
1415   internal_sigprocmask(SIG_SETMASK, &set, &old);
1416   void *th;
1417   real_pthread_create(&th, nullptr, (void*(*)(void *arg))func, arg);
1418   internal_sigprocmask(SIG_SETMASK, &old, nullptr);
1419   return th;
1420 }
1421
1422 void internal_join_thread(void *th) {
1423   real_pthread_join(th, nullptr);
1424 }
1425 #else
1426 void *internal_start_thread(void (*func)(void *), void *arg) { return 0; }
1427
1428 void internal_join_thread(void *th) {}
1429 #endif
1430
1431 #if defined(__aarch64__)
1432 // Android headers in the older NDK releases miss this definition.
1433 struct __sanitizer_esr_context {
1434   struct _aarch64_ctx head;
1435   uint64_t esr;
1436 };
1437
1438 static bool Aarch64GetESR(ucontext_t *ucontext, u64 *esr) {
1439   static const u32 kEsrMagic = 0x45535201;
1440   u8 *aux = ucontext->uc_mcontext.__reserved;
1441   while (true) {
1442     _aarch64_ctx *ctx = (_aarch64_ctx *)aux;
1443     if (ctx->size == 0) break;
1444     if (ctx->magic == kEsrMagic) {
1445       *esr = ((__sanitizer_esr_context *)ctx)->esr;
1446       return true;
1447     }
1448     aux += ctx->size;
1449   }
1450   return false;
1451 }
1452 #endif
1453
1454 SignalContext::WriteFlag SignalContext::GetWriteFlag(void *context) {
1455   ucontext_t *ucontext = (ucontext_t *)context;
1456 #if defined(__x86_64__) || defined(__i386__)
1457   static const uptr PF_WRITE = 1U << 1;
1458 #if SANITIZER_FREEBSD
1459   uptr err = ucontext->uc_mcontext.mc_err;
1460 #else
1461   uptr err = ucontext->uc_mcontext.gregs[REG_ERR];
1462 #endif
1463   return err & PF_WRITE ? WRITE : READ;
1464 #elif defined(__arm__)
1465   static const uptr FSR_WRITE = 1U << 11;
1466   uptr fsr = ucontext->uc_mcontext.error_code;
1467   return fsr & FSR_WRITE ? WRITE : READ;
1468 #elif defined(__aarch64__)
1469   static const u64 ESR_ELx_WNR = 1U << 6;
1470   u64 esr;
1471   if (!Aarch64GetESR(ucontext, &esr)) return UNKNOWN;
1472   return esr & ESR_ELx_WNR ? WRITE : READ;
1473 #else
1474   (void)ucontext;
1475   return UNKNOWN;  // FIXME: Implement.
1476 #endif
1477 }
1478
1479 void SignalContext::DumpAllRegisters(void *context) {
1480   // FIXME: Implement this.
1481 }
1482
1483 void GetPcSpBp(void *context, uptr *pc, uptr *sp, uptr *bp) {
1484 #if defined(__arm__)
1485   ucontext_t *ucontext = (ucontext_t*)context;
1486   *pc = ucontext->uc_mcontext.arm_pc;
1487   *bp = ucontext->uc_mcontext.arm_fp;
1488   *sp = ucontext->uc_mcontext.arm_sp;
1489 #elif defined(__aarch64__)
1490   ucontext_t *ucontext = (ucontext_t*)context;
1491   *pc = ucontext->uc_mcontext.pc;
1492   *bp = ucontext->uc_mcontext.regs[29];
1493   *sp = ucontext->uc_mcontext.sp;
1494 #elif defined(__hppa__)
1495   ucontext_t *ucontext = (ucontext_t*)context;
1496   *pc = ucontext->uc_mcontext.sc_iaoq[0];
1497   /* GCC uses %r3 whenever a frame pointer is needed.  */
1498   *bp = ucontext->uc_mcontext.sc_gr[3];
1499   *sp = ucontext->uc_mcontext.sc_gr[30];
1500 #elif defined(__x86_64__)
1501 # if SANITIZER_FREEBSD
1502   ucontext_t *ucontext = (ucontext_t*)context;
1503   *pc = ucontext->uc_mcontext.mc_rip;
1504   *bp = ucontext->uc_mcontext.mc_rbp;
1505   *sp = ucontext->uc_mcontext.mc_rsp;
1506 # else
1507   ucontext_t *ucontext = (ucontext_t*)context;
1508   *pc = ucontext->uc_mcontext.gregs[REG_RIP];
1509   *bp = ucontext->uc_mcontext.gregs[REG_RBP];
1510   *sp = ucontext->uc_mcontext.gregs[REG_RSP];
1511 # endif
1512 #elif defined(__i386__)
1513 # if SANITIZER_FREEBSD
1514   ucontext_t *ucontext = (ucontext_t*)context;
1515   *pc = ucontext->uc_mcontext.mc_eip;
1516   *bp = ucontext->uc_mcontext.mc_ebp;
1517   *sp = ucontext->uc_mcontext.mc_esp;
1518 # else
1519   ucontext_t *ucontext = (ucontext_t*)context;
1520   *pc = ucontext->uc_mcontext.gregs[REG_EIP];
1521   *bp = ucontext->uc_mcontext.gregs[REG_EBP];
1522   *sp = ucontext->uc_mcontext.gregs[REG_ESP];
1523 # endif
1524 #elif defined(__powerpc__) || defined(__powerpc64__)
1525   ucontext_t *ucontext = (ucontext_t*)context;
1526   *pc = ucontext->uc_mcontext.regs->nip;
1527   *sp = ucontext->uc_mcontext.regs->gpr[PT_R1];
1528   // The powerpc{,64}-linux ABIs do not specify r31 as the frame
1529   // pointer, but GCC always uses r31 when we need a frame pointer.
1530   *bp = ucontext->uc_mcontext.regs->gpr[PT_R31];
1531 #elif defined(__sparc__)
1532   ucontext_t *ucontext = (ucontext_t*)context;
1533   uptr *stk_ptr;
1534 # if defined (__arch64__)
1535   *pc = ucontext->uc_mcontext.mc_gregs[MC_PC];
1536   *sp = ucontext->uc_mcontext.mc_gregs[MC_O6];
1537   stk_ptr = (uptr *) (*sp + 2047);
1538   *bp = stk_ptr[15];
1539 # else
1540   *pc = ucontext->uc_mcontext.gregs[REG_PC];
1541   *sp = ucontext->uc_mcontext.gregs[REG_O6];
1542   stk_ptr = (uptr *) *sp;
1543   *bp = stk_ptr[15];
1544 # endif
1545 #elif defined(__mips__)
1546   ucontext_t *ucontext = (ucontext_t*)context;
1547   *pc = ucontext->uc_mcontext.pc;
1548   *bp = ucontext->uc_mcontext.gregs[30];
1549   *sp = ucontext->uc_mcontext.gregs[29];
1550 #elif defined(__s390__)
1551   ucontext_t *ucontext = (ucontext_t*)context;
1552 # if defined(__s390x__)
1553   *pc = ucontext->uc_mcontext.psw.addr;
1554 # else
1555   *pc = ucontext->uc_mcontext.psw.addr & 0x7fffffff;
1556 # endif
1557   *bp = ucontext->uc_mcontext.gregs[11];
1558   *sp = ucontext->uc_mcontext.gregs[15];
1559 #else
1560 # error "Unsupported arch"
1561 #endif
1562 }
1563
1564 void MaybeReexec() {
1565   // No need to re-exec on Linux.
1566 }
1567
1568 void PrintModuleMap() { }
1569
1570 void CheckNoDeepBind(const char *filename, int flag) {
1571 #ifdef RTLD_DEEPBIND
1572   if (flag & RTLD_DEEPBIND) {
1573     Report(
1574         "You are trying to dlopen a %s shared library with RTLD_DEEPBIND flag"
1575         " which is incompatibe with sanitizer runtime "
1576         "(see https://github.com/google/sanitizers/issues/611 for details"
1577         "). If you want to run %s library under sanitizers please remove "
1578         "RTLD_DEEPBIND from dlopen flags.\n",
1579         filename, filename);
1580     Die();
1581   }
1582 #endif
1583 }
1584
1585 uptr FindAvailableMemoryRange(uptr size, uptr alignment, uptr left_padding) {
1586   UNREACHABLE("FindAvailableMemoryRange is not available");
1587   return 0;
1588 }
1589
1590 } // namespace __sanitizer
1591
1592 #endif // SANITIZER_FREEBSD || SANITIZER_LINUX