Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 1 | //===-- msan_interceptors.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 a part of MemorySanitizer. |
| 11 | // |
| 12 | // Interceptors for standard library functions. |
Kostya Serebryany | 9a58d39 | 2012-12-12 09:54:35 +0000 | [diff] [blame] | 13 | // |
| 14 | // FIXME: move as many interceptors as possible into |
| 15 | // sanitizer_common/sanitizer_common_interceptors.h |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 16 | //===----------------------------------------------------------------------===// |
| 17 | |
| 18 | #include "interception/interception.h" |
| 19 | #include "msan.h" |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 20 | #include "sanitizer_common/sanitizer_platform_limits_posix.h" |
Kostya Serebryany | fa79cd6 | 2013-01-25 11:46:22 +0000 | [diff] [blame] | 21 | #include "sanitizer_common/sanitizer_allocator.h" |
Evgeniy Stepanov | bb25af5 | 2013-08-27 11:34:05 +0000 | [diff] [blame] | 22 | #include "sanitizer_common/sanitizer_atomic.h" |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 23 | #include "sanitizer_common/sanitizer_common.h" |
Evgeniy Stepanov | 9f7e8c1 | 2013-01-28 13:52:49 +0000 | [diff] [blame] | 24 | #include "sanitizer_common/sanitizer_stackdepot.h" |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 25 | #include "sanitizer_common/sanitizer_libc.h" |
Evgeniy Stepanov | b4a218d | 2013-03-19 09:30:52 +0000 | [diff] [blame] | 26 | #include "sanitizer_common/sanitizer_linux.h" |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 27 | |
| 28 | #include <stdarg.h> |
| 29 | // ACHTUNG! No other system header includes in this file. |
| 30 | // Ideally, we should get rid of stdarg.h as well. |
| 31 | |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 32 | using namespace __msan; |
| 33 | |
Evgeniy Stepanov | bb25af5 | 2013-08-27 11:34:05 +0000 | [diff] [blame] | 34 | using __sanitizer::memory_order; |
| 35 | using __sanitizer::atomic_load; |
| 36 | using __sanitizer::atomic_store; |
| 37 | using __sanitizer::atomic_uintptr_t; |
| 38 | |
Evgeniy Stepanov | bfd2122 | 2013-05-22 12:50:26 +0000 | [diff] [blame] | 39 | // True if this is a nested interceptor. |
| 40 | static THREADLOCAL int in_interceptor_scope; |
| 41 | |
| 42 | struct InterceptorScope { |
| 43 | InterceptorScope() { ++in_interceptor_scope; } |
| 44 | ~InterceptorScope() { --in_interceptor_scope; } |
| 45 | }; |
| 46 | |
| 47 | bool IsInInterceptorScope() { |
| 48 | return in_interceptor_scope; |
| 49 | } |
| 50 | |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 51 | #define ENSURE_MSAN_INITED() do { \ |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 52 | CHECK(!msan_init_is_running); \ |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 53 | if (!msan_inited) { \ |
| 54 | __msan_init(); \ |
| 55 | } \ |
| 56 | } while (0) |
| 57 | |
Evgeniy Stepanov | bfd2122 | 2013-05-22 12:50:26 +0000 | [diff] [blame] | 58 | // Check that [x, x+n) range is unpoisoned. |
Evgeniy Stepanov | ae4e1ec | 2013-06-21 12:37:58 +0000 | [diff] [blame] | 59 | #define CHECK_UNPOISONED_0(x, n) \ |
| 60 | do { \ |
| 61 | sptr offset = __msan_test_shadow(x, n); \ |
| 62 | if (__msan::IsInSymbolizer()) break; \ |
| 63 | if (offset >= 0 && __msan::flags()->report_umrs) { \ |
| 64 | GET_CALLER_PC_BP_SP; \ |
| 65 | (void) sp; \ |
| 66 | Printf("UMR in %s at offset %d inside [%p, +%d) \n", __FUNCTION__, \ |
| 67 | offset, x, n); \ |
| 68 | __msan::PrintWarningWithOrigin(pc, bp, \ |
| 69 | __msan_get_origin((char *)x + offset)); \ |
Dmitry Vyukov | 97f0eae | 2013-08-13 15:33:00 +0000 | [diff] [blame] | 70 | if (__msan::flags()->halt_on_error) { \ |
Evgeniy Stepanov | ae4e1ec | 2013-06-21 12:37:58 +0000 | [diff] [blame] | 71 | Printf("Exiting\n"); \ |
| 72 | Die(); \ |
| 73 | } \ |
| 74 | } \ |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 75 | } while (0) |
| 76 | |
Evgeniy Stepanov | bfd2122 | 2013-05-22 12:50:26 +0000 | [diff] [blame] | 77 | // Check that [x, x+n) range is unpoisoned unless we are in a nested |
| 78 | // interceptor. |
Evgeniy Stepanov | dec0f76 | 2013-06-28 11:02:43 +0000 | [diff] [blame] | 79 | #define CHECK_UNPOISONED(x, n) \ |
| 80 | do { \ |
| 81 | if (!IsInInterceptorScope()) CHECK_UNPOISONED_0(x, n); \ |
| 82 | } while (0); |
Evgeniy Stepanov | bfd2122 | 2013-05-22 12:50:26 +0000 | [diff] [blame] | 83 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 84 | static void *fast_memset(void *ptr, int c, SIZE_T n); |
| 85 | static void *fast_memcpy(void *dst, const void *src, SIZE_T n); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 86 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 87 | INTERCEPTOR(SIZE_T, fread, void *ptr, SIZE_T size, SIZE_T nmemb, void *file) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 88 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 89 | SIZE_T res = REAL(fread)(ptr, size, nmemb, file); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 90 | if (res > 0) |
| 91 | __msan_unpoison(ptr, res *size); |
| 92 | return res; |
| 93 | } |
| 94 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 95 | INTERCEPTOR(SIZE_T, fread_unlocked, void *ptr, SIZE_T size, SIZE_T nmemb, |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 96 | void *file) { |
| 97 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 98 | SIZE_T res = REAL(fread_unlocked)(ptr, size, nmemb, file); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 99 | if (res > 0) |
| 100 | __msan_unpoison(ptr, res *size); |
| 101 | return res; |
| 102 | } |
| 103 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 104 | INTERCEPTOR(SSIZE_T, readlink, const char *path, char *buf, SIZE_T bufsiz) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 105 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 106 | SSIZE_T res = REAL(readlink)(path, buf, bufsiz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 107 | if (res > 0) |
| 108 | __msan_unpoison(buf, res); |
| 109 | return res; |
| 110 | } |
| 111 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 112 | INTERCEPTOR(void *, memcpy, void *dest, const void *src, SIZE_T n) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 113 | return __msan_memcpy(dest, src, n); |
| 114 | } |
| 115 | |
Evgeniy Stepanov | aa23bd8 | 2013-07-02 14:49:24 +0000 | [diff] [blame] | 116 | INTERCEPTOR(void *, mempcpy, void *dest, const void *src, SIZE_T n) { |
| 117 | return (char *)__msan_memcpy(dest, src, n) + n; |
| 118 | } |
| 119 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 120 | INTERCEPTOR(void *, memmove, void *dest, const void *src, SIZE_T n) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 121 | return __msan_memmove(dest, src, n); |
| 122 | } |
| 123 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 124 | INTERCEPTOR(void *, memset, void *s, int c, SIZE_T n) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 125 | return __msan_memset(s, c, n); |
| 126 | } |
| 127 | |
Evgeniy Stepanov | db615c1 | 2013-07-04 13:19:41 +0000 | [diff] [blame] | 128 | INTERCEPTOR(void *, bcopy, const void *src, void *dest, SIZE_T n) { |
| 129 | return __msan_memmove(dest, src, n); |
| 130 | } |
| 131 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 132 | INTERCEPTOR(int, posix_memalign, void **memptr, SIZE_T alignment, SIZE_T size) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 133 | GET_MALLOC_STACK_TRACE; |
| 134 | CHECK_EQ(alignment & (alignment - 1), 0); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 135 | CHECK_NE(memptr, 0); |
Evgeniy Stepanov | 455abf4 | 2013-04-23 13:27:36 +0000 | [diff] [blame] | 136 | *memptr = MsanReallocate(&stack, 0, size, alignment, false); |
| 137 | CHECK_NE(*memptr, 0); |
| 138 | __msan_unpoison(memptr, sizeof(*memptr)); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 139 | return 0; |
| 140 | } |
| 141 | |
Evgeniy Stepanov | e3eaa7b | 2013-09-02 09:24:53 +0000 | [diff] [blame] | 142 | INTERCEPTOR(void *, memalign, SIZE_T boundary, SIZE_T size) { |
| 143 | GET_MALLOC_STACK_TRACE; |
| 144 | CHECK_EQ(boundary & (boundary - 1), 0); |
| 145 | void *ptr = MsanReallocate(&stack, 0, size, boundary, false); |
| 146 | return ptr; |
| 147 | } |
| 148 | |
| 149 | INTERCEPTOR(void *, valloc, SIZE_T size) { |
| 150 | GET_MALLOC_STACK_TRACE; |
| 151 | void *ptr = MsanReallocate(&stack, 0, size, GetPageSizeCached(), false); |
| 152 | return ptr; |
| 153 | } |
| 154 | |
| 155 | INTERCEPTOR(void *, pvalloc, SIZE_T size) { |
| 156 | GET_MALLOC_STACK_TRACE; |
| 157 | uptr PageSize = GetPageSizeCached(); |
| 158 | size = RoundUpTo(size, PageSize); |
| 159 | if (size == 0) { |
| 160 | // pvalloc(0) should allocate one page. |
| 161 | size = PageSize; |
| 162 | } |
| 163 | void *ptr = MsanReallocate(&stack, 0, size, PageSize, false); |
| 164 | return ptr; |
| 165 | } |
| 166 | |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 167 | INTERCEPTOR(void, free, void *ptr) { |
Evgeniy Stepanov | 65562f5 | 2013-09-16 11:03:31 +0000 | [diff] [blame] | 168 | GET_MALLOC_STACK_TRACE; |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 169 | if (ptr == 0) return; |
Evgeniy Stepanov | 65562f5 | 2013-09-16 11:03:31 +0000 | [diff] [blame] | 170 | MsanDeallocate(&stack, ptr); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 171 | } |
| 172 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 173 | INTERCEPTOR(SIZE_T, strlen, const char *s) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 174 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 175 | SIZE_T res = REAL(strlen)(s); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 176 | CHECK_UNPOISONED(s, res + 1); |
| 177 | return res; |
| 178 | } |
| 179 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 180 | INTERCEPTOR(SIZE_T, strnlen, const char *s, SIZE_T n) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 181 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 182 | SIZE_T res = REAL(strnlen)(s, n); |
| 183 | SIZE_T scan_size = (res == n) ? res : res + 1; |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 184 | CHECK_UNPOISONED(s, scan_size); |
| 185 | return res; |
| 186 | } |
| 187 | |
| 188 | // FIXME: Add stricter shadow checks in str* interceptors (ex.: strcpy should |
| 189 | // check the shadow of the terminating \0 byte). |
| 190 | |
| 191 | INTERCEPTOR(char *, strcpy, char *dest, const char *src) { // NOLINT |
| 192 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 193 | SIZE_T n = REAL(strlen)(src); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 194 | char *res = REAL(strcpy)(dest, src); // NOLINT |
| 195 | __msan_copy_poison(dest, src, n + 1); |
| 196 | return res; |
| 197 | } |
| 198 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 199 | INTERCEPTOR(char *, strncpy, char *dest, const char *src, SIZE_T n) { // NOLINT |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 200 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 201 | SIZE_T copy_size = REAL(strnlen)(src, n); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 202 | if (copy_size < n) |
| 203 | copy_size++; // trailing \0 |
| 204 | char *res = REAL(strncpy)(dest, src, n); // NOLINT |
| 205 | __msan_copy_poison(dest, src, copy_size); |
| 206 | return res; |
| 207 | } |
| 208 | |
Evgeniy Stepanov | 654688b | 2013-07-01 15:19:37 +0000 | [diff] [blame] | 209 | INTERCEPTOR(char *, stpcpy, char *dest, const char *src) { // NOLINT |
| 210 | ENSURE_MSAN_INITED(); |
| 211 | SIZE_T n = REAL(strlen)(src); |
| 212 | char *res = REAL(stpcpy)(dest, src); // NOLINT |
| 213 | __msan_copy_poison(dest, src, n + 1); |
| 214 | return res; |
| 215 | } |
| 216 | |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 217 | INTERCEPTOR(char *, strdup, char *src) { |
| 218 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 219 | SIZE_T n = REAL(strlen)(src); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 220 | char *res = REAL(strdup)(src); |
| 221 | __msan_copy_poison(res, src, n + 1); |
| 222 | return res; |
| 223 | } |
| 224 | |
Evgeniy Stepanov | ead6cf7 | 2013-03-14 11:10:36 +0000 | [diff] [blame] | 225 | INTERCEPTOR(char *, __strdup, char *src) { |
| 226 | ENSURE_MSAN_INITED(); |
| 227 | SIZE_T n = REAL(strlen)(src); |
| 228 | char *res = REAL(__strdup)(src); |
| 229 | __msan_copy_poison(res, src, n + 1); |
| 230 | return res; |
| 231 | } |
| 232 | |
| 233 | INTERCEPTOR(char *, strndup, char *src, SIZE_T n) { |
| 234 | ENSURE_MSAN_INITED(); |
| 235 | SIZE_T copy_size = REAL(strnlen)(src, n); |
| 236 | char *res = REAL(strndup)(src, n); |
| 237 | __msan_copy_poison(res, src, copy_size); |
| 238 | __msan_unpoison(res + copy_size, 1); // \0 |
| 239 | return res; |
| 240 | } |
| 241 | |
| 242 | INTERCEPTOR(char *, __strndup, char *src, SIZE_T n) { |
| 243 | ENSURE_MSAN_INITED(); |
| 244 | SIZE_T copy_size = REAL(strnlen)(src, n); |
| 245 | char *res = REAL(__strndup)(src, n); |
| 246 | __msan_copy_poison(res, src, copy_size); |
| 247 | __msan_unpoison(res + copy_size, 1); // \0 |
| 248 | return res; |
| 249 | } |
| 250 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 251 | INTERCEPTOR(char *, gcvt, double number, SIZE_T ndigit, char *buf) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 252 | ENSURE_MSAN_INITED(); |
| 253 | char *res = REAL(gcvt)(number, ndigit, buf); |
| 254 | // DynamoRio tool will take care of unpoisoning gcvt result for us. |
| 255 | if (!__msan_has_dynamic_component()) { |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 256 | SIZE_T n = REAL(strlen)(buf); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 257 | __msan_unpoison(buf, n + 1); |
| 258 | } |
| 259 | return res; |
| 260 | } |
| 261 | |
| 262 | INTERCEPTOR(char *, strcat, char *dest, const char *src) { // NOLINT |
| 263 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 264 | SIZE_T src_size = REAL(strlen)(src); |
| 265 | SIZE_T dest_size = REAL(strlen)(dest); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 266 | char *res = REAL(strcat)(dest, src); // NOLINT |
| 267 | __msan_copy_poison(dest + dest_size, src, src_size + 1); |
| 268 | return res; |
| 269 | } |
| 270 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 271 | INTERCEPTOR(char *, strncat, char *dest, const char *src, SIZE_T n) { // NOLINT |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 272 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 273 | SIZE_T dest_size = REAL(strlen)(dest); |
| 274 | SIZE_T copy_size = REAL(strlen)(src); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 275 | if (copy_size < n) |
| 276 | copy_size++; // trailing \0 |
| 277 | char *res = REAL(strncat)(dest, src, n); // NOLINT |
| 278 | __msan_copy_poison(dest + dest_size, src, copy_size); |
| 279 | return res; |
| 280 | } |
| 281 | |
| 282 | INTERCEPTOR(long, strtol, const char *nptr, char **endptr, // NOLINT |
| 283 | int base) { |
| 284 | ENSURE_MSAN_INITED(); |
| 285 | long res = REAL(strtol)(nptr, endptr, base); // NOLINT |
| 286 | if (!__msan_has_dynamic_component()) { |
| 287 | __msan_unpoison(endptr, sizeof(*endptr)); |
| 288 | } |
| 289 | return res; |
| 290 | } |
| 291 | |
| 292 | INTERCEPTOR(long long, strtoll, const char *nptr, char **endptr, // NOLINT |
| 293 | int base) { |
| 294 | ENSURE_MSAN_INITED(); |
| 295 | long res = REAL(strtoll)(nptr, endptr, base); //NOLINT |
| 296 | if (!__msan_has_dynamic_component()) { |
| 297 | __msan_unpoison(endptr, sizeof(*endptr)); |
| 298 | } |
| 299 | return res; |
| 300 | } |
| 301 | |
| 302 | INTERCEPTOR(unsigned long, strtoul, const char *nptr, char **endptr, // NOLINT |
| 303 | int base) { |
| 304 | ENSURE_MSAN_INITED(); |
| 305 | unsigned long res = REAL(strtoul)(nptr, endptr, base); // NOLINT |
| 306 | if (!__msan_has_dynamic_component()) { |
| 307 | __msan_unpoison(endptr, sizeof(*endptr)); |
| 308 | } |
| 309 | return res; |
| 310 | } |
| 311 | |
| 312 | INTERCEPTOR(unsigned long long, strtoull, const char *nptr, // NOLINT |
| 313 | char **endptr, int base) { |
| 314 | ENSURE_MSAN_INITED(); |
| 315 | unsigned long res = REAL(strtoull)(nptr, endptr, base); // NOLINT |
| 316 | if (!__msan_has_dynamic_component()) { |
| 317 | __msan_unpoison(endptr, sizeof(*endptr)); |
| 318 | } |
| 319 | return res; |
| 320 | } |
| 321 | |
Evgeniy Stepanov | e3a3251 | 2013-01-17 13:42:17 +0000 | [diff] [blame] | 322 | INTERCEPTOR(double, strtod, const char *nptr, char **endptr) { // NOLINT |
| 323 | ENSURE_MSAN_INITED(); |
| 324 | double res = REAL(strtod)(nptr, endptr); // NOLINT |
| 325 | if (!__msan_has_dynamic_component()) { |
| 326 | __msan_unpoison(endptr, sizeof(*endptr)); |
| 327 | } |
| 328 | return res; |
| 329 | } |
| 330 | |
| 331 | INTERCEPTOR(float, strtof, const char *nptr, char **endptr) { // NOLINT |
| 332 | ENSURE_MSAN_INITED(); |
| 333 | float res = REAL(strtof)(nptr, endptr); // NOLINT |
| 334 | if (!__msan_has_dynamic_component()) { |
| 335 | __msan_unpoison(endptr, sizeof(*endptr)); |
| 336 | } |
| 337 | return res; |
| 338 | } |
| 339 | |
| 340 | INTERCEPTOR(long double, strtold, const char *nptr, char **endptr) { // NOLINT |
| 341 | ENSURE_MSAN_INITED(); |
| 342 | long double res = REAL(strtold)(nptr, endptr); // NOLINT |
| 343 | if (!__msan_has_dynamic_component()) { |
| 344 | __msan_unpoison(endptr, sizeof(*endptr)); |
| 345 | } |
| 346 | return res; |
| 347 | } |
| 348 | |
Evgeniy Stepanov | b58b72e | 2013-06-21 13:32:26 +0000 | [diff] [blame] | 349 | INTERCEPTOR(int, vasprintf, char **strp, const char *format, va_list ap) { |
| 350 | ENSURE_MSAN_INITED(); |
| 351 | int res = REAL(vasprintf)(strp, format, ap); |
| 352 | if (res >= 0 && !__msan_has_dynamic_component()) { |
| 353 | __msan_unpoison(strp, sizeof(*strp)); |
| 354 | __msan_unpoison(*strp, res + 1); |
| 355 | } |
| 356 | return res; |
| 357 | } |
| 358 | |
| 359 | INTERCEPTOR(int, asprintf, char **strp, const char *format, ...) { // NOLINT |
| 360 | ENSURE_MSAN_INITED(); |
| 361 | va_list ap; |
| 362 | va_start(ap, format); |
| 363 | int res = vasprintf(strp, format, ap); // NOLINT |
| 364 | va_end(ap); |
| 365 | return res; |
| 366 | } |
| 367 | |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 368 | INTERCEPTOR(int, vsnprintf, char *str, uptr size, |
| 369 | const char *format, va_list ap) { |
| 370 | ENSURE_MSAN_INITED(); |
| 371 | int res = REAL(vsnprintf)(str, size, format, ap); |
Evgeniy Stepanov | b58b72e | 2013-06-21 13:32:26 +0000 | [diff] [blame] | 372 | if (res >= 0 && !__msan_has_dynamic_component()) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 373 | __msan_unpoison(str, res + 1); |
| 374 | } |
| 375 | return res; |
| 376 | } |
| 377 | |
| 378 | INTERCEPTOR(int, vsprintf, char *str, const char *format, va_list ap) { |
| 379 | ENSURE_MSAN_INITED(); |
| 380 | int res = REAL(vsprintf)(str, format, ap); |
Evgeniy Stepanov | b58b72e | 2013-06-21 13:32:26 +0000 | [diff] [blame] | 381 | if (res >= 0 && !__msan_has_dynamic_component()) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 382 | __msan_unpoison(str, res + 1); |
| 383 | } |
| 384 | return res; |
| 385 | } |
| 386 | |
| 387 | INTERCEPTOR(int, vswprintf, void *str, uptr size, void *format, va_list ap) { |
| 388 | ENSURE_MSAN_INITED(); |
| 389 | int res = REAL(vswprintf)(str, size, format, ap); |
Evgeniy Stepanov | b58b72e | 2013-06-21 13:32:26 +0000 | [diff] [blame] | 390 | if (res >= 0 && !__msan_has_dynamic_component()) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 391 | __msan_unpoison(str, 4 * (res + 1)); |
| 392 | } |
| 393 | return res; |
| 394 | } |
| 395 | |
| 396 | INTERCEPTOR(int, sprintf, char *str, const char *format, ...) { // NOLINT |
| 397 | ENSURE_MSAN_INITED(); |
| 398 | va_list ap; |
| 399 | va_start(ap, format); |
| 400 | int res = vsprintf(str, format, ap); // NOLINT |
| 401 | va_end(ap); |
| 402 | return res; |
| 403 | } |
| 404 | |
| 405 | INTERCEPTOR(int, snprintf, char *str, uptr size, const char *format, ...) { |
| 406 | ENSURE_MSAN_INITED(); |
| 407 | va_list ap; |
| 408 | va_start(ap, format); |
| 409 | int res = vsnprintf(str, size, format, ap); |
| 410 | va_end(ap); |
| 411 | return res; |
| 412 | } |
| 413 | |
| 414 | INTERCEPTOR(int, swprintf, void *str, uptr size, void *format, ...) { |
| 415 | ENSURE_MSAN_INITED(); |
| 416 | va_list ap; |
| 417 | va_start(ap, format); |
| 418 | int res = vswprintf(str, size, format, ap); |
| 419 | va_end(ap); |
| 420 | return res; |
| 421 | } |
| 422 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 423 | // SIZE_T strftime(char *s, SIZE_T max, const char *format,const struct tm *tm); |
| 424 | INTERCEPTOR(SIZE_T, strftime, char *s, SIZE_T max, const char *format, |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 425 | void *tm) { |
| 426 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 427 | SIZE_T res = REAL(strftime)(s, max, format, tm); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 428 | if (res) __msan_unpoison(s, res + 1); |
| 429 | return res; |
| 430 | } |
| 431 | |
Evgeniy Stepanov | 165d759 | 2013-07-02 13:34:44 +0000 | [diff] [blame] | 432 | INTERCEPTOR(int, mbtowc, wchar_t *dest, const char *src, SIZE_T n) { |
| 433 | ENSURE_MSAN_INITED(); |
| 434 | int res = REAL(mbtowc)(dest, src, n); |
| 435 | if (res != -1 && dest) __msan_unpoison(dest, sizeof(wchar_t)); |
| 436 | return res; |
| 437 | } |
| 438 | |
| 439 | INTERCEPTOR(int, mbrtowc, wchar_t *dest, const char *src, SIZE_T n, void *ps) { |
| 440 | ENSURE_MSAN_INITED(); |
| 441 | SIZE_T res = REAL(mbrtowc)(dest, src, n, ps); |
| 442 | if (res != (SIZE_T)-1 && dest) __msan_unpoison(dest, sizeof(wchar_t)); |
| 443 | return res; |
| 444 | } |
| 445 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 446 | INTERCEPTOR(SIZE_T, wcslen, const wchar_t *s) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 447 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 448 | SIZE_T res = REAL(wcslen)(s); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 449 | CHECK_UNPOISONED(s, sizeof(wchar_t) * (res + 1)); |
| 450 | return res; |
| 451 | } |
| 452 | |
| 453 | // wchar_t *wcschr(const wchar_t *wcs, wchar_t wc); |
| 454 | INTERCEPTOR(wchar_t *, wcschr, void *s, wchar_t wc, void *ps) { |
| 455 | ENSURE_MSAN_INITED(); |
| 456 | wchar_t *res = REAL(wcschr)(s, wc, ps); |
| 457 | return res; |
| 458 | } |
| 459 | |
| 460 | // wchar_t *wcscpy(wchar_t *dest, const wchar_t *src); |
| 461 | INTERCEPTOR(wchar_t *, wcscpy, wchar_t *dest, const wchar_t *src) { |
| 462 | ENSURE_MSAN_INITED(); |
| 463 | wchar_t *res = REAL(wcscpy)(dest, src); |
| 464 | __msan_copy_poison(dest, src, sizeof(wchar_t) * (REAL(wcslen)(src) + 1)); |
| 465 | return res; |
| 466 | } |
| 467 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 468 | // wchar_t *wmemcpy(wchar_t *dest, const wchar_t *src, SIZE_T n); |
| 469 | INTERCEPTOR(wchar_t *, wmemcpy, wchar_t *dest, const wchar_t *src, SIZE_T n) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 470 | ENSURE_MSAN_INITED(); |
| 471 | wchar_t *res = REAL(wmemcpy)(dest, src, n); |
| 472 | __msan_copy_poison(dest, src, n * sizeof(wchar_t)); |
| 473 | return res; |
| 474 | } |
| 475 | |
Evgeniy Stepanov | aa23bd8 | 2013-07-02 14:49:24 +0000 | [diff] [blame] | 476 | INTERCEPTOR(wchar_t *, wmempcpy, wchar_t *dest, const wchar_t *src, SIZE_T n) { |
| 477 | ENSURE_MSAN_INITED(); |
| 478 | wchar_t *res = REAL(wmempcpy)(dest, src, n); |
| 479 | __msan_copy_poison(dest, src, n * sizeof(wchar_t)); |
| 480 | return res; |
| 481 | } |
| 482 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 483 | INTERCEPTOR(wchar_t *, wmemset, wchar_t *s, wchar_t c, SIZE_T n) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 484 | CHECK(MEM_IS_APP(s)); |
| 485 | ENSURE_MSAN_INITED(); |
| 486 | wchar_t *res = (wchar_t *)fast_memset(s, c, n * sizeof(wchar_t)); |
| 487 | __msan_unpoison(s, n * sizeof(wchar_t)); |
| 488 | return res; |
| 489 | } |
| 490 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 491 | INTERCEPTOR(wchar_t *, wmemmove, wchar_t *dest, const wchar_t *src, SIZE_T n) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 492 | ENSURE_MSAN_INITED(); |
| 493 | wchar_t *res = REAL(wmemmove)(dest, src, n); |
| 494 | __msan_move_poison(dest, src, n * sizeof(wchar_t)); |
| 495 | return res; |
| 496 | } |
| 497 | |
| 498 | INTERCEPTOR(int, wcscmp, const wchar_t *s1, const wchar_t *s2) { |
| 499 | ENSURE_MSAN_INITED(); |
| 500 | int res = REAL(wcscmp)(s1, s2); |
| 501 | return res; |
| 502 | } |
| 503 | |
| 504 | INTERCEPTOR(double, wcstod, const wchar_t *nptr, wchar_t **endptr) { |
| 505 | ENSURE_MSAN_INITED(); |
| 506 | double res = REAL(wcstod)(nptr, endptr); |
| 507 | __msan_unpoison(endptr, sizeof(*endptr)); |
| 508 | return res; |
| 509 | } |
| 510 | |
| 511 | // #define UNSUPPORTED(name) \ |
| 512 | // INTERCEPTOR(void, name, void) { \ |
| 513 | // Printf("MSAN: Unsupported %s\n", __FUNCTION__); \ |
| 514 | // Die(); \ |
| 515 | // } |
| 516 | |
| 517 | // FIXME: intercept the following functions: |
| 518 | // Note, they only matter when running without a dynamic tool. |
| 519 | // UNSUPPORTED(wcscoll_l) |
| 520 | // UNSUPPORTED(wcsnrtombs) |
| 521 | // UNSUPPORTED(wcstol) |
| 522 | // UNSUPPORTED(wcstoll) |
| 523 | // UNSUPPORTED(wcstold) |
| 524 | // UNSUPPORTED(wcstoul) |
| 525 | // UNSUPPORTED(wcstoull) |
| 526 | // UNSUPPORTED(wcsxfrm_l) |
| 527 | // UNSUPPORTED(wcsdup) |
| 528 | // UNSUPPORTED(wcsftime) |
| 529 | // UNSUPPORTED(wcsstr) |
| 530 | // UNSUPPORTED(wcsrchr) |
| 531 | // UNSUPPORTED(wctob) |
| 532 | |
| 533 | INTERCEPTOR(int, gettimeofday, void *tv, void *tz) { |
| 534 | ENSURE_MSAN_INITED(); |
| 535 | int res = REAL(gettimeofday)(tv, tz); |
| 536 | if (tv) |
| 537 | __msan_unpoison(tv, 16); |
| 538 | if (tz) |
| 539 | __msan_unpoison(tz, 8); |
| 540 | return res; |
| 541 | } |
| 542 | |
| 543 | INTERCEPTOR(char *, fcvt, double x, int a, int *b, int *c) { |
| 544 | ENSURE_MSAN_INITED(); |
| 545 | char *res = REAL(fcvt)(x, a, b, c); |
| 546 | if (!__msan_has_dynamic_component()) { |
| 547 | __msan_unpoison(b, sizeof(*b)); |
| 548 | __msan_unpoison(c, sizeof(*c)); |
| 549 | } |
| 550 | return res; |
| 551 | } |
| 552 | |
| 553 | INTERCEPTOR(char *, getenv, char *name) { |
| 554 | ENSURE_MSAN_INITED(); |
| 555 | char *res = REAL(getenv)(name); |
| 556 | if (!__msan_has_dynamic_component()) { |
| 557 | if (res) |
| 558 | __msan_unpoison(res, REAL(strlen)(res) + 1); |
| 559 | } |
| 560 | return res; |
| 561 | } |
| 562 | |
Evgeniy Stepanov | 5dbedf6 | 2013-08-01 11:14:14 +0000 | [diff] [blame] | 563 | extern char **environ; |
| 564 | |
| 565 | static void UnpoisonEnviron() { |
| 566 | char **envp = environ; |
| 567 | for (; *envp; ++envp) { |
| 568 | __msan_unpoison(envp, sizeof(*envp)); |
| 569 | __msan_unpoison(*envp, REAL(strlen)(*envp) + 1); |
| 570 | } |
| 571 | // Trailing NULL pointer. |
| 572 | __msan_unpoison(envp, sizeof(*envp)); |
| 573 | } |
| 574 | |
| 575 | INTERCEPTOR(int, setenv, const char *name, const char *value, int overwrite) { |
| 576 | ENSURE_MSAN_INITED(); |
| 577 | int res = REAL(setenv)(name, value, overwrite); |
| 578 | if (!res) UnpoisonEnviron(); |
| 579 | return res; |
| 580 | } |
| 581 | |
| 582 | INTERCEPTOR(int, putenv, char *string) { |
| 583 | ENSURE_MSAN_INITED(); |
| 584 | int res = REAL(putenv)(string); |
| 585 | if (!res) UnpoisonEnviron(); |
| 586 | return res; |
| 587 | } |
| 588 | |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 589 | INTERCEPTOR(int, __fxstat, int magic, int fd, void *buf) { |
| 590 | ENSURE_MSAN_INITED(); |
| 591 | int res = REAL(__fxstat)(magic, fd, buf); |
| 592 | if (!res) |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 593 | __msan_unpoison(buf, __sanitizer::struct_stat_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 594 | return res; |
| 595 | } |
| 596 | |
| 597 | INTERCEPTOR(int, __fxstat64, int magic, int fd, void *buf) { |
| 598 | ENSURE_MSAN_INITED(); |
| 599 | int res = REAL(__fxstat64)(magic, fd, buf); |
| 600 | if (!res) |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 601 | __msan_unpoison(buf, __sanitizer::struct_stat64_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 602 | return res; |
| 603 | } |
| 604 | |
Evgeniy Stepanov | 842fa3f | 2013-09-09 13:40:41 +0000 | [diff] [blame] | 605 | INTERCEPTOR(int, __fxstatat, int magic, int fd, char *pathname, void *buf, |
| 606 | int flags) { |
| 607 | ENSURE_MSAN_INITED(); |
| 608 | int res = REAL(__fxstatat)(magic, fd, pathname, buf, flags); |
| 609 | if (!res) __msan_unpoison(buf, __sanitizer::struct_stat_sz); |
| 610 | return res; |
| 611 | } |
| 612 | |
| 613 | INTERCEPTOR(int, __fxstatat64, int magic, int fd, char *pathname, void *buf, |
| 614 | int flags) { |
| 615 | ENSURE_MSAN_INITED(); |
| 616 | int res = REAL(__fxstatat64)(magic, fd, pathname, buf, flags); |
| 617 | if (!res) __msan_unpoison(buf, __sanitizer::struct_stat64_sz); |
| 618 | return res; |
| 619 | } |
| 620 | |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 621 | INTERCEPTOR(int, __xstat, int magic, char *path, void *buf) { |
| 622 | ENSURE_MSAN_INITED(); |
| 623 | int res = REAL(__xstat)(magic, path, buf); |
| 624 | if (!res) |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 625 | __msan_unpoison(buf, __sanitizer::struct_stat_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 626 | return res; |
| 627 | } |
| 628 | |
| 629 | INTERCEPTOR(int, __xstat64, int magic, char *path, void *buf) { |
| 630 | ENSURE_MSAN_INITED(); |
| 631 | int res = REAL(__xstat64)(magic, path, buf); |
| 632 | if (!res) |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 633 | __msan_unpoison(buf, __sanitizer::struct_stat64_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 634 | return res; |
| 635 | } |
| 636 | |
| 637 | INTERCEPTOR(int, __lxstat, int magic, char *path, void *buf) { |
| 638 | ENSURE_MSAN_INITED(); |
| 639 | int res = REAL(__lxstat)(magic, path, buf); |
| 640 | if (!res) |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 641 | __msan_unpoison(buf, __sanitizer::struct_stat_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 642 | return res; |
| 643 | } |
| 644 | |
| 645 | INTERCEPTOR(int, __lxstat64, int magic, char *path, void *buf) { |
| 646 | ENSURE_MSAN_INITED(); |
| 647 | int res = REAL(__lxstat64)(magic, path, buf); |
| 648 | if (!res) |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 649 | __msan_unpoison(buf, __sanitizer::struct_stat64_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 650 | return res; |
| 651 | } |
| 652 | |
| 653 | INTERCEPTOR(int, pipe, int pipefd[2]) { |
| 654 | if (msan_init_is_running) |
| 655 | return REAL(pipe)(pipefd); |
| 656 | ENSURE_MSAN_INITED(); |
| 657 | int res = REAL(pipe)(pipefd); |
| 658 | if (!res) |
| 659 | __msan_unpoison(pipefd, sizeof(int[2])); |
| 660 | return res; |
| 661 | } |
| 662 | |
Evgeniy Stepanov | b4eac2f | 2013-04-08 13:45:12 +0000 | [diff] [blame] | 663 | INTERCEPTOR(int, pipe2, int pipefd[2], int flags) { |
| 664 | ENSURE_MSAN_INITED(); |
| 665 | int res = REAL(pipe2)(pipefd, flags); |
| 666 | if (!res) |
| 667 | __msan_unpoison(pipefd, sizeof(int[2])); |
| 668 | return res; |
| 669 | } |
| 670 | |
| 671 | INTERCEPTOR(int, socketpair, int domain, int type, int protocol, int sv[2]) { |
| 672 | ENSURE_MSAN_INITED(); |
| 673 | int res = REAL(socketpair)(domain, type, protocol, sv); |
| 674 | if (!res) |
| 675 | __msan_unpoison(sv, sizeof(int[2])); |
| 676 | return res; |
| 677 | } |
| 678 | |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 679 | INTERCEPTOR(char *, fgets, char *s, int size, void *stream) { |
| 680 | ENSURE_MSAN_INITED(); |
| 681 | char *res = REAL(fgets)(s, size, stream); |
| 682 | if (res) |
| 683 | __msan_unpoison(s, REAL(strlen)(s) + 1); |
| 684 | return res; |
| 685 | } |
| 686 | |
| 687 | INTERCEPTOR(char *, fgets_unlocked, char *s, int size, void *stream) { |
| 688 | ENSURE_MSAN_INITED(); |
| 689 | char *res = REAL(fgets_unlocked)(s, size, stream); |
| 690 | if (res) |
| 691 | __msan_unpoison(s, REAL(strlen)(s) + 1); |
| 692 | return res; |
| 693 | } |
| 694 | |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 695 | INTERCEPTOR(int, getrlimit, int resource, void *rlim) { |
| 696 | if (msan_init_is_running) |
| 697 | return REAL(getrlimit)(resource, rlim); |
| 698 | ENSURE_MSAN_INITED(); |
| 699 | int res = REAL(getrlimit)(resource, rlim); |
| 700 | if (!res) |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 701 | __msan_unpoison(rlim, __sanitizer::struct_rlimit_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 702 | return res; |
| 703 | } |
| 704 | |
| 705 | INTERCEPTOR(int, getrlimit64, int resource, void *rlim) { |
| 706 | if (msan_init_is_running) |
| 707 | return REAL(getrlimit64)(resource, rlim); |
| 708 | ENSURE_MSAN_INITED(); |
| 709 | int res = REAL(getrlimit64)(resource, rlim); |
| 710 | if (!res) |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 711 | __msan_unpoison(rlim, __sanitizer::struct_rlimit64_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 712 | return res; |
| 713 | } |
| 714 | |
| 715 | INTERCEPTOR(int, statfs, const char *s, void *buf) { |
| 716 | ENSURE_MSAN_INITED(); |
| 717 | int res = REAL(statfs)(s, buf); |
| 718 | if (!res) |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 719 | __msan_unpoison(buf, __sanitizer::struct_statfs_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 720 | return res; |
| 721 | } |
| 722 | |
| 723 | INTERCEPTOR(int, fstatfs, int fd, void *buf) { |
| 724 | ENSURE_MSAN_INITED(); |
| 725 | int res = REAL(fstatfs)(fd, buf); |
| 726 | if (!res) |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 727 | __msan_unpoison(buf, __sanitizer::struct_statfs_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 728 | return res; |
| 729 | } |
| 730 | |
| 731 | INTERCEPTOR(int, statfs64, const char *s, void *buf) { |
| 732 | ENSURE_MSAN_INITED(); |
| 733 | int res = REAL(statfs64)(s, buf); |
| 734 | if (!res) |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 735 | __msan_unpoison(buf, __sanitizer::struct_statfs64_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 736 | return res; |
| 737 | } |
| 738 | |
| 739 | INTERCEPTOR(int, fstatfs64, int fd, void *buf) { |
| 740 | ENSURE_MSAN_INITED(); |
| 741 | int res = REAL(fstatfs64)(fd, buf); |
| 742 | if (!res) |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 743 | __msan_unpoison(buf, __sanitizer::struct_statfs64_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 744 | return res; |
| 745 | } |
| 746 | |
| 747 | INTERCEPTOR(int, uname, void *utsname) { |
| 748 | ENSURE_MSAN_INITED(); |
| 749 | int res = REAL(uname)(utsname); |
| 750 | if (!res) { |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 751 | __msan_unpoison(utsname, __sanitizer::struct_utsname_sz); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 752 | } |
| 753 | return res; |
| 754 | } |
| 755 | |
Evgeniy Stepanov | ffffbef | 2013-01-23 10:43:38 +0000 | [diff] [blame] | 756 | INTERCEPTOR(int, gethostname, char *name, SIZE_T len) { |
| 757 | ENSURE_MSAN_INITED(); |
| 758 | int res = REAL(gethostname)(name, len); |
| 759 | if (!res) { |
| 760 | SIZE_T real_len = REAL(strnlen)(name, len); |
| 761 | if (real_len < len) |
| 762 | ++real_len; |
| 763 | __msan_unpoison(name, real_len); |
| 764 | } |
| 765 | return res; |
| 766 | } |
| 767 | |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 768 | INTERCEPTOR(int, epoll_wait, int epfd, void *events, int maxevents, |
| 769 | int timeout) { |
| 770 | ENSURE_MSAN_INITED(); |
| 771 | int res = REAL(epoll_wait)(epfd, events, maxevents, timeout); |
| 772 | if (res > 0) { |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 773 | __msan_unpoison(events, __sanitizer::struct_epoll_event_sz * res); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 774 | } |
| 775 | return res; |
| 776 | } |
| 777 | |
| 778 | INTERCEPTOR(int, epoll_pwait, int epfd, void *events, int maxevents, |
| 779 | int timeout, void *sigmask) { |
| 780 | ENSURE_MSAN_INITED(); |
| 781 | int res = REAL(epoll_pwait)(epfd, events, maxevents, timeout, sigmask); |
| 782 | if (res > 0) { |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 783 | __msan_unpoison(events, __sanitizer::struct_epoll_event_sz * res); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 784 | } |
| 785 | return res; |
| 786 | } |
| 787 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 788 | INTERCEPTOR(SSIZE_T, recv, int fd, void *buf, SIZE_T len, int flags) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 789 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 790 | SSIZE_T res = REAL(recv)(fd, buf, len, flags); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 791 | if (res > 0) |
| 792 | __msan_unpoison(buf, res); |
| 793 | return res; |
| 794 | } |
| 795 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 796 | INTERCEPTOR(SSIZE_T, recvfrom, int fd, void *buf, SIZE_T len, int flags, |
Evgeniy Stepanov | a4d08c4 | 2013-05-29 11:30:00 +0000 | [diff] [blame] | 797 | void *srcaddr, int *addrlen) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 798 | ENSURE_MSAN_INITED(); |
Evgeniy Stepanov | 84f46d9 | 2012-12-25 16:51:57 +0000 | [diff] [blame] | 799 | SIZE_T srcaddr_sz; |
Evgeniy Stepanov | a4d08c4 | 2013-05-29 11:30:00 +0000 | [diff] [blame] | 800 | if (srcaddr) srcaddr_sz = *addrlen; |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 801 | SSIZE_T res = REAL(recvfrom)(fd, buf, len, flags, srcaddr, addrlen); |
Evgeniy Stepanov | 84f46d9 | 2012-12-25 16:51:57 +0000 | [diff] [blame] | 802 | if (res > 0) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 803 | __msan_unpoison(buf, res); |
Evgeniy Stepanov | 84f46d9 | 2012-12-25 16:51:57 +0000 | [diff] [blame] | 804 | if (srcaddr) { |
Evgeniy Stepanov | a4d08c4 | 2013-05-29 11:30:00 +0000 | [diff] [blame] | 805 | SIZE_T sz = *addrlen; |
Evgeniy Stepanov | 84f46d9 | 2012-12-25 16:51:57 +0000 | [diff] [blame] | 806 | __msan_unpoison(srcaddr, (sz < srcaddr_sz) ? sz : srcaddr_sz); |
| 807 | } |
| 808 | } |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 809 | return res; |
| 810 | } |
| 811 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 812 | INTERCEPTOR(void *, calloc, SIZE_T nmemb, SIZE_T size) { |
Kostya Serebryany | 6c5b034 | 2013-09-06 10:58:55 +0000 | [diff] [blame] | 813 | if (CallocShouldReturnNullDueToOverflow(size, nmemb)) |
| 814 | return AllocatorReturnNull(); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 815 | GET_MALLOC_STACK_TRACE; |
| 816 | if (!msan_inited) { |
| 817 | // Hack: dlsym calls calloc before REAL(calloc) is retrieved from dlsym. |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 818 | const SIZE_T kCallocPoolSize = 1024; |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 819 | static uptr calloc_memory_for_dlsym[kCallocPoolSize]; |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 820 | static SIZE_T allocated; |
| 821 | SIZE_T size_in_words = ((nmemb * size) + kWordSize - 1) / kWordSize; |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 822 | void *mem = (void*)&calloc_memory_for_dlsym[allocated]; |
| 823 | allocated += size_in_words; |
| 824 | CHECK(allocated < kCallocPoolSize); |
| 825 | return mem; |
| 826 | } |
| 827 | return MsanReallocate(&stack, 0, nmemb * size, sizeof(u64), true); |
| 828 | } |
| 829 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 830 | INTERCEPTOR(void *, realloc, void *ptr, SIZE_T size) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 831 | GET_MALLOC_STACK_TRACE; |
| 832 | return MsanReallocate(&stack, ptr, size, sizeof(u64), false); |
| 833 | } |
| 834 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 835 | INTERCEPTOR(void *, malloc, SIZE_T size) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 836 | GET_MALLOC_STACK_TRACE; |
| 837 | return MsanReallocate(&stack, 0, size, sizeof(u64), false); |
| 838 | } |
| 839 | |
Alexey Samsonov | 34093f0 | 2013-04-23 13:34:19 +0000 | [diff] [blame] | 840 | void __msan_allocated_memory(const void* data, uptr size) { |
Evgeniy Stepanov | 9f7e8c1 | 2013-01-28 13:52:49 +0000 | [diff] [blame] | 841 | GET_MALLOC_STACK_TRACE; |
| 842 | if (flags()->poison_in_malloc) |
| 843 | __msan_poison(data, size); |
| 844 | if (__msan_get_track_origins()) { |
| 845 | u32 stack_id = StackDepotPut(stack.trace, stack.size); |
| 846 | CHECK(stack_id); |
| 847 | CHECK_EQ((stack_id >> 31), 0); // Higher bit is occupied by stack origins. |
| 848 | __msan_set_origin(data, size, stack_id); |
| 849 | } |
| 850 | } |
| 851 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 852 | INTERCEPTOR(void *, mmap, void *addr, SIZE_T length, int prot, int flags, |
| 853 | int fd, OFF_T offset) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 854 | ENSURE_MSAN_INITED(); |
| 855 | void *res = REAL(mmap)(addr, length, prot, flags, fd, offset); |
| 856 | if (res != (void*)-1) |
| 857 | __msan_unpoison(res, RoundUpTo(length, GetPageSize())); |
| 858 | return res; |
| 859 | } |
| 860 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 861 | INTERCEPTOR(void *, mmap64, void *addr, SIZE_T length, int prot, int flags, |
| 862 | int fd, OFF64_T offset) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 863 | ENSURE_MSAN_INITED(); |
| 864 | void *res = REAL(mmap64)(addr, length, prot, flags, fd, offset); |
| 865 | if (res != (void*)-1) |
| 866 | __msan_unpoison(res, RoundUpTo(length, GetPageSize())); |
| 867 | return res; |
| 868 | } |
| 869 | |
Evgeniy Stepanov | e3a3251 | 2013-01-17 13:42:17 +0000 | [diff] [blame] | 870 | struct dlinfo { |
| 871 | char *dli_fname; |
| 872 | void *dli_fbase; |
| 873 | char *dli_sname; |
| 874 | void *dli_saddr; |
| 875 | }; |
| 876 | |
| 877 | INTERCEPTOR(int, dladdr, void *addr, dlinfo *info) { |
| 878 | ENSURE_MSAN_INITED(); |
| 879 | int res = REAL(dladdr)(addr, info); |
| 880 | if (res != 0) { |
| 881 | __msan_unpoison(info, sizeof(*info)); |
| 882 | if (info->dli_fname) |
| 883 | __msan_unpoison(info->dli_fname, REAL(strlen)(info->dli_fname) + 1); |
| 884 | if (info->dli_sname) |
| 885 | __msan_unpoison(info->dli_sname, REAL(strlen)(info->dli_sname) + 1); |
| 886 | } |
| 887 | return res; |
| 888 | } |
| 889 | |
Reid Kleckner | c9d382b | 2013-03-11 18:07:42 +0000 | [diff] [blame] | 890 | // dlopen() ultimately calls mmap() down inside the loader, which generally |
| 891 | // doesn't participate in dynamic symbol resolution. Therefore we won't |
| 892 | // intercept its calls to mmap, and we have to hook it here. The loader |
| 893 | // initializes the module before returning, so without the dynamic component, we |
| 894 | // won't be able to clear the shadow before the initializers. Fixing this would |
| 895 | // require putting our own initializer first to clear the shadow. |
| 896 | INTERCEPTOR(void *, dlopen, const char *filename, int flag) { |
| 897 | ENSURE_MSAN_INITED(); |
| 898 | EnterLoader(); |
| 899 | link_map *map = (link_map *)REAL(dlopen)(filename, flag); |
| 900 | ExitLoader(); |
Evgeniy Stepanov | ecbf646 | 2013-03-22 11:59:49 +0000 | [diff] [blame] | 901 | if (!__msan_has_dynamic_component() && map) { |
Reid Kleckner | c9d382b | 2013-03-11 18:07:42 +0000 | [diff] [blame] | 902 | // If msandr didn't clear the shadow before the initializers ran, we do it |
| 903 | // ourselves afterwards. |
Peter Collingbourne | 9058c07 | 2013-07-29 19:09:49 +0000 | [diff] [blame] | 904 | ForEachMappedRegion(map, __msan_unpoison); |
Reid Kleckner | c9d382b | 2013-03-11 18:07:42 +0000 | [diff] [blame] | 905 | } |
| 906 | return (void *)map; |
| 907 | } |
| 908 | |
Evgeniy Stepanov | ff6728f | 2013-05-17 12:51:13 +0000 | [diff] [blame] | 909 | typedef int (*dl_iterate_phdr_cb)(__sanitizer_dl_phdr_info *info, SIZE_T size, |
| 910 | void *data); |
Evgeniy Stepanov | 87c6ec7 | 2013-05-16 13:00:25 +0000 | [diff] [blame] | 911 | struct dl_iterate_phdr_data { |
| 912 | dl_iterate_phdr_cb callback; |
| 913 | void *data; |
| 914 | }; |
| 915 | |
Evgeniy Stepanov | ff6728f | 2013-05-17 12:51:13 +0000 | [diff] [blame] | 916 | static int msan_dl_iterate_phdr_cb(__sanitizer_dl_phdr_info *info, SIZE_T size, |
| 917 | void *data) { |
| 918 | if (info) { |
Evgeniy Stepanov | 87c6ec7 | 2013-05-16 13:00:25 +0000 | [diff] [blame] | 919 | __msan_unpoison(info, size); |
Evgeniy Stepanov | ff6728f | 2013-05-17 12:51:13 +0000 | [diff] [blame] | 920 | if (info->dlpi_name) |
| 921 | __msan_unpoison(info->dlpi_name, REAL(strlen)(info->dlpi_name) + 1); |
| 922 | } |
Evgeniy Stepanov | 87c6ec7 | 2013-05-16 13:00:25 +0000 | [diff] [blame] | 923 | dl_iterate_phdr_data *cbdata = (dl_iterate_phdr_data *)data; |
Alexey Samsonov | ed4594b | 2013-06-27 07:50:56 +0000 | [diff] [blame] | 924 | UnpoisonParam(3); |
Evgeniy Stepanov | 87c6ec7 | 2013-05-16 13:00:25 +0000 | [diff] [blame] | 925 | return cbdata->callback(info, size, cbdata->data); |
| 926 | } |
| 927 | |
| 928 | INTERCEPTOR(int, dl_iterate_phdr, dl_iterate_phdr_cb callback, void *data) { |
| 929 | ENSURE_MSAN_INITED(); |
| 930 | EnterLoader(); |
| 931 | dl_iterate_phdr_data cbdata; |
| 932 | cbdata.callback = callback; |
| 933 | cbdata.data = data; |
| 934 | int res = REAL(dl_iterate_phdr)(msan_dl_iterate_phdr_cb, (void *)&cbdata); |
| 935 | ExitLoader(); |
| 936 | return res; |
| 937 | } |
| 938 | |
Evgeniy Stepanov | e3a3251 | 2013-01-17 13:42:17 +0000 | [diff] [blame] | 939 | INTERCEPTOR(int, getrusage, int who, void *usage) { |
| 940 | ENSURE_MSAN_INITED(); |
| 941 | int res = REAL(getrusage)(who, usage); |
| 942 | if (res == 0) { |
Evgeniy Stepanov | ea61d08 | 2013-02-19 09:19:16 +0000 | [diff] [blame] | 943 | __msan_unpoison(usage, __sanitizer::struct_rusage_sz); |
Evgeniy Stepanov | e3a3251 | 2013-01-17 13:42:17 +0000 | [diff] [blame] | 944 | } |
| 945 | return res; |
| 946 | } |
| 947 | |
Evgeniy Stepanov | bb25af5 | 2013-08-27 11:34:05 +0000 | [diff] [blame] | 948 | // sigactions_mu guarantees atomicity of sigaction() and signal() calls. |
| 949 | // Access to sigactions[] is gone with relaxed atomics to avoid data race with |
| 950 | // the signal handler. |
Evgeniy Stepanov | 1f8b353 | 2013-04-04 08:22:52 +0000 | [diff] [blame] | 951 | const int kMaxSignals = 1024; |
Evgeniy Stepanov | bb25af5 | 2013-08-27 11:34:05 +0000 | [diff] [blame] | 952 | static atomic_uintptr_t sigactions[kMaxSignals]; |
Evgeniy Stepanov | 1f8b353 | 2013-04-04 08:22:52 +0000 | [diff] [blame] | 953 | static StaticSpinMutex sigactions_mu; |
| 954 | |
| 955 | static void SignalHandler(int signo) { |
Evgeniy Stepanov | cb22c67 | 2013-08-27 14:08:15 +0000 | [diff] [blame] | 956 | ScopedThreadLocalStateBackup stlsb; |
Evgeniy Stepanov | c417981 | 2013-08-27 12:59:39 +0000 | [diff] [blame] | 957 | UnpoisonParam(1); |
| 958 | |
Kostya Serebryany | 5b4267f | 2013-04-05 14:40:25 +0000 | [diff] [blame] | 959 | typedef void (*signal_cb)(int x); |
Evgeniy Stepanov | bb25af5 | 2013-08-27 11:34:05 +0000 | [diff] [blame] | 960 | signal_cb cb = |
| 961 | (signal_cb)atomic_load(&sigactions[signo], memory_order_relaxed); |
Evgeniy Stepanov | 1f8b353 | 2013-04-04 08:22:52 +0000 | [diff] [blame] | 962 | cb(signo); |
| 963 | } |
| 964 | |
| 965 | static void SignalAction(int signo, void *si, void *uc) { |
Evgeniy Stepanov | cb22c67 | 2013-08-27 14:08:15 +0000 | [diff] [blame] | 966 | ScopedThreadLocalStateBackup stlsb; |
Alexey Samsonov | ed4594b | 2013-06-27 07:50:56 +0000 | [diff] [blame] | 967 | UnpoisonParam(3); |
Evgeniy Stepanov | a40a058 | 2013-08-27 11:10:04 +0000 | [diff] [blame] | 968 | __msan_unpoison(si, sizeof(__sanitizer_sigaction)); |
Evgeniy Stepanov | 7b0e132 | 2013-04-04 09:03:56 +0000 | [diff] [blame] | 969 | __msan_unpoison(uc, __sanitizer::ucontext_t_sz); |
Evgeniy Stepanov | 1f8b353 | 2013-04-04 08:22:52 +0000 | [diff] [blame] | 970 | |
| 971 | typedef void (*sigaction_cb)(int, void *, void *); |
Evgeniy Stepanov | bb25af5 | 2013-08-27 11:34:05 +0000 | [diff] [blame] | 972 | sigaction_cb cb = |
| 973 | (sigaction_cb)atomic_load(&sigactions[signo], memory_order_relaxed); |
Evgeniy Stepanov | 1f8b353 | 2013-04-04 08:22:52 +0000 | [diff] [blame] | 974 | cb(signo, si, uc); |
| 975 | } |
| 976 | |
| 977 | INTERCEPTOR(int, sigaction, int signo, const __sanitizer_sigaction *act, |
| 978 | __sanitizer_sigaction *oldact) { |
Evgeniy Stepanov | f8213b4 | 2013-04-01 14:47:21 +0000 | [diff] [blame] | 979 | ENSURE_MSAN_INITED(); |
Kostya Serebryany | c3525ce | 2013-04-04 06:52:40 +0000 | [diff] [blame] | 980 | // FIXME: check that *act is unpoisoned. |
Evgeniy Stepanov | f8213b4 | 2013-04-01 14:47:21 +0000 | [diff] [blame] | 981 | // That requires intercepting all of sigemptyset, sigfillset, etc. |
Evgeniy Stepanov | 7948c64 | 2013-04-05 11:59:16 +0000 | [diff] [blame] | 982 | int res; |
| 983 | if (flags()->wrap_signals) { |
| 984 | SpinMutexLock lock(&sigactions_mu); |
| 985 | CHECK_LT(signo, kMaxSignals); |
Evgeniy Stepanov | bb25af5 | 2013-08-27 11:34:05 +0000 | [diff] [blame] | 986 | uptr old_cb = atomic_load(&sigactions[signo], memory_order_relaxed); |
Evgeniy Stepanov | 7948c64 | 2013-04-05 11:59:16 +0000 | [diff] [blame] | 987 | __sanitizer_sigaction new_act; |
| 988 | __sanitizer_sigaction *pnew_act = act ? &new_act : 0; |
| 989 | if (act) { |
Evgeniy Stepanov | a40a058 | 2013-08-27 11:10:04 +0000 | [diff] [blame] | 990 | internal_memcpy(pnew_act, act, sizeof(__sanitizer_sigaction)); |
| 991 | uptr cb = (uptr)pnew_act->sa_sigaction; |
| 992 | uptr new_cb = (pnew_act->sa_flags & __sanitizer::sa_siginfo) |
| 993 | ? (uptr)SignalAction |
| 994 | : (uptr)SignalHandler; |
Evgeniy Stepanov | 7948c64 | 2013-04-05 11:59:16 +0000 | [diff] [blame] | 995 | if (cb != __sanitizer::sig_ign && cb != __sanitizer::sig_dfl) { |
Evgeniy Stepanov | bb25af5 | 2013-08-27 11:34:05 +0000 | [diff] [blame] | 996 | atomic_store(&sigactions[signo], cb, memory_order_relaxed); |
Evgeniy Stepanov | a40a058 | 2013-08-27 11:10:04 +0000 | [diff] [blame] | 997 | pnew_act->sa_sigaction = (void (*)(int, void *, void *))new_cb; |
Evgeniy Stepanov | 7948c64 | 2013-04-05 11:59:16 +0000 | [diff] [blame] | 998 | } |
Evgeniy Stepanov | 1f8b353 | 2013-04-04 08:22:52 +0000 | [diff] [blame] | 999 | } |
Evgeniy Stepanov | 7948c64 | 2013-04-05 11:59:16 +0000 | [diff] [blame] | 1000 | res = REAL(sigaction)(signo, pnew_act, oldact); |
| 1001 | if (res == 0 && oldact) { |
Evgeniy Stepanov | a40a058 | 2013-08-27 11:10:04 +0000 | [diff] [blame] | 1002 | uptr cb = (uptr)oldact->sa_sigaction; |
Evgeniy Stepanov | e7a7a9b | 2013-04-05 12:58:07 +0000 | [diff] [blame] | 1003 | if (cb != __sanitizer::sig_ign && cb != __sanitizer::sig_dfl) { |
Evgeniy Stepanov | a40a058 | 2013-08-27 11:10:04 +0000 | [diff] [blame] | 1004 | oldact->sa_sigaction = (void (*)(int, void *, void *))old_cb; |
Evgeniy Stepanov | e7a7a9b | 2013-04-05 12:58:07 +0000 | [diff] [blame] | 1005 | } |
Evgeniy Stepanov | 7948c64 | 2013-04-05 11:59:16 +0000 | [diff] [blame] | 1006 | } |
| 1007 | } else { |
| 1008 | res = REAL(sigaction)(signo, act, oldact); |
Evgeniy Stepanov | 1f8b353 | 2013-04-04 08:22:52 +0000 | [diff] [blame] | 1009 | } |
Evgeniy Stepanov | 7948c64 | 2013-04-05 11:59:16 +0000 | [diff] [blame] | 1010 | |
Evgeniy Stepanov | 1f8b353 | 2013-04-04 08:22:52 +0000 | [diff] [blame] | 1011 | if (res == 0 && oldact) { |
Evgeniy Stepanov | a40a058 | 2013-08-27 11:10:04 +0000 | [diff] [blame] | 1012 | __msan_unpoison(oldact, sizeof(__sanitizer_sigaction)); |
Evgeniy Stepanov | f8213b4 | 2013-04-01 14:47:21 +0000 | [diff] [blame] | 1013 | } |
| 1014 | return res; |
| 1015 | } |
| 1016 | |
Evgeniy Stepanov | 1f8b353 | 2013-04-04 08:22:52 +0000 | [diff] [blame] | 1017 | INTERCEPTOR(int, signal, int signo, uptr cb) { |
| 1018 | ENSURE_MSAN_INITED(); |
Evgeniy Stepanov | 7948c64 | 2013-04-05 11:59:16 +0000 | [diff] [blame] | 1019 | if (flags()->wrap_signals) { |
| 1020 | CHECK_LT(signo, kMaxSignals); |
| 1021 | SpinMutexLock lock(&sigactions_mu); |
| 1022 | if (cb != __sanitizer::sig_ign && cb != __sanitizer::sig_dfl) { |
Evgeniy Stepanov | bb25af5 | 2013-08-27 11:34:05 +0000 | [diff] [blame] | 1023 | atomic_store(&sigactions[signo], cb, memory_order_relaxed); |
Evgeniy Stepanov | 7948c64 | 2013-04-05 11:59:16 +0000 | [diff] [blame] | 1024 | cb = (uptr) SignalHandler; |
| 1025 | } |
| 1026 | return REAL(signal)(signo, cb); |
| 1027 | } else { |
| 1028 | return REAL(signal)(signo, cb); |
Evgeniy Stepanov | 1f8b353 | 2013-04-04 08:22:52 +0000 | [diff] [blame] | 1029 | } |
Evgeniy Stepanov | 1f8b353 | 2013-04-04 08:22:52 +0000 | [diff] [blame] | 1030 | } |
| 1031 | |
Evgeniy Stepanov | 98f5ea0 | 2013-03-13 09:01:40 +0000 | [diff] [blame] | 1032 | extern "C" int pthread_attr_init(void *attr); |
| 1033 | extern "C" int pthread_attr_destroy(void *attr); |
| 1034 | extern "C" int pthread_attr_setstacksize(void *attr, uptr stacksize); |
Evgeniy Stepanov | b4a218d | 2013-03-19 09:30:52 +0000 | [diff] [blame] | 1035 | extern "C" int pthread_attr_getstack(void *attr, uptr *stack, uptr *stacksize); |
Evgeniy Stepanov | 98f5ea0 | 2013-03-13 09:01:40 +0000 | [diff] [blame] | 1036 | |
| 1037 | INTERCEPTOR(int, pthread_create, void *th, void *attr, void *(*callback)(void*), |
| 1038 | void * param) { |
| 1039 | ENSURE_MSAN_INITED(); // for GetTlsSize() |
| 1040 | __sanitizer_pthread_attr_t myattr; |
| 1041 | if (attr == 0) { |
| 1042 | pthread_attr_init(&myattr); |
| 1043 | attr = &myattr; |
| 1044 | } |
Evgeniy Stepanov | b4a218d | 2013-03-19 09:30:52 +0000 | [diff] [blame] | 1045 | |
| 1046 | AdjustStackSizeLinux(attr, flags()->verbosity); |
Evgeniy Stepanov | 98f5ea0 | 2013-03-13 09:01:40 +0000 | [diff] [blame] | 1047 | |
| 1048 | int res = REAL(pthread_create)(th, attr, callback, param); |
| 1049 | if (attr == &myattr) |
| 1050 | pthread_attr_destroy(&myattr); |
Evgeniy Stepanov | f8213b4 | 2013-04-01 14:47:21 +0000 | [diff] [blame] | 1051 | if (!res) { |
| 1052 | __msan_unpoison(th, __sanitizer::pthread_t_sz); |
| 1053 | } |
Evgeniy Stepanov | 98f5ea0 | 2013-03-13 09:01:40 +0000 | [diff] [blame] | 1054 | return res; |
| 1055 | } |
| 1056 | |
Timur Iskhodzhanov | d7eb677 | 2013-07-15 16:11:39 +0000 | [diff] [blame] | 1057 | INTERCEPTOR(int, pthread_key_create, __sanitizer_pthread_key_t *key, |
| 1058 | void (*dtor)(void *value)) { |
Evgeniy Stepanov | 628d727 | 2013-07-05 12:31:07 +0000 | [diff] [blame] | 1059 | ENSURE_MSAN_INITED(); |
| 1060 | int res = REAL(pthread_key_create)(key, dtor); |
| 1061 | if (!res && key) |
| 1062 | __msan_unpoison(key, sizeof(*key)); |
| 1063 | return res; |
| 1064 | } |
| 1065 | |
Evgeniy Stepanov | 807e131 | 2013-07-30 12:54:34 +0000 | [diff] [blame] | 1066 | INTERCEPTOR(int, pthread_join, void *th, void **retval) { |
| 1067 | ENSURE_MSAN_INITED(); |
| 1068 | int res = REAL(pthread_join)(th, retval); |
| 1069 | if (!res && retval) |
| 1070 | __msan_unpoison(retval, sizeof(*retval)); |
| 1071 | return res; |
| 1072 | } |
| 1073 | |
Evgeniy Stepanov | 9be91e9 | 2013-09-13 11:27:38 +0000 | [diff] [blame] | 1074 | extern char *tzname[2]; |
| 1075 | |
| 1076 | INTERCEPTOR(void, tzset) { |
| 1077 | ENSURE_MSAN_INITED(); |
| 1078 | REAL(tzset)(); |
| 1079 | if (tzname[0]) |
| 1080 | __msan_unpoison(tzname[0], REAL(strlen)(tzname[0]) + 1); |
| 1081 | if (tzname[1]) |
| 1082 | __msan_unpoison(tzname[1], REAL(strlen)(tzname[1]) + 1); |
| 1083 | return; |
| 1084 | } |
| 1085 | |
Evgeniy Stepanov | bfd2122 | 2013-05-22 12:50:26 +0000 | [diff] [blame] | 1086 | struct MSanInterceptorContext { |
| 1087 | bool in_interceptor_scope; |
| 1088 | }; |
| 1089 | |
Evgeniy Stepanov | a343d1d | 2013-05-23 11:51:47 +0000 | [diff] [blame] | 1090 | // A version of CHECK_UNPOISED using a saved scope value. Used in common |
| 1091 | // interceptors. |
Evgeniy Stepanov | dec0f76 | 2013-06-28 11:02:43 +0000 | [diff] [blame] | 1092 | #define CHECK_UNPOISONED_CTX(ctx, x, n) \ |
| 1093 | do { \ |
| 1094 | if (!((MSanInterceptorContext *)ctx)->in_interceptor_scope) \ |
| 1095 | CHECK_UNPOISONED_0(x, n); \ |
| 1096 | } while (0) |
Evgeniy Stepanov | bfd2122 | 2013-05-22 12:50:26 +0000 | [diff] [blame] | 1097 | |
Evgeniy Stepanov | faba61a | 2013-07-09 12:07:59 +0000 | [diff] [blame] | 1098 | #define COMMON_INTERCEPTOR_UNPOISON_PARAM(ctx, count) \ |
| 1099 | UnpoisonParam(count) |
Evgeniy Stepanov | 222076e | 2013-01-18 11:17:23 +0000 | [diff] [blame] | 1100 | #define COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ptr, size) \ |
Evgeniy Stepanov | bfd2122 | 2013-05-22 12:50:26 +0000 | [diff] [blame] | 1101 | __msan_unpoison(ptr, size) |
| 1102 | #define COMMON_INTERCEPTOR_READ_RANGE(ctx, ptr, size) \ |
Evgeniy Stepanov | dec0f76 | 2013-06-28 11:02:43 +0000 | [diff] [blame] | 1103 | CHECK_UNPOISONED_CTX(ctx, ptr, size) |
Evgeniy Stepanov | 08f6628 | 2013-05-29 09:09:58 +0000 | [diff] [blame] | 1104 | #define COMMON_INTERCEPTOR_ENTER(ctx, func, ...) \ |
| 1105 | if (msan_init_is_running) return REAL(func)(__VA_ARGS__); \ |
| 1106 | MSanInterceptorContext msan_ctx = {IsInInterceptorScope()}; \ |
| 1107 | ctx = (void *)&msan_ctx; \ |
| 1108 | InterceptorScope interceptor_scope; \ |
Evgeniy Stepanov | bfd2122 | 2013-05-22 12:50:26 +0000 | [diff] [blame] | 1109 | ENSURE_MSAN_INITED(); |
| 1110 | #define COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd) \ |
| 1111 | do { \ |
Evgeniy Stepanov | 01671c3 | 2013-01-18 13:12:56 +0000 | [diff] [blame] | 1112 | } while (false) |
Evgeniy Stepanov | 08f6628 | 2013-05-29 09:09:58 +0000 | [diff] [blame] | 1113 | #define COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd) \ |
| 1114 | do { \ |
| 1115 | } while (false) |
| 1116 | #define COMMON_INTERCEPTOR_FD_SOCKET_ACCEPT(ctx, fd, newfd) \ |
| 1117 | do { \ |
| 1118 | } while (false) |
Evgeniy Stepanov | 01671c3 | 2013-01-18 13:12:56 +0000 | [diff] [blame] | 1119 | #define COMMON_INTERCEPTOR_SET_THREAD_NAME(ctx, name) \ |
Evgeniy Stepanov | 08f6628 | 2013-05-29 09:09:58 +0000 | [diff] [blame] | 1120 | do { \ |
| 1121 | } while (false) // FIXME |
Evgeniy Stepanov | 64cace0 | 2013-08-12 13:19:53 +0000 | [diff] [blame] | 1122 | #define COMMON_INTERCEPTOR_BLOCK_REAL(name) REAL(name) |
Evgeniy Stepanov | a6c4a38 | 2013-01-18 13:01:18 +0000 | [diff] [blame] | 1123 | #include "sanitizer_common/sanitizer_common_interceptors.inc" |
Kostya Serebryany | 69fe0ba | 2013-01-18 06:43:13 +0000 | [diff] [blame] | 1124 | |
Evgeniy Stepanov | f552311 | 2013-04-11 14:37:04 +0000 | [diff] [blame] | 1125 | #define COMMON_SYSCALL_PRE_READ_RANGE(p, s) CHECK_UNPOISONED(p, s) |
Alexey Samsonov | 786ee9b | 2013-09-19 08:35:16 +0000 | [diff] [blame^] | 1126 | #define COMMON_SYSCALL_PRE_WRITE_RANGE(p, s) \ |
| 1127 | do { \ |
| 1128 | } while (false) |
| 1129 | #define COMMON_SYSCALL_POST_READ_RANGE(p, s) \ |
| 1130 | do { \ |
| 1131 | } while (false) |
Evgeniy Stepanov | f552311 | 2013-04-11 14:37:04 +0000 | [diff] [blame] | 1132 | #define COMMON_SYSCALL_POST_WRITE_RANGE(p, s) __msan_unpoison(p, s) |
| 1133 | #include "sanitizer_common/sanitizer_common_syscalls.inc" |
| 1134 | |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 1135 | // static |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 1136 | void *fast_memset(void *ptr, int c, SIZE_T n) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 1137 | // hack until we have a really fast internal_memset |
| 1138 | if (sizeof(uptr) == 8 && |
| 1139 | (n % 8) == 0 && |
| 1140 | ((uptr)ptr % 8) == 0 && |
| 1141 | (c == 0 || c == -1)) { |
| 1142 | // Printf("memset %p %zd %x\n", ptr, n, c); |
| 1143 | uptr to_store = c ? -1L : 0L; |
| 1144 | uptr *p = (uptr*)ptr; |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 1145 | for (SIZE_T i = 0; i < n / 8; i++) |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 1146 | p[i] = to_store; |
| 1147 | return ptr; |
| 1148 | } |
| 1149 | return internal_memset(ptr, c, n); |
| 1150 | } |
| 1151 | |
| 1152 | // static |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 1153 | void *fast_memcpy(void *dst, const void *src, SIZE_T n) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 1154 | // Same hack as in fast_memset above. |
| 1155 | if (sizeof(uptr) == 8 && |
| 1156 | (n % 8) == 0 && |
| 1157 | ((uptr)dst % 8) == 0 && |
| 1158 | ((uptr)src % 8) == 0) { |
| 1159 | uptr *d = (uptr*)dst; |
| 1160 | uptr *s = (uptr*)src; |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 1161 | for (SIZE_T i = 0; i < n / 8; i++) |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 1162 | d[i] = s[i]; |
| 1163 | return dst; |
| 1164 | } |
| 1165 | return internal_memcpy(dst, src, n); |
| 1166 | } |
| 1167 | |
| 1168 | // These interface functions reside here so that they can use |
| 1169 | // fast_memset, etc. |
Alexey Samsonov | 34093f0 | 2013-04-23 13:34:19 +0000 | [diff] [blame] | 1170 | void __msan_unpoison(const void *a, uptr size) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 1171 | if (!MEM_IS_APP(a)) return; |
| 1172 | fast_memset((void*)MEM_TO_SHADOW((uptr)a), 0, size); |
| 1173 | } |
| 1174 | |
Alexey Samsonov | 34093f0 | 2013-04-23 13:34:19 +0000 | [diff] [blame] | 1175 | void __msan_poison(const void *a, uptr size) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 1176 | if (!MEM_IS_APP(a)) return; |
| 1177 | fast_memset((void*)MEM_TO_SHADOW((uptr)a), |
| 1178 | __msan::flags()->poison_heap_with_zeroes ? 0 : -1, size); |
| 1179 | } |
| 1180 | |
| 1181 | void __msan_poison_stack(void *a, uptr size) { |
| 1182 | if (!MEM_IS_APP(a)) return; |
| 1183 | fast_memset((void*)MEM_TO_SHADOW((uptr)a), |
| 1184 | __msan::flags()->poison_stack_with_zeroes ? 0 : -1, size); |
| 1185 | } |
| 1186 | |
| 1187 | void __msan_clear_and_unpoison(void *a, uptr size) { |
| 1188 | fast_memset(a, 0, size); |
| 1189 | fast_memset((void*)MEM_TO_SHADOW((uptr)a), 0, size); |
| 1190 | } |
| 1191 | |
| 1192 | void __msan_copy_origin(void *dst, const void *src, uptr size) { |
| 1193 | if (!__msan_get_track_origins()) return; |
| 1194 | if (!MEM_IS_APP(dst) || !MEM_IS_APP(src)) return; |
| 1195 | uptr d = MEM_TO_ORIGIN(dst); |
| 1196 | uptr s = MEM_TO_ORIGIN(src); |
| 1197 | uptr beg = d & ~3UL; // align down. |
| 1198 | uptr end = (d + size + 3) & ~3UL; // align up. |
| 1199 | s = s & ~3UL; // align down. |
| 1200 | fast_memcpy((void*)beg, (void*)s, end - beg); |
| 1201 | } |
| 1202 | |
| 1203 | void __msan_copy_poison(void *dst, const void *src, uptr size) { |
| 1204 | if (!MEM_IS_APP(dst)) return; |
| 1205 | if (!MEM_IS_APP(src)) return; |
| 1206 | fast_memcpy((void*)MEM_TO_SHADOW((uptr)dst), |
| 1207 | (void*)MEM_TO_SHADOW((uptr)src), size); |
| 1208 | __msan_copy_origin(dst, src, size); |
| 1209 | } |
| 1210 | |
| 1211 | void __msan_move_poison(void *dst, const void *src, uptr size) { |
| 1212 | if (!MEM_IS_APP(dst)) return; |
| 1213 | if (!MEM_IS_APP(src)) return; |
| 1214 | internal_memmove((void*)MEM_TO_SHADOW((uptr)dst), |
| 1215 | (void*)MEM_TO_SHADOW((uptr)src), size); |
| 1216 | __msan_copy_origin(dst, src, size); |
| 1217 | } |
| 1218 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 1219 | void *__msan_memcpy(void *dest, const void *src, SIZE_T n) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 1220 | ENSURE_MSAN_INITED(); |
| 1221 | void *res = fast_memcpy(dest, src, n); |
| 1222 | __msan_copy_poison(dest, src, n); |
| 1223 | return res; |
| 1224 | } |
| 1225 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 1226 | void *__msan_memset(void *s, int c, SIZE_T n) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 1227 | ENSURE_MSAN_INITED(); |
| 1228 | void *res = fast_memset(s, c, n); |
| 1229 | __msan_unpoison(s, n); |
| 1230 | return res; |
| 1231 | } |
| 1232 | |
Kostya Serebryany | 07bb392 | 2012-12-13 06:31:40 +0000 | [diff] [blame] | 1233 | void *__msan_memmove(void *dest, const void *src, SIZE_T n) { |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 1234 | ENSURE_MSAN_INITED(); |
| 1235 | void *res = REAL(memmove)(dest, src, n); |
| 1236 | __msan_move_poison(dest, src, n); |
| 1237 | return res; |
| 1238 | } |
| 1239 | |
| 1240 | namespace __msan { |
| 1241 | void InitializeInterceptors() { |
| 1242 | static int inited = 0; |
| 1243 | CHECK_EQ(inited, 0); |
Kostya Serebryany | 69fe0ba | 2013-01-18 06:43:13 +0000 | [diff] [blame] | 1244 | SANITIZER_COMMON_INTERCEPTORS_INIT; |
| 1245 | |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1246 | INTERCEPT_FUNCTION(mmap); |
| 1247 | INTERCEPT_FUNCTION(mmap64); |
| 1248 | INTERCEPT_FUNCTION(posix_memalign); |
Evgeniy Stepanov | e3eaa7b | 2013-09-02 09:24:53 +0000 | [diff] [blame] | 1249 | INTERCEPT_FUNCTION(memalign); |
| 1250 | INTERCEPT_FUNCTION(valloc); |
| 1251 | INTERCEPT_FUNCTION(pvalloc); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1252 | INTERCEPT_FUNCTION(malloc); |
| 1253 | INTERCEPT_FUNCTION(calloc); |
| 1254 | INTERCEPT_FUNCTION(realloc); |
| 1255 | INTERCEPT_FUNCTION(free); |
| 1256 | INTERCEPT_FUNCTION(fread); |
| 1257 | INTERCEPT_FUNCTION(fread_unlocked); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1258 | INTERCEPT_FUNCTION(readlink); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1259 | INTERCEPT_FUNCTION(memcpy); |
Evgeniy Stepanov | aa23bd8 | 2013-07-02 14:49:24 +0000 | [diff] [blame] | 1260 | INTERCEPT_FUNCTION(mempcpy); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1261 | INTERCEPT_FUNCTION(memset); |
| 1262 | INTERCEPT_FUNCTION(memmove); |
Evgeniy Stepanov | db615c1 | 2013-07-04 13:19:41 +0000 | [diff] [blame] | 1263 | INTERCEPT_FUNCTION(bcopy); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1264 | INTERCEPT_FUNCTION(wmemset); |
| 1265 | INTERCEPT_FUNCTION(wmemcpy); |
Evgeniy Stepanov | aa23bd8 | 2013-07-02 14:49:24 +0000 | [diff] [blame] | 1266 | INTERCEPT_FUNCTION(wmempcpy); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1267 | INTERCEPT_FUNCTION(wmemmove); |
| 1268 | INTERCEPT_FUNCTION(strcpy); // NOLINT |
Evgeniy Stepanov | 654688b | 2013-07-01 15:19:37 +0000 | [diff] [blame] | 1269 | INTERCEPT_FUNCTION(stpcpy); // NOLINT |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1270 | INTERCEPT_FUNCTION(strdup); |
Evgeniy Stepanov | ead6cf7 | 2013-03-14 11:10:36 +0000 | [diff] [blame] | 1271 | INTERCEPT_FUNCTION(__strdup); |
| 1272 | INTERCEPT_FUNCTION(strndup); |
| 1273 | INTERCEPT_FUNCTION(__strndup); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1274 | INTERCEPT_FUNCTION(strncpy); // NOLINT |
| 1275 | INTERCEPT_FUNCTION(strlen); |
| 1276 | INTERCEPT_FUNCTION(strnlen); |
| 1277 | INTERCEPT_FUNCTION(gcvt); |
| 1278 | INTERCEPT_FUNCTION(strcat); // NOLINT |
| 1279 | INTERCEPT_FUNCTION(strncat); // NOLINT |
| 1280 | INTERCEPT_FUNCTION(strtol); |
| 1281 | INTERCEPT_FUNCTION(strtoll); |
| 1282 | INTERCEPT_FUNCTION(strtoul); |
| 1283 | INTERCEPT_FUNCTION(strtoull); |
Evgeniy Stepanov | e3a3251 | 2013-01-17 13:42:17 +0000 | [diff] [blame] | 1284 | INTERCEPT_FUNCTION(strtod); |
| 1285 | INTERCEPT_FUNCTION(strtof); |
| 1286 | INTERCEPT_FUNCTION(strtold); |
Evgeniy Stepanov | b58b72e | 2013-06-21 13:32:26 +0000 | [diff] [blame] | 1287 | INTERCEPT_FUNCTION(vasprintf); |
| 1288 | INTERCEPT_FUNCTION(asprintf); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1289 | INTERCEPT_FUNCTION(vsprintf); |
| 1290 | INTERCEPT_FUNCTION(vsnprintf); |
| 1291 | INTERCEPT_FUNCTION(vswprintf); |
| 1292 | INTERCEPT_FUNCTION(sprintf); // NOLINT |
| 1293 | INTERCEPT_FUNCTION(snprintf); |
| 1294 | INTERCEPT_FUNCTION(swprintf); |
| 1295 | INTERCEPT_FUNCTION(strftime); |
Evgeniy Stepanov | 165d759 | 2013-07-02 13:34:44 +0000 | [diff] [blame] | 1296 | INTERCEPT_FUNCTION(mbtowc); |
| 1297 | INTERCEPT_FUNCTION(mbrtowc); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1298 | INTERCEPT_FUNCTION(wcslen); |
| 1299 | INTERCEPT_FUNCTION(wcschr); |
| 1300 | INTERCEPT_FUNCTION(wcscpy); |
| 1301 | INTERCEPT_FUNCTION(wcscmp); |
| 1302 | INTERCEPT_FUNCTION(wcstod); |
| 1303 | INTERCEPT_FUNCTION(getenv); |
Evgeniy Stepanov | 5dbedf6 | 2013-08-01 11:14:14 +0000 | [diff] [blame] | 1304 | INTERCEPT_FUNCTION(setenv); |
| 1305 | INTERCEPT_FUNCTION(putenv); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1306 | INTERCEPT_FUNCTION(gettimeofday); |
| 1307 | INTERCEPT_FUNCTION(fcvt); |
| 1308 | INTERCEPT_FUNCTION(__fxstat); |
Evgeniy Stepanov | 842fa3f | 2013-09-09 13:40:41 +0000 | [diff] [blame] | 1309 | INTERCEPT_FUNCTION(__fxstatat); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1310 | INTERCEPT_FUNCTION(__xstat); |
| 1311 | INTERCEPT_FUNCTION(__lxstat); |
| 1312 | INTERCEPT_FUNCTION(__fxstat64); |
Evgeniy Stepanov | 842fa3f | 2013-09-09 13:40:41 +0000 | [diff] [blame] | 1313 | INTERCEPT_FUNCTION(__fxstatat64); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1314 | INTERCEPT_FUNCTION(__xstat64); |
| 1315 | INTERCEPT_FUNCTION(__lxstat64); |
| 1316 | INTERCEPT_FUNCTION(pipe); |
Evgeniy Stepanov | b4eac2f | 2013-04-08 13:45:12 +0000 | [diff] [blame] | 1317 | INTERCEPT_FUNCTION(pipe2); |
| 1318 | INTERCEPT_FUNCTION(socketpair); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1319 | INTERCEPT_FUNCTION(fgets); |
| 1320 | INTERCEPT_FUNCTION(fgets_unlocked); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1321 | INTERCEPT_FUNCTION(getrlimit); |
| 1322 | INTERCEPT_FUNCTION(getrlimit64); |
| 1323 | INTERCEPT_FUNCTION(statfs); |
| 1324 | INTERCEPT_FUNCTION(fstatfs); |
| 1325 | INTERCEPT_FUNCTION(statfs64); |
| 1326 | INTERCEPT_FUNCTION(fstatfs64); |
| 1327 | INTERCEPT_FUNCTION(uname); |
Evgeniy Stepanov | ffffbef | 2013-01-23 10:43:38 +0000 | [diff] [blame] | 1328 | INTERCEPT_FUNCTION(gethostname); |
Alexey Samsonov | cd1e68e | 2012-12-14 11:52:02 +0000 | [diff] [blame] | 1329 | INTERCEPT_FUNCTION(epoll_wait); |
| 1330 | INTERCEPT_FUNCTION(epoll_pwait); |
| 1331 | INTERCEPT_FUNCTION(recv); |
| 1332 | INTERCEPT_FUNCTION(recvfrom); |
Evgeniy Stepanov | e3a3251 | 2013-01-17 13:42:17 +0000 | [diff] [blame] | 1333 | INTERCEPT_FUNCTION(dladdr); |
Reid Kleckner | c9d382b | 2013-03-11 18:07:42 +0000 | [diff] [blame] | 1334 | INTERCEPT_FUNCTION(dlopen); |
Evgeniy Stepanov | 87c6ec7 | 2013-05-16 13:00:25 +0000 | [diff] [blame] | 1335 | INTERCEPT_FUNCTION(dl_iterate_phdr); |
Evgeniy Stepanov | e3a3251 | 2013-01-17 13:42:17 +0000 | [diff] [blame] | 1336 | INTERCEPT_FUNCTION(getrusage); |
Evgeniy Stepanov | f8213b4 | 2013-04-01 14:47:21 +0000 | [diff] [blame] | 1337 | INTERCEPT_FUNCTION(sigaction); |
Evgeniy Stepanov | 1f8b353 | 2013-04-04 08:22:52 +0000 | [diff] [blame] | 1338 | INTERCEPT_FUNCTION(signal); |
Evgeniy Stepanov | 98f5ea0 | 2013-03-13 09:01:40 +0000 | [diff] [blame] | 1339 | INTERCEPT_FUNCTION(pthread_create); |
Evgeniy Stepanov | 628d727 | 2013-07-05 12:31:07 +0000 | [diff] [blame] | 1340 | INTERCEPT_FUNCTION(pthread_key_create); |
Evgeniy Stepanov | 807e131 | 2013-07-30 12:54:34 +0000 | [diff] [blame] | 1341 | INTERCEPT_FUNCTION(pthread_join); |
Evgeniy Stepanov | 9be91e9 | 2013-09-13 11:27:38 +0000 | [diff] [blame] | 1342 | INTERCEPT_FUNCTION(tzset); |
Evgeniy Stepanov | c503378 | 2012-12-11 12:27:27 +0000 | [diff] [blame] | 1343 | inited = 1; |
| 1344 | } |
| 1345 | } // namespace __msan |