1490215a3Smrg //===-- sanitizer_linux.cc ------------------------------------------------===//
2490215a3Smrg //
3490215a3Smrg // This file is distributed under the University of Illinois Open Source
4490215a3Smrg // License. See LICENSE.TXT for details.
5490215a3Smrg //
6490215a3Smrg //===----------------------------------------------------------------------===//
7490215a3Smrg //
8490215a3Smrg // This file is shared between AddressSanitizer and ThreadSanitizer
9490215a3Smrg // run-time libraries and implements linux-specific functions from
10490215a3Smrg // sanitizer_libc.h.
11490215a3Smrg //===----------------------------------------------------------------------===//
12490215a3Smrg 
13490215a3Smrg #include "sanitizer_platform.h"
14490215a3Smrg 
15490215a3Smrg #if SANITIZER_FREEBSD || SANITIZER_LINUX || SANITIZER_NETBSD || \
16490215a3Smrg     SANITIZER_OPENBSD || SANITIZER_SOLARIS
17490215a3Smrg 
18490215a3Smrg #include "sanitizer_common.h"
19490215a3Smrg #include "sanitizer_flags.h"
20490215a3Smrg #include "sanitizer_getauxval.h"
21490215a3Smrg #include "sanitizer_internal_defs.h"
22490215a3Smrg #include "sanitizer_libc.h"
23490215a3Smrg #include "sanitizer_linux.h"
24490215a3Smrg #include "sanitizer_mutex.h"
25490215a3Smrg #include "sanitizer_placement_new.h"
26490215a3Smrg #include "sanitizer_procmaps.h"
27490215a3Smrg 
28490215a3Smrg #if SANITIZER_LINUX
29490215a3Smrg #include <asm/param.h>
30490215a3Smrg #endif
31490215a3Smrg 
32490215a3Smrg // For mips64, syscall(__NR_stat) fills the buffer in the 'struct kernel_stat'
33490215a3Smrg // format. Struct kernel_stat is defined as 'struct stat' in asm/stat.h. To
34490215a3Smrg // access stat from asm/stat.h, without conflicting with definition in
35490215a3Smrg // sys/stat.h, we use this trick.
36490215a3Smrg #if SANITIZER_LINUX
37490215a3Smrg #if defined(__mips64)
38490215a3Smrg #include <asm/unistd.h>
39490215a3Smrg #include <sys/types.h>
40490215a3Smrg #define stat kernel_stat
41490215a3Smrg #include <asm/stat.h>
42490215a3Smrg #undef stat
43490215a3Smrg #endif
44490215a3Smrg #endif
45490215a3Smrg 
46490215a3Smrg #if SANITIZER_NETBSD
47490215a3Smrg #include <lwp.h>
48490215a3Smrg #endif
49490215a3Smrg 
50490215a3Smrg #include <dlfcn.h>
51490215a3Smrg #include <errno.h>
52490215a3Smrg #include <fcntl.h>
53490215a3Smrg #include <link.h>
54490215a3Smrg #include <pthread.h>
55490215a3Smrg #include <sched.h>
56490215a3Smrg #include <signal.h>
57490215a3Smrg #include <sys/mman.h>
58490215a3Smrg #include <sys/param.h>
59490215a3Smrg #if !SANITIZER_SOLARIS
60490215a3Smrg #include <sys/ptrace.h>
61490215a3Smrg #endif
62490215a3Smrg #include <sys/resource.h>
63490215a3Smrg #include <sys/stat.h>
64490215a3Smrg #include <sys/syscall.h>
65490215a3Smrg #include <sys/time.h>
66490215a3Smrg #include <sys/types.h>
67490215a3Smrg #if !SANITIZER_OPENBSD
68490215a3Smrg #include <ucontext.h>
69490215a3Smrg #endif
70490215a3Smrg #if SANITIZER_OPENBSD
71490215a3Smrg #include <sys/futex.h>
72490215a3Smrg #include <sys/sysctl.h>
73490215a3Smrg #endif
74490215a3Smrg #include <unistd.h>
75490215a3Smrg 
76490215a3Smrg #if SANITIZER_LINUX
77490215a3Smrg #include <sys/utsname.h>
78490215a3Smrg #endif
79490215a3Smrg 
80490215a3Smrg #if SANITIZER_LINUX && !SANITIZER_ANDROID
81490215a3Smrg #include <sys/personality.h>
82490215a3Smrg #endif
83490215a3Smrg 
84490215a3Smrg #if SANITIZER_FREEBSD
85490215a3Smrg #include <sys/exec.h>
86490215a3Smrg #include <sys/sysctl.h>
87490215a3Smrg #include <machine/atomic.h>
88490215a3Smrg extern "C" {
89490215a3Smrg // <sys/umtx.h> must be included after <errno.h> and <sys/types.h> on
90490215a3Smrg // FreeBSD 9.2 and 10.0.
91490215a3Smrg #include <sys/umtx.h>
92490215a3Smrg }
93490215a3Smrg #include <sys/thr.h>
94490215a3Smrg #endif  // SANITIZER_FREEBSD
95490215a3Smrg #if SANITIZER_NETBSD
96490215a3Smrg #include <limits.h>	// For NAME_MAX
97490215a3Smrg #include <sys/sysctl.h>
98490215a3Smrg #include <sys/exec.h>
99490215a3Smrg extern struct ps_strings *__ps_strings;
100490215a3Smrg extern char **environ;  // provided by crt1
101490215a3Smrg #endif  // SANITIZER_NETBSD
102490215a3Smrg 
103490215a3Smrg #if SANITIZER_NETBSD
104490215a3Smrg #include <limits.h>  // For NAME_MAX
105490215a3Smrg #include <sys/sysctl.h>
106490215a3Smrg #include <sys/exec.h>
107490215a3Smrg extern struct ps_strings *__ps_strings;
108490215a3Smrg #endif  // SANITIZER_NETBSD
109490215a3Smrg 
110490215a3Smrg #if SANITIZER_SOLARIS
111490215a3Smrg #include <stdlib.h>
112490215a3Smrg #include <thread.h>
113490215a3Smrg #define environ _environ
114490215a3Smrg #endif
115490215a3Smrg 
116490215a3Smrg extern char **environ;
117490215a3Smrg 
118490215a3Smrg #if SANITIZER_LINUX
119490215a3Smrg // <linux/time.h>
120490215a3Smrg struct kernel_timeval {
121490215a3Smrg   long tv_sec;
122490215a3Smrg   long tv_usec;
123490215a3Smrg };
124490215a3Smrg 
125490215a3Smrg // <linux/futex.h> is broken on some linux distributions.
126490215a3Smrg const int FUTEX_WAIT = 0;
127490215a3Smrg const int FUTEX_WAKE = 1;
128490215a3Smrg const int FUTEX_PRIVATE_FLAG = 128;
129490215a3Smrg const int FUTEX_WAIT_PRIVATE = FUTEX_WAIT | FUTEX_PRIVATE_FLAG;
130490215a3Smrg const int FUTEX_WAKE_PRIVATE = FUTEX_WAKE | FUTEX_PRIVATE_FLAG;
131490215a3Smrg #endif  // SANITIZER_LINUX
132490215a3Smrg 
133490215a3Smrg // Are we using 32-bit or 64-bit Linux syscalls?
134490215a3Smrg // x32 (which defines __x86_64__) has SANITIZER_WORDSIZE == 32
135490215a3Smrg // but it still needs to use 64-bit syscalls.
136490215a3Smrg #if SANITIZER_LINUX && (defined(__x86_64__) || defined(__powerpc64__) ||       \
137490215a3Smrg                         SANITIZER_WORDSIZE == 64)
138490215a3Smrg # define SANITIZER_LINUX_USES_64BIT_SYSCALLS 1
139490215a3Smrg #else
140490215a3Smrg # define SANITIZER_LINUX_USES_64BIT_SYSCALLS 0
141490215a3Smrg #endif
142490215a3Smrg 
143490215a3Smrg #if defined(__x86_64__) || SANITIZER_MIPS64
144490215a3Smrg extern "C" {
145490215a3Smrg extern void internal_sigreturn();
146490215a3Smrg }
147490215a3Smrg #endif
148490215a3Smrg 
149490215a3Smrg // Note : FreeBSD had implemented both
150490215a3Smrg // Linux and OpenBSD apis, available from
151490215a3Smrg // future 12.x version most likely
152490215a3Smrg #if SANITIZER_LINUX && defined(__NR_getrandom)
153490215a3Smrg # if !defined(GRND_NONBLOCK)
154490215a3Smrg #  define GRND_NONBLOCK 1
155490215a3Smrg # endif
156490215a3Smrg # define SANITIZER_USE_GETRANDOM 1
157490215a3Smrg #else
158490215a3Smrg # define SANITIZER_USE_GETRANDOM 0
159490215a3Smrg #endif  // SANITIZER_LINUX && defined(__NR_getrandom)
160490215a3Smrg 
161490215a3Smrg #if SANITIZER_OPENBSD
162490215a3Smrg # define SANITIZER_USE_GETENTROPY 1
163490215a3Smrg #else
164490215a3Smrg # if SANITIZER_FREEBSD && __FreeBSD_version >= 1200000
165490215a3Smrg #   define SANITIZER_USE_GETENTROPY 1
166490215a3Smrg # else
167490215a3Smrg #   define SANITIZER_USE_GETENTROPY 0
168490215a3Smrg # endif
169490215a3Smrg #endif // SANITIZER_USE_GETENTROPY
170490215a3Smrg 
171490215a3Smrg namespace __sanitizer {
172490215a3Smrg 
173490215a3Smrg #if SANITIZER_LINUX && defined(__x86_64__)
174490215a3Smrg #include "sanitizer_syscall_linux_x86_64.inc"
175490215a3Smrg #elif SANITIZER_LINUX && defined(__aarch64__)
176490215a3Smrg #include "sanitizer_syscall_linux_aarch64.inc"
177490215a3Smrg #elif SANITIZER_LINUX && defined(__arm__)
178490215a3Smrg #include "sanitizer_syscall_linux_arm.inc"
179490215a3Smrg #else
180490215a3Smrg #include "sanitizer_syscall_generic.inc"
181490215a3Smrg #endif
182490215a3Smrg 
183490215a3Smrg // --------------- sanitizer_libc.h
184490215a3Smrg #if !SANITIZER_SOLARIS && !SANITIZER_NETBSD
185490215a3Smrg #if !SANITIZER_S390 && !SANITIZER_OPENBSD
internal_mmap(void * addr,uptr length,int prot,int flags,int fd,OFF_T offset)186490215a3Smrg uptr internal_mmap(void *addr, uptr length, int prot, int flags, int fd,
187490215a3Smrg                    OFF_T offset) {
188490215a3Smrg #if SANITIZER_NETBSD
189490215a3Smrg 
190490215a3Smrg #if !defined(_LP64) && BYTE_ORDER == _BIG_ENDIAN
191490215a3Smrg #define __SYSCALL_TO_UINTPTR_T(V)       ((uintptr_t)((V)>>32))
192490215a3Smrg #else
193490215a3Smrg #define __SYSCALL_TO_UINTPTR_T(V)       ((uintptr_t)(V))
194490215a3Smrg #endif
195490215a3Smrg 
196490215a3Smrg   return __SYSCALL_TO_UINTPTR_T(
197490215a3Smrg 	    internal_syscall64(SYSCALL(mmap), addr, length, prot, flags, fd,
198490215a3Smrg 			      (long)0, offset));
199490215a3Smrg   //return internal_syscall_ptr(SYSCALL(mmap), addr, length, prot, flags, fd,
200490215a3Smrg   //                            (long)0, offset);
201490215a3Smrg #elif SANITIZER_FREEBSD || SANITIZER_LINUX_USES_64BIT_SYSCALLS
202490215a3Smrg   return internal_syscall(SYSCALL(mmap), (uptr)addr, length, prot, flags, fd,
203490215a3Smrg                           offset, 0);
204490215a3Smrg #else
205490215a3Smrg   // mmap2 specifies file offset in 4096-byte units.
206490215a3Smrg   CHECK(IsAligned(offset, 4096));
207490215a3Smrg   return internal_syscall(SYSCALL(mmap2), addr, length, prot, flags, fd,
208490215a3Smrg                           offset / 4096);
209490215a3Smrg #endif
210490215a3Smrg }
211490215a3Smrg #endif // !SANITIZER_S390 && !SANITIZER_OPENBSD
212490215a3Smrg 
213490215a3Smrg #if !SANITIZER_OPENBSD
internal_munmap(void * addr,uptr length)214490215a3Smrg uptr internal_munmap(void *addr, uptr length) {
215490215a3Smrg   return internal_syscall_ptr(SYSCALL(munmap), (uptr)addr, length);
216490215a3Smrg }
217490215a3Smrg 
internal_mprotect(void * addr,uptr length,int prot)218490215a3Smrg int internal_mprotect(void *addr, uptr length, int prot) {
219490215a3Smrg   return internal_syscall_ptr(SYSCALL(mprotect), (uptr)addr, length, prot);
220490215a3Smrg }
221490215a3Smrg #endif
222490215a3Smrg 
internal_close(fd_t fd)223490215a3Smrg uptr internal_close(fd_t fd) {
224490215a3Smrg   return internal_syscall(SYSCALL(close), fd);
225490215a3Smrg }
226490215a3Smrg 
internal_open(const char * filename,int flags)227490215a3Smrg uptr internal_open(const char *filename, int flags) {
228490215a3Smrg #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
229490215a3Smrg   return internal_syscall(SYSCALL(openat), AT_FDCWD, (uptr)filename, flags);
230490215a3Smrg #else
231490215a3Smrg   return internal_syscall_ptr(SYSCALL(open), (uptr)filename, flags);
232490215a3Smrg #endif
233490215a3Smrg }
234490215a3Smrg 
internal_open(const char * filename,int flags,u32 mode)235490215a3Smrg uptr internal_open(const char *filename, int flags, u32 mode) {
236490215a3Smrg #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
237490215a3Smrg   return internal_syscall(SYSCALL(openat), AT_FDCWD, (uptr)filename, flags,
238490215a3Smrg                           mode);
239490215a3Smrg #else
240490215a3Smrg   return internal_syscall_ptr(SYSCALL(open), (uptr)filename, flags, mode);
241490215a3Smrg #endif
242490215a3Smrg }
243490215a3Smrg 
internal_read(fd_t fd,void * buf,uptr count)244490215a3Smrg uptr internal_read(fd_t fd, void *buf, uptr count) {
245490215a3Smrg   sptr res;
246490215a3Smrg   HANDLE_EINTR(res,
247490215a3Smrg                (sptr)internal_syscall(SYSCALL(read), fd, (uptr)buf, count));
248490215a3Smrg   return res;
249490215a3Smrg }
250490215a3Smrg 
internal_write(fd_t fd,const void * buf,uptr count)251490215a3Smrg uptr internal_write(fd_t fd, const void *buf, uptr count) {
252490215a3Smrg   sptr res;
253490215a3Smrg   HANDLE_EINTR(res,
254490215a3Smrg                (sptr)internal_syscall(SYSCALL(write), fd, (uptr)buf, count));
255490215a3Smrg   return res;
256490215a3Smrg }
257490215a3Smrg 
internal_ftruncate(fd_t fd,uptr size)258490215a3Smrg uptr internal_ftruncate(fd_t fd, uptr size) {
259490215a3Smrg   sptr res;
260490215a3Smrg #if SANITIZER_NETBSD
261490215a3Smrg   //HANDLE_EINTR(res, internal_syscall64(SYSCALL(ftruncate), fd, 0, (s64)size));
262490215a3Smrg   HANDLE_EINTR(res, internal_syscall(SYSCALL(ftruncate), fd, 0, (s64)size));
263490215a3Smrg #else
264490215a3Smrg   HANDLE_EINTR(res, (sptr)internal_syscall(SYSCALL(ftruncate), fd,
265490215a3Smrg                (OFF_T)size));
266490215a3Smrg #endif
267490215a3Smrg   return res;
268490215a3Smrg }
269490215a3Smrg 
270490215a3Smrg #if !SANITIZER_LINUX_USES_64BIT_SYSCALLS && SANITIZER_LINUX
stat64_to_stat(struct stat64 * in,struct stat * out)271490215a3Smrg static void stat64_to_stat(struct stat64 *in, struct stat *out) {
272490215a3Smrg   internal_memset(out, 0, sizeof(*out));
273490215a3Smrg   out->st_dev = in->st_dev;
274490215a3Smrg   out->st_ino = in->st_ino;
275490215a3Smrg   out->st_mode = in->st_mode;
276490215a3Smrg   out->st_nlink = in->st_nlink;
277490215a3Smrg   out->st_uid = in->st_uid;
278490215a3Smrg   out->st_gid = in->st_gid;
279490215a3Smrg   out->st_rdev = in->st_rdev;
280490215a3Smrg   out->st_size = in->st_size;
281490215a3Smrg   out->st_blksize = in->st_blksize;
282490215a3Smrg   out->st_blocks = in->st_blocks;
283490215a3Smrg   out->st_atime = in->st_atime;
284490215a3Smrg   out->st_mtime = in->st_mtime;
285490215a3Smrg   out->st_ctime = in->st_ctime;
286490215a3Smrg }
287490215a3Smrg #endif
288490215a3Smrg 
289490215a3Smrg #if defined(__mips64) && SANITIZER_LINUX
290490215a3Smrg // Undefine compatibility macros from <sys/stat.h>
291490215a3Smrg // so that they would not clash with the kernel_stat
292490215a3Smrg // st_[a|m|c]time fields
293490215a3Smrg #undef st_atime
294490215a3Smrg #undef st_mtime
295490215a3Smrg #undef st_ctime
296490215a3Smrg #if defined(SANITIZER_ANDROID)
297490215a3Smrg // Bionic sys/stat.h defines additional macros
298490215a3Smrg // for compatibility with the old NDKs and
299490215a3Smrg // they clash with the kernel_stat structure
300490215a3Smrg // st_[a|m|c]time_nsec fields.
301490215a3Smrg #undef st_atime_nsec
302490215a3Smrg #undef st_mtime_nsec
303490215a3Smrg #undef st_ctime_nsec
304490215a3Smrg #endif
305490215a3Smrg 
kernel_stat_to_stat(struct kernel_stat * in,struct stat * out)306490215a3Smrg static void kernel_stat_to_stat(struct kernel_stat *in, struct stat *out) {
307490215a3Smrg   internal_memset(out, 0, sizeof(*out));
308490215a3Smrg   out->st_dev = in->st_dev;
309490215a3Smrg   out->st_ino = in->st_ino;
310490215a3Smrg   out->st_mode = in->st_mode;
311490215a3Smrg   out->st_nlink = in->st_nlink;
312490215a3Smrg   out->st_uid = in->st_uid;
313490215a3Smrg   out->st_gid = in->st_gid;
314490215a3Smrg   out->st_rdev = in->st_rdev;
315490215a3Smrg   out->st_size = in->st_size;
316490215a3Smrg   out->st_blksize = in->st_blksize;
317490215a3Smrg   out->st_blocks = in->st_blocks;
318490215a3Smrg #if defined(__USE_MISC)     || \
319490215a3Smrg     defined(__USE_XOPEN2K8) || \
320490215a3Smrg     defined(SANITIZER_ANDROID)
321490215a3Smrg   out->st_atim.tv_sec = in->st_atime;
322490215a3Smrg   out->st_atim.tv_nsec = in->st_atime_nsec;
323490215a3Smrg   out->st_mtim.tv_sec = in->st_mtime;
324490215a3Smrg   out->st_mtim.tv_nsec = in->st_mtime_nsec;
325490215a3Smrg   out->st_ctim.tv_sec = in->st_ctime;
326490215a3Smrg   out->st_ctim.tv_nsec = in->st_ctime_nsec;
327490215a3Smrg #else
328490215a3Smrg   out->st_atime = in->st_atime;
329490215a3Smrg   out->st_atimensec = in->st_atime_nsec;
330490215a3Smrg   out->st_mtime = in->st_mtime;
331490215a3Smrg   out->st_mtimensec = in->st_mtime_nsec;
332490215a3Smrg   out->st_ctime = in->st_ctime;
333490215a3Smrg   out->st_atimensec = in->st_ctime_nsec;
334490215a3Smrg #endif
335490215a3Smrg }
336490215a3Smrg #endif
337490215a3Smrg 
internal_stat(const char * path,void * buf)338490215a3Smrg uptr internal_stat(const char *path, void *buf) {
339490215a3Smrg #if SANITIZER_FREEBSD || SANITIZER_NETBSD || SANITIZER_OPENBSD
340490215a3Smrg   return internal_syscall(SYSCALL(fstatat), AT_FDCWD, (uptr)path, (uptr)buf, 0);
341490215a3Smrg #elif SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
342490215a3Smrg   return internal_syscall(SYSCALL(newfstatat), AT_FDCWD, (uptr)path, (uptr)buf,
343490215a3Smrg                           0);
344490215a3Smrg #elif SANITIZER_LINUX_USES_64BIT_SYSCALLS
345490215a3Smrg # if defined(__mips64)
346490215a3Smrg   // For mips64, stat syscall fills buffer in the format of kernel_stat
347490215a3Smrg   struct kernel_stat kbuf;
348490215a3Smrg   int res = internal_syscall(SYSCALL(stat), path, &kbuf);
349490215a3Smrg   kernel_stat_to_stat(&kbuf, (struct stat *)buf);
350490215a3Smrg   return res;
351490215a3Smrg # else
352490215a3Smrg   return internal_syscall(SYSCALL(stat), (uptr)path, (uptr)buf);
353490215a3Smrg # endif
354490215a3Smrg #else
355490215a3Smrg   struct stat64 buf64;
356490215a3Smrg   int res = internal_syscall(SYSCALL(stat64), path, &buf64);
357490215a3Smrg   stat64_to_stat(&buf64, (struct stat *)buf);
358490215a3Smrg   return res;
359490215a3Smrg #endif
360490215a3Smrg }
361490215a3Smrg 
internal_lstat(const char * path,void * buf)362490215a3Smrg uptr internal_lstat(const char *path, void *buf) {
363490215a3Smrg #if SANITIZER_NETBSD
364490215a3Smrg   return internal_syscall(SYSCALL(lstat), path, buf);
365490215a3Smrg #elif SANITIZER_FREEBSD || SANITIZER_OPENBSD
366490215a3Smrg   return internal_syscall(SYSCALL(fstatat), AT_FDCWD, (uptr)path, (uptr)buf,
367490215a3Smrg                           AT_SYMLINK_NOFOLLOW);
368490215a3Smrg #elif SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
369490215a3Smrg   return internal_syscall(SYSCALL(newfstatat), AT_FDCWD, (uptr)path, (uptr)buf,
370490215a3Smrg                           AT_SYMLINK_NOFOLLOW);
371490215a3Smrg #elif SANITIZER_LINUX_USES_64BIT_SYSCALLS
372490215a3Smrg # if SANITIZER_MIPS64
373490215a3Smrg   // For mips64, lstat syscall fills buffer in the format of kernel_stat
374490215a3Smrg   struct kernel_stat kbuf;
375490215a3Smrg   int res = internal_syscall(SYSCALL(lstat), path, &kbuf);
376490215a3Smrg   kernel_stat_to_stat(&kbuf, (struct stat *)buf);
377490215a3Smrg   return res;
378490215a3Smrg # else
379490215a3Smrg   return internal_syscall(SYSCALL(lstat), (uptr)path, (uptr)buf);
380490215a3Smrg # endif
381490215a3Smrg #else
382490215a3Smrg   struct stat64 buf64;
383490215a3Smrg   int res = internal_syscall(SYSCALL(lstat64), path, &buf64);
384490215a3Smrg   stat64_to_stat(&buf64, (struct stat *)buf);
385490215a3Smrg   return res;
386490215a3Smrg #endif
387490215a3Smrg }
388490215a3Smrg 
internal_fstat(fd_t fd,void * buf)389490215a3Smrg uptr internal_fstat(fd_t fd, void *buf) {
390490215a3Smrg #if SANITIZER_NETBSD
391490215a3Smrg   return internal_syscall_ptr(SYSCALL(fstat), fd, (uptr)buf);
392490215a3Smrg #elif SANITIZER_FREEBSD || SANITIZER_OPENBSD || \
393490215a3Smrg     SANITIZER_LINUX_USES_64BIT_SYSCALLS
394490215a3Smrg #if SANITIZER_MIPS64 && !SANITIZER_OPENBSD
395490215a3Smrg   // For mips64, fstat syscall fills buffer in the format of kernel_stat
396490215a3Smrg   struct kernel_stat kbuf;
397490215a3Smrg   int res = internal_syscall(SYSCALL(fstat), fd, &kbuf);
398490215a3Smrg   kernel_stat_to_stat(&kbuf, (struct stat *)buf);
399490215a3Smrg   return res;
400490215a3Smrg # else
401490215a3Smrg   return internal_syscall(SYSCALL(fstat), fd, (uptr)buf);
402490215a3Smrg # endif
403490215a3Smrg #else
404490215a3Smrg   struct stat64 buf64;
405490215a3Smrg   int res = internal_syscall(SYSCALL(fstat64), fd, &buf64);
406490215a3Smrg   stat64_to_stat(&buf64, (struct stat *)buf);
407490215a3Smrg   return res;
408490215a3Smrg #endif
409490215a3Smrg }
410490215a3Smrg 
internal_filesize(fd_t fd)411490215a3Smrg uptr internal_filesize(fd_t fd) {
412490215a3Smrg   struct stat st;
413490215a3Smrg   if (internal_fstat(fd, &st))
414490215a3Smrg     return -1;
415490215a3Smrg   return (uptr)st.st_size;
416490215a3Smrg }
417490215a3Smrg 
internal_dup2(int oldfd,int newfd)418490215a3Smrg uptr internal_dup2(int oldfd, int newfd) {
419490215a3Smrg #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
420490215a3Smrg   return internal_syscall(SYSCALL(dup3), oldfd, newfd, 0);
421490215a3Smrg #else
422490215a3Smrg   return internal_syscall(SYSCALL(dup2), oldfd, newfd);
423490215a3Smrg #endif
424490215a3Smrg }
425490215a3Smrg 
internal_readlink(const char * path,char * buf,uptr bufsize)426490215a3Smrg uptr internal_readlink(const char *path, char *buf, uptr bufsize) {
427490215a3Smrg #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
428490215a3Smrg   return internal_syscall(SYSCALL(readlinkat), AT_FDCWD, (uptr)path, (uptr)buf,
429490215a3Smrg                           bufsize);
430490215a3Smrg #elif SANITIZER_OPENBSD
431490215a3Smrg   return internal_syscall(SYSCALL(readlinkat), AT_FDCWD, (uptr)path, (uptr)buf,
432490215a3Smrg                           bufsize);
433490215a3Smrg #else
434490215a3Smrg   return internal_syscall(SYSCALL(readlink), (uptr)path, (uptr)buf, bufsize);
435490215a3Smrg #endif
436490215a3Smrg }
437490215a3Smrg 
internal_unlink(const char * path)438490215a3Smrg uptr internal_unlink(const char *path) {
439490215a3Smrg #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS || SANITIZER_OPENBSD
440490215a3Smrg   return internal_syscall(SYSCALL(unlinkat), AT_FDCWD, (uptr)path, 0);
441490215a3Smrg #else
442490215a3Smrg   return internal_syscall_ptr(SYSCALL(unlink), (uptr)path);
443490215a3Smrg #endif
444490215a3Smrg }
445490215a3Smrg 
internal_rename(const char * oldpath,const char * newpath)446490215a3Smrg uptr internal_rename(const char *oldpath, const char *newpath) {
447490215a3Smrg #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS || SANITIZER_OPENBSD
448490215a3Smrg   return internal_syscall(SYSCALL(renameat), AT_FDCWD, (uptr)oldpath, AT_FDCWD,
449490215a3Smrg                           (uptr)newpath);
450490215a3Smrg #else
451490215a3Smrg   return internal_syscall_ptr(SYSCALL(rename), (uptr)oldpath, (uptr)newpath);
452490215a3Smrg #endif
453490215a3Smrg }
454490215a3Smrg 
internal_sched_yield()455490215a3Smrg uptr internal_sched_yield() {
456490215a3Smrg   return internal_syscall(SYSCALL(sched_yield));
457490215a3Smrg }
458490215a3Smrg 
internal__exit(int exitcode)459490215a3Smrg void internal__exit(int exitcode) {
460490215a3Smrg #if SANITIZER_FREEBSD || SANITIZER_OPENBSD
461490215a3Smrg   internal_syscall(SYSCALL(exit), exitcode);
462490215a3Smrg #else
463490215a3Smrg   internal_syscall(SYSCALL(exit_group), exitcode);
464490215a3Smrg #endif
465490215a3Smrg   Die();  // Unreachable.
466490215a3Smrg }
467490215a3Smrg 
internal_sleep(unsigned int seconds)468490215a3Smrg unsigned int internal_sleep(unsigned int seconds) {
469490215a3Smrg   struct timespec ts;
470490215a3Smrg   ts.tv_sec = 1;
471490215a3Smrg   ts.tv_nsec = 0;
472490215a3Smrg   int res = internal_syscall_ptr(SYSCALL(nanosleep), &ts, &ts);
473490215a3Smrg   if (res) return ts.tv_sec;
474490215a3Smrg   return 0;
475490215a3Smrg }
476490215a3Smrg 
internal_execve(const char * filename,char * const argv[],char * const envp[])477490215a3Smrg uptr internal_execve(const char *filename, char *const argv[],
478490215a3Smrg                      char *const envp[]) {
479490215a3Smrg   return internal_syscall_ptr(SYSCALL(execve), (uptr)filename, (uptr)argv,
480490215a3Smrg                           (uptr)envp);
481490215a3Smrg }
482490215a3Smrg #endif  // !SANITIZER_SOLARIS && !SANITIZER_NETBSD
483490215a3Smrg 
484490215a3Smrg // ----------------- sanitizer_common.h
FileExists(const char * filename)485490215a3Smrg bool FileExists(const char *filename) {
486490215a3Smrg   struct stat st;
487490215a3Smrg #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
488490215a3Smrg   if (internal_syscall(SYSCALL(newfstatat), AT_FDCWD, filename, &st, 0))
489490215a3Smrg #else
490490215a3Smrg   if (internal_stat(filename, &st))
491490215a3Smrg #endif
492490215a3Smrg     return false;
493490215a3Smrg   // Sanity check: filename is a regular file.
494490215a3Smrg   return S_ISREG(st.st_mode);
495490215a3Smrg }
496490215a3Smrg 
497490215a3Smrg #if !SANITIZER_NETBSD
GetTid()498490215a3Smrg tid_t GetTid() {
499490215a3Smrg #if SANITIZER_FREEBSD
500490215a3Smrg   long Tid;
501490215a3Smrg   thr_self(&Tid);
502490215a3Smrg   return Tid;
503490215a3Smrg #elif SANITIZER_OPENBSD
504490215a3Smrg   return internal_syscall(SYSCALL(getthrid));
505490215a3Smrg #elif SANITIZER_SOLARIS
506490215a3Smrg   return thr_self();
507490215a3Smrg #else
508490215a3Smrg   return internal_syscall(SYSCALL(gettid));
509490215a3Smrg #endif
510490215a3Smrg }
511490215a3Smrg 
TgKill(pid_t pid,tid_t tid,int sig)512490215a3Smrg int TgKill(pid_t pid, tid_t tid, int sig) {
513490215a3Smrg #if SANITIZER_LINUX
514490215a3Smrg   return internal_syscall(SYSCALL(tgkill), pid, tid, sig);
515490215a3Smrg #elif SANITIZER_FREEBSD
516490215a3Smrg   return internal_syscall(SYSCALL(thr_kill2), pid, tid, sig);
517490215a3Smrg #elif SANITIZER_OPENBSD
518490215a3Smrg   (void)pid;
519490215a3Smrg   return internal_syscall(SYSCALL(thrkill), tid, sig, nullptr);
520490215a3Smrg #elif SANITIZER_SOLARIS
521490215a3Smrg   (void)pid;
522490215a3Smrg   return thr_kill(tid, sig);
523490215a3Smrg #endif
524490215a3Smrg }
525490215a3Smrg #endif
526490215a3Smrg 
527490215a3Smrg #if !SANITIZER_SOLARIS && !SANITIZER_NETBSD
NanoTime()528490215a3Smrg u64 NanoTime() {
529490215a3Smrg #if SANITIZER_FREEBSD || SANITIZER_OPENBSD
530490215a3Smrg   timeval tv;
531490215a3Smrg #else
532490215a3Smrg   kernel_timeval tv;
533490215a3Smrg #endif
534490215a3Smrg   internal_memset(&tv, 0, sizeof(tv));
535490215a3Smrg   internal_syscall(SYSCALL(gettimeofday), &tv, 0);
536490215a3Smrg   return (u64)tv.tv_sec * 1000*1000*1000 + tv.tv_usec * 1000;
537490215a3Smrg }
538490215a3Smrg 
internal_clock_gettime(__sanitizer_clockid_t clk_id,void * tp)539490215a3Smrg uptr internal_clock_gettime(__sanitizer_clockid_t clk_id, void *tp) {
540490215a3Smrg   return internal_syscall(SYSCALL(clock_gettime), clk_id, tp);
541490215a3Smrg }
542490215a3Smrg #endif  // !SANITIZER_SOLARIS && !SANITIZER_NETBSD
543490215a3Smrg 
544490215a3Smrg // Like getenv, but reads env directly from /proc (on Linux) or parses the
545490215a3Smrg // 'environ' array (on some others) and does not use libc. This function
546490215a3Smrg // should be called first inside __asan_init.
GetEnv(const char * name)547490215a3Smrg const char *GetEnv(const char *name) {
548490215a3Smrg #if SANITIZER_FREEBSD || SANITIZER_NETBSD || SANITIZER_OPENBSD || \
549490215a3Smrg     SANITIZER_SOLARIS
550490215a3Smrg   if (::environ != 0) {
551490215a3Smrg     uptr NameLen = internal_strlen(name);
552490215a3Smrg     for (char **Env = ::environ; *Env != 0; Env++) {
553490215a3Smrg       if (internal_strncmp(*Env, name, NameLen) == 0 && (*Env)[NameLen] == '=')
554490215a3Smrg         return (*Env) + NameLen + 1;
555490215a3Smrg     }
556490215a3Smrg   }
557490215a3Smrg   return 0;  // Not found.
558490215a3Smrg #elif SANITIZER_LINUX
559490215a3Smrg   static char *environ;
560490215a3Smrg   static uptr len;
561490215a3Smrg   static bool inited;
562490215a3Smrg   if (!inited) {
563490215a3Smrg     inited = true;
564490215a3Smrg     uptr environ_size;
565490215a3Smrg     if (!ReadFileToBuffer("/proc/self/environ", &environ, &environ_size, &len))
566490215a3Smrg       environ = nullptr;
567490215a3Smrg   }
568490215a3Smrg   if (!environ || len == 0) return nullptr;
569490215a3Smrg   uptr namelen = internal_strlen(name);
570490215a3Smrg   const char *p = environ;
571490215a3Smrg   while (*p != '\0') {  // will happen at the \0\0 that terminates the buffer
572490215a3Smrg     // proc file has the format NAME=value\0NAME=value\0NAME=value\0...
573490215a3Smrg     const char* endp =
574490215a3Smrg         (char*)internal_memchr(p, '\0', len - (p - environ));
575490215a3Smrg     if (!endp)  // this entry isn't NUL terminated
576490215a3Smrg       return nullptr;
577490215a3Smrg     else if (!internal_memcmp(p, name, namelen) && p[namelen] == '=')  // Match.
578490215a3Smrg       return p + namelen + 1;  // point after =
579490215a3Smrg     p = endp + 1;
580490215a3Smrg   }
581490215a3Smrg   return nullptr;  // Not found.
582490215a3Smrg #else
583490215a3Smrg #error "Unsupported platform"
584490215a3Smrg #endif
585490215a3Smrg }
586490215a3Smrg 
587490215a3Smrg #if !SANITIZER_FREEBSD && !SANITIZER_NETBSD && !SANITIZER_OPENBSD
588490215a3Smrg extern "C" {
589490215a3Smrg SANITIZER_WEAK_ATTRIBUTE extern void *__libc_stack_end;
590490215a3Smrg }
591490215a3Smrg #endif
592490215a3Smrg 
593490215a3Smrg #if !SANITIZER_GO && !SANITIZER_FREEBSD && !SANITIZER_NETBSD &&                \
594490215a3Smrg     !SANITIZER_OPENBSD
ReadNullSepFileToArray(const char * path,char *** arr,int arr_size)595490215a3Smrg static void ReadNullSepFileToArray(const char *path, char ***arr,
596490215a3Smrg                                    int arr_size) {
597490215a3Smrg   char *buff;
598490215a3Smrg   uptr buff_size;
599490215a3Smrg   uptr buff_len;
600490215a3Smrg   *arr = (char **)MmapOrDie(arr_size * sizeof(char *), "NullSepFileArray");
601490215a3Smrg   if (!ReadFileToBuffer(path, &buff, &buff_size, &buff_len, 1024 * 1024)) {
602490215a3Smrg     (*arr)[0] = nullptr;
603490215a3Smrg     return;
604490215a3Smrg   }
605490215a3Smrg   (*arr)[0] = buff;
606490215a3Smrg   int count, i;
607490215a3Smrg   for (count = 1, i = 1; ; i++) {
608490215a3Smrg     if (buff[i] == 0) {
609490215a3Smrg       if (buff[i+1] == 0) break;
610490215a3Smrg       (*arr)[count] = &buff[i+1];
611490215a3Smrg       CHECK_LE(count, arr_size - 1);  // FIXME: make this more flexible.
612490215a3Smrg       count++;
613490215a3Smrg     }
614490215a3Smrg   }
615490215a3Smrg   (*arr)[count] = nullptr;
616490215a3Smrg }
617490215a3Smrg #endif
618490215a3Smrg 
619490215a3Smrg #if !SANITIZER_OPENBSD
GetArgsAndEnv(char *** argv,char *** envp)620490215a3Smrg static void GetArgsAndEnv(char ***argv, char ***envp) {
621490215a3Smrg #if SANITIZER_FREEBSD
622490215a3Smrg   // On FreeBSD, retrieving the argument and environment arrays is done via the
623490215a3Smrg   // kern.ps_strings sysctl, which returns a pointer to a structure containing
624490215a3Smrg   // this information. See also <sys/exec.h>.
625490215a3Smrg   ps_strings *pss;
626490215a3Smrg   uptr sz = sizeof(pss);
627490215a3Smrg   if (internal_sysctlbyname("kern.ps_strings", &pss, &sz, NULL, 0) == -1) {
628490215a3Smrg     Printf("sysctl kern.ps_strings failed\n");
629490215a3Smrg     Die();
630490215a3Smrg   }
631490215a3Smrg   *argv = pss->ps_argvstr;
632490215a3Smrg   *envp = pss->ps_envstr;
633490215a3Smrg #elif SANITIZER_NETBSD
634490215a3Smrg   *argv = __ps_strings->ps_argvstr;
635490215a3Smrg   *envp = __ps_strings->ps_envstr;
636490215a3Smrg #else // SANITIZER_FREEBSD
637490215a3Smrg #if !SANITIZER_GO
638490215a3Smrg   if (&__libc_stack_end) {
639490215a3Smrg #endif // !SANITIZER_GO
640490215a3Smrg     uptr* stack_end = (uptr*)__libc_stack_end;
641490215a3Smrg     int argc = *stack_end;
642490215a3Smrg     *argv = (char**)(stack_end + 1);
643490215a3Smrg     *envp = (char**)(stack_end + argc + 2);
644490215a3Smrg #if !SANITIZER_GO
645490215a3Smrg   } else {
646490215a3Smrg     static const int kMaxArgv = 2000, kMaxEnvp = 2000;
647490215a3Smrg     ReadNullSepFileToArray("/proc/self/cmdline", argv, kMaxArgv);
648490215a3Smrg     ReadNullSepFileToArray("/proc/self/environ", envp, kMaxEnvp);
649490215a3Smrg   }
650490215a3Smrg #endif // !SANITIZER_GO
651490215a3Smrg #endif // SANITIZER_FREEBSD
652490215a3Smrg }
653490215a3Smrg 
GetArgv()654490215a3Smrg char **GetArgv() {
655490215a3Smrg   char **argv, **envp;
656490215a3Smrg   GetArgsAndEnv(&argv, &envp);
657490215a3Smrg   return argv;
658490215a3Smrg }
659490215a3Smrg 
ReExec()660490215a3Smrg void ReExec() {
661490215a3Smrg   char **argv, **envp;
662490215a3Smrg   const char *pathname = "/proc/self/exe";
663490215a3Smrg 
664490215a3Smrg #if SANITIZER_NETBSD
665490215a3Smrg   static const int name[] = {
666490215a3Smrg     CTL_KERN, KERN_PROC_ARGS, -1, KERN_PROC_PATHNAME,
667490215a3Smrg   };
668490215a3Smrg   char path[400];
669490215a3Smrg   uptr len;
670490215a3Smrg 
671490215a3Smrg   len = sizeof(path);
672490215a3Smrg   if (internal_sysctl(name, ARRAY_SIZE(name), path, &len, NULL, 0) != -1)
673490215a3Smrg     pathname = path;
674490215a3Smrg #elif SANITIZER_SOLARIS
675490215a3Smrg   pathname = getexecname();
676490215a3Smrg   CHECK_NE(pathname, NULL);
677490215a3Smrg #endif
678490215a3Smrg 
679490215a3Smrg   GetArgsAndEnv(&argv, &envp);
680490215a3Smrg   uptr rv = internal_execve(pathname, argv, envp);
681490215a3Smrg   int rverrno;
682490215a3Smrg   CHECK_EQ(internal_iserror(rv, &rverrno), true);
683490215a3Smrg   Printf("execve failed, errno %d\n", rverrno);
684490215a3Smrg   Die();
685490215a3Smrg }
686490215a3Smrg #endif
687490215a3Smrg 
688490215a3Smrg #if !SANITIZER_SOLARIS
689490215a3Smrg enum MutexState {
690490215a3Smrg   MtxUnlocked = 0,
691490215a3Smrg   MtxLocked = 1,
692490215a3Smrg   MtxSleeping = 2
693490215a3Smrg };
694490215a3Smrg 
BlockingMutex()695490215a3Smrg BlockingMutex::BlockingMutex() {
696490215a3Smrg   internal_memset(this, 0, sizeof(*this));
697490215a3Smrg }
698490215a3Smrg 
Lock()699490215a3Smrg void BlockingMutex::Lock() {
700490215a3Smrg   CHECK_EQ(owner_, 0);
701490215a3Smrg   atomic_uint32_t *m = reinterpret_cast<atomic_uint32_t *>(&opaque_storage_);
702490215a3Smrg   if (atomic_exchange(m, MtxLocked, memory_order_acquire) == MtxUnlocked)
703490215a3Smrg     return;
704490215a3Smrg   while (atomic_exchange(m, MtxSleeping, memory_order_acquire) != MtxUnlocked) {
705490215a3Smrg #if SANITIZER_FREEBSD
706490215a3Smrg     _umtx_op(m, UMTX_OP_WAIT_UINT, MtxSleeping, 0, 0);
707490215a3Smrg #elif SANITIZER_NETBSD
708490215a3Smrg     sched_yield(); /* No userspace futex-like synchronization */
709490215a3Smrg #else
710490215a3Smrg     internal_syscall(SYSCALL(futex), (uptr)m, FUTEX_WAIT_PRIVATE, MtxSleeping,
711490215a3Smrg                      0, 0, 0);
712490215a3Smrg #endif
713490215a3Smrg   }
714490215a3Smrg }
715490215a3Smrg 
Unlock()716490215a3Smrg void BlockingMutex::Unlock() {
717490215a3Smrg   atomic_uint32_t *m = reinterpret_cast<atomic_uint32_t *>(&opaque_storage_);
718490215a3Smrg   u32 v = atomic_exchange(m, MtxUnlocked, memory_order_release);
719490215a3Smrg   CHECK_NE(v, MtxUnlocked);
720490215a3Smrg   if (v == MtxSleeping) {
721490215a3Smrg #if SANITIZER_FREEBSD
722490215a3Smrg     _umtx_op(m, UMTX_OP_WAKE, 1, 0, 0);
723490215a3Smrg #elif SANITIZER_NETBSD
724490215a3Smrg                    /* No userspace futex-like synchronization */
725490215a3Smrg #else
726490215a3Smrg     internal_syscall(SYSCALL(futex), (uptr)m, FUTEX_WAKE_PRIVATE, 1, 0, 0, 0);
727490215a3Smrg #endif
728490215a3Smrg   }
729490215a3Smrg }
730490215a3Smrg 
CheckLocked()731490215a3Smrg void BlockingMutex::CheckLocked() {
732490215a3Smrg   atomic_uint32_t *m = reinterpret_cast<atomic_uint32_t *>(&opaque_storage_);
733490215a3Smrg   CHECK_NE(MtxUnlocked, atomic_load(m, memory_order_relaxed));
734490215a3Smrg }
735490215a3Smrg #endif // !SANITIZER_SOLARIS
736490215a3Smrg 
737490215a3Smrg // ----------------- sanitizer_linux.h
738490215a3Smrg // The actual size of this structure is specified by d_reclen.
739490215a3Smrg // Note that getdents64 uses a different structure format. We only provide the
740490215a3Smrg // 32-bit syscall here.
741490215a3Smrg #if SANITIZER_NETBSD
742490215a3Smrg // Not used
743490215a3Smrg #elif SANITIZER_OPENBSD
744490215a3Smrg // struct dirent is different for Linux and us. At this moment, we use only
745490215a3Smrg // d_fileno (Linux call this d_ino), d_reclen, and d_name.
746490215a3Smrg struct linux_dirent {
747490215a3Smrg   u64 d_ino;  // d_fileno
748490215a3Smrg   u16 d_reclen;
749490215a3Smrg   u16 d_namlen;  // not used
750490215a3Smrg   u8 d_type;     // not used
751490215a3Smrg   char d_name[NAME_MAX + 1];
752490215a3Smrg };
753490215a3Smrg #else
754490215a3Smrg struct linux_dirent {
755490215a3Smrg #if SANITIZER_X32 || defined(__aarch64__)
756490215a3Smrg   u64 d_ino;
757490215a3Smrg   u64 d_off;
758490215a3Smrg #else
759490215a3Smrg   unsigned long      d_ino;
760490215a3Smrg   unsigned long      d_off;
761490215a3Smrg #endif
762490215a3Smrg   unsigned short     d_reclen;
763490215a3Smrg #ifdef __aarch64__
764490215a3Smrg   unsigned char      d_type;
765490215a3Smrg #endif
766490215a3Smrg   char               d_name[256];
767490215a3Smrg };
768490215a3Smrg #endif
769490215a3Smrg 
770490215a3Smrg #if !SANITIZER_SOLARIS && !SANITIZER_NETBSD
771490215a3Smrg // Syscall wrappers.
internal_ptrace(int request,int pid,void * addr,void * data)772490215a3Smrg uptr internal_ptrace(int request, int pid, void *addr, void *data) {
773490215a3Smrg   return internal_syscall(SYSCALL(ptrace), request, pid, (uptr)addr,
774490215a3Smrg                           (uptr)data);
775490215a3Smrg }
776490215a3Smrg 
internal_waitpid(int pid,int * status,int options)777490215a3Smrg uptr internal_waitpid(int pid, int *status, int options) {
778490215a3Smrg   return internal_syscall(SYSCALL(wait4), pid, (uptr)status, options,
779490215a3Smrg                           0 /* rusage */);
780490215a3Smrg }
781490215a3Smrg 
internal_getpid()782490215a3Smrg uptr internal_getpid() {
783490215a3Smrg   return internal_syscall(SYSCALL(getpid));
784490215a3Smrg }
785490215a3Smrg 
internal_getppid()786490215a3Smrg uptr internal_getppid() {
787490215a3Smrg   return internal_syscall(SYSCALL(getppid));
788490215a3Smrg }
789490215a3Smrg 
internal_dlinfo(void * handle,int request,void * p)790490215a3Smrg int internal_dlinfo(void *handle, int request, void *p) {
791490215a3Smrg #if SANITIZER_FREEBSD || (SANITIZER_LINUX && !SANITIZER_ANDROID) || \
792490215a3Smrg     SANITIZER_SOLARIS
793490215a3Smrg   return dlinfo(handle, request, p);
794490215a3Smrg #else
795490215a3Smrg   UNIMPLEMENTED();
796490215a3Smrg #endif
797490215a3Smrg }
798490215a3Smrg 
internal_getdents(fd_t fd,struct linux_dirent * dirp,unsigned int count)799490215a3Smrg uptr internal_getdents(fd_t fd, struct linux_dirent *dirp, unsigned int count) {
800490215a3Smrg #if SANITIZER_FREEBSD
801490215a3Smrg   return internal_syscall(SYSCALL(getdirentries), fd, (uptr)dirp, count, NULL);
802490215a3Smrg #elif SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
803490215a3Smrg   return internal_syscall(SYSCALL(getdents64), fd, (uptr)dirp, count);
804490215a3Smrg #else
805490215a3Smrg   return internal_syscall(SYSCALL(getdents), fd, (uptr)dirp, count);
806490215a3Smrg #endif
807490215a3Smrg }
808490215a3Smrg 
internal_lseek(fd_t fd,OFF_T offset,int whence)809490215a3Smrg uptr internal_lseek(fd_t fd, OFF_T offset, int whence) {
810490215a3Smrg   return internal_syscall(SYSCALL(lseek), fd, offset, whence);
811490215a3Smrg }
812490215a3Smrg 
813490215a3Smrg #if SANITIZER_LINUX
internal_prctl(int option,uptr arg2,uptr arg3,uptr arg4,uptr arg5)814490215a3Smrg uptr internal_prctl(int option, uptr arg2, uptr arg3, uptr arg4, uptr arg5) {
815490215a3Smrg   return internal_syscall(SYSCALL(prctl), option, arg2, arg3, arg4, arg5);
816490215a3Smrg }
817490215a3Smrg #endif
818490215a3Smrg 
internal_sigaltstack(const void * ss,void * oss)819490215a3Smrg uptr internal_sigaltstack(const void *ss, void *oss) {
820490215a3Smrg   return internal_syscall_ptr(SYSCALL(sigaltstack), (uptr)ss, (uptr)oss);
821490215a3Smrg }
822490215a3Smrg 
internal_fork()823490215a3Smrg int internal_fork() {
824490215a3Smrg #if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
825490215a3Smrg   return internal_syscall(SYSCALL(clone), SIGCHLD, 0);
826490215a3Smrg #else
827490215a3Smrg   return internal_syscall(SYSCALL(fork));
828490215a3Smrg #endif
829490215a3Smrg }
830490215a3Smrg 
831490215a3Smrg #if SANITIZER_NETBSD
832490215a3Smrg #include <signal.h>
833*c35a2b90Schristos extern "C" int __sigaction_siginfo(int, const struct sigaction *, struct sigaction *);
internal_sigaction_norestorer(int signum,const void * act,void * oldact)834490215a3Smrg int internal_sigaction_norestorer(int signum, const void *act, void *oldact) {
835490215a3Smrg 
836*c35a2b90Schristos     return __sigaction_siginfo(signum,
837490215a3Smrg 	reinterpret_cast<const struct sigaction *>(act),
838490215a3Smrg 	reinterpret_cast<struct sigaction *>(oldact));
839490215a3Smrg }
840490215a3Smrg #endif
841490215a3Smrg 
842490215a3Smrg #if SANITIZER_FREEBSD || SANITIZER_OPENBSD
internal_sysctl(const int * name,unsigned int namelen,void * oldp,uptr * oldlenp,const void * newp,uptr newlen)843490215a3Smrg int internal_sysctl(const int *name, unsigned int namelen, void *oldp,
844490215a3Smrg                     uptr *oldlenp, const void *newp, uptr newlen) {
845490215a3Smrg #if SANITIZER_OPENBSD
846490215a3Smrg   return sysctl(name, namelen, oldp, (size_t *)oldlenp, (void *)newp,
847490215a3Smrg                 (size_t)newlen);
848490215a3Smrg #else
849490215a3Smrg   return sysctl(name, namelen, oldp, (size_t *)oldlenp, newp, (size_t)newlen);
850490215a3Smrg #endif
851490215a3Smrg }
852490215a3Smrg 
853490215a3Smrg #if SANITIZER_FREEBSD
internal_sysctlbyname(const char * sname,void * oldp,uptr * oldlenp,const void * newp,uptr newlen)854490215a3Smrg int internal_sysctlbyname(const char *sname, void *oldp, uptr *oldlenp,
855490215a3Smrg                           const void *newp, uptr newlen) {
856490215a3Smrg   return sysctlbyname(sname, oldp, (size_t *)oldlenp, newp, (size_t)newlen);
857490215a3Smrg }
858490215a3Smrg #endif
859490215a3Smrg #endif
860490215a3Smrg 
861490215a3Smrg #if SANITIZER_LINUX
862490215a3Smrg #define SA_RESTORER 0x04000000
863490215a3Smrg // Doesn't set sa_restorer if the caller did not set it, so use with caution
864490215a3Smrg //(see below).
internal_sigaction_norestorer(int signum,const void * act,void * oldact)865490215a3Smrg int internal_sigaction_norestorer(int signum, const void *act, void *oldact) {
866490215a3Smrg   __sanitizer_kernel_sigaction_t k_act, k_oldact;
867490215a3Smrg   internal_memset(&k_act, 0, sizeof(__sanitizer_kernel_sigaction_t));
868490215a3Smrg   internal_memset(&k_oldact, 0, sizeof(__sanitizer_kernel_sigaction_t));
869490215a3Smrg   const __sanitizer_sigaction *u_act = (const __sanitizer_sigaction *)act;
870490215a3Smrg   __sanitizer_sigaction *u_oldact = (__sanitizer_sigaction *)oldact;
871490215a3Smrg   if (u_act) {
872490215a3Smrg     k_act.handler = u_act->handler;
873490215a3Smrg     k_act.sigaction = u_act->sigaction;
874490215a3Smrg     internal_memcpy(&k_act.sa_mask, &u_act->sa_mask,
875490215a3Smrg                     sizeof(__sanitizer_kernel_sigset_t));
876490215a3Smrg     // Without SA_RESTORER kernel ignores the calls (probably returns EINVAL).
877490215a3Smrg     k_act.sa_flags = u_act->sa_flags | SA_RESTORER;
878490215a3Smrg     // FIXME: most often sa_restorer is unset, however the kernel requires it
879490215a3Smrg     // to point to a valid signal restorer that calls the rt_sigreturn syscall.
880490215a3Smrg     // If sa_restorer passed to the kernel is NULL, the program may crash upon
881490215a3Smrg     // signal delivery or fail to unwind the stack in the signal handler.
882490215a3Smrg     // libc implementation of sigaction() passes its own restorer to
883490215a3Smrg     // rt_sigaction, so we need to do the same (we'll need to reimplement the
884490215a3Smrg     // restorers; for x86_64 the restorer address can be obtained from
885490215a3Smrg     // oldact->sa_restorer upon a call to sigaction(xxx, NULL, oldact).
886490215a3Smrg #if !SANITIZER_ANDROID || !SANITIZER_MIPS32
887490215a3Smrg     k_act.sa_restorer = u_act->sa_restorer;
888490215a3Smrg #endif
889490215a3Smrg   }
890490215a3Smrg 
891490215a3Smrg   uptr result = internal_syscall(SYSCALL(rt_sigaction), (uptr)signum,
892490215a3Smrg       (uptr)(u_act ? &k_act : nullptr),
893490215a3Smrg       (uptr)(u_oldact ? &k_oldact : nullptr),
894490215a3Smrg       (uptr)sizeof(__sanitizer_kernel_sigset_t));
895490215a3Smrg 
896490215a3Smrg   if ((result == 0) && u_oldact) {
897490215a3Smrg     u_oldact->handler = k_oldact.handler;
898490215a3Smrg     u_oldact->sigaction = k_oldact.sigaction;
899490215a3Smrg     internal_memcpy(&u_oldact->sa_mask, &k_oldact.sa_mask,
900490215a3Smrg                     sizeof(__sanitizer_kernel_sigset_t));
901490215a3Smrg     u_oldact->sa_flags = k_oldact.sa_flags;
902490215a3Smrg #if !SANITIZER_ANDROID || !SANITIZER_MIPS32
903490215a3Smrg     u_oldact->sa_restorer = k_oldact.sa_restorer;
904490215a3Smrg #endif
905490215a3Smrg   }
906490215a3Smrg   return result;
907490215a3Smrg }
908490215a3Smrg 
909490215a3Smrg // Invokes sigaction via a raw syscall with a restorer, but does not support
910490215a3Smrg // all platforms yet.
911490215a3Smrg // We disable for Go simply because we have not yet added to buildgo.sh.
912490215a3Smrg #if (defined(__x86_64__) || SANITIZER_MIPS64) && !SANITIZER_GO
internal_sigaction_syscall(int signum,const void * act,void * oldact)913490215a3Smrg int internal_sigaction_syscall(int signum, const void *act, void *oldact) {
914490215a3Smrg   if (act == nullptr)
915490215a3Smrg     return internal_sigaction_norestorer(signum, act, oldact);
916490215a3Smrg   __sanitizer_sigaction u_adjust;
917490215a3Smrg   internal_memcpy(&u_adjust, act, sizeof(u_adjust));
918490215a3Smrg #if !SANITIZER_ANDROID || !SANITIZER_MIPS32
919490215a3Smrg   if (u_adjust.sa_restorer == nullptr) {
920490215a3Smrg     u_adjust.sa_restorer = internal_sigreturn;
921490215a3Smrg   }
922490215a3Smrg #endif
923490215a3Smrg   return internal_sigaction_norestorer(signum, (const void *)&u_adjust, oldact);
924490215a3Smrg }
925490215a3Smrg #endif  // defined(__x86_64__) && !SANITIZER_GO
926490215a3Smrg #endif  // SANITIZER_LINUX
927490215a3Smrg 
internal_sigprocmask(int how,__sanitizer_sigset_t * set,__sanitizer_sigset_t * oldset)928490215a3Smrg uptr internal_sigprocmask(int how, __sanitizer_sigset_t *set,
929490215a3Smrg                           __sanitizer_sigset_t *oldset) {
930490215a3Smrg #if SANITIZER_FREEBSD || SANITIZER_OPENBSD
931490215a3Smrg   return internal_syscall(SYSCALL(sigprocmask), how, set, oldset);
932490215a3Smrg #else
933490215a3Smrg   __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
934490215a3Smrg   __sanitizer_kernel_sigset_t *k_oldset = (__sanitizer_kernel_sigset_t *)oldset;
935490215a3Smrg   return internal_syscall(SYSCALL(rt_sigprocmask), (uptr)how,
936490215a3Smrg                           (uptr)&k_set->sig[0], (uptr)&k_oldset->sig[0],
937490215a3Smrg                           sizeof(__sanitizer_kernel_sigset_t));
938490215a3Smrg #endif
939490215a3Smrg }
940490215a3Smrg 
internal_sigfillset(__sanitizer_sigset_t * set)941490215a3Smrg void internal_sigfillset(__sanitizer_sigset_t *set) {
942490215a3Smrg   internal_memset(set, 0xff, sizeof(*set));
943490215a3Smrg }
944490215a3Smrg 
internal_sigemptyset(__sanitizer_sigset_t * set)945490215a3Smrg void internal_sigemptyset(__sanitizer_sigset_t *set) {
946490215a3Smrg   internal_memset(set, 0, sizeof(*set));
947490215a3Smrg }
948490215a3Smrg 
949490215a3Smrg #if SANITIZER_LINUX
internal_sigdelset(__sanitizer_sigset_t * set,int signum)950490215a3Smrg void internal_sigdelset(__sanitizer_sigset_t *set, int signum) {
951490215a3Smrg   signum -= 1;
952490215a3Smrg   CHECK_GE(signum, 0);
953490215a3Smrg   CHECK_LT(signum, sizeof(*set) * 8);
954490215a3Smrg   __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
955490215a3Smrg   const uptr idx = signum / (sizeof(k_set->sig[0]) * 8);
956490215a3Smrg   const uptr bit = signum % (sizeof(k_set->sig[0]) * 8);
957490215a3Smrg   k_set->sig[idx] &= ~(1 << bit);
958490215a3Smrg }
959490215a3Smrg 
internal_sigismember(__sanitizer_sigset_t * set,int signum)960490215a3Smrg bool internal_sigismember(__sanitizer_sigset_t *set, int signum) {
961490215a3Smrg   signum -= 1;
962490215a3Smrg   CHECK_GE(signum, 0);
963490215a3Smrg   CHECK_LT(signum, sizeof(*set) * 8);
964490215a3Smrg   __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
965490215a3Smrg   const uptr idx = signum / (sizeof(k_set->sig[0]) * 8);
966490215a3Smrg   const uptr bit = signum % (sizeof(k_set->sig[0]) * 8);
967490215a3Smrg   return k_set->sig[idx] & (1 << bit);
968490215a3Smrg }
969490215a3Smrg #elif SANITIZER_FREEBSD
internal_sigdelset(__sanitizer_sigset_t * set,int signum)970490215a3Smrg void internal_sigdelset(__sanitizer_sigset_t *set, int signum) {
971490215a3Smrg   sigset_t *rset = reinterpret_cast<sigset_t *>(set);
972490215a3Smrg   sigdelset(rset, signum);
973490215a3Smrg }
974490215a3Smrg 
internal_sigismember(__sanitizer_sigset_t * set,int signum)975490215a3Smrg bool internal_sigismember(__sanitizer_sigset_t *set, int signum) {
976490215a3Smrg   sigset_t *rset = reinterpret_cast<sigset_t *>(set);
977490215a3Smrg   return sigismember(rset, signum);
978490215a3Smrg }
979490215a3Smrg #endif
980490215a3Smrg #endif // !SANITIZER_SOLARIS
981490215a3Smrg 
982490215a3Smrg #if !SANITIZER_NETBSD
983490215a3Smrg // ThreadLister implementation.
ThreadLister(pid_t pid)984490215a3Smrg ThreadLister::ThreadLister(pid_t pid) : pid_(pid), buffer_(4096) {
985490215a3Smrg   char task_directory_path[80];
986490215a3Smrg   internal_snprintf(task_directory_path, sizeof(task_directory_path),
987490215a3Smrg                     "/proc/%d/task/", pid);
988490215a3Smrg   descriptor_ = internal_open(task_directory_path, O_RDONLY | O_DIRECTORY);
989490215a3Smrg   if (internal_iserror(descriptor_)) {
990490215a3Smrg     Report("Can't open /proc/%d/task for reading.\n", pid);
991490215a3Smrg   }
992490215a3Smrg }
993490215a3Smrg 
ListThreads(InternalMmapVector<tid_t> * threads)994490215a3Smrg ThreadLister::Result ThreadLister::ListThreads(
995490215a3Smrg     InternalMmapVector<tid_t> *threads) {
996490215a3Smrg   if (internal_iserror(descriptor_))
997490215a3Smrg     return Error;
998490215a3Smrg   internal_lseek(descriptor_, 0, SEEK_SET);
999490215a3Smrg   threads->clear();
1000490215a3Smrg 
1001490215a3Smrg   Result result = Ok;
1002490215a3Smrg   for (bool first_read = true;; first_read = false) {
1003490215a3Smrg     // Resize to max capacity if it was downsized by IsAlive.
1004490215a3Smrg     buffer_.resize(buffer_.capacity());
1005490215a3Smrg     CHECK_GE(buffer_.size(), 4096);
1006490215a3Smrg     uptr read = internal_getdents(
1007490215a3Smrg         descriptor_, (struct linux_dirent *)buffer_.data(), buffer_.size());
1008490215a3Smrg     if (!read)
1009490215a3Smrg       return result;
1010490215a3Smrg     if (internal_iserror(read)) {
1011490215a3Smrg       Report("Can't read directory entries from /proc/%d/task.\n", pid_);
1012490215a3Smrg       return Error;
1013490215a3Smrg     }
1014490215a3Smrg 
1015490215a3Smrg     for (uptr begin = (uptr)buffer_.data(), end = begin + read; begin < end;) {
1016490215a3Smrg       struct linux_dirent *entry = (struct linux_dirent *)begin;
1017490215a3Smrg       begin += entry->d_reclen;
1018490215a3Smrg       if (entry->d_ino == 1) {
1019490215a3Smrg         // Inode 1 is for bad blocks and also can be a reason for early return.
1020490215a3Smrg         // Should be emitted if kernel tried to output terminating thread.
1021490215a3Smrg         // See proc_task_readdir implementation in Linux.
1022490215a3Smrg         result = Incomplete;
1023490215a3Smrg       }
1024490215a3Smrg       if (entry->d_ino && *entry->d_name >= '0' && *entry->d_name <= '9')
1025490215a3Smrg         threads->push_back(internal_atoll(entry->d_name));
1026490215a3Smrg     }
1027490215a3Smrg 
1028490215a3Smrg     // Now we are going to detect short-read or early EOF. In such cases Linux
1029490215a3Smrg     // can return inconsistent list with missing alive threads.
1030490215a3Smrg     // Code will just remember that the list can be incomplete but it will
1031490215a3Smrg     // continue reads to return as much as possible.
1032490215a3Smrg     if (!first_read) {
1033490215a3Smrg       // The first one was a short-read by definition.
1034490215a3Smrg       result = Incomplete;
1035490215a3Smrg     } else if (read > buffer_.size() - 1024) {
1036490215a3Smrg       // Read was close to the buffer size. So double the size and assume the
1037490215a3Smrg       // worst.
1038490215a3Smrg       buffer_.resize(buffer_.size() * 2);
1039490215a3Smrg       result = Incomplete;
1040490215a3Smrg     } else if (!threads->empty() && !IsAlive(threads->back())) {
1041490215a3Smrg       // Maybe Linux early returned from read on terminated thread (!pid_alive)
1042490215a3Smrg       // and failed to restore read position.
1043490215a3Smrg       // See next_tid and proc_task_instantiate in Linux.
1044490215a3Smrg       result = Incomplete;
1045490215a3Smrg     }
1046490215a3Smrg   }
1047490215a3Smrg }
1048490215a3Smrg 
IsAlive(int tid)1049490215a3Smrg bool ThreadLister::IsAlive(int tid) {
1050490215a3Smrg   // /proc/%d/task/%d/status uses same call to detect alive threads as
1051490215a3Smrg   // proc_task_readdir. See task_state implementation in Linux.
1052490215a3Smrg   char path[80];
1053490215a3Smrg   internal_snprintf(path, sizeof(path), "/proc/%d/task/%d/status", pid_, tid);
1054490215a3Smrg   if (!ReadFileToVector(path, &buffer_) || buffer_.empty())
1055490215a3Smrg     return false;
1056490215a3Smrg   buffer_.push_back(0);
1057490215a3Smrg   static const char kPrefix[] = "\nPPid:";
1058490215a3Smrg   const char *field = internal_strstr(buffer_.data(), kPrefix);
1059490215a3Smrg   if (!field)
1060490215a3Smrg     return false;
1061490215a3Smrg   field += internal_strlen(kPrefix);
1062490215a3Smrg   return (int)internal_atoll(field) != 0;
1063490215a3Smrg }
1064490215a3Smrg 
~ThreadLister()1065490215a3Smrg ThreadLister::~ThreadLister() {
1066490215a3Smrg   if (!internal_iserror(descriptor_))
1067490215a3Smrg     internal_close(descriptor_);
1068490215a3Smrg }
1069490215a3Smrg #endif
1070490215a3Smrg 
1071490215a3Smrg #if SANITIZER_WORDSIZE == 32
1072490215a3Smrg // Take care of unusable kernel area in top gigabyte.
GetKernelAreaSize()1073490215a3Smrg static uptr GetKernelAreaSize() {
1074490215a3Smrg #if SANITIZER_LINUX && !SANITIZER_X32
1075490215a3Smrg   const uptr gbyte = 1UL << 30;
1076490215a3Smrg 
1077490215a3Smrg   // Firstly check if there are writable segments
1078490215a3Smrg   // mapped to top gigabyte (e.g. stack).
1079490215a3Smrg   MemoryMappingLayout proc_maps(/*cache_enabled*/true);
1080490215a3Smrg   MemoryMappedSegment segment;
1081490215a3Smrg   while (proc_maps.Next(&segment)) {
1082490215a3Smrg     if ((segment.end >= 3 * gbyte) && segment.IsWritable()) return 0;
1083490215a3Smrg   }
1084490215a3Smrg 
1085490215a3Smrg #if !SANITIZER_ANDROID
1086490215a3Smrg   // Even if nothing is mapped, top Gb may still be accessible
1087490215a3Smrg   // if we are running on 64-bit kernel.
1088490215a3Smrg   // Uname may report misleading results if personality type
1089490215a3Smrg   // is modified (e.g. under schroot) so check this as well.
1090490215a3Smrg   struct utsname uname_info;
1091490215a3Smrg   int pers = personality(0xffffffffUL);
1092490215a3Smrg   if (!(pers & PER_MASK)
1093490215a3Smrg       && uname(&uname_info) == 0
1094490215a3Smrg       && internal_strstr(uname_info.machine, "64"))
1095490215a3Smrg     return 0;
1096490215a3Smrg #endif  // SANITIZER_ANDROID
1097490215a3Smrg 
1098490215a3Smrg   // Top gigabyte is reserved for kernel.
1099490215a3Smrg   return gbyte;
1100490215a3Smrg #else
1101490215a3Smrg   return 0;
1102490215a3Smrg #endif  // SANITIZER_LINUX && !SANITIZER_X32
1103490215a3Smrg }
1104490215a3Smrg #endif  // SANITIZER_WORDSIZE == 32
1105490215a3Smrg 
GetMaxVirtualAddress()1106490215a3Smrg uptr GetMaxVirtualAddress() {
1107490215a3Smrg #if (SANITIZER_NETBSD || SANITIZER_OPENBSD) && defined(__x86_64__)
1108490215a3Smrg   return 0x7f7ffffff000ULL;  // (0x00007f8000000000 - PAGE_SIZE)
1109490215a3Smrg #elif SANITIZER_WORDSIZE == 64
1110490215a3Smrg # if defined(__powerpc64__) || defined(__aarch64__)
1111490215a3Smrg   // On PowerPC64 we have two different address space layouts: 44- and 46-bit.
1112490215a3Smrg   // We somehow need to figure out which one we are using now and choose
1113490215a3Smrg   // one of 0x00000fffffffffffUL and 0x00003fffffffffffUL.
1114490215a3Smrg   // Note that with 'ulimit -s unlimited' the stack is moved away from the top
1115490215a3Smrg   // of the address space, so simply checking the stack address is not enough.
1116490215a3Smrg   // This should (does) work for both PowerPC64 Endian modes.
1117490215a3Smrg   // Similarly, aarch64 has multiple address space layouts: 39, 42 and 47-bit.
1118490215a3Smrg   return (1ULL << (MostSignificantSetBitIndex(GET_CURRENT_FRAME()) + 1)) - 1;
1119490215a3Smrg # elif defined(__mips64)
1120490215a3Smrg   return (1ULL << 40) - 1;  // 0x000000ffffffffffUL;
1121490215a3Smrg # elif defined(__s390x__)
1122490215a3Smrg   return (1ULL << 53) - 1;  // 0x001fffffffffffffUL;
1123490215a3Smrg # elif defined(__sparc__)
1124490215a3Smrg   return ~(uptr)0;
1125490215a3Smrg # else
1126490215a3Smrg   return (1ULL << 47) - 1;  // 0x00007fffffffffffUL;
1127490215a3Smrg # endif
1128490215a3Smrg #else  // SANITIZER_WORDSIZE == 32
1129490215a3Smrg # if defined(__s390__)
1130490215a3Smrg   return (1ULL << 31) - 1;  // 0x7fffffff;
1131490215a3Smrg # else
1132490215a3Smrg   return (1ULL << 32) - 1;  // 0xffffffff;
1133490215a3Smrg # endif
1134490215a3Smrg #endif  // SANITIZER_WORDSIZE
1135490215a3Smrg }
1136490215a3Smrg 
GetMaxUserVirtualAddress()1137490215a3Smrg uptr GetMaxUserVirtualAddress() {
1138490215a3Smrg   uptr addr = GetMaxVirtualAddress();
1139490215a3Smrg #if SANITIZER_WORDSIZE == 32 && !defined(__s390__)
1140490215a3Smrg   if (!common_flags()->full_address_space)
1141490215a3Smrg     addr -= GetKernelAreaSize();
1142490215a3Smrg   CHECK_LT(reinterpret_cast<uptr>(&addr), addr);
1143490215a3Smrg #endif
1144490215a3Smrg   return addr;
1145490215a3Smrg }
1146490215a3Smrg 
GetPageSize()1147490215a3Smrg uptr GetPageSize() {
1148490215a3Smrg // Android post-M sysconf(_SC_PAGESIZE) crashes if called from .preinit_array.
1149490215a3Smrg #if SANITIZER_ANDROID
1150490215a3Smrg   return 4096;
1151490215a3Smrg #elif SANITIZER_FREEBSD || SANITIZER_NETBSD
1152490215a3Smrg // Use sysctl as sysconf can trigger interceptors internally.
1153490215a3Smrg   int pz = 0;
1154490215a3Smrg   uptr pzl = sizeof(pz);
1155490215a3Smrg   int mib[2] = {CTL_HW, HW_PAGESIZE};
1156490215a3Smrg   int rv = internal_sysctl(mib, 2, &pz, &pzl, nullptr, 0);
1157490215a3Smrg   CHECK_EQ(rv, 0);
1158490215a3Smrg   return (uptr)pz;
1159490215a3Smrg #elif SANITIZER_LINUX && (defined(__x86_64__) || defined(__i386__))
1160490215a3Smrg   return EXEC_PAGESIZE;
1161490215a3Smrg #elif SANITIZER_USE_GETAUXVAL
1162490215a3Smrg   return getauxval(AT_PAGESZ);
1163490215a3Smrg #else
1164490215a3Smrg   return sysconf(_SC_PAGESIZE);  // EXEC_PAGESIZE may not be trustworthy.
1165490215a3Smrg #endif
1166490215a3Smrg }
1167490215a3Smrg 
1168490215a3Smrg #if !SANITIZER_OPENBSD
ReadBinaryName(char * buf,uptr buf_len)1169490215a3Smrg uptr ReadBinaryName(/*out*/char *buf, uptr buf_len) {
1170490215a3Smrg #if SANITIZER_SOLARIS
1171490215a3Smrg   const char *default_module_name = getexecname();
1172490215a3Smrg   CHECK_NE(default_module_name, NULL);
1173490215a3Smrg   return internal_snprintf(buf, buf_len, "%s", default_module_name);
1174490215a3Smrg #else
1175490215a3Smrg #if SANITIZER_FREEBSD || SANITIZER_NETBSD
1176490215a3Smrg #if SANITIZER_FREEBSD
1177490215a3Smrg   const int Mib[4] = {CTL_KERN, KERN_PROC, KERN_PROC_PATHNAME, -1};
1178490215a3Smrg #else
1179490215a3Smrg   const int Mib[4] = {CTL_KERN, KERN_PROC_ARGS, -1, KERN_PROC_PATHNAME};
1180490215a3Smrg #endif
1181490215a3Smrg   const char *default_module_name = "kern.proc.pathname";
1182490215a3Smrg   uptr Size = buf_len;
1183490215a3Smrg   bool IsErr =
1184490215a3Smrg       (internal_sysctl(Mib, ARRAY_SIZE(Mib), buf, &Size, NULL, 0) != 0);
1185490215a3Smrg   int readlink_error = IsErr ? errno : 0;
1186490215a3Smrg   uptr module_name_len = Size;
1187490215a3Smrg #else
1188490215a3Smrg   const char *default_module_name = "/proc/self/exe";
1189490215a3Smrg   uptr module_name_len = internal_readlink(
1190490215a3Smrg       default_module_name, buf, buf_len);
1191490215a3Smrg   int readlink_error;
1192490215a3Smrg   bool IsErr = internal_iserror(module_name_len, &readlink_error);
1193490215a3Smrg #endif  // SANITIZER_SOLARIS
1194490215a3Smrg   if (IsErr) {
1195490215a3Smrg     // We can't read binary name for some reason, assume it's unknown.
1196490215a3Smrg     Report("WARNING: reading executable name failed with errno %d, "
1197490215a3Smrg            "some stack frames may not be symbolized\n", readlink_error);
1198490215a3Smrg     module_name_len = internal_snprintf(buf, buf_len, "%s",
1199490215a3Smrg                                         default_module_name);
1200490215a3Smrg     CHECK_LT(module_name_len, buf_len);
1201490215a3Smrg   }
1202490215a3Smrg   return module_name_len;
1203490215a3Smrg #endif
1204490215a3Smrg }
1205490215a3Smrg #endif // !SANITIZER_OPENBSD
1206490215a3Smrg 
ReadLongProcessName(char * buf,uptr buf_len)1207490215a3Smrg uptr ReadLongProcessName(/*out*/ char *buf, uptr buf_len) {
1208490215a3Smrg #if SANITIZER_LINUX
1209490215a3Smrg   char *tmpbuf;
1210490215a3Smrg   uptr tmpsize;
1211490215a3Smrg   uptr tmplen;
1212490215a3Smrg   if (ReadFileToBuffer("/proc/self/cmdline", &tmpbuf, &tmpsize, &tmplen,
1213490215a3Smrg                        1024 * 1024)) {
1214490215a3Smrg     internal_strncpy(buf, tmpbuf, buf_len);
1215490215a3Smrg     UnmapOrDie(tmpbuf, tmpsize);
1216490215a3Smrg     return internal_strlen(buf);
1217490215a3Smrg   }
1218490215a3Smrg #endif
1219490215a3Smrg   return ReadBinaryName(buf, buf_len);
1220490215a3Smrg }
1221490215a3Smrg 
1222490215a3Smrg // Match full names of the form /path/to/base_name{-,.}*
LibraryNameIs(const char * full_name,const char * base_name)1223490215a3Smrg bool LibraryNameIs(const char *full_name, const char *base_name) {
1224490215a3Smrg   const char *name = full_name;
1225490215a3Smrg   // Strip path.
1226490215a3Smrg   while (*name != '\0') name++;
1227490215a3Smrg   while (name > full_name && *name != '/') name--;
1228490215a3Smrg   if (*name == '/') name++;
1229490215a3Smrg   uptr base_name_length = internal_strlen(base_name);
1230490215a3Smrg   if (internal_strncmp(name, base_name, base_name_length)) return false;
1231490215a3Smrg   return (name[base_name_length] == '-' || name[base_name_length] == '.');
1232490215a3Smrg }
1233490215a3Smrg 
1234490215a3Smrg #if !SANITIZER_ANDROID
1235490215a3Smrg // Call cb for each region mapped by map.
ForEachMappedRegion(link_map * map,void (* cb)(const void *,uptr))1236490215a3Smrg void ForEachMappedRegion(link_map *map, void (*cb)(const void *, uptr)) {
1237490215a3Smrg   CHECK_NE(map, nullptr);
1238490215a3Smrg #if !SANITIZER_FREEBSD && !SANITIZER_OPENBSD
1239490215a3Smrg   typedef ElfW(Phdr) Elf_Phdr;
1240490215a3Smrg   typedef ElfW(Ehdr) Elf_Ehdr;
1241490215a3Smrg #endif // !SANITIZER_FREEBSD && !SANITIZER_OPENBSD
1242490215a3Smrg   char *base = (char *)map->l_addr;
1243490215a3Smrg   Elf_Ehdr *ehdr = (Elf_Ehdr *)base;
1244490215a3Smrg   char *phdrs = base + ehdr->e_phoff;
1245490215a3Smrg   char *phdrs_end = phdrs + ehdr->e_phnum * ehdr->e_phentsize;
1246490215a3Smrg 
1247490215a3Smrg   // Find the segment with the minimum base so we can "relocate" the p_vaddr
1248490215a3Smrg   // fields.  Typically ET_DYN objects (DSOs) have base of zero and ET_EXEC
1249490215a3Smrg   // objects have a non-zero base.
1250490215a3Smrg   uptr preferred_base = (uptr)-1;
1251490215a3Smrg   for (char *iter = phdrs; iter != phdrs_end; iter += ehdr->e_phentsize) {
1252490215a3Smrg     Elf_Phdr *phdr = (Elf_Phdr *)iter;
1253490215a3Smrg     if (phdr->p_type == PT_LOAD && preferred_base > (uptr)phdr->p_vaddr)
1254490215a3Smrg       preferred_base = (uptr)phdr->p_vaddr;
1255490215a3Smrg   }
1256490215a3Smrg 
1257490215a3Smrg   // Compute the delta from the real base to get a relocation delta.
1258490215a3Smrg   sptr delta = (uptr)base - preferred_base;
1259490215a3Smrg   // Now we can figure out what the loader really mapped.
1260490215a3Smrg   for (char *iter = phdrs; iter != phdrs_end; iter += ehdr->e_phentsize) {
1261490215a3Smrg     Elf_Phdr *phdr = (Elf_Phdr *)iter;
1262490215a3Smrg     if (phdr->p_type == PT_LOAD) {
1263490215a3Smrg       uptr seg_start = phdr->p_vaddr + delta;
1264490215a3Smrg       uptr seg_end = seg_start + phdr->p_memsz;
1265490215a3Smrg       // None of these values are aligned.  We consider the ragged edges of the
1266490215a3Smrg       // load command as defined, since they are mapped from the file.
1267490215a3Smrg       seg_start = RoundDownTo(seg_start, GetPageSizeCached());
1268490215a3Smrg       seg_end = RoundUpTo(seg_end, GetPageSizeCached());
1269490215a3Smrg       cb((void *)seg_start, seg_end - seg_start);
1270490215a3Smrg     }
1271490215a3Smrg   }
1272490215a3Smrg }
1273490215a3Smrg #endif
1274490215a3Smrg 
1275490215a3Smrg #if defined(__x86_64__) && SANITIZER_LINUX
1276490215a3Smrg // We cannot use glibc's clone wrapper, because it messes with the child
1277490215a3Smrg // task's TLS. It writes the PID and TID of the child task to its thread
1278490215a3Smrg // descriptor, but in our case the child task shares the thread descriptor with
1279490215a3Smrg // the parent (because we don't know how to allocate a new thread
1280490215a3Smrg // descriptor to keep glibc happy). So the stock version of clone(), when
1281490215a3Smrg // 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)1282490215a3Smrg uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1283490215a3Smrg                     int *parent_tidptr, void *newtls, int *child_tidptr) {
1284490215a3Smrg   long long res;
1285490215a3Smrg   if (!fn || !child_stack)
1286490215a3Smrg     return -EINVAL;
1287490215a3Smrg   CHECK_EQ(0, (uptr)child_stack % 16);
1288490215a3Smrg   child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
1289490215a3Smrg   ((unsigned long long *)child_stack)[0] = (uptr)fn;
1290490215a3Smrg   ((unsigned long long *)child_stack)[1] = (uptr)arg;
1291490215a3Smrg   register void *r8 __asm__("r8") = newtls;
1292490215a3Smrg   register int *r10 __asm__("r10") = child_tidptr;
1293490215a3Smrg   __asm__ __volatile__(
1294490215a3Smrg                        /* %rax = syscall(%rax = SYSCALL(clone),
1295490215a3Smrg                         *                %rdi = flags,
1296490215a3Smrg                         *                %rsi = child_stack,
1297490215a3Smrg                         *                %rdx = parent_tidptr,
1298490215a3Smrg                         *                %r8  = new_tls,
1299490215a3Smrg                         *                %r10 = child_tidptr)
1300490215a3Smrg                         */
1301490215a3Smrg                        "syscall\n"
1302490215a3Smrg 
1303490215a3Smrg                        /* if (%rax != 0)
1304490215a3Smrg                         *   return;
1305490215a3Smrg                         */
1306490215a3Smrg                        "testq  %%rax,%%rax\n"
1307490215a3Smrg                        "jnz    1f\n"
1308490215a3Smrg 
1309490215a3Smrg                        /* In the child. Terminate unwind chain. */
1310490215a3Smrg                        // XXX: We should also terminate the CFI unwind chain
1311490215a3Smrg                        // here. Unfortunately clang 3.2 doesn't support the
1312490215a3Smrg                        // necessary CFI directives, so we skip that part.
1313490215a3Smrg                        "xorq   %%rbp,%%rbp\n"
1314490215a3Smrg 
1315490215a3Smrg                        /* Call "fn(arg)". */
1316490215a3Smrg                        "popq   %%rax\n"
1317490215a3Smrg                        "popq   %%rdi\n"
1318490215a3Smrg                        "call   *%%rax\n"
1319490215a3Smrg 
1320490215a3Smrg                        /* Call _exit(%rax). */
1321490215a3Smrg                        "movq   %%rax,%%rdi\n"
1322490215a3Smrg                        "movq   %2,%%rax\n"
1323490215a3Smrg                        "syscall\n"
1324490215a3Smrg 
1325490215a3Smrg                        /* Return to parent. */
1326490215a3Smrg                      "1:\n"
1327490215a3Smrg                        : "=a" (res)
1328490215a3Smrg                        : "a"(SYSCALL(clone)), "i"(SYSCALL(exit)),
1329490215a3Smrg                          "S"(child_stack),
1330490215a3Smrg                          "D"(flags),
1331490215a3Smrg                          "d"(parent_tidptr),
1332490215a3Smrg                          "r"(r8),
1333490215a3Smrg                          "r"(r10)
1334490215a3Smrg                        : "memory", "r11", "rcx");
1335490215a3Smrg   return res;
1336490215a3Smrg }
1337490215a3Smrg #elif defined(__mips__) && SANITIZER_LINUX
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)1338490215a3Smrg uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1339490215a3Smrg                     int *parent_tidptr, void *newtls, int *child_tidptr) {
1340490215a3Smrg   long long res;
1341490215a3Smrg   if (!fn || !child_stack)
1342490215a3Smrg     return -EINVAL;
1343490215a3Smrg   CHECK_EQ(0, (uptr)child_stack % 16);
1344490215a3Smrg   child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
1345490215a3Smrg   ((unsigned long long *)child_stack)[0] = (uptr)fn;
1346490215a3Smrg   ((unsigned long long *)child_stack)[1] = (uptr)arg;
1347490215a3Smrg   register void *a3 __asm__("$7") = newtls;
1348490215a3Smrg   register int *a4 __asm__("$8") = child_tidptr;
1349490215a3Smrg   // We don't have proper CFI directives here because it requires alot of code
1350490215a3Smrg   // for very marginal benefits.
1351490215a3Smrg   __asm__ __volatile__(
1352490215a3Smrg                        /* $v0 = syscall($v0 = __NR_clone,
1353490215a3Smrg                         * $a0 = flags,
1354490215a3Smrg                         * $a1 = child_stack,
1355490215a3Smrg                         * $a2 = parent_tidptr,
1356490215a3Smrg                         * $a3 = new_tls,
1357490215a3Smrg                         * $a4 = child_tidptr)
1358490215a3Smrg                         */
1359490215a3Smrg                        ".cprestore 16;\n"
1360490215a3Smrg                        "move $4,%1;\n"
1361490215a3Smrg                        "move $5,%2;\n"
1362490215a3Smrg                        "move $6,%3;\n"
1363490215a3Smrg                        "move $7,%4;\n"
1364490215a3Smrg                        /* Store the fifth argument on stack
1365490215a3Smrg                         * if we are using 32-bit abi.
1366490215a3Smrg                         */
1367490215a3Smrg #if SANITIZER_WORDSIZE == 32
1368490215a3Smrg                        "lw %5,16($29);\n"
1369490215a3Smrg #else
1370490215a3Smrg                        "move $8,%5;\n"
1371490215a3Smrg #endif
1372490215a3Smrg                        "li $2,%6;\n"
1373490215a3Smrg                        "syscall;\n"
1374490215a3Smrg 
1375490215a3Smrg                        /* if ($v0 != 0)
1376490215a3Smrg                         * return;
1377490215a3Smrg                         */
1378490215a3Smrg                        "bnez $2,1f;\n"
1379490215a3Smrg 
1380490215a3Smrg                        /* Call "fn(arg)". */
1381490215a3Smrg #if SANITIZER_WORDSIZE == 32
1382490215a3Smrg #ifdef __BIG_ENDIAN__
1383490215a3Smrg                        "lw $25,4($29);\n"
1384490215a3Smrg                        "lw $4,12($29);\n"
1385490215a3Smrg #else
1386490215a3Smrg                        "lw $25,0($29);\n"
1387490215a3Smrg                        "lw $4,8($29);\n"
1388490215a3Smrg #endif
1389490215a3Smrg #else
1390490215a3Smrg                        "ld $25,0($29);\n"
1391490215a3Smrg                        "ld $4,8($29);\n"
1392490215a3Smrg #endif
1393490215a3Smrg                        "jal $25;\n"
1394490215a3Smrg 
1395490215a3Smrg                        /* Call _exit($v0). */
1396490215a3Smrg                        "move $4,$2;\n"
1397490215a3Smrg                        "li $2,%7;\n"
1398490215a3Smrg                        "syscall;\n"
1399490215a3Smrg 
1400490215a3Smrg                        /* Return to parent. */
1401490215a3Smrg                      "1:\n"
1402490215a3Smrg                        : "=r" (res)
1403490215a3Smrg                        : "r"(flags),
1404490215a3Smrg                          "r"(child_stack),
1405490215a3Smrg                          "r"(parent_tidptr),
1406490215a3Smrg                          "r"(a3),
1407490215a3Smrg                          "r"(a4),
1408490215a3Smrg                          "i"(__NR_clone),
1409490215a3Smrg                          "i"(__NR_exit)
1410490215a3Smrg                        : "memory", "$29" );
1411490215a3Smrg   return res;
1412490215a3Smrg }
1413490215a3Smrg #elif defined(__aarch64__) && SANITIZER_LINUX
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)1414490215a3Smrg uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1415490215a3Smrg                     int *parent_tidptr, void *newtls, int *child_tidptr) {
1416490215a3Smrg   long long res;
1417490215a3Smrg   if (!fn || !child_stack)
1418490215a3Smrg     return -EINVAL;
1419490215a3Smrg   CHECK_EQ(0, (uptr)child_stack % 16);
1420490215a3Smrg   child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
1421490215a3Smrg   ((unsigned long long *)child_stack)[0] = (uptr)fn;
1422490215a3Smrg   ((unsigned long long *)child_stack)[1] = (uptr)arg;
1423490215a3Smrg 
1424490215a3Smrg   register int (*__fn)(void *)  __asm__("x0") = fn;
1425490215a3Smrg   register void *__stack __asm__("x1") = child_stack;
1426490215a3Smrg   register int   __flags __asm__("x2") = flags;
1427490215a3Smrg   register void *__arg   __asm__("x3") = arg;
1428490215a3Smrg   register int  *__ptid  __asm__("x4") = parent_tidptr;
1429490215a3Smrg   register void *__tls   __asm__("x5") = newtls;
1430490215a3Smrg   register int  *__ctid  __asm__("x6") = child_tidptr;
1431490215a3Smrg 
1432490215a3Smrg   __asm__ __volatile__(
1433490215a3Smrg                        "mov x0,x2\n" /* flags  */
1434490215a3Smrg                        "mov x2,x4\n" /* ptid  */
1435490215a3Smrg                        "mov x3,x5\n" /* tls  */
1436490215a3Smrg                        "mov x4,x6\n" /* ctid  */
1437490215a3Smrg                        "mov x8,%9\n" /* clone  */
1438490215a3Smrg 
1439490215a3Smrg                        "svc 0x0\n"
1440490215a3Smrg 
1441490215a3Smrg                        /* if (%r0 != 0)
1442490215a3Smrg                         *   return %r0;
1443490215a3Smrg                         */
1444490215a3Smrg                        "cmp x0, #0\n"
1445490215a3Smrg                        "bne 1f\n"
1446490215a3Smrg 
1447490215a3Smrg                        /* In the child, now. Call "fn(arg)". */
1448490215a3Smrg                        "ldp x1, x0, [sp], #16\n"
1449490215a3Smrg                        "blr x1\n"
1450490215a3Smrg 
1451490215a3Smrg                        /* Call _exit(%r0).  */
1452490215a3Smrg                        "mov x8, %10\n"
1453490215a3Smrg                        "svc 0x0\n"
1454490215a3Smrg                      "1:\n"
1455490215a3Smrg 
1456490215a3Smrg                        : "=r" (res)
1457490215a3Smrg                        : "i"(-EINVAL),
1458490215a3Smrg                          "r"(__fn), "r"(__stack), "r"(__flags), "r"(__arg),
1459490215a3Smrg                          "r"(__ptid), "r"(__tls), "r"(__ctid),
1460490215a3Smrg                          "i"(__NR_clone), "i"(__NR_exit)
1461490215a3Smrg                        : "x30", "memory");
1462490215a3Smrg   return res;
1463490215a3Smrg }
1464490215a3Smrg #elif defined(__powerpc64__) && SANITIZER_LINUX
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)1465490215a3Smrg uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1466490215a3Smrg                    int *parent_tidptr, void *newtls, int *child_tidptr) {
1467490215a3Smrg   long long res;
1468490215a3Smrg // Stack frame structure.
1469490215a3Smrg #if SANITIZER_PPC64V1
1470490215a3Smrg //   Back chain == 0        (SP + 112)
1471490215a3Smrg // Frame (112 bytes):
1472490215a3Smrg //   Parameter save area    (SP + 48), 8 doublewords
1473490215a3Smrg //   TOC save area          (SP + 40)
1474490215a3Smrg //   Link editor doubleword (SP + 32)
1475490215a3Smrg //   Compiler doubleword    (SP + 24)
1476490215a3Smrg //   LR save area           (SP + 16)
1477490215a3Smrg //   CR save area           (SP + 8)
1478490215a3Smrg //   Back chain             (SP + 0)
1479490215a3Smrg # define FRAME_SIZE 112
1480490215a3Smrg # define FRAME_TOC_SAVE_OFFSET 40
1481490215a3Smrg #elif SANITIZER_PPC64V2
1482490215a3Smrg //   Back chain == 0        (SP + 32)
1483490215a3Smrg // Frame (32 bytes):
1484490215a3Smrg //   TOC save area          (SP + 24)
1485490215a3Smrg //   LR save area           (SP + 16)
1486490215a3Smrg //   CR save area           (SP + 8)
1487490215a3Smrg //   Back chain             (SP + 0)
1488490215a3Smrg # define FRAME_SIZE 32
1489490215a3Smrg # define FRAME_TOC_SAVE_OFFSET 24
1490490215a3Smrg #else
1491490215a3Smrg # error "Unsupported PPC64 ABI"
1492490215a3Smrg #endif
1493490215a3Smrg   if (!fn || !child_stack)
1494490215a3Smrg     return -EINVAL;
1495490215a3Smrg   CHECK_EQ(0, (uptr)child_stack % 16);
1496490215a3Smrg 
1497490215a3Smrg   register int (*__fn)(void *) __asm__("r3") = fn;
1498490215a3Smrg   register void *__cstack      __asm__("r4") = child_stack;
1499490215a3Smrg   register int __flags         __asm__("r5") = flags;
1500490215a3Smrg   register void *__arg         __asm__("r6") = arg;
1501490215a3Smrg   register int *__ptidptr      __asm__("r7") = parent_tidptr;
1502490215a3Smrg   register void *__newtls      __asm__("r8") = newtls;
1503490215a3Smrg   register int *__ctidptr      __asm__("r9") = child_tidptr;
1504490215a3Smrg 
1505490215a3Smrg  __asm__ __volatile__(
1506490215a3Smrg            /* fn and arg are saved across the syscall */
1507490215a3Smrg            "mr 28, %5\n\t"
1508490215a3Smrg            "mr 27, %8\n\t"
1509490215a3Smrg 
1510490215a3Smrg            /* syscall
1511490215a3Smrg              r0 == __NR_clone
1512490215a3Smrg              r3 == flags
1513490215a3Smrg              r4 == child_stack
1514490215a3Smrg              r5 == parent_tidptr
1515490215a3Smrg              r6 == newtls
1516490215a3Smrg              r7 == child_tidptr */
1517490215a3Smrg            "mr 3, %7\n\t"
1518490215a3Smrg            "mr 5, %9\n\t"
1519490215a3Smrg            "mr 6, %10\n\t"
1520490215a3Smrg            "mr 7, %11\n\t"
1521490215a3Smrg            "li 0, %3\n\t"
1522490215a3Smrg            "sc\n\t"
1523490215a3Smrg 
1524490215a3Smrg            /* Test if syscall was successful */
1525490215a3Smrg            "cmpdi  cr1, 3, 0\n\t"
1526490215a3Smrg            "crandc cr1*4+eq, cr1*4+eq, cr0*4+so\n\t"
1527490215a3Smrg            "bne-   cr1, 1f\n\t"
1528490215a3Smrg 
1529490215a3Smrg            /* Set up stack frame */
1530490215a3Smrg            "li    29, 0\n\t"
1531490215a3Smrg            "stdu  29, -8(1)\n\t"
1532490215a3Smrg            "stdu  1, -%12(1)\n\t"
1533490215a3Smrg            /* Do the function call */
1534490215a3Smrg            "std   2, %13(1)\n\t"
1535490215a3Smrg #if SANITIZER_PPC64V1
1536490215a3Smrg            "ld    0, 0(28)\n\t"
1537490215a3Smrg            "ld    2, 8(28)\n\t"
1538490215a3Smrg            "mtctr 0\n\t"
1539490215a3Smrg #elif SANITIZER_PPC64V2
1540490215a3Smrg            "mr    12, 28\n\t"
1541490215a3Smrg            "mtctr 12\n\t"
1542490215a3Smrg #else
1543490215a3Smrg # error "Unsupported PPC64 ABI"
1544490215a3Smrg #endif
1545490215a3Smrg            "mr    3, 27\n\t"
1546490215a3Smrg            "bctrl\n\t"
1547490215a3Smrg            "ld    2, %13(1)\n\t"
1548490215a3Smrg 
1549490215a3Smrg            /* Call _exit(r3) */
1550490215a3Smrg            "li 0, %4\n\t"
1551490215a3Smrg            "sc\n\t"
1552490215a3Smrg 
1553490215a3Smrg            /* Return to parent */
1554490215a3Smrg            "1:\n\t"
1555490215a3Smrg            "mr %0, 3\n\t"
1556490215a3Smrg              : "=r" (res)
1557490215a3Smrg              : "0" (-1),
1558490215a3Smrg                "i" (EINVAL),
1559490215a3Smrg                "i" (__NR_clone),
1560490215a3Smrg                "i" (__NR_exit),
1561490215a3Smrg                "r" (__fn),
1562490215a3Smrg                "r" (__cstack),
1563490215a3Smrg                "r" (__flags),
1564490215a3Smrg                "r" (__arg),
1565490215a3Smrg                "r" (__ptidptr),
1566490215a3Smrg                "r" (__newtls),
1567490215a3Smrg                "r" (__ctidptr),
1568490215a3Smrg                "i" (FRAME_SIZE),
1569490215a3Smrg                "i" (FRAME_TOC_SAVE_OFFSET)
1570490215a3Smrg              : "cr0", "cr1", "memory", "ctr", "r0", "r27", "r28", "r29");
1571490215a3Smrg   return res;
1572490215a3Smrg }
1573490215a3Smrg #elif defined(__i386__) && SANITIZER_LINUX
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)1574490215a3Smrg uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1575490215a3Smrg                     int *parent_tidptr, void *newtls, int *child_tidptr) {
1576490215a3Smrg   int res;
1577490215a3Smrg   if (!fn || !child_stack)
1578490215a3Smrg     return -EINVAL;
1579490215a3Smrg   CHECK_EQ(0, (uptr)child_stack % 16);
1580490215a3Smrg   child_stack = (char *)child_stack - 7 * sizeof(unsigned int);
1581490215a3Smrg   ((unsigned int *)child_stack)[0] = (uptr)flags;
1582490215a3Smrg   ((unsigned int *)child_stack)[1] = (uptr)0;
1583490215a3Smrg   ((unsigned int *)child_stack)[2] = (uptr)fn;
1584490215a3Smrg   ((unsigned int *)child_stack)[3] = (uptr)arg;
1585490215a3Smrg   __asm__ __volatile__(
1586490215a3Smrg                        /* %eax = syscall(%eax = SYSCALL(clone),
1587490215a3Smrg                         *                %ebx = flags,
1588490215a3Smrg                         *                %ecx = child_stack,
1589490215a3Smrg                         *                %edx = parent_tidptr,
1590490215a3Smrg                         *                %esi  = new_tls,
1591490215a3Smrg                         *                %edi = child_tidptr)
1592490215a3Smrg                         */
1593490215a3Smrg 
1594490215a3Smrg                         /* Obtain flags */
1595490215a3Smrg                         "movl    (%%ecx), %%ebx\n"
1596490215a3Smrg                         /* Do the system call */
1597490215a3Smrg                         "pushl   %%ebx\n"
1598490215a3Smrg                         "pushl   %%esi\n"
1599490215a3Smrg                         "pushl   %%edi\n"
1600490215a3Smrg                         /* Remember the flag value.  */
1601490215a3Smrg                         "movl    %%ebx, (%%ecx)\n"
1602490215a3Smrg                         "int     $0x80\n"
1603490215a3Smrg                         "popl    %%edi\n"
1604490215a3Smrg                         "popl    %%esi\n"
1605490215a3Smrg                         "popl    %%ebx\n"
1606490215a3Smrg 
1607490215a3Smrg                         /* if (%eax != 0)
1608490215a3Smrg                          *   return;
1609490215a3Smrg                          */
1610490215a3Smrg 
1611490215a3Smrg                         "test    %%eax,%%eax\n"
1612490215a3Smrg                         "jnz    1f\n"
1613490215a3Smrg 
1614490215a3Smrg                         /* terminate the stack frame */
1615490215a3Smrg                         "xorl   %%ebp,%%ebp\n"
1616490215a3Smrg                         /* Call FN. */
1617490215a3Smrg                         "call    *%%ebx\n"
1618490215a3Smrg #ifdef PIC
1619490215a3Smrg                         "call    here\n"
1620490215a3Smrg                         "here:\n"
1621490215a3Smrg                         "popl    %%ebx\n"
1622490215a3Smrg                         "addl    $_GLOBAL_OFFSET_TABLE_+[.-here], %%ebx\n"
1623490215a3Smrg #endif
1624490215a3Smrg                         /* Call exit */
1625490215a3Smrg                         "movl    %%eax, %%ebx\n"
1626490215a3Smrg                         "movl    %2, %%eax\n"
1627490215a3Smrg                         "int     $0x80\n"
1628490215a3Smrg                         "1:\n"
1629490215a3Smrg                        : "=a" (res)
1630490215a3Smrg                        : "a"(SYSCALL(clone)), "i"(SYSCALL(exit)),
1631490215a3Smrg                          "c"(child_stack),
1632490215a3Smrg                          "d"(parent_tidptr),
1633490215a3Smrg                          "S"(newtls),
1634490215a3Smrg                          "D"(child_tidptr)
1635490215a3Smrg                        : "memory");
1636490215a3Smrg   return res;
1637490215a3Smrg }
1638490215a3Smrg #elif defined(__arm__) && SANITIZER_LINUX
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)1639490215a3Smrg uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1640490215a3Smrg                     int *parent_tidptr, void *newtls, int *child_tidptr) {
1641490215a3Smrg   unsigned int res;
1642490215a3Smrg   if (!fn || !child_stack)
1643490215a3Smrg     return -EINVAL;
1644490215a3Smrg   child_stack = (char *)child_stack - 2 * sizeof(unsigned int);
1645490215a3Smrg   ((unsigned int *)child_stack)[0] = (uptr)fn;
1646490215a3Smrg   ((unsigned int *)child_stack)[1] = (uptr)arg;
1647490215a3Smrg   register int r0 __asm__("r0") = flags;
1648490215a3Smrg   register void *r1 __asm__("r1") = child_stack;
1649490215a3Smrg   register int *r2 __asm__("r2") = parent_tidptr;
1650490215a3Smrg   register void *r3 __asm__("r3") = newtls;
1651490215a3Smrg   register int *r4 __asm__("r4") = child_tidptr;
1652490215a3Smrg   register int r7 __asm__("r7") = __NR_clone;
1653490215a3Smrg 
1654490215a3Smrg #if __ARM_ARCH > 4 || defined (__ARM_ARCH_4T__)
1655490215a3Smrg # define ARCH_HAS_BX
1656490215a3Smrg #endif
1657490215a3Smrg #if __ARM_ARCH > 4
1658490215a3Smrg # define ARCH_HAS_BLX
1659490215a3Smrg #endif
1660490215a3Smrg 
1661490215a3Smrg #ifdef ARCH_HAS_BX
1662490215a3Smrg # ifdef ARCH_HAS_BLX
1663490215a3Smrg #  define BLX(R) "blx "  #R "\n"
1664490215a3Smrg # else
1665490215a3Smrg #  define BLX(R) "mov lr, pc; bx " #R "\n"
1666490215a3Smrg # endif
1667490215a3Smrg #else
1668490215a3Smrg # define BLX(R)  "mov lr, pc; mov pc," #R "\n"
1669490215a3Smrg #endif
1670490215a3Smrg 
1671490215a3Smrg   __asm__ __volatile__(
1672490215a3Smrg                        /* %r0 = syscall(%r7 = SYSCALL(clone),
1673490215a3Smrg                         *               %r0 = flags,
1674490215a3Smrg                         *               %r1 = child_stack,
1675490215a3Smrg                         *               %r2 = parent_tidptr,
1676490215a3Smrg                         *               %r3  = new_tls,
1677490215a3Smrg                         *               %r4 = child_tidptr)
1678490215a3Smrg                         */
1679490215a3Smrg 
1680490215a3Smrg                        /* Do the system call */
1681490215a3Smrg                        "swi 0x0\n"
1682490215a3Smrg 
1683490215a3Smrg                        /* if (%r0 != 0)
1684490215a3Smrg                         *   return %r0;
1685490215a3Smrg                         */
1686490215a3Smrg                        "cmp r0, #0\n"
1687490215a3Smrg                        "bne 1f\n"
1688490215a3Smrg 
1689490215a3Smrg                        /* In the child, now. Call "fn(arg)". */
1690490215a3Smrg                        "ldr r0, [sp, #4]\n"
1691490215a3Smrg                        "ldr ip, [sp], #8\n"
1692490215a3Smrg                        BLX(ip)
1693490215a3Smrg                        /* Call _exit(%r0). */
1694490215a3Smrg                        "mov r7, %7\n"
1695490215a3Smrg                        "swi 0x0\n"
1696490215a3Smrg                        "1:\n"
1697490215a3Smrg                        "mov %0, r0\n"
1698490215a3Smrg                        : "=r"(res)
1699490215a3Smrg                        : "r"(r0), "r"(r1), "r"(r2), "r"(r3), "r"(r4), "r"(r7),
1700490215a3Smrg                          "i"(__NR_exit)
1701490215a3Smrg                        : "memory");
1702490215a3Smrg   return res;
1703490215a3Smrg }
1704490215a3Smrg #endif  // defined(__x86_64__) && SANITIZER_LINUX
1705490215a3Smrg 
1706490215a3Smrg #if SANITIZER_ANDROID
1707490215a3Smrg #if __ANDROID_API__ < 21
1708490215a3Smrg extern "C" __attribute__((weak)) int dl_iterate_phdr(
1709490215a3Smrg     int (*)(struct dl_phdr_info *, size_t, void *), void *);
1710490215a3Smrg #endif
1711490215a3Smrg 
dl_iterate_phdr_test_cb(struct dl_phdr_info * info,size_t size,void * data)1712490215a3Smrg static int dl_iterate_phdr_test_cb(struct dl_phdr_info *info, size_t size,
1713490215a3Smrg                                    void *data) {
1714490215a3Smrg   // Any name starting with "lib" indicates a bug in L where library base names
1715490215a3Smrg   // are returned instead of paths.
1716490215a3Smrg   if (info->dlpi_name && info->dlpi_name[0] == 'l' &&
1717490215a3Smrg       info->dlpi_name[1] == 'i' && info->dlpi_name[2] == 'b') {
1718490215a3Smrg     *(bool *)data = true;
1719490215a3Smrg     return 1;
1720490215a3Smrg   }
1721490215a3Smrg   return 0;
1722490215a3Smrg }
1723490215a3Smrg 
1724490215a3Smrg static atomic_uint32_t android_api_level;
1725490215a3Smrg 
AndroidDetectApiLevelStatic()1726490215a3Smrg static AndroidApiLevel AndroidDetectApiLevelStatic() {
1727490215a3Smrg #if __ANDROID_API__ <= 19
1728490215a3Smrg   return ANDROID_KITKAT;
1729490215a3Smrg #elif __ANDROID_API__ <= 22
1730490215a3Smrg   return ANDROID_LOLLIPOP_MR1;
1731490215a3Smrg #else
1732490215a3Smrg   return ANDROID_POST_LOLLIPOP;
1733490215a3Smrg #endif
1734490215a3Smrg }
1735490215a3Smrg 
AndroidDetectApiLevel()1736490215a3Smrg static AndroidApiLevel AndroidDetectApiLevel() {
1737490215a3Smrg   if (!&dl_iterate_phdr)
1738490215a3Smrg     return ANDROID_KITKAT; // K or lower
1739490215a3Smrg   bool base_name_seen = false;
1740490215a3Smrg   dl_iterate_phdr(dl_iterate_phdr_test_cb, &base_name_seen);
1741490215a3Smrg   if (base_name_seen)
1742490215a3Smrg     return ANDROID_LOLLIPOP_MR1; // L MR1
1743490215a3Smrg   return ANDROID_POST_LOLLIPOP;   // post-L
1744490215a3Smrg   // Plain L (API level 21) is completely broken wrt ASan and not very
1745490215a3Smrg   // interesting to detect.
1746490215a3Smrg }
1747490215a3Smrg 
1748490215a3Smrg extern "C" __attribute__((weak)) void* _DYNAMIC;
1749490215a3Smrg 
AndroidGetApiLevel()1750490215a3Smrg AndroidApiLevel AndroidGetApiLevel() {
1751490215a3Smrg   AndroidApiLevel level =
1752490215a3Smrg       (AndroidApiLevel)atomic_load(&android_api_level, memory_order_relaxed);
1753490215a3Smrg   if (level) return level;
1754490215a3Smrg   level = &_DYNAMIC == nullptr ? AndroidDetectApiLevelStatic()
1755490215a3Smrg                                : AndroidDetectApiLevel();
1756490215a3Smrg   atomic_store(&android_api_level, level, memory_order_relaxed);
1757490215a3Smrg   return level;
1758490215a3Smrg }
1759490215a3Smrg 
1760490215a3Smrg #endif
1761490215a3Smrg 
GetHandleSignalModeImpl(int signum)1762490215a3Smrg static HandleSignalMode GetHandleSignalModeImpl(int signum) {
1763490215a3Smrg   switch (signum) {
1764490215a3Smrg     case SIGABRT:
1765490215a3Smrg       return common_flags()->handle_abort;
1766490215a3Smrg     case SIGILL:
1767490215a3Smrg       return common_flags()->handle_sigill;
1768490215a3Smrg     case SIGTRAP:
1769490215a3Smrg       return common_flags()->handle_sigtrap;
1770490215a3Smrg     case SIGFPE:
1771490215a3Smrg       return common_flags()->handle_sigfpe;
1772490215a3Smrg     case SIGSEGV:
1773490215a3Smrg       return common_flags()->handle_segv;
1774490215a3Smrg     case SIGBUS:
1775490215a3Smrg       return common_flags()->handle_sigbus;
1776490215a3Smrg   }
1777490215a3Smrg   return kHandleSignalNo;
1778490215a3Smrg }
1779490215a3Smrg 
GetHandleSignalMode(int signum)1780490215a3Smrg HandleSignalMode GetHandleSignalMode(int signum) {
1781490215a3Smrg   HandleSignalMode result = GetHandleSignalModeImpl(signum);
1782490215a3Smrg   if (result == kHandleSignalYes && !common_flags()->allow_user_segv_handler)
1783490215a3Smrg     return kHandleSignalExclusive;
1784490215a3Smrg   return result;
1785490215a3Smrg }
1786490215a3Smrg 
1787490215a3Smrg #if !SANITIZER_GO
internal_start_thread(void (* func)(void * arg),void * arg)1788490215a3Smrg void *internal_start_thread(void(*func)(void *arg), void *arg) {
1789490215a3Smrg   // Start the thread with signals blocked, otherwise it can steal user signals.
1790490215a3Smrg   __sanitizer_sigset_t set, old;
1791490215a3Smrg   internal_sigfillset(&set);
1792490215a3Smrg #if SANITIZER_LINUX && !SANITIZER_ANDROID
1793490215a3Smrg   // Glibc uses SIGSETXID signal during setuid call. If this signal is blocked
1794490215a3Smrg   // on any thread, setuid call hangs (see test/tsan/setuid.c).
1795490215a3Smrg   internal_sigdelset(&set, 33);
1796490215a3Smrg #endif
1797490215a3Smrg   internal_sigprocmask(SIG_SETMASK, &set, &old);
1798490215a3Smrg   void *th;
1799490215a3Smrg   real_pthread_create(&th, nullptr, (void*(*)(void *arg))func, arg);
1800490215a3Smrg   internal_sigprocmask(SIG_SETMASK, &old, nullptr);
1801490215a3Smrg   return th;
1802490215a3Smrg }
1803490215a3Smrg 
internal_join_thread(void * th)1804490215a3Smrg void internal_join_thread(void *th) {
1805490215a3Smrg   real_pthread_join(th, nullptr);
1806490215a3Smrg }
1807490215a3Smrg #else
internal_start_thread(void (* func)(void *),void * arg)1808490215a3Smrg void *internal_start_thread(void (*func)(void *), void *arg) { return 0; }
1809490215a3Smrg 
internal_join_thread(void * th)1810490215a3Smrg void internal_join_thread(void *th) {}
1811490215a3Smrg #endif
1812490215a3Smrg 
1813490215a3Smrg #if defined(__aarch64__)
1814490215a3Smrg #if SANITIZER_LINUX
1815490215a3Smrg // Android headers in the older NDK releases miss this definition.
1816490215a3Smrg struct __sanitizer_esr_context {
1817490215a3Smrg   struct _aarch64_ctx head;
1818490215a3Smrg   uint64_t esr;
1819490215a3Smrg };
1820490215a3Smrg 
Aarch64GetESR(ucontext_t * ucontext,u64 * esr)1821490215a3Smrg static bool Aarch64GetESR(ucontext_t *ucontext, u64 *esr) {
1822490215a3Smrg   static const u32 kEsrMagic = 0x45535201;
1823490215a3Smrg   u8 *aux = ucontext->uc_mcontext.__reserved;
1824490215a3Smrg   while (true) {
1825490215a3Smrg     _aarch64_ctx *ctx = (_aarch64_ctx *)aux;
1826490215a3Smrg     if (ctx->size == 0) break;
1827490215a3Smrg     if (ctx->magic == kEsrMagic) {
1828490215a3Smrg       *esr = ((__sanitizer_esr_context *)ctx)->esr;
1829490215a3Smrg       return true;
1830490215a3Smrg     }
1831490215a3Smrg     aux += ctx->size;
1832490215a3Smrg   }
1833490215a3Smrg   return false;
1834490215a3Smrg }
1835490215a3Smrg #else
Aarch64GetESR(ucontext_t * ucontext,u64 * esr)1836490215a3Smrg static bool Aarch64GetESR(ucontext_t *ucontext, u64 *esr) {
1837490215a3Smrg   return false;
1838490215a3Smrg }
1839490215a3Smrg #endif
1840490215a3Smrg #endif
1841490215a3Smrg 
1842490215a3Smrg #if SANITIZER_OPENBSD
1843490215a3Smrg using Context = sigcontext;
1844490215a3Smrg #else
1845490215a3Smrg using Context = ucontext_t;
1846490215a3Smrg #endif
1847490215a3Smrg 
GetWriteFlag() const1848490215a3Smrg SignalContext::WriteFlag SignalContext::GetWriteFlag() const {
1849490215a3Smrg   Context *ucontext = (Context *)context;
1850490215a3Smrg #if defined(__x86_64__) || defined(__i386__)
1851490215a3Smrg   static const uptr PF_WRITE = 1U << 1;
1852490215a3Smrg #if SANITIZER_FREEBSD
1853490215a3Smrg   uptr err = ucontext->uc_mcontext.mc_err;
1854490215a3Smrg #elif SANITIZER_NETBSD
1855490215a3Smrg   uptr err = ucontext->uc_mcontext.__gregs[_REG_ERR];
1856490215a3Smrg #elif SANITIZER_OPENBSD
1857490215a3Smrg   uptr err = ucontext->sc_err;
1858490215a3Smrg #elif SANITIZER_SOLARIS && defined(__i386__)
1859490215a3Smrg   const int Err = 13;
1860490215a3Smrg   uptr err = ucontext->uc_mcontext.gregs[Err];
1861490215a3Smrg #else
1862490215a3Smrg   uptr err = ucontext->uc_mcontext.gregs[REG_ERR];
1863490215a3Smrg #endif // SANITIZER_FREEBSD
1864490215a3Smrg   return err & PF_WRITE ? WRITE : READ;
1865490215a3Smrg #elif defined(__mips__)
1866490215a3Smrg   uint32_t *exception_source;
1867490215a3Smrg   uint32_t faulty_instruction;
1868490215a3Smrg   uint32_t op_code;
1869490215a3Smrg 
1870490215a3Smrg #if SANITIZER_NETBSD
1871490215a3Smrg   ucontext_t *nucontext = (ucontext_t *)ucontext;
1872490215a3Smrg   exception_source = (uint32_t *)_UC_MACHINE_PC(nucontext);
1873490215a3Smrg #else
1874490215a3Smrg   exception_source = (uint32_t *)ucontext->uc_mcontext.pc;
1875490215a3Smrg #endif
1876490215a3Smrg   faulty_instruction = (uint32_t)(*exception_source);
1877490215a3Smrg 
1878490215a3Smrg   op_code = (faulty_instruction >> 26) & 0x3f;
1879490215a3Smrg 
1880490215a3Smrg   // FIXME: Add support for FPU, microMIPS, DSP, MSA memory instructions.
1881490215a3Smrg   switch (op_code) {
1882490215a3Smrg     case 0x28:  // sb
1883490215a3Smrg     case 0x29:  // sh
1884490215a3Smrg     case 0x2b:  // sw
1885490215a3Smrg     case 0x3f:  // sd
1886490215a3Smrg #if __mips_isa_rev < 6
1887490215a3Smrg     case 0x2c:  // sdl
1888490215a3Smrg     case 0x2d:  // sdr
1889490215a3Smrg     case 0x2a:  // swl
1890490215a3Smrg     case 0x2e:  // swr
1891490215a3Smrg #endif
1892490215a3Smrg       return SignalContext::WRITE;
1893490215a3Smrg 
1894490215a3Smrg     case 0x20:  // lb
1895490215a3Smrg     case 0x24:  // lbu
1896490215a3Smrg     case 0x21:  // lh
1897490215a3Smrg     case 0x25:  // lhu
1898490215a3Smrg     case 0x23:  // lw
1899490215a3Smrg     case 0x27:  // lwu
1900490215a3Smrg     case 0x37:  // ld
1901490215a3Smrg #if __mips_isa_rev < 6
1902490215a3Smrg     case 0x1a:  // ldl
1903490215a3Smrg     case 0x1b:  // ldr
1904490215a3Smrg     case 0x22:  // lwl
1905490215a3Smrg     case 0x26:  // lwr
1906490215a3Smrg #endif
1907490215a3Smrg       return SignalContext::READ;
1908490215a3Smrg #if __mips_isa_rev == 6
1909490215a3Smrg     case 0x3b:  // pcrel
1910490215a3Smrg       op_code = (faulty_instruction >> 19) & 0x3;
1911490215a3Smrg       switch (op_code) {
1912490215a3Smrg         case 0x1:  // lwpc
1913490215a3Smrg         case 0x2:  // lwupc
1914490215a3Smrg           return SignalContext::READ;
1915490215a3Smrg       }
1916490215a3Smrg #endif
1917490215a3Smrg   }
1918490215a3Smrg   return SignalContext::UNKNOWN;
1919490215a3Smrg #elif defined(__arm__) && !SANITIZER_NETBSD
1920490215a3Smrg   static const uptr FSR_WRITE = 1U << 11;
1921490215a3Smrg   uptr fsr = ucontext->uc_mcontext.error_code;
1922490215a3Smrg   return fsr & FSR_WRITE ? WRITE : READ;
1923490215a3Smrg #elif defined(__aarch64__)
1924490215a3Smrg   static const u64 ESR_ELx_WNR = 1U << 6;
1925490215a3Smrg   u64 esr;
1926490215a3Smrg   if (!Aarch64GetESR(ucontext, &esr)) return UNKNOWN;
1927490215a3Smrg   return esr & ESR_ELx_WNR ? WRITE : READ;
1928490215a3Smrg #elif defined(__sparc__)
1929490215a3Smrg   // Decode the instruction to determine the access type.
1930490215a3Smrg   // From OpenSolaris $SRC/uts/sun4/os/trap.c (get_accesstype).
1931490215a3Smrg # if SANITIZER_SOLARIS
1932490215a3Smrg   uptr pc = ucontext->uc_mcontext.gregs[REG_PC];
19331c4120beSchristos # elif SANITIZER_NETBSD
19341c4120beSchristos   uptr pc = ucontext->uc_mcontext.__gregs[_REG_PC];
1935490215a3Smrg # else
1936490215a3Smrg   // Historical BSDism here.
1937490215a3Smrg   struct sigcontext *scontext = (struct sigcontext *)ucontext;
19381c4120beSchristos #  if defined(__arch64__)
1939490215a3Smrg   uptr pc = scontext->sigc_regs.tpc;
1940490215a3Smrg #  else
1941490215a3Smrg   uptr pc = scontext->si_regs.pc;
1942490215a3Smrg #  endif
1943490215a3Smrg # endif
1944490215a3Smrg   u32 instr = *(u32 *)pc;
1945490215a3Smrg   return (instr >> 21) & 1 ? WRITE: READ;
1946490215a3Smrg #else
1947490215a3Smrg   (void)ucontext;
1948490215a3Smrg   return UNKNOWN;  // FIXME: Implement.
1949490215a3Smrg #endif
1950490215a3Smrg }
1951490215a3Smrg 
DumpAllRegisters(void * context)1952490215a3Smrg void SignalContext::DumpAllRegisters(void *context) {
1953490215a3Smrg   // FIXME: Implement this.
1954490215a3Smrg }
1955490215a3Smrg 
GetPcSpBp(void * context,uptr * pc,uptr * sp,uptr * bp)1956490215a3Smrg static void GetPcSpBp(void *context, uptr *pc, uptr *sp, uptr *bp) {
1957490215a3Smrg #if SANITIZER_NETBSD
1958490215a3Smrg   // This covers all NetBSD architectures
1959490215a3Smrg   ucontext_t *ucontext = (ucontext_t *)context;
1960490215a3Smrg   *pc = _UC_MACHINE_PC(ucontext);
1961490215a3Smrg   *bp = _UC_MACHINE_FP(ucontext);
1962490215a3Smrg   *sp = _UC_MACHINE_SP(ucontext);
1963490215a3Smrg #elif defined(__arm__)
1964490215a3Smrg   ucontext_t *ucontext = (ucontext_t*)context;
1965490215a3Smrg   *pc = ucontext->uc_mcontext.arm_pc;
1966490215a3Smrg   *bp = ucontext->uc_mcontext.arm_fp;
1967490215a3Smrg   *sp = ucontext->uc_mcontext.arm_sp;
1968490215a3Smrg #elif defined(__aarch64__)
1969490215a3Smrg   ucontext_t *ucontext = (ucontext_t*)context;
1970490215a3Smrg   *pc = ucontext->uc_mcontext.pc;
1971490215a3Smrg   *bp = ucontext->uc_mcontext.regs[29];
1972490215a3Smrg   *sp = ucontext->uc_mcontext.sp;
1973490215a3Smrg #elif defined(__hppa__)
1974490215a3Smrg   ucontext_t *ucontext = (ucontext_t*)context;
1975490215a3Smrg   *pc = ucontext->uc_mcontext.sc_iaoq[0];
1976490215a3Smrg   /* GCC uses %r3 whenever a frame pointer is needed.  */
1977490215a3Smrg   *bp = ucontext->uc_mcontext.sc_gr[3];
1978490215a3Smrg   *sp = ucontext->uc_mcontext.sc_gr[30];
1979490215a3Smrg #elif defined(__x86_64__)
1980490215a3Smrg # if SANITIZER_FREEBSD
1981490215a3Smrg   ucontext_t *ucontext = (ucontext_t*)context;
1982490215a3Smrg   *pc = ucontext->uc_mcontext.mc_rip;
1983490215a3Smrg   *bp = ucontext->uc_mcontext.mc_rbp;
1984490215a3Smrg   *sp = ucontext->uc_mcontext.mc_rsp;
1985490215a3Smrg #elif SANITIZER_OPENBSD
1986490215a3Smrg   sigcontext *ucontext = (sigcontext *)context;
1987490215a3Smrg   *pc = ucontext->sc_rip;
1988490215a3Smrg   *bp = ucontext->sc_rbp;
1989490215a3Smrg   *sp = ucontext->sc_rsp;
1990490215a3Smrg # else
1991490215a3Smrg   ucontext_t *ucontext = (ucontext_t*)context;
1992490215a3Smrg   *pc = ucontext->uc_mcontext.gregs[REG_RIP];
1993490215a3Smrg   *bp = ucontext->uc_mcontext.gregs[REG_RBP];
1994490215a3Smrg   *sp = ucontext->uc_mcontext.gregs[REG_RSP];
1995490215a3Smrg # endif
1996490215a3Smrg #elif defined(__i386__)
1997490215a3Smrg # if SANITIZER_FREEBSD
1998490215a3Smrg   ucontext_t *ucontext = (ucontext_t*)context;
1999490215a3Smrg   *pc = ucontext->uc_mcontext.mc_eip;
2000490215a3Smrg   *bp = ucontext->uc_mcontext.mc_ebp;
2001490215a3Smrg   *sp = ucontext->uc_mcontext.mc_esp;
2002490215a3Smrg #elif SANITIZER_OPENBSD
2003490215a3Smrg   sigcontext *ucontext = (sigcontext *)context;
2004490215a3Smrg   *pc = ucontext->sc_eip;
2005490215a3Smrg   *bp = ucontext->sc_ebp;
2006490215a3Smrg   *sp = ucontext->sc_esp;
2007490215a3Smrg # else
2008490215a3Smrg   ucontext_t *ucontext = (ucontext_t*)context;
2009490215a3Smrg # if SANITIZER_SOLARIS
2010490215a3Smrg   /* Use the numeric values: the symbolic ones are undefined by llvm
2011490215a3Smrg      include/llvm/Support/Solaris.h.  */
2012490215a3Smrg # ifndef REG_EIP
2013490215a3Smrg #  define REG_EIP 14 // REG_PC
2014490215a3Smrg # endif
2015490215a3Smrg # ifndef REG_EBP
2016490215a3Smrg #  define REG_EBP  6 // REG_FP
2017490215a3Smrg # endif
2018490215a3Smrg # ifndef REG_ESP
2019490215a3Smrg #  define REG_ESP 17 // REG_SP
2020490215a3Smrg # endif
2021490215a3Smrg # endif
2022490215a3Smrg   *pc = ucontext->uc_mcontext.gregs[REG_EIP];
2023490215a3Smrg   *bp = ucontext->uc_mcontext.gregs[REG_EBP];
2024490215a3Smrg   *sp = ucontext->uc_mcontext.gregs[REG_ESP];
2025490215a3Smrg # endif
2026490215a3Smrg #elif defined(__powerpc__) || defined(__powerpc64__)
2027490215a3Smrg   ucontext_t *ucontext = (ucontext_t*)context;
2028490215a3Smrg   *pc = ucontext->uc_mcontext.regs->nip;
2029490215a3Smrg   *sp = ucontext->uc_mcontext.regs->gpr[PT_R1];
2030490215a3Smrg   // The powerpc{,64}-linux ABIs do not specify r31 as the frame
2031490215a3Smrg   // pointer, but GCC always uses r31 when we need a frame pointer.
2032490215a3Smrg   *bp = ucontext->uc_mcontext.regs->gpr[PT_R31];
2033490215a3Smrg #elif defined(__sparc__)
2034490215a3Smrg # if defined(__arch64__) || defined(__sparcv9)
2035490215a3Smrg #  define STACK_BIAS 2047
2036490215a3Smrg # else
2037490215a3Smrg #  define STACK_BIAS 0
2038490215a3Smrg # endif
2039490215a3Smrg # if SANITIZER_SOLARIS
2040490215a3Smrg   ucontext_t *ucontext = (ucontext_t*)context;
2041490215a3Smrg   *pc = ucontext->uc_mcontext.gregs[REG_PC];
2042490215a3Smrg   *sp = ucontext->uc_mcontext.gregs[REG_O6] + STACK_BIAS;
20431c4120beSchristos # elif SANITIZER_NETBSD
2044b3aff75cSchristos   *pc = ucontext->uc_mcontext.__gregs[_REG_PC];
2045b3aff75cSchristos   *sp = ucontext->uc_mcontext.__gregs[_REG_O6] + STACK_BIAS;
2046490215a3Smrg # else
2047490215a3Smrg   // Historical BSDism here.
2048490215a3Smrg   struct sigcontext *scontext = (struct sigcontext *)context;
2049490215a3Smrg #  if defined(__arch64__)
2050490215a3Smrg   *pc = scontext->sigc_regs.tpc;
2051490215a3Smrg   *sp = scontext->sigc_regs.u_regs[14] + STACK_BIAS;
2052490215a3Smrg #  else
2053490215a3Smrg   *pc = scontext->si_regs.pc;
2054490215a3Smrg   *sp = scontext->si_regs.u_regs[14];
2055490215a3Smrg #  endif
2056490215a3Smrg # endif
2057490215a3Smrg   *bp = (uptr) ((uhwptr *) *sp)[14] + STACK_BIAS;
2058490215a3Smrg #elif defined(__mips__)
2059490215a3Smrg   ucontext_t *ucontext = (ucontext_t*)context;
2060490215a3Smrg   *pc = ucontext->uc_mcontext.pc;
2061490215a3Smrg   *bp = ucontext->uc_mcontext.gregs[30];
2062490215a3Smrg   *sp = ucontext->uc_mcontext.gregs[29];
2063490215a3Smrg #elif defined(__s390__)
2064490215a3Smrg   ucontext_t *ucontext = (ucontext_t*)context;
2065490215a3Smrg # if defined(__s390x__)
2066490215a3Smrg   *pc = ucontext->uc_mcontext.psw.addr;
2067490215a3Smrg # else
2068490215a3Smrg   *pc = ucontext->uc_mcontext.psw.addr & 0x7fffffff;
2069490215a3Smrg # endif
2070490215a3Smrg   *bp = ucontext->uc_mcontext.gregs[11];
2071490215a3Smrg   *sp = ucontext->uc_mcontext.gregs[15];
2072490215a3Smrg #else
2073490215a3Smrg # error "Unsupported arch"
2074490215a3Smrg #endif
2075490215a3Smrg }
2076490215a3Smrg 
InitPcSpBp()2077490215a3Smrg void SignalContext::InitPcSpBp() { GetPcSpBp(context, &pc, &sp, &bp); }
2078490215a3Smrg 
MaybeReexec()2079490215a3Smrg void MaybeReexec() {
2080490215a3Smrg   // No need to re-exec on Linux.
2081490215a3Smrg }
2082490215a3Smrg 
CheckASLR()2083490215a3Smrg void CheckASLR() {
2084490215a3Smrg #if SANITIZER_NETBSD
2085490215a3Smrg   int mib[3];
2086490215a3Smrg   int paxflags;
2087490215a3Smrg   uptr len = sizeof(paxflags);
2088490215a3Smrg 
2089490215a3Smrg   mib[0] = CTL_PROC;
2090490215a3Smrg   mib[1] = internal_getpid();
2091490215a3Smrg   mib[2] = PROC_PID_PAXFLAGS;
2092490215a3Smrg 
2093490215a3Smrg   if (UNLIKELY(internal_sysctl(mib, 3, &paxflags, &len, NULL, 0) == -1)) {
2094490215a3Smrg     Printf("sysctl failed\n");
2095490215a3Smrg     Die();
2096490215a3Smrg   }
2097490215a3Smrg 
2098490215a3Smrg   if (UNLIKELY(paxflags & CTL_PROC_PAXFLAGS_ASLR)) {
2099490215a3Smrg     Printf("This sanitizer is not compatible with enabled ASLR\n");
2100490215a3Smrg     Die();
2101490215a3Smrg   }
2102490215a3Smrg #elif SANITIZER_PPC64V2
2103490215a3Smrg   // Disable ASLR for Linux PPC64LE.
2104490215a3Smrg   int old_personality = personality(0xffffffff);
2105490215a3Smrg   if (old_personality != -1 && (old_personality & ADDR_NO_RANDOMIZE) == 0) {
2106490215a3Smrg     VReport(1, "WARNING: Program is being run with address space layout "
2107490215a3Smrg                "randomization (ASLR) enabled which prevents the thread and "
2108490215a3Smrg                "memory sanitizers from working on powerpc64le.\n"
2109490215a3Smrg                "ASLR will be disabled and the program re-executed.\n");
2110490215a3Smrg     CHECK_NE(personality(old_personality | ADDR_NO_RANDOMIZE), -1);
2111490215a3Smrg     ReExec();
2112490215a3Smrg   }
2113490215a3Smrg #else
2114490215a3Smrg   // Do nothing
2115490215a3Smrg #endif
2116490215a3Smrg }
2117490215a3Smrg 
PrintModuleMap()2118490215a3Smrg void PrintModuleMap() { }
2119490215a3Smrg 
CheckNoDeepBind(const char * filename,int flag)2120490215a3Smrg void CheckNoDeepBind(const char *filename, int flag) {
2121490215a3Smrg #ifdef RTLD_DEEPBIND
2122490215a3Smrg   if (flag & RTLD_DEEPBIND) {
2123490215a3Smrg     Report(
2124490215a3Smrg         "You are trying to dlopen a %s shared library with RTLD_DEEPBIND flag"
2125490215a3Smrg         " which is incompatibe with sanitizer runtime "
2126490215a3Smrg         "(see https://github.com/google/sanitizers/issues/611 for details"
2127490215a3Smrg         "). If you want to run %s library under sanitizers please remove "
2128490215a3Smrg         "RTLD_DEEPBIND from dlopen flags.\n",
2129490215a3Smrg         filename, filename);
2130490215a3Smrg     Die();
2131490215a3Smrg   }
2132490215a3Smrg #endif
2133490215a3Smrg }
2134490215a3Smrg 
FindAvailableMemoryRange(uptr size,uptr alignment,uptr left_padding,uptr * largest_gap_found,uptr * max_occupied_addr)2135490215a3Smrg uptr FindAvailableMemoryRange(uptr size, uptr alignment, uptr left_padding,
2136490215a3Smrg                               uptr *largest_gap_found,
2137490215a3Smrg                               uptr *max_occupied_addr) {
2138490215a3Smrg   UNREACHABLE("FindAvailableMemoryRange is not available");
2139490215a3Smrg   return 0;
2140490215a3Smrg }
2141490215a3Smrg 
GetRandom(void * buffer,uptr length,bool blocking)2142490215a3Smrg bool GetRandom(void *buffer, uptr length, bool blocking) {
2143490215a3Smrg   if (!buffer || !length || length > 256)
2144490215a3Smrg     return false;
2145490215a3Smrg #if SANITIZER_USE_GETENTROPY
2146490215a3Smrg   uptr rnd = getentropy(buffer, length);
2147490215a3Smrg   int rverrno = 0;
2148490215a3Smrg   if (internal_iserror(rnd, &rverrno) && rverrno == EFAULT)
2149490215a3Smrg     return false;
2150490215a3Smrg   else if (rnd == 0)
2151490215a3Smrg     return true;
2152490215a3Smrg #endif // SANITIZER_USE_GETENTROPY
2153490215a3Smrg 
2154490215a3Smrg #if SANITIZER_USE_GETRANDOM
2155490215a3Smrg   static atomic_uint8_t skip_getrandom_syscall;
2156490215a3Smrg   if (!atomic_load_relaxed(&skip_getrandom_syscall)) {
2157490215a3Smrg     // Up to 256 bytes, getrandom will not be interrupted.
2158490215a3Smrg     uptr res = internal_syscall(SYSCALL(getrandom), buffer, length,
2159490215a3Smrg                                 blocking ? 0 : GRND_NONBLOCK);
2160490215a3Smrg     int rverrno = 0;
2161490215a3Smrg     if (internal_iserror(res, &rverrno) && rverrno == ENOSYS)
2162490215a3Smrg       atomic_store_relaxed(&skip_getrandom_syscall, 1);
2163490215a3Smrg     else if (res == length)
2164490215a3Smrg       return true;
2165490215a3Smrg   }
2166490215a3Smrg #endif // SANITIZER_USE_GETRANDOM
2167490215a3Smrg   // Up to 256 bytes, a read off /dev/urandom will not be interrupted.
2168490215a3Smrg   // blocking is moot here, O_NONBLOCK has no effect when opening /dev/urandom.
2169490215a3Smrg   uptr fd = internal_open("/dev/urandom", O_RDONLY);
2170490215a3Smrg   if (internal_iserror(fd))
2171490215a3Smrg     return false;
2172490215a3Smrg   uptr res = internal_read(fd, buffer, length);
2173490215a3Smrg   if (internal_iserror(res))
2174490215a3Smrg     return false;
2175490215a3Smrg   internal_close(fd);
2176490215a3Smrg   return true;
2177490215a3Smrg }
2178490215a3Smrg 
2179490215a3Smrg } // namespace __sanitizer
2180490215a3Smrg 
2181490215a3Smrg #endif
2182