annotate libsanitizer/sanitizer_common/sanitizer_posix_libcdep.cpp @ 158:494b0b89df80 default tip

...
author Shinji KONO <kono@ie.u-ryukyu.ac.jp>
date Mon, 25 May 2020 18:13:55 +0900
parents 1830386684a0
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
145
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
1 //===-- sanitizer_posix_libcdep.cpp ---------------------------------------===//
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
2 //
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
4 // See https://llvm.org/LICENSE.txt for license information.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
6 //
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
7 //===----------------------------------------------------------------------===//
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
8 //
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
9 // This file is shared between AddressSanitizer and ThreadSanitizer
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
10 // run-time libraries and implements libc-dependent POSIX-specific functions
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
11 // from sanitizer_libc.h.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
12 //===----------------------------------------------------------------------===//
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
13
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
14 #include "sanitizer_platform.h"
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
15
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
16 #if SANITIZER_POSIX
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
17
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
18 #include "sanitizer_common.h"
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
19 #include "sanitizer_flags.h"
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
20 #include "sanitizer_platform_limits_netbsd.h"
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
21 #include "sanitizer_platform_limits_openbsd.h"
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
22 #include "sanitizer_platform_limits_posix.h"
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
23 #include "sanitizer_platform_limits_solaris.h"
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
24 #include "sanitizer_posix.h"
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
25 #include "sanitizer_procmaps.h"
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
26
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
27 #include <errno.h>
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
28 #include <fcntl.h>
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
29 #include <pthread.h>
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
30 #include <signal.h>
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
31 #include <stdlib.h>
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
32 #include <sys/mman.h>
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
33 #include <sys/resource.h>
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
34 #include <sys/stat.h>
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
35 #include <sys/time.h>
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
36 #include <sys/types.h>
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
37 #include <sys/wait.h>
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
38 #include <unistd.h>
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
39
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
40 #if SANITIZER_FREEBSD
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
41 // The MAP_NORESERVE define has been removed in FreeBSD 11.x, and even before
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
42 // that, it was never implemented. So just define it to zero.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
43 #undef MAP_NORESERVE
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
44 #define MAP_NORESERVE 0
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
45 #endif
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
46
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
47 typedef void (*sa_sigaction_t)(int, siginfo_t *, void *);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
48
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
49 namespace __sanitizer {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
50
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
51 u32 GetUid() {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
52 return getuid();
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
53 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
54
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
55 uptr GetThreadSelf() {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
56 return (uptr)pthread_self();
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
57 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
58
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
59 void ReleaseMemoryPagesToOS(uptr beg, uptr end) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
60 uptr page_size = GetPageSizeCached();
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
61 uptr beg_aligned = RoundUpTo(beg, page_size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
62 uptr end_aligned = RoundDownTo(end, page_size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
63 if (beg_aligned < end_aligned)
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
64 // In the default Solaris compilation environment, madvise() is declared
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
65 // to take a caddr_t arg; casting it to void * results in an invalid
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
66 // conversion error, so use char * instead.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
67 madvise((char *)beg_aligned, end_aligned - beg_aligned,
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
68 SANITIZER_MADVISE_DONTNEED);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
69 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
70
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
71 void SetShadowRegionHugePageMode(uptr addr, uptr size) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
72 #ifdef MADV_NOHUGEPAGE // May not be defined on old systems.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
73 if (common_flags()->no_huge_pages_for_shadow)
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
74 madvise((char *)addr, size, MADV_NOHUGEPAGE);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
75 else
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
76 madvise((char *)addr, size, MADV_HUGEPAGE);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
77 #endif // MADV_NOHUGEPAGE
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
78 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
79
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
80 bool DontDumpShadowMemory(uptr addr, uptr length) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
81 #if defined(MADV_DONTDUMP)
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
82 return madvise((char *)addr, length, MADV_DONTDUMP) == 0;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
83 #elif defined(MADV_NOCORE)
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
84 return madvise((char *)addr, length, MADV_NOCORE) == 0;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
85 #else
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
86 return true;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
87 #endif // MADV_DONTDUMP
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
88 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
89
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
90 static rlim_t getlim(int res) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
91 rlimit rlim;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
92 CHECK_EQ(0, getrlimit(res, &rlim));
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
93 return rlim.rlim_cur;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
94 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
95
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
96 static void setlim(int res, rlim_t lim) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
97 struct rlimit rlim;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
98 if (getrlimit(res, const_cast<struct rlimit *>(&rlim))) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
99 Report("ERROR: %s getrlimit() failed %d\n", SanitizerToolName, errno);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
100 Die();
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
101 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
102 rlim.rlim_cur = lim;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
103 if (setrlimit(res, const_cast<struct rlimit *>(&rlim))) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
104 Report("ERROR: %s setrlimit() failed %d\n", SanitizerToolName, errno);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
105 Die();
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
106 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
107 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
108
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
109 void DisableCoreDumperIfNecessary() {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
110 if (common_flags()->disable_coredump) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
111 setlim(RLIMIT_CORE, 0);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
112 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
113 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
114
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
115 bool StackSizeIsUnlimited() {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
116 rlim_t stack_size = getlim(RLIMIT_STACK);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
117 return (stack_size == RLIM_INFINITY);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
118 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
119
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
120 void SetStackSizeLimitInBytes(uptr limit) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
121 setlim(RLIMIT_STACK, (rlim_t)limit);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
122 CHECK(!StackSizeIsUnlimited());
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
123 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
124
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
125 bool AddressSpaceIsUnlimited() {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
126 rlim_t as_size = getlim(RLIMIT_AS);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
127 return (as_size == RLIM_INFINITY);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
128 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
129
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
130 void SetAddressSpaceUnlimited() {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
131 setlim(RLIMIT_AS, RLIM_INFINITY);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
132 CHECK(AddressSpaceIsUnlimited());
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
133 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
134
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
135 void SleepForSeconds(int seconds) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
136 sleep(seconds);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
137 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
138
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
139 void SleepForMillis(int millis) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
140 usleep(millis * 1000);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
141 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
142
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
143 void Abort() {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
144 #if !SANITIZER_GO
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
145 // If we are handling SIGABRT, unhandle it first.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
146 // TODO(vitalybuka): Check if handler belongs to sanitizer.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
147 if (GetHandleSignalMode(SIGABRT) != kHandleSignalNo) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
148 struct sigaction sigact;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
149 internal_memset(&sigact, 0, sizeof(sigact));
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
150 sigact.sa_sigaction = (sa_sigaction_t)SIG_DFL;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
151 internal_sigaction(SIGABRT, &sigact, nullptr);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
152 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
153 #endif
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
154
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
155 abort();
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
156 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
157
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
158 int Atexit(void (*function)(void)) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
159 #if !SANITIZER_GO
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
160 return atexit(function);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
161 #else
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
162 return 0;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
163 #endif
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
164 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
165
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
166 bool SupportsColoredOutput(fd_t fd) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
167 return isatty(fd) != 0;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
168 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
169
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
170 #if !SANITIZER_GO
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
171 // TODO(glider): different tools may require different altstack size.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
172 static const uptr kAltStackSize = SIGSTKSZ * 4; // SIGSTKSZ is not enough.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
173
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
174 void SetAlternateSignalStack() {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
175 stack_t altstack, oldstack;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
176 CHECK_EQ(0, sigaltstack(nullptr, &oldstack));
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
177 // If the alternate stack is already in place, do nothing.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
178 // Android always sets an alternate stack, but it's too small for us.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
179 if (!SANITIZER_ANDROID && !(oldstack.ss_flags & SS_DISABLE)) return;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
180 // TODO(glider): the mapped stack should have the MAP_STACK flag in the
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
181 // future. It is not required by man 2 sigaltstack now (they're using
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
182 // malloc()).
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
183 void* base = MmapOrDie(kAltStackSize, __func__);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
184 altstack.ss_sp = (char*) base;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
185 altstack.ss_flags = 0;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
186 altstack.ss_size = kAltStackSize;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
187 CHECK_EQ(0, sigaltstack(&altstack, nullptr));
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
188 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
189
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
190 void UnsetAlternateSignalStack() {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
191 stack_t altstack, oldstack;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
192 altstack.ss_sp = nullptr;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
193 altstack.ss_flags = SS_DISABLE;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
194 altstack.ss_size = kAltStackSize; // Some sane value required on Darwin.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
195 CHECK_EQ(0, sigaltstack(&altstack, &oldstack));
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
196 UnmapOrDie(oldstack.ss_sp, oldstack.ss_size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
197 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
198
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
199 static void MaybeInstallSigaction(int signum,
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
200 SignalHandlerType handler) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
201 if (GetHandleSignalMode(signum) == kHandleSignalNo) return;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
202
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
203 struct sigaction sigact;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
204 internal_memset(&sigact, 0, sizeof(sigact));
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
205 sigact.sa_sigaction = (sa_sigaction_t)handler;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
206 // Do not block the signal from being received in that signal's handler.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
207 // Clients are responsible for handling this correctly.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
208 sigact.sa_flags = SA_SIGINFO | SA_NODEFER;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
209 if (common_flags()->use_sigaltstack) sigact.sa_flags |= SA_ONSTACK;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
210 CHECK_EQ(0, internal_sigaction(signum, &sigact, nullptr));
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
211 VReport(1, "Installed the sigaction for signal %d\n", signum);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
212 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
213
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
214 void InstallDeadlySignalHandlers(SignalHandlerType handler) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
215 // Set the alternate signal stack for the main thread.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
216 // This will cause SetAlternateSignalStack to be called twice, but the stack
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
217 // will be actually set only once.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
218 if (common_flags()->use_sigaltstack) SetAlternateSignalStack();
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
219 MaybeInstallSigaction(SIGSEGV, handler);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
220 MaybeInstallSigaction(SIGBUS, handler);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
221 MaybeInstallSigaction(SIGABRT, handler);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
222 MaybeInstallSigaction(SIGFPE, handler);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
223 MaybeInstallSigaction(SIGILL, handler);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
224 MaybeInstallSigaction(SIGTRAP, handler);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
225 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
226
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
227 bool SignalContext::IsStackOverflow() const {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
228 // Access at a reasonable offset above SP, or slightly below it (to account
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
229 // for x86_64 or PowerPC redzone, ARM push of multiple registers, etc) is
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
230 // probably a stack overflow.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
231 #ifdef __s390__
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
232 // On s390, the fault address in siginfo points to start of the page, not
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
233 // to the precise word that was accessed. Mask off the low bits of sp to
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
234 // take it into account.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
235 bool IsStackAccess = addr >= (sp & ~0xFFF) && addr < sp + 0xFFFF;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
236 #else
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
237 // Let's accept up to a page size away from top of stack. Things like stack
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
238 // probing can trigger accesses with such large offsets.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
239 bool IsStackAccess = addr + GetPageSizeCached() > sp && addr < sp + 0xFFFF;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
240 #endif
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
241
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
242 #if __powerpc__
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
243 // Large stack frames can be allocated with e.g.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
244 // lis r0,-10000
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
245 // stdux r1,r1,r0 # store sp to [sp-10000] and update sp by -10000
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
246 // If the store faults then sp will not have been updated, so test above
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
247 // will not work, because the fault address will be more than just "slightly"
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
248 // below sp.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
249 if (!IsStackAccess && IsAccessibleMemoryRange(pc, 4)) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
250 u32 inst = *(unsigned *)pc;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
251 u32 ra = (inst >> 16) & 0x1F;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
252 u32 opcd = inst >> 26;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
253 u32 xo = (inst >> 1) & 0x3FF;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
254 // Check for store-with-update to sp. The instructions we accept are:
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
255 // stbu rs,d(ra) stbux rs,ra,rb
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
256 // sthu rs,d(ra) sthux rs,ra,rb
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
257 // stwu rs,d(ra) stwux rs,ra,rb
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
258 // stdu rs,ds(ra) stdux rs,ra,rb
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
259 // where ra is r1 (the stack pointer).
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
260 if (ra == 1 &&
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
261 (opcd == 39 || opcd == 45 || opcd == 37 || opcd == 62 ||
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
262 (opcd == 31 && (xo == 247 || xo == 439 || xo == 183 || xo == 181))))
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
263 IsStackAccess = true;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
264 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
265 #endif // __powerpc__
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
266
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
267 // We also check si_code to filter out SEGV caused by something else other
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
268 // then hitting the guard page or unmapped memory, like, for example,
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
269 // unaligned memory access.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
270 auto si = static_cast<const siginfo_t *>(siginfo);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
271 return IsStackAccess &&
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
272 (si->si_code == si_SEGV_MAPERR || si->si_code == si_SEGV_ACCERR);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
273 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
274
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
275 #endif // SANITIZER_GO
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
276
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
277 bool IsAccessibleMemoryRange(uptr beg, uptr size) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
278 uptr page_size = GetPageSizeCached();
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
279 // Checking too large memory ranges is slow.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
280 CHECK_LT(size, page_size * 10);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
281 int sock_pair[2];
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
282 if (pipe(sock_pair))
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
283 return false;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
284 uptr bytes_written =
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
285 internal_write(sock_pair[1], reinterpret_cast<void *>(beg), size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
286 int write_errno;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
287 bool result;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
288 if (internal_iserror(bytes_written, &write_errno)) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
289 CHECK_EQ(EFAULT, write_errno);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
290 result = false;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
291 } else {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
292 result = (bytes_written == size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
293 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
294 internal_close(sock_pair[0]);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
295 internal_close(sock_pair[1]);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
296 return result;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
297 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
298
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
299 void PlatformPrepareForSandboxing(__sanitizer_sandbox_arguments *args) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
300 // Some kinds of sandboxes may forbid filesystem access, so we won't be able
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
301 // to read the file mappings from /proc/self/maps. Luckily, neither the
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
302 // process will be able to load additional libraries, so it's fine to use the
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
303 // cached mappings.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
304 MemoryMappingLayout::CacheMemoryMappings();
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
305 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
306
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
307 static bool MmapFixed(uptr fixed_addr, uptr size, int additional_flags,
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
308 const char *name) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
309 size = RoundUpTo(size, GetPageSizeCached());
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
310 fixed_addr = RoundDownTo(fixed_addr, GetPageSizeCached());
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
311 uptr p =
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
312 MmapNamed((void *)fixed_addr, size, PROT_READ | PROT_WRITE,
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
313 MAP_PRIVATE | MAP_FIXED | additional_flags | MAP_ANON, name);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
314 int reserrno;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
315 if (internal_iserror(p, &reserrno)) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
316 Report("ERROR: %s failed to "
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
317 "allocate 0x%zx (%zd) bytes at address %zx (errno: %d)\n",
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
318 SanitizerToolName, size, size, fixed_addr, reserrno);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
319 return false;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
320 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
321 IncreaseTotalMmap(size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
322 return true;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
323 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
324
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
325 bool MmapFixedNoReserve(uptr fixed_addr, uptr size, const char *name) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
326 return MmapFixed(fixed_addr, size, MAP_NORESERVE, name);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
327 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
328
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
329 bool MmapFixedSuperNoReserve(uptr fixed_addr, uptr size, const char *name) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
330 #if SANITIZER_FREEBSD
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
331 if (common_flags()->no_huge_pages_for_shadow)
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
332 return MmapFixedNoReserve(fixed_addr, size, name);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
333 // MAP_NORESERVE is implicit with FreeBSD
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
334 return MmapFixed(fixed_addr, size, MAP_ALIGNED_SUPER, name);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
335 #else
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
336 bool r = MmapFixedNoReserve(fixed_addr, size, name);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
337 if (r)
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
338 SetShadowRegionHugePageMode(fixed_addr, size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
339 return r;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
340 #endif
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
341 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
342
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
343 uptr ReservedAddressRange::Init(uptr size, const char *name, uptr fixed_addr) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
344 base_ = fixed_addr ? MmapFixedNoAccess(fixed_addr, size, name)
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
345 : MmapNoAccess(size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
346 size_ = size;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
347 name_ = name;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
348 (void)os_handle_; // unsupported
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
349 return reinterpret_cast<uptr>(base_);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
350 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
351
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
352 // Uses fixed_addr for now.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
353 // Will use offset instead once we've implemented this function for real.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
354 uptr ReservedAddressRange::Map(uptr fixed_addr, uptr size, const char *name) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
355 return reinterpret_cast<uptr>(
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
356 MmapFixedOrDieOnFatalError(fixed_addr, size, name));
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
357 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
358
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
359 uptr ReservedAddressRange::MapOrDie(uptr fixed_addr, uptr size,
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
360 const char *name) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
361 return reinterpret_cast<uptr>(MmapFixedOrDie(fixed_addr, size, name));
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
362 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
363
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
364 void ReservedAddressRange::Unmap(uptr addr, uptr size) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
365 CHECK_LE(size, size_);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
366 if (addr == reinterpret_cast<uptr>(base_))
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
367 // If we unmap the whole range, just null out the base.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
368 base_ = (size == size_) ? nullptr : reinterpret_cast<void*>(addr + size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
369 else
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
370 CHECK_EQ(addr + size, reinterpret_cast<uptr>(base_) + size_);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
371 size_ -= size;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
372 UnmapOrDie(reinterpret_cast<void*>(addr), size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
373 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
374
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
375 void *MmapFixedNoAccess(uptr fixed_addr, uptr size, const char *name) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
376 return (void *)MmapNamed((void *)fixed_addr, size, PROT_NONE,
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
377 MAP_PRIVATE | MAP_FIXED | MAP_NORESERVE | MAP_ANON,
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
378 name);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
379 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
380
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
381 void *MmapNoAccess(uptr size) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
382 unsigned flags = MAP_PRIVATE | MAP_ANON | MAP_NORESERVE;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
383 return (void *)internal_mmap(nullptr, size, PROT_NONE, flags, -1, 0);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
384 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
385
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
386 // This function is defined elsewhere if we intercepted pthread_attr_getstack.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
387 extern "C" {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
388 SANITIZER_WEAK_ATTRIBUTE int
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
389 real_pthread_attr_getstack(void *attr, void **addr, size_t *size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
390 } // extern "C"
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
391
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
392 int my_pthread_attr_getstack(void *attr, void **addr, uptr *size) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
393 #if !SANITIZER_GO && !SANITIZER_MAC
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
394 if (&real_pthread_attr_getstack)
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
395 return real_pthread_attr_getstack((pthread_attr_t *)attr, addr,
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
396 (size_t *)size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
397 #endif
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
398 return pthread_attr_getstack((pthread_attr_t *)attr, addr, (size_t *)size);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
399 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
400
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
401 #if !SANITIZER_GO
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
402 void AdjustStackSize(void *attr_) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
403 pthread_attr_t *attr = (pthread_attr_t *)attr_;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
404 uptr stackaddr = 0;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
405 uptr stacksize = 0;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
406 my_pthread_attr_getstack(attr, (void**)&stackaddr, &stacksize);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
407 // GLibC will return (0 - stacksize) as the stack address in the case when
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
408 // stacksize is set, but stackaddr is not.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
409 bool stack_set = (stackaddr != 0) && (stackaddr + stacksize != 0);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
410 // We place a lot of tool data into TLS, account for that.
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
411 const uptr minstacksize = GetTlsSize() + 128*1024;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
412 if (stacksize < minstacksize) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
413 if (!stack_set) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
414 if (stacksize != 0) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
415 VPrintf(1, "Sanitizer: increasing stacksize %zu->%zu\n", stacksize,
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
416 minstacksize);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
417 pthread_attr_setstacksize(attr, minstacksize);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
418 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
419 } else {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
420 Printf("Sanitizer: pre-allocated stack size is insufficient: "
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
421 "%zu < %zu\n", stacksize, minstacksize);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
422 Printf("Sanitizer: pthread_create is likely to fail.\n");
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
423 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
424 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
425 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
426 #endif // !SANITIZER_GO
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
427
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
428 pid_t StartSubprocess(const char *program, const char *const argv[],
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
429 fd_t stdin_fd, fd_t stdout_fd, fd_t stderr_fd) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
430 auto file_closer = at_scope_exit([&] {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
431 if (stdin_fd != kInvalidFd) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
432 internal_close(stdin_fd);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
433 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
434 if (stdout_fd != kInvalidFd) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
435 internal_close(stdout_fd);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
436 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
437 if (stderr_fd != kInvalidFd) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
438 internal_close(stderr_fd);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
439 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
440 });
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
441
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
442 int pid = internal_fork();
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
443
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
444 if (pid < 0) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
445 int rverrno;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
446 if (internal_iserror(pid, &rverrno)) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
447 Report("WARNING: failed to fork (errno %d)\n", rverrno);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
448 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
449 return pid;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
450 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
451
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
452 if (pid == 0) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
453 // Child subprocess
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
454 if (stdin_fd != kInvalidFd) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
455 internal_close(STDIN_FILENO);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
456 internal_dup2(stdin_fd, STDIN_FILENO);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
457 internal_close(stdin_fd);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
458 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
459 if (stdout_fd != kInvalidFd) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
460 internal_close(STDOUT_FILENO);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
461 internal_dup2(stdout_fd, STDOUT_FILENO);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
462 internal_close(stdout_fd);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
463 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
464 if (stderr_fd != kInvalidFd) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
465 internal_close(STDERR_FILENO);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
466 internal_dup2(stderr_fd, STDERR_FILENO);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
467 internal_close(stderr_fd);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
468 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
469
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
470 for (int fd = sysconf(_SC_OPEN_MAX); fd > 2; fd--) internal_close(fd);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
471
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
472 execv(program, const_cast<char **>(&argv[0]));
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
473 internal__exit(1);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
474 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
475
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
476 return pid;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
477 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
478
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
479 bool IsProcessRunning(pid_t pid) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
480 int process_status;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
481 uptr waitpid_status = internal_waitpid(pid, &process_status, WNOHANG);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
482 int local_errno;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
483 if (internal_iserror(waitpid_status, &local_errno)) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
484 VReport(1, "Waiting on the process failed (errno %d).\n", local_errno);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
485 return false;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
486 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
487 return waitpid_status == 0;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
488 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
489
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
490 int WaitForProcess(pid_t pid) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
491 int process_status;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
492 uptr waitpid_status = internal_waitpid(pid, &process_status, 0);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
493 int local_errno;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
494 if (internal_iserror(waitpid_status, &local_errno)) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
495 VReport(1, "Waiting on the process failed (errno %d).\n", local_errno);
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
496 return -1;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
497 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
498 return process_status;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
499 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
500
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
501 bool IsStateDetached(int state) {
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
502 return state == PTHREAD_CREATE_DETACHED;
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
503 }
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
504
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
505 } // namespace __sanitizer
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
506
1830386684a0 gcc-9.2.0
anatofuz
parents:
diff changeset
507 #endif // SANITIZER_POSIX