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