13cab2bb3Spatrick //===-- sanitizer_linux.cpp -----------------------------------------------===//
23cab2bb3Spatrick //
33cab2bb3Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
43cab2bb3Spatrick // See https://llvm.org/LICENSE.txt for license information.
53cab2bb3Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
63cab2bb3Spatrick //
73cab2bb3Spatrick //===----------------------------------------------------------------------===//
83cab2bb3Spatrick //
93cab2bb3Spatrick // This file is shared between AddressSanitizer and ThreadSanitizer
103cab2bb3Spatrick // run-time libraries and implements linux-specific functions from
113cab2bb3Spatrick // sanitizer_libc.h.
123cab2bb3Spatrick //===----------------------------------------------------------------------===//
133cab2bb3Spatrick
143cab2bb3Spatrick #include "sanitizer_platform.h"
153cab2bb3Spatrick
163cab2bb3Spatrick #if SANITIZER_FREEBSD || SANITIZER_LINUX || SANITIZER_NETBSD || \
17*68dd5bb1Srobert SANITIZER_OPENBSD || SANITIZER_SOLARIS
183cab2bb3Spatrick
193cab2bb3Spatrick #include "sanitizer_common.h"
203cab2bb3Spatrick #include "sanitizer_flags.h"
213cab2bb3Spatrick #include "sanitizer_getauxval.h"
223cab2bb3Spatrick #include "sanitizer_internal_defs.h"
233cab2bb3Spatrick #include "sanitizer_libc.h"
243cab2bb3Spatrick #include "sanitizer_linux.h"
253cab2bb3Spatrick #include "sanitizer_mutex.h"
263cab2bb3Spatrick #include "sanitizer_placement_new.h"
273cab2bb3Spatrick #include "sanitizer_procmaps.h"
283cab2bb3Spatrick
291f9cb04fSpatrick #if SANITIZER_LINUX && !SANITIZER_GO
303cab2bb3Spatrick #include <asm/param.h>
313cab2bb3Spatrick #endif
323cab2bb3Spatrick
333cab2bb3Spatrick // For mips64, syscall(__NR_stat) fills the buffer in the 'struct kernel_stat'
343cab2bb3Spatrick // format. Struct kernel_stat is defined as 'struct stat' in asm/stat.h. To
353cab2bb3Spatrick // access stat from asm/stat.h, without conflicting with definition in
363cab2bb3Spatrick // sys/stat.h, we use this trick.
37*68dd5bb1Srobert #if SANITIZER_MIPS64
383cab2bb3Spatrick #include <asm/unistd.h>
393cab2bb3Spatrick #include <sys/types.h>
403cab2bb3Spatrick #define stat kernel_stat
41d89ec533Spatrick #if SANITIZER_GO
42d89ec533Spatrick #undef st_atime
43d89ec533Spatrick #undef st_mtime
44d89ec533Spatrick #undef st_ctime
45d89ec533Spatrick #define st_atime st_atim
46d89ec533Spatrick #define st_mtime st_mtim
47d89ec533Spatrick #define st_ctime st_ctim
48d89ec533Spatrick #endif
493cab2bb3Spatrick #include <asm/stat.h>
503cab2bb3Spatrick #undef stat
513cab2bb3Spatrick #endif
523cab2bb3Spatrick
533cab2bb3Spatrick #include <dlfcn.h>
543cab2bb3Spatrick #include <errno.h>
553cab2bb3Spatrick #include <fcntl.h>
563cab2bb3Spatrick #include <link.h>
573cab2bb3Spatrick #include <pthread.h>
583cab2bb3Spatrick #include <sched.h>
593cab2bb3Spatrick #include <signal.h>
603cab2bb3Spatrick #include <sys/mman.h>
613cab2bb3Spatrick #include <sys/param.h>
623cab2bb3Spatrick #if !SANITIZER_SOLARIS
633cab2bb3Spatrick #include <sys/ptrace.h>
643cab2bb3Spatrick #endif
653cab2bb3Spatrick #include <sys/resource.h>
663cab2bb3Spatrick #include <sys/stat.h>
673cab2bb3Spatrick #include <sys/syscall.h>
683cab2bb3Spatrick #include <sys/time.h>
693cab2bb3Spatrick #include <sys/types.h>
703cab2bb3Spatrick #include <ucontext.h>
713cab2bb3Spatrick #include <unistd.h>
723cab2bb3Spatrick
733cab2bb3Spatrick #if SANITIZER_LINUX
743cab2bb3Spatrick #include <sys/utsname.h>
753cab2bb3Spatrick #endif
763cab2bb3Spatrick
773cab2bb3Spatrick #if SANITIZER_LINUX && !SANITIZER_ANDROID
783cab2bb3Spatrick #include <sys/personality.h>
793cab2bb3Spatrick #endif
803cab2bb3Spatrick
81*68dd5bb1Srobert #if SANITIZER_LINUX && defined(__loongarch__)
82*68dd5bb1Srobert # include <sys/sysmacros.h>
83*68dd5bb1Srobert #endif
84*68dd5bb1Srobert
853cab2bb3Spatrick #if SANITIZER_FREEBSD
863cab2bb3Spatrick #include <sys/exec.h>
87*68dd5bb1Srobert #include <sys/procctl.h>
883cab2bb3Spatrick #include <sys/sysctl.h>
893cab2bb3Spatrick #include <machine/atomic.h>
903cab2bb3Spatrick extern "C" {
913cab2bb3Spatrick // <sys/umtx.h> must be included after <errno.h> and <sys/types.h> on
923cab2bb3Spatrick // FreeBSD 9.2 and 10.0.
933cab2bb3Spatrick #include <sys/umtx.h>
943cab2bb3Spatrick }
953cab2bb3Spatrick #include <sys/thr.h>
963cab2bb3Spatrick #endif // SANITIZER_FREEBSD
973cab2bb3Spatrick
983cab2bb3Spatrick #if SANITIZER_NETBSD
993cab2bb3Spatrick #include <limits.h> // For NAME_MAX
1003cab2bb3Spatrick #include <sys/sysctl.h>
1013cab2bb3Spatrick #include <sys/exec.h>
1023cab2bb3Spatrick extern struct ps_strings *__ps_strings;
1033cab2bb3Spatrick #endif // SANITIZER_NETBSD
1043cab2bb3Spatrick
1053cab2bb3Spatrick #if SANITIZER_SOLARIS
1063cab2bb3Spatrick #include <stdlib.h>
1073cab2bb3Spatrick #include <thread.h>
1083cab2bb3Spatrick #define environ _environ
1093cab2bb3Spatrick #endif
1103cab2bb3Spatrick
1113cab2bb3Spatrick extern char **environ;
1123cab2bb3Spatrick
1133cab2bb3Spatrick #if SANITIZER_LINUX
1143cab2bb3Spatrick // <linux/time.h>
1153cab2bb3Spatrick struct kernel_timeval {
1163cab2bb3Spatrick long tv_sec;
1173cab2bb3Spatrick long tv_usec;
1183cab2bb3Spatrick };
1193cab2bb3Spatrick
1203cab2bb3Spatrick // <linux/futex.h> is broken on some linux distributions.
1213cab2bb3Spatrick const int FUTEX_WAIT = 0;
1223cab2bb3Spatrick const int FUTEX_WAKE = 1;
1233cab2bb3Spatrick const int FUTEX_PRIVATE_FLAG = 128;
1243cab2bb3Spatrick const int FUTEX_WAIT_PRIVATE = FUTEX_WAIT | FUTEX_PRIVATE_FLAG;
1253cab2bb3Spatrick const int FUTEX_WAKE_PRIVATE = FUTEX_WAKE | FUTEX_PRIVATE_FLAG;
1263cab2bb3Spatrick #endif // SANITIZER_LINUX
1273cab2bb3Spatrick
1283cab2bb3Spatrick // Are we using 32-bit or 64-bit Linux syscalls?
1293cab2bb3Spatrick // x32 (which defines __x86_64__) has SANITIZER_WORDSIZE == 32
1303cab2bb3Spatrick // but it still needs to use 64-bit syscalls.
1313cab2bb3Spatrick #if SANITIZER_LINUX && (defined(__x86_64__) || defined(__powerpc64__) || \
132*68dd5bb1Srobert SANITIZER_WORDSIZE == 64 || \
133*68dd5bb1Srobert (defined(__mips__) && _MIPS_SIM == _ABIN32))
1343cab2bb3Spatrick # define SANITIZER_LINUX_USES_64BIT_SYSCALLS 1
1353cab2bb3Spatrick #else
1363cab2bb3Spatrick # define SANITIZER_LINUX_USES_64BIT_SYSCALLS 0
1373cab2bb3Spatrick #endif
1383cab2bb3Spatrick
1393cab2bb3Spatrick // Note : FreeBSD had implemented both
140d89ec533Spatrick // Linux apis, available from
1413cab2bb3Spatrick // future 12.x version most likely
1423cab2bb3Spatrick #if SANITIZER_LINUX && defined(__NR_getrandom)
1433cab2bb3Spatrick # if !defined(GRND_NONBLOCK)
1443cab2bb3Spatrick # define GRND_NONBLOCK 1
1453cab2bb3Spatrick # endif
1463cab2bb3Spatrick # define SANITIZER_USE_GETRANDOM 1
1473cab2bb3Spatrick #else
1483cab2bb3Spatrick # define SANITIZER_USE_GETRANDOM 0
1493cab2bb3Spatrick #endif // SANITIZER_LINUX && defined(__NR_getrandom)
1503cab2bb3Spatrick
1513cab2bb3Spatrick #if SANITIZER_FREEBSD && __FreeBSD_version >= 1200000
1523cab2bb3Spatrick # define SANITIZER_USE_GETENTROPY 1
1533cab2bb3Spatrick #else
1543cab2bb3Spatrick # define SANITIZER_USE_GETENTROPY 0
1553cab2bb3Spatrick #endif
1563cab2bb3Spatrick
1573cab2bb3Spatrick namespace __sanitizer {
1583cab2bb3Spatrick
SetSigProcMask(__sanitizer_sigset_t * set,__sanitizer_sigset_t * old)159*68dd5bb1Srobert void SetSigProcMask(__sanitizer_sigset_t *set, __sanitizer_sigset_t *old) {
160*68dd5bb1Srobert CHECK_EQ(0, internal_sigprocmask(SIG_SETMASK, set, old));
161*68dd5bb1Srobert }
162*68dd5bb1Srobert
ScopedBlockSignals(__sanitizer_sigset_t * copy)163*68dd5bb1Srobert ScopedBlockSignals::ScopedBlockSignals(__sanitizer_sigset_t *copy) {
164*68dd5bb1Srobert __sanitizer_sigset_t set;
165*68dd5bb1Srobert internal_sigfillset(&set);
166*68dd5bb1Srobert # if SANITIZER_LINUX && !SANITIZER_ANDROID
167*68dd5bb1Srobert // Glibc uses SIGSETXID signal during setuid call. If this signal is blocked
168*68dd5bb1Srobert // on any thread, setuid call hangs.
169*68dd5bb1Srobert // See test/sanitizer_common/TestCases/Linux/setuid.c.
170*68dd5bb1Srobert internal_sigdelset(&set, 33);
171*68dd5bb1Srobert # endif
172*68dd5bb1Srobert # if SANITIZER_LINUX
173*68dd5bb1Srobert // Seccomp-BPF-sandboxed processes rely on SIGSYS to handle trapped syscalls.
174*68dd5bb1Srobert // If this signal is blocked, such calls cannot be handled and the process may
175*68dd5bb1Srobert // hang.
176*68dd5bb1Srobert internal_sigdelset(&set, 31);
177*68dd5bb1Srobert # endif
178*68dd5bb1Srobert SetSigProcMask(&set, &saved_);
179*68dd5bb1Srobert if (copy)
180*68dd5bb1Srobert internal_memcpy(copy, &saved_, sizeof(saved_));
181*68dd5bb1Srobert }
182*68dd5bb1Srobert
~ScopedBlockSignals()183*68dd5bb1Srobert ScopedBlockSignals::~ScopedBlockSignals() { SetSigProcMask(&saved_, nullptr); }
184*68dd5bb1Srobert
1853cab2bb3Spatrick # if SANITIZER_LINUX && defined(__x86_64__)
1863cab2bb3Spatrick # include "sanitizer_syscall_linux_x86_64.inc"
187d89ec533Spatrick # elif SANITIZER_LINUX && SANITIZER_RISCV64
188d89ec533Spatrick # include "sanitizer_syscall_linux_riscv64.inc"
1893cab2bb3Spatrick # elif SANITIZER_LINUX && defined(__aarch64__)
1903cab2bb3Spatrick # include "sanitizer_syscall_linux_aarch64.inc"
1913cab2bb3Spatrick # elif SANITIZER_LINUX && defined(__arm__)
1923cab2bb3Spatrick # include "sanitizer_syscall_linux_arm.inc"
193*68dd5bb1Srobert # elif SANITIZER_LINUX && defined(__hexagon__)
194*68dd5bb1Srobert # include "sanitizer_syscall_linux_hexagon.inc"
195*68dd5bb1Srobert # elif SANITIZER_LINUX && SANITIZER_LOONGARCH64
196*68dd5bb1Srobert # include "sanitizer_syscall_linux_loongarch64.inc"
1973cab2bb3Spatrick # else
1983cab2bb3Spatrick # include "sanitizer_syscall_generic.inc"
1993cab2bb3Spatrick # endif
2003cab2bb3Spatrick
2013cab2bb3Spatrick // --------------- sanitizer_libc.h
2023cab2bb3Spatrick #if !SANITIZER_SOLARIS && !SANITIZER_NETBSD
203d89ec533Spatrick #if !SANITIZER_S390
internal_mmap(void * addr,uptr length,int prot,int flags,int fd,u64 offset)2043cab2bb3Spatrick uptr internal_mmap(void *addr, uptr length, int prot, int flags, int fd,
2053cab2bb3Spatrick u64 offset) {
2063cab2bb3Spatrick #if SANITIZER_FREEBSD || SANITIZER_LINUX_USES_64BIT_SYSCALLS
2073cab2bb3Spatrick return internal_syscall(SYSCALL(mmap), (uptr)addr, length, prot, flags, fd,
2083cab2bb3Spatrick offset);
2093cab2bb3Spatrick #else
2103cab2bb3Spatrick // mmap2 specifies file offset in 4096-byte units.
2113cab2bb3Spatrick CHECK(IsAligned(offset, 4096));
2123cab2bb3Spatrick return internal_syscall(SYSCALL(mmap2), addr, length, prot, flags, fd,
2133cab2bb3Spatrick offset / 4096);
2143cab2bb3Spatrick #endif
2153cab2bb3Spatrick }
216d89ec533Spatrick #endif // !SANITIZER_S390
2173cab2bb3Spatrick
internal_munmap(void * addr,uptr length)2183cab2bb3Spatrick uptr internal_munmap(void *addr, uptr length) {
2193cab2bb3Spatrick return internal_syscall(SYSCALL(munmap), (uptr)addr, length);
2203cab2bb3Spatrick }
2213cab2bb3Spatrick
222d89ec533Spatrick #if SANITIZER_LINUX
internal_mremap(void * old_address,uptr old_size,uptr new_size,int flags,void * new_address)223d89ec533Spatrick uptr internal_mremap(void *old_address, uptr old_size, uptr new_size, int flags,
224d89ec533Spatrick void *new_address) {
225d89ec533Spatrick return internal_syscall(SYSCALL(mremap), (uptr)old_address, old_size,
226d89ec533Spatrick new_size, flags, (uptr)new_address);
227d89ec533Spatrick }
228d89ec533Spatrick #endif
229d89ec533Spatrick
internal_mprotect(void * addr,uptr length,int prot)2303cab2bb3Spatrick int internal_mprotect(void *addr, uptr length, int prot) {
2313cab2bb3Spatrick return internal_syscall(SYSCALL(mprotect), (uptr)addr, length, prot);
2323cab2bb3Spatrick }
233d89ec533Spatrick
internal_madvise(uptr addr,uptr length,int advice)234d89ec533Spatrick int internal_madvise(uptr addr, uptr length, int advice) {
235d89ec533Spatrick return internal_syscall(SYSCALL(madvise), addr, length, advice);
236d89ec533Spatrick }
2373cab2bb3Spatrick
internal_close(fd_t fd)2383cab2bb3Spatrick uptr internal_close(fd_t fd) {
2393cab2bb3Spatrick return internal_syscall(SYSCALL(close), fd);
2403cab2bb3Spatrick }
2413cab2bb3Spatrick
internal_open(const char * filename,int flags)2423cab2bb3Spatrick uptr internal_open(const char *filename, int flags) {
243*68dd5bb1Srobert # if SANITIZER_LINUX
2443cab2bb3Spatrick return internal_syscall(SYSCALL(openat), AT_FDCWD, (uptr)filename, flags);
2453cab2bb3Spatrick #else
2463cab2bb3Spatrick return internal_syscall(SYSCALL(open), (uptr)filename, flags);
2473cab2bb3Spatrick #endif
2483cab2bb3Spatrick }
2493cab2bb3Spatrick
internal_open(const char * filename,int flags,u32 mode)2503cab2bb3Spatrick uptr internal_open(const char *filename, int flags, u32 mode) {
251*68dd5bb1Srobert # if SANITIZER_LINUX
2523cab2bb3Spatrick return internal_syscall(SYSCALL(openat), AT_FDCWD, (uptr)filename, flags,
2533cab2bb3Spatrick mode);
2543cab2bb3Spatrick #else
2553cab2bb3Spatrick return internal_syscall(SYSCALL(open), (uptr)filename, flags, mode);
2563cab2bb3Spatrick #endif
2573cab2bb3Spatrick }
2583cab2bb3Spatrick
internal_read(fd_t fd,void * buf,uptr count)2593cab2bb3Spatrick uptr internal_read(fd_t fd, void *buf, uptr count) {
2603cab2bb3Spatrick sptr res;
2613cab2bb3Spatrick HANDLE_EINTR(res,
2623cab2bb3Spatrick (sptr)internal_syscall(SYSCALL(read), fd, (uptr)buf, count));
2633cab2bb3Spatrick return res;
2643cab2bb3Spatrick }
2653cab2bb3Spatrick
internal_write(fd_t fd,const void * buf,uptr count)2663cab2bb3Spatrick uptr internal_write(fd_t fd, const void *buf, uptr count) {
2673cab2bb3Spatrick sptr res;
2683cab2bb3Spatrick HANDLE_EINTR(res,
2693cab2bb3Spatrick (sptr)internal_syscall(SYSCALL(write), fd, (uptr)buf, count));
2703cab2bb3Spatrick return res;
2713cab2bb3Spatrick }
2723cab2bb3Spatrick
internal_ftruncate(fd_t fd,uptr size)2733cab2bb3Spatrick uptr internal_ftruncate(fd_t fd, uptr size) {
2743cab2bb3Spatrick sptr res;
2753cab2bb3Spatrick HANDLE_EINTR(res, (sptr)internal_syscall(SYSCALL(ftruncate), fd,
2763cab2bb3Spatrick (OFF_T)size));
2773cab2bb3Spatrick return res;
2783cab2bb3Spatrick }
2793cab2bb3Spatrick
280*68dd5bb1Srobert #if (!SANITIZER_LINUX_USES_64BIT_SYSCALLS || SANITIZER_SPARC) && SANITIZER_LINUX
stat64_to_stat(struct stat64 * in,struct stat * out)2813cab2bb3Spatrick static void stat64_to_stat(struct stat64 *in, struct stat *out) {
2823cab2bb3Spatrick internal_memset(out, 0, sizeof(*out));
2833cab2bb3Spatrick out->st_dev = in->st_dev;
2843cab2bb3Spatrick out->st_ino = in->st_ino;
2853cab2bb3Spatrick out->st_mode = in->st_mode;
2863cab2bb3Spatrick out->st_nlink = in->st_nlink;
2873cab2bb3Spatrick out->st_uid = in->st_uid;
2883cab2bb3Spatrick out->st_gid = in->st_gid;
2893cab2bb3Spatrick out->st_rdev = in->st_rdev;
2903cab2bb3Spatrick out->st_size = in->st_size;
2913cab2bb3Spatrick out->st_blksize = in->st_blksize;
2923cab2bb3Spatrick out->st_blocks = in->st_blocks;
2933cab2bb3Spatrick out->st_atime = in->st_atime;
2943cab2bb3Spatrick out->st_mtime = in->st_mtime;
2953cab2bb3Spatrick out->st_ctime = in->st_ctime;
2963cab2bb3Spatrick }
2973cab2bb3Spatrick #endif
2983cab2bb3Spatrick
299*68dd5bb1Srobert #if SANITIZER_LINUX && defined(__loongarch__)
statx_to_stat(struct statx * in,struct stat * out)300*68dd5bb1Srobert static void statx_to_stat(struct statx *in, struct stat *out) {
301*68dd5bb1Srobert internal_memset(out, 0, sizeof(*out));
302*68dd5bb1Srobert out->st_dev = makedev(in->stx_dev_major, in->stx_dev_minor);
303*68dd5bb1Srobert out->st_ino = in->stx_ino;
304*68dd5bb1Srobert out->st_mode = in->stx_mode;
305*68dd5bb1Srobert out->st_nlink = in->stx_nlink;
306*68dd5bb1Srobert out->st_uid = in->stx_uid;
307*68dd5bb1Srobert out->st_gid = in->stx_gid;
308*68dd5bb1Srobert out->st_rdev = makedev(in->stx_rdev_major, in->stx_rdev_minor);
309*68dd5bb1Srobert out->st_size = in->stx_size;
310*68dd5bb1Srobert out->st_blksize = in->stx_blksize;
311*68dd5bb1Srobert out->st_blocks = in->stx_blocks;
312*68dd5bb1Srobert out->st_atime = in->stx_atime.tv_sec;
313*68dd5bb1Srobert out->st_atim.tv_nsec = in->stx_atime.tv_nsec;
314*68dd5bb1Srobert out->st_mtime = in->stx_mtime.tv_sec;
315*68dd5bb1Srobert out->st_mtim.tv_nsec = in->stx_mtime.tv_nsec;
316*68dd5bb1Srobert out->st_ctime = in->stx_ctime.tv_sec;
317*68dd5bb1Srobert out->st_ctim.tv_nsec = in->stx_ctime.tv_nsec;
318*68dd5bb1Srobert }
319*68dd5bb1Srobert #endif
320*68dd5bb1Srobert
321*68dd5bb1Srobert #if SANITIZER_MIPS64
3223cab2bb3Spatrick // Undefine compatibility macros from <sys/stat.h>
3233cab2bb3Spatrick // so that they would not clash with the kernel_stat
3243cab2bb3Spatrick // st_[a|m|c]time fields
325d89ec533Spatrick #if !SANITIZER_GO
3263cab2bb3Spatrick #undef st_atime
3273cab2bb3Spatrick #undef st_mtime
3283cab2bb3Spatrick #undef st_ctime
329d89ec533Spatrick #endif
3303cab2bb3Spatrick #if defined(SANITIZER_ANDROID)
3313cab2bb3Spatrick // Bionic sys/stat.h defines additional macros
3323cab2bb3Spatrick // for compatibility with the old NDKs and
3333cab2bb3Spatrick // they clash with the kernel_stat structure
3343cab2bb3Spatrick // st_[a|m|c]time_nsec fields.
3353cab2bb3Spatrick #undef st_atime_nsec
3363cab2bb3Spatrick #undef st_mtime_nsec
3373cab2bb3Spatrick #undef st_ctime_nsec
3383cab2bb3Spatrick #endif
kernel_stat_to_stat(struct kernel_stat * in,struct stat * out)3393cab2bb3Spatrick static void kernel_stat_to_stat(struct kernel_stat *in, struct stat *out) {
3403cab2bb3Spatrick internal_memset(out, 0, sizeof(*out));
3413cab2bb3Spatrick out->st_dev = in->st_dev;
3423cab2bb3Spatrick out->st_ino = in->st_ino;
3433cab2bb3Spatrick out->st_mode = in->st_mode;
3443cab2bb3Spatrick out->st_nlink = in->st_nlink;
3453cab2bb3Spatrick out->st_uid = in->st_uid;
3463cab2bb3Spatrick out->st_gid = in->st_gid;
3473cab2bb3Spatrick out->st_rdev = in->st_rdev;
3483cab2bb3Spatrick out->st_size = in->st_size;
3493cab2bb3Spatrick out->st_blksize = in->st_blksize;
3503cab2bb3Spatrick out->st_blocks = in->st_blocks;
3513cab2bb3Spatrick #if defined(__USE_MISC) || \
3523cab2bb3Spatrick defined(__USE_XOPEN2K8) || \
3533cab2bb3Spatrick defined(SANITIZER_ANDROID)
3543cab2bb3Spatrick out->st_atim.tv_sec = in->st_atime;
3553cab2bb3Spatrick out->st_atim.tv_nsec = in->st_atime_nsec;
3563cab2bb3Spatrick out->st_mtim.tv_sec = in->st_mtime;
3573cab2bb3Spatrick out->st_mtim.tv_nsec = in->st_mtime_nsec;
3583cab2bb3Spatrick out->st_ctim.tv_sec = in->st_ctime;
3593cab2bb3Spatrick out->st_ctim.tv_nsec = in->st_ctime_nsec;
3603cab2bb3Spatrick #else
3613cab2bb3Spatrick out->st_atime = in->st_atime;
3623cab2bb3Spatrick out->st_atimensec = in->st_atime_nsec;
3633cab2bb3Spatrick out->st_mtime = in->st_mtime;
3643cab2bb3Spatrick out->st_mtimensec = in->st_mtime_nsec;
3653cab2bb3Spatrick out->st_ctime = in->st_ctime;
3663cab2bb3Spatrick out->st_atimensec = in->st_ctime_nsec;
3673cab2bb3Spatrick #endif
3683cab2bb3Spatrick }
3693cab2bb3Spatrick #endif
3703cab2bb3Spatrick
internal_stat(const char * path,void * buf)3713cab2bb3Spatrick uptr internal_stat(const char *path, void *buf) {
372d89ec533Spatrick # if SANITIZER_FREEBSD
3733cab2bb3Spatrick return internal_syscall(SYSCALL(fstatat), AT_FDCWD, (uptr)path, (uptr)buf, 0);
374*68dd5bb1Srobert # elif SANITIZER_LINUX
375*68dd5bb1Srobert # if defined(__loongarch__)
376*68dd5bb1Srobert struct statx bufx;
377*68dd5bb1Srobert int res = internal_syscall(SYSCALL(statx), AT_FDCWD, (uptr)path,
378*68dd5bb1Srobert AT_NO_AUTOMOUNT, STATX_BASIC_STATS, (uptr)&bufx);
379*68dd5bb1Srobert statx_to_stat(&bufx, (struct stat *)buf);
380*68dd5bb1Srobert return res;
381*68dd5bb1Srobert # elif (SANITIZER_WORDSIZE == 64 || SANITIZER_X32 || \
382*68dd5bb1Srobert (defined(__mips__) && _MIPS_SIM == _ABIN32)) && \
383*68dd5bb1Srobert !SANITIZER_SPARC
3843cab2bb3Spatrick return internal_syscall(SYSCALL(newfstatat), AT_FDCWD, (uptr)path, (uptr)buf,
3853cab2bb3Spatrick 0);
3863cab2bb3Spatrick # else
387*68dd5bb1Srobert struct stat64 buf64;
388*68dd5bb1Srobert int res = internal_syscall(SYSCALL(fstatat64), AT_FDCWD, (uptr)path,
389*68dd5bb1Srobert (uptr)&buf64, 0);
390*68dd5bb1Srobert stat64_to_stat(&buf64, (struct stat *)buf);
391*68dd5bb1Srobert return res;
3923cab2bb3Spatrick # endif
3933cab2bb3Spatrick # else
3943cab2bb3Spatrick struct stat64 buf64;
3953cab2bb3Spatrick int res = internal_syscall(SYSCALL(stat64), path, &buf64);
3963cab2bb3Spatrick stat64_to_stat(&buf64, (struct stat *)buf);
3973cab2bb3Spatrick return res;
3983cab2bb3Spatrick # endif
3993cab2bb3Spatrick }
4003cab2bb3Spatrick
internal_lstat(const char * path,void * buf)4013cab2bb3Spatrick uptr internal_lstat(const char *path, void *buf) {
402d89ec533Spatrick # if SANITIZER_FREEBSD
4033cab2bb3Spatrick return internal_syscall(SYSCALL(fstatat), AT_FDCWD, (uptr)path, (uptr)buf,
4043cab2bb3Spatrick AT_SYMLINK_NOFOLLOW);
405*68dd5bb1Srobert # elif SANITIZER_LINUX
406*68dd5bb1Srobert # if defined(__loongarch__)
407*68dd5bb1Srobert struct statx bufx;
408*68dd5bb1Srobert int res = internal_syscall(SYSCALL(statx), AT_FDCWD, (uptr)path,
409*68dd5bb1Srobert AT_SYMLINK_NOFOLLOW | AT_NO_AUTOMOUNT,
410*68dd5bb1Srobert STATX_BASIC_STATS, (uptr)&bufx);
411*68dd5bb1Srobert statx_to_stat(&bufx, (struct stat *)buf);
412*68dd5bb1Srobert return res;
413*68dd5bb1Srobert # elif (defined(_LP64) || SANITIZER_X32 || \
414*68dd5bb1Srobert (defined(__mips__) && _MIPS_SIM == _ABIN32)) && \
415*68dd5bb1Srobert !SANITIZER_SPARC
4163cab2bb3Spatrick return internal_syscall(SYSCALL(newfstatat), AT_FDCWD, (uptr)path, (uptr)buf,
4173cab2bb3Spatrick AT_SYMLINK_NOFOLLOW);
4183cab2bb3Spatrick # else
419*68dd5bb1Srobert struct stat64 buf64;
420*68dd5bb1Srobert int res = internal_syscall(SYSCALL(fstatat64), AT_FDCWD, (uptr)path,
421*68dd5bb1Srobert (uptr)&buf64, AT_SYMLINK_NOFOLLOW);
422*68dd5bb1Srobert stat64_to_stat(&buf64, (struct stat *)buf);
423*68dd5bb1Srobert return res;
4243cab2bb3Spatrick # endif
4253cab2bb3Spatrick # else
4263cab2bb3Spatrick struct stat64 buf64;
4273cab2bb3Spatrick int res = internal_syscall(SYSCALL(lstat64), path, &buf64);
4283cab2bb3Spatrick stat64_to_stat(&buf64, (struct stat *)buf);
4293cab2bb3Spatrick return res;
4303cab2bb3Spatrick # endif
4313cab2bb3Spatrick }
4323cab2bb3Spatrick
internal_fstat(fd_t fd,void * buf)4333cab2bb3Spatrick uptr internal_fstat(fd_t fd, void *buf) {
434d89ec533Spatrick #if SANITIZER_FREEBSD || SANITIZER_LINUX_USES_64BIT_SYSCALLS
435d89ec533Spatrick #if SANITIZER_MIPS64
4363cab2bb3Spatrick // For mips64, fstat syscall fills buffer in the format of kernel_stat
4373cab2bb3Spatrick struct kernel_stat kbuf;
4383cab2bb3Spatrick int res = internal_syscall(SYSCALL(fstat), fd, &kbuf);
4393cab2bb3Spatrick kernel_stat_to_stat(&kbuf, (struct stat *)buf);
4403cab2bb3Spatrick return res;
441*68dd5bb1Srobert # elif SANITIZER_LINUX && defined(__loongarch__)
442*68dd5bb1Srobert struct statx bufx;
443*68dd5bb1Srobert int res = internal_syscall(SYSCALL(statx), fd, "", AT_EMPTY_PATH,
444*68dd5bb1Srobert STATX_BASIC_STATS, (uptr)&bufx);
445*68dd5bb1Srobert statx_to_stat(&bufx, (struct stat *)buf);
446*68dd5bb1Srobert return res;
4473cab2bb3Spatrick # else
4483cab2bb3Spatrick return internal_syscall(SYSCALL(fstat), fd, (uptr)buf);
4493cab2bb3Spatrick # endif
4503cab2bb3Spatrick #else
4513cab2bb3Spatrick struct stat64 buf64;
4523cab2bb3Spatrick int res = internal_syscall(SYSCALL(fstat64), fd, &buf64);
4533cab2bb3Spatrick stat64_to_stat(&buf64, (struct stat *)buf);
4543cab2bb3Spatrick return res;
4553cab2bb3Spatrick #endif
4563cab2bb3Spatrick }
4573cab2bb3Spatrick
internal_filesize(fd_t fd)4583cab2bb3Spatrick uptr internal_filesize(fd_t fd) {
4593cab2bb3Spatrick struct stat st;
4603cab2bb3Spatrick if (internal_fstat(fd, &st))
4613cab2bb3Spatrick return -1;
4623cab2bb3Spatrick return (uptr)st.st_size;
4633cab2bb3Spatrick }
4643cab2bb3Spatrick
internal_dup(int oldfd)4653cab2bb3Spatrick uptr internal_dup(int oldfd) {
4663cab2bb3Spatrick return internal_syscall(SYSCALL(dup), oldfd);
4673cab2bb3Spatrick }
4683cab2bb3Spatrick
internal_dup2(int oldfd,int newfd)4693cab2bb3Spatrick uptr internal_dup2(int oldfd, int newfd) {
470*68dd5bb1Srobert # if SANITIZER_LINUX
4713cab2bb3Spatrick return internal_syscall(SYSCALL(dup3), oldfd, newfd, 0);
4723cab2bb3Spatrick #else
4733cab2bb3Spatrick return internal_syscall(SYSCALL(dup2), oldfd, newfd);
4743cab2bb3Spatrick #endif
4753cab2bb3Spatrick }
4763cab2bb3Spatrick
internal_readlink(const char * path,char * buf,uptr bufsize)4773cab2bb3Spatrick uptr internal_readlink(const char *path, char *buf, uptr bufsize) {
478*68dd5bb1Srobert # if SANITIZER_LINUX
4793cab2bb3Spatrick return internal_syscall(SYSCALL(readlinkat), AT_FDCWD, (uptr)path, (uptr)buf,
4803cab2bb3Spatrick bufsize);
4813cab2bb3Spatrick #else
4823cab2bb3Spatrick return internal_syscall(SYSCALL(readlink), (uptr)path, (uptr)buf, bufsize);
4833cab2bb3Spatrick #endif
4843cab2bb3Spatrick }
4853cab2bb3Spatrick
internal_unlink(const char * path)4863cab2bb3Spatrick uptr internal_unlink(const char *path) {
487*68dd5bb1Srobert # if SANITIZER_LINUX
4883cab2bb3Spatrick return internal_syscall(SYSCALL(unlinkat), AT_FDCWD, (uptr)path, 0);
4893cab2bb3Spatrick #else
4903cab2bb3Spatrick return internal_syscall(SYSCALL(unlink), (uptr)path);
4913cab2bb3Spatrick #endif
4923cab2bb3Spatrick }
4933cab2bb3Spatrick
internal_rename(const char * oldpath,const char * newpath)4943cab2bb3Spatrick uptr internal_rename(const char *oldpath, const char *newpath) {
495*68dd5bb1Srobert # if (defined(__riscv) || defined(__loongarch__)) && defined(__linux__)
4963cab2bb3Spatrick return internal_syscall(SYSCALL(renameat2), AT_FDCWD, (uptr)oldpath, AT_FDCWD,
4973cab2bb3Spatrick (uptr)newpath, 0);
498*68dd5bb1Srobert # elif SANITIZER_LINUX
4993cab2bb3Spatrick return internal_syscall(SYSCALL(renameat), AT_FDCWD, (uptr)oldpath, AT_FDCWD,
5003cab2bb3Spatrick (uptr)newpath);
5013cab2bb3Spatrick # else
5023cab2bb3Spatrick return internal_syscall(SYSCALL(rename), (uptr)oldpath, (uptr)newpath);
5033cab2bb3Spatrick # endif
5043cab2bb3Spatrick }
5053cab2bb3Spatrick
internal_sched_yield()5063cab2bb3Spatrick uptr internal_sched_yield() {
5073cab2bb3Spatrick return internal_syscall(SYSCALL(sched_yield));
5083cab2bb3Spatrick }
5093cab2bb3Spatrick
internal_usleep(u64 useconds)510d89ec533Spatrick void internal_usleep(u64 useconds) {
5113cab2bb3Spatrick struct timespec ts;
512d89ec533Spatrick ts.tv_sec = useconds / 1000000;
513d89ec533Spatrick ts.tv_nsec = (useconds % 1000000) * 1000;
514d89ec533Spatrick internal_syscall(SYSCALL(nanosleep), &ts, &ts);
5153cab2bb3Spatrick }
5163cab2bb3Spatrick
internal_execve(const char * filename,char * const argv[],char * const envp[])5173cab2bb3Spatrick uptr internal_execve(const char *filename, char *const argv[],
5183cab2bb3Spatrick char *const envp[]) {
5193cab2bb3Spatrick return internal_syscall(SYSCALL(execve), (uptr)filename, (uptr)argv,
5203cab2bb3Spatrick (uptr)envp);
5213cab2bb3Spatrick }
5223cab2bb3Spatrick #endif // !SANITIZER_SOLARIS && !SANITIZER_NETBSD
5233cab2bb3Spatrick
524d89ec533Spatrick #if !SANITIZER_NETBSD
internal__exit(int exitcode)525d89ec533Spatrick void internal__exit(int exitcode) {
526d89ec533Spatrick #if SANITIZER_FREEBSD || SANITIZER_SOLARIS
527d89ec533Spatrick internal_syscall(SYSCALL(exit), exitcode);
528d89ec533Spatrick #else
529d89ec533Spatrick internal_syscall(SYSCALL(exit_group), exitcode);
530d89ec533Spatrick #endif
531d89ec533Spatrick Die(); // Unreachable.
532d89ec533Spatrick }
533d89ec533Spatrick #endif // !SANITIZER_NETBSD
534d89ec533Spatrick
5353cab2bb3Spatrick // ----------------- sanitizer_common.h
FileExists(const char * filename)5363cab2bb3Spatrick bool FileExists(const char *filename) {
5373cab2bb3Spatrick if (ShouldMockFailureToOpen(filename))
5383cab2bb3Spatrick return false;
5393cab2bb3Spatrick struct stat st;
5403cab2bb3Spatrick if (internal_stat(filename, &st))
5413cab2bb3Spatrick return false;
5423cab2bb3Spatrick // Sanity check: filename is a regular file.
5433cab2bb3Spatrick return S_ISREG(st.st_mode);
5443cab2bb3Spatrick }
5453cab2bb3Spatrick
DirExists(const char * path)546*68dd5bb1Srobert bool DirExists(const char *path) {
547*68dd5bb1Srobert struct stat st;
548*68dd5bb1Srobert if (internal_stat(path, &st))
549*68dd5bb1Srobert return false;
550*68dd5bb1Srobert return S_ISDIR(st.st_mode);
551*68dd5bb1Srobert }
552*68dd5bb1Srobert
5533cab2bb3Spatrick # if !SANITIZER_NETBSD
GetTid()5543cab2bb3Spatrick tid_t GetTid() {
5553cab2bb3Spatrick #if SANITIZER_FREEBSD
5563cab2bb3Spatrick long Tid;
5573cab2bb3Spatrick thr_self(&Tid);
5583cab2bb3Spatrick return Tid;
5593cab2bb3Spatrick #elif SANITIZER_SOLARIS
5603cab2bb3Spatrick return thr_self();
5613cab2bb3Spatrick #else
5623cab2bb3Spatrick return internal_syscall(SYSCALL(gettid));
5633cab2bb3Spatrick #endif
5643cab2bb3Spatrick }
5653cab2bb3Spatrick
TgKill(pid_t pid,tid_t tid,int sig)5663cab2bb3Spatrick int TgKill(pid_t pid, tid_t tid, int sig) {
5673cab2bb3Spatrick #if SANITIZER_LINUX
5683cab2bb3Spatrick return internal_syscall(SYSCALL(tgkill), pid, tid, sig);
5693cab2bb3Spatrick #elif SANITIZER_FREEBSD
5703cab2bb3Spatrick return internal_syscall(SYSCALL(thr_kill2), pid, tid, sig);
5713cab2bb3Spatrick #elif SANITIZER_SOLARIS
5723cab2bb3Spatrick (void)pid;
5733cab2bb3Spatrick return thr_kill(tid, sig);
5743cab2bb3Spatrick #endif
5753cab2bb3Spatrick }
5763cab2bb3Spatrick #endif
5773cab2bb3Spatrick
578d89ec533Spatrick #if SANITIZER_GLIBC
NanoTime()5793cab2bb3Spatrick u64 NanoTime() {
5803cab2bb3Spatrick kernel_timeval tv;
5813cab2bb3Spatrick internal_memset(&tv, 0, sizeof(tv));
5823cab2bb3Spatrick internal_syscall(SYSCALL(gettimeofday), &tv, 0);
5833cab2bb3Spatrick return (u64)tv.tv_sec * 1000 * 1000 * 1000 + tv.tv_usec * 1000;
5843cab2bb3Spatrick }
585d89ec533Spatrick // Used by real_clock_gettime.
internal_clock_gettime(__sanitizer_clockid_t clk_id,void * tp)5863cab2bb3Spatrick uptr internal_clock_gettime(__sanitizer_clockid_t clk_id, void *tp) {
5873cab2bb3Spatrick return internal_syscall(SYSCALL(clock_gettime), clk_id, tp);
5883cab2bb3Spatrick }
589d89ec533Spatrick #elif !SANITIZER_SOLARIS && !SANITIZER_NETBSD
NanoTime()590d89ec533Spatrick u64 NanoTime() {
591d89ec533Spatrick struct timespec ts;
592d89ec533Spatrick clock_gettime(CLOCK_REALTIME, &ts);
593d89ec533Spatrick return (u64)ts.tv_sec * 1000 * 1000 * 1000 + ts.tv_nsec;
594d89ec533Spatrick }
595d89ec533Spatrick #endif
5963cab2bb3Spatrick
5973cab2bb3Spatrick // Like getenv, but reads env directly from /proc (on Linux) or parses the
5983cab2bb3Spatrick // 'environ' array (on some others) and does not use libc. This function
5993cab2bb3Spatrick // should be called first inside __asan_init.
GetEnv(const char * name)6003cab2bb3Spatrick const char *GetEnv(const char *name) {
601*68dd5bb1Srobert #if SANITIZER_FREEBSD || SANITIZER_NETBSD || SANITIZER_SOLARIS
6023cab2bb3Spatrick if (::environ != 0) {
6033cab2bb3Spatrick uptr NameLen = internal_strlen(name);
6043cab2bb3Spatrick for (char **Env = ::environ; *Env != 0; Env++) {
6053cab2bb3Spatrick if (internal_strncmp(*Env, name, NameLen) == 0 && (*Env)[NameLen] == '=')
6063cab2bb3Spatrick return (*Env) + NameLen + 1;
6073cab2bb3Spatrick }
6083cab2bb3Spatrick }
6093cab2bb3Spatrick return 0; // Not found.
6103cab2bb3Spatrick #elif SANITIZER_LINUX
6113cab2bb3Spatrick static char *environ;
6123cab2bb3Spatrick static uptr len;
6133cab2bb3Spatrick static bool inited;
6143cab2bb3Spatrick if (!inited) {
6153cab2bb3Spatrick inited = true;
6163cab2bb3Spatrick uptr environ_size;
6173cab2bb3Spatrick if (!ReadFileToBuffer("/proc/self/environ", &environ, &environ_size, &len))
6183cab2bb3Spatrick environ = nullptr;
6193cab2bb3Spatrick }
6203cab2bb3Spatrick if (!environ || len == 0) return nullptr;
6213cab2bb3Spatrick uptr namelen = internal_strlen(name);
6223cab2bb3Spatrick const char *p = environ;
6233cab2bb3Spatrick while (*p != '\0') { // will happen at the \0\0 that terminates the buffer
6243cab2bb3Spatrick // proc file has the format NAME=value\0NAME=value\0NAME=value\0...
6253cab2bb3Spatrick const char* endp =
6263cab2bb3Spatrick (char*)internal_memchr(p, '\0', len - (p - environ));
6273cab2bb3Spatrick if (!endp) // this entry isn't NUL terminated
6283cab2bb3Spatrick return nullptr;
6293cab2bb3Spatrick else if (!internal_memcmp(p, name, namelen) && p[namelen] == '=') // Match.
6303cab2bb3Spatrick return p + namelen + 1; // point after =
6313cab2bb3Spatrick p = endp + 1;
6323cab2bb3Spatrick }
6333cab2bb3Spatrick return nullptr; // Not found.
6343cab2bb3Spatrick #else
6353cab2bb3Spatrick #error "Unsupported platform"
6363cab2bb3Spatrick #endif
6373cab2bb3Spatrick }
6383cab2bb3Spatrick
639d89ec533Spatrick #if !SANITIZER_FREEBSD && !SANITIZER_NETBSD && !SANITIZER_GO
6403cab2bb3Spatrick extern "C" {
6413cab2bb3Spatrick SANITIZER_WEAK_ATTRIBUTE extern void *__libc_stack_end;
6423cab2bb3Spatrick }
6433cab2bb3Spatrick #endif
6443cab2bb3Spatrick
645d89ec533Spatrick #if !SANITIZER_FREEBSD && !SANITIZER_NETBSD
ReadNullSepFileToArray(const char * path,char *** arr,int arr_size)6463cab2bb3Spatrick static void ReadNullSepFileToArray(const char *path, char ***arr,
6473cab2bb3Spatrick int arr_size) {
6483cab2bb3Spatrick char *buff;
6493cab2bb3Spatrick uptr buff_size;
6503cab2bb3Spatrick uptr buff_len;
6513cab2bb3Spatrick *arr = (char **)MmapOrDie(arr_size * sizeof(char *), "NullSepFileArray");
6523cab2bb3Spatrick if (!ReadFileToBuffer(path, &buff, &buff_size, &buff_len, 1024 * 1024)) {
6533cab2bb3Spatrick (*arr)[0] = nullptr;
6543cab2bb3Spatrick return;
6553cab2bb3Spatrick }
6563cab2bb3Spatrick (*arr)[0] = buff;
6573cab2bb3Spatrick int count, i;
6583cab2bb3Spatrick for (count = 1, i = 1; ; i++) {
6593cab2bb3Spatrick if (buff[i] == 0) {
6603cab2bb3Spatrick if (buff[i+1] == 0) break;
6613cab2bb3Spatrick (*arr)[count] = &buff[i+1];
6623cab2bb3Spatrick CHECK_LE(count, arr_size - 1); // FIXME: make this more flexible.
6633cab2bb3Spatrick count++;
6643cab2bb3Spatrick }
6653cab2bb3Spatrick }
6663cab2bb3Spatrick (*arr)[count] = nullptr;
6673cab2bb3Spatrick }
6683cab2bb3Spatrick #endif
6693cab2bb3Spatrick
GetArgsAndEnv(char *** argv,char *** envp)6703cab2bb3Spatrick static void GetArgsAndEnv(char ***argv, char ***envp) {
6713cab2bb3Spatrick #if SANITIZER_FREEBSD
6723cab2bb3Spatrick // On FreeBSD, retrieving the argument and environment arrays is done via the
6733cab2bb3Spatrick // kern.ps_strings sysctl, which returns a pointer to a structure containing
6743cab2bb3Spatrick // this information. See also <sys/exec.h>.
6753cab2bb3Spatrick ps_strings *pss;
6763cab2bb3Spatrick uptr sz = sizeof(pss);
6773cab2bb3Spatrick if (internal_sysctlbyname("kern.ps_strings", &pss, &sz, NULL, 0) == -1) {
6783cab2bb3Spatrick Printf("sysctl kern.ps_strings failed\n");
6793cab2bb3Spatrick Die();
6803cab2bb3Spatrick }
6813cab2bb3Spatrick *argv = pss->ps_argvstr;
6823cab2bb3Spatrick *envp = pss->ps_envstr;
6833cab2bb3Spatrick #elif SANITIZER_NETBSD
6843cab2bb3Spatrick *argv = __ps_strings->ps_argvstr;
6853cab2bb3Spatrick *envp = __ps_strings->ps_envstr;
6863cab2bb3Spatrick #else // SANITIZER_FREEBSD
6873cab2bb3Spatrick #if !SANITIZER_GO
6883cab2bb3Spatrick if (&__libc_stack_end) {
6893cab2bb3Spatrick uptr* stack_end = (uptr*)__libc_stack_end;
6901f9cb04fSpatrick // Normally argc can be obtained from *stack_end, however, on ARM glibc's
6911f9cb04fSpatrick // _start clobbers it:
6921f9cb04fSpatrick // https://sourceware.org/git/?p=glibc.git;a=blob;f=sysdeps/arm/start.S;hb=refs/heads/release/2.31/master#l75
6931f9cb04fSpatrick // Do not special-case ARM and infer argc from argv everywhere.
6941f9cb04fSpatrick int argc = 0;
6951f9cb04fSpatrick while (stack_end[argc + 1]) argc++;
6963cab2bb3Spatrick *argv = (char**)(stack_end + 1);
6973cab2bb3Spatrick *envp = (char**)(stack_end + argc + 2);
6983cab2bb3Spatrick } else {
6991f9cb04fSpatrick #endif // !SANITIZER_GO
7003cab2bb3Spatrick static const int kMaxArgv = 2000, kMaxEnvp = 2000;
7013cab2bb3Spatrick ReadNullSepFileToArray("/proc/self/cmdline", argv, kMaxArgv);
7023cab2bb3Spatrick ReadNullSepFileToArray("/proc/self/environ", envp, kMaxEnvp);
7031f9cb04fSpatrick #if !SANITIZER_GO
7043cab2bb3Spatrick }
7053cab2bb3Spatrick #endif // !SANITIZER_GO
7063cab2bb3Spatrick #endif // SANITIZER_FREEBSD
7073cab2bb3Spatrick }
7083cab2bb3Spatrick
GetArgv()7093cab2bb3Spatrick char **GetArgv() {
7103cab2bb3Spatrick char **argv, **envp;
7113cab2bb3Spatrick GetArgsAndEnv(&argv, &envp);
7123cab2bb3Spatrick return argv;
7133cab2bb3Spatrick }
7143cab2bb3Spatrick
GetEnviron()7153cab2bb3Spatrick char **GetEnviron() {
7163cab2bb3Spatrick char **argv, **envp;
7173cab2bb3Spatrick GetArgsAndEnv(&argv, &envp);
7183cab2bb3Spatrick return envp;
7193cab2bb3Spatrick }
7203cab2bb3Spatrick
7213cab2bb3Spatrick #if !SANITIZER_SOLARIS
FutexWait(atomic_uint32_t * p,u32 cmp)722d89ec533Spatrick void FutexWait(atomic_uint32_t *p, u32 cmp) {
723d89ec533Spatrick # if SANITIZER_FREEBSD
724d89ec533Spatrick _umtx_op(p, UMTX_OP_WAIT_UINT, cmp, 0, 0);
725d89ec533Spatrick # elif SANITIZER_NETBSD
726d89ec533Spatrick sched_yield(); /* No userspace futex-like synchronization */
727d89ec533Spatrick # else
728d89ec533Spatrick internal_syscall(SYSCALL(futex), (uptr)p, FUTEX_WAIT_PRIVATE, cmp, 0, 0, 0);
729d89ec533Spatrick # endif
730d89ec533Spatrick }
731d89ec533Spatrick
FutexWake(atomic_uint32_t * p,u32 count)732d89ec533Spatrick void FutexWake(atomic_uint32_t *p, u32 count) {
733d89ec533Spatrick # if SANITIZER_FREEBSD
734d89ec533Spatrick _umtx_op(p, UMTX_OP_WAKE, count, 0, 0);
735d89ec533Spatrick # elif SANITIZER_NETBSD
736d89ec533Spatrick /* No userspace futex-like synchronization */
737d89ec533Spatrick # else
738d89ec533Spatrick internal_syscall(SYSCALL(futex), (uptr)p, FUTEX_WAKE_PRIVATE, count, 0, 0, 0);
739d89ec533Spatrick # endif
740d89ec533Spatrick }
741d89ec533Spatrick
7423cab2bb3Spatrick # endif // !SANITIZER_SOLARIS
7433cab2bb3Spatrick
7443cab2bb3Spatrick // ----------------- sanitizer_linux.h
7453cab2bb3Spatrick // The actual size of this structure is specified by d_reclen.
7463cab2bb3Spatrick // Note that getdents64 uses a different structure format. We only provide the
7473cab2bb3Spatrick // 32-bit syscall here.
7483cab2bb3Spatrick #if SANITIZER_NETBSD
7493cab2bb3Spatrick // Not used
7503cab2bb3Spatrick #else
7513cab2bb3Spatrick struct linux_dirent {
752*68dd5bb1Srobert # if SANITIZER_X32 || SANITIZER_LINUX
7533cab2bb3Spatrick u64 d_ino;
7543cab2bb3Spatrick u64 d_off;
7553cab2bb3Spatrick # else
7563cab2bb3Spatrick unsigned long d_ino;
7573cab2bb3Spatrick unsigned long d_off;
7583cab2bb3Spatrick # endif
7593cab2bb3Spatrick unsigned short d_reclen;
760*68dd5bb1Srobert # if SANITIZER_LINUX
7613cab2bb3Spatrick unsigned char d_type;
7623cab2bb3Spatrick # endif
7633cab2bb3Spatrick char d_name[256];
7643cab2bb3Spatrick };
7653cab2bb3Spatrick #endif
7663cab2bb3Spatrick
7673cab2bb3Spatrick #if !SANITIZER_SOLARIS && !SANITIZER_NETBSD
7683cab2bb3Spatrick // Syscall wrappers.
internal_ptrace(int request,int pid,void * addr,void * data)7693cab2bb3Spatrick uptr internal_ptrace(int request, int pid, void *addr, void *data) {
7703cab2bb3Spatrick return internal_syscall(SYSCALL(ptrace), request, pid, (uptr)addr,
7713cab2bb3Spatrick (uptr)data);
7723cab2bb3Spatrick }
7733cab2bb3Spatrick
internal_waitpid(int pid,int * status,int options)7743cab2bb3Spatrick uptr internal_waitpid(int pid, int *status, int options) {
7753cab2bb3Spatrick return internal_syscall(SYSCALL(wait4), pid, (uptr)status, options,
7763cab2bb3Spatrick 0 /* rusage */);
7773cab2bb3Spatrick }
7783cab2bb3Spatrick
internal_getpid()7793cab2bb3Spatrick uptr internal_getpid() {
7803cab2bb3Spatrick return internal_syscall(SYSCALL(getpid));
7813cab2bb3Spatrick }
7823cab2bb3Spatrick
internal_getppid()7833cab2bb3Spatrick uptr internal_getppid() {
7843cab2bb3Spatrick return internal_syscall(SYSCALL(getppid));
7853cab2bb3Spatrick }
7863cab2bb3Spatrick
internal_dlinfo(void * handle,int request,void * p)7871f9cb04fSpatrick int internal_dlinfo(void *handle, int request, void *p) {
7881f9cb04fSpatrick #if SANITIZER_FREEBSD
7891f9cb04fSpatrick return dlinfo(handle, request, p);
7901f9cb04fSpatrick #else
7911f9cb04fSpatrick UNIMPLEMENTED();
7921f9cb04fSpatrick #endif
7931f9cb04fSpatrick }
7941f9cb04fSpatrick
internal_getdents(fd_t fd,struct linux_dirent * dirp,unsigned int count)7953cab2bb3Spatrick uptr internal_getdents(fd_t fd, struct linux_dirent *dirp, unsigned int count) {
7963cab2bb3Spatrick #if SANITIZER_FREEBSD
7973cab2bb3Spatrick return internal_syscall(SYSCALL(getdirentries), fd, (uptr)dirp, count, NULL);
798*68dd5bb1Srobert # elif SANITIZER_LINUX
7993cab2bb3Spatrick return internal_syscall(SYSCALL(getdents64), fd, (uptr)dirp, count);
8003cab2bb3Spatrick # else
8013cab2bb3Spatrick return internal_syscall(SYSCALL(getdents), fd, (uptr)dirp, count);
8023cab2bb3Spatrick # endif
8033cab2bb3Spatrick }
8043cab2bb3Spatrick
internal_lseek(fd_t fd,OFF_T offset,int whence)8053cab2bb3Spatrick uptr internal_lseek(fd_t fd, OFF_T offset, int whence) {
8063cab2bb3Spatrick return internal_syscall(SYSCALL(lseek), fd, offset, whence);
8073cab2bb3Spatrick }
8083cab2bb3Spatrick
8093cab2bb3Spatrick #if SANITIZER_LINUX
internal_prctl(int option,uptr arg2,uptr arg3,uptr arg4,uptr arg5)8103cab2bb3Spatrick uptr internal_prctl(int option, uptr arg2, uptr arg3, uptr arg4, uptr arg5) {
8113cab2bb3Spatrick return internal_syscall(SYSCALL(prctl), option, arg2, arg3, arg4, arg5);
8123cab2bb3Spatrick }
813*68dd5bb1Srobert # if defined(__x86_64__)
814*68dd5bb1Srobert # include <asm/unistd_64.h>
815*68dd5bb1Srobert // Currently internal_arch_prctl() is only needed on x86_64.
internal_arch_prctl(int option,uptr arg2)816*68dd5bb1Srobert uptr internal_arch_prctl(int option, uptr arg2) {
817*68dd5bb1Srobert return internal_syscall(__NR_arch_prctl, option, arg2);
818*68dd5bb1Srobert }
819*68dd5bb1Srobert # endif
8203cab2bb3Spatrick # endif
8213cab2bb3Spatrick
internal_sigaltstack(const void * ss,void * oss)8223cab2bb3Spatrick uptr internal_sigaltstack(const void *ss, void *oss) {
8233cab2bb3Spatrick return internal_syscall(SYSCALL(sigaltstack), (uptr)ss, (uptr)oss);
8243cab2bb3Spatrick }
8253cab2bb3Spatrick
internal_fork()8263cab2bb3Spatrick int internal_fork() {
827*68dd5bb1Srobert # if SANITIZER_LINUX
828*68dd5bb1Srobert # if SANITIZER_S390
829*68dd5bb1Srobert return internal_syscall(SYSCALL(clone), 0, SIGCHLD);
830*68dd5bb1Srobert # else
8313cab2bb3Spatrick return internal_syscall(SYSCALL(clone), SIGCHLD, 0);
832*68dd5bb1Srobert # endif
8333cab2bb3Spatrick # else
8343cab2bb3Spatrick return internal_syscall(SYSCALL(fork));
8353cab2bb3Spatrick # endif
8363cab2bb3Spatrick }
8373cab2bb3Spatrick
838d89ec533Spatrick #if SANITIZER_FREEBSD
internal_sysctl(const int * name,unsigned int namelen,void * oldp,uptr * oldlenp,const void * newp,uptr newlen)8393cab2bb3Spatrick int internal_sysctl(const int *name, unsigned int namelen, void *oldp,
8403cab2bb3Spatrick uptr *oldlenp, const void *newp, uptr newlen) {
8413cab2bb3Spatrick return internal_syscall(SYSCALL(__sysctl), name, namelen, oldp,
8423cab2bb3Spatrick (size_t *)oldlenp, newp, (size_t)newlen);
8433cab2bb3Spatrick }
8443cab2bb3Spatrick
internal_sysctlbyname(const char * sname,void * oldp,uptr * oldlenp,const void * newp,uptr newlen)8453cab2bb3Spatrick int internal_sysctlbyname(const char *sname, void *oldp, uptr *oldlenp,
8463cab2bb3Spatrick const void *newp, uptr newlen) {
847d89ec533Spatrick // Note: this function can be called during startup, so we need to avoid
848d89ec533Spatrick // calling any interceptable functions. On FreeBSD >= 1300045 sysctlbyname()
849d89ec533Spatrick // is a real syscall, but for older versions it calls sysctlnametomib()
850d89ec533Spatrick // followed by sysctl(). To avoid calling the intercepted version and
851d89ec533Spatrick // asserting if this happens during startup, call the real sysctlnametomib()
852d89ec533Spatrick // followed by internal_sysctl() if the syscall is not available.
853d89ec533Spatrick #ifdef SYS___sysctlbyname
854d89ec533Spatrick return internal_syscall(SYSCALL(__sysctlbyname), sname,
855d89ec533Spatrick internal_strlen(sname), oldp, (size_t *)oldlenp, newp,
856d89ec533Spatrick (size_t)newlen);
857d89ec533Spatrick #else
858d89ec533Spatrick static decltype(sysctlnametomib) *real_sysctlnametomib = nullptr;
859d89ec533Spatrick if (!real_sysctlnametomib)
860d89ec533Spatrick real_sysctlnametomib =
861d89ec533Spatrick (decltype(sysctlnametomib) *)dlsym(RTLD_NEXT, "sysctlnametomib");
862d89ec533Spatrick CHECK(real_sysctlnametomib);
863d89ec533Spatrick
864d89ec533Spatrick int oid[CTL_MAXNAME];
865d89ec533Spatrick size_t len = CTL_MAXNAME;
866d89ec533Spatrick if (real_sysctlnametomib(sname, oid, &len) == -1)
867d89ec533Spatrick return (-1);
868d89ec533Spatrick return internal_sysctl(oid, len, oldp, oldlenp, newp, newlen);
8693cab2bb3Spatrick #endif
870d89ec533Spatrick }
8713cab2bb3Spatrick #endif
8723cab2bb3Spatrick
8733cab2bb3Spatrick #if SANITIZER_LINUX
8743cab2bb3Spatrick #define SA_RESTORER 0x04000000
8753cab2bb3Spatrick // Doesn't set sa_restorer if the caller did not set it, so use with caution
8763cab2bb3Spatrick //(see below).
internal_sigaction_norestorer(int signum,const void * act,void * oldact)8773cab2bb3Spatrick int internal_sigaction_norestorer(int signum, const void *act, void *oldact) {
8783cab2bb3Spatrick __sanitizer_kernel_sigaction_t k_act, k_oldact;
8793cab2bb3Spatrick internal_memset(&k_act, 0, sizeof(__sanitizer_kernel_sigaction_t));
8803cab2bb3Spatrick internal_memset(&k_oldact, 0, sizeof(__sanitizer_kernel_sigaction_t));
8813cab2bb3Spatrick const __sanitizer_sigaction *u_act = (const __sanitizer_sigaction *)act;
8823cab2bb3Spatrick __sanitizer_sigaction *u_oldact = (__sanitizer_sigaction *)oldact;
8833cab2bb3Spatrick if (u_act) {
8843cab2bb3Spatrick k_act.handler = u_act->handler;
8853cab2bb3Spatrick k_act.sigaction = u_act->sigaction;
8863cab2bb3Spatrick internal_memcpy(&k_act.sa_mask, &u_act->sa_mask,
8873cab2bb3Spatrick sizeof(__sanitizer_kernel_sigset_t));
8883cab2bb3Spatrick // Without SA_RESTORER kernel ignores the calls (probably returns EINVAL).
8893cab2bb3Spatrick k_act.sa_flags = u_act->sa_flags | SA_RESTORER;
8903cab2bb3Spatrick // FIXME: most often sa_restorer is unset, however the kernel requires it
8913cab2bb3Spatrick // to point to a valid signal restorer that calls the rt_sigreturn syscall.
8923cab2bb3Spatrick // If sa_restorer passed to the kernel is NULL, the program may crash upon
8933cab2bb3Spatrick // signal delivery or fail to unwind the stack in the signal handler.
8943cab2bb3Spatrick // libc implementation of sigaction() passes its own restorer to
8953cab2bb3Spatrick // rt_sigaction, so we need to do the same (we'll need to reimplement the
8963cab2bb3Spatrick // restorers; for x86_64 the restorer address can be obtained from
8973cab2bb3Spatrick // oldact->sa_restorer upon a call to sigaction(xxx, NULL, oldact).
8983cab2bb3Spatrick #if !SANITIZER_ANDROID || !SANITIZER_MIPS32
8993cab2bb3Spatrick k_act.sa_restorer = u_act->sa_restorer;
9003cab2bb3Spatrick #endif
9013cab2bb3Spatrick }
9023cab2bb3Spatrick
9033cab2bb3Spatrick uptr result = internal_syscall(SYSCALL(rt_sigaction), (uptr)signum,
9043cab2bb3Spatrick (uptr)(u_act ? &k_act : nullptr),
9053cab2bb3Spatrick (uptr)(u_oldact ? &k_oldact : nullptr),
9063cab2bb3Spatrick (uptr)sizeof(__sanitizer_kernel_sigset_t));
9073cab2bb3Spatrick
9083cab2bb3Spatrick if ((result == 0) && u_oldact) {
9093cab2bb3Spatrick u_oldact->handler = k_oldact.handler;
9103cab2bb3Spatrick u_oldact->sigaction = k_oldact.sigaction;
9113cab2bb3Spatrick internal_memcpy(&u_oldact->sa_mask, &k_oldact.sa_mask,
9123cab2bb3Spatrick sizeof(__sanitizer_kernel_sigset_t));
9133cab2bb3Spatrick u_oldact->sa_flags = k_oldact.sa_flags;
9143cab2bb3Spatrick #if !SANITIZER_ANDROID || !SANITIZER_MIPS32
9153cab2bb3Spatrick u_oldact->sa_restorer = k_oldact.sa_restorer;
9163cab2bb3Spatrick #endif
9173cab2bb3Spatrick }
9183cab2bb3Spatrick return result;
9193cab2bb3Spatrick }
9203cab2bb3Spatrick #endif // SANITIZER_LINUX
9213cab2bb3Spatrick
internal_sigprocmask(int how,__sanitizer_sigset_t * set,__sanitizer_sigset_t * oldset)9223cab2bb3Spatrick uptr internal_sigprocmask(int how, __sanitizer_sigset_t *set,
9233cab2bb3Spatrick __sanitizer_sigset_t *oldset) {
924d89ec533Spatrick #if SANITIZER_FREEBSD
9253cab2bb3Spatrick return internal_syscall(SYSCALL(sigprocmask), how, set, oldset);
9263cab2bb3Spatrick #else
9273cab2bb3Spatrick __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
9283cab2bb3Spatrick __sanitizer_kernel_sigset_t *k_oldset = (__sanitizer_kernel_sigset_t *)oldset;
9291f9cb04fSpatrick return internal_syscall(SYSCALL(rt_sigprocmask), (uptr)how, (uptr)k_set,
9301f9cb04fSpatrick (uptr)k_oldset, sizeof(__sanitizer_kernel_sigset_t));
9313cab2bb3Spatrick #endif
9323cab2bb3Spatrick }
9333cab2bb3Spatrick
internal_sigfillset(__sanitizer_sigset_t * set)9343cab2bb3Spatrick void internal_sigfillset(__sanitizer_sigset_t *set) {
9353cab2bb3Spatrick internal_memset(set, 0xff, sizeof(*set));
9363cab2bb3Spatrick }
9373cab2bb3Spatrick
internal_sigemptyset(__sanitizer_sigset_t * set)9383cab2bb3Spatrick void internal_sigemptyset(__sanitizer_sigset_t *set) {
9393cab2bb3Spatrick internal_memset(set, 0, sizeof(*set));
9403cab2bb3Spatrick }
9413cab2bb3Spatrick
9423cab2bb3Spatrick #if SANITIZER_LINUX
internal_sigdelset(__sanitizer_sigset_t * set,int signum)9433cab2bb3Spatrick void internal_sigdelset(__sanitizer_sigset_t *set, int signum) {
9443cab2bb3Spatrick signum -= 1;
9453cab2bb3Spatrick CHECK_GE(signum, 0);
9463cab2bb3Spatrick CHECK_LT(signum, sizeof(*set) * 8);
9473cab2bb3Spatrick __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
9483cab2bb3Spatrick const uptr idx = signum / (sizeof(k_set->sig[0]) * 8);
9493cab2bb3Spatrick const uptr bit = signum % (sizeof(k_set->sig[0]) * 8);
950d89ec533Spatrick k_set->sig[idx] &= ~((uptr)1 << bit);
9513cab2bb3Spatrick }
9523cab2bb3Spatrick
internal_sigismember(__sanitizer_sigset_t * set,int signum)9533cab2bb3Spatrick bool internal_sigismember(__sanitizer_sigset_t *set, int signum) {
9543cab2bb3Spatrick signum -= 1;
9553cab2bb3Spatrick CHECK_GE(signum, 0);
9563cab2bb3Spatrick CHECK_LT(signum, sizeof(*set) * 8);
9573cab2bb3Spatrick __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
9583cab2bb3Spatrick const uptr idx = signum / (sizeof(k_set->sig[0]) * 8);
9593cab2bb3Spatrick const uptr bit = signum % (sizeof(k_set->sig[0]) * 8);
960d89ec533Spatrick return k_set->sig[idx] & ((uptr)1 << bit);
9613cab2bb3Spatrick }
9623cab2bb3Spatrick #elif SANITIZER_FREEBSD
internal_procctl(int type,int id,int cmd,void * data)963*68dd5bb1Srobert uptr internal_procctl(int type, int id, int cmd, void *data) {
964*68dd5bb1Srobert return internal_syscall(SYSCALL(procctl), type, id, cmd, data);
965*68dd5bb1Srobert }
966*68dd5bb1Srobert
internal_sigdelset(__sanitizer_sigset_t * set,int signum)9673cab2bb3Spatrick void internal_sigdelset(__sanitizer_sigset_t *set, int signum) {
9683cab2bb3Spatrick sigset_t *rset = reinterpret_cast<sigset_t *>(set);
9693cab2bb3Spatrick sigdelset(rset, signum);
9703cab2bb3Spatrick }
9713cab2bb3Spatrick
internal_sigismember(__sanitizer_sigset_t * set,int signum)9723cab2bb3Spatrick bool internal_sigismember(__sanitizer_sigset_t *set, int signum) {
9733cab2bb3Spatrick sigset_t *rset = reinterpret_cast<sigset_t *>(set);
9743cab2bb3Spatrick return sigismember(rset, signum);
9753cab2bb3Spatrick }
9763cab2bb3Spatrick #endif
9773cab2bb3Spatrick #endif // !SANITIZER_SOLARIS
9783cab2bb3Spatrick
9793cab2bb3Spatrick #if !SANITIZER_NETBSD
9803cab2bb3Spatrick // ThreadLister implementation.
ThreadLister(pid_t pid)9813cab2bb3Spatrick ThreadLister::ThreadLister(pid_t pid) : pid_(pid), buffer_(4096) {
9823cab2bb3Spatrick char task_directory_path[80];
9833cab2bb3Spatrick internal_snprintf(task_directory_path, sizeof(task_directory_path),
9843cab2bb3Spatrick "/proc/%d/task/", pid);
9853cab2bb3Spatrick descriptor_ = internal_open(task_directory_path, O_RDONLY | O_DIRECTORY);
9863cab2bb3Spatrick if (internal_iserror(descriptor_)) {
9873cab2bb3Spatrick Report("Can't open /proc/%d/task for reading.\n", pid);
9883cab2bb3Spatrick }
9893cab2bb3Spatrick }
9903cab2bb3Spatrick
ListThreads(InternalMmapVector<tid_t> * threads)9913cab2bb3Spatrick ThreadLister::Result ThreadLister::ListThreads(
9923cab2bb3Spatrick InternalMmapVector<tid_t> *threads) {
9933cab2bb3Spatrick if (internal_iserror(descriptor_))
9943cab2bb3Spatrick return Error;
9953cab2bb3Spatrick internal_lseek(descriptor_, 0, SEEK_SET);
9963cab2bb3Spatrick threads->clear();
9973cab2bb3Spatrick
9983cab2bb3Spatrick Result result = Ok;
9993cab2bb3Spatrick for (bool first_read = true;; first_read = false) {
10003cab2bb3Spatrick // Resize to max capacity if it was downsized by IsAlive.
10013cab2bb3Spatrick buffer_.resize(buffer_.capacity());
10023cab2bb3Spatrick CHECK_GE(buffer_.size(), 4096);
10033cab2bb3Spatrick uptr read = internal_getdents(
10043cab2bb3Spatrick descriptor_, (struct linux_dirent *)buffer_.data(), buffer_.size());
10053cab2bb3Spatrick if (!read)
10063cab2bb3Spatrick return result;
10073cab2bb3Spatrick if (internal_iserror(read)) {
10083cab2bb3Spatrick Report("Can't read directory entries from /proc/%d/task.\n", pid_);
10093cab2bb3Spatrick return Error;
10103cab2bb3Spatrick }
10113cab2bb3Spatrick
10123cab2bb3Spatrick for (uptr begin = (uptr)buffer_.data(), end = begin + read; begin < end;) {
10133cab2bb3Spatrick struct linux_dirent *entry = (struct linux_dirent *)begin;
10143cab2bb3Spatrick begin += entry->d_reclen;
10153cab2bb3Spatrick if (entry->d_ino == 1) {
10163cab2bb3Spatrick // Inode 1 is for bad blocks and also can be a reason for early return.
10173cab2bb3Spatrick // Should be emitted if kernel tried to output terminating thread.
10183cab2bb3Spatrick // See proc_task_readdir implementation in Linux.
10193cab2bb3Spatrick result = Incomplete;
10203cab2bb3Spatrick }
10213cab2bb3Spatrick if (entry->d_ino && *entry->d_name >= '0' && *entry->d_name <= '9')
10223cab2bb3Spatrick threads->push_back(internal_atoll(entry->d_name));
10233cab2bb3Spatrick }
10243cab2bb3Spatrick
10253cab2bb3Spatrick // Now we are going to detect short-read or early EOF. In such cases Linux
10263cab2bb3Spatrick // can return inconsistent list with missing alive threads.
10273cab2bb3Spatrick // Code will just remember that the list can be incomplete but it will
10283cab2bb3Spatrick // continue reads to return as much as possible.
10293cab2bb3Spatrick if (!first_read) {
10303cab2bb3Spatrick // The first one was a short-read by definition.
10313cab2bb3Spatrick result = Incomplete;
10323cab2bb3Spatrick } else if (read > buffer_.size() - 1024) {
10333cab2bb3Spatrick // Read was close to the buffer size. So double the size and assume the
10343cab2bb3Spatrick // worst.
10353cab2bb3Spatrick buffer_.resize(buffer_.size() * 2);
10363cab2bb3Spatrick result = Incomplete;
10373cab2bb3Spatrick } else if (!threads->empty() && !IsAlive(threads->back())) {
10383cab2bb3Spatrick // Maybe Linux early returned from read on terminated thread (!pid_alive)
10393cab2bb3Spatrick // and failed to restore read position.
10403cab2bb3Spatrick // See next_tid and proc_task_instantiate in Linux.
10413cab2bb3Spatrick result = Incomplete;
10423cab2bb3Spatrick }
10433cab2bb3Spatrick }
10443cab2bb3Spatrick }
10453cab2bb3Spatrick
IsAlive(int tid)10463cab2bb3Spatrick bool ThreadLister::IsAlive(int tid) {
10473cab2bb3Spatrick // /proc/%d/task/%d/status uses same call to detect alive threads as
10483cab2bb3Spatrick // proc_task_readdir. See task_state implementation in Linux.
10493cab2bb3Spatrick char path[80];
10503cab2bb3Spatrick internal_snprintf(path, sizeof(path), "/proc/%d/task/%d/status", pid_, tid);
10513cab2bb3Spatrick if (!ReadFileToVector(path, &buffer_) || buffer_.empty())
10523cab2bb3Spatrick return false;
10533cab2bb3Spatrick buffer_.push_back(0);
10543cab2bb3Spatrick static const char kPrefix[] = "\nPPid:";
10553cab2bb3Spatrick const char *field = internal_strstr(buffer_.data(), kPrefix);
10563cab2bb3Spatrick if (!field)
10573cab2bb3Spatrick return false;
10583cab2bb3Spatrick field += internal_strlen(kPrefix);
10593cab2bb3Spatrick return (int)internal_atoll(field) != 0;
10603cab2bb3Spatrick }
10613cab2bb3Spatrick
~ThreadLister()10623cab2bb3Spatrick ThreadLister::~ThreadLister() {
10633cab2bb3Spatrick if (!internal_iserror(descriptor_))
10643cab2bb3Spatrick internal_close(descriptor_);
10653cab2bb3Spatrick }
10663cab2bb3Spatrick #endif
10673cab2bb3Spatrick
10683cab2bb3Spatrick #if SANITIZER_WORDSIZE == 32
10693cab2bb3Spatrick // Take care of unusable kernel area in top gigabyte.
GetKernelAreaSize()10703cab2bb3Spatrick static uptr GetKernelAreaSize() {
10713cab2bb3Spatrick #if SANITIZER_LINUX && !SANITIZER_X32
10723cab2bb3Spatrick const uptr gbyte = 1UL << 30;
10733cab2bb3Spatrick
10743cab2bb3Spatrick // Firstly check if there are writable segments
10753cab2bb3Spatrick // mapped to top gigabyte (e.g. stack).
10763cab2bb3Spatrick MemoryMappingLayout proc_maps(/*cache_enabled*/true);
10773cab2bb3Spatrick if (proc_maps.Error())
10783cab2bb3Spatrick return 0;
10793cab2bb3Spatrick MemoryMappedSegment segment;
10803cab2bb3Spatrick while (proc_maps.Next(&segment)) {
10813cab2bb3Spatrick if ((segment.end >= 3 * gbyte) && segment.IsWritable()) return 0;
10823cab2bb3Spatrick }
10833cab2bb3Spatrick
10843cab2bb3Spatrick #if !SANITIZER_ANDROID
10853cab2bb3Spatrick // Even if nothing is mapped, top Gb may still be accessible
10863cab2bb3Spatrick // if we are running on 64-bit kernel.
10873cab2bb3Spatrick // Uname may report misleading results if personality type
10883cab2bb3Spatrick // is modified (e.g. under schroot) so check this as well.
10893cab2bb3Spatrick struct utsname uname_info;
10903cab2bb3Spatrick int pers = personality(0xffffffffUL);
10911f9cb04fSpatrick if (!(pers & PER_MASK) && internal_uname(&uname_info) == 0 &&
10921f9cb04fSpatrick internal_strstr(uname_info.machine, "64"))
10933cab2bb3Spatrick return 0;
10943cab2bb3Spatrick #endif // SANITIZER_ANDROID
10953cab2bb3Spatrick
10963cab2bb3Spatrick // Top gigabyte is reserved for kernel.
10973cab2bb3Spatrick return gbyte;
10983cab2bb3Spatrick #else
10993cab2bb3Spatrick return 0;
11003cab2bb3Spatrick #endif // SANITIZER_LINUX && !SANITIZER_X32
11013cab2bb3Spatrick }
11023cab2bb3Spatrick #endif // SANITIZER_WORDSIZE == 32
11033cab2bb3Spatrick
GetMaxVirtualAddress()11043cab2bb3Spatrick uptr GetMaxVirtualAddress() {
1105d89ec533Spatrick #if SANITIZER_NETBSD && defined(__x86_64__)
11063cab2bb3Spatrick return 0x7f7ffffff000ULL; // (0x00007f8000000000 - PAGE_SIZE)
11073cab2bb3Spatrick #elif SANITIZER_WORDSIZE == 64
1108*68dd5bb1Srobert # if defined(__powerpc64__) || defined(__aarch64__) || defined(__loongarch__)
11093cab2bb3Spatrick // On PowerPC64 we have two different address space layouts: 44- and 46-bit.
11103cab2bb3Spatrick // We somehow need to figure out which one we are using now and choose
11113cab2bb3Spatrick // one of 0x00000fffffffffffUL and 0x00003fffffffffffUL.
11123cab2bb3Spatrick // Note that with 'ulimit -s unlimited' the stack is moved away from the top
11133cab2bb3Spatrick // of the address space, so simply checking the stack address is not enough.
11143cab2bb3Spatrick // This should (does) work for both PowerPC64 Endian modes.
11153cab2bb3Spatrick // Similarly, aarch64 has multiple address space layouts: 39, 42 and 47-bit.
1116*68dd5bb1Srobert // loongarch64 also has multiple address space layouts: default is 47-bit.
11173cab2bb3Spatrick return (1ULL << (MostSignificantSetBitIndex(GET_CURRENT_FRAME()) + 1)) - 1;
1118d89ec533Spatrick #elif SANITIZER_RISCV64
1119d89ec533Spatrick return (1ULL << 38) - 1;
1120*68dd5bb1Srobert # elif SANITIZER_MIPS64
11213cab2bb3Spatrick return (1ULL << 40) - 1; // 0x000000ffffffffffUL;
11223cab2bb3Spatrick # elif defined(__s390x__)
11233cab2bb3Spatrick return (1ULL << 53) - 1; // 0x001fffffffffffffUL;
11243cab2bb3Spatrick #elif defined(__sparc__)
11253cab2bb3Spatrick return ~(uptr)0;
11263cab2bb3Spatrick # else
11273cab2bb3Spatrick return (1ULL << 47) - 1; // 0x00007fffffffffffUL;
11283cab2bb3Spatrick # endif
11293cab2bb3Spatrick #else // SANITIZER_WORDSIZE == 32
11303cab2bb3Spatrick # if defined(__s390__)
11313cab2bb3Spatrick return (1ULL << 31) - 1; // 0x7fffffff;
11323cab2bb3Spatrick # else
11333cab2bb3Spatrick return (1ULL << 32) - 1; // 0xffffffff;
11343cab2bb3Spatrick # endif
11353cab2bb3Spatrick #endif // SANITIZER_WORDSIZE
11363cab2bb3Spatrick }
11373cab2bb3Spatrick
GetMaxUserVirtualAddress()11383cab2bb3Spatrick uptr GetMaxUserVirtualAddress() {
11393cab2bb3Spatrick uptr addr = GetMaxVirtualAddress();
11403cab2bb3Spatrick #if SANITIZER_WORDSIZE == 32 && !defined(__s390__)
11413cab2bb3Spatrick if (!common_flags()->full_address_space)
11423cab2bb3Spatrick addr -= GetKernelAreaSize();
11433cab2bb3Spatrick CHECK_LT(reinterpret_cast<uptr>(&addr), addr);
11443cab2bb3Spatrick #endif
11453cab2bb3Spatrick return addr;
11463cab2bb3Spatrick }
11473cab2bb3Spatrick
11483cab2bb3Spatrick #if !SANITIZER_ANDROID
GetPageSize()11493cab2bb3Spatrick uptr GetPageSize() {
11501f9cb04fSpatrick #if SANITIZER_LINUX && (defined(__x86_64__) || defined(__i386__)) && \
11511f9cb04fSpatrick defined(EXEC_PAGESIZE)
11523cab2bb3Spatrick return EXEC_PAGESIZE;
11533cab2bb3Spatrick #elif SANITIZER_FREEBSD || SANITIZER_NETBSD
11543cab2bb3Spatrick // Use sysctl as sysconf can trigger interceptors internally.
11553cab2bb3Spatrick int pz = 0;
11563cab2bb3Spatrick uptr pzl = sizeof(pz);
11573cab2bb3Spatrick int mib[2] = {CTL_HW, HW_PAGESIZE};
11583cab2bb3Spatrick int rv = internal_sysctl(mib, 2, &pz, &pzl, nullptr, 0);
11593cab2bb3Spatrick CHECK_EQ(rv, 0);
11603cab2bb3Spatrick return (uptr)pz;
11613cab2bb3Spatrick #elif SANITIZER_USE_GETAUXVAL
11623cab2bb3Spatrick return getauxval(AT_PAGESZ);
11633cab2bb3Spatrick #else
11643cab2bb3Spatrick return sysconf(_SC_PAGESIZE); // EXEC_PAGESIZE may not be trustworthy.
11653cab2bb3Spatrick #endif
11663cab2bb3Spatrick }
11673cab2bb3Spatrick #endif // !SANITIZER_ANDROID
11683cab2bb3Spatrick
ReadBinaryName(char * buf,uptr buf_len)11693cab2bb3Spatrick uptr ReadBinaryName(/*out*/char *buf, uptr buf_len) {
11703cab2bb3Spatrick #if SANITIZER_SOLARIS
11713cab2bb3Spatrick const char *default_module_name = getexecname();
11723cab2bb3Spatrick CHECK_NE(default_module_name, NULL);
11733cab2bb3Spatrick return internal_snprintf(buf, buf_len, "%s", default_module_name);
11743cab2bb3Spatrick #else
11753cab2bb3Spatrick #if SANITIZER_FREEBSD || SANITIZER_NETBSD
11763cab2bb3Spatrick #if SANITIZER_FREEBSD
11773cab2bb3Spatrick const int Mib[4] = {CTL_KERN, KERN_PROC, KERN_PROC_PATHNAME, -1};
11783cab2bb3Spatrick #else
11793cab2bb3Spatrick const int Mib[4] = {CTL_KERN, KERN_PROC_ARGS, -1, KERN_PROC_PATHNAME};
11803cab2bb3Spatrick #endif
11813cab2bb3Spatrick const char *default_module_name = "kern.proc.pathname";
11823cab2bb3Spatrick uptr Size = buf_len;
11833cab2bb3Spatrick bool IsErr =
11843cab2bb3Spatrick (internal_sysctl(Mib, ARRAY_SIZE(Mib), buf, &Size, NULL, 0) != 0);
11853cab2bb3Spatrick int readlink_error = IsErr ? errno : 0;
11863cab2bb3Spatrick uptr module_name_len = Size;
11873cab2bb3Spatrick #else
11883cab2bb3Spatrick const char *default_module_name = "/proc/self/exe";
11893cab2bb3Spatrick uptr module_name_len = internal_readlink(
11903cab2bb3Spatrick default_module_name, buf, buf_len);
11913cab2bb3Spatrick int readlink_error;
11923cab2bb3Spatrick bool IsErr = internal_iserror(module_name_len, &readlink_error);
11933cab2bb3Spatrick #endif // SANITIZER_SOLARIS
11943cab2bb3Spatrick if (IsErr) {
11953cab2bb3Spatrick // We can't read binary name for some reason, assume it's unknown.
11963cab2bb3Spatrick Report("WARNING: reading executable name failed with errno %d, "
11973cab2bb3Spatrick "some stack frames may not be symbolized\n", readlink_error);
11983cab2bb3Spatrick module_name_len = internal_snprintf(buf, buf_len, "%s",
11993cab2bb3Spatrick default_module_name);
12003cab2bb3Spatrick CHECK_LT(module_name_len, buf_len);
12013cab2bb3Spatrick }
12023cab2bb3Spatrick return module_name_len;
12033cab2bb3Spatrick #endif
12043cab2bb3Spatrick }
12053cab2bb3Spatrick
ReadLongProcessName(char * buf,uptr buf_len)12063cab2bb3Spatrick uptr ReadLongProcessName(/*out*/ char *buf, uptr buf_len) {
12073cab2bb3Spatrick #if SANITIZER_LINUX
12083cab2bb3Spatrick char *tmpbuf;
12093cab2bb3Spatrick uptr tmpsize;
12103cab2bb3Spatrick uptr tmplen;
12113cab2bb3Spatrick if (ReadFileToBuffer("/proc/self/cmdline", &tmpbuf, &tmpsize, &tmplen,
12123cab2bb3Spatrick 1024 * 1024)) {
12133cab2bb3Spatrick internal_strncpy(buf, tmpbuf, buf_len);
12143cab2bb3Spatrick UnmapOrDie(tmpbuf, tmpsize);
12153cab2bb3Spatrick return internal_strlen(buf);
12163cab2bb3Spatrick }
12173cab2bb3Spatrick #endif
12183cab2bb3Spatrick return ReadBinaryName(buf, buf_len);
12193cab2bb3Spatrick }
12203cab2bb3Spatrick
12213cab2bb3Spatrick // Match full names of the form /path/to/base_name{-,.}*
LibraryNameIs(const char * full_name,const char * base_name)12223cab2bb3Spatrick bool LibraryNameIs(const char *full_name, const char *base_name) {
12233cab2bb3Spatrick const char *name = full_name;
12243cab2bb3Spatrick // Strip path.
12253cab2bb3Spatrick while (*name != '\0') name++;
12263cab2bb3Spatrick while (name > full_name && *name != '/') name--;
12273cab2bb3Spatrick if (*name == '/') name++;
12283cab2bb3Spatrick uptr base_name_length = internal_strlen(base_name);
12293cab2bb3Spatrick if (internal_strncmp(name, base_name, base_name_length)) return false;
12303cab2bb3Spatrick return (name[base_name_length] == '-' || name[base_name_length] == '.');
12313cab2bb3Spatrick }
12323cab2bb3Spatrick
12333cab2bb3Spatrick #if !SANITIZER_ANDROID
12343cab2bb3Spatrick // Call cb for each region mapped by map.
ForEachMappedRegion(link_map * map,void (* cb)(const void *,uptr))12353cab2bb3Spatrick void ForEachMappedRegion(link_map *map, void (*cb)(const void *, uptr)) {
12363cab2bb3Spatrick CHECK_NE(map, nullptr);
1237d89ec533Spatrick #if !SANITIZER_FREEBSD
12383cab2bb3Spatrick typedef ElfW(Phdr) Elf_Phdr;
12393cab2bb3Spatrick typedef ElfW(Ehdr) Elf_Ehdr;
1240d89ec533Spatrick #endif // !SANITIZER_FREEBSD
12413cab2bb3Spatrick char *base = (char *)map->l_addr;
12423cab2bb3Spatrick Elf_Ehdr *ehdr = (Elf_Ehdr *)base;
12433cab2bb3Spatrick char *phdrs = base + ehdr->e_phoff;
12443cab2bb3Spatrick char *phdrs_end = phdrs + ehdr->e_phnum * ehdr->e_phentsize;
12453cab2bb3Spatrick
12463cab2bb3Spatrick // Find the segment with the minimum base so we can "relocate" the p_vaddr
12473cab2bb3Spatrick // fields. Typically ET_DYN objects (DSOs) have base of zero and ET_EXEC
12483cab2bb3Spatrick // objects have a non-zero base.
12493cab2bb3Spatrick uptr preferred_base = (uptr)-1;
12503cab2bb3Spatrick for (char *iter = phdrs; iter != phdrs_end; iter += ehdr->e_phentsize) {
12513cab2bb3Spatrick Elf_Phdr *phdr = (Elf_Phdr *)iter;
12523cab2bb3Spatrick if (phdr->p_type == PT_LOAD && preferred_base > (uptr)phdr->p_vaddr)
12533cab2bb3Spatrick preferred_base = (uptr)phdr->p_vaddr;
12543cab2bb3Spatrick }
12553cab2bb3Spatrick
12563cab2bb3Spatrick // Compute the delta from the real base to get a relocation delta.
12573cab2bb3Spatrick sptr delta = (uptr)base - preferred_base;
12583cab2bb3Spatrick // Now we can figure out what the loader really mapped.
12593cab2bb3Spatrick for (char *iter = phdrs; iter != phdrs_end; iter += ehdr->e_phentsize) {
12603cab2bb3Spatrick Elf_Phdr *phdr = (Elf_Phdr *)iter;
12613cab2bb3Spatrick if (phdr->p_type == PT_LOAD) {
12623cab2bb3Spatrick uptr seg_start = phdr->p_vaddr + delta;
12633cab2bb3Spatrick uptr seg_end = seg_start + phdr->p_memsz;
12643cab2bb3Spatrick // None of these values are aligned. We consider the ragged edges of the
12653cab2bb3Spatrick // load command as defined, since they are mapped from the file.
12663cab2bb3Spatrick seg_start = RoundDownTo(seg_start, GetPageSizeCached());
12673cab2bb3Spatrick seg_end = RoundUpTo(seg_end, GetPageSizeCached());
12683cab2bb3Spatrick cb((void *)seg_start, seg_end - seg_start);
12693cab2bb3Spatrick }
12703cab2bb3Spatrick }
12713cab2bb3Spatrick }
12723cab2bb3Spatrick #endif
12733cab2bb3Spatrick
1274*68dd5bb1Srobert #if SANITIZER_LINUX
1275*68dd5bb1Srobert #if defined(__x86_64__)
12763cab2bb3Spatrick // We cannot use glibc's clone wrapper, because it messes with the child
12773cab2bb3Spatrick // task's TLS. It writes the PID and TID of the child task to its thread
12783cab2bb3Spatrick // descriptor, but in our case the child task shares the thread descriptor with
12793cab2bb3Spatrick // the parent (because we don't know how to allocate a new thread
12803cab2bb3Spatrick // descriptor to keep glibc happy). So the stock version of clone(), when
12813cab2bb3Spatrick // 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)12823cab2bb3Spatrick uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
12833cab2bb3Spatrick int *parent_tidptr, void *newtls, int *child_tidptr) {
12843cab2bb3Spatrick long long res;
12853cab2bb3Spatrick if (!fn || !child_stack)
12863cab2bb3Spatrick return -EINVAL;
12873cab2bb3Spatrick CHECK_EQ(0, (uptr)child_stack % 16);
12883cab2bb3Spatrick child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
12893cab2bb3Spatrick ((unsigned long long *)child_stack)[0] = (uptr)fn;
12903cab2bb3Spatrick ((unsigned long long *)child_stack)[1] = (uptr)arg;
12913cab2bb3Spatrick register void *r8 __asm__("r8") = newtls;
12923cab2bb3Spatrick register int *r10 __asm__("r10") = child_tidptr;
12933cab2bb3Spatrick __asm__ __volatile__(
12943cab2bb3Spatrick /* %rax = syscall(%rax = SYSCALL(clone),
12953cab2bb3Spatrick * %rdi = flags,
12963cab2bb3Spatrick * %rsi = child_stack,
12973cab2bb3Spatrick * %rdx = parent_tidptr,
12983cab2bb3Spatrick * %r8 = new_tls,
12993cab2bb3Spatrick * %r10 = child_tidptr)
13003cab2bb3Spatrick */
13013cab2bb3Spatrick "syscall\n"
13023cab2bb3Spatrick
13033cab2bb3Spatrick /* if (%rax != 0)
13043cab2bb3Spatrick * return;
13053cab2bb3Spatrick */
13063cab2bb3Spatrick "testq %%rax,%%rax\n"
13073cab2bb3Spatrick "jnz 1f\n"
13083cab2bb3Spatrick
13093cab2bb3Spatrick /* In the child. Terminate unwind chain. */
13103cab2bb3Spatrick // XXX: We should also terminate the CFI unwind chain
13113cab2bb3Spatrick // here. Unfortunately clang 3.2 doesn't support the
13123cab2bb3Spatrick // necessary CFI directives, so we skip that part.
13133cab2bb3Spatrick "xorq %%rbp,%%rbp\n"
13143cab2bb3Spatrick
13153cab2bb3Spatrick /* Call "fn(arg)". */
13163cab2bb3Spatrick "popq %%rax\n"
13173cab2bb3Spatrick "popq %%rdi\n"
13183cab2bb3Spatrick "call *%%rax\n"
13193cab2bb3Spatrick
13203cab2bb3Spatrick /* Call _exit(%rax). */
13213cab2bb3Spatrick "movq %%rax,%%rdi\n"
13223cab2bb3Spatrick "movq %2,%%rax\n"
13233cab2bb3Spatrick "syscall\n"
13243cab2bb3Spatrick
13253cab2bb3Spatrick /* Return to parent. */
13263cab2bb3Spatrick "1:\n"
13273cab2bb3Spatrick : "=a" (res)
13283cab2bb3Spatrick : "a"(SYSCALL(clone)), "i"(SYSCALL(exit)),
13293cab2bb3Spatrick "S"(child_stack),
13303cab2bb3Spatrick "D"(flags),
13313cab2bb3Spatrick "d"(parent_tidptr),
13323cab2bb3Spatrick "r"(r8),
13333cab2bb3Spatrick "r"(r10)
13343cab2bb3Spatrick : "memory", "r11", "rcx");
13353cab2bb3Spatrick return res;
13363cab2bb3Spatrick }
13373cab2bb3Spatrick #elif defined(__mips__)
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)13383cab2bb3Spatrick uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
13393cab2bb3Spatrick int *parent_tidptr, void *newtls, int *child_tidptr) {
13403cab2bb3Spatrick long long res;
13413cab2bb3Spatrick if (!fn || !child_stack)
13423cab2bb3Spatrick return -EINVAL;
13433cab2bb3Spatrick CHECK_EQ(0, (uptr)child_stack % 16);
13443cab2bb3Spatrick child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
13453cab2bb3Spatrick ((unsigned long long *)child_stack)[0] = (uptr)fn;
13463cab2bb3Spatrick ((unsigned long long *)child_stack)[1] = (uptr)arg;
13473cab2bb3Spatrick register void *a3 __asm__("$7") = newtls;
13483cab2bb3Spatrick register int *a4 __asm__("$8") = child_tidptr;
13493cab2bb3Spatrick // We don't have proper CFI directives here because it requires alot of code
13503cab2bb3Spatrick // for very marginal benefits.
13513cab2bb3Spatrick __asm__ __volatile__(
13523cab2bb3Spatrick /* $v0 = syscall($v0 = __NR_clone,
13533cab2bb3Spatrick * $a0 = flags,
13543cab2bb3Spatrick * $a1 = child_stack,
13553cab2bb3Spatrick * $a2 = parent_tidptr,
13563cab2bb3Spatrick * $a3 = new_tls,
13573cab2bb3Spatrick * $a4 = child_tidptr)
13583cab2bb3Spatrick */
13593cab2bb3Spatrick ".cprestore 16;\n"
13603cab2bb3Spatrick "move $4,%1;\n"
13613cab2bb3Spatrick "move $5,%2;\n"
13623cab2bb3Spatrick "move $6,%3;\n"
13633cab2bb3Spatrick "move $7,%4;\n"
13643cab2bb3Spatrick /* Store the fifth argument on stack
13653cab2bb3Spatrick * if we are using 32-bit abi.
13663cab2bb3Spatrick */
13673cab2bb3Spatrick #if SANITIZER_WORDSIZE == 32
13683cab2bb3Spatrick "lw %5,16($29);\n"
13693cab2bb3Spatrick #else
13703cab2bb3Spatrick "move $8,%5;\n"
13713cab2bb3Spatrick #endif
13723cab2bb3Spatrick "li $2,%6;\n"
13733cab2bb3Spatrick "syscall;\n"
13743cab2bb3Spatrick
13753cab2bb3Spatrick /* if ($v0 != 0)
13763cab2bb3Spatrick * return;
13773cab2bb3Spatrick */
13783cab2bb3Spatrick "bnez $2,1f;\n"
13793cab2bb3Spatrick
13803cab2bb3Spatrick /* Call "fn(arg)". */
13813cab2bb3Spatrick #if SANITIZER_WORDSIZE == 32
13823cab2bb3Spatrick #ifdef __BIG_ENDIAN__
13833cab2bb3Spatrick "lw $25,4($29);\n"
13843cab2bb3Spatrick "lw $4,12($29);\n"
13853cab2bb3Spatrick #else
13863cab2bb3Spatrick "lw $25,0($29);\n"
13873cab2bb3Spatrick "lw $4,8($29);\n"
13883cab2bb3Spatrick #endif
13893cab2bb3Spatrick #else
13903cab2bb3Spatrick "ld $25,0($29);\n"
13913cab2bb3Spatrick "ld $4,8($29);\n"
13923cab2bb3Spatrick #endif
13933cab2bb3Spatrick "jal $25;\n"
13943cab2bb3Spatrick
13953cab2bb3Spatrick /* Call _exit($v0). */
13963cab2bb3Spatrick "move $4,$2;\n"
13973cab2bb3Spatrick "li $2,%7;\n"
13983cab2bb3Spatrick "syscall;\n"
13993cab2bb3Spatrick
14003cab2bb3Spatrick /* Return to parent. */
14013cab2bb3Spatrick "1:\n"
14023cab2bb3Spatrick : "=r" (res)
14033cab2bb3Spatrick : "r"(flags),
14043cab2bb3Spatrick "r"(child_stack),
14053cab2bb3Spatrick "r"(parent_tidptr),
14063cab2bb3Spatrick "r"(a3),
14073cab2bb3Spatrick "r"(a4),
14083cab2bb3Spatrick "i"(__NR_clone),
14093cab2bb3Spatrick "i"(__NR_exit)
14103cab2bb3Spatrick : "memory", "$29" );
14113cab2bb3Spatrick return res;
14123cab2bb3Spatrick }
1413d89ec533Spatrick #elif SANITIZER_RISCV64
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)1414d89ec533Spatrick uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1415d89ec533Spatrick int *parent_tidptr, void *newtls, int *child_tidptr) {
1416d89ec533Spatrick if (!fn || !child_stack)
1417d89ec533Spatrick return -EINVAL;
1418d89ec533Spatrick
1419d89ec533Spatrick CHECK_EQ(0, (uptr)child_stack % 16);
1420d89ec533Spatrick
1421d89ec533Spatrick register int res __asm__("a0");
1422d89ec533Spatrick register int __flags __asm__("a0") = flags;
1423d89ec533Spatrick register void *__stack __asm__("a1") = child_stack;
1424d89ec533Spatrick register int *__ptid __asm__("a2") = parent_tidptr;
1425d89ec533Spatrick register void *__tls __asm__("a3") = newtls;
1426d89ec533Spatrick register int *__ctid __asm__("a4") = child_tidptr;
1427d89ec533Spatrick register int (*__fn)(void *) __asm__("a5") = fn;
1428d89ec533Spatrick register void *__arg __asm__("a6") = arg;
1429d89ec533Spatrick register int nr_clone __asm__("a7") = __NR_clone;
1430d89ec533Spatrick
1431d89ec533Spatrick __asm__ __volatile__(
1432d89ec533Spatrick "ecall\n"
1433d89ec533Spatrick
1434d89ec533Spatrick /* if (a0 != 0)
1435d89ec533Spatrick * return a0;
1436d89ec533Spatrick */
1437d89ec533Spatrick "bnez a0, 1f\n"
1438d89ec533Spatrick
1439d89ec533Spatrick // In the child, now. Call "fn(arg)".
1440d89ec533Spatrick "mv a0, a6\n"
1441d89ec533Spatrick "jalr a5\n"
1442d89ec533Spatrick
1443d89ec533Spatrick // Call _exit(a0).
1444d89ec533Spatrick "addi a7, zero, %9\n"
1445d89ec533Spatrick "ecall\n"
1446d89ec533Spatrick "1:\n"
1447d89ec533Spatrick
1448d89ec533Spatrick : "=r"(res)
1449d89ec533Spatrick : "0"(__flags), "r"(__stack), "r"(__ptid), "r"(__tls), "r"(__ctid),
1450d89ec533Spatrick "r"(__fn), "r"(__arg), "r"(nr_clone), "i"(__NR_exit)
1451d89ec533Spatrick : "memory");
1452d89ec533Spatrick return res;
1453d89ec533Spatrick }
14543cab2bb3Spatrick #elif defined(__aarch64__)
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)14553cab2bb3Spatrick uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
14563cab2bb3Spatrick int *parent_tidptr, void *newtls, int *child_tidptr) {
1457*68dd5bb1Srobert register long long res __asm__("x0");
14583cab2bb3Spatrick if (!fn || !child_stack)
14593cab2bb3Spatrick return -EINVAL;
14603cab2bb3Spatrick CHECK_EQ(0, (uptr)child_stack % 16);
14613cab2bb3Spatrick child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
14623cab2bb3Spatrick ((unsigned long long *)child_stack)[0] = (uptr)fn;
14633cab2bb3Spatrick ((unsigned long long *)child_stack)[1] = (uptr)arg;
14643cab2bb3Spatrick
14653cab2bb3Spatrick register int (*__fn)(void *) __asm__("x0") = fn;
14663cab2bb3Spatrick register void *__stack __asm__("x1") = child_stack;
14673cab2bb3Spatrick register int __flags __asm__("x2") = flags;
14683cab2bb3Spatrick register void *__arg __asm__("x3") = arg;
14693cab2bb3Spatrick register int *__ptid __asm__("x4") = parent_tidptr;
14703cab2bb3Spatrick register void *__tls __asm__("x5") = newtls;
14713cab2bb3Spatrick register int *__ctid __asm__("x6") = child_tidptr;
14723cab2bb3Spatrick
14733cab2bb3Spatrick __asm__ __volatile__(
14743cab2bb3Spatrick "mov x0,x2\n" /* flags */
14753cab2bb3Spatrick "mov x2,x4\n" /* ptid */
14763cab2bb3Spatrick "mov x3,x5\n" /* tls */
14773cab2bb3Spatrick "mov x4,x6\n" /* ctid */
14783cab2bb3Spatrick "mov x8,%9\n" /* clone */
14793cab2bb3Spatrick
14803cab2bb3Spatrick "svc 0x0\n"
14813cab2bb3Spatrick
14823cab2bb3Spatrick /* if (%r0 != 0)
14833cab2bb3Spatrick * return %r0;
14843cab2bb3Spatrick */
14853cab2bb3Spatrick "cmp x0, #0\n"
14863cab2bb3Spatrick "bne 1f\n"
14873cab2bb3Spatrick
14883cab2bb3Spatrick /* In the child, now. Call "fn(arg)". */
14893cab2bb3Spatrick "ldp x1, x0, [sp], #16\n"
14903cab2bb3Spatrick "blr x1\n"
14913cab2bb3Spatrick
14923cab2bb3Spatrick /* Call _exit(%r0). */
14933cab2bb3Spatrick "mov x8, %10\n"
14943cab2bb3Spatrick "svc 0x0\n"
14953cab2bb3Spatrick "1:\n"
14963cab2bb3Spatrick
14973cab2bb3Spatrick : "=r" (res)
14983cab2bb3Spatrick : "i"(-EINVAL),
14993cab2bb3Spatrick "r"(__fn), "r"(__stack), "r"(__flags), "r"(__arg),
15003cab2bb3Spatrick "r"(__ptid), "r"(__tls), "r"(__ctid),
15013cab2bb3Spatrick "i"(__NR_clone), "i"(__NR_exit)
15023cab2bb3Spatrick : "x30", "memory");
15033cab2bb3Spatrick return res;
15043cab2bb3Spatrick }
1505*68dd5bb1Srobert #elif SANITIZER_LOONGARCH64
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)1506*68dd5bb1Srobert uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1507*68dd5bb1Srobert int *parent_tidptr, void *newtls, int *child_tidptr) {
1508*68dd5bb1Srobert if (!fn || !child_stack)
1509*68dd5bb1Srobert return -EINVAL;
1510*68dd5bb1Srobert
1511*68dd5bb1Srobert CHECK_EQ(0, (uptr)child_stack % 16);
1512*68dd5bb1Srobert
1513*68dd5bb1Srobert register int res __asm__("$a0");
1514*68dd5bb1Srobert register int __flags __asm__("$a0") = flags;
1515*68dd5bb1Srobert register void *__stack __asm__("$a1") = child_stack;
1516*68dd5bb1Srobert register int *__ptid __asm__("$a2") = parent_tidptr;
1517*68dd5bb1Srobert register int *__ctid __asm__("$a3") = child_tidptr;
1518*68dd5bb1Srobert register void *__tls __asm__("$a4") = newtls;
1519*68dd5bb1Srobert register int (*__fn)(void *) __asm__("$a5") = fn;
1520*68dd5bb1Srobert register void *__arg __asm__("$a6") = arg;
1521*68dd5bb1Srobert register int nr_clone __asm__("$a7") = __NR_clone;
1522*68dd5bb1Srobert
1523*68dd5bb1Srobert __asm__ __volatile__(
1524*68dd5bb1Srobert "syscall 0\n"
1525*68dd5bb1Srobert
1526*68dd5bb1Srobert // if ($a0 != 0)
1527*68dd5bb1Srobert // return $a0;
1528*68dd5bb1Srobert "bnez $a0, 1f\n"
1529*68dd5bb1Srobert
1530*68dd5bb1Srobert // In the child, now. Call "fn(arg)".
1531*68dd5bb1Srobert "move $a0, $a6\n"
1532*68dd5bb1Srobert "jirl $ra, $a5, 0\n"
1533*68dd5bb1Srobert
1534*68dd5bb1Srobert // Call _exit($a0).
1535*68dd5bb1Srobert "addi.d $a7, $zero, %9\n"
1536*68dd5bb1Srobert "syscall 0\n"
1537*68dd5bb1Srobert
1538*68dd5bb1Srobert "1:\n"
1539*68dd5bb1Srobert
1540*68dd5bb1Srobert : "=r"(res)
1541*68dd5bb1Srobert : "0"(__flags), "r"(__stack), "r"(__ptid), "r"(__ctid), "r"(__tls),
1542*68dd5bb1Srobert "r"(__fn), "r"(__arg), "r"(nr_clone), "i"(__NR_exit)
1543*68dd5bb1Srobert : "memory", "$t0", "$t1", "$t2", "$t3", "$t4", "$t5", "$t6", "$t7", "$t8");
1544*68dd5bb1Srobert return res;
1545*68dd5bb1Srobert }
15463cab2bb3Spatrick #elif defined(__powerpc64__)
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)15473cab2bb3Spatrick uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
15483cab2bb3Spatrick int *parent_tidptr, void *newtls, int *child_tidptr) {
15493cab2bb3Spatrick long long res;
15503cab2bb3Spatrick // Stack frame structure.
15513cab2bb3Spatrick #if SANITIZER_PPC64V1
15523cab2bb3Spatrick // Back chain == 0 (SP + 112)
15533cab2bb3Spatrick // Frame (112 bytes):
15543cab2bb3Spatrick // Parameter save area (SP + 48), 8 doublewords
15553cab2bb3Spatrick // TOC save area (SP + 40)
15563cab2bb3Spatrick // Link editor doubleword (SP + 32)
15573cab2bb3Spatrick // Compiler doubleword (SP + 24)
15583cab2bb3Spatrick // LR save area (SP + 16)
15593cab2bb3Spatrick // CR save area (SP + 8)
15603cab2bb3Spatrick // Back chain (SP + 0)
15613cab2bb3Spatrick # define FRAME_SIZE 112
15623cab2bb3Spatrick # define FRAME_TOC_SAVE_OFFSET 40
15633cab2bb3Spatrick #elif SANITIZER_PPC64V2
15643cab2bb3Spatrick // Back chain == 0 (SP + 32)
15653cab2bb3Spatrick // Frame (32 bytes):
15663cab2bb3Spatrick // TOC save area (SP + 24)
15673cab2bb3Spatrick // LR save area (SP + 16)
15683cab2bb3Spatrick // CR save area (SP + 8)
15693cab2bb3Spatrick // Back chain (SP + 0)
15703cab2bb3Spatrick # define FRAME_SIZE 32
15713cab2bb3Spatrick # define FRAME_TOC_SAVE_OFFSET 24
15723cab2bb3Spatrick #else
15733cab2bb3Spatrick # error "Unsupported PPC64 ABI"
15743cab2bb3Spatrick #endif
15753cab2bb3Spatrick if (!fn || !child_stack)
15763cab2bb3Spatrick return -EINVAL;
15773cab2bb3Spatrick CHECK_EQ(0, (uptr)child_stack % 16);
15783cab2bb3Spatrick
15793cab2bb3Spatrick register int (*__fn)(void *) __asm__("r3") = fn;
15803cab2bb3Spatrick register void *__cstack __asm__("r4") = child_stack;
15813cab2bb3Spatrick register int __flags __asm__("r5") = flags;
15823cab2bb3Spatrick register void *__arg __asm__("r6") = arg;
15833cab2bb3Spatrick register int *__ptidptr __asm__("r7") = parent_tidptr;
15843cab2bb3Spatrick register void *__newtls __asm__("r8") = newtls;
15853cab2bb3Spatrick register int *__ctidptr __asm__("r9") = child_tidptr;
15863cab2bb3Spatrick
15873cab2bb3Spatrick __asm__ __volatile__(
15883cab2bb3Spatrick /* fn and arg are saved across the syscall */
15893cab2bb3Spatrick "mr 28, %5\n\t"
15903cab2bb3Spatrick "mr 27, %8\n\t"
15913cab2bb3Spatrick
15923cab2bb3Spatrick /* syscall
15933cab2bb3Spatrick r0 == __NR_clone
15943cab2bb3Spatrick r3 == flags
15953cab2bb3Spatrick r4 == child_stack
15963cab2bb3Spatrick r5 == parent_tidptr
15973cab2bb3Spatrick r6 == newtls
15983cab2bb3Spatrick r7 == child_tidptr */
15993cab2bb3Spatrick "mr 3, %7\n\t"
16003cab2bb3Spatrick "mr 5, %9\n\t"
16013cab2bb3Spatrick "mr 6, %10\n\t"
16023cab2bb3Spatrick "mr 7, %11\n\t"
16033cab2bb3Spatrick "li 0, %3\n\t"
16043cab2bb3Spatrick "sc\n\t"
16053cab2bb3Spatrick
16063cab2bb3Spatrick /* Test if syscall was successful */
16073cab2bb3Spatrick "cmpdi cr1, 3, 0\n\t"
16083cab2bb3Spatrick "crandc cr1*4+eq, cr1*4+eq, cr0*4+so\n\t"
16093cab2bb3Spatrick "bne- cr1, 1f\n\t"
16103cab2bb3Spatrick
16113cab2bb3Spatrick /* Set up stack frame */
16123cab2bb3Spatrick "li 29, 0\n\t"
16133cab2bb3Spatrick "stdu 29, -8(1)\n\t"
16143cab2bb3Spatrick "stdu 1, -%12(1)\n\t"
16153cab2bb3Spatrick /* Do the function call */
16163cab2bb3Spatrick "std 2, %13(1)\n\t"
16173cab2bb3Spatrick #if SANITIZER_PPC64V1
16183cab2bb3Spatrick "ld 0, 0(28)\n\t"
16193cab2bb3Spatrick "ld 2, 8(28)\n\t"
16203cab2bb3Spatrick "mtctr 0\n\t"
16213cab2bb3Spatrick #elif SANITIZER_PPC64V2
16223cab2bb3Spatrick "mr 12, 28\n\t"
16233cab2bb3Spatrick "mtctr 12\n\t"
16243cab2bb3Spatrick #else
16253cab2bb3Spatrick # error "Unsupported PPC64 ABI"
16263cab2bb3Spatrick #endif
16273cab2bb3Spatrick "mr 3, 27\n\t"
16283cab2bb3Spatrick "bctrl\n\t"
16293cab2bb3Spatrick "ld 2, %13(1)\n\t"
16303cab2bb3Spatrick
16313cab2bb3Spatrick /* Call _exit(r3) */
16323cab2bb3Spatrick "li 0, %4\n\t"
16333cab2bb3Spatrick "sc\n\t"
16343cab2bb3Spatrick
16353cab2bb3Spatrick /* Return to parent */
16363cab2bb3Spatrick "1:\n\t"
16373cab2bb3Spatrick "mr %0, 3\n\t"
16383cab2bb3Spatrick : "=r" (res)
16393cab2bb3Spatrick : "0" (-1),
16403cab2bb3Spatrick "i" (EINVAL),
16413cab2bb3Spatrick "i" (__NR_clone),
16423cab2bb3Spatrick "i" (__NR_exit),
16433cab2bb3Spatrick "r" (__fn),
16443cab2bb3Spatrick "r" (__cstack),
16453cab2bb3Spatrick "r" (__flags),
16463cab2bb3Spatrick "r" (__arg),
16473cab2bb3Spatrick "r" (__ptidptr),
16483cab2bb3Spatrick "r" (__newtls),
16493cab2bb3Spatrick "r" (__ctidptr),
16503cab2bb3Spatrick "i" (FRAME_SIZE),
16513cab2bb3Spatrick "i" (FRAME_TOC_SAVE_OFFSET)
16523cab2bb3Spatrick : "cr0", "cr1", "memory", "ctr", "r0", "r27", "r28", "r29");
16533cab2bb3Spatrick return res;
16543cab2bb3Spatrick }
1655*68dd5bb1Srobert #elif defined(__i386__)
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)16563cab2bb3Spatrick uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
16573cab2bb3Spatrick int *parent_tidptr, void *newtls, int *child_tidptr) {
16583cab2bb3Spatrick int res;
16593cab2bb3Spatrick if (!fn || !child_stack)
16603cab2bb3Spatrick return -EINVAL;
16613cab2bb3Spatrick CHECK_EQ(0, (uptr)child_stack % 16);
16623cab2bb3Spatrick child_stack = (char *)child_stack - 7 * sizeof(unsigned int);
16633cab2bb3Spatrick ((unsigned int *)child_stack)[0] = (uptr)flags;
16643cab2bb3Spatrick ((unsigned int *)child_stack)[1] = (uptr)0;
16653cab2bb3Spatrick ((unsigned int *)child_stack)[2] = (uptr)fn;
16663cab2bb3Spatrick ((unsigned int *)child_stack)[3] = (uptr)arg;
16673cab2bb3Spatrick __asm__ __volatile__(
16683cab2bb3Spatrick /* %eax = syscall(%eax = SYSCALL(clone),
16693cab2bb3Spatrick * %ebx = flags,
16703cab2bb3Spatrick * %ecx = child_stack,
16713cab2bb3Spatrick * %edx = parent_tidptr,
16723cab2bb3Spatrick * %esi = new_tls,
16733cab2bb3Spatrick * %edi = child_tidptr)
16743cab2bb3Spatrick */
16753cab2bb3Spatrick
16763cab2bb3Spatrick /* Obtain flags */
16773cab2bb3Spatrick "movl (%%ecx), %%ebx\n"
16783cab2bb3Spatrick /* Do the system call */
16793cab2bb3Spatrick "pushl %%ebx\n"
16803cab2bb3Spatrick "pushl %%esi\n"
16813cab2bb3Spatrick "pushl %%edi\n"
16823cab2bb3Spatrick /* Remember the flag value. */
16833cab2bb3Spatrick "movl %%ebx, (%%ecx)\n"
16843cab2bb3Spatrick "int $0x80\n"
16853cab2bb3Spatrick "popl %%edi\n"
16863cab2bb3Spatrick "popl %%esi\n"
16873cab2bb3Spatrick "popl %%ebx\n"
16883cab2bb3Spatrick
16893cab2bb3Spatrick /* if (%eax != 0)
16903cab2bb3Spatrick * return;
16913cab2bb3Spatrick */
16923cab2bb3Spatrick
16933cab2bb3Spatrick "test %%eax,%%eax\n"
16943cab2bb3Spatrick "jnz 1f\n"
16953cab2bb3Spatrick
16963cab2bb3Spatrick /* terminate the stack frame */
16973cab2bb3Spatrick "xorl %%ebp,%%ebp\n"
16983cab2bb3Spatrick /* Call FN. */
16993cab2bb3Spatrick "call *%%ebx\n"
17003cab2bb3Spatrick #ifdef PIC
17013cab2bb3Spatrick "call here\n"
17023cab2bb3Spatrick "here:\n"
17033cab2bb3Spatrick "popl %%ebx\n"
17043cab2bb3Spatrick "addl $_GLOBAL_OFFSET_TABLE_+[.-here], %%ebx\n"
17053cab2bb3Spatrick #endif
17063cab2bb3Spatrick /* Call exit */
17073cab2bb3Spatrick "movl %%eax, %%ebx\n"
17083cab2bb3Spatrick "movl %2, %%eax\n"
17093cab2bb3Spatrick "int $0x80\n"
17103cab2bb3Spatrick "1:\n"
17113cab2bb3Spatrick : "=a" (res)
17123cab2bb3Spatrick : "a"(SYSCALL(clone)), "i"(SYSCALL(exit)),
17133cab2bb3Spatrick "c"(child_stack),
17143cab2bb3Spatrick "d"(parent_tidptr),
17153cab2bb3Spatrick "S"(newtls),
17163cab2bb3Spatrick "D"(child_tidptr)
17173cab2bb3Spatrick : "memory");
17183cab2bb3Spatrick return res;
17193cab2bb3Spatrick }
1720*68dd5bb1Srobert #elif defined(__arm__)
internal_clone(int (* fn)(void *),void * child_stack,int flags,void * arg,int * parent_tidptr,void * newtls,int * child_tidptr)17213cab2bb3Spatrick uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
17223cab2bb3Spatrick int *parent_tidptr, void *newtls, int *child_tidptr) {
17233cab2bb3Spatrick unsigned int res;
17243cab2bb3Spatrick if (!fn || !child_stack)
17253cab2bb3Spatrick return -EINVAL;
17263cab2bb3Spatrick child_stack = (char *)child_stack - 2 * sizeof(unsigned int);
17273cab2bb3Spatrick ((unsigned int *)child_stack)[0] = (uptr)fn;
17283cab2bb3Spatrick ((unsigned int *)child_stack)[1] = (uptr)arg;
17293cab2bb3Spatrick register int r0 __asm__("r0") = flags;
17303cab2bb3Spatrick register void *r1 __asm__("r1") = child_stack;
17313cab2bb3Spatrick register int *r2 __asm__("r2") = parent_tidptr;
17323cab2bb3Spatrick register void *r3 __asm__("r3") = newtls;
17333cab2bb3Spatrick register int *r4 __asm__("r4") = child_tidptr;
17343cab2bb3Spatrick register int r7 __asm__("r7") = __NR_clone;
17353cab2bb3Spatrick
17363cab2bb3Spatrick #if __ARM_ARCH > 4 || defined (__ARM_ARCH_4T__)
17373cab2bb3Spatrick # define ARCH_HAS_BX
17383cab2bb3Spatrick #endif
17393cab2bb3Spatrick #if __ARM_ARCH > 4
17403cab2bb3Spatrick # define ARCH_HAS_BLX
17413cab2bb3Spatrick #endif
17423cab2bb3Spatrick
17433cab2bb3Spatrick #ifdef ARCH_HAS_BX
17443cab2bb3Spatrick # ifdef ARCH_HAS_BLX
17453cab2bb3Spatrick # define BLX(R) "blx " #R "\n"
17463cab2bb3Spatrick # else
17473cab2bb3Spatrick # define BLX(R) "mov lr, pc; bx " #R "\n"
17483cab2bb3Spatrick # endif
17493cab2bb3Spatrick #else
17503cab2bb3Spatrick # define BLX(R) "mov lr, pc; mov pc," #R "\n"
17513cab2bb3Spatrick #endif
17523cab2bb3Spatrick
17533cab2bb3Spatrick __asm__ __volatile__(
17543cab2bb3Spatrick /* %r0 = syscall(%r7 = SYSCALL(clone),
17553cab2bb3Spatrick * %r0 = flags,
17563cab2bb3Spatrick * %r1 = child_stack,
17573cab2bb3Spatrick * %r2 = parent_tidptr,
17583cab2bb3Spatrick * %r3 = new_tls,
17593cab2bb3Spatrick * %r4 = child_tidptr)
17603cab2bb3Spatrick */
17613cab2bb3Spatrick
17623cab2bb3Spatrick /* Do the system call */
17633cab2bb3Spatrick "swi 0x0\n"
17643cab2bb3Spatrick
17653cab2bb3Spatrick /* if (%r0 != 0)
17663cab2bb3Spatrick * return %r0;
17673cab2bb3Spatrick */
17683cab2bb3Spatrick "cmp r0, #0\n"
17693cab2bb3Spatrick "bne 1f\n"
17703cab2bb3Spatrick
17713cab2bb3Spatrick /* In the child, now. Call "fn(arg)". */
17723cab2bb3Spatrick "ldr r0, [sp, #4]\n"
17733cab2bb3Spatrick "ldr ip, [sp], #8\n"
17743cab2bb3Spatrick BLX(ip)
17753cab2bb3Spatrick /* Call _exit(%r0). */
17763cab2bb3Spatrick "mov r7, %7\n"
17773cab2bb3Spatrick "swi 0x0\n"
17783cab2bb3Spatrick "1:\n"
17793cab2bb3Spatrick "mov %0, r0\n"
17803cab2bb3Spatrick : "=r"(res)
17813cab2bb3Spatrick : "r"(r0), "r"(r1), "r"(r2), "r"(r3), "r"(r4), "r"(r7),
17823cab2bb3Spatrick "i"(__NR_exit)
17833cab2bb3Spatrick : "memory");
17843cab2bb3Spatrick return res;
17853cab2bb3Spatrick }
1786*68dd5bb1Srobert #endif
1787*68dd5bb1Srobert #endif // SANITIZER_LINUX
17883cab2bb3Spatrick
17891f9cb04fSpatrick #if SANITIZER_LINUX
internal_uname(struct utsname * buf)17901f9cb04fSpatrick int internal_uname(struct utsname *buf) {
17911f9cb04fSpatrick return internal_syscall(SYSCALL(uname), buf);
17921f9cb04fSpatrick }
17931f9cb04fSpatrick #endif
17941f9cb04fSpatrick
17953cab2bb3Spatrick #if SANITIZER_ANDROID
17963cab2bb3Spatrick #if __ANDROID_API__ < 21
17973cab2bb3Spatrick extern "C" __attribute__((weak)) int dl_iterate_phdr(
17983cab2bb3Spatrick int (*)(struct dl_phdr_info *, size_t, void *), void *);
17993cab2bb3Spatrick #endif
18003cab2bb3Spatrick
dl_iterate_phdr_test_cb(struct dl_phdr_info * info,size_t size,void * data)18013cab2bb3Spatrick static int dl_iterate_phdr_test_cb(struct dl_phdr_info *info, size_t size,
18023cab2bb3Spatrick void *data) {
18033cab2bb3Spatrick // Any name starting with "lib" indicates a bug in L where library base names
18043cab2bb3Spatrick // are returned instead of paths.
18053cab2bb3Spatrick if (info->dlpi_name && info->dlpi_name[0] == 'l' &&
18063cab2bb3Spatrick info->dlpi_name[1] == 'i' && info->dlpi_name[2] == 'b') {
18073cab2bb3Spatrick *(bool *)data = true;
18083cab2bb3Spatrick return 1;
18093cab2bb3Spatrick }
18103cab2bb3Spatrick return 0;
18113cab2bb3Spatrick }
18123cab2bb3Spatrick
18133cab2bb3Spatrick static atomic_uint32_t android_api_level;
18143cab2bb3Spatrick
AndroidDetectApiLevelStatic()18153cab2bb3Spatrick static AndroidApiLevel AndroidDetectApiLevelStatic() {
18163cab2bb3Spatrick #if __ANDROID_API__ <= 19
18173cab2bb3Spatrick return ANDROID_KITKAT;
18183cab2bb3Spatrick #elif __ANDROID_API__ <= 22
18193cab2bb3Spatrick return ANDROID_LOLLIPOP_MR1;
18203cab2bb3Spatrick #else
18213cab2bb3Spatrick return ANDROID_POST_LOLLIPOP;
18223cab2bb3Spatrick #endif
18233cab2bb3Spatrick }
18243cab2bb3Spatrick
AndroidDetectApiLevel()18253cab2bb3Spatrick static AndroidApiLevel AndroidDetectApiLevel() {
18263cab2bb3Spatrick if (!&dl_iterate_phdr)
18273cab2bb3Spatrick return ANDROID_KITKAT; // K or lower
18283cab2bb3Spatrick bool base_name_seen = false;
18293cab2bb3Spatrick dl_iterate_phdr(dl_iterate_phdr_test_cb, &base_name_seen);
18303cab2bb3Spatrick if (base_name_seen)
18313cab2bb3Spatrick return ANDROID_LOLLIPOP_MR1; // L MR1
18323cab2bb3Spatrick return ANDROID_POST_LOLLIPOP; // post-L
18333cab2bb3Spatrick // Plain L (API level 21) is completely broken wrt ASan and not very
18343cab2bb3Spatrick // interesting to detect.
18353cab2bb3Spatrick }
18363cab2bb3Spatrick
18373cab2bb3Spatrick extern "C" __attribute__((weak)) void* _DYNAMIC;
18383cab2bb3Spatrick
AndroidGetApiLevel()18393cab2bb3Spatrick AndroidApiLevel AndroidGetApiLevel() {
18403cab2bb3Spatrick AndroidApiLevel level =
18413cab2bb3Spatrick (AndroidApiLevel)atomic_load(&android_api_level, memory_order_relaxed);
18423cab2bb3Spatrick if (level) return level;
18433cab2bb3Spatrick level = &_DYNAMIC == nullptr ? AndroidDetectApiLevelStatic()
18443cab2bb3Spatrick : AndroidDetectApiLevel();
18453cab2bb3Spatrick atomic_store(&android_api_level, level, memory_order_relaxed);
18463cab2bb3Spatrick return level;
18473cab2bb3Spatrick }
18483cab2bb3Spatrick
18493cab2bb3Spatrick #endif
18503cab2bb3Spatrick
GetHandleSignalModeImpl(int signum)18513cab2bb3Spatrick static HandleSignalMode GetHandleSignalModeImpl(int signum) {
18523cab2bb3Spatrick switch (signum) {
18533cab2bb3Spatrick case SIGABRT:
18543cab2bb3Spatrick return common_flags()->handle_abort;
18553cab2bb3Spatrick case SIGILL:
18563cab2bb3Spatrick return common_flags()->handle_sigill;
18573cab2bb3Spatrick case SIGTRAP:
18583cab2bb3Spatrick return common_flags()->handle_sigtrap;
18593cab2bb3Spatrick case SIGFPE:
18603cab2bb3Spatrick return common_flags()->handle_sigfpe;
18613cab2bb3Spatrick case SIGSEGV:
18623cab2bb3Spatrick return common_flags()->handle_segv;
18633cab2bb3Spatrick case SIGBUS:
18643cab2bb3Spatrick return common_flags()->handle_sigbus;
18653cab2bb3Spatrick }
18663cab2bb3Spatrick return kHandleSignalNo;
18673cab2bb3Spatrick }
18683cab2bb3Spatrick
GetHandleSignalMode(int signum)18693cab2bb3Spatrick HandleSignalMode GetHandleSignalMode(int signum) {
18703cab2bb3Spatrick HandleSignalMode result = GetHandleSignalModeImpl(signum);
18713cab2bb3Spatrick if (result == kHandleSignalYes && !common_flags()->allow_user_segv_handler)
18723cab2bb3Spatrick return kHandleSignalExclusive;
18733cab2bb3Spatrick return result;
18743cab2bb3Spatrick }
18753cab2bb3Spatrick
18763cab2bb3Spatrick #if !SANITIZER_GO
internal_start_thread(void * (* func)(void * arg),void * arg)18771f9cb04fSpatrick void *internal_start_thread(void *(*func)(void *arg), void *arg) {
1878*68dd5bb1Srobert if (&real_pthread_create == 0)
1879*68dd5bb1Srobert return nullptr;
18803cab2bb3Spatrick // Start the thread with signals blocked, otherwise it can steal user signals.
1881*68dd5bb1Srobert ScopedBlockSignals block(nullptr);
18823cab2bb3Spatrick void *th;
18831f9cb04fSpatrick real_pthread_create(&th, nullptr, func, arg);
18843cab2bb3Spatrick return th;
18853cab2bb3Spatrick }
18863cab2bb3Spatrick
internal_join_thread(void * th)18873cab2bb3Spatrick void internal_join_thread(void *th) {
1888*68dd5bb1Srobert if (&real_pthread_join)
18893cab2bb3Spatrick real_pthread_join(th, nullptr);
18903cab2bb3Spatrick }
18913cab2bb3Spatrick #else
internal_start_thread(void * (* func)(void *),void * arg)18921f9cb04fSpatrick void *internal_start_thread(void *(*func)(void *), void *arg) { return 0; }
18933cab2bb3Spatrick
internal_join_thread(void * th)18943cab2bb3Spatrick void internal_join_thread(void *th) {}
18953cab2bb3Spatrick #endif
18963cab2bb3Spatrick
1897*68dd5bb1Srobert #if SANITIZER_LINUX && defined(__aarch64__)
18983cab2bb3Spatrick // Android headers in the older NDK releases miss this definition.
18993cab2bb3Spatrick struct __sanitizer_esr_context {
19003cab2bb3Spatrick struct _aarch64_ctx head;
19013cab2bb3Spatrick uint64_t esr;
19023cab2bb3Spatrick };
19033cab2bb3Spatrick
Aarch64GetESR(ucontext_t * ucontext,u64 * esr)19043cab2bb3Spatrick static bool Aarch64GetESR(ucontext_t *ucontext, u64 *esr) {
19053cab2bb3Spatrick static const u32 kEsrMagic = 0x45535201;
1906*68dd5bb1Srobert u8 *aux = reinterpret_cast<u8 *>(ucontext->uc_mcontext.__reserved);
19073cab2bb3Spatrick while (true) {
19083cab2bb3Spatrick _aarch64_ctx *ctx = (_aarch64_ctx *)aux;
19093cab2bb3Spatrick if (ctx->size == 0) break;
19103cab2bb3Spatrick if (ctx->magic == kEsrMagic) {
19113cab2bb3Spatrick *esr = ((__sanitizer_esr_context *)ctx)->esr;
19123cab2bb3Spatrick return true;
19133cab2bb3Spatrick }
19143cab2bb3Spatrick aux += ctx->size;
19153cab2bb3Spatrick }
19163cab2bb3Spatrick return false;
19173cab2bb3Spatrick }
1918*68dd5bb1Srobert #elif SANITIZER_FREEBSD && defined(__aarch64__)
1919*68dd5bb1Srobert // FreeBSD doesn't provide ESR in the ucontext.
Aarch64GetESR(ucontext_t * ucontext,u64 * esr)1920*68dd5bb1Srobert static bool Aarch64GetESR(ucontext_t *ucontext, u64 *esr) {
1921*68dd5bb1Srobert return false;
1922*68dd5bb1Srobert }
19233cab2bb3Spatrick #endif
19243cab2bb3Spatrick
19253cab2bb3Spatrick using Context = ucontext_t;
19263cab2bb3Spatrick
GetWriteFlag() const19273cab2bb3Spatrick SignalContext::WriteFlag SignalContext::GetWriteFlag() const {
19283cab2bb3Spatrick Context *ucontext = (Context *)context;
19293cab2bb3Spatrick #if defined(__x86_64__) || defined(__i386__)
19303cab2bb3Spatrick static const uptr PF_WRITE = 1U << 1;
19313cab2bb3Spatrick #if SANITIZER_FREEBSD
19323cab2bb3Spatrick uptr err = ucontext->uc_mcontext.mc_err;
19333cab2bb3Spatrick #elif SANITIZER_NETBSD
19343cab2bb3Spatrick uptr err = ucontext->uc_mcontext.__gregs[_REG_ERR];
19353cab2bb3Spatrick #elif SANITIZER_SOLARIS && defined(__i386__)
19363cab2bb3Spatrick const int Err = 13;
19373cab2bb3Spatrick uptr err = ucontext->uc_mcontext.gregs[Err];
19383cab2bb3Spatrick #else
19393cab2bb3Spatrick uptr err = ucontext->uc_mcontext.gregs[REG_ERR];
19403cab2bb3Spatrick #endif // SANITIZER_FREEBSD
1941*68dd5bb1Srobert return err & PF_WRITE ? Write : Read;
19423cab2bb3Spatrick #elif defined(__mips__)
19433cab2bb3Spatrick uint32_t *exception_source;
19443cab2bb3Spatrick uint32_t faulty_instruction;
19453cab2bb3Spatrick uint32_t op_code;
19463cab2bb3Spatrick
19473cab2bb3Spatrick exception_source = (uint32_t *)ucontext->uc_mcontext.pc;
19483cab2bb3Spatrick faulty_instruction = (uint32_t)(*exception_source);
19493cab2bb3Spatrick
19503cab2bb3Spatrick op_code = (faulty_instruction >> 26) & 0x3f;
19513cab2bb3Spatrick
19523cab2bb3Spatrick // FIXME: Add support for FPU, microMIPS, DSP, MSA memory instructions.
19533cab2bb3Spatrick switch (op_code) {
19543cab2bb3Spatrick case 0x28: // sb
19553cab2bb3Spatrick case 0x29: // sh
19563cab2bb3Spatrick case 0x2b: // sw
19573cab2bb3Spatrick case 0x3f: // sd
19583cab2bb3Spatrick #if __mips_isa_rev < 6
19593cab2bb3Spatrick case 0x2c: // sdl
19603cab2bb3Spatrick case 0x2d: // sdr
19613cab2bb3Spatrick case 0x2a: // swl
19623cab2bb3Spatrick case 0x2e: // swr
19633cab2bb3Spatrick #endif
1964*68dd5bb1Srobert return SignalContext::Write;
19653cab2bb3Spatrick
19663cab2bb3Spatrick case 0x20: // lb
19673cab2bb3Spatrick case 0x24: // lbu
19683cab2bb3Spatrick case 0x21: // lh
19693cab2bb3Spatrick case 0x25: // lhu
19703cab2bb3Spatrick case 0x23: // lw
19713cab2bb3Spatrick case 0x27: // lwu
19723cab2bb3Spatrick case 0x37: // ld
19733cab2bb3Spatrick #if __mips_isa_rev < 6
19743cab2bb3Spatrick case 0x1a: // ldl
19753cab2bb3Spatrick case 0x1b: // ldr
19763cab2bb3Spatrick case 0x22: // lwl
19773cab2bb3Spatrick case 0x26: // lwr
19783cab2bb3Spatrick #endif
1979*68dd5bb1Srobert return SignalContext::Read;
19803cab2bb3Spatrick #if __mips_isa_rev == 6
19813cab2bb3Spatrick case 0x3b: // pcrel
19823cab2bb3Spatrick op_code = (faulty_instruction >> 19) & 0x3;
19833cab2bb3Spatrick switch (op_code) {
19843cab2bb3Spatrick case 0x1: // lwpc
19853cab2bb3Spatrick case 0x2: // lwupc
1986*68dd5bb1Srobert return SignalContext::Read;
19873cab2bb3Spatrick }
19883cab2bb3Spatrick #endif
19893cab2bb3Spatrick }
1990*68dd5bb1Srobert return SignalContext::Unknown;
19913cab2bb3Spatrick #elif defined(__arm__)
19923cab2bb3Spatrick static const uptr FSR_WRITE = 1U << 11;
19933cab2bb3Spatrick uptr fsr = ucontext->uc_mcontext.error_code;
1994*68dd5bb1Srobert return fsr & FSR_WRITE ? Write : Read;
19953cab2bb3Spatrick #elif defined(__aarch64__)
19963cab2bb3Spatrick static const u64 ESR_ELx_WNR = 1U << 6;
19973cab2bb3Spatrick u64 esr;
1998*68dd5bb1Srobert if (!Aarch64GetESR(ucontext, &esr)) return Unknown;
1999*68dd5bb1Srobert return esr & ESR_ELx_WNR ? Write : Read;
2000*68dd5bb1Srobert #elif defined(__loongarch__)
2001*68dd5bb1Srobert u32 flags = ucontext->uc_mcontext.__flags;
2002*68dd5bb1Srobert if (flags & SC_ADDRERR_RD)
2003*68dd5bb1Srobert return SignalContext::Read;
2004*68dd5bb1Srobert if (flags & SC_ADDRERR_WR)
2005*68dd5bb1Srobert return SignalContext::Write;
2006*68dd5bb1Srobert return SignalContext::Unknown;
20073cab2bb3Spatrick #elif defined(__sparc__)
20083cab2bb3Spatrick // Decode the instruction to determine the access type.
20093cab2bb3Spatrick // From OpenSolaris $SRC/uts/sun4/os/trap.c (get_accesstype).
20103cab2bb3Spatrick #if SANITIZER_SOLARIS
20113cab2bb3Spatrick uptr pc = ucontext->uc_mcontext.gregs[REG_PC];
20123cab2bb3Spatrick #else
20133cab2bb3Spatrick // Historical BSDism here.
20143cab2bb3Spatrick struct sigcontext *scontext = (struct sigcontext *)context;
20153cab2bb3Spatrick #if defined(__arch64__)
20163cab2bb3Spatrick uptr pc = scontext->sigc_regs.tpc;
20173cab2bb3Spatrick #else
20183cab2bb3Spatrick uptr pc = scontext->si_regs.pc;
20193cab2bb3Spatrick #endif
20203cab2bb3Spatrick #endif
20213cab2bb3Spatrick u32 instr = *(u32 *)pc;
2022*68dd5bb1Srobert return (instr >> 21) & 1 ? Write: Read;
20231f9cb04fSpatrick #elif defined(__riscv)
2024*68dd5bb1Srobert #if SANITIZER_FREEBSD
2025*68dd5bb1Srobert unsigned long pc = ucontext->uc_mcontext.mc_gpregs.gp_sepc;
2026*68dd5bb1Srobert #else
20271f9cb04fSpatrick unsigned long pc = ucontext->uc_mcontext.__gregs[REG_PC];
2028*68dd5bb1Srobert #endif
20291f9cb04fSpatrick unsigned faulty_instruction = *(uint16_t *)pc;
20301f9cb04fSpatrick
20311f9cb04fSpatrick #if defined(__riscv_compressed)
20321f9cb04fSpatrick if ((faulty_instruction & 0x3) != 0x3) { // it's a compressed instruction
20331f9cb04fSpatrick // set op_bits to the instruction bits [1, 0, 15, 14, 13]
20341f9cb04fSpatrick unsigned op_bits =
20351f9cb04fSpatrick ((faulty_instruction & 0x3) << 3) | (faulty_instruction >> 13);
20361f9cb04fSpatrick unsigned rd = faulty_instruction & 0xF80; // bits 7-11, inclusive
20371f9cb04fSpatrick switch (op_bits) {
20381f9cb04fSpatrick case 0b10'010: // c.lwsp (rd != x0)
20391f9cb04fSpatrick #if __riscv_xlen == 64
20401f9cb04fSpatrick case 0b10'011: // c.ldsp (rd != x0)
20411f9cb04fSpatrick #endif
2042*68dd5bb1Srobert return rd ? SignalContext::Read : SignalContext::Unknown;
20431f9cb04fSpatrick case 0b00'010: // c.lw
20441f9cb04fSpatrick #if __riscv_flen >= 32 && __riscv_xlen == 32
20451f9cb04fSpatrick case 0b10'011: // c.flwsp
20461f9cb04fSpatrick #endif
20471f9cb04fSpatrick #if __riscv_flen >= 32 || __riscv_xlen == 64
20481f9cb04fSpatrick case 0b00'011: // c.flw / c.ld
20491f9cb04fSpatrick #endif
20501f9cb04fSpatrick #if __riscv_flen == 64
20511f9cb04fSpatrick case 0b00'001: // c.fld
20521f9cb04fSpatrick case 0b10'001: // c.fldsp
20531f9cb04fSpatrick #endif
2054*68dd5bb1Srobert return SignalContext::Read;
20551f9cb04fSpatrick case 0b00'110: // c.sw
20561f9cb04fSpatrick case 0b10'110: // c.swsp
20571f9cb04fSpatrick #if __riscv_flen >= 32 || __riscv_xlen == 64
20581f9cb04fSpatrick case 0b00'111: // c.fsw / c.sd
20591f9cb04fSpatrick case 0b10'111: // c.fswsp / c.sdsp
20601f9cb04fSpatrick #endif
20611f9cb04fSpatrick #if __riscv_flen == 64
20621f9cb04fSpatrick case 0b00'101: // c.fsd
20631f9cb04fSpatrick case 0b10'101: // c.fsdsp
20641f9cb04fSpatrick #endif
2065*68dd5bb1Srobert return SignalContext::Write;
20661f9cb04fSpatrick default:
2067*68dd5bb1Srobert return SignalContext::Unknown;
20681f9cb04fSpatrick }
20691f9cb04fSpatrick }
20701f9cb04fSpatrick #endif
20711f9cb04fSpatrick
20721f9cb04fSpatrick unsigned opcode = faulty_instruction & 0x7f; // lower 7 bits
20731f9cb04fSpatrick unsigned funct3 = (faulty_instruction >> 12) & 0x7; // bits 12-14, inclusive
20741f9cb04fSpatrick switch (opcode) {
20751f9cb04fSpatrick case 0b0000011: // loads
20761f9cb04fSpatrick switch (funct3) {
20771f9cb04fSpatrick case 0b000: // lb
20781f9cb04fSpatrick case 0b001: // lh
20791f9cb04fSpatrick case 0b010: // lw
20801f9cb04fSpatrick #if __riscv_xlen == 64
20811f9cb04fSpatrick case 0b011: // ld
20821f9cb04fSpatrick #endif
20831f9cb04fSpatrick case 0b100: // lbu
20841f9cb04fSpatrick case 0b101: // lhu
2085*68dd5bb1Srobert return SignalContext::Read;
20861f9cb04fSpatrick default:
2087*68dd5bb1Srobert return SignalContext::Unknown;
20881f9cb04fSpatrick }
20891f9cb04fSpatrick case 0b0100011: // stores
20901f9cb04fSpatrick switch (funct3) {
20911f9cb04fSpatrick case 0b000: // sb
20921f9cb04fSpatrick case 0b001: // sh
20931f9cb04fSpatrick case 0b010: // sw
20941f9cb04fSpatrick #if __riscv_xlen == 64
20951f9cb04fSpatrick case 0b011: // sd
20961f9cb04fSpatrick #endif
2097*68dd5bb1Srobert return SignalContext::Write;
20981f9cb04fSpatrick default:
2099*68dd5bb1Srobert return SignalContext::Unknown;
21001f9cb04fSpatrick }
21011f9cb04fSpatrick #if __riscv_flen >= 32
21021f9cb04fSpatrick case 0b0000111: // floating-point loads
21031f9cb04fSpatrick switch (funct3) {
21041f9cb04fSpatrick case 0b010: // flw
21051f9cb04fSpatrick #if __riscv_flen == 64
21061f9cb04fSpatrick case 0b011: // fld
21071f9cb04fSpatrick #endif
2108*68dd5bb1Srobert return SignalContext::Read;
21091f9cb04fSpatrick default:
2110*68dd5bb1Srobert return SignalContext::Unknown;
21111f9cb04fSpatrick }
21121f9cb04fSpatrick case 0b0100111: // floating-point stores
21131f9cb04fSpatrick switch (funct3) {
21141f9cb04fSpatrick case 0b010: // fsw
21151f9cb04fSpatrick #if __riscv_flen == 64
21161f9cb04fSpatrick case 0b011: // fsd
21171f9cb04fSpatrick #endif
2118*68dd5bb1Srobert return SignalContext::Write;
21191f9cb04fSpatrick default:
2120*68dd5bb1Srobert return SignalContext::Unknown;
21211f9cb04fSpatrick }
21221f9cb04fSpatrick #endif
21231f9cb04fSpatrick default:
2124*68dd5bb1Srobert return SignalContext::Unknown;
21251f9cb04fSpatrick }
21263cab2bb3Spatrick #else
21273cab2bb3Spatrick (void)ucontext;
2128*68dd5bb1Srobert return Unknown; // FIXME: Implement.
21293cab2bb3Spatrick #endif
21303cab2bb3Spatrick }
21313cab2bb3Spatrick
IsTrueFaultingAddress() const21323cab2bb3Spatrick bool SignalContext::IsTrueFaultingAddress() const {
21333cab2bb3Spatrick auto si = static_cast<const siginfo_t *>(siginfo);
21343cab2bb3Spatrick // SIGSEGV signals without a true fault address have si_code set to 128.
21353cab2bb3Spatrick return si->si_signo == SIGSEGV && si->si_code != 128;
21363cab2bb3Spatrick }
21373cab2bb3Spatrick
DumpAllRegisters(void * context)21383cab2bb3Spatrick void SignalContext::DumpAllRegisters(void *context) {
21393cab2bb3Spatrick // FIXME: Implement this.
21403cab2bb3Spatrick }
21413cab2bb3Spatrick
GetPcSpBp(void * context,uptr * pc,uptr * sp,uptr * bp)21423cab2bb3Spatrick static void GetPcSpBp(void *context, uptr *pc, uptr *sp, uptr *bp) {
21433cab2bb3Spatrick #if SANITIZER_NETBSD
21443cab2bb3Spatrick // This covers all NetBSD architectures
21453cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t *)context;
21463cab2bb3Spatrick *pc = _UC_MACHINE_PC(ucontext);
21473cab2bb3Spatrick *bp = _UC_MACHINE_FP(ucontext);
21483cab2bb3Spatrick *sp = _UC_MACHINE_SP(ucontext);
21493cab2bb3Spatrick #elif defined(__arm__)
21503cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t*)context;
21513cab2bb3Spatrick *pc = ucontext->uc_mcontext.arm_pc;
21523cab2bb3Spatrick *bp = ucontext->uc_mcontext.arm_fp;
21533cab2bb3Spatrick *sp = ucontext->uc_mcontext.arm_sp;
21543cab2bb3Spatrick #elif defined(__aarch64__)
2155*68dd5bb1Srobert # if SANITIZER_FREEBSD
2156*68dd5bb1Srobert ucontext_t *ucontext = (ucontext_t*)context;
2157*68dd5bb1Srobert *pc = ucontext->uc_mcontext.mc_gpregs.gp_elr;
2158*68dd5bb1Srobert *bp = ucontext->uc_mcontext.mc_gpregs.gp_x[29];
2159*68dd5bb1Srobert *sp = ucontext->uc_mcontext.mc_gpregs.gp_sp;
2160*68dd5bb1Srobert # else
21613cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t*)context;
21623cab2bb3Spatrick *pc = ucontext->uc_mcontext.pc;
21633cab2bb3Spatrick *bp = ucontext->uc_mcontext.regs[29];
21643cab2bb3Spatrick *sp = ucontext->uc_mcontext.sp;
2165*68dd5bb1Srobert # endif
21663cab2bb3Spatrick #elif defined(__hppa__)
21673cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t*)context;
21683cab2bb3Spatrick *pc = ucontext->uc_mcontext.sc_iaoq[0];
21693cab2bb3Spatrick /* GCC uses %r3 whenever a frame pointer is needed. */
21703cab2bb3Spatrick *bp = ucontext->uc_mcontext.sc_gr[3];
21713cab2bb3Spatrick *sp = ucontext->uc_mcontext.sc_gr[30];
21723cab2bb3Spatrick #elif defined(__x86_64__)
21733cab2bb3Spatrick # if SANITIZER_FREEBSD
21743cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t*)context;
21753cab2bb3Spatrick *pc = ucontext->uc_mcontext.mc_rip;
21763cab2bb3Spatrick *bp = ucontext->uc_mcontext.mc_rbp;
21773cab2bb3Spatrick *sp = ucontext->uc_mcontext.mc_rsp;
21783cab2bb3Spatrick # else
21793cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t*)context;
21803cab2bb3Spatrick *pc = ucontext->uc_mcontext.gregs[REG_RIP];
21813cab2bb3Spatrick *bp = ucontext->uc_mcontext.gregs[REG_RBP];
21823cab2bb3Spatrick *sp = ucontext->uc_mcontext.gregs[REG_RSP];
21833cab2bb3Spatrick # endif
21843cab2bb3Spatrick #elif defined(__i386__)
21853cab2bb3Spatrick # if SANITIZER_FREEBSD
21863cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t*)context;
21873cab2bb3Spatrick *pc = ucontext->uc_mcontext.mc_eip;
21883cab2bb3Spatrick *bp = ucontext->uc_mcontext.mc_ebp;
21893cab2bb3Spatrick *sp = ucontext->uc_mcontext.mc_esp;
21903cab2bb3Spatrick # else
21913cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t*)context;
21923cab2bb3Spatrick # if SANITIZER_SOLARIS
21933cab2bb3Spatrick /* Use the numeric values: the symbolic ones are undefined by llvm
21943cab2bb3Spatrick include/llvm/Support/Solaris.h. */
21953cab2bb3Spatrick # ifndef REG_EIP
21963cab2bb3Spatrick # define REG_EIP 14 // REG_PC
21973cab2bb3Spatrick # endif
21983cab2bb3Spatrick # ifndef REG_EBP
21993cab2bb3Spatrick # define REG_EBP 6 // REG_FP
22003cab2bb3Spatrick # endif
22011f9cb04fSpatrick # ifndef REG_UESP
22021f9cb04fSpatrick # define REG_UESP 17 // REG_SP
22033cab2bb3Spatrick # endif
22043cab2bb3Spatrick # endif
22053cab2bb3Spatrick *pc = ucontext->uc_mcontext.gregs[REG_EIP];
22063cab2bb3Spatrick *bp = ucontext->uc_mcontext.gregs[REG_EBP];
22071f9cb04fSpatrick *sp = ucontext->uc_mcontext.gregs[REG_UESP];
22083cab2bb3Spatrick # endif
22093cab2bb3Spatrick #elif defined(__powerpc__) || defined(__powerpc64__)
2210*68dd5bb1Srobert # if SANITIZER_FREEBSD
2211*68dd5bb1Srobert ucontext_t *ucontext = (ucontext_t *)context;
2212*68dd5bb1Srobert *pc = ucontext->uc_mcontext.mc_srr0;
2213*68dd5bb1Srobert *sp = ucontext->uc_mcontext.mc_frame[1];
2214*68dd5bb1Srobert *bp = ucontext->uc_mcontext.mc_frame[31];
2215*68dd5bb1Srobert # else
22163cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t*)context;
22173cab2bb3Spatrick *pc = ucontext->uc_mcontext.regs->nip;
22183cab2bb3Spatrick *sp = ucontext->uc_mcontext.regs->gpr[PT_R1];
22193cab2bb3Spatrick // The powerpc{,64}-linux ABIs do not specify r31 as the frame
22203cab2bb3Spatrick // pointer, but GCC always uses r31 when we need a frame pointer.
22213cab2bb3Spatrick *bp = ucontext->uc_mcontext.regs->gpr[PT_R31];
2222*68dd5bb1Srobert # endif
22233cab2bb3Spatrick #elif defined(__sparc__)
22243cab2bb3Spatrick #if defined(__arch64__) || defined(__sparcv9)
22253cab2bb3Spatrick #define STACK_BIAS 2047
22263cab2bb3Spatrick #else
22273cab2bb3Spatrick #define STACK_BIAS 0
22283cab2bb3Spatrick # endif
22293cab2bb3Spatrick # if SANITIZER_SOLARIS
22303cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t *)context;
22313cab2bb3Spatrick *pc = ucontext->uc_mcontext.gregs[REG_PC];
22323cab2bb3Spatrick *sp = ucontext->uc_mcontext.gregs[REG_O6] + STACK_BIAS;
22333cab2bb3Spatrick #else
22343cab2bb3Spatrick // Historical BSDism here.
22353cab2bb3Spatrick struct sigcontext *scontext = (struct sigcontext *)context;
22363cab2bb3Spatrick #if defined(__arch64__)
22373cab2bb3Spatrick *pc = scontext->sigc_regs.tpc;
22383cab2bb3Spatrick *sp = scontext->sigc_regs.u_regs[14] + STACK_BIAS;
22393cab2bb3Spatrick #else
22403cab2bb3Spatrick *pc = scontext->si_regs.pc;
22413cab2bb3Spatrick *sp = scontext->si_regs.u_regs[14];
22423cab2bb3Spatrick #endif
22433cab2bb3Spatrick # endif
22443cab2bb3Spatrick *bp = (uptr)((uhwptr *)*sp)[14] + STACK_BIAS;
22453cab2bb3Spatrick #elif defined(__mips__)
22463cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t*)context;
22473cab2bb3Spatrick *pc = ucontext->uc_mcontext.pc;
22483cab2bb3Spatrick *bp = ucontext->uc_mcontext.gregs[30];
22493cab2bb3Spatrick *sp = ucontext->uc_mcontext.gregs[29];
22503cab2bb3Spatrick #elif defined(__s390__)
22513cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t*)context;
22523cab2bb3Spatrick # if defined(__s390x__)
22533cab2bb3Spatrick *pc = ucontext->uc_mcontext.psw.addr;
22543cab2bb3Spatrick # else
22553cab2bb3Spatrick *pc = ucontext->uc_mcontext.psw.addr & 0x7fffffff;
22563cab2bb3Spatrick # endif
22573cab2bb3Spatrick *bp = ucontext->uc_mcontext.gregs[11];
22583cab2bb3Spatrick *sp = ucontext->uc_mcontext.gregs[15];
22593cab2bb3Spatrick #elif defined(__riscv)
22603cab2bb3Spatrick ucontext_t *ucontext = (ucontext_t*)context;
2261*68dd5bb1Srobert # if SANITIZER_FREEBSD
2262*68dd5bb1Srobert *pc = ucontext->uc_mcontext.mc_gpregs.gp_sepc;
2263*68dd5bb1Srobert *bp = ucontext->uc_mcontext.mc_gpregs.gp_s[0];
2264*68dd5bb1Srobert *sp = ucontext->uc_mcontext.mc_gpregs.gp_sp;
2265*68dd5bb1Srobert # else
22663cab2bb3Spatrick *pc = ucontext->uc_mcontext.__gregs[REG_PC];
22673cab2bb3Spatrick *bp = ucontext->uc_mcontext.__gregs[REG_S0];
22683cab2bb3Spatrick *sp = ucontext->uc_mcontext.__gregs[REG_SP];
2269*68dd5bb1Srobert # endif
2270*68dd5bb1Srobert # elif defined(__hexagon__)
2271*68dd5bb1Srobert ucontext_t *ucontext = (ucontext_t *)context;
2272*68dd5bb1Srobert *pc = ucontext->uc_mcontext.pc;
2273*68dd5bb1Srobert *bp = ucontext->uc_mcontext.r30;
2274*68dd5bb1Srobert *sp = ucontext->uc_mcontext.r29;
2275*68dd5bb1Srobert # elif defined(__loongarch__)
2276*68dd5bb1Srobert ucontext_t *ucontext = (ucontext_t *)context;
2277*68dd5bb1Srobert *pc = ucontext->uc_mcontext.__pc;
2278*68dd5bb1Srobert *bp = ucontext->uc_mcontext.__gregs[22];
2279*68dd5bb1Srobert *sp = ucontext->uc_mcontext.__gregs[3];
22803cab2bb3Spatrick # else
22813cab2bb3Spatrick # error "Unsupported arch"
22823cab2bb3Spatrick # endif
22833cab2bb3Spatrick }
22843cab2bb3Spatrick
InitPcSpBp()22853cab2bb3Spatrick void SignalContext::InitPcSpBp() { GetPcSpBp(context, &pc, &sp, &bp); }
22863cab2bb3Spatrick
InitializePlatformEarly()22873cab2bb3Spatrick void InitializePlatformEarly() {
22883cab2bb3Spatrick // Do nothing.
22893cab2bb3Spatrick }
22903cab2bb3Spatrick
CheckASLR()22913cab2bb3Spatrick void CheckASLR() {
22923cab2bb3Spatrick #if SANITIZER_NETBSD
22933cab2bb3Spatrick int mib[3];
22943cab2bb3Spatrick int paxflags;
22953cab2bb3Spatrick uptr len = sizeof(paxflags);
22963cab2bb3Spatrick
22973cab2bb3Spatrick mib[0] = CTL_PROC;
22983cab2bb3Spatrick mib[1] = internal_getpid();
22993cab2bb3Spatrick mib[2] = PROC_PID_PAXFLAGS;
23003cab2bb3Spatrick
23013cab2bb3Spatrick if (UNLIKELY(internal_sysctl(mib, 3, &paxflags, &len, NULL, 0) == -1)) {
23023cab2bb3Spatrick Printf("sysctl failed\n");
23033cab2bb3Spatrick Die();
23043cab2bb3Spatrick }
23053cab2bb3Spatrick
23063cab2bb3Spatrick if (UNLIKELY(paxflags & CTL_PROC_PAXFLAGS_ASLR)) {
23071f9cb04fSpatrick Printf("This sanitizer is not compatible with enabled ASLR.\n"
23081f9cb04fSpatrick "To disable ASLR, please run \"paxctl +a %s\" and try again.\n",
23091f9cb04fSpatrick GetArgv()[0]);
23103cab2bb3Spatrick Die();
23113cab2bb3Spatrick }
2312*68dd5bb1Srobert #elif SANITIZER_FREEBSD
2313*68dd5bb1Srobert int aslr_status;
2314*68dd5bb1Srobert int r = internal_procctl(P_PID, 0, PROC_ASLR_STATUS, &aslr_status);
2315*68dd5bb1Srobert if (UNLIKELY(r == -1)) {
2316*68dd5bb1Srobert // We're making things less 'dramatic' here since
2317*68dd5bb1Srobert // the cmd is not necessarily guaranteed to be here
2318*68dd5bb1Srobert // just yet regarding FreeBSD release
2319*68dd5bb1Srobert return;
2320*68dd5bb1Srobert }
2321*68dd5bb1Srobert if ((aslr_status & PROC_ASLR_ACTIVE) != 0) {
2322*68dd5bb1Srobert Printf("This sanitizer is not compatible with enabled ASLR "
2323*68dd5bb1Srobert "and binaries compiled with PIE\n");
2324*68dd5bb1Srobert Die();
2325*68dd5bb1Srobert }
23263cab2bb3Spatrick # elif SANITIZER_PPC64V2
23273cab2bb3Spatrick // Disable ASLR for Linux PPC64LE.
23283cab2bb3Spatrick int old_personality = personality(0xffffffff);
23293cab2bb3Spatrick if (old_personality != -1 && (old_personality & ADDR_NO_RANDOMIZE) == 0) {
2330*68dd5bb1Srobert VReport(1,
2331*68dd5bb1Srobert "WARNING: Program is being run with address space layout "
23323cab2bb3Spatrick "randomization (ASLR) enabled which prevents the thread and "
23333cab2bb3Spatrick "memory sanitizers from working on powerpc64le.\n"
23343cab2bb3Spatrick "ASLR will be disabled and the program re-executed.\n");
23353cab2bb3Spatrick CHECK_NE(personality(old_personality | ADDR_NO_RANDOMIZE), -1);
23363cab2bb3Spatrick ReExec();
23373cab2bb3Spatrick }
23383cab2bb3Spatrick # else
23393cab2bb3Spatrick // Do nothing
23403cab2bb3Spatrick # endif
23413cab2bb3Spatrick }
23423cab2bb3Spatrick
CheckMPROTECT()23433cab2bb3Spatrick void CheckMPROTECT() {
23443cab2bb3Spatrick #if SANITIZER_NETBSD
23453cab2bb3Spatrick int mib[3];
23463cab2bb3Spatrick int paxflags;
23473cab2bb3Spatrick uptr len = sizeof(paxflags);
23483cab2bb3Spatrick
23493cab2bb3Spatrick mib[0] = CTL_PROC;
23503cab2bb3Spatrick mib[1] = internal_getpid();
23513cab2bb3Spatrick mib[2] = PROC_PID_PAXFLAGS;
23523cab2bb3Spatrick
23533cab2bb3Spatrick if (UNLIKELY(internal_sysctl(mib, 3, &paxflags, &len, NULL, 0) == -1)) {
23543cab2bb3Spatrick Printf("sysctl failed\n");
23553cab2bb3Spatrick Die();
23563cab2bb3Spatrick }
23573cab2bb3Spatrick
23583cab2bb3Spatrick if (UNLIKELY(paxflags & CTL_PROC_PAXFLAGS_MPROTECT)) {
23593cab2bb3Spatrick Printf("This sanitizer is not compatible with enabled MPROTECT\n");
23603cab2bb3Spatrick Die();
23613cab2bb3Spatrick }
23623cab2bb3Spatrick #else
23633cab2bb3Spatrick // Do nothing
23643cab2bb3Spatrick #endif
23653cab2bb3Spatrick }
23663cab2bb3Spatrick
CheckNoDeepBind(const char * filename,int flag)23673cab2bb3Spatrick void CheckNoDeepBind(const char *filename, int flag) {
23683cab2bb3Spatrick #ifdef RTLD_DEEPBIND
23693cab2bb3Spatrick if (flag & RTLD_DEEPBIND) {
23703cab2bb3Spatrick Report(
23713cab2bb3Spatrick "You are trying to dlopen a %s shared library with RTLD_DEEPBIND flag"
23721f9cb04fSpatrick " which is incompatible with sanitizer runtime "
23733cab2bb3Spatrick "(see https://github.com/google/sanitizers/issues/611 for details"
23743cab2bb3Spatrick "). If you want to run %s library under sanitizers please remove "
23753cab2bb3Spatrick "RTLD_DEEPBIND from dlopen flags.\n",
23763cab2bb3Spatrick filename, filename);
23773cab2bb3Spatrick Die();
23783cab2bb3Spatrick }
23793cab2bb3Spatrick #endif
23803cab2bb3Spatrick }
23813cab2bb3Spatrick
FindAvailableMemoryRange(uptr size,uptr alignment,uptr left_padding,uptr * largest_gap_found,uptr * max_occupied_addr)23823cab2bb3Spatrick uptr FindAvailableMemoryRange(uptr size, uptr alignment, uptr left_padding,
23833cab2bb3Spatrick uptr *largest_gap_found,
23843cab2bb3Spatrick uptr *max_occupied_addr) {
23853cab2bb3Spatrick UNREACHABLE("FindAvailableMemoryRange is not available");
23863cab2bb3Spatrick return 0;
23873cab2bb3Spatrick }
23883cab2bb3Spatrick
GetRandom(void * buffer,uptr length,bool blocking)23893cab2bb3Spatrick bool GetRandom(void *buffer, uptr length, bool blocking) {
23903cab2bb3Spatrick if (!buffer || !length || length > 256)
23913cab2bb3Spatrick return false;
23923cab2bb3Spatrick #if SANITIZER_USE_GETENTROPY
23933cab2bb3Spatrick uptr rnd = getentropy(buffer, length);
23943cab2bb3Spatrick int rverrno = 0;
23953cab2bb3Spatrick if (internal_iserror(rnd, &rverrno) && rverrno == EFAULT)
23963cab2bb3Spatrick return false;
23973cab2bb3Spatrick else if (rnd == 0)
23983cab2bb3Spatrick return true;
23993cab2bb3Spatrick #endif // SANITIZER_USE_GETENTROPY
24003cab2bb3Spatrick
24013cab2bb3Spatrick #if SANITIZER_USE_GETRANDOM
24023cab2bb3Spatrick static atomic_uint8_t skip_getrandom_syscall;
24033cab2bb3Spatrick if (!atomic_load_relaxed(&skip_getrandom_syscall)) {
24043cab2bb3Spatrick // Up to 256 bytes, getrandom will not be interrupted.
24053cab2bb3Spatrick uptr res = internal_syscall(SYSCALL(getrandom), buffer, length,
24063cab2bb3Spatrick blocking ? 0 : GRND_NONBLOCK);
24073cab2bb3Spatrick int rverrno = 0;
24083cab2bb3Spatrick if (internal_iserror(res, &rverrno) && rverrno == ENOSYS)
24093cab2bb3Spatrick atomic_store_relaxed(&skip_getrandom_syscall, 1);
24103cab2bb3Spatrick else if (res == length)
24113cab2bb3Spatrick return true;
24123cab2bb3Spatrick }
24133cab2bb3Spatrick #endif // SANITIZER_USE_GETRANDOM
24143cab2bb3Spatrick // Up to 256 bytes, a read off /dev/urandom will not be interrupted.
24153cab2bb3Spatrick // blocking is moot here, O_NONBLOCK has no effect when opening /dev/urandom.
24163cab2bb3Spatrick uptr fd = internal_open("/dev/urandom", O_RDONLY);
24173cab2bb3Spatrick if (internal_iserror(fd))
24183cab2bb3Spatrick return false;
24193cab2bb3Spatrick uptr res = internal_read(fd, buffer, length);
24203cab2bb3Spatrick if (internal_iserror(res))
24213cab2bb3Spatrick return false;
24223cab2bb3Spatrick internal_close(fd);
24233cab2bb3Spatrick return true;
24243cab2bb3Spatrick }
24253cab2bb3Spatrick
24263cab2bb3Spatrick } // namespace __sanitizer
24273cab2bb3Spatrick
24283cab2bb3Spatrick #endif
2429