1 //===-- sanitizer_linux.cc ------------------------------------------------===//
2 //
3 // This file is distributed under the University of Illinois Open Source
4 // License. See LICENSE.TXT for details.
5 //
6 //===----------------------------------------------------------------------===//
7 //
8 // This file is shared between AddressSanitizer and ThreadSanitizer
9 // run-time libraries and implements linux-specific functions from
10 // sanitizer_libc.h.
11 //===----------------------------------------------------------------------===//
12 
13 #include "sanitizer_platform.h"
14 
15 #if SANITIZER_FREEBSD || SANITIZER_LINUX
16 
17 #include "sanitizer_common.h"
18 #include "sanitizer_flags.h"
19 #include "sanitizer_internal_defs.h"
20 #include "sanitizer_libc.h"
21 #include "sanitizer_linux.h"
22 #include "sanitizer_mutex.h"
23 #include "sanitizer_placement_new.h"
24 #include "sanitizer_procmaps.h"
25 #include "sanitizer_stacktrace.h"
26 #include "sanitizer_symbolizer.h"
27 
28 #if !SANITIZER_FREEBSD
29 #include <asm/param.h>
30 #endif
31 
32 // For mips64, syscall(__NR_stat) fills the buffer in the 'struct kernel_stat'
33 // format. Struct kernel_stat is defined as 'struct stat' in asm/stat.h. To
34 // access stat from asm/stat.h, without conflicting with definition in
35 // sys/stat.h, we use this trick.
36 #if defined(__mips64)
37 #include <asm/unistd.h>
38 #include <sys/types.h>
39 #define stat kernel_stat
40 #include <asm/stat.h>
41 #undef stat
42 #endif
43 
44 #include <dlfcn.h>
45 #include <errno.h>
46 #include <fcntl.h>
47 #include <link.h>
48 #include <pthread.h>
49 #include <sched.h>
50 #include <sys/mman.h>
51 #include <sys/ptrace.h>
52 #include <sys/resource.h>
53 #include <sys/stat.h>
54 #include <sys/syscall.h>
55 #include <sys/time.h>
56 #include <sys/types.h>
57 #include <ucontext.h>
58 #include <unistd.h>
59 
60 #if SANITIZER_FREEBSD
61 #include <sys/sysctl.h>
62 #include <machine/atomic.h>
63 extern "C" {
64 // <sys/umtx.h> must be included after <errno.h> and <sys/types.h> on
65 // FreeBSD 9.2 and 10.0.
66 #include <sys/umtx.h>
67 }
68 extern char **environ;  // provided by crt1
69 #endif  // SANITIZER_FREEBSD
70 
71 #if !SANITIZER_ANDROID
72 #include <sys/signal.h>
73 #endif
74 
75 #if SANITIZER_LINUX
76 // <linux/time.h>
77 struct kernel_timeval {
78   long tv_sec;
79   long tv_usec;
80 };
81 
82 // <linux/futex.h> is broken on some linux distributions.
83 const int FUTEX_WAIT = 0;
84 const int FUTEX_WAKE = 1;
85 #endif  // SANITIZER_LINUX
86 
87 // Are we using 32-bit or 64-bit Linux syscalls?
88 // x32 (which defines __x86_64__) has SANITIZER_WORDSIZE == 32
89 // but it still needs to use 64-bit syscalls.
90 #if SANITIZER_LINUX && (defined(__x86_64__) || SANITIZER_WORDSIZE == 64)
91 # define SANITIZER_LINUX_USES_64BIT_SYSCALLS 1
92 #else
93 # define SANITIZER_LINUX_USES_64BIT_SYSCALLS 0
94 #endif
95 
96 namespace __sanitizer {
97 
98 #if SANITIZER_LINUX && defined(__x86_64__)
99 #include "sanitizer_syscall_linux_x86_64.inc"
100 #elif SANITIZER_LINUX && defined(__aarch64__)
101 #include "sanitizer_syscall_linux_aarch64.inc"
102 #else
103 #include "sanitizer_syscall_generic.inc"
104 #endif
105 
106 // --------------- sanitizer_libc.h
internal_mmap(void * addr,uptr length,int prot,int flags,int fd,OFF_T offset)107 uptr internal_mmap(void *addr, uptr length, int prot, int flags, int fd,
108                    OFF_T offset) {
109 #if SANITIZER_FREEBSD || SANITIZER_LINUX_USES_64BIT_SYSCALLS
110   return internal_syscall(SYSCALL(mmap), (uptr)addr, length, prot, flags, fd,
111                           offset);
112 #else
113   // mmap2 specifies file offset in 4096-byte units.
114   CHECK(IsAligned(offset, 4096));
115   return internal_syscall(SYSCALL(mmap2), addr, length, prot, flags, fd,
116                           offset / 4096);
117 #endif
118 }
119 
internal_munmap(void * addr,uptr length)120 uptr internal_munmap(void *addr, uptr length) {
121   return internal_syscall(SYSCALL(munmap), (uptr)addr, length);
122 }
123 
internal_mprotect(void * addr,uptr length,int prot)124 int internal_mprotect(void *addr, uptr length, int prot) {
125   return internal_syscall(SYSCALL(mprotect), (uptr)addr, length, prot);
126 }
127 
internal_close(fd_t fd)128 uptr internal_close(fd_t fd) {
129   return internal_syscall(SYSCALL(close), fd);
130 }
131 
internal_open(const char * filename,int flags)132 uptr internal_open(const char *filename, int flags) {
133 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
134   return internal_syscall(SYSCALL(openat), AT_FDCWD, (uptr)filename, flags);
135 #else
136   return internal_syscall(SYSCALL(open), (uptr)filename, flags);
137 #endif
138 }
139 
internal_open(const char * filename,int flags,u32 mode)140 uptr internal_open(const char *filename, int flags, u32 mode) {
141 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
142   return internal_syscall(SYSCALL(openat), AT_FDCWD, (uptr)filename, flags,
143                           mode);
144 #else
145   return internal_syscall(SYSCALL(open), (uptr)filename, flags, mode);
146 #endif
147 }
148 
internal_read(fd_t fd,void * buf,uptr count)149 uptr internal_read(fd_t fd, void *buf, uptr count) {
150   sptr res;
151   HANDLE_EINTR(res, (sptr)internal_syscall(SYSCALL(read), fd, (uptr)buf,
152                count));
153   return res;
154 }
155 
internal_write(fd_t fd,const void * buf,uptr count)156 uptr internal_write(fd_t fd, const void *buf, uptr count) {
157   sptr res;
158   HANDLE_EINTR(res, (sptr)internal_syscall(SYSCALL(write), fd, (uptr)buf,
159                count));
160   return res;
161 }
162 
internal_ftruncate(fd_t fd,uptr size)163 uptr internal_ftruncate(fd_t fd, uptr size) {
164   sptr res;
165   HANDLE_EINTR(res, (sptr)internal_syscall(SYSCALL(ftruncate), fd,
166                (OFF_T)size));
167   return res;
168 }
169 
170 #if !SANITIZER_LINUX_USES_64BIT_SYSCALLS && !SANITIZER_FREEBSD
stat64_to_stat(struct stat64 * in,struct stat * out)171 static void stat64_to_stat(struct stat64 *in, struct stat *out) {
172   internal_memset(out, 0, sizeof(*out));
173   out->st_dev = in->st_dev;
174   out->st_ino = in->st_ino;
175   out->st_mode = in->st_mode;
176   out->st_nlink = in->st_nlink;
177   out->st_uid = in->st_uid;
178   out->st_gid = in->st_gid;
179   out->st_rdev = in->st_rdev;
180   out->st_size = in->st_size;
181   out->st_blksize = in->st_blksize;
182   out->st_blocks = in->st_blocks;
183   out->st_atime = in->st_atime;
184   out->st_mtime = in->st_mtime;
185   out->st_ctime = in->st_ctime;
186   out->st_ino = in->st_ino;
187 }
188 #endif
189 
190 #if defined(__mips64)
kernel_stat_to_stat(struct kernel_stat * in,struct stat * out)191 static void kernel_stat_to_stat(struct kernel_stat *in, struct stat *out) {
192   internal_memset(out, 0, sizeof(*out));
193   out->st_dev = in->st_dev;
194   out->st_ino = in->st_ino;
195   out->st_mode = in->st_mode;
196   out->st_nlink = in->st_nlink;
197   out->st_uid = in->st_uid;
198   out->st_gid = in->st_gid;
199   out->st_rdev = in->st_rdev;
200   out->st_size = in->st_size;
201   out->st_blksize = in->st_blksize;
202   out->st_blocks = in->st_blocks;
203   out->st_atime = in->st_atime_nsec;
204   out->st_mtime = in->st_mtime_nsec;
205   out->st_ctime = in->st_ctime_nsec;
206   out->st_ino = in->st_ino;
207 }
208 #endif
209 
internal_stat(const char * path,void * buf)210 uptr internal_stat(const char *path, void *buf) {
211 #if SANITIZER_FREEBSD
212   return internal_syscall(SYSCALL(stat), path, buf);
213 #elif SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
214   return internal_syscall(SYSCALL(newfstatat), AT_FDCWD, (uptr)path,
215                           (uptr)buf, 0);
216 #elif SANITIZER_LINUX_USES_64BIT_SYSCALLS
217 # if defined(__mips64)
218   // For mips64, stat syscall fills buffer in the format of kernel_stat
219   struct kernel_stat kbuf;
220   int res = internal_syscall(SYSCALL(stat), path, &kbuf);
221   kernel_stat_to_stat(&kbuf, (struct stat *)buf);
222   return res;
223 # else
224   return internal_syscall(SYSCALL(stat), (uptr)path, (uptr)buf);
225 # endif
226 #else
227   struct stat64 buf64;
228   int res = internal_syscall(SYSCALL(stat64), path, &buf64);
229   stat64_to_stat(&buf64, (struct stat *)buf);
230   return res;
231 #endif
232 }
233 
internal_lstat(const char * path,void * buf)234 uptr internal_lstat(const char *path, void *buf) {
235 #if SANITIZER_FREEBSD
236   return internal_syscall(SYSCALL(lstat), path, buf);
237 #elif SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
238   return internal_syscall(SYSCALL(newfstatat), AT_FDCWD, (uptr)path,
239                          (uptr)buf, AT_SYMLINK_NOFOLLOW);
240 #elif SANITIZER_LINUX_USES_64BIT_SYSCALLS
241   return internal_syscall(SYSCALL(lstat), (uptr)path, (uptr)buf);
242 #else
243   struct stat64 buf64;
244   int res = internal_syscall(SYSCALL(lstat64), path, &buf64);
245   stat64_to_stat(&buf64, (struct stat *)buf);
246   return res;
247 #endif
248 }
249 
internal_fstat(fd_t fd,void * buf)250 uptr internal_fstat(fd_t fd, void *buf) {
251 #if SANITIZER_FREEBSD || SANITIZER_LINUX_USES_64BIT_SYSCALLS
252   return internal_syscall(SYSCALL(fstat), fd, (uptr)buf);
253 #else
254   struct stat64 buf64;
255   int res = internal_syscall(SYSCALL(fstat64), fd, &buf64);
256   stat64_to_stat(&buf64, (struct stat *)buf);
257   return res;
258 #endif
259 }
260 
internal_filesize(fd_t fd)261 uptr internal_filesize(fd_t fd) {
262   struct stat st;
263   if (internal_fstat(fd, &st))
264     return -1;
265   return (uptr)st.st_size;
266 }
267 
internal_dup2(int oldfd,int newfd)268 uptr internal_dup2(int oldfd, int newfd) {
269 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
270   return internal_syscall(SYSCALL(dup3), oldfd, newfd, 0);
271 #else
272   return internal_syscall(SYSCALL(dup2), oldfd, newfd);
273 #endif
274 }
275 
internal_readlink(const char * path,char * buf,uptr bufsize)276 uptr internal_readlink(const char *path, char *buf, uptr bufsize) {
277 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
278   return internal_syscall(SYSCALL(readlinkat), AT_FDCWD,
279                           (uptr)path, (uptr)buf, bufsize);
280 #else
281   return internal_syscall(SYSCALL(readlink), (uptr)path, (uptr)buf, bufsize);
282 #endif
283 }
284 
internal_unlink(const char * path)285 uptr internal_unlink(const char *path) {
286 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
287   return internal_syscall(SYSCALL(unlinkat), AT_FDCWD, (uptr)path, 0);
288 #else
289   return internal_syscall(SYSCALL(unlink), (uptr)path);
290 #endif
291 }
292 
internal_rename(const char * oldpath,const char * newpath)293 uptr internal_rename(const char *oldpath, const char *newpath) {
294 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
295   return internal_syscall(SYSCALL(renameat), AT_FDCWD, (uptr)oldpath, AT_FDCWD,
296                           (uptr)newpath);
297 #else
298   return internal_syscall(SYSCALL(rename), (uptr)oldpath, (uptr)newpath);
299 #endif
300 }
301 
internal_sched_yield()302 uptr internal_sched_yield() {
303   return internal_syscall(SYSCALL(sched_yield));
304 }
305 
internal__exit(int exitcode)306 void internal__exit(int exitcode) {
307 #if SANITIZER_FREEBSD
308   internal_syscall(SYSCALL(exit), exitcode);
309 #else
310   internal_syscall(SYSCALL(exit_group), exitcode);
311 #endif
312   Die();  // Unreachable.
313 }
314 
internal_execve(const char * filename,char * const argv[],char * const envp[])315 uptr internal_execve(const char *filename, char *const argv[],
316                      char *const envp[]) {
317   return internal_syscall(SYSCALL(execve), (uptr)filename, (uptr)argv,
318                           (uptr)envp);
319 }
320 
321 // ----------------- sanitizer_common.h
FileExists(const char * filename)322 bool FileExists(const char *filename) {
323   struct stat st;
324 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
325   if (internal_syscall(SYSCALL(newfstatat), AT_FDCWD, filename, &st, 0))
326 #else
327   if (internal_stat(filename, &st))
328 #endif
329     return false;
330   // Sanity check: filename is a regular file.
331   return S_ISREG(st.st_mode);
332 }
333 
GetTid()334 uptr GetTid() {
335 #if SANITIZER_FREEBSD
336   return (uptr)pthread_self();
337 #else
338   return internal_syscall(SYSCALL(gettid));
339 #endif
340 }
341 
NanoTime()342 u64 NanoTime() {
343 #if SANITIZER_FREEBSD
344   timeval tv;
345 #else
346   kernel_timeval tv;
347 #endif
348   internal_memset(&tv, 0, sizeof(tv));
349   internal_syscall(SYSCALL(gettimeofday), (uptr)&tv, 0);
350   return (u64)tv.tv_sec * 1000*1000*1000 + tv.tv_usec * 1000;
351 }
352 
353 // Like getenv, but reads env directly from /proc (on Linux) or parses the
354 // 'environ' array (on FreeBSD) and does not use libc. This function should be
355 // called first inside __asan_init.
GetEnv(const char * name)356 const char *GetEnv(const char *name) {
357 #if SANITIZER_FREEBSD
358   if (::environ != 0) {
359     uptr NameLen = internal_strlen(name);
360     for (char **Env = ::environ; *Env != 0; Env++) {
361       if (internal_strncmp(*Env, name, NameLen) == 0 && (*Env)[NameLen] == '=')
362         return (*Env) + NameLen + 1;
363     }
364   }
365   return 0;  // Not found.
366 #elif SANITIZER_LINUX
367   static char *environ;
368   static uptr len;
369   static bool inited;
370   if (!inited) {
371     inited = true;
372     uptr environ_size;
373     if (!ReadFileToBuffer("/proc/self/environ", &environ, &environ_size, &len))
374       environ = nullptr;
375   }
376   if (!environ || len == 0) return nullptr;
377   uptr namelen = internal_strlen(name);
378   const char *p = environ;
379   while (*p != '\0') {  // will happen at the \0\0 that terminates the buffer
380     // proc file has the format NAME=value\0NAME=value\0NAME=value\0...
381     const char* endp =
382         (char*)internal_memchr(p, '\0', len - (p - environ));
383     if (!endp)  // this entry isn't NUL terminated
384       return nullptr;
385     else if (!internal_memcmp(p, name, namelen) && p[namelen] == '=')  // Match.
386       return p + namelen + 1;  // point after =
387     p = endp + 1;
388   }
389   return nullptr;  // Not found.
390 #else
391 #error "Unsupported platform"
392 #endif
393 }
394 
395 extern "C" {
396   SANITIZER_WEAK_ATTRIBUTE extern void *__libc_stack_end;
397 }
398 
399 #if !SANITIZER_GO
ReadNullSepFileToArray(const char * path,char *** arr,int arr_size)400 static void ReadNullSepFileToArray(const char *path, char ***arr,
401                                    int arr_size) {
402   char *buff;
403   uptr buff_size;
404   uptr buff_len;
405   *arr = (char **)MmapOrDie(arr_size * sizeof(char *), "NullSepFileArray");
406   if (!ReadFileToBuffer(path, &buff, &buff_size, &buff_len, 1024 * 1024)) {
407     (*arr)[0] = nullptr;
408     return;
409   }
410   (*arr)[0] = buff;
411   int count, i;
412   for (count = 1, i = 1; ; i++) {
413     if (buff[i] == 0) {
414       if (buff[i+1] == 0) break;
415       (*arr)[count] = &buff[i+1];
416       CHECK_LE(count, arr_size - 1);  // FIXME: make this more flexible.
417       count++;
418     }
419   }
420   (*arr)[count] = nullptr;
421 }
422 #endif
423 
GetArgsAndEnv(char *** argv,char *** envp)424 static void GetArgsAndEnv(char*** argv, char*** envp) {
425 #if !SANITIZER_GO
426   if (&__libc_stack_end) {
427 #endif
428     uptr* stack_end = (uptr*)__libc_stack_end;
429     int argc = *stack_end;
430     *argv = (char**)(stack_end + 1);
431     *envp = (char**)(stack_end + argc + 2);
432 #if !SANITIZER_GO
433   } else {
434     static const int kMaxArgv = 2000, kMaxEnvp = 2000;
435     ReadNullSepFileToArray("/proc/self/cmdline", argv, kMaxArgv);
436     ReadNullSepFileToArray("/proc/self/environ", envp, kMaxEnvp);
437   }
438 #endif
439 }
440 
ReExec()441 void ReExec() {
442   char **argv, **envp;
443   GetArgsAndEnv(&argv, &envp);
444   uptr rv = internal_execve("/proc/self/exe", argv, envp);
445   int rverrno;
446   CHECK_EQ(internal_iserror(rv, &rverrno), true);
447   Printf("execve failed, errno %d\n", rverrno);
448   Die();
449 }
450 
451 enum MutexState {
452   MtxUnlocked = 0,
453   MtxLocked = 1,
454   MtxSleeping = 2
455 };
456 
BlockingMutex()457 BlockingMutex::BlockingMutex() {
458   internal_memset(this, 0, sizeof(*this));
459 }
460 
Lock()461 void BlockingMutex::Lock() {
462   CHECK_EQ(owner_, 0);
463   atomic_uint32_t *m = reinterpret_cast<atomic_uint32_t *>(&opaque_storage_);
464   if (atomic_exchange(m, MtxLocked, memory_order_acquire) == MtxUnlocked)
465     return;
466   while (atomic_exchange(m, MtxSleeping, memory_order_acquire) != MtxUnlocked) {
467 #if SANITIZER_FREEBSD
468     _umtx_op(m, UMTX_OP_WAIT_UINT, MtxSleeping, 0, 0);
469 #else
470     internal_syscall(SYSCALL(futex), (uptr)m, FUTEX_WAIT, MtxSleeping, 0, 0, 0);
471 #endif
472   }
473 }
474 
Unlock()475 void BlockingMutex::Unlock() {
476   atomic_uint32_t *m = reinterpret_cast<atomic_uint32_t *>(&opaque_storage_);
477   u32 v = atomic_exchange(m, MtxUnlocked, memory_order_relaxed);
478   CHECK_NE(v, MtxUnlocked);
479   if (v == MtxSleeping) {
480 #if SANITIZER_FREEBSD
481     _umtx_op(m, UMTX_OP_WAKE, 1, 0, 0);
482 #else
483     internal_syscall(SYSCALL(futex), (uptr)m, FUTEX_WAKE, 1, 0, 0, 0);
484 #endif
485   }
486 }
487 
CheckLocked()488 void BlockingMutex::CheckLocked() {
489   atomic_uint32_t *m = reinterpret_cast<atomic_uint32_t *>(&opaque_storage_);
490   CHECK_NE(MtxUnlocked, atomic_load(m, memory_order_relaxed));
491 }
492 
493 // ----------------- sanitizer_linux.h
494 // The actual size of this structure is specified by d_reclen.
495 // Note that getdents64 uses a different structure format. We only provide the
496 // 32-bit syscall here.
497 struct linux_dirent {
498 #if SANITIZER_X32 || defined(__aarch64__)
499   u64 d_ino;
500   u64 d_off;
501 #else
502   unsigned long      d_ino;
503   unsigned long      d_off;
504 #endif
505   unsigned short     d_reclen;
506 #ifdef __aarch64__
507   unsigned char      d_type;
508 #endif
509   char               d_name[256];
510 };
511 
512 // Syscall wrappers.
internal_ptrace(int request,int pid,void * addr,void * data)513 uptr internal_ptrace(int request, int pid, void *addr, void *data) {
514   return internal_syscall(SYSCALL(ptrace), request, pid, (uptr)addr,
515                           (uptr)data);
516 }
517 
internal_waitpid(int pid,int * status,int options)518 uptr internal_waitpid(int pid, int *status, int options) {
519   return internal_syscall(SYSCALL(wait4), pid, (uptr)status, options,
520                           0 /* rusage */);
521 }
522 
internal_getpid()523 uptr internal_getpid() {
524   return internal_syscall(SYSCALL(getpid));
525 }
526 
internal_getppid()527 uptr internal_getppid() {
528   return internal_syscall(SYSCALL(getppid));
529 }
530 
internal_getdents(fd_t fd,struct linux_dirent * dirp,unsigned int count)531 uptr internal_getdents(fd_t fd, struct linux_dirent *dirp, unsigned int count) {
532 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
533   return internal_syscall(SYSCALL(getdents64), fd, (uptr)dirp, count);
534 #else
535   return internal_syscall(SYSCALL(getdents), fd, (uptr)dirp, count);
536 #endif
537 }
538 
internal_lseek(fd_t fd,OFF_T offset,int whence)539 uptr internal_lseek(fd_t fd, OFF_T offset, int whence) {
540   return internal_syscall(SYSCALL(lseek), fd, offset, whence);
541 }
542 
543 #if SANITIZER_LINUX
internal_prctl(int option,uptr arg2,uptr arg3,uptr arg4,uptr arg5)544 uptr internal_prctl(int option, uptr arg2, uptr arg3, uptr arg4, uptr arg5) {
545   return internal_syscall(SYSCALL(prctl), option, arg2, arg3, arg4, arg5);
546 }
547 #endif
548 
internal_sigaltstack(const void * ss,void * oss)549 uptr internal_sigaltstack(const void *ss, void *oss) {
550   return internal_syscall(SYSCALL(sigaltstack), (uptr)ss, (uptr)oss);
551 }
552 
internal_fork()553 int internal_fork() {
554 #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
555   return internal_syscall(SYSCALL(clone), SIGCHLD, 0);
556 #else
557   return internal_syscall(SYSCALL(fork));
558 #endif
559 }
560 
561 #if SANITIZER_LINUX
562 #define SA_RESTORER 0x04000000
563 // Doesn't set sa_restorer, use with caution (see below).
internal_sigaction_norestorer(int signum,const void * act,void * oldact)564 int internal_sigaction_norestorer(int signum, const void *act, void *oldact) {
565   __sanitizer_kernel_sigaction_t k_act, k_oldact;
566   internal_memset(&k_act, 0, sizeof(__sanitizer_kernel_sigaction_t));
567   internal_memset(&k_oldact, 0, sizeof(__sanitizer_kernel_sigaction_t));
568   const __sanitizer_sigaction *u_act = (const __sanitizer_sigaction *)act;
569   __sanitizer_sigaction *u_oldact = (__sanitizer_sigaction *)oldact;
570   if (u_act) {
571     k_act.handler = u_act->handler;
572     k_act.sigaction = u_act->sigaction;
573     internal_memcpy(&k_act.sa_mask, &u_act->sa_mask,
574                     sizeof(__sanitizer_kernel_sigset_t));
575     // Without SA_RESTORER kernel ignores the calls (probably returns EINVAL).
576     k_act.sa_flags = u_act->sa_flags | SA_RESTORER;
577     // FIXME: most often sa_restorer is unset, however the kernel requires it
578     // to point to a valid signal restorer that calls the rt_sigreturn syscall.
579     // If sa_restorer passed to the kernel is NULL, the program may crash upon
580     // signal delivery or fail to unwind the stack in the signal handler.
581     // libc implementation of sigaction() passes its own restorer to
582     // rt_sigaction, so we need to do the same (we'll need to reimplement the
583     // restorers; for x86_64 the restorer address can be obtained from
584     // oldact->sa_restorer upon a call to sigaction(xxx, NULL, oldact).
585     k_act.sa_restorer = u_act->sa_restorer;
586   }
587 
588   uptr result = internal_syscall(SYSCALL(rt_sigaction), (uptr)signum,
589       (uptr)(u_act ? &k_act : nullptr),
590       (uptr)(u_oldact ? &k_oldact : nullptr),
591       (uptr)sizeof(__sanitizer_kernel_sigset_t));
592 
593   if ((result == 0) && u_oldact) {
594     u_oldact->handler = k_oldact.handler;
595     u_oldact->sigaction = k_oldact.sigaction;
596     internal_memcpy(&u_oldact->sa_mask, &k_oldact.sa_mask,
597                     sizeof(__sanitizer_kernel_sigset_t));
598     u_oldact->sa_flags = k_oldact.sa_flags;
599     u_oldact->sa_restorer = k_oldact.sa_restorer;
600   }
601   return result;
602 }
603 #endif  // SANITIZER_LINUX
604 
internal_sigprocmask(int how,__sanitizer_sigset_t * set,__sanitizer_sigset_t * oldset)605 uptr internal_sigprocmask(int how, __sanitizer_sigset_t *set,
606     __sanitizer_sigset_t *oldset) {
607 #if SANITIZER_FREEBSD
608   return internal_syscall(SYSCALL(sigprocmask), how, set, oldset);
609 #else
610   __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
611   __sanitizer_kernel_sigset_t *k_oldset = (__sanitizer_kernel_sigset_t *)oldset;
612   return internal_syscall(SYSCALL(rt_sigprocmask), (uptr)how,
613                           (uptr)&k_set->sig[0], (uptr)&k_oldset->sig[0],
614                           sizeof(__sanitizer_kernel_sigset_t));
615 #endif
616 }
617 
internal_sigfillset(__sanitizer_sigset_t * set)618 void internal_sigfillset(__sanitizer_sigset_t *set) {
619   internal_memset(set, 0xff, sizeof(*set));
620 }
621 
622 #if SANITIZER_LINUX
internal_sigdelset(__sanitizer_sigset_t * set,int signum)623 void internal_sigdelset(__sanitizer_sigset_t *set, int signum) {
624   signum -= 1;
625   CHECK_GE(signum, 0);
626   CHECK_LT(signum, sizeof(*set) * 8);
627   __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
628   const uptr idx = signum / (sizeof(k_set->sig[0]) * 8);
629   const uptr bit = signum % (sizeof(k_set->sig[0]) * 8);
630   k_set->sig[idx] &= ~(1 << bit);
631 }
632 #endif  // SANITIZER_LINUX
633 
634 // ThreadLister implementation.
ThreadLister(int pid)635 ThreadLister::ThreadLister(int pid)
636   : pid_(pid),
637     descriptor_(-1),
638     buffer_(4096),
639     error_(true),
640     entry_((struct linux_dirent *)buffer_.data()),
641     bytes_read_(0) {
642   char task_directory_path[80];
643   internal_snprintf(task_directory_path, sizeof(task_directory_path),
644                     "/proc/%d/task/", pid);
645   uptr openrv = internal_open(task_directory_path, O_RDONLY | O_DIRECTORY);
646   if (internal_iserror(openrv)) {
647     error_ = true;
648     Report("Can't open /proc/%d/task for reading.\n", pid);
649   } else {
650     error_ = false;
651     descriptor_ = openrv;
652   }
653 }
654 
GetNextTID()655 int ThreadLister::GetNextTID() {
656   int tid = -1;
657   do {
658     if (error_)
659       return -1;
660     if ((char *)entry_ >= &buffer_[bytes_read_] && !GetDirectoryEntries())
661       return -1;
662     if (entry_->d_ino != 0 && entry_->d_name[0] >= '0' &&
663         entry_->d_name[0] <= '9') {
664       // Found a valid tid.
665       tid = (int)internal_atoll(entry_->d_name);
666     }
667     entry_ = (struct linux_dirent *)(((char *)entry_) + entry_->d_reclen);
668   } while (tid < 0);
669   return tid;
670 }
671 
Reset()672 void ThreadLister::Reset() {
673   if (error_ || descriptor_ < 0)
674     return;
675   internal_lseek(descriptor_, 0, SEEK_SET);
676 }
677 
~ThreadLister()678 ThreadLister::~ThreadLister() {
679   if (descriptor_ >= 0)
680     internal_close(descriptor_);
681 }
682 
error()683 bool ThreadLister::error() { return error_; }
684 
GetDirectoryEntries()685 bool ThreadLister::GetDirectoryEntries() {
686   CHECK_GE(descriptor_, 0);
687   CHECK_NE(error_, true);
688   bytes_read_ = internal_getdents(descriptor_,
689                                   (struct linux_dirent *)buffer_.data(),
690                                   buffer_.size());
691   if (internal_iserror(bytes_read_)) {
692     Report("Can't read directory entries from /proc/%d/task.\n", pid_);
693     error_ = true;
694     return false;
695   } else if (bytes_read_ == 0) {
696     return false;
697   }
698   entry_ = (struct linux_dirent *)buffer_.data();
699   return true;
700 }
701 
GetPageSize()702 uptr GetPageSize() {
703 #if SANITIZER_LINUX && (defined(__x86_64__) || defined(__i386__))
704   return EXEC_PAGESIZE;
705 #else
706   return sysconf(_SC_PAGESIZE);  // EXEC_PAGESIZE may not be trustworthy.
707 #endif
708 }
709 
ReadBinaryName(char * buf,uptr buf_len)710 uptr ReadBinaryName(/*out*/char *buf, uptr buf_len) {
711 #if SANITIZER_FREEBSD
712   const int Mib[] = { CTL_KERN, KERN_PROC, KERN_PROC_PATHNAME, -1 };
713   const char *default_module_name = "kern.proc.pathname";
714   size_t Size = buf_len;
715   bool IsErr = (sysctl(Mib, ARRAY_SIZE(Mib), buf, &Size, NULL, 0) != 0);
716   int readlink_error = IsErr ? errno : 0;
717   uptr module_name_len = Size;
718 #else
719   const char *default_module_name = "/proc/self/exe";
720   uptr module_name_len = internal_readlink(
721       default_module_name, buf, buf_len);
722   int readlink_error;
723   bool IsErr = internal_iserror(module_name_len, &readlink_error);
724 #endif
725   if (IsErr) {
726     // We can't read binary name for some reason, assume it's unknown.
727     Report("WARNING: reading executable name failed with errno %d, "
728            "some stack frames may not be symbolized\n", readlink_error);
729     module_name_len = internal_snprintf(buf, buf_len, "%s",
730                                         default_module_name);
731     CHECK_LT(module_name_len, buf_len);
732   }
733   return module_name_len;
734 }
735 
ReadLongProcessName(char * buf,uptr buf_len)736 uptr ReadLongProcessName(/*out*/ char *buf, uptr buf_len) {
737 #if SANITIZER_LINUX
738   char *tmpbuf;
739   uptr tmpsize;
740   uptr tmplen;
741   if (ReadFileToBuffer("/proc/self/cmdline", &tmpbuf, &tmpsize, &tmplen,
742                        1024 * 1024)) {
743     internal_strncpy(buf, tmpbuf, buf_len);
744     UnmapOrDie(tmpbuf, tmpsize);
745     return internal_strlen(buf);
746   }
747 #endif
748   return ReadBinaryName(buf, buf_len);
749 }
750 
751 // Match full names of the form /path/to/base_name{-,.}*
LibraryNameIs(const char * full_name,const char * base_name)752 bool LibraryNameIs(const char *full_name, const char *base_name) {
753   const char *name = full_name;
754   // Strip path.
755   while (*name != '\0') name++;
756   while (name > full_name && *name != '/') name--;
757   if (*name == '/') name++;
758   uptr base_name_length = internal_strlen(base_name);
759   if (internal_strncmp(name, base_name, base_name_length)) return false;
760   return (name[base_name_length] == '-' || name[base_name_length] == '.');
761 }
762 
763 #if !SANITIZER_ANDROID
764 // Call cb for each region mapped by map.
ForEachMappedRegion(link_map * map,void (* cb)(const void *,uptr))765 void ForEachMappedRegion(link_map *map, void (*cb)(const void *, uptr)) {
766   CHECK_NE(map, nullptr);
767 #if !SANITIZER_FREEBSD
768   typedef ElfW(Phdr) Elf_Phdr;
769   typedef ElfW(Ehdr) Elf_Ehdr;
770 #endif  // !SANITIZER_FREEBSD
771   char *base = (char *)map->l_addr;
772   Elf_Ehdr *ehdr = (Elf_Ehdr *)base;
773   char *phdrs = base + ehdr->e_phoff;
774   char *phdrs_end = phdrs + ehdr->e_phnum * ehdr->e_phentsize;
775 
776   // Find the segment with the minimum base so we can "relocate" the p_vaddr
777   // fields.  Typically ET_DYN objects (DSOs) have base of zero and ET_EXEC
778   // objects have a non-zero base.
779   uptr preferred_base = (uptr)-1;
780   for (char *iter = phdrs; iter != phdrs_end; iter += ehdr->e_phentsize) {
781     Elf_Phdr *phdr = (Elf_Phdr *)iter;
782     if (phdr->p_type == PT_LOAD && preferred_base > (uptr)phdr->p_vaddr)
783       preferred_base = (uptr)phdr->p_vaddr;
784   }
785 
786   // Compute the delta from the real base to get a relocation delta.
787   sptr delta = (uptr)base - preferred_base;
788   // Now we can figure out what the loader really mapped.
789   for (char *iter = phdrs; iter != phdrs_end; iter += ehdr->e_phentsize) {
790     Elf_Phdr *phdr = (Elf_Phdr *)iter;
791     if (phdr->p_type == PT_LOAD) {
792       uptr seg_start = phdr->p_vaddr + delta;
793       uptr seg_end = seg_start + phdr->p_memsz;
794       // None of these values are aligned.  We consider the ragged edges of the
795       // load command as defined, since they are mapped from the file.
796       seg_start = RoundDownTo(seg_start, GetPageSizeCached());
797       seg_end = RoundUpTo(seg_end, GetPageSizeCached());
798       cb((void *)seg_start, seg_end - seg_start);
799     }
800   }
801 }
802 #endif
803 
804 #if defined(__x86_64__) && SANITIZER_LINUX
805 // We cannot use glibc's clone wrapper, because it messes with the child
806 // task's TLS. It writes the PID and TID of the child task to its thread
807 // descriptor, but in our case the child task shares the thread descriptor with
808 // the parent (because we don't know how to allocate a new thread
809 // descriptor to keep glibc happy). So the stock version of clone(), when
810 // used with CLONE_VM, would end up corrupting the parent's thread descriptor.
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)811 uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
812                     int *parent_tidptr, void *newtls, int *child_tidptr) {
813   long long res;
814   if (!fn || !child_stack)
815     return -EINVAL;
816   CHECK_EQ(0, (uptr)child_stack % 16);
817   child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
818   ((unsigned long long *)child_stack)[0] = (uptr)fn;
819   ((unsigned long long *)child_stack)[1] = (uptr)arg;
820   register void *r8 __asm__("r8") = newtls;
821   register int *r10 __asm__("r10") = child_tidptr;
822   __asm__ __volatile__(
823                        /* %rax = syscall(%rax = SYSCALL(clone),
824                         *                %rdi = flags,
825                         *                %rsi = child_stack,
826                         *                %rdx = parent_tidptr,
827                         *                %r8  = new_tls,
828                         *                %r10 = child_tidptr)
829                         */
830                        "syscall\n"
831 
832                        /* if (%rax != 0)
833                         *   return;
834                         */
835                        "testq  %%rax,%%rax\n"
836                        "jnz    1f\n"
837 
838                        /* In the child. Terminate unwind chain. */
839                        // XXX: We should also terminate the CFI unwind chain
840                        // here. Unfortunately clang 3.2 doesn't support the
841                        // necessary CFI directives, so we skip that part.
842                        "xorq   %%rbp,%%rbp\n"
843 
844                        /* Call "fn(arg)". */
845                        "popq   %%rax\n"
846                        "popq   %%rdi\n"
847                        "call   *%%rax\n"
848 
849                        /* Call _exit(%rax). */
850                        "movq   %%rax,%%rdi\n"
851                        "movq   %2,%%rax\n"
852                        "syscall\n"
853 
854                        /* Return to parent. */
855                      "1:\n"
856                        : "=a" (res)
857                        : "a"(SYSCALL(clone)), "i"(SYSCALL(exit)),
858                          "S"(child_stack),
859                          "D"(flags),
860                          "d"(parent_tidptr),
861                          "r"(r8),
862                          "r"(r10)
863                        : "rsp", "memory", "r11", "rcx");
864   return res;
865 }
866 #elif defined(__mips__)
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)867 uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
868                     int *parent_tidptr, void *newtls, int *child_tidptr) {
869   long long res;
870   if (!fn || !child_stack)
871     return -EINVAL;
872   CHECK_EQ(0, (uptr)child_stack % 16);
873   child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
874   ((unsigned long long *)child_stack)[0] = (uptr)fn;
875   ((unsigned long long *)child_stack)[1] = (uptr)arg;
876   register void *a3 __asm__("$7") = newtls;
877   register int *a4 __asm__("$8") = child_tidptr;
878   // We don't have proper CFI directives here because it requires alot of code
879   // for very marginal benefits.
880   __asm__ __volatile__(
881                        /* $v0 = syscall($v0 = __NR_clone,
882                         * $a0 = flags,
883                         * $a1 = child_stack,
884                         * $a2 = parent_tidptr,
885                         * $a3 = new_tls,
886                         * $a4 = child_tidptr)
887                         */
888                        ".cprestore 16;\n"
889                        "move $4,%1;\n"
890                        "move $5,%2;\n"
891                        "move $6,%3;\n"
892                        "move $7,%4;\n"
893                        /* Store the fifth argument on stack
894                         * if we are using 32-bit abi.
895                         */
896 #if SANITIZER_WORDSIZE == 32
897                        "lw %5,16($29);\n"
898 #else
899                        "move $8,%5;\n"
900 #endif
901                        "li $2,%6;\n"
902                        "syscall;\n"
903 
904                        /* if ($v0 != 0)
905                         * return;
906                         */
907                        "bnez $2,1f;\n"
908 
909                        /* Call "fn(arg)". */
910                        "ld $25,0($29);\n"
911                        "ld $4,8($29);\n"
912                        "jal $25;\n"
913 
914                        /* Call _exit($v0). */
915                        "move $4,$2;\n"
916                        "li $2,%7;\n"
917                        "syscall;\n"
918 
919                        /* Return to parent. */
920                      "1:\n"
921                        : "=r" (res)
922                        : "r"(flags),
923                          "r"(child_stack),
924                          "r"(parent_tidptr),
925                          "r"(a3),
926                          "r"(a4),
927                          "i"(__NR_clone),
928                          "i"(__NR_exit)
929                        : "memory", "$29" );
930   return res;
931 }
932 #elif defined(__aarch64__)
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)933 uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
934                     int *parent_tidptr, void *newtls, int *child_tidptr) {
935   long long res;
936   if (!fn || !child_stack)
937     return -EINVAL;
938   CHECK_EQ(0, (uptr)child_stack % 16);
939   child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
940   ((unsigned long long *)child_stack)[0] = (uptr)fn;
941   ((unsigned long long *)child_stack)[1] = (uptr)arg;
942 
943   register int (*__fn)(void *)  __asm__("x0") = fn;
944   register void *__stack __asm__("x1") = child_stack;
945   register int   __flags __asm__("x2") = flags;
946   register void *__arg   __asm__("x3") = arg;
947   register int  *__ptid  __asm__("x4") = parent_tidptr;
948   register void *__tls   __asm__("x5") = newtls;
949   register int  *__ctid  __asm__("x6") = child_tidptr;
950 
951   __asm__ __volatile__(
952                        "mov x0,x2\n" /* flags  */
953                        "mov x2,x4\n" /* ptid  */
954                        "mov x3,x5\n" /* tls  */
955                        "mov x4,x6\n" /* ctid  */
956                        "mov x8,%9\n" /* clone  */
957 
958                        "svc 0x0\n"
959 
960                        /* if (%r0 != 0)
961                         *   return %r0;
962                         */
963                        "cmp x0, #0\n"
964                        "bne 1f\n"
965 
966                        /* In the child, now. Call "fn(arg)". */
967                        "ldp x1, x0, [sp], #16\n"
968                        "blr x1\n"
969 
970                        /* Call _exit(%r0).  */
971                        "mov x8, %10\n"
972                        "svc 0x0\n"
973                      "1:\n"
974 
975                        : "=r" (res)
976                        : "i"(-EINVAL),
977                          "r"(__fn), "r"(__stack), "r"(__flags), "r"(__arg),
978                          "r"(__ptid), "r"(__tls), "r"(__ctid),
979                          "i"(__NR_clone), "i"(__NR_exit)
980                        : "x30", "memory");
981   return res;
982 }
983 #endif  // defined(__x86_64__) && SANITIZER_LINUX
984 
985 #if SANITIZER_ANDROID
986 #define PROP_VALUE_MAX 92
987 extern "C" SANITIZER_WEAK_ATTRIBUTE int __system_property_get(const char *name,
988                                                               char *value);
GetExtraActivationFlags(char * buf,uptr size)989 void GetExtraActivationFlags(char *buf, uptr size) {
990   CHECK(size > PROP_VALUE_MAX);
991   CHECK(&__system_property_get);
992   __system_property_get("asan.options", buf);
993 }
994 
995 #if __ANDROID_API__ < 21
996 extern "C" __attribute__((weak)) int dl_iterate_phdr(
997     int (*)(struct dl_phdr_info *, size_t, void *), void *);
998 #endif
999 
dl_iterate_phdr_test_cb(struct dl_phdr_info * info,size_t size,void * data)1000 static int dl_iterate_phdr_test_cb(struct dl_phdr_info *info, size_t size,
1001                                    void *data) {
1002   // Any name starting with "lib" indicates a bug in L where library base names
1003   // are returned instead of paths.
1004   if (info->dlpi_name && info->dlpi_name[0] == 'l' &&
1005       info->dlpi_name[1] == 'i' && info->dlpi_name[2] == 'b') {
1006     *(bool *)data = true;
1007     return 1;
1008   }
1009   return 0;
1010 }
1011 
1012 static atomic_uint32_t android_api_level;
1013 
AndroidDetectApiLevel()1014 static AndroidApiLevel AndroidDetectApiLevel() {
1015   if (!&dl_iterate_phdr)
1016     return ANDROID_KITKAT; // K or lower
1017   bool base_name_seen = false;
1018   dl_iterate_phdr(dl_iterate_phdr_test_cb, &base_name_seen);
1019   if (base_name_seen)
1020     return ANDROID_LOLLIPOP_MR1; // L MR1
1021   return ANDROID_POST_LOLLIPOP;   // post-L
1022   // Plain L (API level 21) is completely broken wrt ASan and not very
1023   // interesting to detect.
1024 }
1025 
AndroidGetApiLevel()1026 AndroidApiLevel AndroidGetApiLevel() {
1027   AndroidApiLevel level =
1028       (AndroidApiLevel)atomic_load(&android_api_level, memory_order_relaxed);
1029   if (level) return level;
1030   level = AndroidDetectApiLevel();
1031   atomic_store(&android_api_level, level, memory_order_relaxed);
1032   return level;
1033 }
1034 
1035 #endif
1036 
IsDeadlySignal(int signum)1037 bool IsDeadlySignal(int signum) {
1038   if (common_flags()->handle_abort && signum == SIGABRT)
1039     return true;
1040   if (common_flags()->handle_sigfpe && signum == SIGFPE)
1041     return true;
1042   return (signum == SIGSEGV || signum == SIGBUS) && common_flags()->handle_segv;
1043 }
1044 
1045 #ifndef SANITIZER_GO
internal_start_thread(void (* func)(void * arg),void * arg)1046 void *internal_start_thread(void(*func)(void *arg), void *arg) {
1047   // Start the thread with signals blocked, otherwise it can steal user signals.
1048   __sanitizer_sigset_t set, old;
1049   internal_sigfillset(&set);
1050 #if SANITIZER_LINUX && !SANITIZER_ANDROID
1051   // Glibc uses SIGSETXID signal during setuid call. If this signal is blocked
1052   // on any thread, setuid call hangs (see test/tsan/setuid.c).
1053   internal_sigdelset(&set, 33);
1054 #endif
1055   internal_sigprocmask(SIG_SETMASK, &set, &old);
1056   void *th;
1057   real_pthread_create(&th, nullptr, (void*(*)(void *arg))func, arg);
1058   internal_sigprocmask(SIG_SETMASK, &old, nullptr);
1059   return th;
1060 }
1061 
internal_join_thread(void * th)1062 void internal_join_thread(void *th) {
1063   real_pthread_join(th, nullptr);
1064 }
1065 #else
internal_start_thread(void (* func)(void *),void * arg)1066 void *internal_start_thread(void (*func)(void *), void *arg) { return 0; }
1067 
internal_join_thread(void * th)1068 void internal_join_thread(void *th) {}
1069 #endif
1070 
GetPcSpBp(void * context,uptr * pc,uptr * sp,uptr * bp)1071 void GetPcSpBp(void *context, uptr *pc, uptr *sp, uptr *bp) {
1072 #if defined(__arm__)
1073   ucontext_t *ucontext = (ucontext_t*)context;
1074   *pc = ucontext->uc_mcontext.arm_pc;
1075   *bp = ucontext->uc_mcontext.arm_fp;
1076   *sp = ucontext->uc_mcontext.arm_sp;
1077 #elif defined(__aarch64__)
1078   ucontext_t *ucontext = (ucontext_t*)context;
1079   *pc = ucontext->uc_mcontext.pc;
1080   *bp = ucontext->uc_mcontext.regs[29];
1081   *sp = ucontext->uc_mcontext.sp;
1082 #elif defined(__hppa__)
1083   ucontext_t *ucontext = (ucontext_t*)context;
1084   *pc = ucontext->uc_mcontext.sc_iaoq[0];
1085   /* GCC uses %r3 whenever a frame pointer is needed.  */
1086   *bp = ucontext->uc_mcontext.sc_gr[3];
1087   *sp = ucontext->uc_mcontext.sc_gr[30];
1088 #elif defined(__x86_64__)
1089 # if SANITIZER_FREEBSD
1090   ucontext_t *ucontext = (ucontext_t*)context;
1091   *pc = ucontext->uc_mcontext.mc_rip;
1092   *bp = ucontext->uc_mcontext.mc_rbp;
1093   *sp = ucontext->uc_mcontext.mc_rsp;
1094 # else
1095   ucontext_t *ucontext = (ucontext_t*)context;
1096   *pc = ucontext->uc_mcontext.gregs[REG_RIP];
1097   *bp = ucontext->uc_mcontext.gregs[REG_RBP];
1098   *sp = ucontext->uc_mcontext.gregs[REG_RSP];
1099 # endif
1100 #elif defined(__i386__)
1101 # if SANITIZER_FREEBSD
1102   ucontext_t *ucontext = (ucontext_t*)context;
1103   *pc = ucontext->uc_mcontext.mc_eip;
1104   *bp = ucontext->uc_mcontext.mc_ebp;
1105   *sp = ucontext->uc_mcontext.mc_esp;
1106 # else
1107   ucontext_t *ucontext = (ucontext_t*)context;
1108   *pc = ucontext->uc_mcontext.gregs[REG_EIP];
1109   *bp = ucontext->uc_mcontext.gregs[REG_EBP];
1110   *sp = ucontext->uc_mcontext.gregs[REG_ESP];
1111 # endif
1112 #elif defined(__powerpc__) || defined(__powerpc64__)
1113   ucontext_t *ucontext = (ucontext_t*)context;
1114   *pc = ucontext->uc_mcontext.regs->nip;
1115   *sp = ucontext->uc_mcontext.regs->gpr[PT_R1];
1116   // The powerpc{,64}-linux ABIs do not specify r31 as the frame
1117   // pointer, but GCC always uses r31 when we need a frame pointer.
1118   *bp = ucontext->uc_mcontext.regs->gpr[PT_R31];
1119 #elif defined(__sparc__)
1120   ucontext_t *ucontext = (ucontext_t*)context;
1121   uptr *stk_ptr;
1122 # if defined (__arch64__)
1123   *pc = ucontext->uc_mcontext.mc_gregs[MC_PC];
1124   *sp = ucontext->uc_mcontext.mc_gregs[MC_O6];
1125   stk_ptr = (uptr *) (*sp + 2047);
1126   *bp = stk_ptr[15];
1127 # else
1128   *pc = ucontext->uc_mcontext.gregs[REG_PC];
1129   *sp = ucontext->uc_mcontext.gregs[REG_O6];
1130   stk_ptr = (uptr *) *sp;
1131   *bp = stk_ptr[15];
1132 # endif
1133 #elif defined(__mips__)
1134   ucontext_t *ucontext = (ucontext_t*)context;
1135   *pc = ucontext->uc_mcontext.pc;
1136   *bp = ucontext->uc_mcontext.gregs[30];
1137   *sp = ucontext->uc_mcontext.gregs[29];
1138 #else
1139 # error "Unsupported arch"
1140 #endif
1141 }
1142 
1143 } // namespace __sanitizer
1144 
1145 #endif // SANITIZER_FREEBSD || SANITIZER_LINUX
1146