]> git.ipfire.org Git - thirdparty/gcc.git/blame - libsanitizer/sanitizer_common/sanitizer_linux.cpp
Correct a function pre/postcondition [PR102403].
[thirdparty/gcc.git] / libsanitizer / sanitizer_common / sanitizer_linux.cpp
CommitLineData
b667dd70 1//===-- sanitizer_linux.cpp -----------------------------------------------===//
f35db108 2//
b667dd70
ML
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
f35db108
WM
6//
7//===----------------------------------------------------------------------===//
8//
9// This file is shared between AddressSanitizer and ThreadSanitizer
10// run-time libraries and implements linux-specific functions from
11// sanitizer_libc.h.
12//===----------------------------------------------------------------------===//
ef1b3fda
KS
13
14#include "sanitizer_platform.h"
696d846a 15
eac97531 16#if SANITIZER_FREEBSD || SANITIZER_LINUX || SANITIZER_NETBSD || \
98f792ff 17 SANITIZER_SOLARIS
f35db108
WM
18
19#include "sanitizer_common.h"
dee5ea7a 20#include "sanitizer_flags.h"
eac97531 21#include "sanitizer_getauxval.h"
f35db108
WM
22#include "sanitizer_internal_defs.h"
23#include "sanitizer_libc.h"
ef1b3fda 24#include "sanitizer_linux.h"
a0408454 25#include "sanitizer_mutex.h"
f35db108
WM
26#include "sanitizer_placement_new.h"
27#include "sanitizer_procmaps.h"
28
3c6331c2 29#if SANITIZER_LINUX && !SANITIZER_GO
ef1b3fda 30#include <asm/param.h>
dee5ea7a
KS
31#endif
32
696d846a
MO
33// For mips64, syscall(__NR_stat) fills the buffer in the 'struct kernel_stat'
34// format. Struct kernel_stat is defined as 'struct stat' in asm/stat.h. To
35// access stat from asm/stat.h, without conflicting with definition in
36// sys/stat.h, we use this trick.
37#if defined(__mips64)
38#include <asm/unistd.h>
39#include <sys/types.h>
40#define stat kernel_stat
98f792ff
ML
41#if SANITIZER_GO
42#undef st_atime
43#undef st_mtime
44#undef st_ctime
45#define st_atime st_atim
46#define st_mtime st_mtim
47#define st_ctime st_ctim
48#endif
696d846a
MO
49#include <asm/stat.h>
50#undef stat
51#endif
52
ef1b3fda
KS
53#include <dlfcn.h>
54#include <errno.h>
f35db108 55#include <fcntl.h>
ef1b3fda 56#include <link.h>
f35db108
WM
57#include <pthread.h>
58#include <sched.h>
eac97531 59#include <signal.h>
f35db108 60#include <sys/mman.h>
eac97531
ML
61#include <sys/param.h>
62#if !SANITIZER_SOLARIS
ef1b3fda 63#include <sys/ptrace.h>
eac97531 64#endif
f35db108
WM
65#include <sys/resource.h>
66#include <sys/stat.h>
67#include <sys/syscall.h>
68#include <sys/time.h>
69#include <sys/types.h>
696d846a 70#include <ucontext.h>
f35db108 71#include <unistd.h>
ef1b3fda 72
5d3805fc
JJ
73#if SANITIZER_LINUX
74#include <sys/utsname.h>
75#endif
76
77#if SANITIZER_LINUX && !SANITIZER_ANDROID
78#include <sys/personality.h>
79#endif
80
dee5ea7a 81#if SANITIZER_FREEBSD
10189819 82#include <sys/exec.h>
866e32ad 83#include <sys/sysctl.h>
dee5ea7a
KS
84#include <machine/atomic.h>
85extern "C" {
86// <sys/umtx.h> must be included after <errno.h> and <sys/types.h> on
87// FreeBSD 9.2 and 10.0.
88#include <sys/umtx.h>
89}
eac97531 90#include <sys/thr.h>
dee5ea7a
KS
91#endif // SANITIZER_FREEBSD
92
5d3805fc
JJ
93#if SANITIZER_NETBSD
94#include <limits.h> // For NAME_MAX
95#include <sys/sysctl.h>
eac97531
ML
96#include <sys/exec.h>
97extern struct ps_strings *__ps_strings;
98#endif // SANITIZER_NETBSD
5d3805fc 99
eac97531
ML
100#if SANITIZER_SOLARIS
101#include <stdlib.h>
102#include <thread.h>
103#define environ _environ
5d3805fc
JJ
104#endif
105
eac97531 106extern char **environ;
5d3805fc 107
dee5ea7a 108#if SANITIZER_LINUX
ef1b3fda
KS
109// <linux/time.h>
110struct kernel_timeval {
111 long tv_sec;
112 long tv_usec;
113};
b4ab7d34
KS
114
115// <linux/futex.h> is broken on some linux distributions.
116const int FUTEX_WAIT = 0;
117const int FUTEX_WAKE = 1;
eac97531
ML
118const int FUTEX_PRIVATE_FLAG = 128;
119const int FUTEX_WAIT_PRIVATE = FUTEX_WAIT | FUTEX_PRIVATE_FLAG;
120const int FUTEX_WAKE_PRIVATE = FUTEX_WAKE | FUTEX_PRIVATE_FLAG;
dee5ea7a 121#endif // SANITIZER_LINUX
f35db108 122
dee5ea7a 123// Are we using 32-bit or 64-bit Linux syscalls?
e297eb60 124// x32 (which defines __x86_64__) has SANITIZER_WORDSIZE == 32
b014e124 125// but it still needs to use 64-bit syscalls.
eac97531
ML
126#if SANITIZER_LINUX && (defined(__x86_64__) || defined(__powerpc64__) || \
127 SANITIZER_WORDSIZE == 64)
b014e124
KS
128# define SANITIZER_LINUX_USES_64BIT_SYSCALLS 1
129#else
130# define SANITIZER_LINUX_USES_64BIT_SYSCALLS 0
131#endif
132
eac97531 133// Note : FreeBSD had implemented both
98f792ff 134// Linux apis, available from
eac97531 135// future 12.x version most likely
5d3805fc
JJ
136#if SANITIZER_LINUX && defined(__NR_getrandom)
137# if !defined(GRND_NONBLOCK)
138# define GRND_NONBLOCK 1
139# endif
140# define SANITIZER_USE_GETRANDOM 1
141#else
142# define SANITIZER_USE_GETRANDOM 0
143#endif // SANITIZER_LINUX && defined(__NR_getrandom)
144
98f792ff
ML
145#if SANITIZER_FREEBSD && __FreeBSD_version >= 1200000
146# define SANITIZER_USE_GETENTROPY 1
eac97531 147#else
98f792ff
ML
148# define SANITIZER_USE_GETENTROPY 0
149#endif
eac97531 150
f35db108
WM
151namespace __sanitizer {
152
dee5ea7a 153#if SANITIZER_LINUX && defined(__x86_64__)
ef1b3fda 154#include "sanitizer_syscall_linux_x86_64.inc"
0b997f6e
ML
155#elif SANITIZER_LINUX && SANITIZER_RISCV64
156#include "sanitizer_syscall_linux_riscv64.inc"
696d846a
MO
157#elif SANITIZER_LINUX && defined(__aarch64__)
158#include "sanitizer_syscall_linux_aarch64.inc"
eac97531
ML
159#elif SANITIZER_LINUX && defined(__arm__)
160#include "sanitizer_syscall_linux_arm.inc"
ef1b3fda
KS
161#else
162#include "sanitizer_syscall_generic.inc"
163#endif
164
f35db108 165// --------------- sanitizer_libc.h
eac97531 166#if !SANITIZER_SOLARIS && !SANITIZER_NETBSD
98f792ff 167#if !SANITIZER_S390
696d846a 168uptr internal_mmap(void *addr, uptr length, int prot, int flags, int fd,
3c6331c2 169 u64 offset) {
eac97531 170#if SANITIZER_FREEBSD || SANITIZER_LINUX_USES_64BIT_SYSCALLS
dee5ea7a 171 return internal_syscall(SYSCALL(mmap), (uptr)addr, length, prot, flags, fd,
df77f0e4 172 offset);
f35db108 173#else
696d846a
MO
174 // mmap2 specifies file offset in 4096-byte units.
175 CHECK(IsAligned(offset, 4096));
dee5ea7a 176 return internal_syscall(SYSCALL(mmap2), addr, length, prot, flags, fd,
696d846a 177 offset / 4096);
f35db108
WM
178#endif
179}
98f792ff 180#endif // !SANITIZER_S390
f35db108 181
ef1b3fda 182uptr internal_munmap(void *addr, uptr length) {
dee5ea7a 183 return internal_syscall(SYSCALL(munmap), (uptr)addr, length);
f35db108
WM
184}
185
d0fee87e
ML
186#if SANITIZER_LINUX
187uptr internal_mremap(void *old_address, uptr old_size, uptr new_size, int flags,
188 void *new_address) {
189 return internal_syscall(SYSCALL(mremap), (uptr)old_address, old_size,
190 new_size, flags, (uptr)new_address);
191}
192#endif
193
696d846a
MO
194int internal_mprotect(void *addr, uptr length, int prot) {
195 return internal_syscall(SYSCALL(mprotect), (uptr)addr, length, prot);
196}
0b997f6e
ML
197
198int internal_madvise(uptr addr, uptr length, int advice) {
199 return internal_syscall(SYSCALL(madvise), addr, length, advice);
200}
696d846a 201
ef1b3fda 202uptr internal_close(fd_t fd) {
dee5ea7a 203 return internal_syscall(SYSCALL(close), fd);
f35db108
WM
204}
205
ef1b3fda 206uptr internal_open(const char *filename, int flags) {
dee5ea7a
KS
207#if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
208 return internal_syscall(SYSCALL(openat), AT_FDCWD, (uptr)filename, flags);
209#else
210 return internal_syscall(SYSCALL(open), (uptr)filename, flags);
211#endif
b4ab7d34
KS
212}
213
ef1b3fda 214uptr internal_open(const char *filename, int flags, u32 mode) {
dee5ea7a
KS
215#if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
216 return internal_syscall(SYSCALL(openat), AT_FDCWD, (uptr)filename, flags,
217 mode);
218#else
219 return internal_syscall(SYSCALL(open), (uptr)filename, flags, mode);
220#endif
b4ab7d34
KS
221}
222
f35db108
WM
223uptr internal_read(fd_t fd, void *buf, uptr count) {
224 sptr res;
eac97531
ML
225 HANDLE_EINTR(res,
226 (sptr)internal_syscall(SYSCALL(read), fd, (uptr)buf, count));
f35db108
WM
227 return res;
228}
229
230uptr internal_write(fd_t fd, const void *buf, uptr count) {
231 sptr res;
eac97531
ML
232 HANDLE_EINTR(res,
233 (sptr)internal_syscall(SYSCALL(write), fd, (uptr)buf, count));
f35db108
WM
234 return res;
235}
236
866e32ad
KS
237uptr internal_ftruncate(fd_t fd, uptr size) {
238 sptr res;
696d846a
MO
239 HANDLE_EINTR(res, (sptr)internal_syscall(SYSCALL(ftruncate), fd,
240 (OFF_T)size));
866e32ad
KS
241 return res;
242}
243
5d3805fc 244#if !SANITIZER_LINUX_USES_64BIT_SYSCALLS && SANITIZER_LINUX
ef1b3fda
KS
245static void stat64_to_stat(struct stat64 *in, struct stat *out) {
246 internal_memset(out, 0, sizeof(*out));
247 out->st_dev = in->st_dev;
248 out->st_ino = in->st_ino;
249 out->st_mode = in->st_mode;
250 out->st_nlink = in->st_nlink;
251 out->st_uid = in->st_uid;
252 out->st_gid = in->st_gid;
253 out->st_rdev = in->st_rdev;
254 out->st_size = in->st_size;
255 out->st_blksize = in->st_blksize;
256 out->st_blocks = in->st_blocks;
257 out->st_atime = in->st_atime;
258 out->st_mtime = in->st_mtime;
259 out->st_ctime = in->st_ctime;
ef1b3fda
KS
260}
261#endif
262
696d846a 263#if defined(__mips64)
5d3805fc
JJ
264// Undefine compatibility macros from <sys/stat.h>
265// so that they would not clash with the kernel_stat
266// st_[a|m|c]time fields
98f792ff 267#if !SANITIZER_GO
5d3805fc
JJ
268#undef st_atime
269#undef st_mtime
270#undef st_ctime
98f792ff 271#endif
5d3805fc
JJ
272#if defined(SANITIZER_ANDROID)
273// Bionic sys/stat.h defines additional macros
274// for compatibility with the old NDKs and
275// they clash with the kernel_stat structure
276// st_[a|m|c]time_nsec fields.
277#undef st_atime_nsec
278#undef st_mtime_nsec
279#undef st_ctime_nsec
280#endif
696d846a
MO
281static void kernel_stat_to_stat(struct kernel_stat *in, struct stat *out) {
282 internal_memset(out, 0, sizeof(*out));
283 out->st_dev = in->st_dev;
284 out->st_ino = in->st_ino;
285 out->st_mode = in->st_mode;
286 out->st_nlink = in->st_nlink;
287 out->st_uid = in->st_uid;
288 out->st_gid = in->st_gid;
289 out->st_rdev = in->st_rdev;
290 out->st_size = in->st_size;
291 out->st_blksize = in->st_blksize;
292 out->st_blocks = in->st_blocks;
5d3805fc
JJ
293#if defined(__USE_MISC) || \
294 defined(__USE_XOPEN2K8) || \
295 defined(SANITIZER_ANDROID)
296 out->st_atim.tv_sec = in->st_atime;
297 out->st_atim.tv_nsec = in->st_atime_nsec;
298 out->st_mtim.tv_sec = in->st_mtime;
299 out->st_mtim.tv_nsec = in->st_mtime_nsec;
300 out->st_ctim.tv_sec = in->st_ctime;
301 out->st_ctim.tv_nsec = in->st_ctime_nsec;
302#else
303 out->st_atime = in->st_atime;
304 out->st_atimensec = in->st_atime_nsec;
305 out->st_mtime = in->st_mtime;
306 out->st_mtimensec = in->st_mtime_nsec;
307 out->st_ctime = in->st_ctime;
308 out->st_atimensec = in->st_ctime_nsec;
309#endif
696d846a
MO
310}
311#endif
312
ef1b3fda 313uptr internal_stat(const char *path, void *buf) {
98f792ff 314#if SANITIZER_FREEBSD
eac97531 315 return internal_syscall(SYSCALL(fstatat), AT_FDCWD, (uptr)path, (uptr)buf, 0);
dee5ea7a 316#elif SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
eac97531
ML
317 return internal_syscall(SYSCALL(newfstatat), AT_FDCWD, (uptr)path, (uptr)buf,
318 0);
dee5ea7a 319#elif SANITIZER_LINUX_USES_64BIT_SYSCALLS
696d846a
MO
320# if defined(__mips64)
321 // For mips64, stat syscall fills buffer in the format of kernel_stat
322 struct kernel_stat kbuf;
323 int res = internal_syscall(SYSCALL(stat), path, &kbuf);
324 kernel_stat_to_stat(&kbuf, (struct stat *)buf);
325 return res;
326# else
dee5ea7a 327 return internal_syscall(SYSCALL(stat), (uptr)path, (uptr)buf);
696d846a 328# endif
b4ab7d34 329#else
ef1b3fda 330 struct stat64 buf64;
dee5ea7a 331 int res = internal_syscall(SYSCALL(stat64), path, &buf64);
ef1b3fda
KS
332 stat64_to_stat(&buf64, (struct stat *)buf);
333 return res;
b4ab7d34
KS
334#endif
335}
336
ef1b3fda 337uptr internal_lstat(const char *path, void *buf) {
98f792ff 338#if SANITIZER_FREEBSD
eac97531
ML
339 return internal_syscall(SYSCALL(fstatat), AT_FDCWD, (uptr)path, (uptr)buf,
340 AT_SYMLINK_NOFOLLOW);
dee5ea7a 341#elif SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
eac97531
ML
342 return internal_syscall(SYSCALL(newfstatat), AT_FDCWD, (uptr)path, (uptr)buf,
343 AT_SYMLINK_NOFOLLOW);
dee5ea7a 344#elif SANITIZER_LINUX_USES_64BIT_SYSCALLS
10189819
MO
345# if SANITIZER_MIPS64
346 // For mips64, lstat syscall fills buffer in the format of kernel_stat
347 struct kernel_stat kbuf;
348 int res = internal_syscall(SYSCALL(lstat), path, &kbuf);
349 kernel_stat_to_stat(&kbuf, (struct stat *)buf);
350 return res;
351# else
dee5ea7a 352 return internal_syscall(SYSCALL(lstat), (uptr)path, (uptr)buf);
10189819 353# endif
b4ab7d34 354#else
ef1b3fda 355 struct stat64 buf64;
dee5ea7a 356 int res = internal_syscall(SYSCALL(lstat64), path, &buf64);
ef1b3fda
KS
357 stat64_to_stat(&buf64, (struct stat *)buf);
358 return res;
b4ab7d34
KS
359#endif
360}
361
ef1b3fda 362uptr internal_fstat(fd_t fd, void *buf) {
98f792ff
ML
363#if SANITIZER_FREEBSD || SANITIZER_LINUX_USES_64BIT_SYSCALLS
364#if SANITIZER_MIPS64
10189819
MO
365 // For mips64, fstat syscall fills buffer in the format of kernel_stat
366 struct kernel_stat kbuf;
367 int res = internal_syscall(SYSCALL(fstat), fd, &kbuf);
368 kernel_stat_to_stat(&kbuf, (struct stat *)buf);
369 return res;
370# else
dee5ea7a 371 return internal_syscall(SYSCALL(fstat), fd, (uptr)buf);
10189819 372# endif
b4ab7d34 373#else
ef1b3fda 374 struct stat64 buf64;
dee5ea7a 375 int res = internal_syscall(SYSCALL(fstat64), fd, &buf64);
ef1b3fda
KS
376 stat64_to_stat(&buf64, (struct stat *)buf);
377 return res;
b4ab7d34
KS
378#endif
379}
380
f35db108 381uptr internal_filesize(fd_t fd) {
f35db108 382 struct stat st;
b4ab7d34
KS
383 if (internal_fstat(fd, &st))
384 return -1;
f35db108
WM
385 return (uptr)st.st_size;
386}
387
b667dd70
ML
388uptr internal_dup(int oldfd) {
389 return internal_syscall(SYSCALL(dup), oldfd);
390}
391
ef1b3fda 392uptr internal_dup2(int oldfd, int newfd) {
dee5ea7a
KS
393#if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
394 return internal_syscall(SYSCALL(dup3), oldfd, newfd, 0);
395#else
396 return internal_syscall(SYSCALL(dup2), oldfd, newfd);
397#endif
f35db108
WM
398}
399
400uptr internal_readlink(const char *path, char *buf, uptr bufsize) {
eac97531
ML
401#if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
402 return internal_syscall(SYSCALL(readlinkat), AT_FDCWD, (uptr)path, (uptr)buf,
403 bufsize);
dee5ea7a 404#else
4dfc5b35 405 return internal_syscall(SYSCALL(readlink), (uptr)path, (uptr)buf, bufsize);
dee5ea7a 406#endif
ef1b3fda
KS
407}
408
409uptr internal_unlink(const char *path) {
98f792ff 410#if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
dee5ea7a
KS
411 return internal_syscall(SYSCALL(unlinkat), AT_FDCWD, (uptr)path, 0);
412#else
413 return internal_syscall(SYSCALL(unlink), (uptr)path);
414#endif
f35db108
WM
415}
416
866e32ad 417uptr internal_rename(const char *oldpath, const char *newpath) {
cb7dc4da
ML
418#if defined(__riscv)
419 return internal_syscall(SYSCALL(renameat2), AT_FDCWD, (uptr)oldpath, AT_FDCWD,
420 (uptr)newpath, 0);
98f792ff 421#elif SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
866e32ad
KS
422 return internal_syscall(SYSCALL(renameat), AT_FDCWD, (uptr)oldpath, AT_FDCWD,
423 (uptr)newpath);
424#else
425 return internal_syscall(SYSCALL(rename), (uptr)oldpath, (uptr)newpath);
426#endif
427}
428
ef1b3fda 429uptr internal_sched_yield() {
dee5ea7a 430 return internal_syscall(SYSCALL(sched_yield));
f35db108
WM
431}
432
90e46074 433void internal_usleep(u64 useconds) {
10189819 434 struct timespec ts;
90e46074
L
435 ts.tv_sec = useconds / 1000000;
436 ts.tv_nsec = (useconds % 1000000) * 1000;
437 internal_syscall(SYSCALL(nanosleep), &ts, &ts);
10189819
MO
438}
439
ef1b3fda
KS
440uptr internal_execve(const char *filename, char *const argv[],
441 char *const envp[]) {
dee5ea7a
KS
442 return internal_syscall(SYSCALL(execve), (uptr)filename, (uptr)argv,
443 (uptr)envp);
ef1b3fda 444}
eac97531 445#endif // !SANITIZER_SOLARIS && !SANITIZER_NETBSD
ef1b3fda 446
0b997f6e
ML
447#if !SANITIZER_NETBSD
448void internal__exit(int exitcode) {
98f792ff 449#if SANITIZER_FREEBSD || SANITIZER_SOLARIS
0b997f6e
ML
450 internal_syscall(SYSCALL(exit), exitcode);
451#else
452 internal_syscall(SYSCALL(exit_group), exitcode);
453#endif
454 Die(); // Unreachable.
455}
456#endif // !SANITIZER_NETBSD
457
f35db108 458// ----------------- sanitizer_common.h
e297eb60 459bool FileExists(const char *filename) {
b667dd70
ML
460 if (ShouldMockFailureToOpen(filename))
461 return false;
dee5ea7a 462 struct stat st;
7697b16f 463#if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
dee5ea7a 464 if (internal_syscall(SYSCALL(newfstatat), AT_FDCWD, filename, &st, 0))
dee5ea7a 465#else
ef1b3fda 466 if (internal_stat(filename, &st))
7697b16f 467#endif
e297eb60 468 return false;
e297eb60
KS
469 // Sanity check: filename is a regular file.
470 return S_ISREG(st.st_mode);
471}
472
eac97531 473#if !SANITIZER_NETBSD
5d3805fc 474tid_t GetTid() {
dee5ea7a 475#if SANITIZER_FREEBSD
eac97531
ML
476 long Tid;
477 thr_self(&Tid);
478 return Tid;
eac97531
ML
479#elif SANITIZER_SOLARIS
480 return thr_self();
dee5ea7a
KS
481#else
482 return internal_syscall(SYSCALL(gettid));
483#endif
ef1b3fda 484}
f35db108 485
eac97531
ML
486int TgKill(pid_t pid, tid_t tid, int sig) {
487#if SANITIZER_LINUX
488 return internal_syscall(SYSCALL(tgkill), pid, tid, sig);
489#elif SANITIZER_FREEBSD
490 return internal_syscall(SYSCALL(thr_kill2), pid, tid, sig);
eac97531
ML
491#elif SANITIZER_SOLARIS
492 (void)pid;
493 return thr_kill(tid, sig);
494#endif
495}
496#endif
497
d0fee87e 498#if SANITIZER_GLIBC
ef1b3fda 499u64 NanoTime() {
df77f0e4
KS
500 kernel_timeval tv;
501 internal_memset(&tv, 0, sizeof(tv));
eac97531 502 internal_syscall(SYSCALL(gettimeofday), &tv, 0);
d0fee87e 503 return (u64)tv.tv_sec * 1000 * 1000 * 1000 + tv.tv_usec * 1000;
f35db108 504}
d0fee87e 505// Used by real_clock_gettime.
eac97531
ML
506uptr internal_clock_gettime(__sanitizer_clockid_t clk_id, void *tp) {
507 return internal_syscall(SYSCALL(clock_gettime), clk_id, tp);
508}
d0fee87e
ML
509#elif !SANITIZER_SOLARIS && !SANITIZER_NETBSD
510u64 NanoTime() {
511 struct timespec ts;
512 clock_gettime(CLOCK_REALTIME, &ts);
513 return (u64)ts.tv_sec * 1000 * 1000 * 1000 + ts.tv_nsec;
514}
515#endif
eac97531 516
866e32ad 517// Like getenv, but reads env directly from /proc (on Linux) or parses the
eac97531
ML
518// 'environ' array (on some others) and does not use libc. This function
519// should be called first inside __asan_init.
f35db108 520const char *GetEnv(const char *name) {
98f792ff 521#if SANITIZER_FREEBSD || SANITIZER_NETBSD || SANITIZER_SOLARIS
866e32ad
KS
522 if (::environ != 0) {
523 uptr NameLen = internal_strlen(name);
524 for (char **Env = ::environ; *Env != 0; Env++) {
525 if (internal_strncmp(*Env, name, NameLen) == 0 && (*Env)[NameLen] == '=')
526 return (*Env) + NameLen + 1;
527 }
528 }
529 return 0; // Not found.
530#elif SANITIZER_LINUX
f35db108
WM
531 static char *environ;
532 static uptr len;
533 static bool inited;
534 if (!inited) {
535 inited = true;
536 uptr environ_size;
696d846a
MO
537 if (!ReadFileToBuffer("/proc/self/environ", &environ, &environ_size, &len))
538 environ = nullptr;
f35db108 539 }
696d846a 540 if (!environ || len == 0) return nullptr;
f35db108
WM
541 uptr namelen = internal_strlen(name);
542 const char *p = environ;
543 while (*p != '\0') { // will happen at the \0\0 that terminates the buffer
544 // proc file has the format NAME=value\0NAME=value\0NAME=value\0...
545 const char* endp =
546 (char*)internal_memchr(p, '\0', len - (p - environ));
696d846a
MO
547 if (!endp) // this entry isn't NUL terminated
548 return nullptr;
f35db108
WM
549 else if (!internal_memcmp(p, name, namelen) && p[namelen] == '=') // Match.
550 return p + namelen + 1; // point after =
551 p = endp + 1;
552 }
696d846a 553 return nullptr; // Not found.
866e32ad
KS
554#else
555#error "Unsupported platform"
556#endif
f35db108
WM
557}
558
98f792ff 559#if !SANITIZER_FREEBSD && !SANITIZER_NETBSD && !SANITIZER_GO
7df59255 560extern "C" {
eac97531 561SANITIZER_WEAK_ATTRIBUTE extern void *__libc_stack_end;
7df59255 562}
10189819 563#endif
7df59255 564
98f792ff 565#if !SANITIZER_FREEBSD && !SANITIZER_NETBSD
2660d12d
KS
566static void ReadNullSepFileToArray(const char *path, char ***arr,
567 int arr_size) {
568 char *buff;
696d846a
MO
569 uptr buff_size;
570 uptr buff_len;
2660d12d 571 *arr = (char **)MmapOrDie(arr_size * sizeof(char *), "NullSepFileArray");
696d846a
MO
572 if (!ReadFileToBuffer(path, &buff, &buff_size, &buff_len, 1024 * 1024)) {
573 (*arr)[0] = nullptr;
574 return;
575 }
2660d12d
KS
576 (*arr)[0] = buff;
577 int count, i;
578 for (count = 1, i = 1; ; i++) {
f35db108
WM
579 if (buff[i] == 0) {
580 if (buff[i+1] == 0) break;
2660d12d
KS
581 (*arr)[count] = &buff[i+1];
582 CHECK_LE(count, arr_size - 1); // FIXME: make this more flexible.
583 count++;
f35db108
WM
584 }
585 }
696d846a 586 (*arr)[count] = nullptr;
2660d12d 587}
ef1b3fda 588#endif
2660d12d 589
10189819 590static void GetArgsAndEnv(char ***argv, char ***envp) {
eac97531
ML
591#if SANITIZER_FREEBSD
592 // On FreeBSD, retrieving the argument and environment arrays is done via the
593 // kern.ps_strings sysctl, which returns a pointer to a structure containing
594 // this information. See also <sys/exec.h>.
595 ps_strings *pss;
596 uptr sz = sizeof(pss);
597 if (internal_sysctlbyname("kern.ps_strings", &pss, &sz, NULL, 0) == -1) {
598 Printf("sysctl kern.ps_strings failed\n");
599 Die();
600 }
601 *argv = pss->ps_argvstr;
602 *envp = pss->ps_envstr;
603#elif SANITIZER_NETBSD
604 *argv = __ps_strings->ps_argvstr;
605 *envp = __ps_strings->ps_envstr;
606#else // SANITIZER_FREEBSD
ef1b3fda
KS
607#if !SANITIZER_GO
608 if (&__libc_stack_end) {
ef1b3fda 609 uptr* stack_end = (uptr*)__libc_stack_end;
3c6331c2
ML
610 // Normally argc can be obtained from *stack_end, however, on ARM glibc's
611 // _start clobbers it:
612 // https://sourceware.org/git/?p=glibc.git;a=blob;f=sysdeps/arm/start.S;hb=refs/heads/release/2.31/master#l75
613 // Do not special-case ARM and infer argc from argv everywhere.
614 int argc = 0;
615 while (stack_end[argc + 1]) argc++;
ef1b3fda
KS
616 *argv = (char**)(stack_end + 1);
617 *envp = (char**)(stack_end + argc + 2);
ef1b3fda 618 } else {
3c6331c2 619#endif // !SANITIZER_GO
ef1b3fda
KS
620 static const int kMaxArgv = 2000, kMaxEnvp = 2000;
621 ReadNullSepFileToArray("/proc/self/cmdline", argv, kMaxArgv);
622 ReadNullSepFileToArray("/proc/self/environ", envp, kMaxEnvp);
3c6331c2 623#if !SANITIZER_GO
ef1b3fda 624 }
eac97531
ML
625#endif // !SANITIZER_GO
626#endif // SANITIZER_FREEBSD
10189819
MO
627}
628
629char **GetArgv() {
630 char **argv, **envp;
631 GetArgsAndEnv(&argv, &envp);
632 return argv;
7df59255
KS
633}
634
b667dd70 635char **GetEnviron() {
2660d12d 636 char **argv, **envp;
7df59255 637 GetArgsAndEnv(&argv, &envp);
b667dd70 638 return envp;
f35db108 639}
b667dd70 640
eac97531 641#if !SANITIZER_SOLARIS
90e46074
L
642void FutexWait(atomic_uint32_t *p, u32 cmp) {
643# if SANITIZER_FREEBSD
644 _umtx_op(p, UMTX_OP_WAIT_UINT, cmp, 0, 0);
645# elif SANITIZER_NETBSD
646 sched_yield(); /* No userspace futex-like synchronization */
647# else
648 internal_syscall(SYSCALL(futex), (uptr)p, FUTEX_WAIT_PRIVATE, cmp, 0, 0, 0);
649# endif
650}
651
652void FutexWake(atomic_uint32_t *p, u32 count) {
653# if SANITIZER_FREEBSD
654 _umtx_op(p, UMTX_OP_WAKE, count, 0, 0);
655# elif SANITIZER_NETBSD
656 /* No userspace futex-like synchronization */
657# else
658 internal_syscall(SYSCALL(futex), (uptr)p, FUTEX_WAKE_PRIVATE, count, 0, 0, 0);
659# endif
660}
661
662enum { MtxUnlocked = 0, MtxLocked = 1, MtxSleeping = 2 };
2660d12d 663
ef1b3fda
KS
664BlockingMutex::BlockingMutex() {
665 internal_memset(this, 0, sizeof(*this));
666}
667
2660d12d 668void BlockingMutex::Lock() {
696d846a 669 CHECK_EQ(owner_, 0);
2660d12d
KS
670 atomic_uint32_t *m = reinterpret_cast<atomic_uint32_t *>(&opaque_storage_);
671 if (atomic_exchange(m, MtxLocked, memory_order_acquire) == MtxUnlocked)
672 return;
dee5ea7a
KS
673 while (atomic_exchange(m, MtxSleeping, memory_order_acquire) != MtxUnlocked) {
674#if SANITIZER_FREEBSD
675 _umtx_op(m, UMTX_OP_WAIT_UINT, MtxSleeping, 0, 0);
5d3805fc 676#elif SANITIZER_NETBSD
eac97531 677 sched_yield(); /* No userspace futex-like synchronization */
dee5ea7a 678#else
eac97531
ML
679 internal_syscall(SYSCALL(futex), (uptr)m, FUTEX_WAIT_PRIVATE, MtxSleeping,
680 0, 0, 0);
dee5ea7a
KS
681#endif
682 }
2660d12d
KS
683}
684
685void BlockingMutex::Unlock() {
686 atomic_uint32_t *m = reinterpret_cast<atomic_uint32_t *>(&opaque_storage_);
5d3805fc 687 u32 v = atomic_exchange(m, MtxUnlocked, memory_order_release);
2660d12d 688 CHECK_NE(v, MtxUnlocked);
dee5ea7a
KS
689 if (v == MtxSleeping) {
690#if SANITIZER_FREEBSD
691 _umtx_op(m, UMTX_OP_WAKE, 1, 0, 0);
5d3805fc 692#elif SANITIZER_NETBSD
eac97531 693 /* No userspace futex-like synchronization */
dee5ea7a 694#else
eac97531 695 internal_syscall(SYSCALL(futex), (uptr)m, FUTEX_WAKE_PRIVATE, 1, 0, 0, 0);
dee5ea7a
KS
696#endif
697 }
ef1b3fda
KS
698}
699
90e46074
L
700void BlockingMutex::CheckLocked() const {
701 auto m = reinterpret_cast<atomic_uint32_t const *>(&opaque_storage_);
ef1b3fda
KS
702 CHECK_NE(MtxUnlocked, atomic_load(m, memory_order_relaxed));
703}
90e46074 704# endif // !SANITIZER_SOLARIS
ef1b3fda
KS
705
706// ----------------- sanitizer_linux.h
707// The actual size of this structure is specified by d_reclen.
708// Note that getdents64 uses a different structure format. We only provide the
709// 32-bit syscall here.
5d3805fc 710#if SANITIZER_NETBSD
eac97531 711// Not used
5d3805fc 712#else
ef1b3fda 713struct linux_dirent {
0b997f6e 714#if SANITIZER_X32 || defined(__aarch64__) || SANITIZER_RISCV64
dee5ea7a
KS
715 u64 d_ino;
716 u64 d_off;
717#else
ef1b3fda
KS
718 unsigned long d_ino;
719 unsigned long d_off;
dee5ea7a 720#endif
ef1b3fda 721 unsigned short d_reclen;
0b997f6e 722#if defined(__aarch64__) || SANITIZER_RISCV64
55aea9f5
MO
723 unsigned char d_type;
724#endif
ef1b3fda
KS
725 char d_name[256];
726};
5d3805fc 727#endif
ef1b3fda 728
eac97531 729#if !SANITIZER_SOLARIS && !SANITIZER_NETBSD
ef1b3fda
KS
730// Syscall wrappers.
731uptr internal_ptrace(int request, int pid, void *addr, void *data) {
dee5ea7a
KS
732 return internal_syscall(SYSCALL(ptrace), request, pid, (uptr)addr,
733 (uptr)data);
ef1b3fda
KS
734}
735
736uptr internal_waitpid(int pid, int *status, int options) {
dee5ea7a 737 return internal_syscall(SYSCALL(wait4), pid, (uptr)status, options,
df77f0e4 738 0 /* rusage */);
ef1b3fda
KS
739}
740
741uptr internal_getpid() {
dee5ea7a 742 return internal_syscall(SYSCALL(getpid));
ef1b3fda
KS
743}
744
745uptr internal_getppid() {
dee5ea7a 746 return internal_syscall(SYSCALL(getppid));
ef1b3fda
KS
747}
748
3c6331c2
ML
749int internal_dlinfo(void *handle, int request, void *p) {
750#if SANITIZER_FREEBSD
751 return dlinfo(handle, request, p);
752#else
753 UNIMPLEMENTED();
754#endif
755}
756
ef1b3fda 757uptr internal_getdents(fd_t fd, struct linux_dirent *dirp, unsigned int count) {
eac97531 758#if SANITIZER_FREEBSD
5d3805fc
JJ
759 return internal_syscall(SYSCALL(getdirentries), fd, (uptr)dirp, count, NULL);
760#elif SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
dee5ea7a
KS
761 return internal_syscall(SYSCALL(getdents64), fd, (uptr)dirp, count);
762#else
763 return internal_syscall(SYSCALL(getdents), fd, (uptr)dirp, count);
764#endif
ef1b3fda
KS
765}
766
767uptr internal_lseek(fd_t fd, OFF_T offset, int whence) {
dee5ea7a 768 return internal_syscall(SYSCALL(lseek), fd, offset, whence);
ef1b3fda
KS
769}
770
dee5ea7a 771#if SANITIZER_LINUX
ef1b3fda 772uptr internal_prctl(int option, uptr arg2, uptr arg3, uptr arg4, uptr arg5) {
dee5ea7a 773 return internal_syscall(SYSCALL(prctl), option, arg2, arg3, arg4, arg5);
ef1b3fda 774}
dee5ea7a 775#endif
ef1b3fda 776
144e36a7 777uptr internal_sigaltstack(const void *ss, void *oss) {
dee5ea7a 778 return internal_syscall(SYSCALL(sigaltstack), (uptr)ss, (uptr)oss);
ef1b3fda
KS
779}
780
dee5ea7a 781int internal_fork() {
866e32ad
KS
782#if SANITIZER_USES_CANONICAL_LINUX_SYSCALLS
783 return internal_syscall(SYSCALL(clone), SIGCHLD, 0);
784#else
dee5ea7a 785 return internal_syscall(SYSCALL(fork));
866e32ad 786#endif
df77f0e4
KS
787}
788
98f792ff 789#if SANITIZER_FREEBSD
eac97531
ML
790int internal_sysctl(const int *name, unsigned int namelen, void *oldp,
791 uptr *oldlenp, const void *newp, uptr newlen) {
b667dd70
ML
792 return internal_syscall(SYSCALL(__sysctl), name, namelen, oldp,
793 (size_t *)oldlenp, newp, (size_t)newlen);
eac97531
ML
794}
795
eac97531
ML
796int internal_sysctlbyname(const char *sname, void *oldp, uptr *oldlenp,
797 const void *newp, uptr newlen) {
0b997f6e
ML
798 // Note: this function can be called during startup, so we need to avoid
799 // calling any interceptable functions. On FreeBSD >= 1300045 sysctlbyname()
800 // is a real syscall, but for older versions it calls sysctlnametomib()
801 // followed by sysctl(). To avoid calling the intercepted version and
802 // asserting if this happens during startup, call the real sysctlnametomib()
803 // followed by internal_sysctl() if the syscall is not available.
804#ifdef SYS___sysctlbyname
805 return internal_syscall(SYSCALL(__sysctlbyname), sname,
806 internal_strlen(sname), oldp, (size_t *)oldlenp, newp,
807 (size_t)newlen);
808#else
809 static decltype(sysctlnametomib) *real_sysctlnametomib = nullptr;
810 if (!real_sysctlnametomib)
811 real_sysctlnametomib =
812 (decltype(sysctlnametomib) *)dlsym(RTLD_NEXT, "sysctlnametomib");
813 CHECK(real_sysctlnametomib);
814
815 int oid[CTL_MAXNAME];
816 size_t len = CTL_MAXNAME;
817 if (real_sysctlnametomib(sname, oid, &len) == -1)
818 return (-1);
819 return internal_sysctl(oid, len, oldp, oldlenp, newp, newlen);
820#endif
eac97531
ML
821}
822#endif
eac97531 823
dee5ea7a 824#if SANITIZER_LINUX
696d846a 825#define SA_RESTORER 0x04000000
10189819
MO
826// Doesn't set sa_restorer if the caller did not set it, so use with caution
827//(see below).
dee5ea7a
KS
828int internal_sigaction_norestorer(int signum, const void *act, void *oldact) {
829 __sanitizer_kernel_sigaction_t k_act, k_oldact;
830 internal_memset(&k_act, 0, sizeof(__sanitizer_kernel_sigaction_t));
831 internal_memset(&k_oldact, 0, sizeof(__sanitizer_kernel_sigaction_t));
696d846a 832 const __sanitizer_sigaction *u_act = (const __sanitizer_sigaction *)act;
dee5ea7a
KS
833 __sanitizer_sigaction *u_oldact = (__sanitizer_sigaction *)oldact;
834 if (u_act) {
835 k_act.handler = u_act->handler;
836 k_act.sigaction = u_act->sigaction;
837 internal_memcpy(&k_act.sa_mask, &u_act->sa_mask,
838 sizeof(__sanitizer_kernel_sigset_t));
696d846a
MO
839 // Without SA_RESTORER kernel ignores the calls (probably returns EINVAL).
840 k_act.sa_flags = u_act->sa_flags | SA_RESTORER;
dee5ea7a
KS
841 // FIXME: most often sa_restorer is unset, however the kernel requires it
842 // to point to a valid signal restorer that calls the rt_sigreturn syscall.
843 // If sa_restorer passed to the kernel is NULL, the program may crash upon
844 // signal delivery or fail to unwind the stack in the signal handler.
845 // libc implementation of sigaction() passes its own restorer to
846 // rt_sigaction, so we need to do the same (we'll need to reimplement the
847 // restorers; for x86_64 the restorer address can be obtained from
848 // oldact->sa_restorer upon a call to sigaction(xxx, NULL, oldact).
10189819 849#if !SANITIZER_ANDROID || !SANITIZER_MIPS32
dee5ea7a 850 k_act.sa_restorer = u_act->sa_restorer;
10189819 851#endif
dee5ea7a
KS
852 }
853
854 uptr result = internal_syscall(SYSCALL(rt_sigaction), (uptr)signum,
696d846a
MO
855 (uptr)(u_act ? &k_act : nullptr),
856 (uptr)(u_oldact ? &k_oldact : nullptr),
dee5ea7a
KS
857 (uptr)sizeof(__sanitizer_kernel_sigset_t));
858
859 if ((result == 0) && u_oldact) {
860 u_oldact->handler = k_oldact.handler;
861 u_oldact->sigaction = k_oldact.sigaction;
862 internal_memcpy(&u_oldact->sa_mask, &k_oldact.sa_mask,
863 sizeof(__sanitizer_kernel_sigset_t));
864 u_oldact->sa_flags = k_oldact.sa_flags;
10189819 865#if !SANITIZER_ANDROID || !SANITIZER_MIPS32
dee5ea7a 866 u_oldact->sa_restorer = k_oldact.sa_restorer;
10189819 867#endif
dee5ea7a
KS
868 }
869 return result;
df77f0e4 870}
dee5ea7a 871#endif // SANITIZER_LINUX
df77f0e4 872
dee5ea7a 873uptr internal_sigprocmask(int how, __sanitizer_sigset_t *set,
eac97531 874 __sanitizer_sigset_t *oldset) {
98f792ff 875#if SANITIZER_FREEBSD
dee5ea7a
KS
876 return internal_syscall(SYSCALL(sigprocmask), how, set, oldset);
877#else
878 __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
879 __sanitizer_kernel_sigset_t *k_oldset = (__sanitizer_kernel_sigset_t *)oldset;
0b997f6e
ML
880 return internal_syscall(SYSCALL(rt_sigprocmask), (uptr)how, (uptr)k_set,
881 (uptr)k_oldset, sizeof(__sanitizer_kernel_sigset_t));
dee5ea7a
KS
882#endif
883}
884
885void internal_sigfillset(__sanitizer_sigset_t *set) {
df77f0e4
KS
886 internal_memset(set, 0xff, sizeof(*set));
887}
888
10189819
MO
889void internal_sigemptyset(__sanitizer_sigset_t *set) {
890 internal_memset(set, 0, sizeof(*set));
891}
892
dee5ea7a
KS
893#if SANITIZER_LINUX
894void internal_sigdelset(__sanitizer_sigset_t *set, int signum) {
df77f0e4
KS
895 signum -= 1;
896 CHECK_GE(signum, 0);
897 CHECK_LT(signum, sizeof(*set) * 8);
dee5ea7a
KS
898 __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
899 const uptr idx = signum / (sizeof(k_set->sig[0]) * 8);
900 const uptr bit = signum % (sizeof(k_set->sig[0]) * 8);
90e46074 901 k_set->sig[idx] &= ~((uptr)1 << bit);
df77f0e4 902}
10189819
MO
903
904bool internal_sigismember(__sanitizer_sigset_t *set, int signum) {
905 signum -= 1;
906 CHECK_GE(signum, 0);
907 CHECK_LT(signum, sizeof(*set) * 8);
908 __sanitizer_kernel_sigset_t *k_set = (__sanitizer_kernel_sigset_t *)set;
909 const uptr idx = signum / (sizeof(k_set->sig[0]) * 8);
910 const uptr bit = signum % (sizeof(k_set->sig[0]) * 8);
90e46074 911 return k_set->sig[idx] & ((uptr)1 << bit);
10189819 912}
eac97531
ML
913#elif SANITIZER_FREEBSD
914void internal_sigdelset(__sanitizer_sigset_t *set, int signum) {
915 sigset_t *rset = reinterpret_cast<sigset_t *>(set);
916 sigdelset(rset, signum);
917}
918
919bool internal_sigismember(__sanitizer_sigset_t *set, int signum) {
920 sigset_t *rset = reinterpret_cast<sigset_t *>(set);
921 return sigismember(rset, signum);
922}
923#endif
924#endif // !SANITIZER_SOLARIS
df77f0e4 925
eac97531 926#if !SANITIZER_NETBSD
ef1b3fda 927// ThreadLister implementation.
eac97531 928ThreadLister::ThreadLister(pid_t pid) : pid_(pid), buffer_(4096) {
ef1b3fda
KS
929 char task_directory_path[80];
930 internal_snprintf(task_directory_path, sizeof(task_directory_path),
931 "/proc/%d/task/", pid);
eac97531
ML
932 descriptor_ = internal_open(task_directory_path, O_RDONLY | O_DIRECTORY);
933 if (internal_iserror(descriptor_)) {
ef1b3fda 934 Report("Can't open /proc/%d/task for reading.\n", pid);
ef1b3fda
KS
935 }
936}
937
eac97531
ML
938ThreadLister::Result ThreadLister::ListThreads(
939 InternalMmapVector<tid_t> *threads) {
940 if (internal_iserror(descriptor_))
941 return Error;
942 internal_lseek(descriptor_, 0, SEEK_SET);
943 threads->clear();
944
945 Result result = Ok;
946 for (bool first_read = true;; first_read = false) {
947 // Resize to max capacity if it was downsized by IsAlive.
948 buffer_.resize(buffer_.capacity());
949 CHECK_GE(buffer_.size(), 4096);
950 uptr read = internal_getdents(
951 descriptor_, (struct linux_dirent *)buffer_.data(), buffer_.size());
952 if (!read)
953 return result;
954 if (internal_iserror(read)) {
955 Report("Can't read directory entries from /proc/%d/task.\n", pid_);
956 return Error;
ef1b3fda 957 }
ef1b3fda 958
eac97531
ML
959 for (uptr begin = (uptr)buffer_.data(), end = begin + read; begin < end;) {
960 struct linux_dirent *entry = (struct linux_dirent *)begin;
961 begin += entry->d_reclen;
962 if (entry->d_ino == 1) {
963 // Inode 1 is for bad blocks and also can be a reason for early return.
964 // Should be emitted if kernel tried to output terminating thread.
965 // See proc_task_readdir implementation in Linux.
966 result = Incomplete;
967 }
968 if (entry->d_ino && *entry->d_name >= '0' && *entry->d_name <= '9')
969 threads->push_back(internal_atoll(entry->d_name));
970 }
ef1b3fda 971
eac97531
ML
972 // Now we are going to detect short-read or early EOF. In such cases Linux
973 // can return inconsistent list with missing alive threads.
974 // Code will just remember that the list can be incomplete but it will
975 // continue reads to return as much as possible.
976 if (!first_read) {
977 // The first one was a short-read by definition.
978 result = Incomplete;
979 } else if (read > buffer_.size() - 1024) {
980 // Read was close to the buffer size. So double the size and assume the
981 // worst.
982 buffer_.resize(buffer_.size() * 2);
983 result = Incomplete;
984 } else if (!threads->empty() && !IsAlive(threads->back())) {
985 // Maybe Linux early returned from read on terminated thread (!pid_alive)
986 // and failed to restore read position.
987 // See next_tid and proc_task_instantiate in Linux.
988 result = Incomplete;
989 }
990 }
ef1b3fda
KS
991}
992
eac97531
ML
993bool ThreadLister::IsAlive(int tid) {
994 // /proc/%d/task/%d/status uses same call to detect alive threads as
995 // proc_task_readdir. See task_state implementation in Linux.
996 char path[80];
997 internal_snprintf(path, sizeof(path), "/proc/%d/task/%d/status", pid_, tid);
998 if (!ReadFileToVector(path, &buffer_) || buffer_.empty())
ef1b3fda 999 return false;
eac97531
ML
1000 buffer_.push_back(0);
1001 static const char kPrefix[] = "\nPPid:";
1002 const char *field = internal_strstr(buffer_.data(), kPrefix);
1003 if (!field)
ef1b3fda 1004 return false;
eac97531
ML
1005 field += internal_strlen(kPrefix);
1006 return (int)internal_atoll(field) != 0;
1007}
1008
1009ThreadLister::~ThreadLister() {
1010 if (!internal_iserror(descriptor_))
1011 internal_close(descriptor_);
ef1b3fda 1012}
eac97531 1013#endif
ef1b3fda 1014
5d3805fc
JJ
1015#if SANITIZER_WORDSIZE == 32
1016// Take care of unusable kernel area in top gigabyte.
1017static uptr GetKernelAreaSize() {
1018#if SANITIZER_LINUX && !SANITIZER_X32
1019 const uptr gbyte = 1UL << 30;
1020
1021 // Firstly check if there are writable segments
1022 // mapped to top gigabyte (e.g. stack).
1023 MemoryMappingLayout proc_maps(/*cache_enabled*/true);
b667dd70
ML
1024 if (proc_maps.Error())
1025 return 0;
5d3805fc
JJ
1026 MemoryMappedSegment segment;
1027 while (proc_maps.Next(&segment)) {
1028 if ((segment.end >= 3 * gbyte) && segment.IsWritable()) return 0;
1029 }
1030
1031#if !SANITIZER_ANDROID
1032 // Even if nothing is mapped, top Gb may still be accessible
1033 // if we are running on 64-bit kernel.
1034 // Uname may report misleading results if personality type
1035 // is modified (e.g. under schroot) so check this as well.
1036 struct utsname uname_info;
1037 int pers = personality(0xffffffffUL);
3c6331c2
ML
1038 if (!(pers & PER_MASK) && internal_uname(&uname_info) == 0 &&
1039 internal_strstr(uname_info.machine, "64"))
5d3805fc
JJ
1040 return 0;
1041#endif // SANITIZER_ANDROID
1042
1043 // Top gigabyte is reserved for kernel.
1044 return gbyte;
1045#else
1046 return 0;
1047#endif // SANITIZER_LINUX && !SANITIZER_X32
1048}
1049#endif // SANITIZER_WORDSIZE == 32
1050
1051uptr GetMaxVirtualAddress() {
98f792ff 1052#if SANITIZER_NETBSD && defined(__x86_64__)
5d3805fc
JJ
1053 return 0x7f7ffffff000ULL; // (0x00007f8000000000 - PAGE_SIZE)
1054#elif SANITIZER_WORDSIZE == 64
1055# if defined(__powerpc64__) || defined(__aarch64__)
1056 // On PowerPC64 we have two different address space layouts: 44- and 46-bit.
1057 // We somehow need to figure out which one we are using now and choose
1058 // one of 0x00000fffffffffffUL and 0x00003fffffffffffUL.
1059 // Note that with 'ulimit -s unlimited' the stack is moved away from the top
1060 // of the address space, so simply checking the stack address is not enough.
1061 // This should (does) work for both PowerPC64 Endian modes.
1062 // Similarly, aarch64 has multiple address space layouts: 39, 42 and 47-bit.
1063 return (1ULL << (MostSignificantSetBitIndex(GET_CURRENT_FRAME()) + 1)) - 1;
0b997f6e
ML
1064#elif SANITIZER_RISCV64
1065 return (1ULL << 38) - 1;
5d3805fc
JJ
1066# elif defined(__mips64)
1067 return (1ULL << 40) - 1; // 0x000000ffffffffffUL;
1068# elif defined(__s390x__)
1069 return (1ULL << 53) - 1; // 0x001fffffffffffffUL;
b667dd70 1070#elif defined(__sparc__)
36b50aeb 1071 return ~(uptr)0;
5d3805fc
JJ
1072# else
1073 return (1ULL << 47) - 1; // 0x00007fffffffffffUL;
1074# endif
1075#else // SANITIZER_WORDSIZE == 32
1076# if defined(__s390__)
1077 return (1ULL << 31) - 1; // 0x7fffffff;
1078# else
eac97531 1079 return (1ULL << 32) - 1; // 0xffffffff;
5d3805fc
JJ
1080# endif
1081#endif // SANITIZER_WORDSIZE
1082}
1083
eac97531
ML
1084uptr GetMaxUserVirtualAddress() {
1085 uptr addr = GetMaxVirtualAddress();
1086#if SANITIZER_WORDSIZE == 32 && !defined(__s390__)
1087 if (!common_flags()->full_address_space)
1088 addr -= GetKernelAreaSize();
1089 CHECK_LT(reinterpret_cast<uptr>(&addr), addr);
1090#endif
1091 return addr;
1092}
1093
b667dd70 1094#if !SANITIZER_ANDROID
ef1b3fda 1095uptr GetPageSize() {
3c6331c2
ML
1096#if SANITIZER_LINUX && (defined(__x86_64__) || defined(__i386__)) && \
1097 defined(EXEC_PAGESIZE)
ef1b3fda 1098 return EXEC_PAGESIZE;
b667dd70
ML
1099#elif SANITIZER_FREEBSD || SANITIZER_NETBSD
1100// Use sysctl as sysconf can trigger interceptors internally.
1101 int pz = 0;
1102 uptr pzl = sizeof(pz);
1103 int mib[2] = {CTL_HW, HW_PAGESIZE};
1104 int rv = internal_sysctl(mib, 2, &pz, &pzl, nullptr, 0);
1105 CHECK_EQ(rv, 0);
1106 return (uptr)pz;
3ca75cd5
ML
1107#elif SANITIZER_USE_GETAUXVAL
1108 return getauxval(AT_PAGESZ);
ef1b3fda
KS
1109#else
1110 return sysconf(_SC_PAGESIZE); // EXEC_PAGESIZE may not be trustworthy.
1111#endif
1112}
b667dd70 1113#endif // !SANITIZER_ANDROID
ef1b3fda 1114
ef1b3fda 1115uptr ReadBinaryName(/*out*/char *buf, uptr buf_len) {
eac97531
ML
1116#if SANITIZER_SOLARIS
1117 const char *default_module_name = getexecname();
1118 CHECK_NE(default_module_name, NULL);
1119 return internal_snprintf(buf, buf_len, "%s", default_module_name);
1120#else
5d3805fc 1121#if SANITIZER_FREEBSD || SANITIZER_NETBSD
866e32ad 1122#if SANITIZER_FREEBSD
5d3805fc
JJ
1123 const int Mib[4] = {CTL_KERN, KERN_PROC, KERN_PROC_PATHNAME, -1};
1124#else
1125 const int Mib[4] = {CTL_KERN, KERN_PROC_ARGS, -1, KERN_PROC_PATHNAME};
1126#endif
696d846a 1127 const char *default_module_name = "kern.proc.pathname";
eac97531
ML
1128 uptr Size = buf_len;
1129 bool IsErr =
1130 (internal_sysctl(Mib, ARRAY_SIZE(Mib), buf, &Size, NULL, 0) != 0);
866e32ad
KS
1131 int readlink_error = IsErr ? errno : 0;
1132 uptr module_name_len = Size;
1133#else
696d846a 1134 const char *default_module_name = "/proc/self/exe";
ef1b3fda 1135 uptr module_name_len = internal_readlink(
696d846a 1136 default_module_name, buf, buf_len);
ef1b3fda 1137 int readlink_error;
866e32ad 1138 bool IsErr = internal_iserror(module_name_len, &readlink_error);
eac97531 1139#endif // SANITIZER_SOLARIS
866e32ad 1140 if (IsErr) {
696d846a
MO
1141 // We can't read binary name for some reason, assume it's unknown.
1142 Report("WARNING: reading executable name failed with errno %d, "
866e32ad 1143 "some stack frames may not be symbolized\n", readlink_error);
696d846a
MO
1144 module_name_len = internal_snprintf(buf, buf_len, "%s",
1145 default_module_name);
ef1b3fda 1146 CHECK_LT(module_name_len, buf_len);
ef1b3fda
KS
1147 }
1148 return module_name_len;
eac97531 1149#endif
2660d12d
KS
1150}
1151
696d846a
MO
1152uptr ReadLongProcessName(/*out*/ char *buf, uptr buf_len) {
1153#if SANITIZER_LINUX
1154 char *tmpbuf;
1155 uptr tmpsize;
1156 uptr tmplen;
1157 if (ReadFileToBuffer("/proc/self/cmdline", &tmpbuf, &tmpsize, &tmplen,
1158 1024 * 1024)) {
1159 internal_strncpy(buf, tmpbuf, buf_len);
1160 UnmapOrDie(tmpbuf, tmpsize);
1161 return internal_strlen(buf);
ef1b3fda 1162 }
696d846a
MO
1163#endif
1164 return ReadBinaryName(buf, buf_len);
ef1b3fda
KS
1165}
1166
1167// Match full names of the form /path/to/base_name{-,.}*
1168bool LibraryNameIs(const char *full_name, const char *base_name) {
1169 const char *name = full_name;
1170 // Strip path.
1171 while (*name != '\0') name++;
1172 while (name > full_name && *name != '/') name--;
1173 if (*name == '/') name++;
1174 uptr base_name_length = internal_strlen(base_name);
1175 if (internal_strncmp(name, base_name, base_name_length)) return false;
1176 return (name[base_name_length] == '-' || name[base_name_length] == '.');
1177}
1178
1179#if !SANITIZER_ANDROID
1180// Call cb for each region mapped by map.
1181void ForEachMappedRegion(link_map *map, void (*cb)(const void *, uptr)) {
696d846a 1182 CHECK_NE(map, nullptr);
98f792ff 1183#if !SANITIZER_FREEBSD
ef1b3fda
KS
1184 typedef ElfW(Phdr) Elf_Phdr;
1185 typedef ElfW(Ehdr) Elf_Ehdr;
98f792ff 1186#endif // !SANITIZER_FREEBSD
ef1b3fda
KS
1187 char *base = (char *)map->l_addr;
1188 Elf_Ehdr *ehdr = (Elf_Ehdr *)base;
1189 char *phdrs = base + ehdr->e_phoff;
1190 char *phdrs_end = phdrs + ehdr->e_phnum * ehdr->e_phentsize;
1191
1192 // Find the segment with the minimum base so we can "relocate" the p_vaddr
1193 // fields. Typically ET_DYN objects (DSOs) have base of zero and ET_EXEC
1194 // objects have a non-zero base.
1195 uptr preferred_base = (uptr)-1;
1196 for (char *iter = phdrs; iter != phdrs_end; iter += ehdr->e_phentsize) {
1197 Elf_Phdr *phdr = (Elf_Phdr *)iter;
1198 if (phdr->p_type == PT_LOAD && preferred_base > (uptr)phdr->p_vaddr)
1199 preferred_base = (uptr)phdr->p_vaddr;
1200 }
1201
1202 // Compute the delta from the real base to get a relocation delta.
1203 sptr delta = (uptr)base - preferred_base;
1204 // Now we can figure out what the loader really mapped.
1205 for (char *iter = phdrs; iter != phdrs_end; iter += ehdr->e_phentsize) {
1206 Elf_Phdr *phdr = (Elf_Phdr *)iter;
1207 if (phdr->p_type == PT_LOAD) {
1208 uptr seg_start = phdr->p_vaddr + delta;
1209 uptr seg_end = seg_start + phdr->p_memsz;
1210 // None of these values are aligned. We consider the ragged edges of the
1211 // load command as defined, since they are mapped from the file.
1212 seg_start = RoundDownTo(seg_start, GetPageSizeCached());
1213 seg_end = RoundUpTo(seg_end, GetPageSizeCached());
1214 cb((void *)seg_start, seg_end - seg_start);
1215 }
1216 }
1217}
1218#endif
1219
dee5ea7a 1220#if defined(__x86_64__) && SANITIZER_LINUX
ef1b3fda
KS
1221// We cannot use glibc's clone wrapper, because it messes with the child
1222// task's TLS. It writes the PID and TID of the child task to its thread
1223// descriptor, but in our case the child task shares the thread descriptor with
1224// the parent (because we don't know how to allocate a new thread
1225// descriptor to keep glibc happy). So the stock version of clone(), when
1226// used with CLONE_VM, would end up corrupting the parent's thread descriptor.
1227uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1228 int *parent_tidptr, void *newtls, int *child_tidptr) {
1229 long long res;
1230 if (!fn || !child_stack)
1231 return -EINVAL;
1232 CHECK_EQ(0, (uptr)child_stack % 16);
e45de14e
L
1233 child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
1234 ((unsigned long long *)child_stack)[0] = (uptr)fn;
1235 ((unsigned long long *)child_stack)[1] = (uptr)arg;
df77f0e4
KS
1236 register void *r8 __asm__("r8") = newtls;
1237 register int *r10 __asm__("r10") = child_tidptr;
ef1b3fda 1238 __asm__ __volatile__(
dee5ea7a 1239 /* %rax = syscall(%rax = SYSCALL(clone),
ef1b3fda
KS
1240 * %rdi = flags,
1241 * %rsi = child_stack,
1242 * %rdx = parent_tidptr,
1243 * %r8 = new_tls,
1244 * %r10 = child_tidptr)
1245 */
ef1b3fda
KS
1246 "syscall\n"
1247
1248 /* if (%rax != 0)
1249 * return;
1250 */
1251 "testq %%rax,%%rax\n"
1252 "jnz 1f\n"
1253
1254 /* In the child. Terminate unwind chain. */
486294d4
MS
1255 // XXX: We should also terminate the CFI unwind chain
1256 // here. Unfortunately clang 3.2 doesn't support the
1257 // necessary CFI directives, so we skip that part.
ef1b3fda
KS
1258 "xorq %%rbp,%%rbp\n"
1259
1260 /* Call "fn(arg)". */
1261 "popq %%rax\n"
1262 "popq %%rdi\n"
1263 "call *%%rax\n"
1264
1265 /* Call _exit(%rax). */
1266 "movq %%rax,%%rdi\n"
1267 "movq %2,%%rax\n"
1268 "syscall\n"
1269
1270 /* Return to parent. */
1271 "1:\n"
1272 : "=a" (res)
dee5ea7a 1273 : "a"(SYSCALL(clone)), "i"(SYSCALL(exit)),
ef1b3fda
KS
1274 "S"(child_stack),
1275 "D"(flags),
1276 "d"(parent_tidptr),
e45de14e
L
1277 "r"(r8),
1278 "r"(r10)
eac97531 1279 : "memory", "r11", "rcx");
ef1b3fda
KS
1280 return res;
1281}
696d846a
MO
1282#elif defined(__mips__)
1283uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1284 int *parent_tidptr, void *newtls, int *child_tidptr) {
1285 long long res;
1286 if (!fn || !child_stack)
1287 return -EINVAL;
1288 CHECK_EQ(0, (uptr)child_stack % 16);
1289 child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
1290 ((unsigned long long *)child_stack)[0] = (uptr)fn;
1291 ((unsigned long long *)child_stack)[1] = (uptr)arg;
1292 register void *a3 __asm__("$7") = newtls;
1293 register int *a4 __asm__("$8") = child_tidptr;
1294 // We don't have proper CFI directives here because it requires alot of code
1295 // for very marginal benefits.
1296 __asm__ __volatile__(
1297 /* $v0 = syscall($v0 = __NR_clone,
1298 * $a0 = flags,
1299 * $a1 = child_stack,
1300 * $a2 = parent_tidptr,
1301 * $a3 = new_tls,
1302 * $a4 = child_tidptr)
1303 */
1304 ".cprestore 16;\n"
1305 "move $4,%1;\n"
1306 "move $5,%2;\n"
1307 "move $6,%3;\n"
1308 "move $7,%4;\n"
1309 /* Store the fifth argument on stack
1310 * if we are using 32-bit abi.
1311 */
1312#if SANITIZER_WORDSIZE == 32
1313 "lw %5,16($29);\n"
1314#else
1315 "move $8,%5;\n"
1316#endif
1317 "li $2,%6;\n"
1318 "syscall;\n"
dee5ea7a 1319
696d846a
MO
1320 /* if ($v0 != 0)
1321 * return;
1322 */
1323 "bnez $2,1f;\n"
1324
1325 /* Call "fn(arg)". */
10189819
MO
1326#if SANITIZER_WORDSIZE == 32
1327#ifdef __BIG_ENDIAN__
1328 "lw $25,4($29);\n"
1329 "lw $4,12($29);\n"
1330#else
1331 "lw $25,0($29);\n"
1332 "lw $4,8($29);\n"
1333#endif
1334#else
696d846a
MO
1335 "ld $25,0($29);\n"
1336 "ld $4,8($29);\n"
10189819 1337#endif
696d846a 1338 "jal $25;\n"
866e32ad 1339
696d846a
MO
1340 /* Call _exit($v0). */
1341 "move $4,$2;\n"
1342 "li $2,%7;\n"
1343 "syscall;\n"
1344
1345 /* Return to parent. */
1346 "1:\n"
1347 : "=r" (res)
1348 : "r"(flags),
1349 "r"(child_stack),
1350 "r"(parent_tidptr),
1351 "r"(a3),
1352 "r"(a4),
1353 "i"(__NR_clone),
1354 "i"(__NR_exit)
1355 : "memory", "$29" );
1356 return res;
866e32ad 1357}
0b997f6e
ML
1358#elif SANITIZER_RISCV64
1359uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1360 int *parent_tidptr, void *newtls, int *child_tidptr) {
0b997f6e
ML
1361 if (!fn || !child_stack)
1362 return -EINVAL;
d0fee87e 1363
0b997f6e 1364 CHECK_EQ(0, (uptr)child_stack % 16);
0b997f6e 1365
d0fee87e
ML
1366 register int res __asm__("a0");
1367 register int __flags __asm__("a0") = flags;
0b997f6e 1368 register void *__stack __asm__("a1") = child_stack;
d0fee87e
ML
1369 register int *__ptid __asm__("a2") = parent_tidptr;
1370 register void *__tls __asm__("a3") = newtls;
1371 register int *__ctid __asm__("a4") = child_tidptr;
1372 register int (*__fn)(void *) __asm__("a5") = fn;
1373 register void *__arg __asm__("a6") = arg;
1374 register int nr_clone __asm__("a7") = __NR_clone;
0b997f6e
ML
1375
1376 __asm__ __volatile__(
0b997f6e
ML
1377 "ecall\n"
1378
d0fee87e
ML
1379 /* if (a0 != 0)
1380 * return a0;
0b997f6e
ML
1381 */
1382 "bnez a0, 1f\n"
1383
d0fee87e
ML
1384 // In the child, now. Call "fn(arg)".
1385 "mv a0, a6\n"
1386 "jalr a5\n"
0b997f6e 1387
d0fee87e
ML
1388 // Call _exit(a0).
1389 "addi a7, zero, %9\n"
0b997f6e
ML
1390 "ecall\n"
1391 "1:\n"
1392
1393 : "=r"(res)
d0fee87e
ML
1394 : "0"(__flags), "r"(__stack), "r"(__ptid), "r"(__tls), "r"(__ctid),
1395 "r"(__fn), "r"(__arg), "r"(nr_clone), "i"(__NR_exit)
1396 : "memory");
0b997f6e
ML
1397 return res;
1398}
696d846a
MO
1399#elif defined(__aarch64__)
1400uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1401 int *parent_tidptr, void *newtls, int *child_tidptr) {
1402 long long res;
1403 if (!fn || !child_stack)
1404 return -EINVAL;
1405 CHECK_EQ(0, (uptr)child_stack % 16);
1406 child_stack = (char *)child_stack - 2 * sizeof(unsigned long long);
1407 ((unsigned long long *)child_stack)[0] = (uptr)fn;
1408 ((unsigned long long *)child_stack)[1] = (uptr)arg;
866e32ad 1409
696d846a
MO
1410 register int (*__fn)(void *) __asm__("x0") = fn;
1411 register void *__stack __asm__("x1") = child_stack;
1412 register int __flags __asm__("x2") = flags;
1413 register void *__arg __asm__("x3") = arg;
1414 register int *__ptid __asm__("x4") = parent_tidptr;
1415 register void *__tls __asm__("x5") = newtls;
1416 register int *__ctid __asm__("x6") = child_tidptr;
1417
1418 __asm__ __volatile__(
1419 "mov x0,x2\n" /* flags */
1420 "mov x2,x4\n" /* ptid */
1421 "mov x3,x5\n" /* tls */
1422 "mov x4,x6\n" /* ctid */
1423 "mov x8,%9\n" /* clone */
1424
1425 "svc 0x0\n"
1426
1427 /* if (%r0 != 0)
1428 * return %r0;
1429 */
1430 "cmp x0, #0\n"
1431 "bne 1f\n"
1432
1433 /* In the child, now. Call "fn(arg)". */
1434 "ldp x1, x0, [sp], #16\n"
1435 "blr x1\n"
1436
1437 /* Call _exit(%r0). */
1438 "mov x8, %10\n"
1439 "svc 0x0\n"
1440 "1:\n"
1441
1442 : "=r" (res)
1443 : "i"(-EINVAL),
1444 "r"(__fn), "r"(__stack), "r"(__flags), "r"(__arg),
1445 "r"(__ptid), "r"(__tls), "r"(__ctid),
1446 "i"(__NR_clone), "i"(__NR_exit)
1447 : "x30", "memory");
1448 return res;
dee5ea7a 1449}
10189819
MO
1450#elif defined(__powerpc64__)
1451uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1452 int *parent_tidptr, void *newtls, int *child_tidptr) {
1453 long long res;
5d3805fc
JJ
1454// Stack frame structure.
1455#if SANITIZER_PPC64V1
1456// Back chain == 0 (SP + 112)
1457// Frame (112 bytes):
1458// Parameter save area (SP + 48), 8 doublewords
1459// TOC save area (SP + 40)
1460// Link editor doubleword (SP + 32)
1461// Compiler doubleword (SP + 24)
1462// LR save area (SP + 16)
1463// CR save area (SP + 8)
1464// Back chain (SP + 0)
1465# define FRAME_SIZE 112
1466# define FRAME_TOC_SAVE_OFFSET 40
1467#elif SANITIZER_PPC64V2
1468// Back chain == 0 (SP + 32)
1469// Frame (32 bytes):
1470// TOC save area (SP + 24)
1471// LR save area (SP + 16)
1472// CR save area (SP + 8)
1473// Back chain (SP + 0)
1474# define FRAME_SIZE 32
1475# define FRAME_TOC_SAVE_OFFSET 24
10189819 1476#else
5d3805fc 1477# error "Unsupported PPC64 ABI"
10189819
MO
1478#endif
1479 if (!fn || !child_stack)
1480 return -EINVAL;
1481 CHECK_EQ(0, (uptr)child_stack % 16);
dee5ea7a 1482
10189819
MO
1483 register int (*__fn)(void *) __asm__("r3") = fn;
1484 register void *__cstack __asm__("r4") = child_stack;
1485 register int __flags __asm__("r5") = flags;
5d3805fc
JJ
1486 register void *__arg __asm__("r6") = arg;
1487 register int *__ptidptr __asm__("r7") = parent_tidptr;
1488 register void *__newtls __asm__("r8") = newtls;
1489 register int *__ctidptr __asm__("r9") = child_tidptr;
10189819
MO
1490
1491 __asm__ __volatile__(
5d3805fc 1492 /* fn and arg are saved across the syscall */
10189819 1493 "mr 28, %5\n\t"
10189819
MO
1494 "mr 27, %8\n\t"
1495
1496 /* syscall
5d3805fc 1497 r0 == __NR_clone
10189819
MO
1498 r3 == flags
1499 r4 == child_stack
1500 r5 == parent_tidptr
1501 r6 == newtls
1502 r7 == child_tidptr */
1503 "mr 3, %7\n\t"
1504 "mr 5, %9\n\t"
1505 "mr 6, %10\n\t"
1506 "mr 7, %11\n\t"
1507 "li 0, %3\n\t"
1508 "sc\n\t"
1509
1510 /* Test if syscall was successful */
1511 "cmpdi cr1, 3, 0\n\t"
1512 "crandc cr1*4+eq, cr1*4+eq, cr0*4+so\n\t"
1513 "bne- cr1, 1f\n\t"
1514
5d3805fc
JJ
1515 /* Set up stack frame */
1516 "li 29, 0\n\t"
1517 "stdu 29, -8(1)\n\t"
1518 "stdu 1, -%12(1)\n\t"
10189819
MO
1519 /* Do the function call */
1520 "std 2, %13(1)\n\t"
5d3805fc 1521#if SANITIZER_PPC64V1
10189819
MO
1522 "ld 0, 0(28)\n\t"
1523 "ld 2, 8(28)\n\t"
1524 "mtctr 0\n\t"
5d3805fc 1525#elif SANITIZER_PPC64V2
10189819
MO
1526 "mr 12, 28\n\t"
1527 "mtctr 12\n\t"
5d3805fc
JJ
1528#else
1529# error "Unsupported PPC64 ABI"
10189819
MO
1530#endif
1531 "mr 3, 27\n\t"
1532 "bctrl\n\t"
1533 "ld 2, %13(1)\n\t"
1534
1535 /* Call _exit(r3) */
1536 "li 0, %4\n\t"
1537 "sc\n\t"
1538
1539 /* Return to parent */
1540 "1:\n\t"
1541 "mr %0, 3\n\t"
1542 : "=r" (res)
5d3805fc
JJ
1543 : "0" (-1),
1544 "i" (EINVAL),
1545 "i" (__NR_clone),
1546 "i" (__NR_exit),
1547 "r" (__fn),
1548 "r" (__cstack),
1549 "r" (__flags),
1550 "r" (__arg),
1551 "r" (__ptidptr),
1552 "r" (__newtls),
1553 "r" (__ctidptr),
1554 "i" (FRAME_SIZE),
1555 "i" (FRAME_TOC_SAVE_OFFSET)
1556 : "cr0", "cr1", "memory", "ctr", "r0", "r27", "r28", "r29");
1557 return res;
1558}
1559#elif defined(__i386__) && SANITIZER_LINUX
1560uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1561 int *parent_tidptr, void *newtls, int *child_tidptr) {
1562 int res;
1563 if (!fn || !child_stack)
1564 return -EINVAL;
1565 CHECK_EQ(0, (uptr)child_stack % 16);
1566 child_stack = (char *)child_stack - 7 * sizeof(unsigned int);
1567 ((unsigned int *)child_stack)[0] = (uptr)flags;
1568 ((unsigned int *)child_stack)[1] = (uptr)0;
1569 ((unsigned int *)child_stack)[2] = (uptr)fn;
1570 ((unsigned int *)child_stack)[3] = (uptr)arg;
1571 __asm__ __volatile__(
1572 /* %eax = syscall(%eax = SYSCALL(clone),
1573 * %ebx = flags,
1574 * %ecx = child_stack,
1575 * %edx = parent_tidptr,
1576 * %esi = new_tls,
1577 * %edi = child_tidptr)
1578 */
1579
1580 /* Obtain flags */
1581 "movl (%%ecx), %%ebx\n"
1582 /* Do the system call */
1583 "pushl %%ebx\n"
1584 "pushl %%esi\n"
1585 "pushl %%edi\n"
1586 /* Remember the flag value. */
1587 "movl %%ebx, (%%ecx)\n"
1588 "int $0x80\n"
1589 "popl %%edi\n"
1590 "popl %%esi\n"
1591 "popl %%ebx\n"
1592
1593 /* if (%eax != 0)
1594 * return;
1595 */
1596
1597 "test %%eax,%%eax\n"
1598 "jnz 1f\n"
1599
1600 /* terminate the stack frame */
1601 "xorl %%ebp,%%ebp\n"
1602 /* Call FN. */
1603 "call *%%ebx\n"
1604#ifdef PIC
1605 "call here\n"
1606 "here:\n"
1607 "popl %%ebx\n"
1608 "addl $_GLOBAL_OFFSET_TABLE_+[.-here], %%ebx\n"
1609#endif
1610 /* Call exit */
1611 "movl %%eax, %%ebx\n"
1612 "movl %2, %%eax\n"
1613 "int $0x80\n"
1614 "1:\n"
1615 : "=a" (res)
1616 : "a"(SYSCALL(clone)), "i"(SYSCALL(exit)),
1617 "c"(child_stack),
1618 "d"(parent_tidptr),
1619 "S"(newtls),
1620 "D"(child_tidptr)
1621 : "memory");
1622 return res;
1623}
1624#elif defined(__arm__) && SANITIZER_LINUX
1625uptr internal_clone(int (*fn)(void *), void *child_stack, int flags, void *arg,
1626 int *parent_tidptr, void *newtls, int *child_tidptr) {
1627 unsigned int res;
1628 if (!fn || !child_stack)
1629 return -EINVAL;
1630 child_stack = (char *)child_stack - 2 * sizeof(unsigned int);
1631 ((unsigned int *)child_stack)[0] = (uptr)fn;
1632 ((unsigned int *)child_stack)[1] = (uptr)arg;
1633 register int r0 __asm__("r0") = flags;
1634 register void *r1 __asm__("r1") = child_stack;
1635 register int *r2 __asm__("r2") = parent_tidptr;
1636 register void *r3 __asm__("r3") = newtls;
1637 register int *r4 __asm__("r4") = child_tidptr;
1638 register int r7 __asm__("r7") = __NR_clone;
1639
1640#if __ARM_ARCH > 4 || defined (__ARM_ARCH_4T__)
1641# define ARCH_HAS_BX
1642#endif
1643#if __ARM_ARCH > 4
1644# define ARCH_HAS_BLX
1645#endif
1646
1647#ifdef ARCH_HAS_BX
1648# ifdef ARCH_HAS_BLX
1649# define BLX(R) "blx " #R "\n"
1650# else
1651# define BLX(R) "mov lr, pc; bx " #R "\n"
1652# endif
1653#else
1654# define BLX(R) "mov lr, pc; mov pc," #R "\n"
1655#endif
1656
1657 __asm__ __volatile__(
1658 /* %r0 = syscall(%r7 = SYSCALL(clone),
1659 * %r0 = flags,
1660 * %r1 = child_stack,
1661 * %r2 = parent_tidptr,
1662 * %r3 = new_tls,
1663 * %r4 = child_tidptr)
1664 */
1665
1666 /* Do the system call */
1667 "swi 0x0\n"
1668
1669 /* if (%r0 != 0)
1670 * return %r0;
1671 */
1672 "cmp r0, #0\n"
1673 "bne 1f\n"
1674
1675 /* In the child, now. Call "fn(arg)". */
1676 "ldr r0, [sp, #4]\n"
1677 "ldr ip, [sp], #8\n"
1678 BLX(ip)
1679 /* Call _exit(%r0). */
1680 "mov r7, %7\n"
1681 "swi 0x0\n"
1682 "1:\n"
1683 "mov %0, r0\n"
1684 : "=r"(res)
1685 : "r"(r0), "r"(r1), "r"(r2), "r"(r3), "r"(r4), "r"(r7),
1686 "i"(__NR_exit)
1687 : "memory");
10189819 1688 return res;
dee5ea7a 1689}
10189819 1690#endif // defined(__x86_64__) && SANITIZER_LINUX
696d846a 1691
3c6331c2
ML
1692#if SANITIZER_LINUX
1693int internal_uname(struct utsname *buf) {
1694 return internal_syscall(SYSCALL(uname), buf);
1695}
1696#endif
1697
10189819 1698#if SANITIZER_ANDROID
696d846a
MO
1699#if __ANDROID_API__ < 21
1700extern "C" __attribute__((weak)) int dl_iterate_phdr(
1701 int (*)(struct dl_phdr_info *, size_t, void *), void *);
1702#endif
1703
1704static int dl_iterate_phdr_test_cb(struct dl_phdr_info *info, size_t size,
1705 void *data) {
1706 // Any name starting with "lib" indicates a bug in L where library base names
1707 // are returned instead of paths.
1708 if (info->dlpi_name && info->dlpi_name[0] == 'l' &&
1709 info->dlpi_name[1] == 'i' && info->dlpi_name[2] == 'b') {
1710 *(bool *)data = true;
1711 return 1;
1712 }
1713 return 0;
1714}
1715
1716static atomic_uint32_t android_api_level;
1717
eac97531
ML
1718static AndroidApiLevel AndroidDetectApiLevelStatic() {
1719#if __ANDROID_API__ <= 19
1720 return ANDROID_KITKAT;
1721#elif __ANDROID_API__ <= 22
1722 return ANDROID_LOLLIPOP_MR1;
1723#else
1724 return ANDROID_POST_LOLLIPOP;
1725#endif
1726}
1727
696d846a
MO
1728static AndroidApiLevel AndroidDetectApiLevel() {
1729 if (!&dl_iterate_phdr)
1730 return ANDROID_KITKAT; // K or lower
1731 bool base_name_seen = false;
1732 dl_iterate_phdr(dl_iterate_phdr_test_cb, &base_name_seen);
1733 if (base_name_seen)
1734 return ANDROID_LOLLIPOP_MR1; // L MR1
1735 return ANDROID_POST_LOLLIPOP; // post-L
1736 // Plain L (API level 21) is completely broken wrt ASan and not very
1737 // interesting to detect.
1738}
1739
eac97531
ML
1740extern "C" __attribute__((weak)) void* _DYNAMIC;
1741
696d846a
MO
1742AndroidApiLevel AndroidGetApiLevel() {
1743 AndroidApiLevel level =
1744 (AndroidApiLevel)atomic_load(&android_api_level, memory_order_relaxed);
1745 if (level) return level;
eac97531
ML
1746 level = &_DYNAMIC == nullptr ? AndroidDetectApiLevelStatic()
1747 : AndroidDetectApiLevel();
696d846a
MO
1748 atomic_store(&android_api_level, level, memory_order_relaxed);
1749 return level;
1750}
1751
dee5ea7a
KS
1752#endif
1753
5d3805fc
JJ
1754static HandleSignalMode GetHandleSignalModeImpl(int signum) {
1755 switch (signum) {
1756 case SIGABRT:
1757 return common_flags()->handle_abort;
1758 case SIGILL:
1759 return common_flags()->handle_sigill;
eac97531
ML
1760 case SIGTRAP:
1761 return common_flags()->handle_sigtrap;
5d3805fc
JJ
1762 case SIGFPE:
1763 return common_flags()->handle_sigfpe;
1764 case SIGSEGV:
1765 return common_flags()->handle_segv;
1766 case SIGBUS:
1767 return common_flags()->handle_sigbus;
1768 }
1769 return kHandleSignalNo;
1770}
1771
1772HandleSignalMode GetHandleSignalMode(int signum) {
1773 HandleSignalMode result = GetHandleSignalModeImpl(signum);
1774 if (result == kHandleSignalYes && !common_flags()->allow_user_segv_handler)
1775 return kHandleSignalExclusive;
1776 return result;
696d846a
MO
1777}
1778
10189819 1779#if !SANITIZER_GO
3c6331c2 1780void *internal_start_thread(void *(*func)(void *arg), void *arg) {
696d846a
MO
1781 // Start the thread with signals blocked, otherwise it can steal user signals.
1782 __sanitizer_sigset_t set, old;
1783 internal_sigfillset(&set);
1784#if SANITIZER_LINUX && !SANITIZER_ANDROID
1785 // Glibc uses SIGSETXID signal during setuid call. If this signal is blocked
1786 // on any thread, setuid call hangs (see test/tsan/setuid.c).
1787 internal_sigdelset(&set, 33);
1788#endif
1789 internal_sigprocmask(SIG_SETMASK, &set, &old);
1790 void *th;
3c6331c2 1791 real_pthread_create(&th, nullptr, func, arg);
696d846a
MO
1792 internal_sigprocmask(SIG_SETMASK, &old, nullptr);
1793 return th;
1794}
1795
1796void internal_join_thread(void *th) {
1797 real_pthread_join(th, nullptr);
1798}
1799#else
3c6331c2 1800void *internal_start_thread(void *(*func)(void *), void *arg) { return 0; }
696d846a
MO
1801
1802void internal_join_thread(void *th) {}
1803#endif
1804
10189819
MO
1805#if defined(__aarch64__)
1806// Android headers in the older NDK releases miss this definition.
1807struct __sanitizer_esr_context {
1808 struct _aarch64_ctx head;
1809 uint64_t esr;
1810};
1811
1812static bool Aarch64GetESR(ucontext_t *ucontext, u64 *esr) {
1813 static const u32 kEsrMagic = 0x45535201;
1814 u8 *aux = ucontext->uc_mcontext.__reserved;
1815 while (true) {
1816 _aarch64_ctx *ctx = (_aarch64_ctx *)aux;
1817 if (ctx->size == 0) break;
1818 if (ctx->magic == kEsrMagic) {
1819 *esr = ((__sanitizer_esr_context *)ctx)->esr;
1820 return true;
1821 }
1822 aux += ctx->size;
1823 }
1824 return false;
1825}
1826#endif
1827
eac97531 1828using Context = ucontext_t;
eac97531 1829
5d3805fc 1830SignalContext::WriteFlag SignalContext::GetWriteFlag() const {
eac97531 1831 Context *ucontext = (Context *)context;
10189819
MO
1832#if defined(__x86_64__) || defined(__i386__)
1833 static const uptr PF_WRITE = 1U << 1;
1834#if SANITIZER_FREEBSD
1835 uptr err = ucontext->uc_mcontext.mc_err;
5d3805fc
JJ
1836#elif SANITIZER_NETBSD
1837 uptr err = ucontext->uc_mcontext.__gregs[_REG_ERR];
eac97531
ML
1838#elif SANITIZER_SOLARIS && defined(__i386__)
1839 const int Err = 13;
1840 uptr err = ucontext->uc_mcontext.gregs[Err];
10189819
MO
1841#else
1842 uptr err = ucontext->uc_mcontext.gregs[REG_ERR];
eac97531 1843#endif // SANITIZER_FREEBSD
10189819 1844 return err & PF_WRITE ? WRITE : READ;
eac97531
ML
1845#elif defined(__mips__)
1846 uint32_t *exception_source;
1847 uint32_t faulty_instruction;
1848 uint32_t op_code;
1849
1850 exception_source = (uint32_t *)ucontext->uc_mcontext.pc;
1851 faulty_instruction = (uint32_t)(*exception_source);
1852
1853 op_code = (faulty_instruction >> 26) & 0x3f;
1854
1855 // FIXME: Add support for FPU, microMIPS, DSP, MSA memory instructions.
1856 switch (op_code) {
1857 case 0x28: // sb
1858 case 0x29: // sh
1859 case 0x2b: // sw
1860 case 0x3f: // sd
1861#if __mips_isa_rev < 6
1862 case 0x2c: // sdl
1863 case 0x2d: // sdr
1864 case 0x2a: // swl
1865 case 0x2e: // swr
1866#endif
1867 return SignalContext::WRITE;
1868
1869 case 0x20: // lb
1870 case 0x24: // lbu
1871 case 0x21: // lh
1872 case 0x25: // lhu
1873 case 0x23: // lw
1874 case 0x27: // lwu
1875 case 0x37: // ld
1876#if __mips_isa_rev < 6
1877 case 0x1a: // ldl
1878 case 0x1b: // ldr
1879 case 0x22: // lwl
1880 case 0x26: // lwr
1881#endif
1882 return SignalContext::READ;
1883#if __mips_isa_rev == 6
1884 case 0x3b: // pcrel
1885 op_code = (faulty_instruction >> 19) & 0x3;
1886 switch (op_code) {
1887 case 0x1: // lwpc
1888 case 0x2: // lwupc
1889 return SignalContext::READ;
1890 }
1891#endif
1892 }
1893 return SignalContext::UNKNOWN;
10189819
MO
1894#elif defined(__arm__)
1895 static const uptr FSR_WRITE = 1U << 11;
1896 uptr fsr = ucontext->uc_mcontext.error_code;
1897 return fsr & FSR_WRITE ? WRITE : READ;
1898#elif defined(__aarch64__)
1899 static const u64 ESR_ELx_WNR = 1U << 6;
1900 u64 esr;
1901 if (!Aarch64GetESR(ucontext, &esr)) return UNKNOWN;
1902 return esr & ESR_ELx_WNR ? WRITE : READ;
017abbe3 1903#elif defined(__sparc__)
eac97531
ML
1904 // Decode the instruction to determine the access type.
1905 // From OpenSolaris $SRC/uts/sun4/os/trap.c (get_accesstype).
b667dd70 1906#if SANITIZER_SOLARIS
eac97531 1907 uptr pc = ucontext->uc_mcontext.gregs[REG_PC];
b667dd70 1908#else
017abbe3
EB
1909 // Historical BSDism here.
1910 struct sigcontext *scontext = (struct sigcontext *)context;
b667dd70 1911#if defined(__arch64__)
017abbe3 1912 uptr pc = scontext->sigc_regs.tpc;
b667dd70 1913#else
017abbe3 1914 uptr pc = scontext->si_regs.pc;
b667dd70
ML
1915#endif
1916#endif
eac97531
ML
1917 u32 instr = *(u32 *)pc;
1918 return (instr >> 21) & 1 ? WRITE: READ;
3c6331c2
ML
1919#elif defined(__riscv)
1920 unsigned long pc = ucontext->uc_mcontext.__gregs[REG_PC];
1921 unsigned faulty_instruction = *(uint16_t *)pc;
1922
1923#if defined(__riscv_compressed)
1924 if ((faulty_instruction & 0x3) != 0x3) { // it's a compressed instruction
1925 // set op_bits to the instruction bits [1, 0, 15, 14, 13]
1926 unsigned op_bits =
1927 ((faulty_instruction & 0x3) << 3) | (faulty_instruction >> 13);
1928 unsigned rd = faulty_instruction & 0xF80; // bits 7-11, inclusive
1929 switch (op_bits) {
1930 case 0b10'010: // c.lwsp (rd != x0)
1931#if __riscv_xlen == 64
1932 case 0b10'011: // c.ldsp (rd != x0)
1933#endif
1934 return rd ? SignalContext::READ : SignalContext::UNKNOWN;
1935 case 0b00'010: // c.lw
1936#if __riscv_flen >= 32 && __riscv_xlen == 32
1937 case 0b10'011: // c.flwsp
1938#endif
1939#if __riscv_flen >= 32 || __riscv_xlen == 64
1940 case 0b00'011: // c.flw / c.ld
1941#endif
1942#if __riscv_flen == 64
1943 case 0b00'001: // c.fld
1944 case 0b10'001: // c.fldsp
1945#endif
1946 return SignalContext::READ;
1947 case 0b00'110: // c.sw
1948 case 0b10'110: // c.swsp
1949#if __riscv_flen >= 32 || __riscv_xlen == 64
1950 case 0b00'111: // c.fsw / c.sd
1951 case 0b10'111: // c.fswsp / c.sdsp
1952#endif
1953#if __riscv_flen == 64
1954 case 0b00'101: // c.fsd
1955 case 0b10'101: // c.fsdsp
1956#endif
1957 return SignalContext::WRITE;
1958 default:
1959 return SignalContext::UNKNOWN;
1960 }
1961 }
1962#endif
1963
1964 unsigned opcode = faulty_instruction & 0x7f; // lower 7 bits
1965 unsigned funct3 = (faulty_instruction >> 12) & 0x7; // bits 12-14, inclusive
1966 switch (opcode) {
1967 case 0b0000011: // loads
1968 switch (funct3) {
1969 case 0b000: // lb
1970 case 0b001: // lh
1971 case 0b010: // lw
1972#if __riscv_xlen == 64
1973 case 0b011: // ld
1974#endif
1975 case 0b100: // lbu
1976 case 0b101: // lhu
1977 return SignalContext::READ;
1978 default:
1979 return SignalContext::UNKNOWN;
1980 }
1981 case 0b0100011: // stores
1982 switch (funct3) {
1983 case 0b000: // sb
1984 case 0b001: // sh
1985 case 0b010: // sw
1986#if __riscv_xlen == 64
1987 case 0b011: // sd
1988#endif
1989 return SignalContext::WRITE;
1990 default:
1991 return SignalContext::UNKNOWN;
1992 }
1993#if __riscv_flen >= 32
1994 case 0b0000111: // floating-point loads
1995 switch (funct3) {
1996 case 0b010: // flw
1997#if __riscv_flen == 64
1998 case 0b011: // fld
1999#endif
2000 return SignalContext::READ;
2001 default:
2002 return SignalContext::UNKNOWN;
2003 }
2004 case 0b0100111: // floating-point stores
2005 switch (funct3) {
2006 case 0b010: // fsw
2007#if __riscv_flen == 64
2008 case 0b011: // fsd
2009#endif
2010 return SignalContext::WRITE;
2011 default:
2012 return SignalContext::UNKNOWN;
2013 }
2014#endif
2015 default:
2016 return SignalContext::UNKNOWN;
2017 }
10189819
MO
2018#else
2019 (void)ucontext;
2020 return UNKNOWN; // FIXME: Implement.
2021#endif
2022}
2023
3ca75cd5
ML
2024bool SignalContext::IsTrueFaultingAddress() const {
2025 auto si = static_cast<const siginfo_t *>(siginfo);
2026 // SIGSEGV signals without a true fault address have si_code set to 128.
2027 return si->si_signo == SIGSEGV && si->si_code != 128;
2028}
2029
5d3805fc
JJ
2030void SignalContext::DumpAllRegisters(void *context) {
2031 // FIXME: Implement this.
2032}
2033
2034static void GetPcSpBp(void *context, uptr *pc, uptr *sp, uptr *bp) {
eac97531
ML
2035#if SANITIZER_NETBSD
2036 // This covers all NetBSD architectures
2037 ucontext_t *ucontext = (ucontext_t *)context;
2038 *pc = _UC_MACHINE_PC(ucontext);
2039 *bp = _UC_MACHINE_FP(ucontext);
2040 *sp = _UC_MACHINE_SP(ucontext);
2041#elif defined(__arm__)
696d846a
MO
2042 ucontext_t *ucontext = (ucontext_t*)context;
2043 *pc = ucontext->uc_mcontext.arm_pc;
2044 *bp = ucontext->uc_mcontext.arm_fp;
2045 *sp = ucontext->uc_mcontext.arm_sp;
2046#elif defined(__aarch64__)
2047 ucontext_t *ucontext = (ucontext_t*)context;
2048 *pc = ucontext->uc_mcontext.pc;
2049 *bp = ucontext->uc_mcontext.regs[29];
2050 *sp = ucontext->uc_mcontext.sp;
2051#elif defined(__hppa__)
2052 ucontext_t *ucontext = (ucontext_t*)context;
2053 *pc = ucontext->uc_mcontext.sc_iaoq[0];
2054 /* GCC uses %r3 whenever a frame pointer is needed. */
2055 *bp = ucontext->uc_mcontext.sc_gr[3];
2056 *sp = ucontext->uc_mcontext.sc_gr[30];
2057#elif defined(__x86_64__)
2058# if SANITIZER_FREEBSD
2059 ucontext_t *ucontext = (ucontext_t*)context;
2060 *pc = ucontext->uc_mcontext.mc_rip;
2061 *bp = ucontext->uc_mcontext.mc_rbp;
2062 *sp = ucontext->uc_mcontext.mc_rsp;
2063# else
2064 ucontext_t *ucontext = (ucontext_t*)context;
2065 *pc = ucontext->uc_mcontext.gregs[REG_RIP];
2066 *bp = ucontext->uc_mcontext.gregs[REG_RBP];
2067 *sp = ucontext->uc_mcontext.gregs[REG_RSP];
2068# endif
2069#elif defined(__i386__)
2070# if SANITIZER_FREEBSD
2071 ucontext_t *ucontext = (ucontext_t*)context;
2072 *pc = ucontext->uc_mcontext.mc_eip;
2073 *bp = ucontext->uc_mcontext.mc_ebp;
2074 *sp = ucontext->uc_mcontext.mc_esp;
2075# else
2076 ucontext_t *ucontext = (ucontext_t*)context;
eac97531
ML
2077# if SANITIZER_SOLARIS
2078 /* Use the numeric values: the symbolic ones are undefined by llvm
2079 include/llvm/Support/Solaris.h. */
2080# ifndef REG_EIP
2081# define REG_EIP 14 // REG_PC
2082# endif
2083# ifndef REG_EBP
2084# define REG_EBP 6 // REG_FP
2085# endif
6805c1c6
RO
2086# ifndef REG_UESP
2087# define REG_UESP 17 // REG_SP
eac97531
ML
2088# endif
2089# endif
696d846a
MO
2090 *pc = ucontext->uc_mcontext.gregs[REG_EIP];
2091 *bp = ucontext->uc_mcontext.gregs[REG_EBP];
6805c1c6 2092 *sp = ucontext->uc_mcontext.gregs[REG_UESP];
696d846a
MO
2093# endif
2094#elif defined(__powerpc__) || defined(__powerpc64__)
2095 ucontext_t *ucontext = (ucontext_t*)context;
2096 *pc = ucontext->uc_mcontext.regs->nip;
2097 *sp = ucontext->uc_mcontext.regs->gpr[PT_R1];
2098 // The powerpc{,64}-linux ABIs do not specify r31 as the frame
2099 // pointer, but GCC always uses r31 when we need a frame pointer.
2100 *bp = ucontext->uc_mcontext.regs->gpr[PT_R31];
2101#elif defined(__sparc__)
b667dd70
ML
2102#if defined(__arch64__) || defined(__sparcv9)
2103#define STACK_BIAS 2047
2104#else
2105#define STACK_BIAS 0
eac97531 2106# endif
6388cb29 2107# if SANITIZER_SOLARIS
b667dd70 2108 ucontext_t *ucontext = (ucontext_t *)context;
696d846a 2109 *pc = ucontext->uc_mcontext.gregs[REG_PC];
017abbe3 2110 *sp = ucontext->uc_mcontext.gregs[REG_O6] + STACK_BIAS;
b667dd70 2111#else
017abbe3
EB
2112 // Historical BSDism here.
2113 struct sigcontext *scontext = (struct sigcontext *)context;
b667dd70 2114#if defined(__arch64__)
017abbe3
EB
2115 *pc = scontext->sigc_regs.tpc;
2116 *sp = scontext->sigc_regs.u_regs[14] + STACK_BIAS;
b667dd70 2117#else
017abbe3
EB
2118 *pc = scontext->si_regs.pc;
2119 *sp = scontext->si_regs.u_regs[14];
b667dd70 2120#endif
696d846a 2121# endif
b667dd70 2122 *bp = (uptr)((uhwptr *)*sp)[14] + STACK_BIAS;
696d846a
MO
2123#elif defined(__mips__)
2124 ucontext_t *ucontext = (ucontext_t*)context;
2125 *pc = ucontext->uc_mcontext.pc;
2126 *bp = ucontext->uc_mcontext.gregs[30];
2127 *sp = ucontext->uc_mcontext.gregs[29];
10189819
MO
2128#elif defined(__s390__)
2129 ucontext_t *ucontext = (ucontext_t*)context;
2130# if defined(__s390x__)
2131 *pc = ucontext->uc_mcontext.psw.addr;
2132# else
2133 *pc = ucontext->uc_mcontext.psw.addr & 0x7fffffff;
2134# endif
2135 *bp = ucontext->uc_mcontext.gregs[11];
2136 *sp = ucontext->uc_mcontext.gregs[15];
cb7dc4da
ML
2137#elif defined(__riscv)
2138 ucontext_t *ucontext = (ucontext_t*)context;
2139 *pc = ucontext->uc_mcontext.__gregs[REG_PC];
2140 *bp = ucontext->uc_mcontext.__gregs[REG_S0];
2141 *sp = ucontext->uc_mcontext.__gregs[REG_SP];
696d846a
MO
2142#else
2143# error "Unsupported arch"
2144#endif
dee5ea7a
KS
2145}
2146
5d3805fc
JJ
2147void SignalContext::InitPcSpBp() { GetPcSpBp(context, &pc, &sp, &bp); }
2148
b667dd70
ML
2149void InitializePlatformEarly() {
2150 // Do nothing.
2151}
2152
10189819
MO
2153void MaybeReexec() {
2154 // No need to re-exec on Linux.
2155}
2156
eac97531
ML
2157void CheckASLR() {
2158#if SANITIZER_NETBSD
2159 int mib[3];
2160 int paxflags;
2161 uptr len = sizeof(paxflags);
2162
2163 mib[0] = CTL_PROC;
2164 mib[1] = internal_getpid();
2165 mib[2] = PROC_PID_PAXFLAGS;
2166
2167 if (UNLIKELY(internal_sysctl(mib, 3, &paxflags, &len, NULL, 0) == -1)) {
2168 Printf("sysctl failed\n");
2169 Die();
2170 }
2171
2172 if (UNLIKELY(paxflags & CTL_PROC_PAXFLAGS_ASLR)) {
3c6331c2
ML
2173 Printf("This sanitizer is not compatible with enabled ASLR.\n"
2174 "To disable ASLR, please run \"paxctl +a %s\" and try again.\n",
2175 GetArgv()[0]);
eac97531
ML
2176 Die();
2177 }
39abbb86
BS
2178#elif SANITIZER_PPC64V2
2179 // Disable ASLR for Linux PPC64LE.
2180 int old_personality = personality(0xffffffff);
2181 if (old_personality != -1 && (old_personality & ADDR_NO_RANDOMIZE) == 0) {
2182 VReport(1, "WARNING: Program is being run with address space layout "
2183 "randomization (ASLR) enabled which prevents the thread and "
2184 "memory sanitizers from working on powerpc64le.\n"
2185 "ASLR will be disabled and the program re-executed.\n");
2186 CHECK_NE(personality(old_personality | ADDR_NO_RANDOMIZE), -1);
2187 ReExec();
2188 }
3ca75cd5
ML
2189#elif SANITIZER_FREEBSD
2190 int aslr_pie;
2191 uptr len = sizeof(aslr_pie);
2192#if SANITIZER_WORDSIZE == 64
2193 if (UNLIKELY(internal_sysctlbyname("kern.elf64.aslr.pie_enable",
2194 &aslr_pie, &len, NULL, 0) == -1)) {
2195 // We're making things less 'dramatic' here since
2196 // the OID is not necessarily guaranteed to be here
2197 // just yet regarding FreeBSD release
2198 return;
2199 }
2200
2201 if (aslr_pie > 0) {
2202 Printf("This sanitizer is not compatible with enabled ASLR "
2203 "and binaries compiled with PIE\n");
2204 Die();
2205 }
2206#endif
2207 // there might be 32 bits compat for 64 bits
2208 if (UNLIKELY(internal_sysctlbyname("kern.elf32.aslr.pie_enable",
2209 &aslr_pie, &len, NULL, 0) == -1)) {
2210 return;
2211 }
2212
2213 if (aslr_pie > 0) {
2214 Printf("This sanitizer is not compatible with enabled ASLR "
2215 "and binaries compiled with PIE\n");
2216 Die();
2217 }
eac97531
ML
2218#else
2219 // Do nothing
2220#endif
2221}
2222
b667dd70
ML
2223void CheckMPROTECT() {
2224#if SANITIZER_NETBSD
2225 int mib[3];
2226 int paxflags;
2227 uptr len = sizeof(paxflags);
2228
2229 mib[0] = CTL_PROC;
2230 mib[1] = internal_getpid();
2231 mib[2] = PROC_PID_PAXFLAGS;
2232
2233 if (UNLIKELY(internal_sysctl(mib, 3, &paxflags, &len, NULL, 0) == -1)) {
2234 Printf("sysctl failed\n");
2235 Die();
2236 }
2237
2238 if (UNLIKELY(paxflags & CTL_PROC_PAXFLAGS_MPROTECT)) {
2239 Printf("This sanitizer is not compatible with enabled MPROTECT\n");
2240 Die();
2241 }
2242#else
2243 // Do nothing
2244#endif
2245}
2246
5d3805fc
JJ
2247void CheckNoDeepBind(const char *filename, int flag) {
2248#ifdef RTLD_DEEPBIND
2249 if (flag & RTLD_DEEPBIND) {
2250 Report(
2251 "You are trying to dlopen a %s shared library with RTLD_DEEPBIND flag"
0b997f6e 2252 " which is incompatible with sanitizer runtime "
5d3805fc
JJ
2253 "(see https://github.com/google/sanitizers/issues/611 for details"
2254 "). If you want to run %s library under sanitizers please remove "
2255 "RTLD_DEEPBIND from dlopen flags.\n",
2256 filename, filename);
2257 Die();
2258 }
2259#endif
2260}
2261
2262uptr FindAvailableMemoryRange(uptr size, uptr alignment, uptr left_padding,
eac97531
ML
2263 uptr *largest_gap_found,
2264 uptr *max_occupied_addr) {
10189819
MO
2265 UNREACHABLE("FindAvailableMemoryRange is not available");
2266 return 0;
2267}
2268
5d3805fc
JJ
2269bool GetRandom(void *buffer, uptr length, bool blocking) {
2270 if (!buffer || !length || length > 256)
2271 return false;
eac97531
ML
2272#if SANITIZER_USE_GETENTROPY
2273 uptr rnd = getentropy(buffer, length);
2274 int rverrno = 0;
2275 if (internal_iserror(rnd, &rverrno) && rverrno == EFAULT)
2276 return false;
2277 else if (rnd == 0)
2278 return true;
2279#endif // SANITIZER_USE_GETENTROPY
2280
5d3805fc
JJ
2281#if SANITIZER_USE_GETRANDOM
2282 static atomic_uint8_t skip_getrandom_syscall;
2283 if (!atomic_load_relaxed(&skip_getrandom_syscall)) {
2284 // Up to 256 bytes, getrandom will not be interrupted.
2285 uptr res = internal_syscall(SYSCALL(getrandom), buffer, length,
2286 blocking ? 0 : GRND_NONBLOCK);
2287 int rverrno = 0;
2288 if (internal_iserror(res, &rverrno) && rverrno == ENOSYS)
2289 atomic_store_relaxed(&skip_getrandom_syscall, 1);
2290 else if (res == length)
2291 return true;
2292 }
eac97531 2293#endif // SANITIZER_USE_GETRANDOM
5d3805fc
JJ
2294 // Up to 256 bytes, a read off /dev/urandom will not be interrupted.
2295 // blocking is moot here, O_NONBLOCK has no effect when opening /dev/urandom.
2296 uptr fd = internal_open("/dev/urandom", O_RDONLY);
2297 if (internal_iserror(fd))
2298 return false;
2299 uptr res = internal_read(fd, buffer, length);
2300 if (internal_iserror(res))
2301 return false;
2302 internal_close(fd);
2303 return true;
2304}
2305
696d846a 2306} // namespace __sanitizer
f35db108 2307
eac97531 2308#endif