| blob | ba3693dbd11f6362a1021fdda20567b1f97b2d8a |
1 //===-- sanitizer_common_interceptors.inc -----------------------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // Common function interceptors for tools like AddressSanitizer,
10 // ThreadSanitizer, MemorySanitizer, etc.
11 //
12 // This file should be included into the tool's interceptor file,
13 // which has to define its own macros:
14 // COMMON_INTERCEPTOR_ENTER
15 // COMMON_INTERCEPTOR_ENTER_NOIGNORE
16 // COMMON_INTERCEPTOR_READ_RANGE
17 // COMMON_INTERCEPTOR_WRITE_RANGE
18 // COMMON_INTERCEPTOR_INITIALIZE_RANGE
19 // COMMON_INTERCEPTOR_DIR_ACQUIRE
20 // COMMON_INTERCEPTOR_FD_ACQUIRE
21 // COMMON_INTERCEPTOR_FD_RELEASE
22 // COMMON_INTERCEPTOR_FD_ACCESS
23 // COMMON_INTERCEPTOR_SET_THREAD_NAME
24 // COMMON_INTERCEPTOR_DLOPEN
25 // COMMON_INTERCEPTOR_ON_EXIT
26 // COMMON_INTERCEPTOR_SET_PTHREAD_NAME
27 // COMMON_INTERCEPTOR_HANDLE_RECVMSG
28 // COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED
29 // COMMON_INTERCEPTOR_MMAP_IMPL
30 // COMMON_INTERCEPTOR_MUNMAP_IMPL
31 // COMMON_INTERCEPTOR_COPY_STRING
32 // COMMON_INTERCEPTOR_STRNDUP_IMPL
33 // COMMON_INTERCEPTOR_STRERROR
34 //===----------------------------------------------------------------------===//
36 #include <stdarg.h>
38 #include "interception/interception.h"
39 #include "sanitizer_addrhashmap.h"
40 #include "sanitizer_dl.h"
41 #include "sanitizer_errno.h"
42 #include "sanitizer_placement_new.h"
43 #include "sanitizer_platform_interceptors.h"
44 #include "sanitizer_platform_limits_posix.h"
45 #include "sanitizer_symbolizer.h"
46 #include "sanitizer_tls_get_addr.h"
48 #if SANITIZER_INTERCEPTOR_HOOKS
49 #define CALL_WEAK_INTERCEPTOR_HOOK(f, ...) f(__VA_ARGS__);
50 #define DECLARE_WEAK_INTERCEPTOR_HOOK(f, ...) \
51 SANITIZER_INTERFACE_WEAK_DEF(void, f, __VA_ARGS__) {}
52 #else
53 #define DECLARE_WEAK_INTERCEPTOR_HOOK(f, ...)
54 #define CALL_WEAK_INTERCEPTOR_HOOK(f, ...)
56 #endif // SANITIZER_INTERCEPTOR_HOOKS
58 #if SANITIZER_WINDOWS && !defined(va_copy)
59 #define va_copy(dst, src) ((dst) = (src))
60 #endif // _WIN32
62 #if SANITIZER_FREEBSD
63 #define pthread_setname_np pthread_set_name_np
64 #define inet_aton __inet_aton
65 #define inet_pton __inet_pton
66 #define iconv __bsd_iconv
67 #endif
69 #if SANITIZER_NETBSD
70 #define clock_getres __clock_getres50
71 #define clock_gettime __clock_gettime50
72 #define clock_settime __clock_settime50
73 #define ctime __ctime50
74 #define ctime_r __ctime_r50
75 #define devname __devname50
76 #define fgetpos __fgetpos50
77 #define fsetpos __fsetpos50
78 #define fstatvfs __fstatvfs90
79 #define fstatvfs1 __fstatvfs190
80 #define fts_children __fts_children60
81 #define fts_close __fts_close60
82 #define fts_open __fts_open60
83 #define fts_read __fts_read60
84 #define fts_set __fts_set60
85 #define getitimer __getitimer50
86 #define getmntinfo __getmntinfo90
87 #define getpwent __getpwent50
88 #define getpwnam __getpwnam50
89 #define getpwnam_r __getpwnam_r50
90 #define getpwuid __getpwuid50
91 #define getpwuid_r __getpwuid_r50
92 #define getutent __getutent50
93 #define getutxent __getutxent50
94 #define getutxid __getutxid50
95 #define getutxline __getutxline50
96 #define getvfsstat __getvfsstat90
97 #define pututxline __pututxline50
98 #define glob __glob30
99 #define gmtime __gmtime50
100 #define gmtime_r __gmtime_r50
101 #define localtime __locatime50
102 #define localtime_r __localtime_r50
103 #define mktime __mktime50
104 #define lstat __lstat50
105 #define opendir __opendir30
106 #define readdir __readdir30
107 #define readdir_r __readdir_r30
108 #define scandir __scandir30
109 #define setitimer __setitimer50
110 #define setlocale __setlocale50
111 #define shmctl __shmctl50
112 #define sigaltstack __sigaltstack14
113 #define sigemptyset __sigemptyset14
114 #define sigfillset __sigfillset14
115 #define sigpending __sigpending14
116 #define sigprocmask __sigprocmask14
117 #define sigtimedwait __sigtimedwait50
118 #define stat __stat50
119 #define statvfs __statvfs90
120 #define statvfs1 __statvfs190
121 #define time __time50
122 #define times __times13
123 #define unvis __unvis50
124 #define wait3 __wait350
125 #define wait4 __wait450
126 extern const unsigned short *_ctype_tab_;
127 extern const short *_toupper_tab_;
128 extern const short *_tolower_tab_;
129 #endif
131 #if SANITIZER_LINUX && SANITIZER_SPARC32
132 // On 32-bit Linux/sparc64, double and long double are identical and glibc
133 // uses a __nldbl_ (no long double) prefix for various stdio functions.
134 # define __isoc23_fscanf __nldbl___isoc23_fscanf
135 # define __isoc23_scanf __nldbl___isoc23_scanf
136 # define __isoc23_sscanf __nldbl___isoc23_sscanf
137 # define __isoc23_vfscanf __nldbl___isoc23_vfscanf
138 # define __isoc23_vscanf __nldbl___isoc23_vscanf
139 # define __isoc23_vsscanf __nldbl___isoc23_vsscanf
140 # define __isoc99_fscanf __nldbl___isoc99_fscanf
141 # define __isoc99_scanf __nldbl___isoc99_scanf
142 # define __isoc99_sscanf __nldbl___isoc99_sscanf
143 # define __isoc99_vfscanf __nldbl___isoc99_vfscanf
144 # define __isoc99_vscanf __nldbl___isoc99_vscanf
145 # define __isoc99_vsscanf __nldbl___isoc99_vsscanf
146 # define asprintf __nldbl_asprintf
147 # define fprintf __nldbl_fprintf
148 # define fscanf __nldbl_fscanf
149 # define printf __nldbl_printf
150 # define scanf __nldbl_scanf
151 # define snprintf __nldbl_snprintf
152 # define sprintf __nldbl_sprintf
153 # define sscanf __nldbl_sscanf
154 # define vasprintf __nldbl_vasprintf
155 # define vfprintf __nldbl_vfprintf
156 # define vfscanf __nldbl_vfscanf
157 # define vprintf __nldbl_vprintf
158 # define vscanf __nldbl_vscanf
159 # define vsnprintf __nldbl_vsnprintf
160 # define vsprintf __nldbl_vsprintf
161 # define vsscanf __nldbl_vsscanf
162 #endif
164 #if SANITIZER_MUSL && \
165 (defined(__i386__) || defined(__arm__) || SANITIZER_MIPS32 || SANITIZER_PPC32)
166 // musl 1.2.0 on existing 32-bit architectures uses new symbol names for the
167 // time-related functions that take 64-bit time_t values. See
168 // https://musl.libc.org/time64.html
169 #define adjtime __adjtime64
170 #define adjtimex __adjtimex_time64
171 #define aio_suspend __aio_suspend_time64
172 #define clock_adjtime __clock_adjtime64
173 #define clock_getres __clock_getres_time64
174 #define clock_gettime __clock_gettime64
175 #define clock_nanosleep __clock_nanosleep_time64
176 #define clock_settime __clock_settime64
177 #define cnd_timedwait __cnd_timedwait_time64
178 #define ctime __ctime64
179 #define ctime_r __ctime64_r
180 #define difftime __difftime64
181 #define dlsym __dlsym_time64
182 #define fstatat __fstatat_time64
183 #define fstat __fstat_time64
184 #define ftime __ftime64
185 #define futimens __futimens_time64
186 #define futimesat __futimesat_time64
187 #define futimes __futimes_time64
188 #define getitimer __getitimer_time64
189 #define getrusage __getrusage_time64
190 #define gettimeofday __gettimeofday_time64
191 #define gmtime __gmtime64
192 #define gmtime_r __gmtime64_r
193 #define localtime __localtime64
194 #define localtime_r __localtime64_r
195 #define lstat __lstat_time64
196 #define lutimes __lutimes_time64
197 #define mktime __mktime64
198 #define mq_timedreceive __mq_timedreceive_time64
199 #define mq_timedsend __mq_timedsend_time64
200 #define mtx_timedlock __mtx_timedlock_time64
201 #define nanosleep __nanosleep_time64
202 #define ppoll __ppoll_time64
203 #define pselect __pselect_time64
204 #define pthread_cond_timedwait __pthread_cond_timedwait_time64
205 #define pthread_mutex_timedlock __pthread_mutex_timedlock_time64
206 #define pthread_rwlock_timedrdlock __pthread_rwlock_timedrdlock_time64
207 #define pthread_rwlock_timedwrlock __pthread_rwlock_timedwrlock_time64
208 #define pthread_timedjoin_np __pthread_timedjoin_np_time64
209 #define recvmmsg __recvmmsg_time64
210 #define sched_rr_get_interval __sched_rr_get_interval_time64
211 #define select __select_time64
212 #define semtimedop __semtimedop_time64
213 #define sem_timedwait __sem_timedwait_time64
214 #define setitimer __setitimer_time64
215 #define settimeofday __settimeofday_time64
216 #define sigtimedwait __sigtimedwait_time64
217 #define stat __stat_time64
218 #define stime __stime64
219 #define thrd_sleep __thrd_sleep_time64
220 #define timegm __timegm_time64
221 #define timerfd_gettime __timerfd_gettime64
222 #define timerfd_settime __timerfd_settime64
223 #define timer_gettime __timer_gettime64
224 #define timer_settime __timer_settime64
225 #define timespec_get __timespec_get_time64
226 #define time __time64
227 #define utimensat __utimensat_time64
228 #define utimes __utimes_time64
229 #define utime __utime64
230 #define wait3 __wait3_time64
231 #define wait4 __wait4_time64
232 #endif
234 #ifndef COMMON_INTERCEPTOR_INITIALIZE_RANGE
235 #define COMMON_INTERCEPTOR_INITIALIZE_RANGE(p, size) {}
236 #endif
238 #ifndef COMMON_INTERCEPTOR_UNPOISON_PARAM
239 #define COMMON_INTERCEPTOR_UNPOISON_PARAM(count) {}
240 #endif
242 #ifndef COMMON_INTERCEPTOR_FD_ACCESS
243 #define COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd) {}
244 #endif
246 #ifndef COMMON_INTERCEPTOR_HANDLE_RECVMSG
247 #define COMMON_INTERCEPTOR_HANDLE_RECVMSG(ctx, msg) ((void)(msg))
248 #endif
250 #ifndef COMMON_INTERCEPTOR_FILE_OPEN
251 #define COMMON_INTERCEPTOR_FILE_OPEN(ctx, file, path) {}
252 #endif
254 #ifndef COMMON_INTERCEPTOR_FILE_CLOSE
255 #define COMMON_INTERCEPTOR_FILE_CLOSE(ctx, file) {}
256 #endif
258 #ifndef COMMON_INTERCEPTOR_LIBRARY_LOADED
259 #define COMMON_INTERCEPTOR_LIBRARY_LOADED(filename, handle) {}
260 #endif
262 #ifndef COMMON_INTERCEPTOR_LIBRARY_UNLOADED
263 #define COMMON_INTERCEPTOR_LIBRARY_UNLOADED() {}
264 #endif
266 #ifndef COMMON_INTERCEPTOR_ENTER_NOIGNORE
267 #define COMMON_INTERCEPTOR_ENTER_NOIGNORE(ctx, ...) \
268 COMMON_INTERCEPTOR_ENTER(ctx, __VA_ARGS__)
269 #endif
271 #ifndef COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED
272 #define COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED (0)
273 #endif
275 #define COMMON_INTERCEPTOR_READ_STRING(ctx, s, n) \
276 COMMON_INTERCEPTOR_READ_RANGE((ctx), (s), \
277 common_flags()->strict_string_checks ? (internal_strlen(s)) + 1 : (n) )
279 #ifndef COMMON_INTERCEPTOR_DLOPEN
280 #define COMMON_INTERCEPTOR_DLOPEN(filename, flag) \
281 ({ CheckNoDeepBind(filename, flag); REAL(dlopen)(filename, flag); })
282 #endif
284 #ifndef COMMON_INTERCEPTOR_GET_TLS_RANGE
285 #define COMMON_INTERCEPTOR_GET_TLS_RANGE(begin, end) *begin = *end = 0;
286 #endif
288 #ifndef COMMON_INTERCEPTOR_ACQUIRE
289 #define COMMON_INTERCEPTOR_ACQUIRE(ctx, u) {}
290 #endif
292 #ifndef COMMON_INTERCEPTOR_RELEASE
293 #define COMMON_INTERCEPTOR_RELEASE(ctx, u) {}
294 #endif
296 #ifndef COMMON_INTERCEPTOR_USER_CALLBACK_START
297 #define COMMON_INTERCEPTOR_USER_CALLBACK_START() {}
298 #endif
300 #ifndef COMMON_INTERCEPTOR_USER_CALLBACK_END
301 #define COMMON_INTERCEPTOR_USER_CALLBACK_END() {}
302 #endif
304 #ifdef SANITIZER_NLDBL_VERSION
305 #define COMMON_INTERCEPT_FUNCTION_LDBL(fn) \
306 COMMON_INTERCEPT_FUNCTION_VER(fn, SANITIZER_NLDBL_VERSION)
307 #else
308 #define COMMON_INTERCEPT_FUNCTION_LDBL(fn) \
309 COMMON_INTERCEPT_FUNCTION(fn)
310 #endif
312 #if SANITIZER_GLIBC
313 // If we could not find the versioned symbol, fall back to an unversioned
314 // lookup. This is needed to work around a GLibc bug that causes dlsym
315 // with RTLD_NEXT to return the oldest versioned symbol.
316 // See https://sourceware.org/bugzilla/show_bug.cgi?id=14932.
317 // For certain symbols (e.g. regexec) we have to perform a versioned lookup,
318 // but that versioned symbol will only exist for architectures where the
319 // oldest Glibc version pre-dates support for that architecture.
320 // For example, regexec@GLIBC_2.3.4 exists on x86_64, but not RISC-V.
321 // See also https://gcc.gnu.org/bugzilla/show_bug.cgi?id=98920.
322 #define COMMON_INTERCEPT_FUNCTION_GLIBC_VER_MIN(fn, ver) \
323 COMMON_INTERCEPT_FUNCTION_VER_UNVERSIONED_FALLBACK(fn, ver)
324 #else
325 #define COMMON_INTERCEPT_FUNCTION_GLIBC_VER_MIN(fn, ver) \
326 COMMON_INTERCEPT_FUNCTION(fn)
327 #endif
329 #ifndef COMMON_INTERCEPTOR_MMAP_IMPL
330 #define COMMON_INTERCEPTOR_MMAP_IMPL(ctx, mmap, addr, sz, prot, flags, fd, \
331 off) \
332 { return REAL(mmap)(addr, sz, prot, flags, fd, off); }
333 #endif
335 #ifndef COMMON_INTERCEPTOR_MUNMAP_IMPL
336 #define COMMON_INTERCEPTOR_MUNMAP_IMPL(ctx, addr, sz) \
337 { return REAL(munmap)(addr, sz); }
338 #endif
340 #ifndef COMMON_INTERCEPTOR_COPY_STRING
341 #define COMMON_INTERCEPTOR_COPY_STRING(ctx, to, from, size) {}
342 #endif
344 #ifndef COMMON_INTERCEPTOR_STRNDUP_IMPL
345 #define COMMON_INTERCEPTOR_STRNDUP_IMPL(ctx, s, size) \
346 COMMON_INTERCEPTOR_ENTER(ctx, strndup, s, size); \
347 uptr copy_length = internal_strnlen(s, size); \
348 char *new_mem = (char *)WRAP(malloc)(copy_length + 1); \
349 if (common_flags()->intercept_strndup) { \
350 COMMON_INTERCEPTOR_READ_STRING(ctx, s, Min(size, copy_length + 1)); \
351 } \
352 if (new_mem) { \
353 COMMON_INTERCEPTOR_COPY_STRING(ctx, new_mem, s, copy_length); \
354 internal_memcpy(new_mem, s, copy_length); \
355 new_mem[copy_length] = '\0'; \
356 } \
357 return new_mem;
358 #endif
360 #ifndef COMMON_INTERCEPTOR_STRERROR
361 #define COMMON_INTERCEPTOR_STRERROR() {}
362 #endif
364 struct FileMetadata {
365 // For open_memstream().
366 char **addr;
367 SIZE_T *size;
368 };
370 struct CommonInterceptorMetadata {
371 enum {
372 CIMT_INVALID = 0,
373 CIMT_FILE
374 } type;
375 union {
376 FileMetadata file;
377 };
378 };
380 #if SI_POSIX
381 typedef AddrHashMap<CommonInterceptorMetadata, 31051> MetadataHashMap;
383 static MetadataHashMap *interceptor_metadata_map;
385 UNUSED static void SetInterceptorMetadata(__sanitizer_FILE *addr,
386 const FileMetadata &file) {
387 MetadataHashMap::Handle h(interceptor_metadata_map, (uptr)addr);
388 CHECK(h.created());
389 h->type = CommonInterceptorMetadata::CIMT_FILE;
390 h->file = file;
391 }
393 UNUSED static const FileMetadata *GetInterceptorMetadata(
394 __sanitizer_FILE *addr) {
395 MetadataHashMap::Handle h(interceptor_metadata_map, (uptr)addr,
396 /* remove */ false,
397 /* create */ false);
398 if (addr && h.exists()) {
399 CHECK(!h.created());
400 CHECK(h->type == CommonInterceptorMetadata::CIMT_FILE);
401 return &h->file;
402 } else {
403 return 0;
404 }
405 }
407 UNUSED static void DeleteInterceptorMetadata(void *addr) {
408 MetadataHashMap::Handle h(interceptor_metadata_map, (uptr)addr, true);
409 CHECK(h.exists());
410 }
411 #endif // SI_POSIX
413 #if SANITIZER_INTERCEPT_STRLEN
414 INTERCEPTOR(SIZE_T, strlen, const char *s) {
415 // Sometimes strlen is called prior to InitializeCommonInterceptors,
416 // in which case the REAL(strlen) typically used in
417 // COMMON_INTERCEPTOR_ENTER will fail. We use internal_strlen here
418 // to handle that.
419 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
420 return internal_strlen(s);
421 void *ctx;
422 COMMON_INTERCEPTOR_ENTER(ctx, strlen, s);
423 SIZE_T result = REAL(strlen)(s);
424 if (common_flags()->intercept_strlen)
425 COMMON_INTERCEPTOR_READ_RANGE(ctx, s, result + 1);
426 return result;
427 }
428 #define INIT_STRLEN COMMON_INTERCEPT_FUNCTION(strlen)
429 #else
430 #define INIT_STRLEN
431 #endif
433 #if SANITIZER_INTERCEPT_STRNLEN
434 INTERCEPTOR(SIZE_T, strnlen, const char *s, SIZE_T maxlen) {
435 void *ctx;
436 COMMON_INTERCEPTOR_ENTER(ctx, strnlen, s, maxlen);
437 SIZE_T length = REAL(strnlen)(s, maxlen);
438 if (common_flags()->intercept_strlen)
439 COMMON_INTERCEPTOR_READ_RANGE(ctx, s, Min(length + 1, maxlen));
440 return length;
441 }
442 #define INIT_STRNLEN COMMON_INTERCEPT_FUNCTION(strnlen)
443 #else
444 #define INIT_STRNLEN
445 #endif
447 #if SANITIZER_INTERCEPT_STRNDUP
448 INTERCEPTOR(char*, strndup, const char *s, uptr size) {
449 void *ctx;
450 COMMON_INTERCEPTOR_STRNDUP_IMPL(ctx, s, size);
451 }
452 #define INIT_STRNDUP COMMON_INTERCEPT_FUNCTION(strndup)
453 #else
454 #define INIT_STRNDUP
455 #endif // SANITIZER_INTERCEPT_STRNDUP
457 #if SANITIZER_INTERCEPT___STRNDUP
458 INTERCEPTOR(char*, __strndup, const char *s, uptr size) {
459 void *ctx;
460 COMMON_INTERCEPTOR_STRNDUP_IMPL(ctx, s, size);
461 }
462 #define INIT___STRNDUP COMMON_INTERCEPT_FUNCTION(__strndup)
463 #else
464 #define INIT___STRNDUP
465 #endif // SANITIZER_INTERCEPT___STRNDUP
467 #if SANITIZER_INTERCEPT_TEXTDOMAIN
468 INTERCEPTOR(char*, textdomain, const char *domainname) {
469 void *ctx;
470 COMMON_INTERCEPTOR_ENTER(ctx, textdomain, domainname);
471 if (domainname) COMMON_INTERCEPTOR_READ_STRING(ctx, domainname, 0);
472 char *domain = REAL(textdomain)(domainname);
473 if (domain) {
474 COMMON_INTERCEPTOR_INITIALIZE_RANGE(domain, internal_strlen(domain) + 1);
475 }
476 return domain;
477 }
478 #define INIT_TEXTDOMAIN COMMON_INTERCEPT_FUNCTION(textdomain)
479 #else
480 #define INIT_TEXTDOMAIN
481 #endif
483 #if SANITIZER_INTERCEPT_STRCMP || SANITIZER_INTERCEPT_MEMCMP
484 static inline int CharCmpX(unsigned char c1, unsigned char c2) {
485 return (c1 == c2) ? 0 : (c1 < c2) ? -1 : 1;
486 }
487 #endif
489 #if SANITIZER_INTERCEPT_STRCMP
490 DECLARE_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_strcmp, uptr called_pc,
491 const char *s1, const char *s2, int result)
493 INTERCEPTOR(int, strcmp, const char *s1, const char *s2) {
494 void *ctx;
495 COMMON_INTERCEPTOR_ENTER(ctx, strcmp, s1, s2);
496 unsigned char c1, c2;
497 uptr i;
498 for (i = 0;; i++) {
499 c1 = (unsigned char)s1[i];
500 c2 = (unsigned char)s2[i];
501 if (c1 != c2 || c1 == '\0') break;
502 }
503 if (common_flags()->intercept_strcmp) {
504 COMMON_INTERCEPTOR_READ_STRING(ctx, s1, i + 1);
505 COMMON_INTERCEPTOR_READ_STRING(ctx, s2, i + 1);
506 }
507 int result = CharCmpX(c1, c2);
508 CALL_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_strcmp, GET_CALLER_PC(), s1,
509 s2, result);
510 return result;
511 }
513 DECLARE_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_strncmp, uptr called_pc,
514 const char *s1, const char *s2, uptr n,
515 int result)
517 INTERCEPTOR(int, strncmp, const char *s1, const char *s2, uptr size) {
518 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
519 return internal_strncmp(s1, s2, size);
520 void *ctx;
521 COMMON_INTERCEPTOR_ENTER(ctx, strncmp, s1, s2, size);
522 unsigned char c1 = 0, c2 = 0;
523 uptr i;
524 for (i = 0; i < size; i++) {
525 c1 = (unsigned char)s1[i];
526 c2 = (unsigned char)s2[i];
527 if (c1 != c2 || c1 == '\0') break;
528 }
529 uptr i1 = i;
530 uptr i2 = i;
531 if (common_flags()->strict_string_checks) {
532 for (; i1 < size && s1[i1]; i1++) {}
533 for (; i2 < size && s2[i2]; i2++) {}
534 }
535 COMMON_INTERCEPTOR_READ_RANGE((ctx), (s1), Min(i1 + 1, size));
536 COMMON_INTERCEPTOR_READ_RANGE((ctx), (s2), Min(i2 + 1, size));
537 int result = CharCmpX(c1, c2);
538 CALL_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_strncmp, GET_CALLER_PC(), s1,
539 s2, size, result);
540 return result;
541 }
543 #define INIT_STRCMP COMMON_INTERCEPT_FUNCTION(strcmp)
544 #define INIT_STRNCMP COMMON_INTERCEPT_FUNCTION(strncmp)
545 #else
546 #define INIT_STRCMP
547 #define INIT_STRNCMP
548 #endif
550 #if SANITIZER_INTERCEPT_STRCASECMP
551 static inline int CharCaseCmp(unsigned char c1, unsigned char c2) {
552 int c1_low = ToLower(c1);
553 int c2_low = ToLower(c2);
554 return c1_low - c2_low;
555 }
557 DECLARE_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_strcasecmp, uptr called_pc,
558 const char *s1, const char *s2, int result)
560 INTERCEPTOR(int, strcasecmp, const char *s1, const char *s2) {
561 void *ctx;
562 COMMON_INTERCEPTOR_ENTER(ctx, strcasecmp, s1, s2);
563 unsigned char c1 = 0, c2 = 0;
564 uptr i;
565 for (i = 0;; i++) {
566 c1 = (unsigned char)s1[i];
567 c2 = (unsigned char)s2[i];
568 if (CharCaseCmp(c1, c2) != 0 || c1 == '\0') break;
569 }
570 COMMON_INTERCEPTOR_READ_STRING(ctx, s1, i + 1);
571 COMMON_INTERCEPTOR_READ_STRING(ctx, s2, i + 1);
572 int result = CharCaseCmp(c1, c2);
573 CALL_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_strcasecmp, GET_CALLER_PC(),
574 s1, s2, result);
575 return result;
576 }
578 DECLARE_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_strncasecmp, uptr called_pc,
579 const char *s1, const char *s2, uptr size,
580 int result)
582 INTERCEPTOR(int, strncasecmp, const char *s1, const char *s2, SIZE_T size) {
583 void *ctx;
584 COMMON_INTERCEPTOR_ENTER(ctx, strncasecmp, s1, s2, size);
585 unsigned char c1 = 0, c2 = 0;
586 uptr i;
587 for (i = 0; i < size; i++) {
588 c1 = (unsigned char)s1[i];
589 c2 = (unsigned char)s2[i];
590 if (CharCaseCmp(c1, c2) != 0 || c1 == '\0') break;
591 }
592 uptr i1 = i;
593 uptr i2 = i;
594 if (common_flags()->strict_string_checks) {
595 for (; i1 < size && s1[i1]; i1++) {}
596 for (; i2 < size && s2[i2]; i2++) {}
597 }
598 COMMON_INTERCEPTOR_READ_RANGE((ctx), (s1), Min(i1 + 1, size));
599 COMMON_INTERCEPTOR_READ_RANGE((ctx), (s2), Min(i2 + 1, size));
600 int result = CharCaseCmp(c1, c2);
601 CALL_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_strncasecmp, GET_CALLER_PC(),
602 s1, s2, size, result);
603 return result;
604 }
606 #define INIT_STRCASECMP COMMON_INTERCEPT_FUNCTION(strcasecmp)
607 #define INIT_STRNCASECMP COMMON_INTERCEPT_FUNCTION(strncasecmp)
608 #else
609 #define INIT_STRCASECMP
610 #define INIT_STRNCASECMP
611 #endif
613 #if SANITIZER_INTERCEPT_STRSTR || SANITIZER_INTERCEPT_STRCASESTR
614 static inline void StrstrCheck(void *ctx, char *r, const char *s1,
615 const char *s2) {
616 uptr len1 = internal_strlen(s1);
617 uptr len2 = internal_strlen(s2);
618 COMMON_INTERCEPTOR_READ_STRING(ctx, s1, r ? r - s1 + len2 : len1 + 1);
619 COMMON_INTERCEPTOR_READ_RANGE(ctx, s2, len2 + 1);
620 }
621 #endif
623 #if SANITIZER_INTERCEPT_STRSTR
625 DECLARE_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_strstr, uptr called_pc,
626 const char *s1, const char *s2, char *result)
628 INTERCEPTOR(char*, strstr, const char *s1, const char *s2) {
629 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
630 return internal_strstr(s1, s2);
631 void *ctx;
632 COMMON_INTERCEPTOR_ENTER(ctx, strstr, s1, s2);
633 char *r = REAL(strstr)(s1, s2);
634 if (common_flags()->intercept_strstr)
635 StrstrCheck(ctx, r, s1, s2);
636 CALL_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_strstr, GET_CALLER_PC(), s1,
637 s2, r);
638 return r;
639 }
641 #define INIT_STRSTR COMMON_INTERCEPT_FUNCTION(strstr);
642 #else
643 #define INIT_STRSTR
644 #endif
646 #if SANITIZER_INTERCEPT_STRCASESTR
648 DECLARE_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_strcasestr, uptr called_pc,
649 const char *s1, const char *s2, char *result)
651 INTERCEPTOR(char*, strcasestr, const char *s1, const char *s2) {
652 void *ctx;
653 COMMON_INTERCEPTOR_ENTER(ctx, strcasestr, s1, s2);
654 char *r = REAL(strcasestr)(s1, s2);
655 if (common_flags()->intercept_strstr)
656 StrstrCheck(ctx, r, s1, s2);
657 CALL_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_strcasestr, GET_CALLER_PC(),
658 s1, s2, r);
659 return r;
660 }
662 #define INIT_STRCASESTR COMMON_INTERCEPT_FUNCTION(strcasestr);
663 #else
664 #define INIT_STRCASESTR
665 #endif
667 #if SANITIZER_INTERCEPT_STRTOK
669 INTERCEPTOR(char*, strtok, char *str, const char *delimiters) {
670 void *ctx;
671 COMMON_INTERCEPTOR_ENTER(ctx, strtok, str, delimiters);
672 if (!common_flags()->intercept_strtok) {
673 return REAL(strtok)(str, delimiters);
674 }
675 if (common_flags()->strict_string_checks) {
676 // If strict_string_checks is enabled, we check the whole first argument
677 // string on the first call (strtok saves this string in a static buffer
678 // for subsequent calls). We do not need to check strtok's result.
679 // As the delimiters can change, we check them every call.
680 if (str != nullptr) {
681 COMMON_INTERCEPTOR_READ_RANGE(ctx, str, internal_strlen(str) + 1);
682 }
683 COMMON_INTERCEPTOR_READ_RANGE(ctx, delimiters,
684 internal_strlen(delimiters) + 1);
685 return REAL(strtok)(str, delimiters);
686 } else {
687 // However, when strict_string_checks is disabled we cannot check the
688 // whole string on the first call. Instead, we check the result string
689 // which is guaranteed to be a NULL-terminated substring of the first
690 // argument. We also conservatively check one character of str and the
691 // delimiters.
692 if (str != nullptr) {
693 COMMON_INTERCEPTOR_READ_STRING(ctx, str, 1);
694 }
695 COMMON_INTERCEPTOR_READ_RANGE(ctx, delimiters, 1);
696 char *result = REAL(strtok)(str, delimiters);
697 if (result != nullptr) {
698 COMMON_INTERCEPTOR_READ_RANGE(ctx, result, internal_strlen(result) + 1);
699 } else if (str != nullptr) {
700 // No delimiter were found, it's safe to assume that the entire str was
701 // scanned.
702 COMMON_INTERCEPTOR_READ_RANGE(ctx, str, internal_strlen(str) + 1);
703 }
704 return result;
705 }
706 }
708 #define INIT_STRTOK COMMON_INTERCEPT_FUNCTION(strtok)
709 #else
710 #define INIT_STRTOK
711 #endif
713 #if SANITIZER_INTERCEPT_MEMMEM
714 DECLARE_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_memmem, uptr called_pc,
715 const void *s1, SIZE_T len1, const void *s2,
716 SIZE_T len2, void *result)
718 INTERCEPTOR(void*, memmem, const void *s1, SIZE_T len1, const void *s2,
719 SIZE_T len2) {
720 void *ctx;
721 COMMON_INTERCEPTOR_ENTER(ctx, memmem, s1, len1, s2, len2);
722 void *r = REAL(memmem)(s1, len1, s2, len2);
723 if (common_flags()->intercept_memmem) {
724 COMMON_INTERCEPTOR_READ_RANGE(ctx, s1, len1);
725 COMMON_INTERCEPTOR_READ_RANGE(ctx, s2, len2);
726 }
727 CALL_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_memmem, GET_CALLER_PC(),
728 s1, len1, s2, len2, r);
729 return r;
730 }
732 #define INIT_MEMMEM COMMON_INTERCEPT_FUNCTION(memmem);
733 #else
734 #define INIT_MEMMEM
735 #endif // SANITIZER_INTERCEPT_MEMMEM
737 #if SANITIZER_INTERCEPT_STRCHR
738 INTERCEPTOR(char*, strchr, const char *s, int c) {
739 void *ctx;
740 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
741 return internal_strchr(s, c);
742 COMMON_INTERCEPTOR_ENTER(ctx, strchr, s, c);
743 char *result = REAL(strchr)(s, c);
744 if (common_flags()->intercept_strchr) {
745 // Keep strlen as macro argument, as macro may ignore it.
746 COMMON_INTERCEPTOR_READ_STRING(ctx, s,
747 (result ? result - s : internal_strlen(s)) + 1);
748 }
749 return result;
750 }
751 #define INIT_STRCHR COMMON_INTERCEPT_FUNCTION(strchr)
752 #else
753 #define INIT_STRCHR
754 #endif
756 #if SANITIZER_INTERCEPT_STRCHRNUL
757 INTERCEPTOR(char*, strchrnul, const char *s, int c) {
758 void *ctx;
759 COMMON_INTERCEPTOR_ENTER(ctx, strchrnul, s, c);
760 char *result = REAL(strchrnul)(s, c);
761 uptr len = result - s + 1;
762 if (common_flags()->intercept_strchr)
763 COMMON_INTERCEPTOR_READ_STRING(ctx, s, len);
764 return result;
765 }
766 #define INIT_STRCHRNUL COMMON_INTERCEPT_FUNCTION(strchrnul)
767 #else
768 #define INIT_STRCHRNUL
769 #endif
771 #if SANITIZER_INTERCEPT_STRRCHR
772 INTERCEPTOR(char*, strrchr, const char *s, int c) {
773 void *ctx;
774 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
775 return internal_strrchr(s, c);
776 COMMON_INTERCEPTOR_ENTER(ctx, strrchr, s, c);
777 if (common_flags()->intercept_strchr)
778 COMMON_INTERCEPTOR_READ_RANGE(ctx, s, internal_strlen(s) + 1);
779 return REAL(strrchr)(s, c);
780 }
781 #define INIT_STRRCHR COMMON_INTERCEPT_FUNCTION(strrchr)
782 #else
783 #define INIT_STRRCHR
784 #endif
786 #if SANITIZER_INTERCEPT_STRSPN
787 INTERCEPTOR(SIZE_T, strspn, const char *s1, const char *s2) {
788 void *ctx;
789 COMMON_INTERCEPTOR_ENTER(ctx, strspn, s1, s2);
790 SIZE_T r = REAL(strspn)(s1, s2);
791 if (common_flags()->intercept_strspn) {
792 COMMON_INTERCEPTOR_READ_RANGE(ctx, s2, internal_strlen(s2) + 1);
793 COMMON_INTERCEPTOR_READ_STRING(ctx, s1, r + 1);
794 }
795 return r;
796 }
798 INTERCEPTOR(SIZE_T, strcspn, const char *s1, const char *s2) {
799 void *ctx;
800 COMMON_INTERCEPTOR_ENTER(ctx, strcspn, s1, s2);
801 SIZE_T r = REAL(strcspn)(s1, s2);
802 if (common_flags()->intercept_strspn) {
803 COMMON_INTERCEPTOR_READ_RANGE(ctx, s2, internal_strlen(s2) + 1);
804 COMMON_INTERCEPTOR_READ_STRING(ctx, s1, r + 1);
805 }
806 return r;
807 }
809 #define INIT_STRSPN \
810 COMMON_INTERCEPT_FUNCTION(strspn); \
811 COMMON_INTERCEPT_FUNCTION(strcspn);
812 #else
813 #define INIT_STRSPN
814 #endif
816 #if SANITIZER_INTERCEPT_STRPBRK
817 INTERCEPTOR(char *, strpbrk, const char *s1, const char *s2) {
818 void *ctx;
819 COMMON_INTERCEPTOR_ENTER(ctx, strpbrk, s1, s2);
820 char *r = REAL(strpbrk)(s1, s2);
821 if (common_flags()->intercept_strpbrk) {
822 COMMON_INTERCEPTOR_READ_RANGE(ctx, s2, internal_strlen(s2) + 1);
823 COMMON_INTERCEPTOR_READ_STRING(ctx, s1,
824 r ? r - s1 + 1 : internal_strlen(s1) + 1);
825 }
826 return r;
827 }
829 #define INIT_STRPBRK COMMON_INTERCEPT_FUNCTION(strpbrk);
830 #else
831 #define INIT_STRPBRK
832 #endif
834 #if SANITIZER_INTERCEPT_MEMCMP
835 DECLARE_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_memcmp, uptr called_pc,
836 const void *s1, const void *s2, uptr n,
837 int result)
839 // Common code for `memcmp` and `bcmp`.
840 int MemcmpInterceptorCommon(void *ctx,
841 int (*real_fn)(const void *, const void *, uptr),
842 const void *a1, const void *a2, uptr size) {
843 if (common_flags()->intercept_memcmp) {
844 if (common_flags()->strict_memcmp) {
845 // Check the entire regions even if the first bytes of the buffers are
846 // different.
847 COMMON_INTERCEPTOR_READ_RANGE(ctx, a1, size);
848 COMMON_INTERCEPTOR_READ_RANGE(ctx, a2, size);
849 // Fallthrough to REAL(memcmp) below.
850 } else {
851 unsigned char c1 = 0, c2 = 0;
852 const unsigned char *s1 = (const unsigned char*)a1;
853 const unsigned char *s2 = (const unsigned char*)a2;
854 uptr i;
855 for (i = 0; i < size; i++) {
856 c1 = s1[i];
857 c2 = s2[i];
858 if (c1 != c2) break;
859 }
860 COMMON_INTERCEPTOR_READ_RANGE(ctx, s1, Min(i + 1, size));
861 COMMON_INTERCEPTOR_READ_RANGE(ctx, s2, Min(i + 1, size));
862 int r = CharCmpX(c1, c2);
863 CALL_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_memcmp, GET_CALLER_PC(),
864 a1, a2, size, r);
865 return r;
866 }
867 }
868 int result = real_fn(a1, a2, size);
869 CALL_WEAK_INTERCEPTOR_HOOK(__sanitizer_weak_hook_memcmp, GET_CALLER_PC(), a1,
870 a2, size, result);
871 return result;
872 }
874 INTERCEPTOR(int, memcmp, const void *a1, const void *a2, uptr size) {
875 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
876 return internal_memcmp(a1, a2, size);
877 void *ctx;
878 COMMON_INTERCEPTOR_ENTER(ctx, memcmp, a1, a2, size);
879 return MemcmpInterceptorCommon(ctx, REAL(memcmp), a1, a2, size);
880 }
882 #define INIT_MEMCMP COMMON_INTERCEPT_FUNCTION(memcmp)
883 #else
884 #define INIT_MEMCMP
885 #endif
887 #if SANITIZER_INTERCEPT_BCMP
888 INTERCEPTOR(int, bcmp, const void *a1, const void *a2, uptr size) {
889 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
890 return internal_memcmp(a1, a2, size);
891 void *ctx;
892 COMMON_INTERCEPTOR_ENTER(ctx, bcmp, a1, a2, size);
893 return MemcmpInterceptorCommon(ctx, REAL(bcmp), a1, a2, size);
894 }
896 #define INIT_BCMP COMMON_INTERCEPT_FUNCTION(bcmp)
897 #else
898 #define INIT_BCMP
899 #endif
901 #if SANITIZER_INTERCEPT_MEMCHR
902 INTERCEPTOR(void*, memchr, const void *s, int c, SIZE_T n) {
903 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
904 return internal_memchr(s, c, n);
905 void *ctx;
906 COMMON_INTERCEPTOR_ENTER(ctx, memchr, s, c, n);
907 #if SANITIZER_WINDOWS
908 void *res;
909 if (REAL(memchr)) {
910 res = REAL(memchr)(s, c, n);
911 } else {
912 res = internal_memchr(s, c, n);
913 }
914 #else
915 void *res = REAL(memchr)(s, c, n);
916 #endif
917 uptr len = res ? (char *)res - (const char *)s + 1 : n;
918 COMMON_INTERCEPTOR_READ_RANGE(ctx, s, len);
919 return res;
920 }
922 #define INIT_MEMCHR COMMON_INTERCEPT_FUNCTION(memchr)
923 #else
924 #define INIT_MEMCHR
925 #endif
927 #if SANITIZER_INTERCEPT_MEMRCHR
928 INTERCEPTOR(void*, memrchr, const void *s, int c, SIZE_T n) {
929 void *ctx;
930 COMMON_INTERCEPTOR_ENTER(ctx, memrchr, s, c, n);
931 COMMON_INTERCEPTOR_READ_RANGE(ctx, s, n);
932 return REAL(memrchr)(s, c, n);
933 }
935 #define INIT_MEMRCHR COMMON_INTERCEPT_FUNCTION(memrchr)
936 #else
937 #define INIT_MEMRCHR
938 #endif
940 #if SANITIZER_INTERCEPT_FREXP
941 INTERCEPTOR(double, frexp, double x, int *exp) {
942 void *ctx;
943 COMMON_INTERCEPTOR_ENTER(ctx, frexp, x, exp);
944 // Assuming frexp() always writes to |exp|.
945 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, exp, sizeof(*exp));
946 double res = REAL(frexp)(x, exp);
947 COMMON_INTERCEPTOR_INITIALIZE_RANGE(exp, sizeof(*exp));
948 return res;
949 }
951 #define INIT_FREXP COMMON_INTERCEPT_FUNCTION(frexp);
952 #else
953 #define INIT_FREXP
954 #endif // SANITIZER_INTERCEPT_FREXP
956 #if SANITIZER_INTERCEPT_FREXPF_FREXPL
957 INTERCEPTOR(float, frexpf, float x, int *exp) {
958 void *ctx;
959 COMMON_INTERCEPTOR_ENTER(ctx, frexpf, x, exp);
960 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, exp, sizeof(*exp));
961 float res = REAL(frexpf)(x, exp);
962 COMMON_INTERCEPTOR_INITIALIZE_RANGE(exp, sizeof(*exp));
963 return res;
964 }
966 INTERCEPTOR(long double, frexpl, long double x, int *exp) {
967 void *ctx;
968 COMMON_INTERCEPTOR_ENTER(ctx, frexpl, x, exp);
969 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, exp, sizeof(*exp));
970 long double res = REAL(frexpl)(x, exp);
971 COMMON_INTERCEPTOR_INITIALIZE_RANGE(exp, sizeof(*exp));
972 return res;
973 }
975 #define INIT_FREXPF_FREXPL \
976 COMMON_INTERCEPT_FUNCTION(frexpf); \
977 COMMON_INTERCEPT_FUNCTION_LDBL(frexpl)
978 #else
979 #define INIT_FREXPF_FREXPL
980 #endif // SANITIZER_INTERCEPT_FREXPF_FREXPL
982 #if SI_POSIX
983 static void write_iovec(void *ctx, struct __sanitizer_iovec *iovec,
984 SIZE_T iovlen, SIZE_T maxlen) {
985 for (SIZE_T i = 0; i < iovlen && maxlen; ++i) {
986 SSIZE_T sz = Min(iovec[i].iov_len, maxlen);
987 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, iovec[i].iov_base, sz);
988 maxlen -= sz;
989 }
990 }
992 static void read_iovec(void *ctx, struct __sanitizer_iovec *iovec,
993 SIZE_T iovlen, SIZE_T maxlen) {
994 COMMON_INTERCEPTOR_READ_RANGE(ctx, iovec, sizeof(*iovec) * iovlen);
995 for (SIZE_T i = 0; i < iovlen && maxlen; ++i) {
996 SSIZE_T sz = Min(iovec[i].iov_len, maxlen);
997 COMMON_INTERCEPTOR_READ_RANGE(ctx, iovec[i].iov_base, sz);
998 maxlen -= sz;
999 }
1000 }
1001 #endif
1003 #if SANITIZER_INTERCEPT_READ
1004 INTERCEPTOR(SSIZE_T, read, int fd, void *ptr, SIZE_T count) {
1005 void *ctx;
1006 COMMON_INTERCEPTOR_ENTER(ctx, read, fd, ptr, count);
1007 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
1008 // FIXME: under ASan the call below may write to freed memory and corrupt
1009 // its metadata. See
1010 // https://github.com/google/sanitizers/issues/321.
1011 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(read)(fd, ptr, count);
1012 if (res > 0) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ptr, res);
1013 if (res >= 0 && fd >= 0) COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
1014 return res;
1015 }
1016 #define INIT_READ COMMON_INTERCEPT_FUNCTION(read)
1017 #else
1018 #define INIT_READ
1019 #endif
1021 #if SANITIZER_INTERCEPT_FREAD
1022 INTERCEPTOR(SIZE_T, fread, void *ptr, SIZE_T size, SIZE_T nmemb, void *file) {
1023 // libc file streams can call user-supplied functions, see fopencookie.
1024 void *ctx;
1025 COMMON_INTERCEPTOR_ENTER(ctx, fread, ptr, size, nmemb, file);
1026 // FIXME: under ASan the call below may write to freed memory and corrupt
1027 // its metadata. See
1028 // https://github.com/google/sanitizers/issues/321.
1029 SIZE_T res = REAL(fread)(ptr, size, nmemb, file);
1030 if (res > 0) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ptr, res * size);
1031 return res;
1032 }
1033 #define INIT_FREAD COMMON_INTERCEPT_FUNCTION(fread)
1034 #else
1035 #define INIT_FREAD
1036 #endif
1038 #if SANITIZER_INTERCEPT_PREAD
1039 INTERCEPTOR(SSIZE_T, pread, int fd, void *ptr, SIZE_T count, OFF_T offset) {
1040 void *ctx;
1041 COMMON_INTERCEPTOR_ENTER(ctx, pread, fd, ptr, count, offset);
1042 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
1043 // FIXME: under ASan the call below may write to freed memory and corrupt
1044 // its metadata. See
1045 // https://github.com/google/sanitizers/issues/321.
1046 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(pread)(fd, ptr, count, offset);
1047 if (res > 0) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ptr, res);
1048 if (res >= 0 && fd >= 0) COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
1049 return res;
1050 }
1051 #define INIT_PREAD COMMON_INTERCEPT_FUNCTION(pread)
1052 #else
1053 #define INIT_PREAD
1054 #endif
1056 #if SANITIZER_INTERCEPT_PREAD64
1057 INTERCEPTOR(SSIZE_T, pread64, int fd, void *ptr, SIZE_T count, OFF64_T offset) {
1058 void *ctx;
1059 COMMON_INTERCEPTOR_ENTER(ctx, pread64, fd, ptr, count, offset);
1060 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
1061 // FIXME: under ASan the call below may write to freed memory and corrupt
1062 // its metadata. See
1063 // https://github.com/google/sanitizers/issues/321.
1064 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(pread64)(fd, ptr, count, offset);
1065 if (res > 0) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ptr, res);
1066 if (res >= 0 && fd >= 0) COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
1067 return res;
1068 }
1069 #define INIT_PREAD64 COMMON_INTERCEPT_FUNCTION(pread64)
1070 #else
1071 #define INIT_PREAD64
1072 #endif
1074 #if SANITIZER_INTERCEPT_READV
1075 INTERCEPTOR_WITH_SUFFIX(SSIZE_T, readv, int fd, __sanitizer_iovec *iov,
1076 int iovcnt) {
1077 void *ctx;
1078 COMMON_INTERCEPTOR_ENTER(ctx, readv, fd, iov, iovcnt);
1079 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
1080 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(readv)(fd, iov, iovcnt);
1081 if (res > 0) write_iovec(ctx, iov, iovcnt, res);
1082 if (res >= 0 && fd >= 0) COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
1083 return res;
1084 }
1085 #define INIT_READV COMMON_INTERCEPT_FUNCTION(readv)
1086 #else
1087 #define INIT_READV
1088 #endif
1090 #if SANITIZER_INTERCEPT_PREADV
1091 INTERCEPTOR(SSIZE_T, preadv, int fd, __sanitizer_iovec *iov, int iovcnt,
1092 OFF_T offset) {
1093 void *ctx;
1094 COMMON_INTERCEPTOR_ENTER(ctx, preadv, fd, iov, iovcnt, offset);
1095 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
1096 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(preadv)(fd, iov, iovcnt, offset);
1097 if (res > 0) write_iovec(ctx, iov, iovcnt, res);
1098 if (res >= 0 && fd >= 0) COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
1099 return res;
1100 }
1101 #define INIT_PREADV COMMON_INTERCEPT_FUNCTION(preadv)
1102 #else
1103 #define INIT_PREADV
1104 #endif
1106 #if SANITIZER_INTERCEPT_PREADV64
1107 INTERCEPTOR(SSIZE_T, preadv64, int fd, __sanitizer_iovec *iov, int iovcnt,
1108 OFF64_T offset) {
1109 void *ctx;
1110 COMMON_INTERCEPTOR_ENTER(ctx, preadv64, fd, iov, iovcnt, offset);
1111 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
1112 SSIZE_T res =
1113 COMMON_INTERCEPTOR_BLOCK_REAL(preadv64)(fd, iov, iovcnt, offset);
1114 if (res > 0) write_iovec(ctx, iov, iovcnt, res);
1115 if (res >= 0 && fd >= 0) COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
1116 return res;
1117 }
1118 #define INIT_PREADV64 COMMON_INTERCEPT_FUNCTION(preadv64)
1119 #else
1120 #define INIT_PREADV64
1121 #endif
1123 #if SANITIZER_INTERCEPT_WRITE
1124 INTERCEPTOR(SSIZE_T, write, int fd, void *ptr, SIZE_T count) {
1125 void *ctx;
1126 COMMON_INTERCEPTOR_ENTER(ctx, write, fd, ptr, count);
1127 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
1128 if (fd >= 0) COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd);
1129 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(write)(fd, ptr, count);
1130 // FIXME: this check should be _before_ the call to
1131 // COMMON_INTERCEPTOR_BLOCK_REAL(write), not after
1132 if (res > 0) COMMON_INTERCEPTOR_READ_RANGE(ctx, ptr, res);
1133 return res;
1134 }
1135 #define INIT_WRITE COMMON_INTERCEPT_FUNCTION(write)
1136 #else
1137 #define INIT_WRITE
1138 #endif
1140 #if SANITIZER_INTERCEPT_FWRITE
1141 INTERCEPTOR(SIZE_T, fwrite, const void *p, uptr size, uptr nmemb, void *file) {
1142 // libc file streams can call user-supplied functions, see fopencookie.
1143 void *ctx;
1144 COMMON_INTERCEPTOR_ENTER(ctx, fwrite, p, size, nmemb, file);
1145 SIZE_T res = REAL(fwrite)(p, size, nmemb, file);
1146 if (res > 0) COMMON_INTERCEPTOR_READ_RANGE(ctx, p, res * size);
1147 return res;
1148 }
1149 #define INIT_FWRITE COMMON_INTERCEPT_FUNCTION(fwrite)
1150 #else
1151 #define INIT_FWRITE
1152 #endif
1154 #if SANITIZER_INTERCEPT_PWRITE
1155 INTERCEPTOR(SSIZE_T, pwrite, int fd, void *ptr, SIZE_T count, OFF_T offset) {
1156 void *ctx;
1157 COMMON_INTERCEPTOR_ENTER(ctx, pwrite, fd, ptr, count, offset);
1158 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
1159 if (fd >= 0) COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd);
1160 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(pwrite)(fd, ptr, count, offset);
1161 if (res > 0) COMMON_INTERCEPTOR_READ_RANGE(ctx, ptr, res);
1162 return res;
1163 }
1164 #define INIT_PWRITE COMMON_INTERCEPT_FUNCTION(pwrite)
1165 #else
1166 #define INIT_PWRITE
1167 #endif
1169 #if SANITIZER_INTERCEPT_PWRITE64
1170 INTERCEPTOR(SSIZE_T, pwrite64, int fd, void *ptr, OFF64_T count,
1171 OFF64_T offset) {
1172 void *ctx;
1173 COMMON_INTERCEPTOR_ENTER(ctx, pwrite64, fd, ptr, count, offset);
1174 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
1175 if (fd >= 0) COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd);
1176 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(pwrite64)(fd, ptr, count, offset);
1177 if (res > 0) COMMON_INTERCEPTOR_READ_RANGE(ctx, ptr, res);
1178 return res;
1179 }
1180 #define INIT_PWRITE64 COMMON_INTERCEPT_FUNCTION(pwrite64)
1181 #else
1182 #define INIT_PWRITE64
1183 #endif
1185 #if SANITIZER_INTERCEPT_WRITEV
1186 INTERCEPTOR_WITH_SUFFIX(SSIZE_T, writev, int fd, __sanitizer_iovec *iov,
1187 int iovcnt) {
1188 void *ctx;
1189 COMMON_INTERCEPTOR_ENTER(ctx, writev, fd, iov, iovcnt);
1190 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
1191 if (fd >= 0) COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd);
1192 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(writev)(fd, iov, iovcnt);
1193 if (res > 0) read_iovec(ctx, iov, iovcnt, res);
1194 return res;
1195 }
1196 #define INIT_WRITEV COMMON_INTERCEPT_FUNCTION(writev)
1197 #else
1198 #define INIT_WRITEV
1199 #endif
1201 #if SANITIZER_INTERCEPT_PWRITEV
1202 INTERCEPTOR(SSIZE_T, pwritev, int fd, __sanitizer_iovec *iov, int iovcnt,
1203 OFF_T offset) {
1204 void *ctx;
1205 COMMON_INTERCEPTOR_ENTER(ctx, pwritev, fd, iov, iovcnt, offset);
1206 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
1207 if (fd >= 0) COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd);
1208 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(pwritev)(fd, iov, iovcnt, offset);
1209 if (res > 0) read_iovec(ctx, iov, iovcnt, res);
1210 return res;
1211 }
1212 #define INIT_PWRITEV COMMON_INTERCEPT_FUNCTION(pwritev)
1213 #else
1214 #define INIT_PWRITEV
1215 #endif
1217 #if SANITIZER_INTERCEPT_PWRITEV64
1218 INTERCEPTOR(SSIZE_T, pwritev64, int fd, __sanitizer_iovec *iov, int iovcnt,
1219 OFF64_T offset) {
1220 void *ctx;
1221 COMMON_INTERCEPTOR_ENTER(ctx, pwritev64, fd, iov, iovcnt, offset);
1222 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
1223 if (fd >= 0) COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd);
1224 SSIZE_T res =
1225 COMMON_INTERCEPTOR_BLOCK_REAL(pwritev64)(fd, iov, iovcnt, offset);
1226 if (res > 0) read_iovec(ctx, iov, iovcnt, res);
1227 return res;
1228 }
1229 #define INIT_PWRITEV64 COMMON_INTERCEPT_FUNCTION(pwritev64)
1230 #else
1231 #define INIT_PWRITEV64
1232 #endif
1234 #if SANITIZER_INTERCEPT_FGETS
1235 INTERCEPTOR(char *, fgets, char *s, SIZE_T size, void *file) {
1236 // libc file streams can call user-supplied functions, see fopencookie.
1237 void *ctx;
1238 COMMON_INTERCEPTOR_ENTER(ctx, fgets, s, size, file);
1239 // FIXME: under ASan the call below may write to freed memory and corrupt
1240 // its metadata. See
1241 // https://github.com/google/sanitizers/issues/321.
1242 char *res = REAL(fgets)(s, size, file);
1243 if (res)
1244 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, s, internal_strlen(s) + 1);
1245 return res;
1246 }
1247 #define INIT_FGETS COMMON_INTERCEPT_FUNCTION(fgets)
1248 #else
1249 #define INIT_FGETS
1250 #endif
1252 #if SANITIZER_INTERCEPT_FPUTS
1253 INTERCEPTOR_WITH_SUFFIX(int, fputs, char *s, void *file) {
1254 // libc file streams can call user-supplied functions, see fopencookie.
1255 void *ctx;
1256 COMMON_INTERCEPTOR_ENTER(ctx, fputs, s, file);
1257 if (!SANITIZER_APPLE || s) { // `fputs(NULL, file)` is supported on Darwin.
1258 COMMON_INTERCEPTOR_READ_RANGE(ctx, s, internal_strlen(s) + 1);
1259 }
1260 return REAL(fputs)(s, file);
1261 }
1262 #define INIT_FPUTS COMMON_INTERCEPT_FUNCTION(fputs)
1263 #else
1264 #define INIT_FPUTS
1265 #endif
1267 #if SANITIZER_INTERCEPT_PUTS
1268 INTERCEPTOR(int, puts, char *s) {
1269 // libc file streams can call user-supplied functions, see fopencookie.
1270 void *ctx;
1271 COMMON_INTERCEPTOR_ENTER(ctx, puts, s);
1272 if (!SANITIZER_APPLE || s) { // `puts(NULL)` is supported on Darwin.
1273 COMMON_INTERCEPTOR_READ_RANGE(ctx, s, internal_strlen(s) + 1);
1274 }
1275 return REAL(puts)(s);
1276 }
1277 #define INIT_PUTS COMMON_INTERCEPT_FUNCTION(puts)
1278 #else
1279 #define INIT_PUTS
1280 #endif
1282 #if SANITIZER_INTERCEPT_PRCTL
1283 INTERCEPTOR(int, prctl, int option, unsigned long arg2, unsigned long arg3,
1284 unsigned long arg4, unsigned long arg5) {
1285 void *ctx;
1286 COMMON_INTERCEPTOR_ENTER(ctx, prctl, option, arg2, arg3, arg4, arg5);
1287 static const int PR_SET_NAME = 15;
1288 static const int PR_GET_NAME = 16;
1289 static const int PR_SET_VMA = 0x53564d41;
1290 static const int PR_SCHED_CORE = 62;
1291 static const int PR_SCHED_CORE_GET = 0;
1292 static const int PR_GET_PDEATHSIG = 2;
1294 # if !SANITIZER_ANDROID
1295 static const int PR_SET_SECCOMP = 22;
1296 static const int SECCOMP_MODE_FILTER = 2;
1297 # endif
1298 if (option == PR_SET_VMA && arg2 == 0UL) {
1299 char *name = (char *)arg5;
1300 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
1301 }
1302 int res = REAL(prctl)(option, arg2, arg3, arg4, arg5);
1303 if (option == PR_SET_NAME) {
1304 char buff[16];
1305 internal_strncpy(buff, (char *)arg2, 15);
1306 buff[15] = 0;
1307 COMMON_INTERCEPTOR_SET_THREAD_NAME(ctx, buff);
1308 } else if (res == 0 && option == PR_GET_NAME) {
1309 char *name = (char *)arg2;
1310 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, name, internal_strlen(name) + 1);
1311 } else if (res != -1 && option == PR_SCHED_CORE &&
1312 arg2 == PR_SCHED_CORE_GET) {
1313 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, (u64 *)(arg5), sizeof(u64));
1314 } else if (res != -1 && option == PR_GET_PDEATHSIG) {
1315 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, (u64 *)(arg2), sizeof(int));
1316 # if !SANITIZER_ANDROID
1317 } else if (res != -1 && option == PR_SET_SECCOMP &&
1318 arg2 == SECCOMP_MODE_FILTER) {
1319 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, (u64 *)(arg3), struct_sock_fprog_sz);
1320 # endif
1321 }
1322 return res;
1323 }
1324 #define INIT_PRCTL COMMON_INTERCEPT_FUNCTION(prctl)
1325 #else
1326 #define INIT_PRCTL
1327 #endif // SANITIZER_INTERCEPT_PRCTL
1329 #if SANITIZER_INTERCEPT_TIME
1330 INTERCEPTOR(unsigned long, time, unsigned long *t) {
1331 void *ctx;
1332 COMMON_INTERCEPTOR_ENTER(ctx, time, t);
1333 unsigned long local_t;
1334 unsigned long res = REAL(time)(&local_t);
1335 if (t && res != (unsigned long)-1) {
1336 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, t, sizeof(*t));
1337 *t = local_t;
1338 }
1339 return res;
1340 }
1341 #define INIT_TIME COMMON_INTERCEPT_FUNCTION(time);
1342 #else
1343 #define INIT_TIME
1344 #endif // SANITIZER_INTERCEPT_TIME
1346 #if SANITIZER_INTERCEPT_LOCALTIME_AND_FRIENDS
1347 static void unpoison_tm(void *ctx, __sanitizer_tm *tm) {
1348 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, tm, sizeof(*tm));
1349 #if !SANITIZER_SOLARIS
1350 if (tm->tm_zone) {
1351 // Can not use COMMON_INTERCEPTOR_WRITE_RANGE here, because tm->tm_zone
1352 // can point to shared memory and tsan would report a data race.
1353 COMMON_INTERCEPTOR_INITIALIZE_RANGE(tm->tm_zone,
1354 internal_strlen(tm->tm_zone) + 1);
1355 }
1356 #endif
1357 }
1358 INTERCEPTOR(__sanitizer_tm *, localtime, unsigned long *timep) {
1359 void *ctx;
1360 COMMON_INTERCEPTOR_ENTER(ctx, localtime, timep);
1361 __sanitizer_tm *res = REAL(localtime)(timep);
1362 if (res) {
1363 COMMON_INTERCEPTOR_READ_RANGE(ctx, timep, sizeof(*timep));
1364 unpoison_tm(ctx, res);
1365 }
1366 return res;
1367 }
1368 INTERCEPTOR(__sanitizer_tm *, localtime_r, unsigned long *timep, void *result) {
1369 void *ctx;
1370 COMMON_INTERCEPTOR_ENTER(ctx, localtime_r, timep, result);
1371 __sanitizer_tm *res = REAL(localtime_r)(timep, result);
1372 if (res) {
1373 COMMON_INTERCEPTOR_READ_RANGE(ctx, timep, sizeof(*timep));
1374 unpoison_tm(ctx, res);
1375 }
1376 return res;
1377 }
1378 INTERCEPTOR(__sanitizer_tm *, gmtime, unsigned long *timep) {
1379 void *ctx;
1380 COMMON_INTERCEPTOR_ENTER(ctx, gmtime, timep);
1381 __sanitizer_tm *res = REAL(gmtime)(timep);
1382 if (res) {
1383 COMMON_INTERCEPTOR_READ_RANGE(ctx, timep, sizeof(*timep));
1384 unpoison_tm(ctx, res);
1385 }
1386 return res;
1387 }
1388 INTERCEPTOR(__sanitizer_tm *, gmtime_r, unsigned long *timep, void *result) {
1389 void *ctx;
1390 COMMON_INTERCEPTOR_ENTER(ctx, gmtime_r, timep, result);
1391 __sanitizer_tm *res = REAL(gmtime_r)(timep, result);
1392 if (res) {
1393 COMMON_INTERCEPTOR_READ_RANGE(ctx, timep, sizeof(*timep));
1394 unpoison_tm(ctx, res);
1395 }
1396 return res;
1397 }
1398 INTERCEPTOR(char *, ctime, unsigned long *timep) {
1399 void *ctx;
1400 COMMON_INTERCEPTOR_ENTER(ctx, ctime, timep);
1401 // FIXME: under ASan the call below may write to freed memory and corrupt
1402 // its metadata. See
1403 // https://github.com/google/sanitizers/issues/321.
1404 char *res = REAL(ctime)(timep);
1405 if (res) {
1406 COMMON_INTERCEPTOR_READ_RANGE(ctx, timep, sizeof(*timep));
1407 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
1408 }
1409 return res;
1410 }
1411 INTERCEPTOR(char *, ctime_r, unsigned long *timep, char *result) {
1412 void *ctx;
1413 COMMON_INTERCEPTOR_ENTER(ctx, ctime_r, timep, result);
1414 // FIXME: under ASan the call below may write to freed memory and corrupt
1415 // its metadata. See
1416 // https://github.com/google/sanitizers/issues/321.
1417 char *res = REAL(ctime_r)(timep, result);
1418 if (res) {
1419 COMMON_INTERCEPTOR_READ_RANGE(ctx, timep, sizeof(*timep));
1420 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
1421 }
1422 return res;
1423 }
1424 INTERCEPTOR(char *, asctime, __sanitizer_tm *tm) {
1425 void *ctx;
1426 COMMON_INTERCEPTOR_ENTER(ctx, asctime, tm);
1427 // FIXME: under ASan the call below may write to freed memory and corrupt
1428 // its metadata. See
1429 // https://github.com/google/sanitizers/issues/321.
1430 char *res = REAL(asctime)(tm);
1431 if (res) {
1432 COMMON_INTERCEPTOR_READ_RANGE(ctx, tm, sizeof(*tm));
1433 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
1434 }
1435 return res;
1436 }
1437 INTERCEPTOR(char *, asctime_r, __sanitizer_tm *tm, char *result) {
1438 void *ctx;
1439 COMMON_INTERCEPTOR_ENTER(ctx, asctime_r, tm, result);
1440 // FIXME: under ASan the call below may write to freed memory and corrupt
1441 // its metadata. See
1442 // https://github.com/google/sanitizers/issues/321.
1443 char *res = REAL(asctime_r)(tm, result);
1444 if (res) {
1445 COMMON_INTERCEPTOR_READ_RANGE(ctx, tm, sizeof(*tm));
1446 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
1447 }
1448 return res;
1449 }
1450 INTERCEPTOR(long, mktime, __sanitizer_tm *tm) {
1451 void *ctx;
1452 COMMON_INTERCEPTOR_ENTER(ctx, mktime, tm);
1453 COMMON_INTERCEPTOR_READ_RANGE(ctx, &tm->tm_sec, sizeof(tm->tm_sec));
1454 COMMON_INTERCEPTOR_READ_RANGE(ctx, &tm->tm_min, sizeof(tm->tm_min));
1455 COMMON_INTERCEPTOR_READ_RANGE(ctx, &tm->tm_hour, sizeof(tm->tm_hour));
1456 COMMON_INTERCEPTOR_READ_RANGE(ctx, &tm->tm_mday, sizeof(tm->tm_mday));
1457 COMMON_INTERCEPTOR_READ_RANGE(ctx, &tm->tm_mon, sizeof(tm->tm_mon));
1458 COMMON_INTERCEPTOR_READ_RANGE(ctx, &tm->tm_year, sizeof(tm->tm_year));
1459 COMMON_INTERCEPTOR_READ_RANGE(ctx, &tm->tm_isdst, sizeof(tm->tm_isdst));
1460 long res = REAL(mktime)(tm);
1461 if (res != -1) unpoison_tm(ctx, tm);
1462 return res;
1463 }
1464 #define INIT_LOCALTIME_AND_FRIENDS \
1465 COMMON_INTERCEPT_FUNCTION(localtime); \
1466 COMMON_INTERCEPT_FUNCTION(localtime_r); \
1467 COMMON_INTERCEPT_FUNCTION(gmtime); \
1468 COMMON_INTERCEPT_FUNCTION(gmtime_r); \
1469 COMMON_INTERCEPT_FUNCTION(ctime); \
1470 COMMON_INTERCEPT_FUNCTION(ctime_r); \
1471 COMMON_INTERCEPT_FUNCTION(asctime); \
1472 COMMON_INTERCEPT_FUNCTION(asctime_r); \
1473 COMMON_INTERCEPT_FUNCTION(mktime);
1474 #else
1475 #define INIT_LOCALTIME_AND_FRIENDS
1476 #endif // SANITIZER_INTERCEPT_LOCALTIME_AND_FRIENDS
1478 #if SANITIZER_INTERCEPT_STRPTIME
1479 INTERCEPTOR(char *, strptime, char *s, char *format, __sanitizer_tm *tm) {
1480 void *ctx;
1481 COMMON_INTERCEPTOR_ENTER(ctx, strptime, s, format, tm);
1482 if (format)
1483 COMMON_INTERCEPTOR_READ_RANGE(ctx, format, internal_strlen(format) + 1);
1484 // FIXME: under ASan the call below may write to freed memory and corrupt
1485 // its metadata. See
1486 // https://github.com/google/sanitizers/issues/321.
1487 char *res = REAL(strptime)(s, format, tm);
1488 COMMON_INTERCEPTOR_READ_STRING(ctx, s, res ? res - s : 0);
1489 if (res && tm) {
1490 // Do not call unpoison_tm here, because strptime does not, in fact,
1491 // initialize the entire struct tm. For example, tm_zone pointer is left
1492 // uninitialized.
1493 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, tm, sizeof(*tm));
1494 }
1495 return res;
1496 }
1497 #define INIT_STRPTIME COMMON_INTERCEPT_FUNCTION(strptime);
1498 #else
1499 #define INIT_STRPTIME
1500 #endif
1502 #if SANITIZER_INTERCEPT_SCANF || SANITIZER_INTERCEPT_PRINTF
1503 #include "sanitizer_common_interceptors_format.inc"
1505 #define FORMAT_INTERCEPTOR_IMPL(name, vname, ...) \
1506 { \
1507 void *ctx; \
1508 va_list ap; \
1509 va_start(ap, format); \
1510 COMMON_INTERCEPTOR_ENTER(ctx, vname, __VA_ARGS__, ap); \
1511 int res = WRAP(vname)(__VA_ARGS__, ap); \
1512 va_end(ap); \
1513 return res; \
1514 }
1516 #endif
1518 #if SANITIZER_INTERCEPT_SCANF
1520 #define VSCANF_INTERCEPTOR_IMPL(vname, allowGnuMalloc, ...) \
1521 { \
1522 void *ctx; \
1523 COMMON_INTERCEPTOR_ENTER(ctx, vname, __VA_ARGS__); \
1524 va_list aq; \
1525 va_copy(aq, ap); \
1526 int res = REAL(vname)(__VA_ARGS__); \
1527 if (res > 0) \
1528 scanf_common(ctx, res, allowGnuMalloc, format, aq); \
1529 va_end(aq); \
1530 return res; \
1531 }
1533 INTERCEPTOR(int, vscanf, const char *format, va_list ap)
1534 VSCANF_INTERCEPTOR_IMPL(vscanf, true, format, ap)
1536 INTERCEPTOR(int, vsscanf, const char *str, const char *format, va_list ap)
1537 VSCANF_INTERCEPTOR_IMPL(vsscanf, true, str, format, ap)
1539 INTERCEPTOR(int, vfscanf, void *stream, const char *format, va_list ap)
1540 VSCANF_INTERCEPTOR_IMPL(vfscanf, true, stream, format, ap)
1542 #if SANITIZER_INTERCEPT_ISOC99_SCANF
1543 INTERCEPTOR(int, __isoc99_vscanf, const char *format, va_list ap)
1544 VSCANF_INTERCEPTOR_IMPL(__isoc99_vscanf, false, format, ap)
1546 INTERCEPTOR(int, __isoc99_vsscanf, const char *str, const char *format,
1547 va_list ap)
1548 VSCANF_INTERCEPTOR_IMPL(__isoc99_vsscanf, false, str, format, ap)
1550 INTERCEPTOR(int, __isoc99_vfscanf, void *stream, const char *format, va_list ap)
1551 VSCANF_INTERCEPTOR_IMPL(__isoc99_vfscanf, false, stream, format, ap)
1553 INTERCEPTOR(int, __isoc23_vscanf, const char *format, va_list ap)
1554 VSCANF_INTERCEPTOR_IMPL(__isoc23_vscanf, false, format, ap)
1556 INTERCEPTOR(int, __isoc23_vsscanf, const char *str, const char *format,
1557 va_list ap)
1558 VSCANF_INTERCEPTOR_IMPL(__isoc23_vsscanf, false, str, format, ap)
1560 INTERCEPTOR(int, __isoc23_vfscanf, void *stream, const char *format, va_list ap)
1561 VSCANF_INTERCEPTOR_IMPL(__isoc23_vfscanf, false, stream, format, ap)
1562 #endif // SANITIZER_INTERCEPT_ISOC99_SCANF
1564 INTERCEPTOR(int, scanf, const char *format, ...)
1565 FORMAT_INTERCEPTOR_IMPL(scanf, vscanf, format)
1567 INTERCEPTOR(int, fscanf, void *stream, const char *format, ...)
1568 FORMAT_INTERCEPTOR_IMPL(fscanf, vfscanf, stream, format)
1570 INTERCEPTOR(int, sscanf, const char *str, const char *format, ...)
1571 FORMAT_INTERCEPTOR_IMPL(sscanf, vsscanf, str, format)
1573 #if SANITIZER_INTERCEPT_ISOC99_SCANF
1574 INTERCEPTOR(int, __isoc99_scanf, const char *format, ...)
1575 FORMAT_INTERCEPTOR_IMPL(__isoc99_scanf, __isoc99_vscanf, format)
1577 INTERCEPTOR(int, __isoc99_fscanf, void *stream, const char *format, ...)
1578 FORMAT_INTERCEPTOR_IMPL(__isoc99_fscanf, __isoc99_vfscanf, stream, format)
1580 INTERCEPTOR(int, __isoc99_sscanf, const char *str, const char *format, ...)
1581 FORMAT_INTERCEPTOR_IMPL(__isoc99_sscanf, __isoc99_vsscanf, str, format)
1583 INTERCEPTOR(int, __isoc23_scanf, const char *format, ...)
1584 FORMAT_INTERCEPTOR_IMPL(__isoc23_scanf, __isoc23_vscanf, format)
1586 INTERCEPTOR(int, __isoc23_fscanf, void *stream, const char *format, ...)
1587 FORMAT_INTERCEPTOR_IMPL(__isoc23_fscanf, __isoc23_vfscanf, stream, format)
1589 INTERCEPTOR(int, __isoc23_sscanf, const char *str, const char *format, ...)
1590 FORMAT_INTERCEPTOR_IMPL(__isoc23_sscanf, __isoc23_vsscanf, str, format)
1591 #endif
1593 #endif
1595 #if SANITIZER_INTERCEPT_SCANF
1596 #define INIT_SCANF \
1597 COMMON_INTERCEPT_FUNCTION_LDBL(scanf); \
1598 COMMON_INTERCEPT_FUNCTION_LDBL(sscanf); \
1599 COMMON_INTERCEPT_FUNCTION_LDBL(fscanf); \
1600 COMMON_INTERCEPT_FUNCTION_LDBL(vscanf); \
1601 COMMON_INTERCEPT_FUNCTION_LDBL(vsscanf); \
1602 COMMON_INTERCEPT_FUNCTION_LDBL(vfscanf);
1603 #else
1604 #define INIT_SCANF
1605 #endif
1607 #if SANITIZER_INTERCEPT_ISOC99_SCANF
1608 #define INIT_ISOC99_SCANF \
1609 COMMON_INTERCEPT_FUNCTION(__isoc99_scanf); \
1610 COMMON_INTERCEPT_FUNCTION(__isoc99_sscanf); \
1611 COMMON_INTERCEPT_FUNCTION(__isoc99_fscanf); \
1612 COMMON_INTERCEPT_FUNCTION(__isoc99_vscanf); \
1613 COMMON_INTERCEPT_FUNCTION(__isoc99_vsscanf); \
1614 COMMON_INTERCEPT_FUNCTION(__isoc99_vfscanf); \
1615 COMMON_INTERCEPT_FUNCTION(__isoc23_scanf); \
1616 COMMON_INTERCEPT_FUNCTION(__isoc23_sscanf); \
1617 COMMON_INTERCEPT_FUNCTION(__isoc23_fscanf); \
1618 COMMON_INTERCEPT_FUNCTION(__isoc23_vscanf); \
1619 COMMON_INTERCEPT_FUNCTION(__isoc23_vsscanf); \
1620 COMMON_INTERCEPT_FUNCTION(__isoc23_vfscanf);
1621 #else
1622 #define INIT_ISOC99_SCANF
1623 #endif
1625 #if SANITIZER_INTERCEPT_PRINTF
1627 #define VPRINTF_INTERCEPTOR_ENTER(vname, ...) \
1628 void *ctx; \
1629 COMMON_INTERCEPTOR_ENTER(ctx, vname, __VA_ARGS__); \
1630 va_list aq; \
1631 va_copy(aq, ap);
1633 #define VPRINTF_INTERCEPTOR_RETURN() \
1634 va_end(aq);
1636 #define VPRINTF_INTERCEPTOR_IMPL(vname, ...) \
1637 { \
1638 VPRINTF_INTERCEPTOR_ENTER(vname, __VA_ARGS__); \
1639 if (common_flags()->check_printf) \
1640 printf_common(ctx, format, aq); \
1641 int res = REAL(vname)(__VA_ARGS__); \
1642 VPRINTF_INTERCEPTOR_RETURN(); \
1643 return res; \
1644 }
1646 // FIXME: under ASan the REAL() call below may write to freed memory and
1647 // corrupt its metadata. See
1648 // https://github.com/google/sanitizers/issues/321.
1649 #define VSPRINTF_INTERCEPTOR_IMPL(vname, str, ...) \
1650 { \
1651 VPRINTF_INTERCEPTOR_ENTER(vname, str, __VA_ARGS__) \
1652 if (common_flags()->check_printf) { \
1653 printf_common(ctx, format, aq); \
1654 } \
1655 int res = REAL(vname)(str, __VA_ARGS__); \
1656 if (res >= 0) { \
1657 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, str, res + 1); \
1658 } \
1659 VPRINTF_INTERCEPTOR_RETURN(); \
1660 return res; \
1661 }
1663 // FIXME: under ASan the REAL() call below may write to freed memory and
1664 // corrupt its metadata. See
1665 // https://github.com/google/sanitizers/issues/321.
1666 #define VSNPRINTF_INTERCEPTOR_IMPL(vname, str, size, ...) \
1667 { \
1668 VPRINTF_INTERCEPTOR_ENTER(vname, str, size, __VA_ARGS__) \
1669 if (common_flags()->check_printf) { \
1670 printf_common(ctx, format, aq); \
1671 } \
1672 int res = REAL(vname)(str, size, __VA_ARGS__); \
1673 if (res >= 0) { \
1674 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, str, Min(size, (SIZE_T)(res + 1))); \
1675 } \
1676 VPRINTF_INTERCEPTOR_RETURN(); \
1677 return res; \
1678 }
1680 // FIXME: under ASan the REAL() call below may write to freed memory and
1681 // corrupt its metadata. See
1682 // https://github.com/google/sanitizers/issues/321.
1683 #define VASPRINTF_INTERCEPTOR_IMPL(vname, strp, ...) \
1684 { \
1685 VPRINTF_INTERCEPTOR_ENTER(vname, strp, __VA_ARGS__) \
1686 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, strp, sizeof(char *)); \
1687 if (common_flags()->check_printf) { \
1688 printf_common(ctx, format, aq); \
1689 } \
1690 int res = REAL(vname)(strp, __VA_ARGS__); \
1691 if (res >= 0) { \
1692 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *strp, res + 1); \
1693 } \
1694 VPRINTF_INTERCEPTOR_RETURN(); \
1695 return res; \
1696 }
1698 INTERCEPTOR(int, vprintf, const char *format, va_list ap)
1699 VPRINTF_INTERCEPTOR_IMPL(vprintf, format, ap)
1701 INTERCEPTOR(int, vfprintf, __sanitizer_FILE *stream, const char *format,
1702 va_list ap)
1703 VPRINTF_INTERCEPTOR_IMPL(vfprintf, stream, format, ap)
1705 INTERCEPTOR(int, vsnprintf, char *str, SIZE_T size, const char *format,
1706 va_list ap)
1707 VSNPRINTF_INTERCEPTOR_IMPL(vsnprintf, str, size, format, ap)
1709 #if SANITIZER_INTERCEPT___PRINTF_CHK
1710 INTERCEPTOR(int, __vsnprintf_chk, char *str, SIZE_T size, int flag,
1711 SIZE_T size_to, const char *format, va_list ap)
1712 VSNPRINTF_INTERCEPTOR_IMPL(vsnprintf, str, size, format, ap)
1713 #endif
1715 #if SANITIZER_INTERCEPT_PRINTF_L
1716 INTERCEPTOR(int, vsnprintf_l, char *str, SIZE_T size, void *loc,
1717 const char *format, va_list ap)
1718 VSNPRINTF_INTERCEPTOR_IMPL(vsnprintf_l, str, size, loc, format, ap)
1720 INTERCEPTOR(int, snprintf_l, char *str, SIZE_T size, void *loc,
1721 const char *format, ...)
1722 FORMAT_INTERCEPTOR_IMPL(snprintf_l, vsnprintf_l, str, size, loc, format)
1723 #endif // SANITIZER_INTERCEPT_PRINTF_L
1725 INTERCEPTOR(int, vsprintf, char *str, const char *format, va_list ap)
1726 VSPRINTF_INTERCEPTOR_IMPL(vsprintf, str, format, ap)
1728 #if SANITIZER_INTERCEPT___PRINTF_CHK
1729 INTERCEPTOR(int, __vsprintf_chk, char *str, int flag, SIZE_T size_to,
1730 const char *format, va_list ap)
1731 VSPRINTF_INTERCEPTOR_IMPL(vsprintf, str, format, ap)
1732 #endif
1734 INTERCEPTOR(int, vasprintf, char **strp, const char *format, va_list ap)
1735 VASPRINTF_INTERCEPTOR_IMPL(vasprintf, strp, format, ap)
1737 #if SANITIZER_INTERCEPT_ISOC99_PRINTF
1738 INTERCEPTOR(int, __isoc99_vprintf, const char *format, va_list ap)
1739 VPRINTF_INTERCEPTOR_IMPL(__isoc99_vprintf, format, ap)
1741 INTERCEPTOR(int, __isoc99_vfprintf, __sanitizer_FILE *stream,
1742 const char *format, va_list ap)
1743 VPRINTF_INTERCEPTOR_IMPL(__isoc99_vfprintf, stream, format, ap)
1745 INTERCEPTOR(int, __isoc99_vsnprintf, char *str, SIZE_T size, const char *format,
1746 va_list ap)
1747 VSNPRINTF_INTERCEPTOR_IMPL(__isoc99_vsnprintf, str, size, format, ap)
1749 INTERCEPTOR(int, __isoc99_vsprintf, char *str, const char *format,
1750 va_list ap)
1751 VSPRINTF_INTERCEPTOR_IMPL(__isoc99_vsprintf, str, format,
1752 ap)
1754 #endif // SANITIZER_INTERCEPT_ISOC99_PRINTF
1756 INTERCEPTOR(int, printf, const char *format, ...)
1757 FORMAT_INTERCEPTOR_IMPL(printf, vprintf, format)
1759 INTERCEPTOR(int, fprintf, __sanitizer_FILE *stream, const char *format, ...)
1760 FORMAT_INTERCEPTOR_IMPL(fprintf, vfprintf, stream, format)
1762 #if SANITIZER_INTERCEPT___PRINTF_CHK
1763 INTERCEPTOR(int, __fprintf_chk, __sanitizer_FILE *stream, SIZE_T size,
1764 const char *format, ...)
1765 FORMAT_INTERCEPTOR_IMPL(__fprintf_chk, vfprintf, stream, format)
1766 #endif
1768 INTERCEPTOR(int, sprintf, char *str, const char *format, ...)
1769 FORMAT_INTERCEPTOR_IMPL(sprintf, vsprintf, str, format)
1771 #if SANITIZER_INTERCEPT___PRINTF_CHK
1772 INTERCEPTOR(int, __sprintf_chk, char *str, int flag, SIZE_T size_to,
1773 const char *format, ...)
1774 FORMAT_INTERCEPTOR_IMPL(__sprintf_chk, vsprintf, str, format)
1775 #endif
1777 INTERCEPTOR(int, snprintf, char *str, SIZE_T size, const char *format, ...)
1778 FORMAT_INTERCEPTOR_IMPL(snprintf, vsnprintf, str, size, format)
1780 #if SANITIZER_INTERCEPT___PRINTF_CHK
1781 INTERCEPTOR(int, __snprintf_chk, char *str, SIZE_T size, int flag,
1782 SIZE_T size_to, const char *format, ...)
1783 FORMAT_INTERCEPTOR_IMPL(__snprintf_chk, vsnprintf, str, size, format)
1784 #endif
1786 INTERCEPTOR(int, asprintf, char **strp, const char *format, ...)
1787 FORMAT_INTERCEPTOR_IMPL(asprintf, vasprintf, strp, format)
1789 #if SANITIZER_INTERCEPT_ISOC99_PRINTF
1790 INTERCEPTOR(int, __isoc99_printf, const char *format, ...)
1791 FORMAT_INTERCEPTOR_IMPL(__isoc99_printf, __isoc99_vprintf, format)
1793 INTERCEPTOR(int, __isoc99_fprintf, __sanitizer_FILE *stream, const char *format,
1794 ...)
1795 FORMAT_INTERCEPTOR_IMPL(__isoc99_fprintf, __isoc99_vfprintf, stream, format)
1797 INTERCEPTOR(int, __isoc99_sprintf, char *str, const char *format, ...)
1798 FORMAT_INTERCEPTOR_IMPL(__isoc99_sprintf, __isoc99_vsprintf, str, format)
1800 INTERCEPTOR(int, __isoc99_snprintf, char *str, SIZE_T size,
1801 const char *format, ...)
1802 FORMAT_INTERCEPTOR_IMPL(__isoc99_snprintf, __isoc99_vsnprintf, str, size,
1803 format)
1805 #endif // SANITIZER_INTERCEPT_ISOC99_PRINTF
1807 #endif // SANITIZER_INTERCEPT_PRINTF
1809 #if SANITIZER_INTERCEPT_PRINTF
1810 #define INIT_PRINTF \
1811 COMMON_INTERCEPT_FUNCTION_LDBL(printf); \
1812 COMMON_INTERCEPT_FUNCTION_LDBL(sprintf); \
1813 COMMON_INTERCEPT_FUNCTION_LDBL(snprintf); \
1814 COMMON_INTERCEPT_FUNCTION_LDBL(asprintf); \
1815 COMMON_INTERCEPT_FUNCTION_LDBL(fprintf); \
1816 COMMON_INTERCEPT_FUNCTION_LDBL(vprintf); \
1817 COMMON_INTERCEPT_FUNCTION_LDBL(vsprintf); \
1818 COMMON_INTERCEPT_FUNCTION_LDBL(vsnprintf); \
1819 COMMON_INTERCEPT_FUNCTION_LDBL(vasprintf); \
1820 COMMON_INTERCEPT_FUNCTION_LDBL(vfprintf);
1821 #else
1822 #define INIT_PRINTF
1823 #endif
1825 #if SANITIZER_INTERCEPT___PRINTF_CHK
1826 #define INIT___PRINTF_CHK \
1827 COMMON_INTERCEPT_FUNCTION(__sprintf_chk); \
1828 COMMON_INTERCEPT_FUNCTION(__snprintf_chk); \
1829 COMMON_INTERCEPT_FUNCTION(__vsprintf_chk); \
1830 COMMON_INTERCEPT_FUNCTION(__vsnprintf_chk); \
1831 COMMON_INTERCEPT_FUNCTION(__fprintf_chk);
1832 #else
1833 #define INIT___PRINTF_CHK
1834 #endif
1836 #if SANITIZER_INTERCEPT_PRINTF_L
1837 #define INIT_PRINTF_L \
1838 COMMON_INTERCEPT_FUNCTION(snprintf_l); \
1839 COMMON_INTERCEPT_FUNCTION(vsnprintf_l);
1840 #else
1841 #define INIT_PRINTF_L
1842 #endif
1844 #if SANITIZER_INTERCEPT_ISOC99_PRINTF
1845 #define INIT_ISOC99_PRINTF \
1846 COMMON_INTERCEPT_FUNCTION(__isoc99_printf); \
1847 COMMON_INTERCEPT_FUNCTION(__isoc99_sprintf); \
1848 COMMON_INTERCEPT_FUNCTION(__isoc99_snprintf); \
1849 COMMON_INTERCEPT_FUNCTION(__isoc99_fprintf); \
1850 COMMON_INTERCEPT_FUNCTION(__isoc99_vprintf); \
1851 COMMON_INTERCEPT_FUNCTION(__isoc99_vsprintf); \
1852 COMMON_INTERCEPT_FUNCTION(__isoc99_vsnprintf); \
1853 COMMON_INTERCEPT_FUNCTION(__isoc99_vfprintf);
1854 #else
1855 #define INIT_ISOC99_PRINTF
1856 #endif
1858 #if SANITIZER_INTERCEPT_IOCTL
1859 #include "sanitizer_common_interceptors_ioctl.inc"
1860 #include "sanitizer_interceptors_ioctl_netbsd.inc"
1861 INTERCEPTOR(int, ioctl, int d, unsigned long request, ...) {
1862 // We need a frame pointer, because we call into ioctl_common_[pre|post] which
1863 // can trigger a report and we need to be able to unwind through this
1864 // function. On Mac in debug mode we might not have a frame pointer, because
1865 // ioctl_common_[pre|post] doesn't get inlined here.
1866 ENABLE_FRAME_POINTER;
1868 void *ctx;
1869 va_list ap;
1870 va_start(ap, request);
1871 void *arg = va_arg(ap, void *);
1872 va_end(ap);
1873 COMMON_INTERCEPTOR_ENTER(ctx, ioctl, d, request, arg);
1875 CHECK(ioctl_initialized);
1877 // Note: TSan does not use common flags, and they are zero-initialized.
1878 // This effectively disables ioctl handling in TSan.
1879 if (!common_flags()->handle_ioctl) return REAL(ioctl)(d, request, arg);
1881 // Although request is unsigned long, the rest of the interceptor uses it
1882 // as just "unsigned" to save space, because we know that all values fit in
1883 // "unsigned" - they are compile-time constants.
1885 const ioctl_desc *desc = ioctl_lookup(request);
1886 ioctl_desc decoded_desc;
1887 if (!desc) {
1888 VPrintf(2, "Decoding unknown ioctl 0x%lx\n", request);
1889 if (!ioctl_decode(request, &decoded_desc))
1890 Printf("WARNING: failed decoding unknown ioctl 0x%lx\n", request);
1891 else
1892 desc = &decoded_desc;
1893 }
1895 if (desc) ioctl_common_pre(ctx, desc, d, request, arg);
1896 int res = REAL(ioctl)(d, request, arg);
1897 // FIXME: some ioctls have different return values for success and failure.
1898 if (desc && res != -1) ioctl_common_post(ctx, desc, res, d, request, arg);
1899 return res;
1900 }
1901 #define INIT_IOCTL \
1902 ioctl_init(); \
1903 COMMON_INTERCEPT_FUNCTION(ioctl);
1904 #else
1905 #define INIT_IOCTL
1906 #endif
1908 #if SANITIZER_POSIX
1909 UNUSED static void unpoison_passwd(void *ctx, __sanitizer_passwd *pwd) {
1910 if (pwd) {
1911 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pwd, sizeof(*pwd));
1912 if (pwd->pw_name)
1913 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pwd->pw_name,
1914 internal_strlen(pwd->pw_name) + 1);
1915 if (pwd->pw_passwd)
1916 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pwd->pw_passwd,
1917 internal_strlen(pwd->pw_passwd) + 1);
1918 #if !SANITIZER_ANDROID
1919 if (pwd->pw_gecos)
1920 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pwd->pw_gecos,
1921 internal_strlen(pwd->pw_gecos) + 1);
1922 #endif
1923 #if SANITIZER_APPLE || SANITIZER_FREEBSD || SANITIZER_NETBSD
1924 if (pwd->pw_class)
1925 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pwd->pw_class,
1926 internal_strlen(pwd->pw_class) + 1);
1927 #endif
1928 if (pwd->pw_dir)
1929 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pwd->pw_dir,
1930 internal_strlen(pwd->pw_dir) + 1);
1931 if (pwd->pw_shell)
1932 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pwd->pw_shell,
1933 internal_strlen(pwd->pw_shell) + 1);
1934 }
1935 }
1937 UNUSED static void unpoison_group(void *ctx, __sanitizer_group *grp) {
1938 if (grp) {
1939 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, grp, sizeof(*grp));
1940 if (grp->gr_name)
1941 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, grp->gr_name,
1942 internal_strlen(grp->gr_name) + 1);
1943 if (grp->gr_passwd)
1944 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, grp->gr_passwd,
1945 internal_strlen(grp->gr_passwd) + 1);
1946 char **p = grp->gr_mem;
1947 for (; *p; ++p) {
1948 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *p, internal_strlen(*p) + 1);
1949 }
1950 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, grp->gr_mem,
1951 (p - grp->gr_mem + 1) * sizeof(*p));
1952 }
1953 }
1954 #endif // SANITIZER_POSIX
1956 #if SANITIZER_INTERCEPT_GETPWNAM_AND_FRIENDS
1957 INTERCEPTOR(__sanitizer_passwd *, getpwnam, const char *name) {
1958 void *ctx;
1959 COMMON_INTERCEPTOR_ENTER(ctx, getpwnam, name);
1960 if (name)
1961 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
1962 __sanitizer_passwd *res = REAL(getpwnam)(name);
1963 unpoison_passwd(ctx, res);
1964 return res;
1965 }
1966 INTERCEPTOR(__sanitizer_passwd *, getpwuid, u32 uid) {
1967 void *ctx;
1968 COMMON_INTERCEPTOR_ENTER(ctx, getpwuid, uid);
1969 __sanitizer_passwd *res = REAL(getpwuid)(uid);
1970 unpoison_passwd(ctx, res);
1971 return res;
1972 }
1973 INTERCEPTOR(__sanitizer_group *, getgrnam, const char *name) {
1974 void *ctx;
1975 COMMON_INTERCEPTOR_ENTER(ctx, getgrnam, name);
1976 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
1977 __sanitizer_group *res = REAL(getgrnam)(name);
1978 unpoison_group(ctx, res);
1979 return res;
1980 }
1981 INTERCEPTOR(__sanitizer_group *, getgrgid, u32 gid) {
1982 void *ctx;
1983 COMMON_INTERCEPTOR_ENTER(ctx, getgrgid, gid);
1984 __sanitizer_group *res = REAL(getgrgid)(gid);
1985 unpoison_group(ctx, res);
1986 return res;
1987 }
1988 #define INIT_GETPWNAM_AND_FRIENDS \
1989 COMMON_INTERCEPT_FUNCTION(getpwnam); \
1990 COMMON_INTERCEPT_FUNCTION(getpwuid); \
1991 COMMON_INTERCEPT_FUNCTION(getgrnam); \
1992 COMMON_INTERCEPT_FUNCTION(getgrgid);
1993 #else
1994 #define INIT_GETPWNAM_AND_FRIENDS
1995 #endif
1997 #if SANITIZER_INTERCEPT_GETPWNAM_R_AND_FRIENDS
1998 INTERCEPTOR(int, getpwnam_r, const char *name, __sanitizer_passwd *pwd,
1999 char *buf, SIZE_T buflen, __sanitizer_passwd **result) {
2000 void *ctx;
2001 COMMON_INTERCEPTOR_ENTER(ctx, getpwnam_r, name, pwd, buf, buflen, result);
2002 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
2003 // FIXME: under ASan the call below may write to freed memory and corrupt
2004 // its metadata. See
2005 // https://github.com/google/sanitizers/issues/321.
2006 int res = REAL(getpwnam_r)(name, pwd, buf, buflen, result);
2007 if (!res && result)
2008 unpoison_passwd(ctx, *result);
2009 if (result) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
2010 return res;
2011 }
2012 INTERCEPTOR(int, getpwuid_r, u32 uid, __sanitizer_passwd *pwd, char *buf,
2013 SIZE_T buflen, __sanitizer_passwd **result) {
2014 void *ctx;
2015 COMMON_INTERCEPTOR_ENTER(ctx, getpwuid_r, uid, pwd, buf, buflen, result);
2016 // FIXME: under ASan the call below may write to freed memory and corrupt
2017 // its metadata. See
2018 // https://github.com/google/sanitizers/issues/321.
2019 int res = REAL(getpwuid_r)(uid, pwd, buf, buflen, result);
2020 if (!res && result)
2021 unpoison_passwd(ctx, *result);
2022 if (result) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
2023 return res;
2024 }
2025 INTERCEPTOR(int, getgrnam_r, const char *name, __sanitizer_group *grp,
2026 char *buf, SIZE_T buflen, __sanitizer_group **result) {
2027 void *ctx;
2028 COMMON_INTERCEPTOR_ENTER(ctx, getgrnam_r, name, grp, buf, buflen, result);
2029 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
2030 // FIXME: under ASan the call below may write to freed memory and corrupt
2031 // its metadata. See
2032 // https://github.com/google/sanitizers/issues/321.
2033 int res = REAL(getgrnam_r)(name, grp, buf, buflen, result);
2034 if (!res && result)
2035 unpoison_group(ctx, *result);
2036 if (result) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
2037 return res;
2038 }
2039 INTERCEPTOR(int, getgrgid_r, u32 gid, __sanitizer_group *grp, char *buf,
2040 SIZE_T buflen, __sanitizer_group **result) {
2041 void *ctx;
2042 COMMON_INTERCEPTOR_ENTER(ctx, getgrgid_r, gid, grp, buf, buflen, result);
2043 // FIXME: under ASan the call below may write to freed memory and corrupt
2044 // its metadata. See
2045 // https://github.com/google/sanitizers/issues/321.
2046 int res = REAL(getgrgid_r)(gid, grp, buf, buflen, result);
2047 if (!res && result)
2048 unpoison_group(ctx, *result);
2049 if (result) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
2050 return res;
2051 }
2052 #define INIT_GETPWNAM_R_AND_FRIENDS \
2053 COMMON_INTERCEPT_FUNCTION(getpwnam_r); \
2054 COMMON_INTERCEPT_FUNCTION(getpwuid_r); \
2055 COMMON_INTERCEPT_FUNCTION(getgrnam_r); \
2056 COMMON_INTERCEPT_FUNCTION(getgrgid_r);
2057 #else
2058 #define INIT_GETPWNAM_R_AND_FRIENDS
2059 #endif
2061 #if SANITIZER_INTERCEPT_GETPWENT
2062 INTERCEPTOR(__sanitizer_passwd *, getpwent, int dummy) {
2063 void *ctx;
2064 COMMON_INTERCEPTOR_ENTER(ctx, getpwent, dummy);
2065 __sanitizer_passwd *res = REAL(getpwent)(dummy);
2066 unpoison_passwd(ctx, res);
2067 return res;
2068 }
2069 INTERCEPTOR(__sanitizer_group *, getgrent, int dummy) {
2070 void *ctx;
2071 COMMON_INTERCEPTOR_ENTER(ctx, getgrent, dummy);
2072 __sanitizer_group *res = REAL(getgrent)(dummy);
2073 unpoison_group(ctx, res);
2074 return res;
2075 }
2076 #define INIT_GETPWENT \
2077 COMMON_INTERCEPT_FUNCTION(getpwent); \
2078 COMMON_INTERCEPT_FUNCTION(getgrent);
2079 #else
2080 #define INIT_GETPWENT
2081 #endif
2083 #if SANITIZER_INTERCEPT_FGETPWENT
2084 INTERCEPTOR(__sanitizer_passwd *, fgetpwent, void *fp) {
2085 void *ctx;
2086 COMMON_INTERCEPTOR_ENTER(ctx, fgetpwent, fp);
2087 __sanitizer_passwd *res = REAL(fgetpwent)(fp);
2088 unpoison_passwd(ctx, res);
2089 return res;
2090 }
2091 INTERCEPTOR(__sanitizer_group *, fgetgrent, void *fp) {
2092 void *ctx;
2093 COMMON_INTERCEPTOR_ENTER(ctx, fgetgrent, fp);
2094 __sanitizer_group *res = REAL(fgetgrent)(fp);
2095 unpoison_group(ctx, res);
2096 return res;
2097 }
2098 #define INIT_FGETPWENT \
2099 COMMON_INTERCEPT_FUNCTION(fgetpwent); \
2100 COMMON_INTERCEPT_FUNCTION(fgetgrent);
2101 #else
2102 #define INIT_FGETPWENT
2103 #endif
2105 #if SANITIZER_INTERCEPT_GETPWENT_R
2106 INTERCEPTOR(int, getpwent_r, __sanitizer_passwd *pwbuf, char *buf,
2107 SIZE_T buflen, __sanitizer_passwd **pwbufp) {
2108 void *ctx;
2109 COMMON_INTERCEPTOR_ENTER(ctx, getpwent_r, pwbuf, buf, buflen, pwbufp);
2110 // FIXME: under ASan the call below may write to freed memory and corrupt
2111 // its metadata. See
2112 // https://github.com/google/sanitizers/issues/321.
2113 int res = REAL(getpwent_r)(pwbuf, buf, buflen, pwbufp);
2114 if (!res && pwbufp)
2115 unpoison_passwd(ctx, *pwbufp);
2116 if (pwbufp) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pwbufp, sizeof(*pwbufp));
2117 return res;
2118 }
2119 INTERCEPTOR(int, getgrent_r, __sanitizer_group *pwbuf, char *buf, SIZE_T buflen,
2120 __sanitizer_group **pwbufp) {
2121 void *ctx;
2122 COMMON_INTERCEPTOR_ENTER(ctx, getgrent_r, pwbuf, buf, buflen, pwbufp);
2123 // FIXME: under ASan the call below may write to freed memory and corrupt
2124 // its metadata. See
2125 // https://github.com/google/sanitizers/issues/321.
2126 int res = REAL(getgrent_r)(pwbuf, buf, buflen, pwbufp);
2127 if (!res && pwbufp)
2128 unpoison_group(ctx, *pwbufp);
2129 if (pwbufp) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pwbufp, sizeof(*pwbufp));
2130 return res;
2131 }
2132 #define INIT_GETPWENT_R \
2133 COMMON_INTERCEPT_FUNCTION(getpwent_r); \
2134 COMMON_INTERCEPT_FUNCTION(getgrent_r);
2135 #else
2136 #define INIT_GETPWENT_R
2137 #endif
2139 #if SANITIZER_INTERCEPT_FGETPWENT_R
2140 INTERCEPTOR(int, fgetpwent_r, void *fp, __sanitizer_passwd *pwbuf, char *buf,
2141 SIZE_T buflen, __sanitizer_passwd **pwbufp) {
2142 void *ctx;
2143 COMMON_INTERCEPTOR_ENTER(ctx, fgetpwent_r, fp, pwbuf, buf, buflen, pwbufp);
2144 // FIXME: under ASan the call below may write to freed memory and corrupt
2145 // its metadata. See
2146 // https://github.com/google/sanitizers/issues/321.
2147 int res = REAL(fgetpwent_r)(fp, pwbuf, buf, buflen, pwbufp);
2148 if (!res && pwbufp)
2149 unpoison_passwd(ctx, *pwbufp);
2150 if (pwbufp) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pwbufp, sizeof(*pwbufp));
2151 return res;
2152 }
2153 #define INIT_FGETPWENT_R \
2154 COMMON_INTERCEPT_FUNCTION(fgetpwent_r);
2155 #else
2156 #define INIT_FGETPWENT_R
2157 #endif
2159 #if SANITIZER_INTERCEPT_FGETGRENT_R
2160 INTERCEPTOR(int, fgetgrent_r, void *fp, __sanitizer_group *pwbuf, char *buf,
2161 SIZE_T buflen, __sanitizer_group **pwbufp) {
2162 void *ctx;
2163 COMMON_INTERCEPTOR_ENTER(ctx, fgetgrent_r, fp, pwbuf, buf, buflen, pwbufp);
2164 // FIXME: under ASan the call below may write to freed memory and corrupt
2165 // its metadata. See
2166 // https://github.com/google/sanitizers/issues/321.
2167 int res = REAL(fgetgrent_r)(fp, pwbuf, buf, buflen, pwbufp);
2168 if (!res && pwbufp)
2169 unpoison_group(ctx, *pwbufp);
2170 if (pwbufp) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pwbufp, sizeof(*pwbufp));
2171 return res;
2172 }
2173 #define INIT_FGETGRENT_R \
2174 COMMON_INTERCEPT_FUNCTION(fgetgrent_r);
2175 #else
2176 #define INIT_FGETGRENT_R
2177 #endif
2179 #if SANITIZER_INTERCEPT_SETPWENT
2180 // The only thing these interceptors do is disable any nested interceptors.
2181 // These functions may open nss modules and call uninstrumented functions from
2182 // them, and we don't want things like strlen() to trigger.
2183 INTERCEPTOR(void, setpwent, int dummy) {
2184 void *ctx;
2185 COMMON_INTERCEPTOR_ENTER(ctx, setpwent, dummy);
2186 REAL(setpwent)(dummy);
2187 }
2188 INTERCEPTOR(void, endpwent, int dummy) {
2189 void *ctx;
2190 COMMON_INTERCEPTOR_ENTER(ctx, endpwent, dummy);
2191 REAL(endpwent)(dummy);
2192 }
2193 INTERCEPTOR(void, setgrent, int dummy) {
2194 void *ctx;
2195 COMMON_INTERCEPTOR_ENTER(ctx, setgrent, dummy);
2196 REAL(setgrent)(dummy);
2197 }
2198 INTERCEPTOR(void, endgrent, int dummy) {
2199 void *ctx;
2200 COMMON_INTERCEPTOR_ENTER(ctx, endgrent, dummy);
2201 REAL(endgrent)(dummy);
2202 }
2203 #define INIT_SETPWENT \
2204 COMMON_INTERCEPT_FUNCTION(setpwent); \
2205 COMMON_INTERCEPT_FUNCTION(endpwent); \
2206 COMMON_INTERCEPT_FUNCTION(setgrent); \
2207 COMMON_INTERCEPT_FUNCTION(endgrent);
2208 #else
2209 #define INIT_SETPWENT
2210 #endif
2212 #if SANITIZER_INTERCEPT_CLOCK_GETTIME
2213 INTERCEPTOR(int, clock_getres, u32 clk_id, void *tp) {
2214 void *ctx;
2215 COMMON_INTERCEPTOR_ENTER(ctx, clock_getres, clk_id, tp);
2216 // FIXME: under ASan the call below may write to freed memory and corrupt
2217 // its metadata. See
2218 // https://github.com/google/sanitizers/issues/321.
2219 int res = REAL(clock_getres)(clk_id, tp);
2220 if (!res && tp) {
2221 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, tp, struct_timespec_sz);
2222 }
2223 return res;
2224 }
2225 INTERCEPTOR(int, clock_gettime, u32 clk_id, void *tp) {
2226 void *ctx;
2227 COMMON_INTERCEPTOR_ENTER(ctx, clock_gettime, clk_id, tp);
2228 // FIXME: under ASan the call below may write to freed memory and corrupt
2229 // its metadata. See
2230 // https://github.com/google/sanitizers/issues/321.
2231 int res = REAL(clock_gettime)(clk_id, tp);
2232 if (!res) {
2233 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, tp, struct_timespec_sz);
2234 }
2235 return res;
2236 }
2237 #if SANITIZER_GLIBC
2238 namespace __sanitizer {
2239 extern "C" {
2240 int real_clock_gettime(u32 clk_id, void *tp) {
2241 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
2242 return internal_clock_gettime(clk_id, tp);
2243 return REAL(clock_gettime)(clk_id, tp);
2244 }
2245 } // extern "C"
2246 } // namespace __sanitizer
2247 #endif
2248 INTERCEPTOR(int, clock_settime, u32 clk_id, const void *tp) {
2249 void *ctx;
2250 COMMON_INTERCEPTOR_ENTER(ctx, clock_settime, clk_id, tp);
2251 COMMON_INTERCEPTOR_READ_RANGE(ctx, tp, struct_timespec_sz);
2252 return REAL(clock_settime)(clk_id, tp);
2253 }
2254 #define INIT_CLOCK_GETTIME \
2255 COMMON_INTERCEPT_FUNCTION(clock_getres); \
2256 COMMON_INTERCEPT_FUNCTION(clock_gettime); \
2257 COMMON_INTERCEPT_FUNCTION(clock_settime);
2258 #else
2259 #define INIT_CLOCK_GETTIME
2260 #endif
2262 #if SANITIZER_INTERCEPT_CLOCK_GETCPUCLOCKID
2263 INTERCEPTOR(int, clock_getcpuclockid, pid_t pid,
2264 __sanitizer_clockid_t *clockid) {
2265 void *ctx;
2266 COMMON_INTERCEPTOR_ENTER(ctx, clock_getcpuclockid, pid, clockid);
2267 int res = REAL(clock_getcpuclockid)(pid, clockid);
2268 if (!res && clockid) {
2269 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, clockid, sizeof *clockid);
2270 }
2271 return res;
2272 }
2274 INTERCEPTOR(int, pthread_getcpuclockid, uptr thread,
2275 __sanitizer_clockid_t *clockid) {
2276 void *ctx;
2277 COMMON_INTERCEPTOR_ENTER(ctx, pthread_getcpuclockid, thread, clockid);
2278 int res = REAL(pthread_getcpuclockid)(thread, clockid);
2279 if (!res && clockid) {
2280 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, clockid, sizeof *clockid);
2281 }
2282 return res;
2283 }
2285 #define INIT_CLOCK_GETCPUCLOCKID \
2286 COMMON_INTERCEPT_FUNCTION(clock_getcpuclockid); \
2287 COMMON_INTERCEPT_FUNCTION(pthread_getcpuclockid);
2288 #else
2289 #define INIT_CLOCK_GETCPUCLOCKID
2290 #endif
2292 #if SANITIZER_INTERCEPT_TIMER_CREATE
2293 INTERCEPTOR(int, timer_create, __sanitizer_clockid_t clockid, void *sevp,
2294 __sanitizer_timer_t *timer) {
2295 void *ctx;
2296 COMMON_INTERCEPTOR_ENTER(ctx, timer_create, clockid, sevp, timer);
2297 int res = REAL(timer_create)(clockid, sevp, timer);
2298 if (!res && timer) {
2299 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, timer, sizeof *timer);
2300 }
2301 return res;
2302 }
2304 INTERCEPTOR(int, timer_delete, __sanitizer_timer_t timer) {
2305 void *ctx;
2306 COMMON_INTERCEPTOR_ENTER(ctx, timer_delete, timer);
2307 int res = REAL(timer_delete)(timer);
2308 return res;
2309 }
2311 INTERCEPTOR(int, timer_gettime, __sanitizer_timer_t timer,
2312 struct __sanitizer_itimerspec *curr_value) {
2313 void *ctx;
2314 COMMON_INTERCEPTOR_ENTER(ctx, timer_gettime, timer, curr_value);
2315 int res = REAL(timer_gettime)(timer, curr_value);
2316 if (!res && curr_value) {
2317 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, curr_value, sizeof *curr_value);
2318 }
2319 return res;
2320 }
2322 INTERCEPTOR(int, timer_settime, __sanitizer_timer_t timer, int flags,
2323 const struct __sanitizer_itimerspec *new_value,
2324 struct __sanitizer_itimerspec *old_value) {
2325 void *ctx;
2326 COMMON_INTERCEPTOR_ENTER(ctx, timer_settime, timer, flags, new_value,
2327 old_value);
2328 int res = REAL(timer_settime)(timer, flags, new_value, old_value);
2329 if (!res) {
2330 if (new_value)
2331 COMMON_INTERCEPTOR_READ_RANGE(ctx, new_value, sizeof *new_value);
2332 if (old_value)
2333 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, old_value, sizeof *old_value);
2334 }
2335 return res;
2336 }
2338 # define INIT_TIMER_CREATE \
2339 COMMON_INTERCEPT_FUNCTION_GLIBC_VER_MIN(timer_create, "GLIBC_2.3.3"); \
2340 COMMON_INTERCEPT_FUNCTION_GLIBC_VER_MIN(timer_delete, "GLIBC_2.3.3"); \
2341 COMMON_INTERCEPT_FUNCTION_GLIBC_VER_MIN(timer_gettime, "GLIBC_2.3.3"); \
2342 COMMON_INTERCEPT_FUNCTION_GLIBC_VER_MIN(timer_settime, "GLIBC_2.3.3");
2343 #else
2344 # define INIT_TIMER_CREATE
2345 #endif
2347 #if SANITIZER_INTERCEPT_GETITIMER
2348 INTERCEPTOR(int, getitimer, int which, void *curr_value) {
2349 void *ctx;
2350 COMMON_INTERCEPTOR_ENTER(ctx, getitimer, which, curr_value);
2351 // FIXME: under ASan the call below may write to freed memory and corrupt
2352 // its metadata. See
2353 // https://github.com/google/sanitizers/issues/321.
2354 int res = REAL(getitimer)(which, curr_value);
2355 if (!res && curr_value) {
2356 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, curr_value, struct_itimerval_sz);
2357 }
2358 return res;
2359 }
2360 INTERCEPTOR(int, setitimer, int which, const void *new_value, void *old_value) {
2361 void *ctx;
2362 COMMON_INTERCEPTOR_ENTER(ctx, setitimer, which, new_value, old_value);
2363 if (new_value) {
2364 // itimerval can contain padding that may be legitimately uninitialized
2365 const struct __sanitizer_itimerval *nv =
2366 (const struct __sanitizer_itimerval *)new_value;
2367 COMMON_INTERCEPTOR_READ_RANGE(ctx, &nv->it_interval.tv_sec,
2368 sizeof(__sanitizer_time_t));
2369 COMMON_INTERCEPTOR_READ_RANGE(ctx, &nv->it_interval.tv_usec,
2370 sizeof(__sanitizer_suseconds_t));
2371 COMMON_INTERCEPTOR_READ_RANGE(ctx, &nv->it_value.tv_sec,
2372 sizeof(__sanitizer_time_t));
2373 COMMON_INTERCEPTOR_READ_RANGE(ctx, &nv->it_value.tv_usec,
2374 sizeof(__sanitizer_suseconds_t));
2375 }
2376 // FIXME: under ASan the call below may write to freed memory and corrupt
2377 // its metadata. See
2378 // https://github.com/google/sanitizers/issues/321.
2379 int res = REAL(setitimer)(which, new_value, old_value);
2380 if (!res && old_value) {
2381 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, old_value, struct_itimerval_sz);
2382 }
2383 return res;
2384 }
2385 #define INIT_GETITIMER \
2386 COMMON_INTERCEPT_FUNCTION(getitimer); \
2387 COMMON_INTERCEPT_FUNCTION(setitimer);
2388 #else
2389 #define INIT_GETITIMER
2390 #endif
2392 #if SANITIZER_INTERCEPT_TIMESPEC_GET
2393 INTERCEPTOR(int, timespec_get, struct __sanitizer_timespec *ts, int base) {
2394 void *ctx;
2395 COMMON_INTERCEPTOR_ENTER(ctx, timespec_get, ts, base);
2396 // We don't yet know if ts is addressable, so we use our own scratch buffer
2397 struct __sanitizer_timespec ts_local;
2398 int res = REAL(timespec_get)(&ts_local, base);
2399 if (res) {
2400 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ts,
2401 sizeof(struct __sanitizer_timespec));
2402 internal_memcpy(ts, &ts_local, sizeof(struct __sanitizer_timespec));
2403 }
2404 return res;
2405 }
2406 # define INIT_TIMESPEC_GET COMMON_INTERCEPT_FUNCTION(timespec_get);
2407 #else
2408 # define INIT_TIMESPEC_GET
2409 #endif
2411 #if SANITIZER_INTERCEPT_GLOB
2412 static void unpoison_glob_t(void *ctx, __sanitizer_glob_t *pglob) {
2413 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pglob, sizeof(*pglob));
2414 // +1 for NULL pointer at the end.
2415 if (pglob->gl_pathv)
2416 COMMON_INTERCEPTOR_WRITE_RANGE(
2417 ctx, pglob->gl_pathv, (pglob->gl_pathc + 1) * sizeof(*pglob->gl_pathv));
2418 for (SIZE_T i = 0; i < pglob->gl_pathc; ++i) {
2419 char *p = pglob->gl_pathv[i];
2420 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p, internal_strlen(p) + 1);
2421 }
2422 }
2424 #if SANITIZER_SOLARIS
2425 INTERCEPTOR(int, glob, const char *pattern, int flags,
2426 int (*errfunc)(const char *epath, int eerrno),
2427 __sanitizer_glob_t *pglob) {
2428 void *ctx;
2429 COMMON_INTERCEPTOR_ENTER(ctx, glob, pattern, flags, errfunc, pglob);
2430 COMMON_INTERCEPTOR_READ_STRING(ctx, pattern, 0);
2431 int res = REAL(glob)(pattern, flags, errfunc, pglob);
2432 if ((!res || res == glob_nomatch) && pglob) unpoison_glob_t(ctx, pglob);
2433 return res;
2434 }
2435 #else
2436 static THREADLOCAL __sanitizer_glob_t *pglob_copy;
2438 static void wrapped_gl_closedir(void *dir) {
2439 COMMON_INTERCEPTOR_UNPOISON_PARAM(1);
2440 pglob_copy->gl_closedir(dir);
2441 }
2443 static void *wrapped_gl_readdir(void *dir) {
2444 COMMON_INTERCEPTOR_UNPOISON_PARAM(1);
2445 return pglob_copy->gl_readdir(dir);
2446 }
2448 static void *wrapped_gl_opendir(const char *s) {
2449 COMMON_INTERCEPTOR_UNPOISON_PARAM(1);
2450 COMMON_INTERCEPTOR_INITIALIZE_RANGE(s, internal_strlen(s) + 1);
2451 return pglob_copy->gl_opendir(s);
2452 }
2454 static int wrapped_gl_lstat(const char *s, void *st) {
2455 COMMON_INTERCEPTOR_UNPOISON_PARAM(2);
2456 COMMON_INTERCEPTOR_INITIALIZE_RANGE(s, internal_strlen(s) + 1);
2457 return pglob_copy->gl_lstat(s, st);
2458 }
2460 static int wrapped_gl_stat(const char *s, void *st) {
2461 COMMON_INTERCEPTOR_UNPOISON_PARAM(2);
2462 COMMON_INTERCEPTOR_INITIALIZE_RANGE(s, internal_strlen(s) + 1);
2463 return pglob_copy->gl_stat(s, st);
2464 }
2466 static const __sanitizer_glob_t kGlobCopy = {
2467 0, 0, 0,
2468 0, wrapped_gl_closedir, wrapped_gl_readdir,
2469 wrapped_gl_opendir, wrapped_gl_lstat, wrapped_gl_stat};
2471 INTERCEPTOR(int, glob, const char *pattern, int flags,
2472 int (*errfunc)(const char *epath, int eerrno),
2473 __sanitizer_glob_t *pglob) {
2474 void *ctx;
2475 COMMON_INTERCEPTOR_ENTER(ctx, glob, pattern, flags, errfunc, pglob);
2476 COMMON_INTERCEPTOR_READ_STRING(ctx, pattern, 0);
2477 __sanitizer_glob_t glob_copy;
2478 internal_memcpy(&glob_copy, &kGlobCopy, sizeof(glob_copy));
2479 if (flags & glob_altdirfunc) {
2480 Swap(pglob->gl_closedir, glob_copy.gl_closedir);
2481 Swap(pglob->gl_readdir, glob_copy.gl_readdir);
2482 Swap(pglob->gl_opendir, glob_copy.gl_opendir);
2483 Swap(pglob->gl_lstat, glob_copy.gl_lstat);
2484 Swap(pglob->gl_stat, glob_copy.gl_stat);
2485 pglob_copy = &glob_copy;
2486 }
2487 int res = REAL(glob)(pattern, flags, errfunc, pglob);
2488 if (flags & glob_altdirfunc) {
2489 Swap(pglob->gl_closedir, glob_copy.gl_closedir);
2490 Swap(pglob->gl_readdir, glob_copy.gl_readdir);
2491 Swap(pglob->gl_opendir, glob_copy.gl_opendir);
2492 Swap(pglob->gl_lstat, glob_copy.gl_lstat);
2493 Swap(pglob->gl_stat, glob_copy.gl_stat);
2494 }
2495 pglob_copy = 0;
2496 if ((!res || res == glob_nomatch) && pglob) unpoison_glob_t(ctx, pglob);
2497 return res;
2498 }
2499 #endif // SANITIZER_SOLARIS
2500 #define INIT_GLOB \
2501 COMMON_INTERCEPT_FUNCTION(glob);
2502 #else // SANITIZER_INTERCEPT_GLOB
2503 #define INIT_GLOB
2504 #endif // SANITIZER_INTERCEPT_GLOB
2506 #if SANITIZER_INTERCEPT_GLOB64
2507 INTERCEPTOR(int, glob64, const char *pattern, int flags,
2508 int (*errfunc)(const char *epath, int eerrno),
2509 __sanitizer_glob_t *pglob) {
2510 void *ctx;
2511 COMMON_INTERCEPTOR_ENTER(ctx, glob64, pattern, flags, errfunc, pglob);
2512 COMMON_INTERCEPTOR_READ_STRING(ctx, pattern, 0);
2513 __sanitizer_glob_t glob_copy;
2514 internal_memcpy(&glob_copy, &kGlobCopy, sizeof(glob_copy));
2515 if (flags & glob_altdirfunc) {
2516 Swap(pglob->gl_closedir, glob_copy.gl_closedir);
2517 Swap(pglob->gl_readdir, glob_copy.gl_readdir);
2518 Swap(pglob->gl_opendir, glob_copy.gl_opendir);
2519 Swap(pglob->gl_lstat, glob_copy.gl_lstat);
2520 Swap(pglob->gl_stat, glob_copy.gl_stat);
2521 pglob_copy = &glob_copy;
2522 }
2523 int res = REAL(glob64)(pattern, flags, errfunc, pglob);
2524 if (flags & glob_altdirfunc) {
2525 Swap(pglob->gl_closedir, glob_copy.gl_closedir);
2526 Swap(pglob->gl_readdir, glob_copy.gl_readdir);
2527 Swap(pglob->gl_opendir, glob_copy.gl_opendir);
2528 Swap(pglob->gl_lstat, glob_copy.gl_lstat);
2529 Swap(pglob->gl_stat, glob_copy.gl_stat);
2530 }
2531 pglob_copy = 0;
2532 if ((!res || res == glob_nomatch) && pglob) unpoison_glob_t(ctx, pglob);
2533 return res;
2534 }
2535 #define INIT_GLOB64 \
2536 COMMON_INTERCEPT_FUNCTION(glob64);
2537 #else // SANITIZER_INTERCEPT_GLOB64
2538 #define INIT_GLOB64
2539 #endif // SANITIZER_INTERCEPT_GLOB64
2541 #if SANITIZER_INTERCEPT___B64_TO
2542 INTERCEPTOR(int, __b64_ntop, unsigned char const *src, SIZE_T srclength,
2543 char *target, SIZE_T targsize) {
2544 void *ctx;
2545 COMMON_INTERCEPTOR_ENTER(ctx, __b64_ntop, src, srclength, target, targsize);
2546 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, srclength);
2547 int res = REAL(__b64_ntop)(src, srclength, target, targsize);
2548 if (res >= 0)
2549 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, target, res + 1);
2550 return res;
2551 }
2552 INTERCEPTOR(int, __b64_pton, char const *src, char *target, SIZE_T targsize) {
2553 void *ctx;
2554 COMMON_INTERCEPTOR_ENTER(ctx, __b64_pton, src, target, targsize);
2555 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, internal_strlen(src) + 1);
2556 int res = REAL(__b64_pton)(src, target, targsize);
2557 if (res >= 0)
2558 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, target, res);
2559 return res;
2560 }
2561 #define INIT___B64_TO \
2562 COMMON_INTERCEPT_FUNCTION(__b64_ntop); \
2563 COMMON_INTERCEPT_FUNCTION(__b64_pton);
2564 #else // SANITIZER_INTERCEPT___B64_TO
2565 #define INIT___B64_TO
2566 #endif // SANITIZER_INTERCEPT___B64_TO
2568 #if SANITIZER_INTERCEPT_DN_COMP_EXPAND
2569 # if __GLIBC_PREREQ(2, 34)
2570 // Changed with https://sourceware.org/git/?p=glibc.git;h=640bbdf
2571 # define DN_COMP_INTERCEPTOR_NAME dn_comp
2572 # define DN_EXPAND_INTERCEPTOR_NAME dn_expand
2573 # else
2574 # define DN_COMP_INTERCEPTOR_NAME __dn_comp
2575 # define DN_EXPAND_INTERCEPTOR_NAME __dn_expand
2576 # endif
2577 INTERCEPTOR(int, DN_COMP_INTERCEPTOR_NAME, unsigned char *exp_dn,
2578 unsigned char *comp_dn, int length, unsigned char **dnptrs,
2579 unsigned char **lastdnptr) {
2580 void *ctx;
2581 COMMON_INTERCEPTOR_ENTER(ctx, DN_COMP_INTERCEPTOR_NAME, exp_dn, comp_dn,
2582 length, dnptrs, lastdnptr);
2583 int res = REAL(DN_COMP_INTERCEPTOR_NAME)(exp_dn, comp_dn, length, dnptrs,
2584 lastdnptr);
2585 if (res >= 0) {
2586 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, comp_dn, res);
2587 if (dnptrs && lastdnptr) {
2588 unsigned char **p = dnptrs;
2589 for (; p != lastdnptr && *p; ++p)
2590 ;
2591 if (p != lastdnptr)
2592 ++p;
2593 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dnptrs, (p - dnptrs) * sizeof(*p));
2594 }
2595 }
2596 return res;
2597 }
2598 INTERCEPTOR(int, DN_EXPAND_INTERCEPTOR_NAME, unsigned char const *base,
2599 unsigned char const *end, unsigned char const *src, char *dest,
2600 int space) {
2601 void *ctx;
2602 COMMON_INTERCEPTOR_ENTER(ctx, DN_EXPAND_INTERCEPTOR_NAME, base, end, src,
2603 dest, space);
2604 // TODO: add read check if __dn_comp intercept added
2605 int res = REAL(DN_EXPAND_INTERCEPTOR_NAME)(base, end, src, dest, space);
2606 if (res >= 0)
2607 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dest, internal_strlen(dest) + 1);
2608 return res;
2609 }
2610 # define INIT_DN_COMP_EXPAND \
2611 COMMON_INTERCEPT_FUNCTION(DN_COMP_INTERCEPTOR_NAME); \
2612 COMMON_INTERCEPT_FUNCTION(DN_EXPAND_INTERCEPTOR_NAME);
2613 #else // SANITIZER_INTERCEPT_DN_COMP_EXPAND
2614 # define INIT_DN_COMP_EXPAND
2615 #endif // SANITIZER_INTERCEPT_DN_COMP_EXPAND
2617 #if SANITIZER_INTERCEPT_POSIX_SPAWN
2619 template <class RealSpawnPtr>
2620 static int PosixSpawnImpl(void *ctx, RealSpawnPtr *real_posix_spawn, pid_t *pid,
2621 const char *file_or_path, const void *file_actions,
2622 const void *attrp, char *const argv[],
2623 char *const envp[]) {
2624 COMMON_INTERCEPTOR_READ_RANGE(ctx, file_or_path,
2625 internal_strlen(file_or_path) + 1);
2626 if (argv) {
2627 for (char *const *s = argv; ; ++s) {
2628 COMMON_INTERCEPTOR_READ_RANGE(ctx, s, sizeof(*s));
2629 if (!*s) break;
2630 COMMON_INTERCEPTOR_READ_RANGE(ctx, *s, internal_strlen(*s) + 1);
2631 }
2632 }
2633 if (envp) {
2634 for (char *const *s = envp; ; ++s) {
2635 COMMON_INTERCEPTOR_READ_RANGE(ctx, s, sizeof(*s));
2636 if (!*s) break;
2637 COMMON_INTERCEPTOR_READ_RANGE(ctx, *s, internal_strlen(*s) + 1);
2638 }
2639 }
2640 int res =
2641 real_posix_spawn(pid, file_or_path, file_actions, attrp, argv, envp);
2642 if (res == 0)
2643 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pid, sizeof(*pid));
2644 return res;
2645 }
2646 INTERCEPTOR(int, posix_spawn, pid_t *pid, const char *path,
2647 const void *file_actions, const void *attrp, char *const argv[],
2648 char *const envp[]) {
2649 void *ctx;
2650 COMMON_INTERCEPTOR_ENTER(ctx, posix_spawn, pid, path, file_actions, attrp,
2651 argv, envp);
2652 return PosixSpawnImpl(ctx, REAL(posix_spawn), pid, path, file_actions, attrp,
2653 argv, envp);
2654 }
2655 INTERCEPTOR(int, posix_spawnp, pid_t *pid, const char *file,
2656 const void *file_actions, const void *attrp, char *const argv[],
2657 char *const envp[]) {
2658 void *ctx;
2659 COMMON_INTERCEPTOR_ENTER(ctx, posix_spawnp, pid, file, file_actions, attrp,
2660 argv, envp);
2661 return PosixSpawnImpl(ctx, REAL(posix_spawnp), pid, file, file_actions, attrp,
2662 argv, envp);
2663 }
2664 # define INIT_POSIX_SPAWN \
2665 COMMON_INTERCEPT_FUNCTION(posix_spawn); \
2666 COMMON_INTERCEPT_FUNCTION(posix_spawnp);
2667 #else // SANITIZER_INTERCEPT_POSIX_SPAWN
2668 # define INIT_POSIX_SPAWN
2669 #endif // SANITIZER_INTERCEPT_POSIX_SPAWN
2671 #if SANITIZER_INTERCEPT_WAIT
2672 // According to sys/wait.h, wait(), waitid(), waitpid() may have symbol version
2673 // suffixes on Darwin. See the declaration of INTERCEPTOR_WITH_SUFFIX for
2674 // details.
2675 INTERCEPTOR_WITH_SUFFIX(int, wait, int *status) {
2676 void *ctx;
2677 COMMON_INTERCEPTOR_ENTER(ctx, wait, status);
2678 // FIXME: under ASan the call below may write to freed memory and corrupt
2679 // its metadata. See
2680 // https://github.com/google/sanitizers/issues/321.
2681 int res = COMMON_INTERCEPTOR_BLOCK_REAL(wait)(status);
2682 if (res != -1 && status)
2683 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, status, sizeof(*status));
2684 return res;
2685 }
2686 // On FreeBSD id_t is always 64-bit wide.
2687 #if SANITIZER_FREEBSD && (SANITIZER_WORDSIZE == 32)
2688 INTERCEPTOR_WITH_SUFFIX(int, waitid, int idtype, long long id, void *infop,
2689 int options) {
2690 #else
2691 INTERCEPTOR_WITH_SUFFIX(int, waitid, int idtype, int id, void *infop,
2692 int options) {
2693 #endif
2694 void *ctx;
2695 COMMON_INTERCEPTOR_ENTER(ctx, waitid, idtype, id, infop, options);
2696 // FIXME: under ASan the call below may write to freed memory and corrupt
2697 // its metadata. See
2698 // https://github.com/google/sanitizers/issues/321.
2699 int res = COMMON_INTERCEPTOR_BLOCK_REAL(waitid)(idtype, id, infop, options);
2700 if (res != -1 && infop)
2701 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, infop, siginfo_t_sz);
2702 return res;
2703 }
2704 INTERCEPTOR_WITH_SUFFIX(int, waitpid, int pid, int *status, int options) {
2705 void *ctx;
2706 COMMON_INTERCEPTOR_ENTER(ctx, waitpid, pid, status, options);
2707 // FIXME: under ASan the call below may write to freed memory and corrupt
2708 // its metadata. See
2709 // https://github.com/google/sanitizers/issues/321.
2710 int res = COMMON_INTERCEPTOR_BLOCK_REAL(waitpid)(pid, status, options);
2711 if (res != -1 && status)
2712 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, status, sizeof(*status));
2713 return res;
2714 }
2715 INTERCEPTOR(int, wait3, int *status, int options, void *rusage) {
2716 void *ctx;
2717 COMMON_INTERCEPTOR_ENTER(ctx, wait3, status, options, rusage);
2718 // FIXME: under ASan the call below may write to freed memory and corrupt
2719 // its metadata. See
2720 // https://github.com/google/sanitizers/issues/321.
2721 int res = COMMON_INTERCEPTOR_BLOCK_REAL(wait3)(status, options, rusage);
2722 if (res != -1) {
2723 if (status) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, status, sizeof(*status));
2724 if (rusage) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, rusage, struct_rusage_sz);
2725 }
2726 return res;
2727 }
2728 #if SANITIZER_ANDROID
2729 INTERCEPTOR(int, __wait4, int pid, int *status, int options, void *rusage) {
2730 void *ctx;
2731 COMMON_INTERCEPTOR_ENTER(ctx, __wait4, pid, status, options, rusage);
2732 // FIXME: under ASan the call below may write to freed memory and corrupt
2733 // its metadata. See
2734 // https://github.com/google/sanitizers/issues/321.
2735 int res =
2736 COMMON_INTERCEPTOR_BLOCK_REAL(__wait4)(pid, status, options, rusage);
2737 if (res != -1) {
2738 if (status) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, status, sizeof(*status));
2739 if (rusage) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, rusage, struct_rusage_sz);
2740 }
2741 return res;
2742 }
2743 #define INIT_WAIT4 COMMON_INTERCEPT_FUNCTION(__wait4);
2744 #else
2745 INTERCEPTOR(int, wait4, int pid, int *status, int options, void *rusage) {
2746 void *ctx;
2747 COMMON_INTERCEPTOR_ENTER(ctx, wait4, pid, status, options, rusage);
2748 // FIXME: under ASan the call below may write to freed memory and corrupt
2749 // its metadata. See
2750 // https://github.com/google/sanitizers/issues/321.
2751 int res = COMMON_INTERCEPTOR_BLOCK_REAL(wait4)(pid, status, options, rusage);
2752 if (res != -1) {
2753 if (status) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, status, sizeof(*status));
2754 if (rusage) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, rusage, struct_rusage_sz);
2755 }
2756 return res;
2757 }
2758 #define INIT_WAIT4 COMMON_INTERCEPT_FUNCTION(wait4);
2759 #endif // SANITIZER_ANDROID
2760 #define INIT_WAIT \
2761 COMMON_INTERCEPT_FUNCTION(wait); \
2762 COMMON_INTERCEPT_FUNCTION(waitid); \
2763 COMMON_INTERCEPT_FUNCTION(waitpid); \
2764 COMMON_INTERCEPT_FUNCTION(wait3);
2765 #else
2766 #define INIT_WAIT
2767 #define INIT_WAIT4
2768 #endif
2770 #if SANITIZER_INTERCEPT_INET
2771 INTERCEPTOR(char *, inet_ntop, int af, const void *src, char *dst, u32 size) {
2772 void *ctx;
2773 COMMON_INTERCEPTOR_ENTER(ctx, inet_ntop, af, src, dst, size);
2774 uptr sz = __sanitizer_in_addr_sz(af);
2775 if (sz) COMMON_INTERCEPTOR_READ_RANGE(ctx, src, sz);
2776 // FIXME: figure out read size based on the address family.
2777 // FIXME: under ASan the call below may write to freed memory and corrupt
2778 // its metadata. See
2779 // https://github.com/google/sanitizers/issues/321.
2780 char *res = REAL(inet_ntop)(af, src, dst, size);
2781 if (res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
2782 return res;
2783 }
2784 INTERCEPTOR(int, inet_pton, int af, const char *src, void *dst) {
2785 void *ctx;
2786 COMMON_INTERCEPTOR_ENTER(ctx, inet_pton, af, src, dst);
2787 COMMON_INTERCEPTOR_READ_STRING(ctx, src, 0);
2788 // FIXME: figure out read size based on the address family.
2789 // FIXME: under ASan the call below may write to freed memory and corrupt
2790 // its metadata. See
2791 // https://github.com/google/sanitizers/issues/321.
2792 int res = REAL(inet_pton)(af, src, dst);
2793 if (res == 1) {
2794 uptr sz = __sanitizer_in_addr_sz(af);
2795 if (sz) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, sz);
2796 }
2797 return res;
2798 }
2799 #define INIT_INET \
2800 COMMON_INTERCEPT_FUNCTION(inet_ntop); \
2801 COMMON_INTERCEPT_FUNCTION(inet_pton);
2802 #else
2803 #define INIT_INET
2804 #endif
2806 #if SANITIZER_INTERCEPT_INET
2807 INTERCEPTOR(int, inet_aton, const char *cp, void *dst) {
2808 void *ctx;
2809 COMMON_INTERCEPTOR_ENTER(ctx, inet_aton, cp, dst);
2810 if (cp) COMMON_INTERCEPTOR_READ_RANGE(ctx, cp, internal_strlen(cp) + 1);
2811 // FIXME: under ASan the call below may write to freed memory and corrupt
2812 // its metadata. See
2813 // https://github.com/google/sanitizers/issues/321.
2814 int res = REAL(inet_aton)(cp, dst);
2815 if (res != 0) {
2816 uptr sz = __sanitizer_in_addr_sz(af_inet);
2817 if (sz) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, sz);
2818 }
2819 return res;
2820 }
2821 #define INIT_INET_ATON COMMON_INTERCEPT_FUNCTION(inet_aton);
2822 #else
2823 #define INIT_INET_ATON
2824 #endif
2826 #if SANITIZER_INTERCEPT_PTHREAD_GETSCHEDPARAM
2827 INTERCEPTOR(int, pthread_getschedparam, uptr thread, int *policy, int *param) {
2828 void *ctx;
2829 COMMON_INTERCEPTOR_ENTER(ctx, pthread_getschedparam, thread, policy, param);
2830 // FIXME: under ASan the call below may write to freed memory and corrupt
2831 // its metadata. See
2832 // https://github.com/google/sanitizers/issues/321.
2833 int res = REAL(pthread_getschedparam)(thread, policy, param);
2834 if (res == 0) {
2835 if (policy) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, policy, sizeof(*policy));
2836 if (param) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, param, sizeof(*param));
2837 }
2838 return res;
2839 }
2840 #define INIT_PTHREAD_GETSCHEDPARAM \
2841 COMMON_INTERCEPT_FUNCTION(pthread_getschedparam);
2842 #else
2843 #define INIT_PTHREAD_GETSCHEDPARAM
2844 #endif
2846 #if SANITIZER_INTERCEPT_GETADDRINFO
2847 INTERCEPTOR(int, getaddrinfo, char *node, char *service,
2848 struct __sanitizer_addrinfo *hints,
2849 struct __sanitizer_addrinfo **out) {
2850 void *ctx;
2851 COMMON_INTERCEPTOR_ENTER(ctx, getaddrinfo, node, service, hints, out);
2852 if (node) COMMON_INTERCEPTOR_READ_RANGE(ctx, node, internal_strlen(node) + 1);
2853 if (service)
2854 COMMON_INTERCEPTOR_READ_RANGE(ctx, service, internal_strlen(service) + 1);
2855 if (hints)
2856 COMMON_INTERCEPTOR_READ_RANGE(ctx, hints, sizeof(__sanitizer_addrinfo));
2857 // FIXME: under ASan the call below may write to freed memory and corrupt
2858 // its metadata. See
2859 // https://github.com/google/sanitizers/issues/321.
2860 int res = REAL(getaddrinfo)(node, service, hints, out);
2861 if (res == 0 && out) {
2862 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, out, sizeof(*out));
2863 struct __sanitizer_addrinfo *p = *out;
2864 while (p) {
2865 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p, sizeof(*p));
2866 if (p->ai_addr)
2867 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p->ai_addr, p->ai_addrlen);
2868 if (p->ai_canonname)
2869 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p->ai_canonname,
2870 internal_strlen(p->ai_canonname) + 1);
2871 p = p->ai_next;
2872 }
2873 }
2874 return res;
2875 }
2876 #define INIT_GETADDRINFO COMMON_INTERCEPT_FUNCTION(getaddrinfo);
2877 #else
2878 #define INIT_GETADDRINFO
2879 #endif
2881 #if SANITIZER_INTERCEPT_GETNAMEINFO
2882 INTERCEPTOR(int, getnameinfo, void *sockaddr, unsigned salen, char *host,
2883 unsigned hostlen, char *serv, unsigned servlen, int flags) {
2884 void *ctx;
2885 COMMON_INTERCEPTOR_ENTER(ctx, getnameinfo, sockaddr, salen, host, hostlen,
2886 serv, servlen, flags);
2887 // FIXME: consider adding READ_RANGE(sockaddr, salen)
2888 // There is padding in in_addr that may make this too noisy
2889 // FIXME: under ASan the call below may write to freed memory and corrupt
2890 // its metadata. See
2891 // https://github.com/google/sanitizers/issues/321.
2892 int res =
2893 REAL(getnameinfo)(sockaddr, salen, host, hostlen, serv, servlen, flags);
2894 if (res == 0) {
2895 if (host && hostlen)
2896 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, host, internal_strlen(host) + 1);
2897 if (serv && servlen)
2898 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, serv, internal_strlen(serv) + 1);
2899 }
2900 return res;
2901 }
2902 #define INIT_GETNAMEINFO COMMON_INTERCEPT_FUNCTION(getnameinfo);
2903 #else
2904 #define INIT_GETNAMEINFO
2905 #endif
2907 #if SANITIZER_INTERCEPT_GETSOCKNAME
2908 INTERCEPTOR(int, getsockname, int sock_fd, void *addr, unsigned *addrlen) {
2909 void *ctx;
2910 COMMON_INTERCEPTOR_ENTER(ctx, getsockname, sock_fd, addr, addrlen);
2911 unsigned addr_sz;
2912 if (addrlen) {
2913 COMMON_INTERCEPTOR_READ_RANGE(ctx, addrlen, sizeof(*addrlen));
2914 addr_sz = *addrlen;
2915 }
2916 // FIXME: under ASan the call below may write to freed memory and corrupt
2917 // its metadata. See
2918 // https://github.com/google/sanitizers/issues/321.
2919 int res = REAL(getsockname)(sock_fd, addr, addrlen);
2920 if (!res && addr && addrlen) {
2921 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, addr, Min(addr_sz, *addrlen));
2922 }
2923 return res;
2924 }
2925 #define INIT_GETSOCKNAME COMMON_INTERCEPT_FUNCTION(getsockname);
2926 #else
2927 #define INIT_GETSOCKNAME
2928 #endif
2930 #if SANITIZER_INTERCEPT_GETHOSTBYNAME || SANITIZER_INTERCEPT_GETHOSTBYNAME_R
2931 static void write_hostent(void *ctx, struct __sanitizer_hostent *h) {
2932 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, h, sizeof(__sanitizer_hostent));
2933 if (h->h_name)
2934 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, h->h_name, internal_strlen(h->h_name) + 1);
2935 char **p = h->h_aliases;
2936 while (*p) {
2937 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *p, internal_strlen(*p) + 1);
2938 ++p;
2939 }
2940 COMMON_INTERCEPTOR_WRITE_RANGE(
2941 ctx, h->h_aliases, (p - h->h_aliases + 1) * sizeof(*h->h_aliases));
2942 p = h->h_addr_list;
2943 while (*p) {
2944 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *p, h->h_length);
2945 ++p;
2946 }
2947 COMMON_INTERCEPTOR_WRITE_RANGE(
2948 ctx, h->h_addr_list, (p - h->h_addr_list + 1) * sizeof(*h->h_addr_list));
2949 }
2950 #endif
2952 #if SANITIZER_INTERCEPT_GETHOSTBYNAME
2953 INTERCEPTOR(struct __sanitizer_hostent *, gethostbyname, char *name) {
2954 void *ctx;
2955 COMMON_INTERCEPTOR_ENTER(ctx, gethostbyname, name);
2956 struct __sanitizer_hostent *res = REAL(gethostbyname)(name);
2957 if (res) write_hostent(ctx, res);
2958 return res;
2959 }
2961 INTERCEPTOR(struct __sanitizer_hostent *, gethostbyaddr, void *addr, int len,
2962 int type) {
2963 void *ctx;
2964 COMMON_INTERCEPTOR_ENTER(ctx, gethostbyaddr, addr, len, type);
2965 COMMON_INTERCEPTOR_READ_RANGE(ctx, addr, len);
2966 struct __sanitizer_hostent *res = REAL(gethostbyaddr)(addr, len, type);
2967 if (res) write_hostent(ctx, res);
2968 return res;
2969 }
2971 INTERCEPTOR(struct __sanitizer_hostent *, gethostent, int fake) {
2972 void *ctx;
2973 COMMON_INTERCEPTOR_ENTER(ctx, gethostent, fake);
2974 struct __sanitizer_hostent *res = REAL(gethostent)(fake);
2975 if (res) write_hostent(ctx, res);
2976 return res;
2977 }
2978 #define INIT_GETHOSTBYNAME \
2979 COMMON_INTERCEPT_FUNCTION(gethostent); \
2980 COMMON_INTERCEPT_FUNCTION(gethostbyaddr); \
2981 COMMON_INTERCEPT_FUNCTION(gethostbyname);
2982 #else
2983 #define INIT_GETHOSTBYNAME
2984 #endif // SANITIZER_INTERCEPT_GETHOSTBYNAME
2986 #if SANITIZER_INTERCEPT_GETHOSTBYNAME2
2987 INTERCEPTOR(struct __sanitizer_hostent *, gethostbyname2, char *name, int af) {
2988 void *ctx;
2989 COMMON_INTERCEPTOR_ENTER(ctx, gethostbyname2, name, af);
2990 struct __sanitizer_hostent *res = REAL(gethostbyname2)(name, af);
2991 if (res) write_hostent(ctx, res);
2992 return res;
2993 }
2994 #define INIT_GETHOSTBYNAME2 COMMON_INTERCEPT_FUNCTION(gethostbyname2);
2995 #else
2996 #define INIT_GETHOSTBYNAME2
2997 #endif // SANITIZER_INTERCEPT_GETHOSTBYNAME2
2999 #if SANITIZER_INTERCEPT_GETHOSTBYNAME_R
3000 INTERCEPTOR(int, gethostbyname_r, char *name, struct __sanitizer_hostent *ret,
3001 char *buf, SIZE_T buflen, __sanitizer_hostent **result,
3002 int *h_errnop) {
3003 void *ctx;
3004 COMMON_INTERCEPTOR_ENTER(ctx, gethostbyname_r, name, ret, buf, buflen, result,
3005 h_errnop);
3006 // FIXME: under ASan the call below may write to freed memory and corrupt
3007 // its metadata. See
3008 // https://github.com/google/sanitizers/issues/321.
3009 int res = REAL(gethostbyname_r)(name, ret, buf, buflen, result, h_errnop);
3010 if (result) {
3011 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
3012 if (res == 0 && *result) write_hostent(ctx, *result);
3013 }
3014 if (h_errnop)
3015 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, h_errnop, sizeof(*h_errnop));
3016 return res;
3017 }
3018 #define INIT_GETHOSTBYNAME_R COMMON_INTERCEPT_FUNCTION(gethostbyname_r);
3019 #else
3020 #define INIT_GETHOSTBYNAME_R
3021 #endif
3023 #if SANITIZER_INTERCEPT_GETHOSTENT_R
3024 INTERCEPTOR(int, gethostent_r, struct __sanitizer_hostent *ret, char *buf,
3025 SIZE_T buflen, __sanitizer_hostent **result, int *h_errnop) {
3026 void *ctx;
3027 COMMON_INTERCEPTOR_ENTER(ctx, gethostent_r, ret, buf, buflen, result,
3028 h_errnop);
3029 // FIXME: under ASan the call below may write to freed memory and corrupt
3030 // its metadata. See
3031 // https://github.com/google/sanitizers/issues/321.
3032 int res = REAL(gethostent_r)(ret, buf, buflen, result, h_errnop);
3033 if (result) {
3034 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
3035 if (res == 0 && *result) write_hostent(ctx, *result);
3036 }
3037 if (h_errnop)
3038 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, h_errnop, sizeof(*h_errnop));
3039 return res;
3040 }
3041 #define INIT_GETHOSTENT_R \
3042 COMMON_INTERCEPT_FUNCTION(gethostent_r);
3043 #else
3044 #define INIT_GETHOSTENT_R
3045 #endif
3047 #if SANITIZER_INTERCEPT_GETHOSTBYADDR_R
3048 INTERCEPTOR(int, gethostbyaddr_r, void *addr, int len, int type,
3049 struct __sanitizer_hostent *ret, char *buf, SIZE_T buflen,
3050 __sanitizer_hostent **result, int *h_errnop) {
3051 void *ctx;
3052 COMMON_INTERCEPTOR_ENTER(ctx, gethostbyaddr_r, addr, len, type, ret, buf,
3053 buflen, result, h_errnop);
3054 COMMON_INTERCEPTOR_READ_RANGE(ctx, addr, len);
3055 // FIXME: under ASan the call below may write to freed memory and corrupt
3056 // its metadata. See
3057 // https://github.com/google/sanitizers/issues/321.
3058 int res = REAL(gethostbyaddr_r)(addr, len, type, ret, buf, buflen, result,
3059 h_errnop);
3060 if (result) {
3061 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
3062 if (res == 0 && *result) write_hostent(ctx, *result);
3063 }
3064 if (h_errnop)
3065 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, h_errnop, sizeof(*h_errnop));
3066 return res;
3067 }
3068 #define INIT_GETHOSTBYADDR_R \
3069 COMMON_INTERCEPT_FUNCTION(gethostbyaddr_r);
3070 #else
3071 #define INIT_GETHOSTBYADDR_R
3072 #endif
3074 #if SANITIZER_INTERCEPT_GETHOSTBYNAME2_R
3075 INTERCEPTOR(int, gethostbyname2_r, char *name, int af,
3076 struct __sanitizer_hostent *ret, char *buf, SIZE_T buflen,
3077 __sanitizer_hostent **result, int *h_errnop) {
3078 void *ctx;
3079 COMMON_INTERCEPTOR_ENTER(ctx, gethostbyname2_r, name, af, ret, buf, buflen,
3080 result, h_errnop);
3081 // FIXME: under ASan the call below may write to freed memory and corrupt
3082 // its metadata. See
3083 // https://github.com/google/sanitizers/issues/321.
3084 int res =
3085 REAL(gethostbyname2_r)(name, af, ret, buf, buflen, result, h_errnop);
3086 if (result) {
3087 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
3088 if (res == 0 && *result) write_hostent(ctx, *result);
3089 }
3090 if (h_errnop)
3091 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, h_errnop, sizeof(*h_errnop));
3092 return res;
3093 }
3094 #define INIT_GETHOSTBYNAME2_R \
3095 COMMON_INTERCEPT_FUNCTION(gethostbyname2_r);
3096 #else
3097 #define INIT_GETHOSTBYNAME2_R
3098 #endif
3100 #if SANITIZER_INTERCEPT_GETSOCKOPT
3101 INTERCEPTOR(int, getsockopt, int sockfd, int level, int optname, void *optval,
3102 int *optlen) {
3103 void *ctx;
3104 COMMON_INTERCEPTOR_ENTER(ctx, getsockopt, sockfd, level, optname, optval,
3105 optlen);
3106 if (optlen) COMMON_INTERCEPTOR_READ_RANGE(ctx, optlen, sizeof(*optlen));
3107 // FIXME: under ASan the call below may write to freed memory and corrupt
3108 // its metadata. See
3109 // https://github.com/google/sanitizers/issues/321.
3110 int res = REAL(getsockopt)(sockfd, level, optname, optval, optlen);
3111 if (res == 0)
3112 if (optval && optlen) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, optval, *optlen);
3113 return res;
3114 }
3115 #define INIT_GETSOCKOPT COMMON_INTERCEPT_FUNCTION(getsockopt);
3116 #else
3117 #define INIT_GETSOCKOPT
3118 #endif
3120 #if SANITIZER_INTERCEPT_ACCEPT
3121 INTERCEPTOR(int, accept, int fd, void *addr, unsigned *addrlen) {
3122 void *ctx;
3123 COMMON_INTERCEPTOR_ENTER(ctx, accept, fd, addr, addrlen);
3124 unsigned addrlen0 = 0;
3125 if (addrlen) {
3126 COMMON_INTERCEPTOR_READ_RANGE(ctx, addrlen, sizeof(*addrlen));
3127 addrlen0 = *addrlen;
3128 }
3129 int fd2 = COMMON_INTERCEPTOR_BLOCK_REAL(accept)(fd, addr, addrlen);
3130 if (fd2 >= 0) {
3131 if (fd >= 0) COMMON_INTERCEPTOR_FD_SOCKET_ACCEPT(ctx, fd, fd2);
3132 if (addr && addrlen)
3133 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, addr, Min(*addrlen, addrlen0));
3134 }
3135 return fd2;
3136 }
3137 #define INIT_ACCEPT COMMON_INTERCEPT_FUNCTION(accept);
3138 #else
3139 #define INIT_ACCEPT
3140 #endif
3142 #if SANITIZER_INTERCEPT_ACCEPT4
3143 INTERCEPTOR(int, accept4, int fd, void *addr, unsigned *addrlen, int f) {
3144 void *ctx;
3145 COMMON_INTERCEPTOR_ENTER(ctx, accept4, fd, addr, addrlen, f);
3146 unsigned addrlen0 = 0;
3147 if (addrlen) {
3148 COMMON_INTERCEPTOR_READ_RANGE(ctx, addrlen, sizeof(*addrlen));
3149 addrlen0 = *addrlen;
3150 }
3151 // FIXME: under ASan the call below may write to freed memory and corrupt
3152 // its metadata. See
3153 // https://github.com/google/sanitizers/issues/321.
3154 int fd2 = COMMON_INTERCEPTOR_BLOCK_REAL(accept4)(fd, addr, addrlen, f);
3155 if (fd2 >= 0) {
3156 if (fd >= 0) COMMON_INTERCEPTOR_FD_SOCKET_ACCEPT(ctx, fd, fd2);
3157 if (addr && addrlen)
3158 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, addr, Min(*addrlen, addrlen0));
3159 }
3160 return fd2;
3161 }
3162 #define INIT_ACCEPT4 COMMON_INTERCEPT_FUNCTION(accept4);
3163 #else
3164 #define INIT_ACCEPT4
3165 #endif
3167 #if SANITIZER_INTERCEPT_PACCEPT
3168 INTERCEPTOR(int, paccept, int fd, void *addr, unsigned *addrlen,
3169 __sanitizer_sigset_t *set, int f) {
3170 void *ctx;
3171 COMMON_INTERCEPTOR_ENTER(ctx, paccept, fd, addr, addrlen, set, f);
3172 unsigned addrlen0 = 0;
3173 if (addrlen) {
3174 COMMON_INTERCEPTOR_READ_RANGE(ctx, addrlen, sizeof(*addrlen));
3175 addrlen0 = *addrlen;
3176 }
3177 if (set) COMMON_INTERCEPTOR_READ_RANGE(ctx, set, sizeof(*set));
3178 int fd2 = COMMON_INTERCEPTOR_BLOCK_REAL(paccept)(fd, addr, addrlen, set, f);
3179 if (fd2 >= 0) {
3180 if (fd >= 0) COMMON_INTERCEPTOR_FD_SOCKET_ACCEPT(ctx, fd, fd2);
3181 if (addr && addrlen)
3182 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, addr, Min(*addrlen, addrlen0));
3183 }
3184 return fd2;
3185 }
3186 #define INIT_PACCEPT COMMON_INTERCEPT_FUNCTION(paccept);
3187 #else
3188 #define INIT_PACCEPT
3189 #endif
3191 #if SANITIZER_INTERCEPT_MODF
3192 INTERCEPTOR(double, modf, double x, double *iptr) {
3193 void *ctx;
3194 COMMON_INTERCEPTOR_ENTER(ctx, modf, x, iptr);
3195 // FIXME: under ASan the call below may write to freed memory and corrupt
3196 // its metadata. See
3197 // https://github.com/google/sanitizers/issues/321.
3198 double res = REAL(modf)(x, iptr);
3199 if (iptr) {
3200 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, iptr, sizeof(*iptr));
3201 }
3202 return res;
3203 }
3204 INTERCEPTOR(float, modff, float x, float *iptr) {
3205 void *ctx;
3206 COMMON_INTERCEPTOR_ENTER(ctx, modff, x, iptr);
3207 // FIXME: under ASan the call below may write to freed memory and corrupt
3208 // its metadata. See
3209 // https://github.com/google/sanitizers/issues/321.
3210 float res = REAL(modff)(x, iptr);
3211 if (iptr) {
3212 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, iptr, sizeof(*iptr));
3213 }
3214 return res;
3215 }
3216 INTERCEPTOR(long double, modfl, long double x, long double *iptr) {
3217 void *ctx;
3218 COMMON_INTERCEPTOR_ENTER(ctx, modfl, x, iptr);
3219 // FIXME: under ASan the call below may write to freed memory and corrupt
3220 // its metadata. See
3221 // https://github.com/google/sanitizers/issues/321.
3222 long double res = REAL(modfl)(x, iptr);
3223 if (iptr) {
3224 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, iptr, sizeof(*iptr));
3225 }
3226 return res;
3227 }
3228 #define INIT_MODF \
3229 COMMON_INTERCEPT_FUNCTION(modf); \
3230 COMMON_INTERCEPT_FUNCTION(modff); \
3231 COMMON_INTERCEPT_FUNCTION_LDBL(modfl);
3232 #else
3233 #define INIT_MODF
3234 #endif
3236 #if SANITIZER_INTERCEPT_RECVMSG || SANITIZER_INTERCEPT_RECVMMSG
3237 static void write_msghdr(void *ctx, struct __sanitizer_msghdr *msg,
3238 SSIZE_T maxlen) {
3239 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, msg, sizeof(*msg));
3240 if (msg->msg_name && msg->msg_namelen)
3241 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, msg->msg_name, msg->msg_namelen);
3242 if (msg->msg_iov && msg->msg_iovlen)
3243 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, msg->msg_iov,
3244 sizeof(*msg->msg_iov) * msg->msg_iovlen);
3245 write_iovec(ctx, msg->msg_iov, msg->msg_iovlen, maxlen);
3246 if (msg->msg_control && msg->msg_controllen)
3247 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, msg->msg_control, msg->msg_controllen);
3248 }
3249 #endif
3251 #if SANITIZER_INTERCEPT_RECVMSG
3252 INTERCEPTOR(SSIZE_T, recvmsg, int fd, struct __sanitizer_msghdr *msg,
3253 int flags) {
3254 void *ctx;
3255 COMMON_INTERCEPTOR_ENTER(ctx, recvmsg, fd, msg, flags);
3256 // FIXME: under ASan the call below may write to freed memory and corrupt
3257 // its metadata. See
3258 // https://github.com/google/sanitizers/issues/321.
3259 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(recvmsg)(fd, msg, flags);
3260 if (res >= 0) {
3261 if (fd >= 0) COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
3262 if (msg) {
3263 write_msghdr(ctx, msg, res);
3264 COMMON_INTERCEPTOR_HANDLE_RECVMSG(ctx, msg);
3265 }
3266 }
3267 return res;
3268 }
3269 #define INIT_RECVMSG COMMON_INTERCEPT_FUNCTION(recvmsg);
3270 #else
3271 #define INIT_RECVMSG
3272 #endif
3274 #if SANITIZER_INTERCEPT_RECVMMSG
3275 INTERCEPTOR(int, recvmmsg, int fd, struct __sanitizer_mmsghdr *msgvec,
3276 unsigned int vlen, int flags, void *timeout) {
3277 void *ctx;
3278 COMMON_INTERCEPTOR_ENTER(ctx, recvmmsg, fd, msgvec, vlen, flags, timeout);
3279 if (timeout) COMMON_INTERCEPTOR_READ_RANGE(ctx, timeout, struct_timespec_sz);
3280 int res =
3281 COMMON_INTERCEPTOR_BLOCK_REAL(recvmmsg)(fd, msgvec, vlen, flags, timeout);
3282 if (res >= 0) {
3283 if (fd >= 0) COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
3284 for (int i = 0; i < res; ++i) {
3285 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, &msgvec[i].msg_len,
3286 sizeof(msgvec[i].msg_len));
3287 write_msghdr(ctx, &msgvec[i].msg_hdr, msgvec[i].msg_len);
3288 COMMON_INTERCEPTOR_HANDLE_RECVMSG(ctx, &msgvec[i].msg_hdr);
3289 }
3290 }
3291 return res;
3292 }
3293 #define INIT_RECVMMSG COMMON_INTERCEPT_FUNCTION(recvmmsg);
3294 #else
3295 #define INIT_RECVMMSG
3296 #endif
3298 #if SANITIZER_INTERCEPT_SENDMSG || SANITIZER_INTERCEPT_SENDMMSG
3299 static void read_msghdr_control(void *ctx, void *control, uptr controllen) {
3300 const unsigned kCmsgDataOffset =
3301 RoundUpTo(sizeof(__sanitizer_cmsghdr), sizeof(uptr));
3303 char *p = (char *)control;
3304 char *const control_end = p + controllen;
3305 while (true) {
3306 if (p + sizeof(__sanitizer_cmsghdr) > control_end) break;
3307 __sanitizer_cmsghdr *cmsg = (__sanitizer_cmsghdr *)p;
3308 COMMON_INTERCEPTOR_READ_RANGE(ctx, &cmsg->cmsg_len, sizeof(cmsg->cmsg_len));
3310 if (p + RoundUpTo(cmsg->cmsg_len, sizeof(uptr)) > control_end) break;
3312 COMMON_INTERCEPTOR_READ_RANGE(ctx, &cmsg->cmsg_level,
3313 sizeof(cmsg->cmsg_level));
3314 COMMON_INTERCEPTOR_READ_RANGE(ctx, &cmsg->cmsg_type,
3315 sizeof(cmsg->cmsg_type));
3317 if (cmsg->cmsg_len > kCmsgDataOffset) {
3318 char *data = p + kCmsgDataOffset;
3319 unsigned data_len = cmsg->cmsg_len - kCmsgDataOffset;
3320 if (data_len > 0) COMMON_INTERCEPTOR_READ_RANGE(ctx, data, data_len);
3321 }
3323 p += RoundUpTo(cmsg->cmsg_len, sizeof(uptr));
3324 }
3325 }
3327 static void read_msghdr(void *ctx, struct __sanitizer_msghdr *msg,
3328 SSIZE_T maxlen) {
3329 #define R(f) \
3330 COMMON_INTERCEPTOR_READ_RANGE(ctx, &msg->msg_##f, sizeof(msg->msg_##f))
3331 R(name);
3332 R(namelen);
3333 R(iov);
3334 R(iovlen);
3335 R(control);
3336 R(controllen);
3337 R(flags);
3338 #undef R
3339 if (msg->msg_name && msg->msg_namelen)
3340 COMMON_INTERCEPTOR_READ_RANGE(ctx, msg->msg_name, msg->msg_namelen);
3341 if (msg->msg_iov && msg->msg_iovlen)
3342 COMMON_INTERCEPTOR_READ_RANGE(ctx, msg->msg_iov,
3343 sizeof(*msg->msg_iov) * msg->msg_iovlen);
3344 read_iovec(ctx, msg->msg_iov, msg->msg_iovlen, maxlen);
3345 if (msg->msg_control && msg->msg_controllen)
3346 read_msghdr_control(ctx, msg->msg_control, msg->msg_controllen);
3347 }
3348 #endif
3350 #if SANITIZER_INTERCEPT_SENDMSG
3351 INTERCEPTOR(SSIZE_T, sendmsg, int fd, struct __sanitizer_msghdr *msg,
3352 int flags) {
3353 void *ctx;
3354 COMMON_INTERCEPTOR_ENTER(ctx, sendmsg, fd, msg, flags);
3355 if (fd >= 0) {
3356 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
3357 COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd);
3358 }
3359 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(sendmsg)(fd, msg, flags);
3360 if (common_flags()->intercept_send && res >= 0 && msg)
3361 read_msghdr(ctx, msg, res);
3362 return res;
3363 }
3364 #define INIT_SENDMSG COMMON_INTERCEPT_FUNCTION(sendmsg);
3365 #else
3366 #define INIT_SENDMSG
3367 #endif
3369 #if SANITIZER_INTERCEPT_SENDMMSG
3370 INTERCEPTOR(int, sendmmsg, int fd, struct __sanitizer_mmsghdr *msgvec,
3371 unsigned vlen, int flags) {
3372 void *ctx;
3373 COMMON_INTERCEPTOR_ENTER(ctx, sendmmsg, fd, msgvec, vlen, flags);
3374 if (fd >= 0) {
3375 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
3376 COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd);
3377 }
3378 int res = COMMON_INTERCEPTOR_BLOCK_REAL(sendmmsg)(fd, msgvec, vlen, flags);
3379 if (res >= 0 && msgvec) {
3380 for (int i = 0; i < res; ++i) {
3381 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, &msgvec[i].msg_len,
3382 sizeof(msgvec[i].msg_len));
3383 if (common_flags()->intercept_send)
3384 read_msghdr(ctx, &msgvec[i].msg_hdr, msgvec[i].msg_len);
3385 }
3386 }
3387 return res;
3388 }
3389 #define INIT_SENDMMSG COMMON_INTERCEPT_FUNCTION(sendmmsg);
3390 #else
3391 #define INIT_SENDMMSG
3392 #endif
3394 #if SANITIZER_INTERCEPT_SYSMSG
3395 INTERCEPTOR(int, msgsnd, int msqid, const void *msgp, SIZE_T msgsz,
3396 int msgflg) {
3397 void *ctx;
3398 COMMON_INTERCEPTOR_ENTER(ctx, msgsnd, msqid, msgp, msgsz, msgflg);
3399 if (msgp)
3400 COMMON_INTERCEPTOR_READ_RANGE(ctx, msgp, sizeof(long) + msgsz);
3401 int res = COMMON_INTERCEPTOR_BLOCK_REAL(msgsnd)(msqid, msgp, msgsz, msgflg);
3402 return res;
3403 }
3405 INTERCEPTOR(SSIZE_T, msgrcv, int msqid, void *msgp, SIZE_T msgsz,
3406 long msgtyp, int msgflg) {
3407 void *ctx;
3408 COMMON_INTERCEPTOR_ENTER(ctx, msgrcv, msqid, msgp, msgsz, msgtyp, msgflg);
3409 SSIZE_T len =
3410 COMMON_INTERCEPTOR_BLOCK_REAL(msgrcv)(msqid, msgp, msgsz, msgtyp, msgflg);
3411 if (len != -1)
3412 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, msgp, sizeof(long) + len);
3413 return len;
3414 }
3416 #define INIT_SYSMSG \
3417 COMMON_INTERCEPT_FUNCTION(msgsnd); \
3418 COMMON_INTERCEPT_FUNCTION(msgrcv);
3419 #else
3420 #define INIT_SYSMSG
3421 #endif
3423 #if SANITIZER_INTERCEPT_GETPEERNAME
3424 INTERCEPTOR(int, getpeername, int sockfd, void *addr, unsigned *addrlen) {
3425 void *ctx;
3426 COMMON_INTERCEPTOR_ENTER(ctx, getpeername, sockfd, addr, addrlen);
3427 unsigned addr_sz;
3428 if (addrlen) {
3429 COMMON_INTERCEPTOR_READ_RANGE(ctx, addrlen, sizeof(*addrlen));
3430 addr_sz = *addrlen;
3431 }
3432 // FIXME: under ASan the call below may write to freed memory and corrupt
3433 // its metadata. See
3434 // https://github.com/google/sanitizers/issues/321.
3435 int res = REAL(getpeername)(sockfd, addr, addrlen);
3436 if (!res && addr && addrlen) {
3437 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, addr, Min(addr_sz, *addrlen));
3438 }
3439 return res;
3440 }
3441 #define INIT_GETPEERNAME COMMON_INTERCEPT_FUNCTION(getpeername);
3442 #else
3443 #define INIT_GETPEERNAME
3444 #endif
3446 #if SANITIZER_INTERCEPT_SYSINFO
3447 INTERCEPTOR(int, sysinfo, void *info) {
3448 void *ctx;
3449 // FIXME: under ASan the call below may write to freed memory and corrupt
3450 // its metadata. See
3451 // https://github.com/google/sanitizers/issues/321.
3452 COMMON_INTERCEPTOR_ENTER(ctx, sysinfo, info);
3453 int res = REAL(sysinfo)(info);
3454 if (!res && info)
3455 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, info, struct_sysinfo_sz);
3456 return res;
3457 }
3458 #define INIT_SYSINFO COMMON_INTERCEPT_FUNCTION(sysinfo);
3459 #else
3460 #define INIT_SYSINFO
3461 #endif
3463 #if SANITIZER_INTERCEPT_READDIR
3464 INTERCEPTOR(__sanitizer_dirent *, opendir, const char *path) {
3465 void *ctx;
3466 COMMON_INTERCEPTOR_ENTER(ctx, opendir, path);
3467 COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
3468 __sanitizer_dirent *res = REAL(opendir)(path);
3469 if (res)
3470 COMMON_INTERCEPTOR_DIR_ACQUIRE(ctx, path);
3471 return res;
3472 }
3474 INTERCEPTOR(__sanitizer_dirent *, readdir, void *dirp) {
3475 void *ctx;
3476 COMMON_INTERCEPTOR_ENTER(ctx, readdir, dirp);
3477 // FIXME: under ASan the call below may write to freed memory and corrupt
3478 // its metadata. See
3479 // https://github.com/google/sanitizers/issues/321.
3480 __sanitizer_dirent *res = REAL(readdir)(dirp);
3481 if (res)
3482 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, __sanitizer_dirsiz(res));
3483 return res;
3484 }
3486 INTERCEPTOR(int, readdir_r, void *dirp, __sanitizer_dirent *entry,
3487 __sanitizer_dirent **result) {
3488 void *ctx;
3489 COMMON_INTERCEPTOR_ENTER(ctx, readdir_r, dirp, entry, result);
3490 // FIXME: under ASan the call below may write to freed memory and corrupt
3491 // its metadata. See
3492 // https://github.com/google/sanitizers/issues/321.
3493 int res = REAL(readdir_r)(dirp, entry, result);
3494 if (!res) {
3495 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
3496 if (*result)
3497 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *result, __sanitizer_dirsiz(*result));
3498 }
3499 return res;
3500 }
3502 #define INIT_READDIR \
3503 COMMON_INTERCEPT_FUNCTION(opendir); \
3504 COMMON_INTERCEPT_FUNCTION(readdir); \
3505 COMMON_INTERCEPT_FUNCTION(readdir_r);
3506 #else
3507 #define INIT_READDIR
3508 #endif
3510 #if SANITIZER_INTERCEPT_READDIR64
3511 INTERCEPTOR(__sanitizer_dirent64 *, readdir64, void *dirp) {
3512 void *ctx;
3513 COMMON_INTERCEPTOR_ENTER(ctx, readdir64, dirp);
3514 // FIXME: under ASan the call below may write to freed memory and corrupt
3515 // its metadata. See
3516 // https://github.com/google/sanitizers/issues/321.
3517 __sanitizer_dirent64 *res = REAL(readdir64)(dirp);
3518 if (res)
3519 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, __sanitizer_dirsiz(res));
3520 return res;
3521 }
3523 INTERCEPTOR(int, readdir64_r, void *dirp, __sanitizer_dirent64 *entry,
3524 __sanitizer_dirent64 **result) {
3525 void *ctx;
3526 COMMON_INTERCEPTOR_ENTER(ctx, readdir64_r, dirp, entry, result);
3527 // FIXME: under ASan the call below may write to freed memory and corrupt
3528 // its metadata. See
3529 // https://github.com/google/sanitizers/issues/321.
3530 int res = REAL(readdir64_r)(dirp, entry, result);
3531 if (!res) {
3532 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
3533 if (*result)
3534 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *result, __sanitizer_dirsiz(*result));
3535 }
3536 return res;
3537 }
3538 #define INIT_READDIR64 \
3539 COMMON_INTERCEPT_FUNCTION(readdir64); \
3540 COMMON_INTERCEPT_FUNCTION(readdir64_r);
3541 #else
3542 #define INIT_READDIR64
3543 #endif
3545 #if SANITIZER_INTERCEPT_PTRACE
3546 INTERCEPTOR(uptr, ptrace, int request, int pid, void *addr, void *data) {
3547 void *ctx;
3548 COMMON_INTERCEPTOR_ENTER(ctx, ptrace, request, pid, addr, data);
3549 __sanitizer_iovec local_iovec;
3551 void *data_arg = ptrace_data_arg(request, addr, data);
3552 if (data_arg) {
3553 if (request == ptrace_setregs) {
3554 COMMON_INTERCEPTOR_READ_RANGE(ctx, data_arg, struct_user_regs_struct_sz);
3555 } else if (request == ptrace_setfpregs) {
3556 COMMON_INTERCEPTOR_READ_RANGE(ctx, data_arg,
3557 struct_user_fpregs_struct_sz);
3558 } else if (request == ptrace_setfpxregs) {
3559 COMMON_INTERCEPTOR_READ_RANGE(ctx, data_arg,
3560 struct_user_fpxregs_struct_sz);
3561 } else if (request == ptrace_setvfpregs) {
3562 COMMON_INTERCEPTOR_READ_RANGE(ctx, data_arg,
3563 struct_user_vfpregs_struct_sz);
3564 } else if (request == ptrace_setsiginfo) {
3565 COMMON_INTERCEPTOR_READ_RANGE(ctx, data_arg, siginfo_t_sz);
3567 // Some kernel might zero the iovec::iov_base in case of invalid
3568 // write access. In this case copy the invalid address for further
3569 // inspection.
3570 } else if (request == ptrace_setregset || request == ptrace_getregset) {
3571 __sanitizer_iovec *iovec = (__sanitizer_iovec *)data_arg;
3572 COMMON_INTERCEPTOR_READ_RANGE(ctx, iovec, sizeof(*iovec));
3573 local_iovec = *iovec;
3574 if (request == ptrace_setregset)
3575 COMMON_INTERCEPTOR_READ_RANGE(ctx, iovec->iov_base, iovec->iov_len);
3576 }
3577 }
3579 // FIXME: under ASan the call below may write to freed memory and corrupt
3580 // its metadata. See
3581 // https://github.com/google/sanitizers/issues/321.
3582 uptr res = REAL(ptrace)(request, pid, addr, data);
3584 if (!res && data_arg) {
3585 // Note that PEEK* requests assign different meaning to the return value.
3586 // This function does not handle them (nor does it need to).
3587 if (request == ptrace_getregs) {
3588 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, data_arg, struct_user_regs_struct_sz);
3589 } else if (request == ptrace_getfpregs) {
3590 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, data_arg,
3591 struct_user_fpregs_struct_sz);
3592 } else if (request == ptrace_getfpxregs) {
3593 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, data_arg,
3594 struct_user_fpxregs_struct_sz);
3595 } else if (request == ptrace_getvfpregs) {
3596 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, data_arg,
3597 struct_user_vfpregs_struct_sz);
3598 } else if (request == ptrace_getsiginfo) {
3599 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, data_arg, siginfo_t_sz);
3600 } else if (request == ptrace_geteventmsg) {
3601 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, data_arg, sizeof(unsigned long));
3602 } else if (request == ptrace_getregset) {
3603 __sanitizer_iovec *iovec = (__sanitizer_iovec *)data_arg;
3604 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, iovec, sizeof(*iovec));
3605 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, local_iovec.iov_base,
3606 local_iovec.iov_len);
3607 }
3608 }
3609 return res;
3610 }
3612 #define INIT_PTRACE COMMON_INTERCEPT_FUNCTION(ptrace);
3613 #else
3614 #define INIT_PTRACE
3615 #endif
3617 #if SANITIZER_INTERCEPT_SETLOCALE
3618 static void unpoison_ctype_arrays(void *ctx) {
3619 #if SANITIZER_NETBSD
3620 // These arrays contain 256 regular elements in unsigned char range + 1 EOF
3621 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, _ctype_tab_, 257 * sizeof(short));
3622 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, _toupper_tab_, 257 * sizeof(short));
3623 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, _tolower_tab_, 257 * sizeof(short));
3624 #endif
3625 }
3627 INTERCEPTOR(char *, setlocale, int category, char *locale) {
3628 void *ctx;
3629 COMMON_INTERCEPTOR_ENTER(ctx, setlocale, category, locale);
3630 if (locale)
3631 COMMON_INTERCEPTOR_READ_RANGE(ctx, locale, internal_strlen(locale) + 1);
3632 char *res = REAL(setlocale)(category, locale);
3633 if (res) {
3634 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, internal_strlen(res) + 1);
3635 unpoison_ctype_arrays(ctx);
3636 }
3637 return res;
3638 }
3640 #define INIT_SETLOCALE COMMON_INTERCEPT_FUNCTION(setlocale);
3641 #else
3642 #define INIT_SETLOCALE
3643 #endif
3645 #if SANITIZER_INTERCEPT_GETCWD
3646 INTERCEPTOR(char *, getcwd, char *buf, SIZE_T size) {
3647 void *ctx;
3648 COMMON_INTERCEPTOR_ENTER(ctx, getcwd, buf, size);
3649 // FIXME: under ASan the call below may write to freed memory and corrupt
3650 // its metadata. See
3651 // https://github.com/google/sanitizers/issues/321.
3652 char *res = REAL(getcwd)(buf, size);
3653 if (res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
3654 return res;
3655 }
3656 #define INIT_GETCWD COMMON_INTERCEPT_FUNCTION(getcwd);
3657 #else
3658 #define INIT_GETCWD
3659 #endif
3661 #if SANITIZER_INTERCEPT_GET_CURRENT_DIR_NAME
3662 INTERCEPTOR(char *, get_current_dir_name, int fake) {
3663 void *ctx;
3664 COMMON_INTERCEPTOR_ENTER(ctx, get_current_dir_name, fake);
3665 // FIXME: under ASan the call below may write to freed memory and corrupt
3666 // its metadata. See
3667 // https://github.com/google/sanitizers/issues/321.
3668 char *res = REAL(get_current_dir_name)(fake);
3669 if (res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
3670 return res;
3671 }
3673 #define INIT_GET_CURRENT_DIR_NAME \
3674 COMMON_INTERCEPT_FUNCTION(get_current_dir_name);
3675 #else
3676 #define INIT_GET_CURRENT_DIR_NAME
3677 #endif
3679 UNUSED static inline void FixRealStrtolEndptr(const char *nptr, char **endptr) {
3680 CHECK(endptr);
3681 if (nptr == *endptr) {
3682 // No digits were found at strtol call, we need to find out the last
3683 // symbol accessed by strtoll on our own.
3684 // We get this symbol by skipping leading blanks and optional +/- sign.
3685 while (IsSpace(*nptr)) nptr++;
3686 if (*nptr == '+' || *nptr == '-') nptr++;
3687 *endptr = const_cast<char *>(nptr);
3688 }
3689 CHECK(*endptr >= nptr);
3690 }
3692 UNUSED static inline void StrtolFixAndCheck(void *ctx, const char *nptr,
3693 char **endptr, char *real_endptr, int base) {
3694 if (endptr) {
3695 *endptr = real_endptr;
3696 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, endptr, sizeof(*endptr));
3697 }
3698 // If base has unsupported value, strtol can exit with EINVAL
3699 // without reading any characters. So do additional checks only
3700 // if base is valid.
3701 bool is_valid_base = (base == 0) || (2 <= base && base <= 36);
3702 if (is_valid_base) {
3703 FixRealStrtolEndptr(nptr, &real_endptr);
3704 }
3705 COMMON_INTERCEPTOR_READ_STRING(ctx, nptr, is_valid_base ?
3706 (real_endptr - nptr) + 1 : 0);
3707 }
3709 #if SANITIZER_INTERCEPT_STRTOIMAX
3710 template <typename Fn>
3711 static ALWAYS_INLINE auto StrtoimaxImpl(void *ctx, Fn real, const char *nptr,
3712 char **endptr, int base)
3713 -> decltype(real(nullptr, nullptr, 0)) {
3714 char *real_endptr;
3715 auto res = real(nptr, &real_endptr, base);
3716 StrtolFixAndCheck(ctx, nptr, endptr, real_endptr, base);
3717 return res;
3718 }
3720 INTERCEPTOR(INTMAX_T, strtoimax, const char *nptr, char **endptr, int base) {
3721 void *ctx;
3722 COMMON_INTERCEPTOR_ENTER(ctx, strtoimax, nptr, endptr, base);
3723 return StrtoimaxImpl(ctx, REAL(strtoimax), nptr, endptr, base);
3724 }
3725 INTERCEPTOR(UINTMAX_T, strtoumax, const char *nptr, char **endptr, int base) {
3726 void *ctx;
3727 COMMON_INTERCEPTOR_ENTER(ctx, strtoumax, nptr, endptr, base);
3728 return StrtoimaxImpl(ctx, REAL(strtoumax), nptr, endptr, base);
3729 }
3731 #define INIT_STRTOIMAX \
3732 COMMON_INTERCEPT_FUNCTION(strtoimax); \
3733 COMMON_INTERCEPT_FUNCTION(strtoumax);
3734 #else
3735 #define INIT_STRTOIMAX
3736 #endif
3738 #if SANITIZER_INTERCEPT_STRTOIMAX && SANITIZER_GLIBC
3739 INTERCEPTOR(INTMAX_T, __isoc23_strtoimax, const char *nptr, char **endptr, int base) {
3740 void *ctx;
3741 COMMON_INTERCEPTOR_ENTER(ctx, __isoc23_strtoimax, nptr, endptr, base);
3742 return StrtoimaxImpl(ctx, REAL(__isoc23_strtoimax), nptr, endptr, base);
3743 }
3744 INTERCEPTOR(UINTMAX_T, __isoc23_strtoumax, const char *nptr, char **endptr, int base) {
3745 void *ctx;
3746 COMMON_INTERCEPTOR_ENTER(ctx, __isoc23_strtoumax, nptr, endptr, base);
3747 return StrtoimaxImpl(ctx, REAL(__isoc23_strtoumax), nptr, endptr, base);
3748 }
3750 # define INIT_STRTOIMAX_C23 \
3751 COMMON_INTERCEPT_FUNCTION(__isoc23_strtoimax); \
3752 COMMON_INTERCEPT_FUNCTION(__isoc23_strtoumax);
3753 #else
3754 # define INIT_STRTOIMAX_C23
3755 #endif
3757 #if SANITIZER_INTERCEPT_MBSTOWCS
3758 INTERCEPTOR(SIZE_T, mbstowcs, wchar_t *dest, const char *src, SIZE_T len) {
3759 void *ctx;
3760 COMMON_INTERCEPTOR_ENTER(ctx, mbstowcs, dest, src, len);
3761 // FIXME: under ASan the call below may write to freed memory and corrupt
3762 // its metadata. See
3763 // https://github.com/google/sanitizers/issues/321.
3764 SIZE_T res = REAL(mbstowcs)(dest, src, len);
3765 if (res != (SIZE_T) - 1 && dest) {
3766 SIZE_T write_cnt = res + (res < len);
3767 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dest, write_cnt * sizeof(wchar_t));
3768 }
3769 return res;
3770 }
3772 INTERCEPTOR(SIZE_T, mbsrtowcs, wchar_t *dest, const char **src, SIZE_T len,
3773 void *ps) {
3774 void *ctx;
3775 COMMON_INTERCEPTOR_ENTER(ctx, mbsrtowcs, dest, src, len, ps);
3776 if (src) COMMON_INTERCEPTOR_READ_RANGE(ctx, src, sizeof(*src));
3777 if (ps) COMMON_INTERCEPTOR_READ_RANGE(ctx, ps, mbstate_t_sz);
3778 // FIXME: under ASan the call below may write to freed memory and corrupt
3779 // its metadata. See
3780 // https://github.com/google/sanitizers/issues/321.
3781 SIZE_T res = REAL(mbsrtowcs)(dest, src, len, ps);
3782 if (res != (SIZE_T)(-1) && dest && src) {
3783 // This function, and several others, may or may not write the terminating
3784 // \0 character. They write it iff they clear *src.
3785 SIZE_T write_cnt = res + !*src;
3786 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dest, write_cnt * sizeof(wchar_t));
3787 }
3788 return res;
3789 }
3791 #define INIT_MBSTOWCS \
3792 COMMON_INTERCEPT_FUNCTION(mbstowcs); \
3793 COMMON_INTERCEPT_FUNCTION(mbsrtowcs);
3794 #else
3795 #define INIT_MBSTOWCS
3796 #endif
3798 #if SANITIZER_INTERCEPT_MBSNRTOWCS
3799 INTERCEPTOR(SIZE_T, mbsnrtowcs, wchar_t *dest, const char **src, SIZE_T nms,
3800 SIZE_T len, void *ps) {
3801 void *ctx;
3802 COMMON_INTERCEPTOR_ENTER(ctx, mbsnrtowcs, dest, src, nms, len, ps);
3803 if (src) {
3804 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, sizeof(*src));
3805 if (nms) COMMON_INTERCEPTOR_READ_RANGE(ctx, *src, nms);
3806 }
3807 if (ps) COMMON_INTERCEPTOR_READ_RANGE(ctx, ps, mbstate_t_sz);
3808 // FIXME: under ASan the call below may write to freed memory and corrupt
3809 // its metadata. See
3810 // https://github.com/google/sanitizers/issues/321.
3811 SIZE_T res = REAL(mbsnrtowcs)(dest, src, nms, len, ps);
3812 if (res != (SIZE_T)(-1) && dest && src) {
3813 SIZE_T write_cnt = res + !*src;
3814 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dest, write_cnt * sizeof(wchar_t));
3815 }
3816 return res;
3817 }
3819 #define INIT_MBSNRTOWCS COMMON_INTERCEPT_FUNCTION(mbsnrtowcs);
3820 #else
3821 #define INIT_MBSNRTOWCS
3822 #endif
3824 #if SANITIZER_INTERCEPT_WCSTOMBS
3825 INTERCEPTOR(SIZE_T, wcstombs, char *dest, const wchar_t *src, SIZE_T len) {
3826 void *ctx;
3827 COMMON_INTERCEPTOR_ENTER(ctx, wcstombs, dest, src, len);
3828 // FIXME: under ASan the call below may write to freed memory and corrupt
3829 // its metadata. See
3830 // https://github.com/google/sanitizers/issues/321.
3831 SIZE_T res = REAL(wcstombs)(dest, src, len);
3832 if (res != (SIZE_T) - 1 && dest) {
3833 SIZE_T write_cnt = res + (res < len);
3834 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dest, write_cnt);
3835 }
3836 return res;
3837 }
3839 INTERCEPTOR(SIZE_T, wcsrtombs, char *dest, const wchar_t **src, SIZE_T len,
3840 void *ps) {
3841 void *ctx;
3842 COMMON_INTERCEPTOR_ENTER(ctx, wcsrtombs, dest, src, len, ps);
3843 if (src) COMMON_INTERCEPTOR_READ_RANGE(ctx, src, sizeof(*src));
3844 if (ps) COMMON_INTERCEPTOR_READ_RANGE(ctx, ps, mbstate_t_sz);
3845 // FIXME: under ASan the call below may write to freed memory and corrupt
3846 // its metadata. See
3847 // https://github.com/google/sanitizers/issues/321.
3848 SIZE_T res = REAL(wcsrtombs)(dest, src, len, ps);
3849 if (res != (SIZE_T) - 1 && dest && src) {
3850 SIZE_T write_cnt = res + !*src;
3851 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dest, write_cnt);
3852 }
3853 return res;
3854 }
3856 #define INIT_WCSTOMBS \
3857 COMMON_INTERCEPT_FUNCTION(wcstombs); \
3858 COMMON_INTERCEPT_FUNCTION(wcsrtombs);
3859 #else
3860 #define INIT_WCSTOMBS
3861 #endif
3863 #if SANITIZER_INTERCEPT_WCSNRTOMBS
3864 INTERCEPTOR(SIZE_T, wcsnrtombs, char *dest, const wchar_t **src, SIZE_T nms,
3865 SIZE_T len, void *ps) {
3866 void *ctx;
3867 COMMON_INTERCEPTOR_ENTER(ctx, wcsnrtombs, dest, src, nms, len, ps);
3868 if (src) {
3869 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, sizeof(*src));
3870 if (nms) COMMON_INTERCEPTOR_READ_RANGE(ctx, *src, nms);
3871 }
3872 if (ps) COMMON_INTERCEPTOR_READ_RANGE(ctx, ps, mbstate_t_sz);
3873 // FIXME: under ASan the call below may write to freed memory and corrupt
3874 // its metadata. See
3875 // https://github.com/google/sanitizers/issues/321.
3876 SIZE_T res = REAL(wcsnrtombs)(dest, src, nms, len, ps);
3877 if (res != ((SIZE_T)-1) && dest && src) {
3878 SIZE_T write_cnt = res + !*src;
3879 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dest, write_cnt);
3880 }
3881 return res;
3882 }
3884 #define INIT_WCSNRTOMBS COMMON_INTERCEPT_FUNCTION(wcsnrtombs);
3885 #else
3886 #define INIT_WCSNRTOMBS
3887 #endif
3890 #if SANITIZER_INTERCEPT_WCRTOMB
3891 INTERCEPTOR(SIZE_T, wcrtomb, char *dest, wchar_t src, void *ps) {
3892 void *ctx;
3893 COMMON_INTERCEPTOR_ENTER(ctx, wcrtomb, dest, src, ps);
3894 if (ps) COMMON_INTERCEPTOR_READ_RANGE(ctx, ps, mbstate_t_sz);
3896 if (!dest)
3897 return REAL(wcrtomb)(dest, src, ps);
3899 char local_dest[32];
3900 SIZE_T res = REAL(wcrtomb)(local_dest, src, ps);
3901 if (res != ((SIZE_T)-1)) {
3902 CHECK_LE(res, sizeof(local_dest));
3903 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dest, res);
3904 REAL(memcpy)(dest, local_dest, res);
3905 }
3906 return res;
3907 }
3909 #define INIT_WCRTOMB COMMON_INTERCEPT_FUNCTION(wcrtomb);
3910 #else
3911 #define INIT_WCRTOMB
3912 #endif
3914 #if SANITIZER_INTERCEPT_WCTOMB
3915 INTERCEPTOR(int, wctomb, char *dest, wchar_t src) {
3916 void *ctx;
3917 COMMON_INTERCEPTOR_ENTER(ctx, wctomb, dest, src);
3918 if (!dest)
3919 return REAL(wctomb)(dest, src);
3921 char local_dest[32];
3922 int res = REAL(wctomb)(local_dest, src);
3923 if (res != -1) {
3924 CHECK_LE(res, sizeof(local_dest));
3925 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dest, res);
3926 REAL(memcpy)(dest, local_dest, res);
3927 }
3928 return res;
3929 }
3931 #define INIT_WCTOMB COMMON_INTERCEPT_FUNCTION(wctomb);
3932 #else
3933 #define INIT_WCTOMB
3934 #endif
3936 #if SANITIZER_INTERCEPT_TCGETATTR
3937 INTERCEPTOR(int, tcgetattr, int fd, void *termios_p) {
3938 void *ctx;
3939 COMMON_INTERCEPTOR_ENTER(ctx, tcgetattr, fd, termios_p);
3940 // FIXME: under ASan the call below may write to freed memory and corrupt
3941 // its metadata. See
3942 // https://github.com/google/sanitizers/issues/321.
3943 int res = REAL(tcgetattr)(fd, termios_p);
3944 if (!res && termios_p)
3945 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, termios_p, struct_termios_sz);
3946 return res;
3947 }
3949 #define INIT_TCGETATTR COMMON_INTERCEPT_FUNCTION(tcgetattr);
3950 #else
3951 #define INIT_TCGETATTR
3952 #endif
3954 #if SANITIZER_INTERCEPT_REALPATH
3955 INTERCEPTOR(char *, realpath, const char *path, char *resolved_path) {
3956 void *ctx;
3957 COMMON_INTERCEPTOR_ENTER(ctx, realpath, path, resolved_path);
3958 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
3960 // Workaround a bug in glibc where dlsym(RTLD_NEXT, ...) returns the oldest
3961 // version of a versioned symbol. For realpath(), this gives us something
3962 // (called __old_realpath) that does not handle NULL in the second argument.
3963 // Handle it as part of the interceptor.
3964 char *allocated_path = nullptr;
3965 if (!resolved_path)
3966 allocated_path = resolved_path = (char *)WRAP(malloc)(path_max + 1);
3968 char *res = REAL(realpath)(path, resolved_path);
3969 if (allocated_path && !res)
3970 WRAP(free)(allocated_path);
3971 if (res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
3972 return res;
3973 }
3974 # define INIT_REALPATH COMMON_INTERCEPT_FUNCTION(realpath);
3975 #else
3976 #define INIT_REALPATH
3977 #endif
3979 #if SANITIZER_INTERCEPT_CANONICALIZE_FILE_NAME
3980 INTERCEPTOR(char *, canonicalize_file_name, const char *path) {
3981 void *ctx;
3982 COMMON_INTERCEPTOR_ENTER(ctx, canonicalize_file_name, path);
3983 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
3984 char *res = REAL(canonicalize_file_name)(path);
3985 if (res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
3986 return res;
3987 }
3988 #define INIT_CANONICALIZE_FILE_NAME \
3989 COMMON_INTERCEPT_FUNCTION(canonicalize_file_name);
3990 #else
3991 #define INIT_CANONICALIZE_FILE_NAME
3992 #endif
3994 #if SANITIZER_INTERCEPT_CONFSTR
3995 INTERCEPTOR(SIZE_T, confstr, int name, char *buf, SIZE_T len) {
3996 void *ctx;
3997 COMMON_INTERCEPTOR_ENTER(ctx, confstr, name, buf, len);
3998 // FIXME: under ASan the call below may write to freed memory and corrupt
3999 // its metadata. See
4000 // https://github.com/google/sanitizers/issues/321.
4001 SIZE_T res = REAL(confstr)(name, buf, len);
4002 if (buf && res)
4003 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, res < len ? res : len);
4004 return res;
4005 }
4006 #define INIT_CONFSTR COMMON_INTERCEPT_FUNCTION(confstr);
4007 #else
4008 #define INIT_CONFSTR
4009 #endif
4011 #if SANITIZER_INTERCEPT_SCHED_GETAFFINITY
4012 INTERCEPTOR(int, sched_getaffinity, int pid, SIZE_T cpusetsize, void *mask) {
4013 void *ctx;
4014 COMMON_INTERCEPTOR_ENTER(ctx, sched_getaffinity, pid, cpusetsize, mask);
4015 // FIXME: under ASan the call below may write to freed memory and corrupt
4016 // its metadata. See
4017 // https://github.com/google/sanitizers/issues/321.
4018 int res = REAL(sched_getaffinity)(pid, cpusetsize, mask);
4019 if (mask && !res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, mask, cpusetsize);
4020 return res;
4021 }
4022 #define INIT_SCHED_GETAFFINITY COMMON_INTERCEPT_FUNCTION(sched_getaffinity);
4023 #else
4024 #define INIT_SCHED_GETAFFINITY
4025 #endif
4027 #if SANITIZER_INTERCEPT_SCHED_GETPARAM
4028 INTERCEPTOR(int, sched_getparam, int pid, void *param) {
4029 void *ctx;
4030 COMMON_INTERCEPTOR_ENTER(ctx, sched_getparam, pid, param);
4031 int res = REAL(sched_getparam)(pid, param);
4032 if (!res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, param, struct_sched_param_sz);
4033 return res;
4034 }
4035 #define INIT_SCHED_GETPARAM COMMON_INTERCEPT_FUNCTION(sched_getparam);
4036 #else
4037 #define INIT_SCHED_GETPARAM
4038 #endif
4040 #if SANITIZER_INTERCEPT_STRERROR
4041 INTERCEPTOR(char *, strerror, int errnum) {
4042 void *ctx;
4043 COMMON_INTERCEPTOR_ENTER(ctx, strerror, errnum);
4044 COMMON_INTERCEPTOR_STRERROR();
4045 char *res = REAL(strerror)(errnum);
4046 if (res) COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, internal_strlen(res) + 1);
4047 return res;
4048 }
4049 #define INIT_STRERROR COMMON_INTERCEPT_FUNCTION(strerror);
4050 #else
4051 #define INIT_STRERROR
4052 #endif
4054 #if SANITIZER_INTERCEPT_STRERROR_R
4055 // There are 2 versions of strerror_r:
4056 // * POSIX version returns 0 on success, negative error code on failure,
4057 // writes message to buf.
4058 // * GNU version returns message pointer, which points to either buf or some
4059 // static storage.
4060 #if ((_POSIX_C_SOURCE >= 200112L || _XOPEN_SOURCE >= 600) && !_GNU_SOURCE) || \
4061 SANITIZER_APPLE || SANITIZER_ANDROID || SANITIZER_NETBSD || \
4062 SANITIZER_FREEBSD
4063 // POSIX version. Spec is not clear on whether buf is NULL-terminated.
4064 // At least on OSX, buf contents are valid even when the call fails.
4065 INTERCEPTOR(int, strerror_r, int errnum, char *buf, SIZE_T buflen) {
4066 void *ctx;
4067 COMMON_INTERCEPTOR_ENTER(ctx, strerror_r, errnum, buf, buflen);
4068 // FIXME: under ASan the call below may write to freed memory and corrupt
4069 // its metadata. See
4070 // https://github.com/google/sanitizers/issues/321.
4071 int res = REAL(strerror_r)(errnum, buf, buflen);
4073 SIZE_T sz = internal_strnlen(buf, buflen);
4074 if (sz < buflen) ++sz;
4075 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, sz);
4076 return res;
4077 }
4078 #else
4079 // GNU version.
4080 INTERCEPTOR(char *, strerror_r, int errnum, char *buf, SIZE_T buflen) {
4081 void *ctx;
4082 COMMON_INTERCEPTOR_ENTER(ctx, strerror_r, errnum, buf, buflen);
4083 // FIXME: under ASan the call below may write to freed memory and corrupt
4084 // its metadata. See
4085 // https://github.com/google/sanitizers/issues/321.
4086 char *res = REAL(strerror_r)(errnum, buf, buflen);
4087 if (res == buf)
4088 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
4089 else
4090 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, internal_strlen(res) + 1);
4091 return res;
4092 }
4093 #endif //(_POSIX_C_SOURCE >= 200112L || _XOPEN_SOURCE >= 600) && !_GNU_SOURCE ||
4094 //SANITIZER_APPLE
4095 #define INIT_STRERROR_R COMMON_INTERCEPT_FUNCTION(strerror_r);
4096 #else
4097 #define INIT_STRERROR_R
4098 #endif
4100 #if SANITIZER_INTERCEPT_XPG_STRERROR_R
4101 INTERCEPTOR(int, __xpg_strerror_r, int errnum, char *buf, SIZE_T buflen) {
4102 void *ctx;
4103 COMMON_INTERCEPTOR_ENTER(ctx, __xpg_strerror_r, errnum, buf, buflen);
4104 // FIXME: under ASan the call below may write to freed memory and corrupt
4105 // its metadata. See
4106 // https://github.com/google/sanitizers/issues/321.
4107 int res = REAL(__xpg_strerror_r)(errnum, buf, buflen);
4108 // This version always returns a null-terminated string.
4109 if (buf && buflen)
4110 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, internal_strlen(buf) + 1);
4111 return res;
4112 }
4113 #define INIT_XPG_STRERROR_R COMMON_INTERCEPT_FUNCTION(__xpg_strerror_r);
4114 #else
4115 #define INIT_XPG_STRERROR_R
4116 #endif
4118 #if SANITIZER_INTERCEPT_SCANDIR
4119 typedef int (*scandir_filter_f)(const struct __sanitizer_dirent *);
4120 typedef int (*scandir_compar_f)(const struct __sanitizer_dirent **,
4121 const struct __sanitizer_dirent **);
4123 static THREADLOCAL scandir_filter_f scandir_filter;
4124 static THREADLOCAL scandir_compar_f scandir_compar;
4126 static int wrapped_scandir_filter(const struct __sanitizer_dirent *dir) {
4127 COMMON_INTERCEPTOR_UNPOISON_PARAM(1);
4128 COMMON_INTERCEPTOR_INITIALIZE_RANGE(dir, __sanitizer_dirsiz(dir));
4129 return scandir_filter(dir);
4130 }
4132 static int wrapped_scandir_compar(const struct __sanitizer_dirent **a,
4133 const struct __sanitizer_dirent **b) {
4134 COMMON_INTERCEPTOR_UNPOISON_PARAM(2);
4135 COMMON_INTERCEPTOR_INITIALIZE_RANGE(a, sizeof(*a));
4136 COMMON_INTERCEPTOR_INITIALIZE_RANGE(*a, __sanitizer_dirsiz(*a));
4137 COMMON_INTERCEPTOR_INITIALIZE_RANGE(b, sizeof(*b));
4138 COMMON_INTERCEPTOR_INITIALIZE_RANGE(*b, __sanitizer_dirsiz(*b));
4139 return scandir_compar(a, b);
4140 }
4142 INTERCEPTOR(int, scandir, char *dirp, __sanitizer_dirent ***namelist,
4143 scandir_filter_f filter, scandir_compar_f compar) {
4144 void *ctx;
4145 COMMON_INTERCEPTOR_ENTER(ctx, scandir, dirp, namelist, filter, compar);
4146 if (dirp) COMMON_INTERCEPTOR_READ_RANGE(ctx, dirp, internal_strlen(dirp) + 1);
4147 scandir_filter = filter;
4148 scandir_compar = compar;
4149 // FIXME: under ASan the call below may write to freed memory and corrupt
4150 // its metadata. See
4151 // https://github.com/google/sanitizers/issues/321.
4152 int res = REAL(scandir)(dirp, namelist,
4153 filter ? wrapped_scandir_filter : nullptr,
4154 compar ? wrapped_scandir_compar : nullptr);
4155 scandir_filter = nullptr;
4156 scandir_compar = nullptr;
4157 if (namelist && res > 0) {
4158 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, namelist, sizeof(*namelist));
4159 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *namelist, sizeof(**namelist) * res);
4160 for (int i = 0; i < res; ++i)
4161 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, (*namelist)[i],
4162 __sanitizer_dirsiz((*namelist)[i]));
4163 }
4164 return res;
4165 }
4166 #define INIT_SCANDIR COMMON_INTERCEPT_FUNCTION(scandir);
4167 #else
4168 #define INIT_SCANDIR
4169 #endif
4171 #if SANITIZER_INTERCEPT_SCANDIR64
4172 typedef int (*scandir64_filter_f)(const struct __sanitizer_dirent64 *);
4173 typedef int (*scandir64_compar_f)(const struct __sanitizer_dirent64 **,
4174 const struct __sanitizer_dirent64 **);
4176 static THREADLOCAL scandir64_filter_f scandir64_filter;
4177 static THREADLOCAL scandir64_compar_f scandir64_compar;
4179 static int wrapped_scandir64_filter(const struct __sanitizer_dirent64 *dir) {
4180 COMMON_INTERCEPTOR_UNPOISON_PARAM(1);
4181 COMMON_INTERCEPTOR_INITIALIZE_RANGE(dir, __sanitizer_dirsiz(dir));
4182 return scandir64_filter(dir);
4183 }
4185 static int wrapped_scandir64_compar(const struct __sanitizer_dirent64 **a,
4186 const struct __sanitizer_dirent64 **b) {
4187 COMMON_INTERCEPTOR_UNPOISON_PARAM(2);
4188 COMMON_INTERCEPTOR_INITIALIZE_RANGE(a, sizeof(*a));
4189 COMMON_INTERCEPTOR_INITIALIZE_RANGE(*a, __sanitizer_dirsiz(*a));
4190 COMMON_INTERCEPTOR_INITIALIZE_RANGE(b, sizeof(*b));
4191 COMMON_INTERCEPTOR_INITIALIZE_RANGE(*b, __sanitizer_dirsiz(*b));
4192 return scandir64_compar(a, b);
4193 }
4195 INTERCEPTOR(int, scandir64, char *dirp, __sanitizer_dirent64 ***namelist,
4196 scandir64_filter_f filter, scandir64_compar_f compar) {
4197 void *ctx;
4198 COMMON_INTERCEPTOR_ENTER(ctx, scandir64, dirp, namelist, filter, compar);
4199 if (dirp) COMMON_INTERCEPTOR_READ_RANGE(ctx, dirp, internal_strlen(dirp) + 1);
4200 scandir64_filter = filter;
4201 scandir64_compar = compar;
4202 // FIXME: under ASan the call below may write to freed memory and corrupt
4203 // its metadata. See
4204 // https://github.com/google/sanitizers/issues/321.
4205 int res =
4206 REAL(scandir64)(dirp, namelist,
4207 filter ? wrapped_scandir64_filter : nullptr,
4208 compar ? wrapped_scandir64_compar : nullptr);
4209 scandir64_filter = nullptr;
4210 scandir64_compar = nullptr;
4211 if (namelist && res > 0) {
4212 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, namelist, sizeof(*namelist));
4213 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *namelist, sizeof(**namelist) * res);
4214 for (int i = 0; i < res; ++i)
4215 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, (*namelist)[i],
4216 __sanitizer_dirsiz((*namelist)[i]));
4217 }
4218 return res;
4219 }
4220 #define INIT_SCANDIR64 COMMON_INTERCEPT_FUNCTION(scandir64);
4221 #else
4222 #define INIT_SCANDIR64
4223 #endif
4225 #if SANITIZER_INTERCEPT_GETGROUPS
4226 INTERCEPTOR(int, getgroups, int size, u32 *lst) {
4227 void *ctx;
4228 COMMON_INTERCEPTOR_ENTER(ctx, getgroups, size, lst);
4229 // FIXME: under ASan the call below may write to freed memory and corrupt
4230 // its metadata. See
4231 // https://github.com/google/sanitizers/issues/321.
4232 int res = REAL(getgroups)(size, lst);
4233 if (res >= 0 && lst && size > 0)
4234 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, lst, res * sizeof(*lst));
4235 return res;
4236 }
4237 #define INIT_GETGROUPS COMMON_INTERCEPT_FUNCTION(getgroups);
4238 #else
4239 #define INIT_GETGROUPS
4240 #endif
4242 #if SANITIZER_INTERCEPT_POLL
4243 static void read_pollfd(void *ctx, __sanitizer_pollfd *fds,
4244 __sanitizer_nfds_t nfds) {
4245 for (unsigned i = 0; i < nfds; ++i) {
4246 COMMON_INTERCEPTOR_READ_RANGE(ctx, &fds[i].fd, sizeof(fds[i].fd));
4247 COMMON_INTERCEPTOR_READ_RANGE(ctx, &fds[i].events, sizeof(fds[i].events));
4248 }
4249 }
4251 static void write_pollfd(void *ctx, __sanitizer_pollfd *fds,
4252 __sanitizer_nfds_t nfds) {
4253 for (unsigned i = 0; i < nfds; ++i)
4254 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, &fds[i].revents,
4255 sizeof(fds[i].revents));
4256 }
4258 INTERCEPTOR(int, poll, __sanitizer_pollfd *fds, __sanitizer_nfds_t nfds,
4259 int timeout) {
4260 void *ctx;
4261 COMMON_INTERCEPTOR_ENTER(ctx, poll, fds, nfds, timeout);
4262 if (fds && nfds) read_pollfd(ctx, fds, nfds);
4263 int res = COMMON_INTERCEPTOR_BLOCK_REAL(poll)(fds, nfds, timeout);
4264 if (fds && nfds) write_pollfd(ctx, fds, nfds);
4265 return res;
4266 }
4267 #define INIT_POLL COMMON_INTERCEPT_FUNCTION(poll);
4268 #else
4269 #define INIT_POLL
4270 #endif
4272 #if SANITIZER_INTERCEPT_PPOLL
4273 INTERCEPTOR(int, ppoll, __sanitizer_pollfd *fds, __sanitizer_nfds_t nfds,
4274 void *timeout_ts, __sanitizer_sigset_t *sigmask) {
4275 void *ctx;
4276 COMMON_INTERCEPTOR_ENTER(ctx, ppoll, fds, nfds, timeout_ts, sigmask);
4277 if (fds && nfds) read_pollfd(ctx, fds, nfds);
4278 if (timeout_ts)
4279 COMMON_INTERCEPTOR_READ_RANGE(ctx, timeout_ts, struct_timespec_sz);
4280 if (sigmask) COMMON_INTERCEPTOR_READ_RANGE(ctx, sigmask, sizeof(*sigmask));
4281 int res =
4282 COMMON_INTERCEPTOR_BLOCK_REAL(ppoll)(fds, nfds, timeout_ts, sigmask);
4283 if (fds && nfds) write_pollfd(ctx, fds, nfds);
4284 return res;
4285 }
4286 #define INIT_PPOLL COMMON_INTERCEPT_FUNCTION(ppoll);
4287 #else
4288 #define INIT_PPOLL
4289 #endif
4291 #if SANITIZER_INTERCEPT_WORDEXP
4292 INTERCEPTOR(int, wordexp, char *s, __sanitizer_wordexp_t *p, int flags) {
4293 void *ctx;
4294 COMMON_INTERCEPTOR_ENTER(ctx, wordexp, s, p, flags);
4295 if (s) COMMON_INTERCEPTOR_READ_RANGE(ctx, s, internal_strlen(s) + 1);
4296 // FIXME: under ASan the call below may write to freed memory and corrupt
4297 // its metadata. See
4298 // https://github.com/google/sanitizers/issues/321.
4299 int res = REAL(wordexp)(s, p, flags);
4300 if (!res && p) {
4301 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p, sizeof(*p));
4302 uptr we_wordc =
4303 ((flags & wordexp_wrde_dooffs) ? p->we_offs : 0) + p->we_wordc;
4304 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p->we_wordv,
4305 sizeof(*p->we_wordv) * (we_wordc + 1));
4306 for (uptr i = 0; i < we_wordc; ++i) {
4307 char *w = p->we_wordv[i];
4308 if (w) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, w, internal_strlen(w) + 1);
4309 }
4310 }
4311 return res;
4312 }
4313 #define INIT_WORDEXP COMMON_INTERCEPT_FUNCTION(wordexp);
4314 #else
4315 #define INIT_WORDEXP
4316 #endif
4318 #if SANITIZER_INTERCEPT_SIGWAIT
4319 INTERCEPTOR(int, sigwait, __sanitizer_sigset_t *set, int *sig) {
4320 void *ctx;
4321 COMMON_INTERCEPTOR_ENTER(ctx, sigwait, set, sig);
4322 if (set) COMMON_INTERCEPTOR_READ_RANGE(ctx, set, sizeof(*set));
4323 // FIXME: under ASan the call below may write to freed memory and corrupt
4324 // its metadata. See
4325 // https://github.com/google/sanitizers/issues/321.
4326 int res = COMMON_INTERCEPTOR_BLOCK_REAL(sigwait)(set, sig);
4327 if (!res && sig) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, sig, sizeof(*sig));
4328 return res;
4329 }
4330 #define INIT_SIGWAIT COMMON_INTERCEPT_FUNCTION(sigwait);
4331 #else
4332 #define INIT_SIGWAIT
4333 #endif
4335 #if SANITIZER_INTERCEPT_SIGWAITINFO
4336 INTERCEPTOR(int, sigwaitinfo, __sanitizer_sigset_t *set, void *info) {
4337 void *ctx;
4338 COMMON_INTERCEPTOR_ENTER(ctx, sigwaitinfo, set, info);
4339 if (set) COMMON_INTERCEPTOR_READ_RANGE(ctx, set, sizeof(*set));
4340 // FIXME: under ASan the call below may write to freed memory and corrupt
4341 // its metadata. See
4342 // https://github.com/google/sanitizers/issues/321.
4343 int res = COMMON_INTERCEPTOR_BLOCK_REAL(sigwaitinfo)(set, info);
4344 if (res > 0 && info) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, info, siginfo_t_sz);
4345 return res;
4346 }
4347 #define INIT_SIGWAITINFO COMMON_INTERCEPT_FUNCTION(sigwaitinfo);
4348 #else
4349 #define INIT_SIGWAITINFO
4350 #endif
4352 #if SANITIZER_INTERCEPT_SIGTIMEDWAIT
4353 INTERCEPTOR(int, sigtimedwait, __sanitizer_sigset_t *set, void *info,
4354 void *timeout) {
4355 void *ctx;
4356 COMMON_INTERCEPTOR_ENTER(ctx, sigtimedwait, set, info, timeout);
4357 if (timeout) COMMON_INTERCEPTOR_READ_RANGE(ctx, timeout, struct_timespec_sz);
4358 if (set) COMMON_INTERCEPTOR_READ_RANGE(ctx, set, sizeof(*set));
4359 // FIXME: under ASan the call below may write to freed memory and corrupt
4360 // its metadata. See
4361 // https://github.com/google/sanitizers/issues/321.
4362 int res = COMMON_INTERCEPTOR_BLOCK_REAL(sigtimedwait)(set, info, timeout);
4363 if (res > 0 && info) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, info, siginfo_t_sz);
4364 return res;
4365 }
4366 #define INIT_SIGTIMEDWAIT COMMON_INTERCEPT_FUNCTION(sigtimedwait);
4367 #else
4368 #define INIT_SIGTIMEDWAIT
4369 #endif
4371 #if SANITIZER_INTERCEPT_SIGSETOPS
4372 INTERCEPTOR(int, sigemptyset, __sanitizer_sigset_t *set) {
4373 void *ctx;
4374 COMMON_INTERCEPTOR_ENTER(ctx, sigemptyset, set);
4375 // FIXME: under ASan the call below may write to freed memory and corrupt
4376 // its metadata. See
4377 // https://github.com/google/sanitizers/issues/321.
4378 int res = REAL(sigemptyset)(set);
4379 if (!res && set) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, set, sizeof(*set));
4380 return res;
4381 }
4383 INTERCEPTOR(int, sigfillset, __sanitizer_sigset_t *set) {
4384 void *ctx;
4385 COMMON_INTERCEPTOR_ENTER(ctx, sigfillset, set);
4386 // FIXME: under ASan the call below may write to freed memory and corrupt
4387 // its metadata. See
4388 // https://github.com/google/sanitizers/issues/321.
4389 int res = REAL(sigfillset)(set);
4390 if (!res && set) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, set, sizeof(*set));
4391 return res;
4392 }
4393 #define INIT_SIGSETOPS \
4394 COMMON_INTERCEPT_FUNCTION(sigemptyset); \
4395 COMMON_INTERCEPT_FUNCTION(sigfillset);
4396 #else
4397 #define INIT_SIGSETOPS
4398 #endif
4400 #if SANITIZER_INTERCEPT_SIGSET_LOGICOPS
4401 INTERCEPTOR(int, sigandset, __sanitizer_sigset_t *dst,
4402 __sanitizer_sigset_t *src1, __sanitizer_sigset_t *src2) {
4403 void *ctx;
4404 COMMON_INTERCEPTOR_ENTER(ctx, sigandset, dst, src1, src2);
4405 if (src1)
4406 COMMON_INTERCEPTOR_READ_RANGE(ctx, src1, sizeof(*src1));
4407 if (src2)
4408 COMMON_INTERCEPTOR_READ_RANGE(ctx, src2, sizeof(*src2));
4409 int res = REAL(sigandset)(dst, src1, src2);
4410 if (!res && dst)
4411 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, sizeof(*dst));
4412 return res;
4413 }
4415 INTERCEPTOR(int, sigorset, __sanitizer_sigset_t *dst,
4416 __sanitizer_sigset_t *src1, __sanitizer_sigset_t *src2) {
4417 void *ctx;
4418 COMMON_INTERCEPTOR_ENTER(ctx, sigorset, dst, src1, src2);
4419 if (src1)
4420 COMMON_INTERCEPTOR_READ_RANGE(ctx, src1, sizeof(*src1));
4421 if (src2)
4422 COMMON_INTERCEPTOR_READ_RANGE(ctx, src2, sizeof(*src2));
4423 int res = REAL(sigorset)(dst, src1, src2);
4424 if (!res && dst)
4425 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, sizeof(*dst));
4426 return res;
4427 }
4428 #define INIT_SIGSET_LOGICOPS \
4429 COMMON_INTERCEPT_FUNCTION(sigandset); \
4430 COMMON_INTERCEPT_FUNCTION(sigorset);
4431 #else
4432 #define INIT_SIGSET_LOGICOPS
4433 #endif
4435 #if SANITIZER_INTERCEPT_SIGPENDING
4436 INTERCEPTOR(int, sigpending, __sanitizer_sigset_t *set) {
4437 void *ctx;
4438 COMMON_INTERCEPTOR_ENTER(ctx, sigpending, set);
4439 // FIXME: under ASan the call below may write to freed memory and corrupt
4440 // its metadata. See
4441 // https://github.com/google/sanitizers/issues/321.
4442 int res = REAL(sigpending)(set);
4443 if (!res && set) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, set, sizeof(*set));
4444 return res;
4445 }
4446 #define INIT_SIGPENDING COMMON_INTERCEPT_FUNCTION(sigpending);
4447 #else
4448 #define INIT_SIGPENDING
4449 #endif
4451 #if SANITIZER_INTERCEPT_SIGPROCMASK
4452 INTERCEPTOR(int, sigprocmask, int how, __sanitizer_sigset_t *set,
4453 __sanitizer_sigset_t *oldset) {
4454 void *ctx;
4455 COMMON_INTERCEPTOR_ENTER(ctx, sigprocmask, how, set, oldset);
4456 if (set) COMMON_INTERCEPTOR_READ_RANGE(ctx, set, sizeof(*set));
4457 // FIXME: under ASan the call below may write to freed memory and corrupt
4458 // its metadata. See
4459 // https://github.com/google/sanitizers/issues/321.
4460 int res = REAL(sigprocmask)(how, set, oldset);
4461 if (!res && oldset)
4462 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, oldset, sizeof(*oldset));
4463 return res;
4464 }
4465 #define INIT_SIGPROCMASK COMMON_INTERCEPT_FUNCTION(sigprocmask);
4466 #else
4467 #define INIT_SIGPROCMASK
4468 #endif
4470 #if SANITIZER_INTERCEPT_PTHREAD_SIGMASK
4471 INTERCEPTOR(int, pthread_sigmask, int how, __sanitizer_sigset_t *set,
4472 __sanitizer_sigset_t *oldset) {
4473 void *ctx;
4474 COMMON_INTERCEPTOR_ENTER(ctx, pthread_sigmask, how, set, oldset);
4475 if (set) COMMON_INTERCEPTOR_READ_RANGE(ctx, set, sizeof(*set));
4476 // FIXME: under ASan the call below may write to freed memory and corrupt
4477 // its metadata. See
4478 // https://github.com/google/sanitizers/issues/321.
4479 int res = REAL(pthread_sigmask)(how, set, oldset);
4480 if (!res && oldset)
4481 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, oldset, sizeof(*oldset));
4482 return res;
4483 }
4484 #define INIT_PTHREAD_SIGMASK COMMON_INTERCEPT_FUNCTION(pthread_sigmask);
4485 #else
4486 #define INIT_PTHREAD_SIGMASK
4487 #endif
4489 #if SANITIZER_INTERCEPT_BACKTRACE
4490 INTERCEPTOR(int, backtrace, void **buffer, int size) {
4491 void *ctx;
4492 COMMON_INTERCEPTOR_ENTER(ctx, backtrace, buffer, size);
4493 // 'buffer' might be freed memory, hence it is unsafe to directly call
4494 // REAL(backtrace)(buffer, size). Instead, we use our own known-good
4495 // scratch buffer.
4496 void **scratch = (void**)InternalAlloc(sizeof(void*) * size);
4497 int res = REAL(backtrace)(scratch, size);
4498 if (res && buffer) {
4499 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buffer, res * sizeof(*buffer));
4500 internal_memcpy(buffer, scratch, res * sizeof(*buffer));
4501 }
4502 InternalFree(scratch);
4503 return res;
4504 }
4506 INTERCEPTOR(char **, backtrace_symbols, void **buffer, int size) {
4507 void *ctx;
4508 COMMON_INTERCEPTOR_ENTER(ctx, backtrace_symbols, buffer, size);
4509 if (buffer && size)
4510 COMMON_INTERCEPTOR_READ_RANGE(ctx, buffer, size * sizeof(*buffer));
4511 // The COMMON_INTERCEPTOR_READ_RANGE above ensures that 'buffer' is
4512 // valid for reading.
4513 char **res = REAL(backtrace_symbols)(buffer, size);
4514 if (res && size) {
4515 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, size * sizeof(*res));
4516 for (int i = 0; i < size; ++i)
4517 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res[i], internal_strlen(res[i]) + 1);
4518 }
4519 return res;
4520 }
4521 #define INIT_BACKTRACE \
4522 COMMON_INTERCEPT_FUNCTION(backtrace); \
4523 COMMON_INTERCEPT_FUNCTION(backtrace_symbols);
4524 #else
4525 #define INIT_BACKTRACE
4526 #endif
4528 #if SANITIZER_INTERCEPT__EXIT
4529 INTERCEPTOR(void, _exit, int status) {
4530 void *ctx;
4531 COMMON_INTERCEPTOR_ENTER(ctx, _exit, status);
4532 COMMON_INTERCEPTOR_USER_CALLBACK_START();
4533 int status1 = COMMON_INTERCEPTOR_ON_EXIT(ctx);
4534 COMMON_INTERCEPTOR_USER_CALLBACK_END();
4535 if (status == 0) status = status1;
4536 REAL(_exit)(status);
4537 }
4538 #define INIT__EXIT COMMON_INTERCEPT_FUNCTION(_exit);
4539 #else
4540 #define INIT__EXIT
4541 #endif
4543 #if SANITIZER_INTERCEPT___LIBC_MUTEX
4544 INTERCEPTOR(int, __libc_thr_setcancelstate, int state, int *oldstate)
4545 ALIAS(WRAP(pthread_setcancelstate));
4547 #define INIT___LIBC_THR_SETCANCELSTATE \
4548 COMMON_INTERCEPT_FUNCTION(__libc_thr_setcancelstate)
4549 #else
4550 #define INIT___LIBC_THR_SETCANCELSTATE
4551 #endif
4553 #if SANITIZER_INTERCEPT_GETMNTENT || SANITIZER_INTERCEPT_GETMNTENT_R
4554 static void write_mntent(void *ctx, __sanitizer_mntent *mnt) {
4555 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, mnt, sizeof(*mnt));
4556 if (mnt->mnt_fsname)
4557 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, mnt->mnt_fsname,
4558 internal_strlen(mnt->mnt_fsname) + 1);
4559 if (mnt->mnt_dir)
4560 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, mnt->mnt_dir,
4561 internal_strlen(mnt->mnt_dir) + 1);
4562 if (mnt->mnt_type)
4563 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, mnt->mnt_type,
4564 internal_strlen(mnt->mnt_type) + 1);
4565 if (mnt->mnt_opts)
4566 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, mnt->mnt_opts,
4567 internal_strlen(mnt->mnt_opts) + 1);
4568 }
4569 #endif
4571 #if SANITIZER_INTERCEPT_GETMNTENT
4572 INTERCEPTOR(__sanitizer_mntent *, getmntent, void *fp) {
4573 void *ctx;
4574 COMMON_INTERCEPTOR_ENTER(ctx, getmntent, fp);
4575 __sanitizer_mntent *res = REAL(getmntent)(fp);
4576 if (res) write_mntent(ctx, res);
4577 return res;
4578 }
4579 #define INIT_GETMNTENT COMMON_INTERCEPT_FUNCTION(getmntent);
4580 #else
4581 #define INIT_GETMNTENT
4582 #endif
4584 #if SANITIZER_INTERCEPT_GETMNTENT_R
4585 INTERCEPTOR(__sanitizer_mntent *, getmntent_r, void *fp,
4586 __sanitizer_mntent *mntbuf, char *buf, int buflen) {
4587 void *ctx;
4588 COMMON_INTERCEPTOR_ENTER(ctx, getmntent_r, fp, mntbuf, buf, buflen);
4589 __sanitizer_mntent *res = REAL(getmntent_r)(fp, mntbuf, buf, buflen);
4590 if (res) write_mntent(ctx, res);
4591 return res;
4592 }
4593 #define INIT_GETMNTENT_R COMMON_INTERCEPT_FUNCTION(getmntent_r);
4594 #else
4595 #define INIT_GETMNTENT_R
4596 #endif
4598 #if SANITIZER_INTERCEPT_STATFS
4599 INTERCEPTOR(int, statfs, char *path, void *buf) {
4600 void *ctx;
4601 COMMON_INTERCEPTOR_ENTER(ctx, statfs, path, buf);
4602 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
4603 // FIXME: under ASan the call below may write to freed memory and corrupt
4604 // its metadata. See
4605 // https://github.com/google/sanitizers/issues/321.
4606 int res = REAL(statfs)(path, buf);
4607 if (!res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, struct_statfs_sz);
4608 return res;
4609 }
4610 INTERCEPTOR(int, fstatfs, int fd, void *buf) {
4611 void *ctx;
4612 COMMON_INTERCEPTOR_ENTER(ctx, fstatfs, fd, buf);
4613 // FIXME: under ASan the call below may write to freed memory and corrupt
4614 // its metadata. See
4615 // https://github.com/google/sanitizers/issues/321.
4616 int res = REAL(fstatfs)(fd, buf);
4617 if (!res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, struct_statfs_sz);
4618 return res;
4619 }
4620 #define INIT_STATFS \
4621 COMMON_INTERCEPT_FUNCTION(statfs); \
4622 COMMON_INTERCEPT_FUNCTION(fstatfs);
4623 #else
4624 #define INIT_STATFS
4625 #endif
4627 #if SANITIZER_INTERCEPT_STATFS64
4628 INTERCEPTOR(int, statfs64, char *path, void *buf) {
4629 void *ctx;
4630 COMMON_INTERCEPTOR_ENTER(ctx, statfs64, path, buf);
4631 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
4632 // FIXME: under ASan the call below may write to freed memory and corrupt
4633 // its metadata. See
4634 // https://github.com/google/sanitizers/issues/321.
4635 int res = REAL(statfs64)(path, buf);
4636 if (!res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, struct_statfs64_sz);
4637 return res;
4638 }
4639 INTERCEPTOR(int, fstatfs64, int fd, void *buf) {
4640 void *ctx;
4641 COMMON_INTERCEPTOR_ENTER(ctx, fstatfs64, fd, buf);
4642 // FIXME: under ASan the call below may write to freed memory and corrupt
4643 // its metadata. See
4644 // https://github.com/google/sanitizers/issues/321.
4645 int res = REAL(fstatfs64)(fd, buf);
4646 if (!res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, struct_statfs64_sz);
4647 return res;
4648 }
4649 #define INIT_STATFS64 \
4650 COMMON_INTERCEPT_FUNCTION(statfs64); \
4651 COMMON_INTERCEPT_FUNCTION(fstatfs64);
4652 #else
4653 #define INIT_STATFS64
4654 #endif
4656 #if SANITIZER_INTERCEPT_STATVFS
4657 INTERCEPTOR(int, statvfs, char *path, void *buf) {
4658 void *ctx;
4659 COMMON_INTERCEPTOR_ENTER(ctx, statvfs, path, buf);
4660 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
4661 // FIXME: under ASan the call below may write to freed memory and corrupt
4662 // its metadata. See
4663 // https://github.com/google/sanitizers/issues/321.
4664 int res = REAL(statvfs)(path, buf);
4665 if (!res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, struct_statvfs_sz);
4666 return res;
4667 }
4668 INTERCEPTOR(int, fstatvfs, int fd, void *buf) {
4669 void *ctx;
4670 COMMON_INTERCEPTOR_ENTER(ctx, fstatvfs, fd, buf);
4671 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
4672 // FIXME: under ASan the call below may write to freed memory and corrupt
4673 // its metadata. See
4674 // https://github.com/google/sanitizers/issues/321.
4675 int res = REAL(fstatvfs)(fd, buf);
4676 if (!res) {
4677 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, struct_statvfs_sz);
4678 if (fd >= 0)
4679 COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
4680 }
4681 return res;
4682 }
4683 #define INIT_STATVFS \
4684 COMMON_INTERCEPT_FUNCTION(statvfs); \
4685 COMMON_INTERCEPT_FUNCTION(fstatvfs);
4686 #else
4687 #define INIT_STATVFS
4688 #endif
4690 #if SANITIZER_INTERCEPT_STATVFS64
4691 INTERCEPTOR(int, statvfs64, char *path, void *buf) {
4692 void *ctx;
4693 COMMON_INTERCEPTOR_ENTER(ctx, statvfs64, path, buf);
4694 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
4695 // FIXME: under ASan the call below may write to freed memory and corrupt
4696 // its metadata. See
4697 // https://github.com/google/sanitizers/issues/321.
4698 int res = REAL(statvfs64)(path, buf);
4699 if (!res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, struct_statvfs64_sz);
4700 return res;
4701 }
4702 INTERCEPTOR(int, fstatvfs64, int fd, void *buf) {
4703 void *ctx;
4704 COMMON_INTERCEPTOR_ENTER(ctx, fstatvfs64, fd, buf);
4705 // FIXME: under ASan the call below may write to freed memory and corrupt
4706 // its metadata. See
4707 // https://github.com/google/sanitizers/issues/321.
4708 int res = REAL(fstatvfs64)(fd, buf);
4709 if (!res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, struct_statvfs64_sz);
4710 return res;
4711 }
4712 #define INIT_STATVFS64 \
4713 COMMON_INTERCEPT_FUNCTION(statvfs64); \
4714 COMMON_INTERCEPT_FUNCTION(fstatvfs64);
4715 #else
4716 #define INIT_STATVFS64
4717 #endif
4719 #if SANITIZER_INTERCEPT_INITGROUPS
4720 INTERCEPTOR(int, initgroups, char *user, u32 group) {
4721 void *ctx;
4722 COMMON_INTERCEPTOR_ENTER(ctx, initgroups, user, group);
4723 if (user) COMMON_INTERCEPTOR_READ_RANGE(ctx, user, internal_strlen(user) + 1);
4724 int res = REAL(initgroups)(user, group);
4725 return res;
4726 }
4727 #define INIT_INITGROUPS COMMON_INTERCEPT_FUNCTION(initgroups);
4728 #else
4729 #define INIT_INITGROUPS
4730 #endif
4732 #if SANITIZER_INTERCEPT_ETHER_NTOA_ATON
4733 INTERCEPTOR(char *, ether_ntoa, __sanitizer_ether_addr *addr) {
4734 void *ctx;
4735 COMMON_INTERCEPTOR_ENTER(ctx, ether_ntoa, addr);
4736 if (addr) COMMON_INTERCEPTOR_READ_RANGE(ctx, addr, sizeof(*addr));
4737 char *res = REAL(ether_ntoa)(addr);
4738 if (res) COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, internal_strlen(res) + 1);
4739 return res;
4740 }
4741 INTERCEPTOR(__sanitizer_ether_addr *, ether_aton, char *buf) {
4742 void *ctx;
4743 COMMON_INTERCEPTOR_ENTER(ctx, ether_aton, buf);
4744 if (buf) COMMON_INTERCEPTOR_READ_RANGE(ctx, buf, internal_strlen(buf) + 1);
4745 __sanitizer_ether_addr *res = REAL(ether_aton)(buf);
4746 if (res) COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, sizeof(*res));
4747 return res;
4748 }
4749 #define INIT_ETHER_NTOA_ATON \
4750 COMMON_INTERCEPT_FUNCTION(ether_ntoa); \
4751 COMMON_INTERCEPT_FUNCTION(ether_aton);
4752 #else
4753 #define INIT_ETHER_NTOA_ATON
4754 #endif
4756 #if SANITIZER_INTERCEPT_ETHER_HOST
4757 INTERCEPTOR(int, ether_ntohost, char *hostname, __sanitizer_ether_addr *addr) {
4758 void *ctx;
4759 COMMON_INTERCEPTOR_ENTER(ctx, ether_ntohost, hostname, addr);
4760 if (addr) COMMON_INTERCEPTOR_READ_RANGE(ctx, addr, sizeof(*addr));
4761 // FIXME: under ASan the call below may write to freed memory and corrupt
4762 // its metadata. See
4763 // https://github.com/google/sanitizers/issues/321.
4764 int res = REAL(ether_ntohost)(hostname, addr);
4765 if (!res && hostname)
4766 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, hostname, internal_strlen(hostname) + 1);
4767 return res;
4768 }
4769 INTERCEPTOR(int, ether_hostton, char *hostname, __sanitizer_ether_addr *addr) {
4770 void *ctx;
4771 COMMON_INTERCEPTOR_ENTER(ctx, ether_hostton, hostname, addr);
4772 if (hostname)
4773 COMMON_INTERCEPTOR_READ_RANGE(ctx, hostname, internal_strlen(hostname) + 1);
4774 // FIXME: under ASan the call below may write to freed memory and corrupt
4775 // its metadata. See
4776 // https://github.com/google/sanitizers/issues/321.
4777 int res = REAL(ether_hostton)(hostname, addr);
4778 if (!res && addr) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, addr, sizeof(*addr));
4779 return res;
4780 }
4781 INTERCEPTOR(int, ether_line, char *line, __sanitizer_ether_addr *addr,
4782 char *hostname) {
4783 void *ctx;
4784 COMMON_INTERCEPTOR_ENTER(ctx, ether_line, line, addr, hostname);
4785 if (line) COMMON_INTERCEPTOR_READ_RANGE(ctx, line, internal_strlen(line) + 1);
4786 // FIXME: under ASan the call below may write to freed memory and corrupt
4787 // its metadata. See
4788 // https://github.com/google/sanitizers/issues/321.
4789 int res = REAL(ether_line)(line, addr, hostname);
4790 if (!res) {
4791 if (addr) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, addr, sizeof(*addr));
4792 if (hostname)
4793 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, hostname, internal_strlen(hostname) + 1);
4794 }
4795 return res;
4796 }
4797 #define INIT_ETHER_HOST \
4798 COMMON_INTERCEPT_FUNCTION(ether_ntohost); \
4799 COMMON_INTERCEPT_FUNCTION(ether_hostton); \
4800 COMMON_INTERCEPT_FUNCTION(ether_line);
4801 #else
4802 #define INIT_ETHER_HOST
4803 #endif
4805 #if SANITIZER_INTERCEPT_ETHER_R
4806 INTERCEPTOR(char *, ether_ntoa_r, __sanitizer_ether_addr *addr, char *buf) {
4807 void *ctx;
4808 COMMON_INTERCEPTOR_ENTER(ctx, ether_ntoa_r, addr, buf);
4809 if (addr) COMMON_INTERCEPTOR_READ_RANGE(ctx, addr, sizeof(*addr));
4810 // FIXME: under ASan the call below may write to freed memory and corrupt
4811 // its metadata. See
4812 // https://github.com/google/sanitizers/issues/321.
4813 char *res = REAL(ether_ntoa_r)(addr, buf);
4814 if (res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
4815 return res;
4816 }
4817 INTERCEPTOR(__sanitizer_ether_addr *, ether_aton_r, char *buf,
4818 __sanitizer_ether_addr *addr) {
4819 void *ctx;
4820 COMMON_INTERCEPTOR_ENTER(ctx, ether_aton_r, buf, addr);
4821 if (buf) COMMON_INTERCEPTOR_READ_RANGE(ctx, buf, internal_strlen(buf) + 1);
4822 // FIXME: under ASan the call below may write to freed memory and corrupt
4823 // its metadata. See
4824 // https://github.com/google/sanitizers/issues/321.
4825 __sanitizer_ether_addr *res = REAL(ether_aton_r)(buf, addr);
4826 if (res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, sizeof(*res));
4827 return res;
4828 }
4829 #define INIT_ETHER_R \
4830 COMMON_INTERCEPT_FUNCTION(ether_ntoa_r); \
4831 COMMON_INTERCEPT_FUNCTION(ether_aton_r);
4832 #else
4833 #define INIT_ETHER_R
4834 #endif
4836 #if SANITIZER_INTERCEPT_SHMCTL
4837 INTERCEPTOR(int, shmctl, int shmid, int cmd, void *buf) {
4838 void *ctx;
4839 COMMON_INTERCEPTOR_ENTER(ctx, shmctl, shmid, cmd, buf);
4840 // FIXME: under ASan the call below may write to freed memory and corrupt
4841 // its metadata. See
4842 // https://github.com/google/sanitizers/issues/321.
4843 int res = REAL(shmctl)(shmid, cmd, buf);
4844 if (res >= 0) {
4845 unsigned sz = 0;
4846 if (cmd == shmctl_ipc_stat || cmd == shmctl_shm_stat)
4847 sz = sizeof(__sanitizer_shmid_ds);
4848 else if (cmd == shmctl_ipc_info)
4849 sz = struct_shminfo_sz;
4850 else if (cmd == shmctl_shm_info)
4851 sz = struct_shm_info_sz;
4852 if (sz) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, sz);
4853 }
4854 return res;
4855 }
4856 #define INIT_SHMCTL COMMON_INTERCEPT_FUNCTION(shmctl);
4857 #else
4858 #define INIT_SHMCTL
4859 #endif
4861 #if SANITIZER_INTERCEPT_RANDOM_R
4862 INTERCEPTOR(int, random_r, void *buf, u32 *result) {
4863 void *ctx;
4864 COMMON_INTERCEPTOR_ENTER(ctx, random_r, buf, result);
4865 // FIXME: under ASan the call below may write to freed memory and corrupt
4866 // its metadata. See
4867 // https://github.com/google/sanitizers/issues/321.
4868 int res = REAL(random_r)(buf, result);
4869 if (!res && result)
4870 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
4871 return res;
4872 }
4873 #define INIT_RANDOM_R COMMON_INTERCEPT_FUNCTION(random_r);
4874 #else
4875 #define INIT_RANDOM_R
4876 #endif
4878 // FIXME: under ASan the REAL() call below may write to freed memory and corrupt
4879 // its metadata. See
4880 // https://github.com/google/sanitizers/issues/321.
4881 #if SANITIZER_INTERCEPT_PTHREAD_ATTR_GET || \
4882 SANITIZER_INTERCEPT_PTHREAD_ATTR_GET_SCHED || \
4883 SANITIZER_INTERCEPT_PTHREAD_ATTR_GETINHERITSSCHED || \
4884 SANITIZER_INTERCEPT_PTHREAD_MUTEXATTR_GET || \
4885 SANITIZER_INTERCEPT_PTHREAD_RWLOCKATTR_GET || \
4886 SANITIZER_INTERCEPT_PTHREAD_CONDATTR_GET || \
4887 SANITIZER_INTERCEPT_PTHREAD_BARRIERATTR_GET
4888 #define INTERCEPTOR_PTHREAD_OBJECT_ATTR_GET(fn, sz) \
4889 INTERCEPTOR(int, fn, void *attr, void *r) { \
4890 void *ctx; \
4891 COMMON_INTERCEPTOR_ENTER(ctx, fn, attr, r); \
4892 int res = REAL(fn)(attr, r); \
4893 if (!res && r) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, r, sz); \
4894 return res; \
4895 }
4896 #define INTERCEPTOR_PTHREAD_ATTR_GET(what, sz) \
4897 INTERCEPTOR_PTHREAD_OBJECT_ATTR_GET(pthread_attr_get##what, sz)
4898 #define INTERCEPTOR_PTHREAD_MUTEXATTR_GET(what, sz) \
4899 INTERCEPTOR_PTHREAD_OBJECT_ATTR_GET(pthread_mutexattr_get##what, sz)
4900 #define INTERCEPTOR_PTHREAD_RWLOCKATTR_GET(what, sz) \
4901 INTERCEPTOR_PTHREAD_OBJECT_ATTR_GET(pthread_rwlockattr_get##what, sz)
4902 #define INTERCEPTOR_PTHREAD_CONDATTR_GET(what, sz) \
4903 INTERCEPTOR_PTHREAD_OBJECT_ATTR_GET(pthread_condattr_get##what, sz)
4904 #define INTERCEPTOR_PTHREAD_BARRIERATTR_GET(what, sz) \
4905 INTERCEPTOR_PTHREAD_OBJECT_ATTR_GET(pthread_barrierattr_get##what, sz)
4906 #endif
4908 #if SANITIZER_INTERCEPT_PTHREAD_ATTR_GET
4909 INTERCEPTOR_PTHREAD_ATTR_GET(detachstate, sizeof(int))
4910 INTERCEPTOR_PTHREAD_ATTR_GET(guardsize, sizeof(SIZE_T))
4911 INTERCEPTOR_PTHREAD_ATTR_GET(scope, sizeof(int))
4912 INTERCEPTOR_PTHREAD_ATTR_GET(stacksize, sizeof(SIZE_T))
4913 INTERCEPTOR(int, pthread_attr_getstack, void *attr, void **addr, SIZE_T *size) {
4914 void *ctx;
4915 COMMON_INTERCEPTOR_ENTER(ctx, pthread_attr_getstack, attr, addr, size);
4916 // FIXME: under ASan the call below may write to freed memory and corrupt
4917 // its metadata. See
4918 // https://github.com/google/sanitizers/issues/321.
4919 int res = REAL(pthread_attr_getstack)(attr, addr, size);
4920 if (!res) {
4921 if (addr) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, addr, sizeof(*addr));
4922 if (size) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, size, sizeof(*size));
4923 }
4924 return res;
4925 }
4927 // We may need to call the real pthread_attr_getstack from the run-time
4928 // in sanitizer_common, but we don't want to include the interception headers
4929 // there. So, just define this function here.
4930 namespace __sanitizer {
4931 extern "C" {
4932 int real_pthread_attr_getstack(void *attr, void **addr, SIZE_T *size) {
4933 return REAL(pthread_attr_getstack)(attr, addr, size);
4934 }
4935 } // extern "C"
4936 } // namespace __sanitizer
4938 #define INIT_PTHREAD_ATTR_GET \
4939 COMMON_INTERCEPT_FUNCTION(pthread_attr_getdetachstate); \
4940 COMMON_INTERCEPT_FUNCTION(pthread_attr_getguardsize); \
4941 COMMON_INTERCEPT_FUNCTION(pthread_attr_getscope); \
4942 COMMON_INTERCEPT_FUNCTION(pthread_attr_getstacksize); \
4943 COMMON_INTERCEPT_FUNCTION(pthread_attr_getstack);
4944 #else
4945 #define INIT_PTHREAD_ATTR_GET
4946 #endif
4948 #if SANITIZER_INTERCEPT_PTHREAD_ATTR_GET_SCHED
4949 INTERCEPTOR_PTHREAD_ATTR_GET(schedparam, struct_sched_param_sz)
4950 INTERCEPTOR_PTHREAD_ATTR_GET(schedpolicy, sizeof(int))
4952 #define INIT_PTHREAD_ATTR_GET_SCHED \
4953 COMMON_INTERCEPT_FUNCTION(pthread_attr_getschedparam); \
4954 COMMON_INTERCEPT_FUNCTION(pthread_attr_getschedpolicy);
4955 #else
4956 #define INIT_PTHREAD_ATTR_GET_SCHED
4957 #endif
4959 #if SANITIZER_INTERCEPT_PTHREAD_ATTR_GETINHERITSCHED
4960 INTERCEPTOR_PTHREAD_ATTR_GET(inheritsched, sizeof(int))
4962 #define INIT_PTHREAD_ATTR_GETINHERITSCHED \
4963 COMMON_INTERCEPT_FUNCTION(pthread_attr_getinheritsched);
4964 #else
4965 #define INIT_PTHREAD_ATTR_GETINHERITSCHED
4966 #endif
4968 #if SANITIZER_INTERCEPT_PTHREAD_ATTR_GETAFFINITY_NP
4969 INTERCEPTOR(int, pthread_attr_getaffinity_np, void *attr, SIZE_T cpusetsize,
4970 void *cpuset) {
4971 void *ctx;
4972 COMMON_INTERCEPTOR_ENTER(ctx, pthread_attr_getaffinity_np, attr, cpusetsize,
4973 cpuset);
4974 // FIXME: under ASan the call below may write to freed memory and corrupt
4975 // its metadata. See
4976 // https://github.com/google/sanitizers/issues/321.
4977 int res = REAL(pthread_attr_getaffinity_np)(attr, cpusetsize, cpuset);
4978 if (!res && cpusetsize && cpuset)
4979 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cpuset, cpusetsize);
4980 return res;
4981 }
4983 #define INIT_PTHREAD_ATTR_GETAFFINITY_NP \
4984 COMMON_INTERCEPT_FUNCTION(pthread_attr_getaffinity_np);
4985 #else
4986 #define INIT_PTHREAD_ATTR_GETAFFINITY_NP
4987 #endif
4989 #if SANITIZER_INTERCEPT_PTHREAD_GETAFFINITY_NP
4990 INTERCEPTOR(int, pthread_getaffinity_np, void *attr, SIZE_T cpusetsize,
4991 void *cpuset) {
4992 void *ctx;
4993 COMMON_INTERCEPTOR_ENTER(ctx, pthread_getaffinity_np, attr, cpusetsize,
4994 cpuset);
4995 // FIXME: under ASan the call below may write to freed memory and corrupt
4996 // its metadata. See
4997 // https://github.com/google/sanitizers/issues/321.
4998 int res = REAL(pthread_getaffinity_np)(attr, cpusetsize, cpuset);
4999 if (!res && cpusetsize && cpuset)
5000 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cpuset, cpusetsize);
5001 return res;
5002 }
5004 #define INIT_PTHREAD_GETAFFINITY_NP \
5005 COMMON_INTERCEPT_FUNCTION(pthread_getaffinity_np);
5006 #else
5007 #define INIT_PTHREAD_GETAFFINITY_NP
5008 #endif
5010 #if SANITIZER_INTERCEPT_PTHREAD_MUTEXATTR_GETPSHARED
5011 INTERCEPTOR_PTHREAD_MUTEXATTR_GET(pshared, sizeof(int))
5012 #define INIT_PTHREAD_MUTEXATTR_GETPSHARED \
5013 COMMON_INTERCEPT_FUNCTION(pthread_mutexattr_getpshared);
5014 #else
5015 #define INIT_PTHREAD_MUTEXATTR_GETPSHARED
5016 #endif
5018 #if SANITIZER_INTERCEPT_PTHREAD_MUTEXATTR_GETTYPE
5019 INTERCEPTOR_PTHREAD_MUTEXATTR_GET(type, sizeof(int))
5020 #define INIT_PTHREAD_MUTEXATTR_GETTYPE \
5021 COMMON_INTERCEPT_FUNCTION(pthread_mutexattr_gettype);
5022 #else
5023 #define INIT_PTHREAD_MUTEXATTR_GETTYPE
5024 #endif
5026 #if SANITIZER_INTERCEPT_PTHREAD_MUTEXATTR_GETPROTOCOL
5027 INTERCEPTOR_PTHREAD_MUTEXATTR_GET(protocol, sizeof(int))
5028 #define INIT_PTHREAD_MUTEXATTR_GETPROTOCOL \
5029 COMMON_INTERCEPT_FUNCTION(pthread_mutexattr_getprotocol);
5030 #else
5031 #define INIT_PTHREAD_MUTEXATTR_GETPROTOCOL
5032 #endif
5034 #if SANITIZER_INTERCEPT_PTHREAD_MUTEXATTR_GETPRIOCEILING
5035 INTERCEPTOR_PTHREAD_MUTEXATTR_GET(prioceiling, sizeof(int))
5036 #define INIT_PTHREAD_MUTEXATTR_GETPRIOCEILING \
5037 COMMON_INTERCEPT_FUNCTION(pthread_mutexattr_getprioceiling);
5038 #else
5039 #define INIT_PTHREAD_MUTEXATTR_GETPRIOCEILING
5040 #endif
5042 #if SANITIZER_INTERCEPT_PTHREAD_MUTEXATTR_GETROBUST
5043 INTERCEPTOR_PTHREAD_MUTEXATTR_GET(robust, sizeof(int))
5044 #define INIT_PTHREAD_MUTEXATTR_GETROBUST \
5045 COMMON_INTERCEPT_FUNCTION(pthread_mutexattr_getrobust);
5046 #else
5047 #define INIT_PTHREAD_MUTEXATTR_GETROBUST
5048 #endif
5050 #if SANITIZER_INTERCEPT_PTHREAD_MUTEXATTR_GETROBUST_NP
5051 INTERCEPTOR_PTHREAD_MUTEXATTR_GET(robust_np, sizeof(int))
5052 #define INIT_PTHREAD_MUTEXATTR_GETROBUST_NP \
5053 COMMON_INTERCEPT_FUNCTION(pthread_mutexattr_getrobust_np);
5054 #else
5055 #define INIT_PTHREAD_MUTEXATTR_GETROBUST_NP
5056 #endif
5058 #if SANITIZER_INTERCEPT_PTHREAD_RWLOCKATTR_GETPSHARED
5059 INTERCEPTOR_PTHREAD_RWLOCKATTR_GET(pshared, sizeof(int))
5060 #define INIT_PTHREAD_RWLOCKATTR_GETPSHARED \
5061 COMMON_INTERCEPT_FUNCTION(pthread_rwlockattr_getpshared);
5062 #else
5063 #define INIT_PTHREAD_RWLOCKATTR_GETPSHARED
5064 #endif
5066 #if SANITIZER_INTERCEPT_PTHREAD_RWLOCKATTR_GETKIND_NP
5067 INTERCEPTOR_PTHREAD_RWLOCKATTR_GET(kind_np, sizeof(int))
5068 #define INIT_PTHREAD_RWLOCKATTR_GETKIND_NP \
5069 COMMON_INTERCEPT_FUNCTION(pthread_rwlockattr_getkind_np);
5070 #else
5071 #define INIT_PTHREAD_RWLOCKATTR_GETKIND_NP
5072 #endif
5074 #if SANITIZER_INTERCEPT_PTHREAD_CONDATTR_GETPSHARED
5075 INTERCEPTOR_PTHREAD_CONDATTR_GET(pshared, sizeof(int))
5076 #define INIT_PTHREAD_CONDATTR_GETPSHARED \
5077 COMMON_INTERCEPT_FUNCTION(pthread_condattr_getpshared);
5078 #else
5079 #define INIT_PTHREAD_CONDATTR_GETPSHARED
5080 #endif
5082 #if SANITIZER_INTERCEPT_PTHREAD_CONDATTR_GETCLOCK
5083 INTERCEPTOR_PTHREAD_CONDATTR_GET(clock, sizeof(int))
5084 #define INIT_PTHREAD_CONDATTR_GETCLOCK \
5085 COMMON_INTERCEPT_FUNCTION(pthread_condattr_getclock);
5086 #else
5087 #define INIT_PTHREAD_CONDATTR_GETCLOCK
5088 #endif
5090 #if SANITIZER_INTERCEPT_PTHREAD_BARRIERATTR_GETPSHARED
5091 INTERCEPTOR_PTHREAD_BARRIERATTR_GET(pshared, sizeof(int)) // !mac !android
5092 #define INIT_PTHREAD_BARRIERATTR_GETPSHARED \
5093 COMMON_INTERCEPT_FUNCTION(pthread_barrierattr_getpshared);
5094 #else
5095 #define INIT_PTHREAD_BARRIERATTR_GETPSHARED
5096 #endif
5098 #if SANITIZER_INTERCEPT_TMPNAM
5099 INTERCEPTOR(char *, tmpnam, char *s) {
5100 void *ctx;
5101 COMMON_INTERCEPTOR_ENTER(ctx, tmpnam, s);
5102 char *res = REAL(tmpnam)(s);
5103 if (res) {
5104 if (s)
5105 // FIXME: under ASan the call below may write to freed memory and corrupt
5106 // its metadata. See
5107 // https://github.com/google/sanitizers/issues/321.
5108 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, s, internal_strlen(s) + 1);
5109 else
5110 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, internal_strlen(res) + 1);
5111 }
5112 return res;
5113 }
5114 #define INIT_TMPNAM COMMON_INTERCEPT_FUNCTION(tmpnam);
5115 #else
5116 #define INIT_TMPNAM
5117 #endif
5119 #if SANITIZER_INTERCEPT_TMPNAM_R
5120 INTERCEPTOR(char *, tmpnam_r, char *s) {
5121 void *ctx;
5122 COMMON_INTERCEPTOR_ENTER(ctx, tmpnam_r, s);
5123 // FIXME: under ASan the call below may write to freed memory and corrupt
5124 // its metadata. See
5125 // https://github.com/google/sanitizers/issues/321.
5126 char *res = REAL(tmpnam_r)(s);
5127 if (res && s) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, s, internal_strlen(s) + 1);
5128 return res;
5129 }
5130 #define INIT_TMPNAM_R COMMON_INTERCEPT_FUNCTION(tmpnam_r);
5131 #else
5132 #define INIT_TMPNAM_R
5133 #endif
5135 #if SANITIZER_INTERCEPT_PTSNAME
5136 INTERCEPTOR(char *, ptsname, int fd) {
5137 void *ctx;
5138 COMMON_INTERCEPTOR_ENTER(ctx, ptsname, fd);
5139 char *res = REAL(ptsname)(fd);
5140 if (res != nullptr)
5141 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, internal_strlen(res) + 1);
5142 return res;
5143 }
5144 #define INIT_PTSNAME COMMON_INTERCEPT_FUNCTION(ptsname);
5145 #else
5146 #define INIT_PTSNAME
5147 #endif
5149 #if SANITIZER_INTERCEPT_PTSNAME_R
5150 INTERCEPTOR(int, ptsname_r, int fd, char *name, SIZE_T namesize) {
5151 void *ctx;
5152 COMMON_INTERCEPTOR_ENTER(ctx, ptsname_r, fd, name, namesize);
5153 int res = REAL(ptsname_r)(fd, name, namesize);
5154 if (res == 0)
5155 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, name, internal_strlen(name) + 1);
5156 return res;
5157 }
5158 #define INIT_PTSNAME_R COMMON_INTERCEPT_FUNCTION(ptsname_r);
5159 #else
5160 #define INIT_PTSNAME_R
5161 #endif
5163 #if SANITIZER_INTERCEPT_TTYNAME
5164 INTERCEPTOR(char *, ttyname, int fd) {
5165 void *ctx;
5166 COMMON_INTERCEPTOR_ENTER(ctx, ttyname, fd);
5167 char *res = REAL(ttyname)(fd);
5168 if (res != nullptr)
5169 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, internal_strlen(res) + 1);
5170 return res;
5171 }
5172 #define INIT_TTYNAME COMMON_INTERCEPT_FUNCTION(ttyname);
5173 #else
5174 #define INIT_TTYNAME
5175 #endif
5177 #if SANITIZER_INTERCEPT_TTYNAME_R
5178 INTERCEPTOR(int, ttyname_r, int fd, char *name, SIZE_T namesize) {
5179 void *ctx;
5180 COMMON_INTERCEPTOR_ENTER(ctx, ttyname_r, fd, name, namesize);
5181 int res = REAL(ttyname_r)(fd, name, namesize);
5182 if (res == 0)
5183 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, name, internal_strlen(name) + 1);
5184 return res;
5185 }
5186 #define INIT_TTYNAME_R COMMON_INTERCEPT_FUNCTION(ttyname_r);
5187 #else
5188 #define INIT_TTYNAME_R
5189 #endif
5191 #if SANITIZER_INTERCEPT_TEMPNAM
5192 INTERCEPTOR(char *, tempnam, char *dir, char *pfx) {
5193 void *ctx;
5194 COMMON_INTERCEPTOR_ENTER(ctx, tempnam, dir, pfx);
5195 if (dir) COMMON_INTERCEPTOR_READ_RANGE(ctx, dir, internal_strlen(dir) + 1);
5196 if (pfx) COMMON_INTERCEPTOR_READ_RANGE(ctx, pfx, internal_strlen(pfx) + 1);
5197 char *res = REAL(tempnam)(dir, pfx);
5198 if (res) COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, internal_strlen(res) + 1);
5199 return res;
5200 }
5201 #define INIT_TEMPNAM COMMON_INTERCEPT_FUNCTION(tempnam);
5202 #else
5203 #define INIT_TEMPNAM
5204 #endif
5206 #if SANITIZER_INTERCEPT_PTHREAD_SETNAME_NP && !SANITIZER_NETBSD
5207 INTERCEPTOR(int, pthread_setname_np, uptr thread, const char *name) {
5208 void *ctx;
5209 COMMON_INTERCEPTOR_ENTER(ctx, pthread_setname_np, thread, name);
5210 COMMON_INTERCEPTOR_READ_STRING(ctx, name, 0);
5211 COMMON_INTERCEPTOR_SET_PTHREAD_NAME(ctx, thread, name);
5212 return REAL(pthread_setname_np)(thread, name);
5213 }
5214 #define INIT_PTHREAD_SETNAME_NP COMMON_INTERCEPT_FUNCTION(pthread_setname_np);
5215 #elif SANITIZER_INTERCEPT_PTHREAD_SETNAME_NP && SANITIZER_NETBSD
5216 INTERCEPTOR(int, pthread_setname_np, uptr thread, const char *name, void *arg) {
5217 void *ctx;
5218 char newname[32]; // PTHREAD_MAX_NAMELEN_NP=32
5219 COMMON_INTERCEPTOR_ENTER(ctx, pthread_setname_np, thread, name, arg);
5220 COMMON_INTERCEPTOR_READ_STRING(ctx, name, 0);
5221 internal_snprintf(newname, sizeof(newname), name, arg);
5222 COMMON_INTERCEPTOR_SET_PTHREAD_NAME(ctx, thread, newname);
5223 return REAL(pthread_setname_np)(thread, name, arg);
5224 }
5225 #define INIT_PTHREAD_SETNAME_NP COMMON_INTERCEPT_FUNCTION(pthread_setname_np);
5226 #else
5227 #define INIT_PTHREAD_SETNAME_NP
5228 #endif
5230 #if SANITIZER_INTERCEPT_PTHREAD_GETNAME_NP
5231 INTERCEPTOR(int, pthread_getname_np, uptr thread, char *name, SIZE_T len) {
5232 void *ctx;
5233 COMMON_INTERCEPTOR_ENTER(ctx, pthread_getname_np, thread, name, len);
5234 int res = REAL(pthread_getname_np)(thread, name, len);
5235 if (!res)
5236 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, name, internal_strnlen(name, len) + 1);
5237 return res;
5238 }
5239 #define INIT_PTHREAD_GETNAME_NP COMMON_INTERCEPT_FUNCTION(pthread_getname_np);
5240 #else
5241 #define INIT_PTHREAD_GETNAME_NP
5242 #endif
5244 #if SANITIZER_INTERCEPT_SINCOS
5245 INTERCEPTOR(void, sincos, double x, double *sin, double *cos) {
5246 void *ctx;
5247 COMMON_INTERCEPTOR_ENTER(ctx, sincos, x, sin, cos);
5248 // FIXME: under ASan the call below may write to freed memory and corrupt
5249 // its metadata. See
5250 // https://github.com/google/sanitizers/issues/321.
5251 REAL(sincos)(x, sin, cos);
5252 if (sin) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, sin, sizeof(*sin));
5253 if (cos) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cos, sizeof(*cos));
5254 }
5255 INTERCEPTOR(void, sincosf, float x, float *sin, float *cos) {
5256 void *ctx;
5257 COMMON_INTERCEPTOR_ENTER(ctx, sincosf, x, sin, cos);
5258 // FIXME: under ASan the call below may write to freed memory and corrupt
5259 // its metadata. See
5260 // https://github.com/google/sanitizers/issues/321.
5261 REAL(sincosf)(x, sin, cos);
5262 if (sin) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, sin, sizeof(*sin));
5263 if (cos) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cos, sizeof(*cos));
5264 }
5265 INTERCEPTOR(void, sincosl, long double x, long double *sin, long double *cos) {
5266 void *ctx;
5267 COMMON_INTERCEPTOR_ENTER(ctx, sincosl, x, sin, cos);
5268 // FIXME: under ASan the call below may write to freed memory and corrupt
5269 // its metadata. See
5270 // https://github.com/google/sanitizers/issues/321.
5271 REAL(sincosl)(x, sin, cos);
5272 if (sin) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, sin, sizeof(*sin));
5273 if (cos) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cos, sizeof(*cos));
5274 }
5275 #define INIT_SINCOS \
5276 COMMON_INTERCEPT_FUNCTION(sincos); \
5277 COMMON_INTERCEPT_FUNCTION(sincosf); \
5278 COMMON_INTERCEPT_FUNCTION_LDBL(sincosl);
5279 #else
5280 #define INIT_SINCOS
5281 #endif
5283 #if SANITIZER_INTERCEPT_REMQUO
5284 INTERCEPTOR(double, remquo, double x, double y, int *quo) {
5285 void *ctx;
5286 COMMON_INTERCEPTOR_ENTER(ctx, remquo, x, y, quo);
5287 // FIXME: under ASan the call below may write to freed memory and corrupt
5288 // its metadata. See
5289 // https://github.com/google/sanitizers/issues/321.
5290 double res = REAL(remquo)(x, y, quo);
5291 if (quo) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, quo, sizeof(*quo));
5292 return res;
5293 }
5294 INTERCEPTOR(float, remquof, float x, float y, int *quo) {
5295 void *ctx;
5296 COMMON_INTERCEPTOR_ENTER(ctx, remquof, x, y, quo);
5297 // FIXME: under ASan the call below may write to freed memory and corrupt
5298 // its metadata. See
5299 // https://github.com/google/sanitizers/issues/321.
5300 float res = REAL(remquof)(x, y, quo);
5301 if (quo) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, quo, sizeof(*quo));
5302 return res;
5303 }
5304 #define INIT_REMQUO \
5305 COMMON_INTERCEPT_FUNCTION(remquo); \
5306 COMMON_INTERCEPT_FUNCTION(remquof);
5307 #else
5308 #define INIT_REMQUO
5309 #endif
5311 #if SANITIZER_INTERCEPT_REMQUOL
5312 INTERCEPTOR(long double, remquol, long double x, long double y, int *quo) {
5313 void *ctx;
5314 COMMON_INTERCEPTOR_ENTER(ctx, remquol, x, y, quo);
5315 // FIXME: under ASan the call below may write to freed memory and corrupt
5316 // its metadata. See
5317 // https://github.com/google/sanitizers/issues/321.
5318 long double res = REAL(remquol)(x, y, quo);
5319 if (quo) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, quo, sizeof(*quo));
5320 return res;
5321 }
5322 #define INIT_REMQUOL \
5323 COMMON_INTERCEPT_FUNCTION_LDBL(remquol);
5324 #else
5325 #define INIT_REMQUOL
5326 #endif
5328 #if SANITIZER_INTERCEPT_LGAMMA
5329 extern int signgam;
5330 INTERCEPTOR(double, lgamma, double x) {
5331 void *ctx;
5332 COMMON_INTERCEPTOR_ENTER(ctx, lgamma, x);
5333 double res = REAL(lgamma)(x);
5334 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, &signgam, sizeof(signgam));
5335 return res;
5336 }
5337 INTERCEPTOR(float, lgammaf, float x) {
5338 void *ctx;
5339 COMMON_INTERCEPTOR_ENTER(ctx, lgammaf, x);
5340 float res = REAL(lgammaf)(x);
5341 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, &signgam, sizeof(signgam));
5342 return res;
5343 }
5344 #define INIT_LGAMMA \
5345 COMMON_INTERCEPT_FUNCTION(lgamma); \
5346 COMMON_INTERCEPT_FUNCTION(lgammaf);
5347 #else
5348 #define INIT_LGAMMA
5349 #endif
5351 #if SANITIZER_INTERCEPT_LGAMMAL
5352 INTERCEPTOR(long double, lgammal, long double x) {
5353 void *ctx;
5354 COMMON_INTERCEPTOR_ENTER(ctx, lgammal, x);
5355 long double res = REAL(lgammal)(x);
5356 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, &signgam, sizeof(signgam));
5357 return res;
5358 }
5359 #define INIT_LGAMMAL \
5360 COMMON_INTERCEPT_FUNCTION_LDBL(lgammal);
5361 #else
5362 #define INIT_LGAMMAL
5363 #endif
5365 #if SANITIZER_INTERCEPT_LGAMMA_R
5366 INTERCEPTOR(double, lgamma_r, double x, int *signp) {
5367 void *ctx;
5368 COMMON_INTERCEPTOR_ENTER(ctx, lgamma_r, x, signp);
5369 // FIXME: under ASan the call below may write to freed memory and corrupt
5370 // its metadata. See
5371 // https://github.com/google/sanitizers/issues/321.
5372 double res = REAL(lgamma_r)(x, signp);
5373 if (signp) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, signp, sizeof(*signp));
5374 return res;
5375 }
5376 INTERCEPTOR(float, lgammaf_r, float x, int *signp) {
5377 void *ctx;
5378 COMMON_INTERCEPTOR_ENTER(ctx, lgammaf_r, x, signp);
5379 // FIXME: under ASan the call below may write to freed memory and corrupt
5380 // its metadata. See
5381 // https://github.com/google/sanitizers/issues/321.
5382 float res = REAL(lgammaf_r)(x, signp);
5383 if (signp) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, signp, sizeof(*signp));
5384 return res;
5385 }
5386 #define INIT_LGAMMA_R \
5387 COMMON_INTERCEPT_FUNCTION(lgamma_r); \
5388 COMMON_INTERCEPT_FUNCTION(lgammaf_r);
5389 #else
5390 #define INIT_LGAMMA_R
5391 #endif
5393 #if SANITIZER_INTERCEPT_LGAMMAL_R
5394 INTERCEPTOR(long double, lgammal_r, long double x, int *signp) {
5395 void *ctx;
5396 COMMON_INTERCEPTOR_ENTER(ctx, lgammal_r, x, signp);
5397 // FIXME: under ASan the call below may write to freed memory and corrupt
5398 // its metadata. See
5399 // https://github.com/google/sanitizers/issues/321.
5400 long double res = REAL(lgammal_r)(x, signp);
5401 if (signp) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, signp, sizeof(*signp));
5402 return res;
5403 }
5404 #define INIT_LGAMMAL_R COMMON_INTERCEPT_FUNCTION_LDBL(lgammal_r);
5405 #else
5406 #define INIT_LGAMMAL_R
5407 #endif
5409 #if SANITIZER_INTERCEPT_DRAND48_R
5410 INTERCEPTOR(int, drand48_r, void *buffer, double *result) {
5411 void *ctx;
5412 COMMON_INTERCEPTOR_ENTER(ctx, drand48_r, buffer, result);
5413 // FIXME: under ASan the call below may write to freed memory and corrupt
5414 // its metadata. See
5415 // https://github.com/google/sanitizers/issues/321.
5416 int res = REAL(drand48_r)(buffer, result);
5417 if (result) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
5418 return res;
5419 }
5420 INTERCEPTOR(int, lrand48_r, void *buffer, long *result) {
5421 void *ctx;
5422 COMMON_INTERCEPTOR_ENTER(ctx, lrand48_r, buffer, result);
5423 // FIXME: under ASan the call below may write to freed memory and corrupt
5424 // its metadata. See
5425 // https://github.com/google/sanitizers/issues/321.
5426 int res = REAL(lrand48_r)(buffer, result);
5427 if (result) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(*result));
5428 return res;
5429 }
5430 #define INIT_DRAND48_R \
5431 COMMON_INTERCEPT_FUNCTION(drand48_r); \
5432 COMMON_INTERCEPT_FUNCTION(lrand48_r);
5433 #else
5434 #define INIT_DRAND48_R
5435 #endif
5437 #if SANITIZER_INTERCEPT_RAND_R
5438 INTERCEPTOR(int, rand_r, unsigned *seedp) {
5439 void *ctx;
5440 COMMON_INTERCEPTOR_ENTER(ctx, rand_r, seedp);
5441 COMMON_INTERCEPTOR_READ_RANGE(ctx, seedp, sizeof(*seedp));
5442 return REAL(rand_r)(seedp);
5443 }
5444 #define INIT_RAND_R COMMON_INTERCEPT_FUNCTION(rand_r);
5445 #else
5446 #define INIT_RAND_R
5447 #endif
5449 #if SANITIZER_INTERCEPT_GETLINE
5450 INTERCEPTOR(SSIZE_T, getline, char **lineptr, SIZE_T *n, void *stream) {
5451 void *ctx;
5452 COMMON_INTERCEPTOR_ENTER(ctx, getline, lineptr, n, stream);
5453 // FIXME: under ASan the call below may write to freed memory and corrupt
5454 // its metadata. See
5455 // https://github.com/google/sanitizers/issues/321.
5456 SSIZE_T res = REAL(getline)(lineptr, n, stream);
5457 if (res > 0) {
5458 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, lineptr, sizeof(*lineptr));
5459 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, n, sizeof(*n));
5460 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *lineptr, res + 1);
5461 }
5462 return res;
5463 }
5465 // FIXME: under ASan the call below may write to freed memory and corrupt its
5466 // metadata. See
5467 // https://github.com/google/sanitizers/issues/321.
5468 #define GETDELIM_INTERCEPTOR_IMPL(vname) \
5469 { \
5470 void *ctx; \
5471 COMMON_INTERCEPTOR_ENTER(ctx, vname, lineptr, n, delim, stream); \
5472 SSIZE_T res = REAL(vname)(lineptr, n, delim, stream); \
5473 if (res > 0) { \
5474 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, lineptr, sizeof(*lineptr)); \
5475 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, n, sizeof(*n)); \
5476 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *lineptr, res + 1); \
5477 } \
5478 return res; \
5479 }
5481 INTERCEPTOR(SSIZE_T, __getdelim, char **lineptr, SIZE_T *n, int delim,
5482 void *stream)
5483 GETDELIM_INTERCEPTOR_IMPL(__getdelim)
5485 // There's no __getdelim() on FreeBSD so we supply the getdelim() interceptor
5486 // with its own body.
5487 INTERCEPTOR(SSIZE_T, getdelim, char **lineptr, SIZE_T *n, int delim,
5488 void *stream)
5489 GETDELIM_INTERCEPTOR_IMPL(getdelim)
5491 #define INIT_GETLINE \
5492 COMMON_INTERCEPT_FUNCTION(getline); \
5493 COMMON_INTERCEPT_FUNCTION(__getdelim); \
5494 COMMON_INTERCEPT_FUNCTION(getdelim);
5495 #else
5496 #define INIT_GETLINE
5497 #endif
5499 #if SANITIZER_INTERCEPT_ICONV
5500 INTERCEPTOR(SIZE_T, iconv, void *cd, char **inbuf, SIZE_T *inbytesleft,
5501 char **outbuf, SIZE_T *outbytesleft) {
5502 void *ctx;
5503 COMMON_INTERCEPTOR_ENTER(ctx, iconv, cd, inbuf, inbytesleft, outbuf,
5504 outbytesleft);
5505 if (inbytesleft)
5506 COMMON_INTERCEPTOR_READ_RANGE(ctx, inbytesleft, sizeof(*inbytesleft));
5507 if (inbuf && inbytesleft)
5508 COMMON_INTERCEPTOR_READ_RANGE(ctx, *inbuf, *inbytesleft);
5509 if (outbytesleft)
5510 COMMON_INTERCEPTOR_READ_RANGE(ctx, outbytesleft, sizeof(*outbytesleft));
5511 void *outbuf_orig = outbuf ? *outbuf : nullptr;
5512 // FIXME: under ASan the call below may write to freed memory and corrupt
5513 // its metadata. See
5514 // https://github.com/google/sanitizers/issues/321.
5515 SIZE_T res = REAL(iconv)(cd, inbuf, inbytesleft, outbuf, outbytesleft);
5516 if (outbuf && *outbuf > outbuf_orig) {
5517 SIZE_T sz = (char *)*outbuf - (char *)outbuf_orig;
5518 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, outbuf_orig, sz);
5519 }
5520 return res;
5521 }
5522 #define INIT_ICONV COMMON_INTERCEPT_FUNCTION(iconv);
5523 #else
5524 #define INIT_ICONV
5525 #endif
5527 #if SANITIZER_INTERCEPT_TIMES
5528 INTERCEPTOR(__sanitizer_clock_t, times, void *tms) {
5529 void *ctx;
5530 COMMON_INTERCEPTOR_ENTER(ctx, times, tms);
5531 // FIXME: under ASan the call below may write to freed memory and corrupt
5532 // its metadata. See
5533 // https://github.com/google/sanitizers/issues/321.
5534 __sanitizer_clock_t res = REAL(times)(tms);
5535 if (res != (__sanitizer_clock_t)-1 && tms)
5536 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, tms, struct_tms_sz);
5537 return res;
5538 }
5539 #define INIT_TIMES COMMON_INTERCEPT_FUNCTION(times);
5540 #else
5541 #define INIT_TIMES
5542 #endif
5544 #if SANITIZER_S390 && \
5545 (SANITIZER_INTERCEPT_TLS_GET_ADDR || SANITIZER_INTERCEPT_TLS_GET_OFFSET)
5546 extern "C" uptr __tls_get_offset_wrapper(void *arg, uptr (*fn)(void *arg));
5547 DEFINE_REAL(uptr, __tls_get_offset, void *arg)
5548 #endif
5550 #if SANITIZER_INTERCEPT_TLS_GET_ADDR
5551 #if !SANITIZER_S390
5552 #define INIT_TLS_GET_ADDR COMMON_INTERCEPT_FUNCTION(__tls_get_addr)
5553 // If you see any crashes around this functions, there are 2 known issues with
5554 // it: 1. __tls_get_addr can be called with mis-aligned stack due to:
5555 // https://gcc.gnu.org/bugzilla/show_bug.cgi?id=58066
5556 // 2. It can be called recursively if sanitizer code uses __tls_get_addr
5557 // to access thread local variables (it should not happen normally,
5558 // because sanitizers use initial-exec tls model).
5559 INTERCEPTOR(void *, __tls_get_addr, void *arg) {
5560 void *ctx;
5561 COMMON_INTERCEPTOR_ENTER(ctx, __tls_get_addr, arg);
5562 void *res = REAL(__tls_get_addr)(arg);
5563 uptr tls_begin, tls_end;
5564 COMMON_INTERCEPTOR_GET_TLS_RANGE(&tls_begin, &tls_end);
5565 DTLS::DTV *dtv = DTLS_on_tls_get_addr(arg, res, tls_begin, tls_end);
5566 if (dtv) {
5567 // New DTLS block has been allocated.
5568 COMMON_INTERCEPTOR_INITIALIZE_RANGE((void *)dtv->beg, dtv->size);
5569 }
5570 return res;
5571 }
5572 #if SANITIZER_PPC
5573 // On PowerPC, we also need to intercept __tls_get_addr_opt, which has
5574 // mostly the same semantics as __tls_get_addr, but its presence enables
5575 // some optimizations in linker (which are safe to ignore here).
5576 INTERCEPTOR(void *, __tls_get_addr_opt, void *arg) ALIAS(WRAP(__tls_get_addr));
5577 #endif
5578 #else // SANITIZER_S390
5579 // On s390, we have to intercept two functions here:
5580 // - __tls_get_addr_internal, which is a glibc-internal function that is like
5581 // the usual __tls_get_addr, but returns a TP-relative offset instead of
5582 // a proper pointer. It is used by dlsym for TLS symbols.
5583 // - __tls_get_offset, which is like the above, but also takes a GOT-relative
5584 // descriptor offset as an argument instead of a pointer. GOT address
5585 // is passed in r12, so it's necessary to write it in assembly. This is
5586 // the function used by the compiler.
5587 #define INIT_TLS_GET_ADDR COMMON_INTERCEPT_FUNCTION(__tls_get_offset)
5588 INTERCEPTOR(uptr, __tls_get_addr_internal, void *arg) {
5589 void *ctx;
5590 COMMON_INTERCEPTOR_ENTER(ctx, __tls_get_addr_internal, arg);
5591 uptr res = __tls_get_offset_wrapper(arg, REAL(__tls_get_offset));
5592 uptr tp = reinterpret_cast<uptr>(__builtin_thread_pointer());
5593 void *ptr = reinterpret_cast<void *>(res + tp);
5594 uptr tls_begin, tls_end;
5595 COMMON_INTERCEPTOR_GET_TLS_RANGE(&tls_begin, &tls_end);
5596 DTLS::DTV *dtv = DTLS_on_tls_get_addr(arg, ptr, tls_begin, tls_end);
5597 if (dtv) {
5598 // New DTLS block has been allocated.
5599 COMMON_INTERCEPTOR_INITIALIZE_RANGE((void *)dtv->beg, dtv->size);
5600 }
5601 return res;
5602 }
5603 #endif // SANITIZER_S390
5604 #else
5605 #define INIT_TLS_GET_ADDR
5606 #endif
5608 #if SANITIZER_S390 && \
5609 (SANITIZER_INTERCEPT_TLS_GET_ADDR || SANITIZER_INTERCEPT_TLS_GET_OFFSET)
5610 // We need a hidden symbol aliasing the above, so that we can jump
5611 // directly to it from the assembly below.
5612 extern "C" __attribute__((visibility("hidden"))) uptr __tls_get_addr_hidden(
5613 void *arg) ALIAS(WRAP(__tls_get_addr_internal));
5614 extern "C" uptr __tls_get_offset(void *arg);
5615 extern "C" uptr TRAMPOLINE(__tls_get_offset)(void *arg);
5616 extern "C" uptr WRAP(__tls_get_offset)(void *arg);
5617 // Now carefully intercept __tls_get_offset.
5618 asm(
5619 ".text\n"
5620 // The __intercept_ version has to exist, so that gen_dynamic_list.py
5621 // exports our symbol.
5622 ".weak __tls_get_offset\n"
5623 ".set __tls_get_offset, __interceptor___tls_get_offset\n"
5624 ".global __interceptor___tls_get_offset\n"
5625 ".type __interceptor___tls_get_offset, @function\n"
5626 "__interceptor___tls_get_offset:\n"
5627 #ifdef __s390x__
5628 "la %r2, 0(%r2,%r12)\n"
5629 "jg __tls_get_addr_hidden\n"
5630 #else
5631 "basr %r3,0\n"
5632 "0: la %r2,0(%r2,%r12)\n"
5633 "l %r4,1f-0b(%r3)\n"
5634 "b 0(%r4,%r3)\n"
5635 "1: .long __tls_get_addr_hidden - 0b\n"
5636 #endif
5637 ".size __interceptor___tls_get_offset, .-__interceptor___tls_get_offset\n"
5638 // Assembly wrapper to call REAL(__tls_get_offset)(arg)
5639 ".type __tls_get_offset_wrapper, @function\n"
5640 "__tls_get_offset_wrapper:\n"
5641 #ifdef __s390x__
5642 "sgr %r2,%r12\n"
5643 #else
5644 "sr %r2,%r12\n"
5645 #endif
5646 "br %r3\n"
5647 ".size __tls_get_offset_wrapper, .-__tls_get_offset_wrapper\n"
5648 );
5649 #endif
5651 #if SANITIZER_INTERCEPT_LISTXATTR
5652 INTERCEPTOR(SSIZE_T, listxattr, const char *path, char *list, SIZE_T size) {
5653 void *ctx;
5654 COMMON_INTERCEPTOR_ENTER(ctx, listxattr, path, list, size);
5655 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
5656 // FIXME: under ASan the call below may write to freed memory and corrupt
5657 // its metadata. See
5658 // https://github.com/google/sanitizers/issues/321.
5659 SSIZE_T res = REAL(listxattr)(path, list, size);
5660 // Here and below, size == 0 is a special case where nothing is written to the
5661 // buffer, and res contains the desired buffer size.
5662 if (size && res > 0 && list) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, list, res);
5663 return res;
5664 }
5665 INTERCEPTOR(SSIZE_T, llistxattr, const char *path, char *list, SIZE_T size) {
5666 void *ctx;
5667 COMMON_INTERCEPTOR_ENTER(ctx, llistxattr, path, list, size);
5668 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
5669 // FIXME: under ASan the call below may write to freed memory and corrupt
5670 // its metadata. See
5671 // https://github.com/google/sanitizers/issues/321.
5672 SSIZE_T res = REAL(llistxattr)(path, list, size);
5673 if (size && res > 0 && list) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, list, res);
5674 return res;
5675 }
5676 INTERCEPTOR(SSIZE_T, flistxattr, int fd, char *list, SIZE_T size) {
5677 void *ctx;
5678 COMMON_INTERCEPTOR_ENTER(ctx, flistxattr, fd, list, size);
5679 // FIXME: under ASan the call below may write to freed memory and corrupt
5680 // its metadata. See
5681 // https://github.com/google/sanitizers/issues/321.
5682 SSIZE_T res = REAL(flistxattr)(fd, list, size);
5683 if (size && res > 0 && list) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, list, res);
5684 return res;
5685 }
5686 #define INIT_LISTXATTR \
5687 COMMON_INTERCEPT_FUNCTION(listxattr); \
5688 COMMON_INTERCEPT_FUNCTION(llistxattr); \
5689 COMMON_INTERCEPT_FUNCTION(flistxattr);
5690 #else
5691 #define INIT_LISTXATTR
5692 #endif
5694 #if SANITIZER_INTERCEPT_GETXATTR
5695 INTERCEPTOR(SSIZE_T, getxattr, const char *path, const char *name, char *value,
5696 SIZE_T size) {
5697 void *ctx;
5698 COMMON_INTERCEPTOR_ENTER(ctx, getxattr, path, name, value, size);
5699 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
5700 if (name) COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
5701 // FIXME: under ASan the call below may write to freed memory and corrupt
5702 // its metadata. See
5703 // https://github.com/google/sanitizers/issues/321.
5704 SSIZE_T res = REAL(getxattr)(path, name, value, size);
5705 if (size && res > 0 && value) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, value, res);
5706 return res;
5707 }
5708 INTERCEPTOR(SSIZE_T, lgetxattr, const char *path, const char *name, char *value,
5709 SIZE_T size) {
5710 void *ctx;
5711 COMMON_INTERCEPTOR_ENTER(ctx, lgetxattr, path, name, value, size);
5712 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
5713 if (name) COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
5714 // FIXME: under ASan the call below may write to freed memory and corrupt
5715 // its metadata. See
5716 // https://github.com/google/sanitizers/issues/321.
5717 SSIZE_T res = REAL(lgetxattr)(path, name, value, size);
5718 if (size && res > 0 && value) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, value, res);
5719 return res;
5720 }
5721 INTERCEPTOR(SSIZE_T, fgetxattr, int fd, const char *name, char *value,
5722 SIZE_T size) {
5723 void *ctx;
5724 COMMON_INTERCEPTOR_ENTER(ctx, fgetxattr, fd, name, value, size);
5725 if (name) COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
5726 // FIXME: under ASan the call below may write to freed memory and corrupt
5727 // its metadata. See
5728 // https://github.com/google/sanitizers/issues/321.
5729 SSIZE_T res = REAL(fgetxattr)(fd, name, value, size);
5730 if (size && res > 0 && value) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, value, res);
5731 return res;
5732 }
5733 #define INIT_GETXATTR \
5734 COMMON_INTERCEPT_FUNCTION(getxattr); \
5735 COMMON_INTERCEPT_FUNCTION(lgetxattr); \
5736 COMMON_INTERCEPT_FUNCTION(fgetxattr);
5737 #else
5738 #define INIT_GETXATTR
5739 #endif
5741 #if SANITIZER_INTERCEPT_GETRESID
5742 INTERCEPTOR(int, getresuid, void *ruid, void *euid, void *suid) {
5743 void *ctx;
5744 COMMON_INTERCEPTOR_ENTER(ctx, getresuid, ruid, euid, suid);
5745 // FIXME: under ASan the call below may write to freed memory and corrupt
5746 // its metadata. See
5747 // https://github.com/google/sanitizers/issues/321.
5748 int res = REAL(getresuid)(ruid, euid, suid);
5749 if (res >= 0) {
5750 if (ruid) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ruid, uid_t_sz);
5751 if (euid) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, euid, uid_t_sz);
5752 if (suid) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, suid, uid_t_sz);
5753 }
5754 return res;
5755 }
5756 INTERCEPTOR(int, getresgid, void *rgid, void *egid, void *sgid) {
5757 void *ctx;
5758 COMMON_INTERCEPTOR_ENTER(ctx, getresgid, rgid, egid, sgid);
5759 // FIXME: under ASan the call below may write to freed memory and corrupt
5760 // its metadata. See
5761 // https://github.com/google/sanitizers/issues/321.
5762 int res = REAL(getresgid)(rgid, egid, sgid);
5763 if (res >= 0) {
5764 if (rgid) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, rgid, gid_t_sz);
5765 if (egid) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, egid, gid_t_sz);
5766 if (sgid) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, sgid, gid_t_sz);
5767 }
5768 return res;
5769 }
5770 #define INIT_GETRESID \
5771 COMMON_INTERCEPT_FUNCTION(getresuid); \
5772 COMMON_INTERCEPT_FUNCTION(getresgid);
5773 #else
5774 #define INIT_GETRESID
5775 #endif
5777 #if SANITIZER_INTERCEPT_GETIFADDRS
5778 // As long as getifaddrs()/freeifaddrs() use calloc()/free(), we don't need to
5779 // intercept freeifaddrs(). If that ceases to be the case, we might need to
5780 // intercept it to poison the memory again.
5781 INTERCEPTOR(int, getifaddrs, __sanitizer_ifaddrs **ifap) {
5782 void *ctx;
5783 COMMON_INTERCEPTOR_ENTER(ctx, getifaddrs, ifap);
5784 // FIXME: under ASan the call below may write to freed memory and corrupt
5785 // its metadata. See
5786 // https://github.com/google/sanitizers/issues/321.
5787 int res = REAL(getifaddrs)(ifap);
5788 if (res == 0 && ifap) {
5789 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ifap, sizeof(void *));
5790 __sanitizer_ifaddrs *p = *ifap;
5791 while (p) {
5792 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p, sizeof(__sanitizer_ifaddrs));
5793 if (p->ifa_name)
5794 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p->ifa_name,
5795 internal_strlen(p->ifa_name) + 1);
5796 if (p->ifa_addr)
5797 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p->ifa_addr, struct_sockaddr_sz);
5798 if (p->ifa_netmask)
5799 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p->ifa_netmask, struct_sockaddr_sz);
5800 // On Linux this is a union, but the other member also points to a
5801 // struct sockaddr, so the following is sufficient.
5802 if (p->ifa_dstaddr)
5803 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p->ifa_dstaddr, struct_sockaddr_sz);
5804 // FIXME(smatveev): Unpoison p->ifa_data as well.
5805 p = p->ifa_next;
5806 }
5807 }
5808 return res;
5809 }
5810 #define INIT_GETIFADDRS \
5811 COMMON_INTERCEPT_FUNCTION(getifaddrs);
5812 #else
5813 #define INIT_GETIFADDRS
5814 #endif
5816 #if SANITIZER_INTERCEPT_IF_INDEXTONAME
5817 INTERCEPTOR(char *, if_indextoname, unsigned int ifindex, char* ifname) {
5818 void *ctx;
5819 COMMON_INTERCEPTOR_ENTER(ctx, if_indextoname, ifindex, ifname);
5820 // FIXME: under ASan the call below may write to freed memory and corrupt
5821 // its metadata. See
5822 // https://github.com/google/sanitizers/issues/321.
5823 char *res = REAL(if_indextoname)(ifindex, ifname);
5824 if (res && ifname)
5825 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ifname, internal_strlen(ifname) + 1);
5826 return res;
5827 }
5828 INTERCEPTOR(unsigned int, if_nametoindex, const char* ifname) {
5829 void *ctx;
5830 COMMON_INTERCEPTOR_ENTER(ctx, if_nametoindex, ifname);
5831 if (ifname)
5832 COMMON_INTERCEPTOR_READ_RANGE(ctx, ifname, internal_strlen(ifname) + 1);
5833 return REAL(if_nametoindex)(ifname);
5834 }
5835 #define INIT_IF_INDEXTONAME \
5836 COMMON_INTERCEPT_FUNCTION(if_indextoname); \
5837 COMMON_INTERCEPT_FUNCTION(if_nametoindex);
5838 #else
5839 #define INIT_IF_INDEXTONAME
5840 #endif
5842 #if SANITIZER_INTERCEPT_CAPGET
5843 INTERCEPTOR(int, capget, void *hdrp, void *datap) {
5844 void *ctx;
5845 COMMON_INTERCEPTOR_ENTER(ctx, capget, hdrp, datap);
5846 if (hdrp)
5847 COMMON_INTERCEPTOR_READ_RANGE(ctx, hdrp, __user_cap_header_struct_sz);
5848 // FIXME: under ASan the call below may write to freed memory and corrupt
5849 // its metadata. See
5850 // https://github.com/google/sanitizers/issues/321.
5851 int res = REAL(capget)(hdrp, datap);
5852 if (res == 0 && datap) {
5853 unsigned datasz = __user_cap_data_struct_sz(hdrp);
5854 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, datap, datasz);
5855 }
5856 // We can also return -1 and write to hdrp->version if the version passed in
5857 // hdrp->version is unsupported. But that's not a trivial condition to check,
5858 // and anyway COMMON_INTERCEPTOR_READ_RANGE protects us to some extent.
5859 return res;
5860 }
5861 INTERCEPTOR(int, capset, void *hdrp, const void *datap) {
5862 void *ctx;
5863 COMMON_INTERCEPTOR_ENTER(ctx, capset, hdrp, datap);
5864 if (hdrp)
5865 COMMON_INTERCEPTOR_READ_RANGE(ctx, hdrp, __user_cap_header_struct_sz);
5866 if (datap) {
5867 unsigned datasz = __user_cap_data_struct_sz(hdrp);
5868 COMMON_INTERCEPTOR_READ_RANGE(ctx, datap, datasz);
5869 }
5870 return REAL(capset)(hdrp, datap);
5871 }
5872 #define INIT_CAPGET \
5873 COMMON_INTERCEPT_FUNCTION(capget); \
5874 COMMON_INTERCEPT_FUNCTION(capset);
5875 #else
5876 #define INIT_CAPGET
5877 #endif
5879 #if SANITIZER_INTERCEPT_FTIME
5880 INTERCEPTOR(int, ftime, __sanitizer_timeb *tp) {
5881 void *ctx;
5882 COMMON_INTERCEPTOR_ENTER(ctx, ftime, tp);
5883 // FIXME: under ASan the call below may write to freed memory and corrupt
5884 // its metadata. See
5885 // https://github.com/google/sanitizers/issues/321.
5886 int res = REAL(ftime)(tp);
5887 if (tp)
5888 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, tp, sizeof(*tp));
5889 return res;
5890 }
5891 #define INIT_FTIME COMMON_INTERCEPT_FUNCTION(ftime);
5892 #else
5893 #define INIT_FTIME
5894 #endif // SANITIZER_INTERCEPT_FTIME
5896 #if SANITIZER_INTERCEPT_XDR
5897 INTERCEPTOR(void, xdrmem_create, __sanitizer_XDR *xdrs, uptr addr,
5898 unsigned size, int op) {
5899 void *ctx;
5900 COMMON_INTERCEPTOR_ENTER(ctx, xdrmem_create, xdrs, addr, size, op);
5901 // FIXME: under ASan the call below may write to freed memory and corrupt
5902 // its metadata. See
5903 // https://github.com/google/sanitizers/issues/321.
5904 REAL(xdrmem_create)(xdrs, addr, size, op);
5905 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, xdrs, sizeof(*xdrs));
5906 if (op == __sanitizer_XDR_ENCODE) {
5907 // It's not obvious how much data individual xdr_ routines write.
5908 // Simply unpoison the entire target buffer in advance.
5909 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, (void *)addr, size);
5910 }
5911 }
5913 INTERCEPTOR(void, xdrstdio_create, __sanitizer_XDR *xdrs, void *file, int op) {
5914 void *ctx;
5915 COMMON_INTERCEPTOR_ENTER(ctx, xdrstdio_create, xdrs, file, op);
5916 // FIXME: under ASan the call below may write to freed memory and corrupt
5917 // its metadata. See
5918 // https://github.com/google/sanitizers/issues/321.
5919 REAL(xdrstdio_create)(xdrs, file, op);
5920 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, xdrs, sizeof(*xdrs));
5921 }
5923 // FIXME: under ASan the call below may write to freed memory and corrupt
5924 // its metadata. See
5925 // https://github.com/google/sanitizers/issues/321.
5926 #define XDR_INTERCEPTOR(F, T) \
5927 INTERCEPTOR(int, F, __sanitizer_XDR *xdrs, T *p) { \
5928 void *ctx; \
5929 COMMON_INTERCEPTOR_ENTER(ctx, F, xdrs, p); \
5930 if (p && xdrs->x_op == __sanitizer_XDR_ENCODE) \
5931 COMMON_INTERCEPTOR_READ_RANGE(ctx, p, sizeof(*p)); \
5932 int res = REAL(F)(xdrs, p); \
5933 if (res && p && xdrs->x_op == __sanitizer_XDR_DECODE) \
5934 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p, sizeof(*p)); \
5935 return res; \
5936 }
5938 XDR_INTERCEPTOR(xdr_short, short)
5939 XDR_INTERCEPTOR(xdr_u_short, unsigned short)
5940 XDR_INTERCEPTOR(xdr_int, int)
5941 XDR_INTERCEPTOR(xdr_u_int, unsigned)
5942 XDR_INTERCEPTOR(xdr_long, long)
5943 XDR_INTERCEPTOR(xdr_u_long, unsigned long)
5944 XDR_INTERCEPTOR(xdr_hyper, long long)
5945 XDR_INTERCEPTOR(xdr_u_hyper, unsigned long long)
5946 XDR_INTERCEPTOR(xdr_longlong_t, long long)
5947 XDR_INTERCEPTOR(xdr_u_longlong_t, unsigned long long)
5948 XDR_INTERCEPTOR(xdr_int8_t, u8)
5949 XDR_INTERCEPTOR(xdr_uint8_t, u8)
5950 XDR_INTERCEPTOR(xdr_int16_t, u16)
5951 XDR_INTERCEPTOR(xdr_uint16_t, u16)
5952 XDR_INTERCEPTOR(xdr_int32_t, u32)
5953 XDR_INTERCEPTOR(xdr_uint32_t, u32)
5954 XDR_INTERCEPTOR(xdr_int64_t, u64)
5955 XDR_INTERCEPTOR(xdr_uint64_t, u64)
5956 XDR_INTERCEPTOR(xdr_quad_t, long long)
5957 XDR_INTERCEPTOR(xdr_u_quad_t, unsigned long long)
5958 XDR_INTERCEPTOR(xdr_bool, bool)
5959 XDR_INTERCEPTOR(xdr_enum, int)
5960 XDR_INTERCEPTOR(xdr_char, char)
5961 XDR_INTERCEPTOR(xdr_u_char, unsigned char)
5962 XDR_INTERCEPTOR(xdr_float, float)
5963 XDR_INTERCEPTOR(xdr_double, double)
5965 // FIXME: intercept xdr_array, opaque, union, vector, reference, pointer,
5966 // wrapstring, sizeof
5968 INTERCEPTOR(int, xdr_bytes, __sanitizer_XDR *xdrs, char **p, unsigned *sizep,
5969 unsigned maxsize) {
5970 void *ctx;
5971 COMMON_INTERCEPTOR_ENTER(ctx, xdr_bytes, xdrs, p, sizep, maxsize);
5972 if (p && sizep && xdrs->x_op == __sanitizer_XDR_ENCODE) {
5973 COMMON_INTERCEPTOR_READ_RANGE(ctx, p, sizeof(*p));
5974 COMMON_INTERCEPTOR_READ_RANGE(ctx, sizep, sizeof(*sizep));
5975 COMMON_INTERCEPTOR_READ_RANGE(ctx, *p, *sizep);
5976 }
5977 // FIXME: under ASan the call below may write to freed memory and corrupt
5978 // its metadata. See
5979 // https://github.com/google/sanitizers/issues/321.
5980 int res = REAL(xdr_bytes)(xdrs, p, sizep, maxsize);
5981 if (p && sizep && xdrs->x_op == __sanitizer_XDR_DECODE) {
5982 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p, sizeof(*p));
5983 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, sizep, sizeof(*sizep));
5984 if (res && *p && *sizep) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *p, *sizep);
5985 }
5986 return res;
5987 }
5989 INTERCEPTOR(int, xdr_string, __sanitizer_XDR *xdrs, char **p,
5990 unsigned maxsize) {
5991 void *ctx;
5992 COMMON_INTERCEPTOR_ENTER(ctx, xdr_string, xdrs, p, maxsize);
5993 if (p && xdrs->x_op == __sanitizer_XDR_ENCODE) {
5994 COMMON_INTERCEPTOR_READ_RANGE(ctx, p, sizeof(*p));
5995 COMMON_INTERCEPTOR_READ_RANGE(ctx, *p, internal_strlen(*p) + 1);
5996 }
5997 // FIXME: under ASan the call below may write to freed memory and corrupt
5998 // its metadata. See
5999 // https://github.com/google/sanitizers/issues/321.
6000 int res = REAL(xdr_string)(xdrs, p, maxsize);
6001 if (p && xdrs->x_op == __sanitizer_XDR_DECODE) {
6002 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p, sizeof(*p));
6003 if (res && *p)
6004 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *p, internal_strlen(*p) + 1);
6005 }
6006 return res;
6007 }
6009 #define INIT_XDR \
6010 COMMON_INTERCEPT_FUNCTION(xdrmem_create); \
6011 COMMON_INTERCEPT_FUNCTION(xdrstdio_create); \
6012 COMMON_INTERCEPT_FUNCTION(xdr_short); \
6013 COMMON_INTERCEPT_FUNCTION(xdr_u_short); \
6014 COMMON_INTERCEPT_FUNCTION(xdr_int); \
6015 COMMON_INTERCEPT_FUNCTION(xdr_u_int); \
6016 COMMON_INTERCEPT_FUNCTION(xdr_long); \
6017 COMMON_INTERCEPT_FUNCTION(xdr_u_long); \
6018 COMMON_INTERCEPT_FUNCTION(xdr_hyper); \
6019 COMMON_INTERCEPT_FUNCTION(xdr_u_hyper); \
6020 COMMON_INTERCEPT_FUNCTION(xdr_longlong_t); \
6021 COMMON_INTERCEPT_FUNCTION(xdr_u_longlong_t); \
6022 COMMON_INTERCEPT_FUNCTION(xdr_int8_t); \
6023 COMMON_INTERCEPT_FUNCTION(xdr_uint8_t); \
6024 COMMON_INTERCEPT_FUNCTION(xdr_int16_t); \
6025 COMMON_INTERCEPT_FUNCTION(xdr_uint16_t); \
6026 COMMON_INTERCEPT_FUNCTION(xdr_int32_t); \
6027 COMMON_INTERCEPT_FUNCTION(xdr_uint32_t); \
6028 COMMON_INTERCEPT_FUNCTION(xdr_int64_t); \
6029 COMMON_INTERCEPT_FUNCTION(xdr_uint64_t); \
6030 COMMON_INTERCEPT_FUNCTION(xdr_quad_t); \
6031 COMMON_INTERCEPT_FUNCTION(xdr_u_quad_t); \
6032 COMMON_INTERCEPT_FUNCTION(xdr_bool); \
6033 COMMON_INTERCEPT_FUNCTION(xdr_enum); \
6034 COMMON_INTERCEPT_FUNCTION(xdr_char); \
6035 COMMON_INTERCEPT_FUNCTION(xdr_u_char); \
6036 COMMON_INTERCEPT_FUNCTION(xdr_float); \
6037 COMMON_INTERCEPT_FUNCTION(xdr_double); \
6038 COMMON_INTERCEPT_FUNCTION(xdr_bytes); \
6039 COMMON_INTERCEPT_FUNCTION(xdr_string);
6040 #else
6041 #define INIT_XDR
6042 #endif // SANITIZER_INTERCEPT_XDR
6044 #if SANITIZER_INTERCEPT_XDRREC
6045 typedef int (*xdrrec_cb)(char*, char*, int);
6046 struct XdrRecWrapper {
6047 char *handle;
6048 xdrrec_cb rd, wr;
6049 };
6050 typedef AddrHashMap<XdrRecWrapper *, 11> XdrRecWrapMap;
6051 static XdrRecWrapMap *xdrrec_wrap_map;
6053 static int xdrrec_wr_wrap(char *handle, char *buf, int count) {
6054 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
6055 COMMON_INTERCEPTOR_INITIALIZE_RANGE(buf, count);
6056 XdrRecWrapper *wrap = (XdrRecWrapper *)handle;
6057 return wrap->wr(wrap->handle, buf, count);
6058 }
6060 static int xdrrec_rd_wrap(char *handle, char *buf, int count) {
6061 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
6062 XdrRecWrapper *wrap = (XdrRecWrapper *)handle;
6063 return wrap->rd(wrap->handle, buf, count);
6064 }
6066 // This doesn't apply to the solaris version as it has a different function
6067 // signature.
6068 INTERCEPTOR(void, xdrrec_create, __sanitizer_XDR *xdr, unsigned sndsize,
6069 unsigned rcvsize, char *handle, int (*rd)(char*, char*, int),
6070 int (*wr)(char*, char*, int)) {
6071 void *ctx;
6072 COMMON_INTERCEPTOR_ENTER(ctx, xdrrec_create, xdr, sndsize, rcvsize,
6073 handle, rd, wr);
6074 COMMON_INTERCEPTOR_READ_RANGE(ctx, &xdr->x_op, sizeof xdr->x_op);
6076 // We can't allocate a wrapper on the stack, as the handle is used outside
6077 // this stack frame. So we put it on the heap, and keep track of it with
6078 // the HashMap (keyed by x_private). When we later need to xdr_destroy,
6079 // we can index the map, free the wrapper, and then clean the map entry.
6080 XdrRecWrapper *wrap_data =
6081 (XdrRecWrapper *)InternalAlloc(sizeof(XdrRecWrapper));
6082 wrap_data->handle = handle;
6083 wrap_data->rd = rd;
6084 wrap_data->wr = wr;
6085 if (wr)
6086 wr = xdrrec_wr_wrap;
6087 if (rd)
6088 rd = xdrrec_rd_wrap;
6089 handle = (char *)wrap_data;
6091 REAL(xdrrec_create)(xdr, sndsize, rcvsize, handle, rd, wr);
6092 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, xdr, sizeof *xdr);
6094 XdrRecWrapMap::Handle wrap(xdrrec_wrap_map, xdr->x_private, false, true);
6095 *wrap = wrap_data;
6096 }
6098 // We have to intercept this to be able to free wrapper memory;
6099 // otherwise it's not necessary.
6100 INTERCEPTOR(void, xdr_destroy, __sanitizer_XDR *xdr) {
6101 void *ctx;
6102 COMMON_INTERCEPTOR_ENTER(ctx, xdr_destroy, xdr);
6104 XdrRecWrapMap::Handle wrap(xdrrec_wrap_map, xdr->x_private, true);
6105 InternalFree(*wrap);
6106 REAL(xdr_destroy)(xdr);
6107 }
6108 #define INIT_XDRREC_LINUX \
6109 static u64 xdrrec_wrap_mem[sizeof(XdrRecWrapMap) / sizeof(u64) + 1]; \
6110 xdrrec_wrap_map = new ((void *)&xdrrec_wrap_mem) XdrRecWrapMap(); \
6111 COMMON_INTERCEPT_FUNCTION(xdrrec_create); \
6112 COMMON_INTERCEPT_FUNCTION(xdr_destroy);
6113 #else
6114 #define INIT_XDRREC_LINUX
6115 #endif
6117 #if SANITIZER_INTERCEPT_TSEARCH
6118 INTERCEPTOR(void *, tsearch, void *key, void **rootp,
6119 int (*compar)(const void *, const void *)) {
6120 void *ctx;
6121 COMMON_INTERCEPTOR_ENTER(ctx, tsearch, key, rootp, compar);
6122 // FIXME: under ASan the call below may write to freed memory and corrupt
6123 // its metadata. See
6124 // https://github.com/google/sanitizers/issues/321.
6125 void *res = REAL(tsearch)(key, rootp, compar);
6126 if (res && *(void **)res == key)
6127 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, sizeof(void *));
6128 return res;
6129 }
6130 #define INIT_TSEARCH COMMON_INTERCEPT_FUNCTION(tsearch);
6131 #else
6132 #define INIT_TSEARCH
6133 #endif
6135 #if SANITIZER_INTERCEPT_LIBIO_INTERNALS || SANITIZER_INTERCEPT_FOPEN || \
6136 SANITIZER_INTERCEPT_OPEN_MEMSTREAM
6137 void unpoison_file(__sanitizer_FILE *fp) {
6138 #if SANITIZER_HAS_STRUCT_FILE
6139 COMMON_INTERCEPTOR_INITIALIZE_RANGE(fp, sizeof(*fp));
6140 #if SANITIZER_NETBSD
6141 if (fp->_bf._base && fp->_bf._size > 0)
6142 COMMON_INTERCEPTOR_INITIALIZE_RANGE(fp->_bf._base,
6143 fp->_bf._size);
6144 #else
6145 if (fp->_IO_read_base && fp->_IO_read_base < fp->_IO_read_end)
6146 COMMON_INTERCEPTOR_INITIALIZE_RANGE(fp->_IO_read_base,
6147 fp->_IO_read_end - fp->_IO_read_base);
6148 if (fp->_IO_write_base && fp->_IO_write_base < fp->_IO_write_end)
6149 COMMON_INTERCEPTOR_INITIALIZE_RANGE(fp->_IO_write_base,
6150 fp->_IO_write_end - fp->_IO_write_base);
6151 #endif
6152 #endif // SANITIZER_HAS_STRUCT_FILE
6153 }
6154 #endif
6156 #if SANITIZER_INTERCEPT_LIBIO_INTERNALS
6157 // These guys are called when a .c source is built with -O2.
6158 INTERCEPTOR(int, __uflow, __sanitizer_FILE *fp) {
6159 void *ctx;
6160 COMMON_INTERCEPTOR_ENTER(ctx, __uflow, fp);
6161 int res = REAL(__uflow)(fp);
6162 unpoison_file(fp);
6163 return res;
6164 }
6165 INTERCEPTOR(int, __underflow, __sanitizer_FILE *fp) {
6166 void *ctx;
6167 COMMON_INTERCEPTOR_ENTER(ctx, __underflow, fp);
6168 int res = REAL(__underflow)(fp);
6169 unpoison_file(fp);
6170 return res;
6171 }
6172 INTERCEPTOR(int, __overflow, __sanitizer_FILE *fp, int ch) {
6173 void *ctx;
6174 COMMON_INTERCEPTOR_ENTER(ctx, __overflow, fp, ch);
6175 int res = REAL(__overflow)(fp, ch);
6176 unpoison_file(fp);
6177 return res;
6178 }
6179 INTERCEPTOR(int, __wuflow, __sanitizer_FILE *fp) {
6180 void *ctx;
6181 COMMON_INTERCEPTOR_ENTER(ctx, __wuflow, fp);
6182 int res = REAL(__wuflow)(fp);
6183 unpoison_file(fp);
6184 return res;
6185 }
6186 INTERCEPTOR(int, __wunderflow, __sanitizer_FILE *fp) {
6187 void *ctx;
6188 COMMON_INTERCEPTOR_ENTER(ctx, __wunderflow, fp);
6189 int res = REAL(__wunderflow)(fp);
6190 unpoison_file(fp);
6191 return res;
6192 }
6193 INTERCEPTOR(int, __woverflow, __sanitizer_FILE *fp, int ch) {
6194 void *ctx;
6195 COMMON_INTERCEPTOR_ENTER(ctx, __woverflow, fp, ch);
6196 int res = REAL(__woverflow)(fp, ch);
6197 unpoison_file(fp);
6198 return res;
6199 }
6200 #define INIT_LIBIO_INTERNALS \
6201 COMMON_INTERCEPT_FUNCTION(__uflow); \
6202 COMMON_INTERCEPT_FUNCTION(__underflow); \
6203 COMMON_INTERCEPT_FUNCTION(__overflow); \
6204 COMMON_INTERCEPT_FUNCTION(__wuflow); \
6205 COMMON_INTERCEPT_FUNCTION(__wunderflow); \
6206 COMMON_INTERCEPT_FUNCTION(__woverflow);
6207 #else
6208 #define INIT_LIBIO_INTERNALS
6209 #endif
6211 #if SANITIZER_INTERCEPT_FOPEN
6212 INTERCEPTOR(__sanitizer_FILE *, fopen, const char *path, const char *mode) {
6213 void *ctx;
6214 COMMON_INTERCEPTOR_ENTER(ctx, fopen, path, mode);
6215 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
6216 COMMON_INTERCEPTOR_READ_RANGE(ctx, mode, internal_strlen(mode) + 1);
6217 __sanitizer_FILE *res = REAL(fopen)(path, mode);
6218 COMMON_INTERCEPTOR_FILE_OPEN(ctx, res, path);
6219 if (res) unpoison_file(res);
6220 return res;
6221 }
6222 INTERCEPTOR(__sanitizer_FILE *, fdopen, int fd, const char *mode) {
6223 void *ctx;
6224 COMMON_INTERCEPTOR_ENTER(ctx, fdopen, fd, mode);
6225 COMMON_INTERCEPTOR_READ_RANGE(ctx, mode, internal_strlen(mode) + 1);
6226 __sanitizer_FILE *res = REAL(fdopen)(fd, mode);
6227 if (res) unpoison_file(res);
6228 return res;
6229 }
6230 INTERCEPTOR(__sanitizer_FILE *, freopen, const char *path, const char *mode,
6231 __sanitizer_FILE *fp) {
6232 void *ctx;
6233 COMMON_INTERCEPTOR_ENTER(ctx, freopen, path, mode, fp);
6234 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
6235 COMMON_INTERCEPTOR_READ_RANGE(ctx, mode, internal_strlen(mode) + 1);
6236 COMMON_INTERCEPTOR_FILE_CLOSE(ctx, fp);
6237 __sanitizer_FILE *res = REAL(freopen)(path, mode, fp);
6238 COMMON_INTERCEPTOR_FILE_OPEN(ctx, res, path);
6239 if (res) unpoison_file(res);
6240 return res;
6241 }
6242 #define INIT_FOPEN \
6243 COMMON_INTERCEPT_FUNCTION(fopen); \
6244 COMMON_INTERCEPT_FUNCTION(fdopen); \
6245 COMMON_INTERCEPT_FUNCTION(freopen);
6246 #else
6247 #define INIT_FOPEN
6248 #endif
6250 #if SANITIZER_INTERCEPT_FLOPEN
6251 INTERCEPTOR(int, flopen, const char *path, int flags, ...) {
6252 void *ctx;
6253 va_list ap;
6254 va_start(ap, flags);
6255 u16 mode = static_cast<u16>(va_arg(ap, u32));
6256 va_end(ap);
6257 COMMON_INTERCEPTOR_ENTER(ctx, flopen, path, flags, mode);
6258 if (path) {
6259 COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
6260 }
6261 return COMMON_INTERCEPTOR_BLOCK_REAL(flopen)(path, flags, mode);
6262 }
6264 INTERCEPTOR(int, flopenat, int dirfd, const char *path, int flags, ...) {
6265 void *ctx;
6266 va_list ap;
6267 va_start(ap, flags);
6268 u16 mode = static_cast<u16>(va_arg(ap, u32));
6269 va_end(ap);
6270 COMMON_INTERCEPTOR_ENTER(ctx, flopen, path, flags, mode);
6271 if (path) {
6272 COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
6273 }
6274 return COMMON_INTERCEPTOR_BLOCK_REAL(flopenat)(dirfd, path, flags, mode);
6275 }
6277 #define INIT_FLOPEN \
6278 COMMON_INTERCEPT_FUNCTION(flopen); \
6279 COMMON_INTERCEPT_FUNCTION(flopenat);
6280 #else
6281 #define INIT_FLOPEN
6282 #endif
6284 #if SANITIZER_INTERCEPT_FOPEN64
6285 INTERCEPTOR(__sanitizer_FILE *, fopen64, const char *path, const char *mode) {
6286 void *ctx;
6287 COMMON_INTERCEPTOR_ENTER(ctx, fopen64, path, mode);
6288 COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
6289 COMMON_INTERCEPTOR_READ_RANGE(ctx, mode, internal_strlen(mode) + 1);
6290 __sanitizer_FILE *res = REAL(fopen64)(path, mode);
6291 COMMON_INTERCEPTOR_FILE_OPEN(ctx, res, path);
6292 if (res) unpoison_file(res);
6293 return res;
6294 }
6295 INTERCEPTOR(__sanitizer_FILE *, freopen64, const char *path, const char *mode,
6296 __sanitizer_FILE *fp) {
6297 void *ctx;
6298 COMMON_INTERCEPTOR_ENTER(ctx, freopen64, path, mode, fp);
6299 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
6300 COMMON_INTERCEPTOR_READ_RANGE(ctx, mode, internal_strlen(mode) + 1);
6301 COMMON_INTERCEPTOR_FILE_CLOSE(ctx, fp);
6302 __sanitizer_FILE *res = REAL(freopen64)(path, mode, fp);
6303 COMMON_INTERCEPTOR_FILE_OPEN(ctx, res, path);
6304 if (res) unpoison_file(res);
6305 return res;
6306 }
6307 #define INIT_FOPEN64 \
6308 COMMON_INTERCEPT_FUNCTION(fopen64); \
6309 COMMON_INTERCEPT_FUNCTION(freopen64);
6310 #else
6311 #define INIT_FOPEN64
6312 #endif
6314 #if SANITIZER_INTERCEPT_OPEN_MEMSTREAM
6315 INTERCEPTOR(__sanitizer_FILE *, open_memstream, char **ptr, SIZE_T *sizeloc) {
6316 void *ctx;
6317 COMMON_INTERCEPTOR_ENTER(ctx, open_memstream, ptr, sizeloc);
6318 // FIXME: under ASan the call below may write to freed memory and corrupt
6319 // its metadata. See
6320 // https://github.com/google/sanitizers/issues/321.
6321 __sanitizer_FILE *res = REAL(open_memstream)(ptr, sizeloc);
6322 if (res) {
6323 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ptr, sizeof(*ptr));
6324 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, sizeloc, sizeof(*sizeloc));
6325 unpoison_file(res);
6326 FileMetadata file = {ptr, sizeloc};
6327 SetInterceptorMetadata(res, file);
6328 }
6329 return res;
6330 }
6331 INTERCEPTOR(__sanitizer_FILE *, open_wmemstream, wchar_t **ptr,
6332 SIZE_T *sizeloc) {
6333 void *ctx;
6334 COMMON_INTERCEPTOR_ENTER(ctx, open_wmemstream, ptr, sizeloc);
6335 __sanitizer_FILE *res = REAL(open_wmemstream)(ptr, sizeloc);
6336 if (res) {
6337 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ptr, sizeof(*ptr));
6338 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, sizeloc, sizeof(*sizeloc));
6339 unpoison_file(res);
6340 FileMetadata file = {(char **)ptr, sizeloc};
6341 SetInterceptorMetadata(res, file);
6342 }
6343 return res;
6344 }
6345 INTERCEPTOR(__sanitizer_FILE *, fmemopen, void *buf, SIZE_T size,
6346 const char *mode) {
6347 void *ctx;
6348 COMMON_INTERCEPTOR_ENTER(ctx, fmemopen, buf, size, mode);
6349 // FIXME: under ASan the call below may write to freed memory and corrupt
6350 // its metadata. See
6351 // https://github.com/google/sanitizers/issues/321.
6352 __sanitizer_FILE *res = REAL(fmemopen)(buf, size, mode);
6353 if (res) unpoison_file(res);
6354 return res;
6355 }
6356 #define INIT_OPEN_MEMSTREAM \
6357 COMMON_INTERCEPT_FUNCTION(open_memstream); \
6358 COMMON_INTERCEPT_FUNCTION(open_wmemstream); \
6359 COMMON_INTERCEPT_FUNCTION(fmemopen);
6360 #else
6361 #define INIT_OPEN_MEMSTREAM
6362 #endif
6364 #if SANITIZER_INTERCEPT_OBSTACK
6365 static void initialize_obstack(__sanitizer_obstack *obstack) {
6366 COMMON_INTERCEPTOR_INITIALIZE_RANGE(obstack, sizeof(*obstack));
6367 if (obstack->chunk)
6368 COMMON_INTERCEPTOR_INITIALIZE_RANGE(obstack->chunk,
6369 sizeof(*obstack->chunk));
6370 }
6372 INTERCEPTOR(int, _obstack_begin_1, __sanitizer_obstack *obstack, int sz,
6373 int align, void *(*alloc_fn)(uptr arg, uptr sz),
6374 void (*free_fn)(uptr arg, void *p)) {
6375 void *ctx;
6376 COMMON_INTERCEPTOR_ENTER(ctx, _obstack_begin_1, obstack, sz, align, alloc_fn,
6377 free_fn);
6378 int res = REAL(_obstack_begin_1)(obstack, sz, align, alloc_fn, free_fn);
6379 if (res) initialize_obstack(obstack);
6380 return res;
6381 }
6382 INTERCEPTOR(int, _obstack_begin, __sanitizer_obstack *obstack, int sz,
6383 int align, void *(*alloc_fn)(uptr sz), void (*free_fn)(void *p)) {
6384 void *ctx;
6385 COMMON_INTERCEPTOR_ENTER(ctx, _obstack_begin, obstack, sz, align, alloc_fn,
6386 free_fn);
6387 int res = REAL(_obstack_begin)(obstack, sz, align, alloc_fn, free_fn);
6388 if (res) initialize_obstack(obstack);
6389 return res;
6390 }
6391 INTERCEPTOR(void, _obstack_newchunk, __sanitizer_obstack *obstack, int length) {
6392 void *ctx;
6393 COMMON_INTERCEPTOR_ENTER(ctx, _obstack_newchunk, obstack, length);
6394 REAL(_obstack_newchunk)(obstack, length);
6395 if (obstack->chunk)
6396 COMMON_INTERCEPTOR_INITIALIZE_RANGE(
6397 obstack->chunk, obstack->next_free - (char *)obstack->chunk);
6398 }
6399 #define INIT_OBSTACK \
6400 COMMON_INTERCEPT_FUNCTION(_obstack_begin_1); \
6401 COMMON_INTERCEPT_FUNCTION(_obstack_begin); \
6402 COMMON_INTERCEPT_FUNCTION(_obstack_newchunk);
6403 #else
6404 #define INIT_OBSTACK
6405 #endif
6407 #if SANITIZER_INTERCEPT_FFLUSH
6408 INTERCEPTOR(int, fflush, __sanitizer_FILE *fp) {
6409 void *ctx;
6410 COMMON_INTERCEPTOR_ENTER(ctx, fflush, fp);
6411 if (fp)
6412 unpoison_file(fp);
6413 int res = REAL(fflush)(fp);
6414 // FIXME: handle fp == NULL
6415 if (fp) {
6416 const FileMetadata *m = GetInterceptorMetadata(fp);
6417 if (m) COMMON_INTERCEPTOR_INITIALIZE_RANGE(*m->addr, *m->size);
6418 }
6419 return res;
6420 }
6421 #define INIT_FFLUSH COMMON_INTERCEPT_FUNCTION(fflush);
6422 #else
6423 #define INIT_FFLUSH
6424 #endif
6426 #if SANITIZER_INTERCEPT_FCLOSE
6427 INTERCEPTOR(int, fclose, __sanitizer_FILE *fp) {
6428 void *ctx;
6429 COMMON_INTERCEPTOR_ENTER(ctx, fclose, fp);
6430 COMMON_INTERCEPTOR_FILE_CLOSE(ctx, fp);
6431 const FileMetadata *m = GetInterceptorMetadata(fp);
6432 if (fp)
6433 unpoison_file(fp);
6434 int res = REAL(fclose)(fp);
6435 if (m) {
6436 COMMON_INTERCEPTOR_INITIALIZE_RANGE(*m->addr, *m->size);
6437 DeleteInterceptorMetadata(fp);
6438 }
6439 return res;
6440 }
6441 #define INIT_FCLOSE COMMON_INTERCEPT_FUNCTION(fclose);
6442 #else
6443 #define INIT_FCLOSE
6444 #endif
6446 #if SANITIZER_INTERCEPT_DLOPEN_DLCLOSE
6447 INTERCEPTOR(void*, dlopen, const char *filename, int flag) {
6448 void *ctx;
6449 COMMON_INTERCEPTOR_ENTER_NOIGNORE(ctx, dlopen, filename, flag);
6451 if (filename) {
6452 COMMON_INTERCEPTOR_READ_STRING(ctx, filename, 0);
6454 # if !SANITIZER_DYNAMIC
6455 // We care about a very specific use-case: dladdr on
6456 // statically-linked ASan may return <main program>
6457 // instead of the library.
6458 // We therefore only take effect if the sanitizer is statically
6459 // linked, and we don't bother canonicalizing paths because
6460 // dladdr should return the same address both times (we assume
6461 // the user did not canonicalize the result from dladdr).
6462 if (common_flags()->test_only_replace_dlopen_main_program) {
6463 VPrintf(1, "dlopen interceptor: filename: %s\n", filename);
6465 const char *SelfFName = DladdrSelfFName();
6466 VPrintf(1, "dlopen interceptor: DladdrSelfFName: %p %s\n",
6467 (const void *)SelfFName, SelfFName);
6469 if (SelfFName && internal_strcmp(SelfFName, filename) == 0) {
6470 // It's possible they copied the string from dladdr, so
6471 // we do a string comparison rather than pointer comparison.
6472 VPrintf(1, "dlopen interceptor: replacing %s because it matches %s\n",
6473 filename, SelfFName);
6474 filename = (char *)0; // RTLD_DEFAULT
6475 }
6476 }
6477 # endif // !SANITIZER_DYNAMIC
6478 }
6480 void *res = COMMON_INTERCEPTOR_DLOPEN(filename, flag);
6481 Symbolizer::GetOrInit()->InvalidateModuleList();
6482 COMMON_INTERCEPTOR_LIBRARY_LOADED(filename, res);
6483 return res;
6484 }
6486 INTERCEPTOR(int, dlclose, void *handle) {
6487 void *ctx;
6488 COMMON_INTERCEPTOR_ENTER_NOIGNORE(ctx, dlclose, handle);
6489 int res = REAL(dlclose)(handle);
6490 Symbolizer::GetOrInit()->InvalidateModuleList();
6491 COMMON_INTERCEPTOR_LIBRARY_UNLOADED();
6492 return res;
6493 }
6494 #define INIT_DLOPEN_DLCLOSE \
6495 COMMON_INTERCEPT_FUNCTION(dlopen); \
6496 COMMON_INTERCEPT_FUNCTION(dlclose);
6497 #else
6498 #define INIT_DLOPEN_DLCLOSE
6499 #endif
6501 #if SANITIZER_INTERCEPT_GETPASS
6502 INTERCEPTOR(char *, getpass, const char *prompt) {
6503 void *ctx;
6504 COMMON_INTERCEPTOR_ENTER(ctx, getpass, prompt);
6505 if (prompt)
6506 COMMON_INTERCEPTOR_READ_RANGE(ctx, prompt, internal_strlen(prompt)+1);
6507 char *res = REAL(getpass)(prompt);
6508 if (res) COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, internal_strlen(res)+1);
6509 return res;
6510 }
6512 #define INIT_GETPASS COMMON_INTERCEPT_FUNCTION(getpass);
6513 #else
6514 #define INIT_GETPASS
6515 #endif
6517 #if SANITIZER_INTERCEPT_TIMERFD
6518 INTERCEPTOR(int, timerfd_settime, int fd, int flags, void *new_value,
6519 void *old_value) {
6520 void *ctx;
6521 COMMON_INTERCEPTOR_ENTER(ctx, timerfd_settime, fd, flags, new_value,
6522 old_value);
6523 COMMON_INTERCEPTOR_READ_RANGE(ctx, new_value, struct_itimerspec_sz);
6524 int res = REAL(timerfd_settime)(fd, flags, new_value, old_value);
6525 if (res != -1 && old_value)
6526 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, old_value, struct_itimerspec_sz);
6527 return res;
6528 }
6530 INTERCEPTOR(int, timerfd_gettime, int fd, void *curr_value) {
6531 void *ctx;
6532 COMMON_INTERCEPTOR_ENTER(ctx, timerfd_gettime, fd, curr_value);
6533 int res = REAL(timerfd_gettime)(fd, curr_value);
6534 if (res != -1 && curr_value)
6535 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, curr_value, struct_itimerspec_sz);
6536 return res;
6537 }
6538 #define INIT_TIMERFD \
6539 COMMON_INTERCEPT_FUNCTION(timerfd_settime); \
6540 COMMON_INTERCEPT_FUNCTION(timerfd_gettime);
6541 #else
6542 #define INIT_TIMERFD
6543 #endif
6545 #if SANITIZER_INTERCEPT_MLOCKX
6546 // Linux kernel has a bug that leads to kernel deadlock if a process
6547 // maps TBs of memory and then calls mlock().
6548 static void MlockIsUnsupported() {
6549 static atomic_uint8_t printed;
6550 if (atomic_exchange(&printed, 1, memory_order_relaxed))
6551 return;
6552 VPrintf(1, "%s ignores mlock/mlockall/munlock/munlockall\n",
6553 SanitizerToolName);
6554 }
6556 INTERCEPTOR(int, mlock, const void *addr, uptr len) {
6557 MlockIsUnsupported();
6558 return 0;
6559 }
6561 INTERCEPTOR(int, munlock, const void *addr, uptr len) {
6562 MlockIsUnsupported();
6563 return 0;
6564 }
6566 INTERCEPTOR(int, mlockall, int flags) {
6567 MlockIsUnsupported();
6568 return 0;
6569 }
6571 INTERCEPTOR(int, munlockall, void) {
6572 MlockIsUnsupported();
6573 return 0;
6574 }
6576 #define INIT_MLOCKX \
6577 COMMON_INTERCEPT_FUNCTION(mlock); \
6578 COMMON_INTERCEPT_FUNCTION(munlock); \
6579 COMMON_INTERCEPT_FUNCTION(mlockall); \
6580 COMMON_INTERCEPT_FUNCTION(munlockall);
6582 #else
6583 #define INIT_MLOCKX
6584 #endif // SANITIZER_INTERCEPT_MLOCKX
6586 #if SANITIZER_INTERCEPT_FOPENCOOKIE
6587 struct WrappedCookie {
6588 void *real_cookie;
6589 __sanitizer_cookie_io_functions_t real_io_funcs;
6590 };
6592 static uptr wrapped_read(void *cookie, char *buf, uptr size) {
6593 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
6594 WrappedCookie *wrapped_cookie = (WrappedCookie *)cookie;
6595 __sanitizer_cookie_io_read real_read = wrapped_cookie->real_io_funcs.read;
6596 return real_read ? real_read(wrapped_cookie->real_cookie, buf, size) : 0;
6597 }
6599 static uptr wrapped_write(void *cookie, const char *buf, uptr size) {
6600 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
6601 WrappedCookie *wrapped_cookie = (WrappedCookie *)cookie;
6602 __sanitizer_cookie_io_write real_write = wrapped_cookie->real_io_funcs.write;
6603 return real_write ? real_write(wrapped_cookie->real_cookie, buf, size) : size;
6604 }
6606 static int wrapped_seek(void *cookie, u64 *offset, int whence) {
6607 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
6608 COMMON_INTERCEPTOR_INITIALIZE_RANGE(offset, sizeof(*offset));
6609 WrappedCookie *wrapped_cookie = (WrappedCookie *)cookie;
6610 __sanitizer_cookie_io_seek real_seek = wrapped_cookie->real_io_funcs.seek;
6611 return real_seek ? real_seek(wrapped_cookie->real_cookie, offset, whence)
6612 : -1;
6613 }
6615 static int wrapped_close(void *cookie) {
6616 COMMON_INTERCEPTOR_UNPOISON_PARAM(1);
6617 WrappedCookie *wrapped_cookie = (WrappedCookie *)cookie;
6618 __sanitizer_cookie_io_close real_close = wrapped_cookie->real_io_funcs.close;
6619 int res = real_close ? real_close(wrapped_cookie->real_cookie) : 0;
6620 InternalFree(wrapped_cookie);
6621 return res;
6622 }
6624 INTERCEPTOR(__sanitizer_FILE *, fopencookie, void *cookie, const char *mode,
6625 __sanitizer_cookie_io_functions_t io_funcs) {
6626 void *ctx;
6627 COMMON_INTERCEPTOR_ENTER(ctx, fopencookie, cookie, mode, io_funcs);
6628 WrappedCookie *wrapped_cookie =
6629 (WrappedCookie *)InternalAlloc(sizeof(WrappedCookie));
6630 wrapped_cookie->real_cookie = cookie;
6631 wrapped_cookie->real_io_funcs = io_funcs;
6632 __sanitizer_FILE *res =
6633 REAL(fopencookie)(wrapped_cookie, mode, {wrapped_read, wrapped_write,
6634 wrapped_seek, wrapped_close});
6635 return res;
6636 }
6638 #define INIT_FOPENCOOKIE COMMON_INTERCEPT_FUNCTION(fopencookie);
6639 #else
6640 #define INIT_FOPENCOOKIE
6641 #endif // SANITIZER_INTERCEPT_FOPENCOOKIE
6643 #if SANITIZER_INTERCEPT_SEM
6644 INTERCEPTOR(int, sem_init, __sanitizer_sem_t *s, int pshared, unsigned value) {
6645 void *ctx;
6646 COMMON_INTERCEPTOR_ENTER(ctx, sem_init, s, pshared, value);
6647 // Workaround a bug in glibc's "old" semaphore implementation by
6648 // zero-initializing the sem_t contents. This has to be done here because
6649 // interceptors bind to the lowest version before glibc 2.36, hitting the
6650 // buggy code path while the non-sanitized build of the same code works fine.
6651 REAL(memset)(s, 0, sizeof(*s));
6652 int res = REAL(sem_init)(s, pshared, value);
6653 return res;
6654 }
6656 INTERCEPTOR(int, sem_destroy, __sanitizer_sem_t *s) {
6657 void *ctx;
6658 COMMON_INTERCEPTOR_ENTER(ctx, sem_destroy, s);
6659 int res = REAL(sem_destroy)(s);
6660 return res;
6661 }
6663 INTERCEPTOR(int, sem_wait, __sanitizer_sem_t *s) {
6664 void *ctx;
6665 COMMON_INTERCEPTOR_ENTER(ctx, sem_wait, s);
6666 int res = COMMON_INTERCEPTOR_BLOCK_REAL(sem_wait)(s);
6667 if (res == 0) {
6668 COMMON_INTERCEPTOR_ACQUIRE(ctx, (uptr)s);
6669 }
6670 return res;
6671 }
6673 INTERCEPTOR(int, sem_trywait, __sanitizer_sem_t *s) {
6674 void *ctx;
6675 COMMON_INTERCEPTOR_ENTER(ctx, sem_trywait, s);
6676 int res = REAL(sem_trywait)(s);
6677 if (res == 0) {
6678 COMMON_INTERCEPTOR_ACQUIRE(ctx, (uptr)s);
6679 }
6680 return res;
6681 }
6683 INTERCEPTOR(int, sem_timedwait, __sanitizer_sem_t *s, void *abstime) {
6684 void *ctx;
6685 COMMON_INTERCEPTOR_ENTER(ctx, sem_timedwait, s, abstime);
6686 COMMON_INTERCEPTOR_READ_RANGE(ctx, abstime, struct_timespec_sz);
6687 int res = COMMON_INTERCEPTOR_BLOCK_REAL(sem_timedwait)(s, abstime);
6688 if (res == 0) {
6689 COMMON_INTERCEPTOR_ACQUIRE(ctx, (uptr)s);
6690 }
6691 return res;
6692 }
6694 INTERCEPTOR(int, sem_post, __sanitizer_sem_t *s) {
6695 void *ctx;
6696 COMMON_INTERCEPTOR_ENTER(ctx, sem_post, s);
6697 COMMON_INTERCEPTOR_RELEASE(ctx, (uptr)s);
6698 int res = REAL(sem_post)(s);
6699 return res;
6700 }
6702 INTERCEPTOR(int, sem_getvalue, __sanitizer_sem_t *s, int *sval) {
6703 void *ctx;
6704 COMMON_INTERCEPTOR_ENTER(ctx, sem_getvalue, s, sval);
6705 int res = REAL(sem_getvalue)(s, sval);
6706 if (res == 0) {
6707 COMMON_INTERCEPTOR_ACQUIRE(ctx, (uptr)s);
6708 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, sval, sizeof(*sval));
6709 }
6710 return res;
6711 }
6713 INTERCEPTOR(__sanitizer_sem_t *, sem_open, const char *name, int oflag, ...) {
6714 void *ctx;
6715 va_list ap;
6716 va_start(ap, oflag);
6717 u32 mode = va_arg(ap, u32);
6718 u32 value = va_arg(ap, u32);
6719 COMMON_INTERCEPTOR_ENTER(ctx, sem_open, name, oflag, mode, value);
6720 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
6721 __sanitizer_sem_t *s = REAL(sem_open)(name, oflag, mode, value);
6722 if (s)
6723 COMMON_INTERCEPTOR_INITIALIZE_RANGE(s, sizeof(*s));
6724 va_end(ap);
6725 return s;
6726 }
6728 INTERCEPTOR(int, sem_unlink, const char *name) {
6729 void *ctx;
6730 COMMON_INTERCEPTOR_ENTER(ctx, sem_unlink, name);
6731 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
6732 return REAL(sem_unlink)(name);
6733 }
6735 # define INIT_SEM \
6736 COMMON_INTERCEPT_FUNCTION(sem_init); \
6737 COMMON_INTERCEPT_FUNCTION(sem_destroy); \
6738 COMMON_INTERCEPT_FUNCTION(sem_wait); \
6739 COMMON_INTERCEPT_FUNCTION(sem_trywait); \
6740 COMMON_INTERCEPT_FUNCTION(sem_timedwait); \
6741 COMMON_INTERCEPT_FUNCTION(sem_post); \
6742 COMMON_INTERCEPT_FUNCTION(sem_getvalue); \
6743 COMMON_INTERCEPT_FUNCTION(sem_open); \
6744 COMMON_INTERCEPT_FUNCTION(sem_unlink);
6745 #else
6746 # define INIT_SEM
6747 #endif // SANITIZER_INTERCEPT_SEM
6749 #if SANITIZER_INTERCEPT_PTHREAD_SETCANCEL
6750 INTERCEPTOR(int, pthread_setcancelstate, int state, int *oldstate) {
6751 void *ctx;
6752 COMMON_INTERCEPTOR_ENTER(ctx, pthread_setcancelstate, state, oldstate);
6753 int res = REAL(pthread_setcancelstate)(state, oldstate);
6754 if (res == 0 && oldstate != nullptr)
6755 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, oldstate, sizeof(*oldstate));
6756 return res;
6757 }
6759 INTERCEPTOR(int, pthread_setcanceltype, int type, int *oldtype) {
6760 void *ctx;
6761 COMMON_INTERCEPTOR_ENTER(ctx, pthread_setcanceltype, type, oldtype);
6762 int res = REAL(pthread_setcanceltype)(type, oldtype);
6763 if (res == 0 && oldtype != nullptr)
6764 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, oldtype, sizeof(*oldtype));
6765 return res;
6766 }
6767 #define INIT_PTHREAD_SETCANCEL \
6768 COMMON_INTERCEPT_FUNCTION(pthread_setcancelstate); \
6769 COMMON_INTERCEPT_FUNCTION(pthread_setcanceltype);
6770 #else
6771 #define INIT_PTHREAD_SETCANCEL
6772 #endif
6774 #if SANITIZER_INTERCEPT_MINCORE
6775 INTERCEPTOR(int, mincore, void *addr, uptr length, unsigned char *vec) {
6776 void *ctx;
6777 COMMON_INTERCEPTOR_ENTER(ctx, mincore, addr, length, vec);
6778 int res = REAL(mincore)(addr, length, vec);
6779 if (res == 0) {
6780 uptr page_size = GetPageSizeCached();
6781 uptr vec_size = ((length + page_size - 1) & (~(page_size - 1))) / page_size;
6782 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, vec, vec_size);
6783 }
6784 return res;
6785 }
6786 #define INIT_MINCORE COMMON_INTERCEPT_FUNCTION(mincore);
6787 #else
6788 #define INIT_MINCORE
6789 #endif
6791 #if SANITIZER_INTERCEPT_PROCESS_VM_READV
6792 INTERCEPTOR(SSIZE_T, process_vm_readv, int pid, __sanitizer_iovec *local_iov,
6793 uptr liovcnt, __sanitizer_iovec *remote_iov, uptr riovcnt,
6794 uptr flags) {
6795 void *ctx;
6796 COMMON_INTERCEPTOR_ENTER(ctx, process_vm_readv, pid, local_iov, liovcnt,
6797 remote_iov, riovcnt, flags);
6798 SSIZE_T res = REAL(process_vm_readv)(pid, local_iov, liovcnt, remote_iov,
6799 riovcnt, flags);
6800 if (res > 0)
6801 write_iovec(ctx, local_iov, liovcnt, res);
6802 return res;
6803 }
6805 INTERCEPTOR(SSIZE_T, process_vm_writev, int pid, __sanitizer_iovec *local_iov,
6806 uptr liovcnt, __sanitizer_iovec *remote_iov, uptr riovcnt,
6807 uptr flags) {
6808 void *ctx;
6809 COMMON_INTERCEPTOR_ENTER(ctx, process_vm_writev, pid, local_iov, liovcnt,
6810 remote_iov, riovcnt, flags);
6811 SSIZE_T res = REAL(process_vm_writev)(pid, local_iov, liovcnt, remote_iov,
6812 riovcnt, flags);
6813 if (res > 0)
6814 read_iovec(ctx, local_iov, liovcnt, res);
6815 return res;
6816 }
6817 #define INIT_PROCESS_VM_READV \
6818 COMMON_INTERCEPT_FUNCTION(process_vm_readv); \
6819 COMMON_INTERCEPT_FUNCTION(process_vm_writev);
6820 #else
6821 #define INIT_PROCESS_VM_READV
6822 #endif
6824 #if SANITIZER_INTERCEPT_CTERMID
6825 INTERCEPTOR(char *, ctermid, char *s) {
6826 void *ctx;
6827 COMMON_INTERCEPTOR_ENTER(ctx, ctermid, s);
6828 char *res = REAL(ctermid)(s);
6829 if (res) {
6830 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, internal_strlen(res) + 1);
6831 }
6832 return res;
6833 }
6834 #define INIT_CTERMID COMMON_INTERCEPT_FUNCTION(ctermid);
6835 #else
6836 #define INIT_CTERMID
6837 #endif
6839 #if SANITIZER_INTERCEPT_CTERMID_R
6840 INTERCEPTOR(char *, ctermid_r, char *s) {
6841 void *ctx;
6842 COMMON_INTERCEPTOR_ENTER(ctx, ctermid_r, s);
6843 char *res = REAL(ctermid_r)(s);
6844 if (res) {
6845 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, internal_strlen(res) + 1);
6846 }
6847 return res;
6848 }
6849 #define INIT_CTERMID_R COMMON_INTERCEPT_FUNCTION(ctermid_r);
6850 #else
6851 #define INIT_CTERMID_R
6852 #endif
6854 #if SANITIZER_INTERCEPT_RECV_RECVFROM
6855 INTERCEPTOR(SSIZE_T, recv, int fd, void *buf, SIZE_T len, int flags) {
6856 void *ctx;
6857 COMMON_INTERCEPTOR_ENTER(ctx, recv, fd, buf, len, flags);
6858 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
6859 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(recv)(fd, buf, len, flags);
6860 if (res > 0) {
6861 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, Min((SIZE_T)res, len));
6862 }
6863 if (res >= 0 && fd >= 0) COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
6864 return res;
6865 }
6867 INTERCEPTOR(SSIZE_T, recvfrom, int fd, void *buf, SIZE_T len, int flags,
6868 void *srcaddr, int *addrlen) {
6869 void *ctx;
6870 COMMON_INTERCEPTOR_ENTER(ctx, recvfrom, fd, buf, len, flags, srcaddr,
6871 addrlen);
6872 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
6873 SIZE_T srcaddr_sz;
6874 if (srcaddr) srcaddr_sz = *addrlen;
6875 (void)srcaddr_sz; // prevent "set but not used" warning
6876 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(recvfrom)(fd, buf, len, flags,
6877 srcaddr, addrlen);
6878 if (res > 0)
6879 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, Min((SIZE_T)res, len));
6880 if (res >= 0 && srcaddr)
6881 COMMON_INTERCEPTOR_INITIALIZE_RANGE(srcaddr,
6882 Min((SIZE_T)*addrlen, srcaddr_sz));
6883 return res;
6884 }
6885 #define INIT_RECV_RECVFROM \
6886 COMMON_INTERCEPT_FUNCTION(recv); \
6887 COMMON_INTERCEPT_FUNCTION(recvfrom);
6888 #else
6889 #define INIT_RECV_RECVFROM
6890 #endif
6892 #if SANITIZER_INTERCEPT_SEND_SENDTO
6893 INTERCEPTOR(SSIZE_T, send, int fd, void *buf, SIZE_T len, int flags) {
6894 void *ctx;
6895 COMMON_INTERCEPTOR_ENTER(ctx, send, fd, buf, len, flags);
6896 if (fd >= 0) {
6897 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
6898 COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd);
6899 }
6900 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(send)(fd, buf, len, flags);
6901 if (common_flags()->intercept_send && res > 0)
6902 COMMON_INTERCEPTOR_READ_RANGE(ctx, buf, Min((SIZE_T)res, len));
6903 return res;
6904 }
6906 INTERCEPTOR(SSIZE_T, sendto, int fd, void *buf, SIZE_T len, int flags,
6907 void *dstaddr, int addrlen) {
6908 void *ctx;
6909 COMMON_INTERCEPTOR_ENTER(ctx, sendto, fd, buf, len, flags, dstaddr, addrlen);
6910 if (fd >= 0) {
6911 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
6912 COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd);
6913 }
6914 // Can't check dstaddr as it may have uninitialized padding at the end.
6915 SSIZE_T res = COMMON_INTERCEPTOR_BLOCK_REAL(sendto)(fd, buf, len, flags,
6916 dstaddr, addrlen);
6917 if (common_flags()->intercept_send && res > 0)
6918 COMMON_INTERCEPTOR_READ_RANGE(ctx, buf, Min((SIZE_T)res, len));
6919 return res;
6920 }
6921 #define INIT_SEND_SENDTO \
6922 COMMON_INTERCEPT_FUNCTION(send); \
6923 COMMON_INTERCEPT_FUNCTION(sendto);
6924 #else
6925 #define INIT_SEND_SENDTO
6926 #endif
6928 #if SANITIZER_INTERCEPT_EVENTFD_READ_WRITE
6929 INTERCEPTOR(int, eventfd_read, int fd, __sanitizer_eventfd_t *value) {
6930 void *ctx;
6931 COMMON_INTERCEPTOR_ENTER(ctx, eventfd_read, fd, value);
6932 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
6933 int res = COMMON_INTERCEPTOR_BLOCK_REAL(eventfd_read)(fd, value);
6934 if (res == 0) {
6935 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, value, sizeof(*value));
6936 if (fd >= 0) COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
6937 }
6938 return res;
6939 }
6940 INTERCEPTOR(int, eventfd_write, int fd, __sanitizer_eventfd_t value) {
6941 void *ctx;
6942 COMMON_INTERCEPTOR_ENTER(ctx, eventfd_write, fd, value);
6943 if (fd >= 0) {
6944 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
6945 COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd);
6946 }
6947 int res = COMMON_INTERCEPTOR_BLOCK_REAL(eventfd_write)(fd, value);
6948 return res;
6949 }
6950 #define INIT_EVENTFD_READ_WRITE \
6951 COMMON_INTERCEPT_FUNCTION(eventfd_read); \
6952 COMMON_INTERCEPT_FUNCTION(eventfd_write)
6953 #else
6954 #define INIT_EVENTFD_READ_WRITE
6955 #endif
6957 #if SANITIZER_INTERCEPT_STAT
6958 INTERCEPTOR(int, stat, const char *path, void *buf) {
6959 void *ctx;
6960 COMMON_INTERCEPTOR_ENTER(ctx, stat, path, buf);
6961 if (common_flags()->intercept_stat)
6962 COMMON_INTERCEPTOR_READ_STRING(ctx, path, 0);
6963 int res = REAL(stat)(path, buf);
6964 if (!res)
6965 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, __sanitizer::struct_stat_sz);
6966 return res;
6967 }
6968 #define INIT_STAT COMMON_INTERCEPT_FUNCTION(stat)
6969 #else
6970 #define INIT_STAT
6971 #endif
6973 #if SANITIZER_INTERCEPT_STAT64
6974 INTERCEPTOR(int, stat64, const char *path, void *buf) {
6975 void *ctx;
6976 COMMON_INTERCEPTOR_ENTER(ctx, stat64, path, buf);
6977 if (common_flags()->intercept_stat)
6978 COMMON_INTERCEPTOR_READ_STRING(ctx, path, 0);
6979 int res = REAL(stat64)(path, buf);
6980 if (!res)
6981 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, __sanitizer::struct_stat64_sz);
6982 return res;
6983 }
6984 #define INIT_STAT64 COMMON_INTERCEPT_FUNCTION(stat64)
6985 #else
6986 #define INIT_STAT64
6987 #endif
6990 #if SANITIZER_INTERCEPT_LSTAT
6991 INTERCEPTOR(int, lstat, const char *path, void *buf) {
6992 void *ctx;
6993 COMMON_INTERCEPTOR_ENTER(ctx, lstat, path, buf);
6994 if (common_flags()->intercept_stat)
6995 COMMON_INTERCEPTOR_READ_STRING(ctx, path, 0);
6996 int res = REAL(lstat)(path, buf);
6997 if (!res)
6998 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, __sanitizer::struct_stat_sz);
6999 return res;
7000 }
7001 #define INIT_LSTAT COMMON_INTERCEPT_FUNCTION(lstat)
7002 #else
7003 #define INIT_LSTAT
7004 #endif
7006 #if SANITIZER_INTERCEPT_STAT64
7007 INTERCEPTOR(int, lstat64, const char *path, void *buf) {
7008 void *ctx;
7009 COMMON_INTERCEPTOR_ENTER(ctx, lstat64, path, buf);
7010 if (common_flags()->intercept_stat)
7011 COMMON_INTERCEPTOR_READ_STRING(ctx, path, 0);
7012 int res = REAL(lstat64)(path, buf);
7013 if (!res)
7014 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, __sanitizer::struct_stat64_sz);
7015 return res;
7016 }
7017 #define INIT_LSTAT64 COMMON_INTERCEPT_FUNCTION(lstat64)
7018 #else
7019 #define INIT_LSTAT64
7020 #endif
7022 #if SANITIZER_INTERCEPT___XSTAT
7023 INTERCEPTOR(int, __xstat, int version, const char *path, void *buf) {
7024 void *ctx;
7025 COMMON_INTERCEPTOR_ENTER(ctx, __xstat, version, path, buf);
7026 if (common_flags()->intercept_stat)
7027 COMMON_INTERCEPTOR_READ_STRING(ctx, path, 0);
7028 int res = REAL(__xstat)(version, path, buf);
7029 if (!res)
7030 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, __sanitizer::struct_stat_sz);
7031 return res;
7032 }
7033 #define INIT___XSTAT COMMON_INTERCEPT_FUNCTION(__xstat)
7034 #else
7035 #define INIT___XSTAT
7036 #endif
7038 #if SANITIZER_INTERCEPT___XSTAT64
7039 INTERCEPTOR(int, __xstat64, int version, const char *path, void *buf) {
7040 void *ctx;
7041 COMMON_INTERCEPTOR_ENTER(ctx, __xstat64, version, path, buf);
7042 if (common_flags()->intercept_stat)
7043 COMMON_INTERCEPTOR_READ_STRING(ctx, path, 0);
7044 int res = REAL(__xstat64)(version, path, buf);
7045 if (!res)
7046 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, __sanitizer::struct_stat64_sz);
7047 return res;
7048 }
7049 #define INIT___XSTAT64 COMMON_INTERCEPT_FUNCTION(__xstat64)
7050 #else
7051 #define INIT___XSTAT64
7052 #endif
7054 #if SANITIZER_INTERCEPT___LXSTAT
7055 INTERCEPTOR(int, __lxstat, int version, const char *path, void *buf) {
7056 void *ctx;
7057 COMMON_INTERCEPTOR_ENTER(ctx, __lxstat, version, path, buf);
7058 if (common_flags()->intercept_stat)
7059 COMMON_INTERCEPTOR_READ_STRING(ctx, path, 0);
7060 int res = REAL(__lxstat)(version, path, buf);
7061 if (!res)
7062 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, __sanitizer::struct_stat_sz);
7063 return res;
7064 }
7065 #define INIT___LXSTAT COMMON_INTERCEPT_FUNCTION(__lxstat)
7066 #else
7067 #define INIT___LXSTAT
7068 #endif
7070 #if SANITIZER_INTERCEPT___LXSTAT64
7071 INTERCEPTOR(int, __lxstat64, int version, const char *path, void *buf) {
7072 void *ctx;
7073 COMMON_INTERCEPTOR_ENTER(ctx, __lxstat64, version, path, buf);
7074 if (common_flags()->intercept_stat)
7075 COMMON_INTERCEPTOR_READ_STRING(ctx, path, 0);
7076 int res = REAL(__lxstat64)(version, path, buf);
7077 if (!res)
7078 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, __sanitizer::struct_stat64_sz);
7079 return res;
7080 }
7081 #define INIT___LXSTAT64 COMMON_INTERCEPT_FUNCTION(__lxstat64)
7082 #else
7083 #define INIT___LXSTAT64
7084 #endif
7086 // FIXME: add other *stat interceptor
7088 #if SANITIZER_INTERCEPT_UTMP
7089 INTERCEPTOR(void *, getutent, int dummy) {
7090 void *ctx;
7091 COMMON_INTERCEPTOR_ENTER(ctx, getutent, dummy);
7092 void *res = REAL(getutent)(dummy);
7093 if (res)
7094 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, __sanitizer::struct_utmp_sz);
7095 return res;
7096 }
7097 INTERCEPTOR(void *, getutid, void *ut) {
7098 void *ctx;
7099 COMMON_INTERCEPTOR_ENTER(ctx, getutid, ut);
7100 void *res = REAL(getutid)(ut);
7101 if (res)
7102 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, __sanitizer::struct_utmp_sz);
7103 return res;
7104 }
7105 INTERCEPTOR(void *, getutline, void *ut) {
7106 void *ctx;
7107 COMMON_INTERCEPTOR_ENTER(ctx, getutline, ut);
7108 void *res = REAL(getutline)(ut);
7109 if (res)
7110 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, __sanitizer::struct_utmp_sz);
7111 return res;
7112 }
7113 #define INIT_UTMP \
7114 COMMON_INTERCEPT_FUNCTION(getutent); \
7115 COMMON_INTERCEPT_FUNCTION(getutid); \
7116 COMMON_INTERCEPT_FUNCTION(getutline);
7117 #else
7118 #define INIT_UTMP
7119 #endif
7121 #if SANITIZER_INTERCEPT_UTMPX
7122 INTERCEPTOR(void *, getutxent, int dummy) {
7123 void *ctx;
7124 COMMON_INTERCEPTOR_ENTER(ctx, getutxent, dummy);
7125 void *res = REAL(getutxent)(dummy);
7126 if (res)
7127 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, __sanitizer::struct_utmpx_sz);
7128 return res;
7129 }
7130 INTERCEPTOR(void *, getutxid, void *ut) {
7131 void *ctx;
7132 COMMON_INTERCEPTOR_ENTER(ctx, getutxid, ut);
7133 void *res = REAL(getutxid)(ut);
7134 if (res)
7135 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, __sanitizer::struct_utmpx_sz);
7136 return res;
7137 }
7138 INTERCEPTOR(void *, getutxline, void *ut) {
7139 void *ctx;
7140 COMMON_INTERCEPTOR_ENTER(ctx, getutxline, ut);
7141 void *res = REAL(getutxline)(ut);
7142 if (res)
7143 COMMON_INTERCEPTOR_INITIALIZE_RANGE(res, __sanitizer::struct_utmpx_sz);
7144 return res;
7145 }
7146 INTERCEPTOR(void *, pututxline, const void *ut) {
7147 void *ctx;
7148 COMMON_INTERCEPTOR_ENTER(ctx, pututxline, ut);
7149 if (ut)
7150 COMMON_INTERCEPTOR_READ_RANGE(ctx, ut, __sanitizer::struct_utmpx_sz);
7151 void *res = REAL(pututxline)(ut);
7152 if (res)
7153 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, __sanitizer::struct_utmpx_sz);
7154 return res;
7155 }
7156 #define INIT_UTMPX \
7157 COMMON_INTERCEPT_FUNCTION(getutxent); \
7158 COMMON_INTERCEPT_FUNCTION(getutxid); \
7159 COMMON_INTERCEPT_FUNCTION(getutxline); \
7160 COMMON_INTERCEPT_FUNCTION(pututxline);
7161 #else
7162 #define INIT_UTMPX
7163 #endif
7165 #if SANITIZER_INTERCEPT_GETLOADAVG
7166 INTERCEPTOR(int, getloadavg, double *loadavg, int nelem) {
7167 void *ctx;
7168 COMMON_INTERCEPTOR_ENTER(ctx, getloadavg, loadavg, nelem);
7169 int res = REAL(getloadavg)(loadavg, nelem);
7170 if (res > 0)
7171 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, loadavg, res * sizeof(*loadavg));
7172 return res;
7173 }
7174 #define INIT_GETLOADAVG \
7175 COMMON_INTERCEPT_FUNCTION(getloadavg);
7176 #else
7177 #define INIT_GETLOADAVG
7178 #endif
7180 #if SANITIZER_INTERCEPT_MCHECK_MPROBE
7181 INTERCEPTOR(int, mcheck, void (*abortfunc)(int mstatus)) {
7182 return 0;
7183 }
7185 INTERCEPTOR(int, mcheck_pedantic, void (*abortfunc)(int mstatus)) {
7186 return 0;
7187 }
7189 INTERCEPTOR(int, mprobe, void *ptr) {
7190 return 0;
7191 }
7192 #endif
7194 #if SANITIZER_INTERCEPT_WCSLEN
7195 INTERCEPTOR(SIZE_T, wcslen, const wchar_t *s) {
7196 void *ctx;
7197 COMMON_INTERCEPTOR_ENTER(ctx, wcslen, s);
7198 SIZE_T res = REAL(wcslen)(s);
7199 COMMON_INTERCEPTOR_READ_RANGE(ctx, s, sizeof(wchar_t) * (res + 1));
7200 return res;
7201 }
7203 INTERCEPTOR(SIZE_T, wcsnlen, const wchar_t *s, SIZE_T n) {
7204 void *ctx;
7205 COMMON_INTERCEPTOR_ENTER(ctx, wcsnlen, s, n);
7206 SIZE_T res = REAL(wcsnlen)(s, n);
7207 COMMON_INTERCEPTOR_READ_RANGE(ctx, s, sizeof(wchar_t) * Min(res + 1, n));
7208 return res;
7209 }
7210 #define INIT_WCSLEN \
7211 COMMON_INTERCEPT_FUNCTION(wcslen); \
7212 COMMON_INTERCEPT_FUNCTION(wcsnlen);
7213 #else
7214 #define INIT_WCSLEN
7215 #endif
7217 #if SANITIZER_INTERCEPT_WCSCAT
7218 INTERCEPTOR(wchar_t *, wcscat, wchar_t *dst, const wchar_t *src) {
7219 void *ctx;
7220 COMMON_INTERCEPTOR_ENTER(ctx, wcscat, dst, src);
7221 SIZE_T src_size = internal_wcslen(src);
7222 SIZE_T dst_size = internal_wcslen(dst);
7223 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, (src_size + 1) * sizeof(wchar_t));
7224 COMMON_INTERCEPTOR_READ_RANGE(ctx, dst, (dst_size + 1) * sizeof(wchar_t));
7225 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst + dst_size,
7226 (src_size + 1) * sizeof(wchar_t));
7227 return REAL(wcscat)(dst, src);
7228 }
7230 INTERCEPTOR(wchar_t *, wcsncat, wchar_t *dst, const wchar_t *src, SIZE_T n) {
7231 void *ctx;
7232 COMMON_INTERCEPTOR_ENTER(ctx, wcsncat, dst, src, n);
7233 SIZE_T src_size = internal_wcsnlen(src, n);
7234 SIZE_T dst_size = internal_wcslen(dst);
7235 COMMON_INTERCEPTOR_READ_RANGE(ctx, src,
7236 Min(src_size + 1, n) * sizeof(wchar_t));
7237 COMMON_INTERCEPTOR_READ_RANGE(ctx, dst, (dst_size + 1) * sizeof(wchar_t));
7238 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst + dst_size,
7239 (src_size + 1) * sizeof(wchar_t));
7240 return REAL(wcsncat)(dst, src, n);
7241 }
7242 #define INIT_WCSCAT \
7243 COMMON_INTERCEPT_FUNCTION(wcscat); \
7244 COMMON_INTERCEPT_FUNCTION(wcsncat);
7245 #else
7246 #define INIT_WCSCAT
7247 #endif
7249 #if SANITIZER_INTERCEPT_WCSDUP
7250 INTERCEPTOR(wchar_t *, wcsdup, wchar_t *s) {
7251 void *ctx;
7252 COMMON_INTERCEPTOR_ENTER(ctx, wcsdup, s);
7253 SIZE_T len = internal_wcslen(s);
7254 COMMON_INTERCEPTOR_READ_RANGE(ctx, s, sizeof(wchar_t) * (len + 1));
7255 wchar_t *result = REAL(wcsdup)(s);
7256 if (result)
7257 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof(wchar_t) * (len + 1));
7258 return result;
7259 }
7261 #define INIT_WCSDUP COMMON_INTERCEPT_FUNCTION(wcsdup);
7262 #else
7263 #define INIT_WCSDUP
7264 #endif
7266 #if SANITIZER_INTERCEPT_STRXFRM
7267 static SIZE_T RealStrLen(const char *str) { return internal_strlen(str); }
7269 static SIZE_T RealStrLen(const wchar_t *str) { return internal_wcslen(str); }
7271 #define STRXFRM_INTERCEPTOR_IMPL(strxfrm, dest, src, len, ...) \
7272 { \
7273 void *ctx; \
7274 COMMON_INTERCEPTOR_ENTER(ctx, strxfrm, dest, src, len, ##__VA_ARGS__); \
7275 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, \
7276 sizeof(*src) * (RealStrLen(src) + 1)); \
7277 SIZE_T res = REAL(strxfrm)(dest, src, len, ##__VA_ARGS__); \
7278 if (res < len) \
7279 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dest, sizeof(*src) * (res + 1)); \
7280 return res; \
7281 }
7283 INTERCEPTOR(SIZE_T, strxfrm, char *dest, const char *src, SIZE_T len) {
7284 STRXFRM_INTERCEPTOR_IMPL(strxfrm, dest, src, len);
7285 }
7287 INTERCEPTOR(SIZE_T, strxfrm_l, char *dest, const char *src, SIZE_T len,
7288 void *locale) {
7289 STRXFRM_INTERCEPTOR_IMPL(strxfrm_l, dest, src, len, locale);
7290 }
7292 #define INIT_STRXFRM \
7293 COMMON_INTERCEPT_FUNCTION(strxfrm); \
7294 COMMON_INTERCEPT_FUNCTION(strxfrm_l);
7295 #else
7296 #define INIT_STRXFRM
7297 #endif
7299 #if SANITIZER_INTERCEPT___STRXFRM_L
7300 INTERCEPTOR(SIZE_T, __strxfrm_l, char *dest, const char *src, SIZE_T len,
7301 void *locale) {
7302 STRXFRM_INTERCEPTOR_IMPL(__strxfrm_l, dest, src, len, locale);
7303 }
7305 #define INIT___STRXFRM_L COMMON_INTERCEPT_FUNCTION(__strxfrm_l);
7306 #else
7307 #define INIT___STRXFRM_L
7308 #endif
7310 #if SANITIZER_INTERCEPT_WCSXFRM
7311 INTERCEPTOR(SIZE_T, wcsxfrm, wchar_t *dest, const wchar_t *src, SIZE_T len) {
7312 STRXFRM_INTERCEPTOR_IMPL(wcsxfrm, dest, src, len);
7313 }
7315 INTERCEPTOR(SIZE_T, wcsxfrm_l, wchar_t *dest, const wchar_t *src, SIZE_T len,
7316 void *locale) {
7317 STRXFRM_INTERCEPTOR_IMPL(wcsxfrm_l, dest, src, len, locale);
7318 }
7320 #define INIT_WCSXFRM \
7321 COMMON_INTERCEPT_FUNCTION(wcsxfrm); \
7322 COMMON_INTERCEPT_FUNCTION(wcsxfrm_l);
7323 #else
7324 #define INIT_WCSXFRM
7325 #endif
7327 #if SANITIZER_INTERCEPT___WCSXFRM_L
7328 INTERCEPTOR(SIZE_T, __wcsxfrm_l, wchar_t *dest, const wchar_t *src, SIZE_T len,
7329 void *locale) {
7330 STRXFRM_INTERCEPTOR_IMPL(__wcsxfrm_l, dest, src, len, locale);
7331 }
7333 #define INIT___WCSXFRM_L COMMON_INTERCEPT_FUNCTION(__wcsxfrm_l);
7334 #else
7335 #define INIT___WCSXFRM_L
7336 #endif
7338 #if SANITIZER_INTERCEPT_ACCT
7339 INTERCEPTOR(int, acct, const char *file) {
7340 void *ctx;
7341 COMMON_INTERCEPTOR_ENTER(ctx, acct, file);
7342 if (file)
7343 COMMON_INTERCEPTOR_READ_RANGE(ctx, file, internal_strlen(file) + 1);
7344 return REAL(acct)(file);
7345 }
7346 #define INIT_ACCT COMMON_INTERCEPT_FUNCTION(acct)
7347 #else
7348 #define INIT_ACCT
7349 #endif
7351 #if SANITIZER_INTERCEPT_USER_FROM_UID
7352 INTERCEPTOR(const char *, user_from_uid, u32 uid, int nouser) {
7353 void *ctx;
7354 const char *user;
7355 COMMON_INTERCEPTOR_ENTER(ctx, user_from_uid, uid, nouser);
7356 user = REAL(user_from_uid)(uid, nouser);
7357 if (user)
7358 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, user, internal_strlen(user) + 1);
7359 return user;
7360 }
7361 #define INIT_USER_FROM_UID COMMON_INTERCEPT_FUNCTION(user_from_uid)
7362 #else
7363 #define INIT_USER_FROM_UID
7364 #endif
7366 #if SANITIZER_INTERCEPT_UID_FROM_USER
7367 INTERCEPTOR(int, uid_from_user, const char *name, u32 *uid) {
7368 void *ctx;
7369 int res;
7370 COMMON_INTERCEPTOR_ENTER(ctx, uid_from_user, name, uid);
7371 if (name)
7372 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
7373 res = REAL(uid_from_user)(name, uid);
7374 if (uid)
7375 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, uid, sizeof(*uid));
7376 return res;
7377 }
7378 #define INIT_UID_FROM_USER COMMON_INTERCEPT_FUNCTION(uid_from_user)
7379 #else
7380 #define INIT_UID_FROM_USER
7381 #endif
7383 #if SANITIZER_INTERCEPT_GROUP_FROM_GID
7384 INTERCEPTOR(const char *, group_from_gid, u32 gid, int nogroup) {
7385 void *ctx;
7386 const char *group;
7387 COMMON_INTERCEPTOR_ENTER(ctx, group_from_gid, gid, nogroup);
7388 group = REAL(group_from_gid)(gid, nogroup);
7389 if (group)
7390 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, group, internal_strlen(group) + 1);
7391 return group;
7392 }
7393 #define INIT_GROUP_FROM_GID COMMON_INTERCEPT_FUNCTION(group_from_gid)
7394 #else
7395 #define INIT_GROUP_FROM_GID
7396 #endif
7398 #if SANITIZER_INTERCEPT_GID_FROM_GROUP
7399 INTERCEPTOR(int, gid_from_group, const char *group, u32 *gid) {
7400 void *ctx;
7401 int res;
7402 COMMON_INTERCEPTOR_ENTER(ctx, gid_from_group, group, gid);
7403 if (group)
7404 COMMON_INTERCEPTOR_READ_RANGE(ctx, group, internal_strlen(group) + 1);
7405 res = REAL(gid_from_group)(group, gid);
7406 if (gid)
7407 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, gid, sizeof(*gid));
7408 return res;
7409 }
7410 #define INIT_GID_FROM_GROUP COMMON_INTERCEPT_FUNCTION(gid_from_group)
7411 #else
7412 #define INIT_GID_FROM_GROUP
7413 #endif
7415 #if SANITIZER_INTERCEPT_ACCESS
7416 INTERCEPTOR(int, access, const char *path, int mode) {
7417 void *ctx;
7418 COMMON_INTERCEPTOR_ENTER(ctx, access, path, mode);
7419 if (path)
7420 COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
7421 return REAL(access)(path, mode);
7422 }
7423 #define INIT_ACCESS COMMON_INTERCEPT_FUNCTION(access)
7424 #else
7425 #define INIT_ACCESS
7426 #endif
7428 #if SANITIZER_INTERCEPT_FACCESSAT
7429 INTERCEPTOR(int, faccessat, int fd, const char *path, int mode, int flags) {
7430 void *ctx;
7431 COMMON_INTERCEPTOR_ENTER(ctx, faccessat, fd, path, mode, flags);
7432 if (path)
7433 COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
7434 return REAL(faccessat)(fd, path, mode, flags);
7435 }
7436 #define INIT_FACCESSAT COMMON_INTERCEPT_FUNCTION(faccessat)
7437 #else
7438 #define INIT_FACCESSAT
7439 #endif
7441 #if SANITIZER_INTERCEPT_GETGROUPLIST
7442 INTERCEPTOR(int, getgrouplist, const char *name, u32 basegid, u32 *groups,
7443 int *ngroups) {
7444 void *ctx;
7445 int res;
7446 COMMON_INTERCEPTOR_ENTER(ctx, getgrouplist, name, basegid, groups, ngroups);
7447 if (name)
7448 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
7449 if (ngroups)
7450 COMMON_INTERCEPTOR_READ_RANGE(ctx, ngroups, sizeof(*ngroups));
7451 res = REAL(getgrouplist)(name, basegid, groups, ngroups);
7452 if (!res && groups && ngroups) {
7453 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, groups, sizeof(*groups) * (*ngroups));
7454 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ngroups, sizeof(*ngroups));
7455 }
7456 return res;
7457 }
7459 #define INIT_GETGROUPLIST COMMON_INTERCEPT_FUNCTION(getgrouplist);
7460 #else
7461 #define INIT_GETGROUPLIST
7462 #endif
7464 #if SANITIZER_INTERCEPT_GETGROUPMEMBERSHIP
7465 INTERCEPTOR(int, getgroupmembership, const char *name, u32 basegid, u32 *groups,
7466 int maxgrp, int *ngroups) {
7467 void *ctx;
7468 int res;
7469 COMMON_INTERCEPTOR_ENTER(ctx, getgroupmembership, name, basegid, groups,
7470 maxgrp, ngroups);
7471 if (name)
7472 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
7473 res = REAL(getgroupmembership)(name, basegid, groups, maxgrp, ngroups);
7474 if (!res && groups && ngroups) {
7475 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, groups, sizeof(*groups) * (*ngroups));
7476 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ngroups, sizeof(*ngroups));
7477 }
7478 return res;
7479 }
7481 #define INIT_GETGROUPMEMBERSHIP COMMON_INTERCEPT_FUNCTION(getgroupmembership);
7482 #else
7483 #define INIT_GETGROUPMEMBERSHIP
7484 #endif
7486 #if SANITIZER_INTERCEPT_READLINK
7487 INTERCEPTOR(SSIZE_T, readlink, const char *path, char *buf, SIZE_T bufsiz) {
7488 void* ctx;
7489 COMMON_INTERCEPTOR_ENTER(ctx, readlink, path, buf, bufsiz);
7490 COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
7491 SSIZE_T res = REAL(readlink)(path, buf, bufsiz);
7492 if (res > 0)
7493 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, res);
7494 return res;
7495 }
7497 #define INIT_READLINK COMMON_INTERCEPT_FUNCTION(readlink)
7498 #else
7499 #define INIT_READLINK
7500 #endif
7502 #if SANITIZER_INTERCEPT_READLINKAT
7503 INTERCEPTOR(SSIZE_T, readlinkat, int dirfd, const char *path, char *buf,
7504 SIZE_T bufsiz) {
7505 void* ctx;
7506 COMMON_INTERCEPTOR_ENTER(ctx, readlinkat, dirfd, path, buf, bufsiz);
7507 COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
7508 SSIZE_T res = REAL(readlinkat)(dirfd, path, buf, bufsiz);
7509 if (res > 0)
7510 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, res);
7511 return res;
7512 }
7514 #define INIT_READLINKAT COMMON_INTERCEPT_FUNCTION(readlinkat)
7515 #else
7516 #define INIT_READLINKAT
7517 #endif
7519 #if SANITIZER_INTERCEPT_NAME_TO_HANDLE_AT
7520 INTERCEPTOR(int, name_to_handle_at, int dirfd, const char *pathname,
7521 struct file_handle *handle, int *mount_id, int flags) {
7522 void* ctx;
7523 COMMON_INTERCEPTOR_ENTER(ctx, name_to_handle_at, dirfd, pathname, handle,
7524 mount_id, flags);
7525 COMMON_INTERCEPTOR_READ_RANGE(ctx, pathname, internal_strlen(pathname) + 1);
7527 __sanitizer_file_handle *sanitizer_handle =
7528 reinterpret_cast<__sanitizer_file_handle*>(handle);
7529 COMMON_INTERCEPTOR_READ_RANGE(
7530 ctx, &sanitizer_handle->handle_bytes,
7531 sizeof(sanitizer_handle->handle_bytes));
7533 int res = REAL(name_to_handle_at)(dirfd, pathname, handle, mount_id, flags);
7534 if (!res) {
7535 COMMON_INTERCEPTOR_WRITE_RANGE(
7536 ctx, &sanitizer_handle->handle_bytes,
7537 sizeof(sanitizer_handle->handle_bytes));
7538 COMMON_INTERCEPTOR_WRITE_RANGE(
7539 ctx, &sanitizer_handle->handle_type,
7540 sizeof(sanitizer_handle->handle_type));
7541 COMMON_INTERCEPTOR_WRITE_RANGE(
7542 ctx, &sanitizer_handle->f_handle, sanitizer_handle->handle_bytes);
7543 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, mount_id, sizeof(*mount_id));
7544 }
7545 return res;
7546 }
7548 #define INIT_NAME_TO_HANDLE_AT COMMON_INTERCEPT_FUNCTION(name_to_handle_at)
7549 #else
7550 #define INIT_NAME_TO_HANDLE_AT
7551 #endif
7553 #if SANITIZER_INTERCEPT_OPEN_BY_HANDLE_AT
7554 INTERCEPTOR(int, open_by_handle_at, int mount_fd, struct file_handle* handle,
7555 int flags) {
7556 void* ctx;
7557 COMMON_INTERCEPTOR_ENTER(ctx, open_by_handle_at, mount_fd, handle, flags);
7559 __sanitizer_file_handle *sanitizer_handle =
7560 reinterpret_cast<__sanitizer_file_handle*>(handle);
7561 COMMON_INTERCEPTOR_READ_RANGE(
7562 ctx, &sanitizer_handle->handle_bytes,
7563 sizeof(sanitizer_handle->handle_bytes));
7564 COMMON_INTERCEPTOR_READ_RANGE(
7565 ctx, &sanitizer_handle->handle_type,
7566 sizeof(sanitizer_handle->handle_type));
7567 COMMON_INTERCEPTOR_READ_RANGE(
7568 ctx, &sanitizer_handle->f_handle, sanitizer_handle->handle_bytes);
7570 return COMMON_INTERCEPTOR_BLOCK_REAL(open_by_handle_at)(mount_fd, handle,
7571 flags);
7572 }
7574 #define INIT_OPEN_BY_HANDLE_AT COMMON_INTERCEPT_FUNCTION(open_by_handle_at)
7575 #else
7576 #define INIT_OPEN_BY_HANDLE_AT
7577 #endif
7579 #if SANITIZER_INTERCEPT_STRLCPY
7580 INTERCEPTOR(SIZE_T, strlcpy, char *dst, char *src, SIZE_T size) {
7581 void *ctx;
7582 SIZE_T res;
7583 COMMON_INTERCEPTOR_ENTER(ctx, strlcpy, dst, src, size);
7584 if (src) {
7585 // Keep strnlen as macro argument, as macro may ignore it.
7586 COMMON_INTERCEPTOR_READ_STRING(
7587 ctx, src, Min(internal_strnlen(src, size), size - 1) + 1);
7588 }
7589 res = REAL(strlcpy)(dst, src, size);
7590 COMMON_INTERCEPTOR_COPY_STRING(ctx, dst, src, internal_strlen(dst) + 1);
7591 return res;
7592 }
7594 INTERCEPTOR(SIZE_T, strlcat, char *dst, char *src, SIZE_T size) {
7595 void *ctx;
7596 SIZE_T len = 0;
7597 COMMON_INTERCEPTOR_ENTER(ctx, strlcat, dst, src, size);
7598 // src is checked in the strlcpy() interceptor
7599 if (dst) {
7600 len = internal_strnlen(dst, size);
7601 COMMON_INTERCEPTOR_READ_STRING(ctx, dst, Min(len, size - 1) + 1);
7602 }
7603 // Reuse the rest of the code in the strlcpy() interceptor
7604 return WRAP(strlcpy)(dst + len, src, size - len) + len;
7605 }
7606 #define INIT_STRLCPY \
7607 COMMON_INTERCEPT_FUNCTION(strlcpy); \
7608 COMMON_INTERCEPT_FUNCTION(strlcat);
7609 #else
7610 #define INIT_STRLCPY
7611 #endif
7613 #if SANITIZER_INTERCEPT_MMAP
7614 INTERCEPTOR(void *, mmap, void *addr, SIZE_T sz, int prot, int flags, int fd,
7615 OFF_T off) {
7616 void *ctx;
7617 if (common_flags()->detect_write_exec)
7618 ReportMmapWriteExec(prot, flags);
7619 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
7620 return (void *)internal_mmap(addr, sz, prot, flags, fd, off);
7621 COMMON_INTERCEPTOR_ENTER(ctx, mmap, addr, sz, prot, flags, fd, off);
7622 COMMON_INTERCEPTOR_MMAP_IMPL(ctx, mmap, addr, sz, prot, flags, fd, off);
7623 }
7625 INTERCEPTOR(int, munmap, void *addr, SIZE_T sz) {
7626 void *ctx;
7627 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
7628 return (int)internal_munmap(addr, sz);
7629 COMMON_INTERCEPTOR_ENTER(ctx, munmap, addr, sz);
7630 COMMON_INTERCEPTOR_MUNMAP_IMPL(ctx, addr, sz);
7631 }
7633 INTERCEPTOR(int, mprotect, void *addr, SIZE_T sz, int prot) {
7634 void *ctx;
7635 if (common_flags()->detect_write_exec)
7636 ReportMmapWriteExec(prot, 0);
7637 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
7638 return (int)internal_mprotect(addr, sz, prot);
7639 COMMON_INTERCEPTOR_ENTER(ctx, mprotect, addr, sz, prot);
7640 MprotectMallocZones(addr, prot);
7641 return REAL(mprotect)(addr, sz, prot);
7642 }
7643 #define INIT_MMAP \
7644 COMMON_INTERCEPT_FUNCTION(mmap); \
7645 COMMON_INTERCEPT_FUNCTION(munmap); \
7646 COMMON_INTERCEPT_FUNCTION(mprotect);
7647 #else
7648 #define INIT_MMAP
7649 #endif
7651 #if SANITIZER_INTERCEPT_MMAP64
7652 INTERCEPTOR(void *, mmap64, void *addr, SIZE_T sz, int prot, int flags, int fd,
7653 OFF64_T off) {
7654 void *ctx;
7655 if (common_flags()->detect_write_exec)
7656 ReportMmapWriteExec(prot, flags);
7657 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
7658 return (void *)internal_mmap(addr, sz, prot, flags, fd, off);
7659 COMMON_INTERCEPTOR_ENTER(ctx, mmap64, addr, sz, prot, flags, fd, off);
7660 COMMON_INTERCEPTOR_MMAP_IMPL(ctx, mmap64, addr, sz, prot, flags, fd, off);
7661 }
7662 #define INIT_MMAP64 COMMON_INTERCEPT_FUNCTION(mmap64);
7663 #else
7664 #define INIT_MMAP64
7665 #endif
7667 #if SANITIZER_INTERCEPT_DEVNAME
7668 INTERCEPTOR(char *, devname, u64 dev, u32 type) {
7669 void *ctx;
7670 char *name;
7671 COMMON_INTERCEPTOR_ENTER(ctx, devname, dev, type);
7672 name = REAL(devname)(dev, type);
7673 if (name)
7674 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, name, internal_strlen(name) + 1);
7675 return name;
7676 }
7677 #define INIT_DEVNAME COMMON_INTERCEPT_FUNCTION(devname);
7678 #else
7679 #define INIT_DEVNAME
7680 #endif
7682 #if SANITIZER_INTERCEPT_DEVNAME_R
7683 #if SANITIZER_NETBSD
7684 #define DEVNAME_R_RETTYPE int
7685 #define DEVNAME_R_SUCCESS(x) (!(x))
7686 #else
7687 #define DEVNAME_R_RETTYPE char*
7688 #define DEVNAME_R_SUCCESS(x) (x)
7689 #endif
7690 INTERCEPTOR(DEVNAME_R_RETTYPE, devname_r, u64 dev, u32 type, char *path,
7691 uptr len) {
7692 void *ctx;
7693 COMMON_INTERCEPTOR_ENTER(ctx, devname_r, dev, type, path, len);
7694 DEVNAME_R_RETTYPE res = REAL(devname_r)(dev, type, path, len);
7695 if (DEVNAME_R_SUCCESS(res))
7696 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, path, internal_strlen(path) + 1);
7697 return res;
7698 }
7699 #define INIT_DEVNAME_R COMMON_INTERCEPT_FUNCTION(devname_r);
7700 #else
7701 #define INIT_DEVNAME_R
7702 #endif
7704 #if SANITIZER_INTERCEPT_FGETLN
7705 INTERCEPTOR(char *, fgetln, __sanitizer_FILE *stream, SIZE_T *len) {
7706 void *ctx;
7707 COMMON_INTERCEPTOR_ENTER(ctx, fgetln, stream, len);
7708 char *str = REAL(fgetln)(stream, len);
7709 if (str && len) {
7710 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, len, sizeof(*len));
7711 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, str, *len);
7712 }
7713 return str;
7714 }
7715 #define INIT_FGETLN COMMON_INTERCEPT_FUNCTION(fgetln)
7716 #else
7717 #define INIT_FGETLN
7718 #endif
7720 #if SANITIZER_INTERCEPT_STRMODE
7721 INTERCEPTOR(void, strmode, u32 mode, char *bp) {
7722 void *ctx;
7723 COMMON_INTERCEPTOR_ENTER(ctx, strmode, mode, bp);
7724 REAL(strmode)(mode, bp);
7725 if (bp)
7726 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, bp, internal_strlen(bp) + 1);
7727 }
7728 #define INIT_STRMODE COMMON_INTERCEPT_FUNCTION(strmode)
7729 #else
7730 #define INIT_STRMODE
7731 #endif
7733 #if SANITIZER_INTERCEPT_TTYENT
7734 INTERCEPTOR(struct __sanitizer_ttyent *, getttyent, void) {
7735 void *ctx;
7736 COMMON_INTERCEPTOR_ENTER(ctx, getttyent);
7737 struct __sanitizer_ttyent *ttyent = REAL(getttyent)();
7738 if (ttyent)
7739 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ttyent, struct_ttyent_sz);
7740 return ttyent;
7741 }
7742 INTERCEPTOR(struct __sanitizer_ttyent *, getttynam, char *name) {
7743 void *ctx;
7744 COMMON_INTERCEPTOR_ENTER(ctx, getttynam, name);
7745 if (name)
7746 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
7747 struct __sanitizer_ttyent *ttyent = REAL(getttynam)(name);
7748 if (ttyent)
7749 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ttyent, struct_ttyent_sz);
7750 return ttyent;
7751 }
7752 #define INIT_TTYENT \
7753 COMMON_INTERCEPT_FUNCTION(getttyent); \
7754 COMMON_INTERCEPT_FUNCTION(getttynam);
7755 #else
7756 #define INIT_TTYENT
7757 #endif
7759 #if SANITIZER_INTERCEPT_TTYENTPATH
7760 INTERCEPTOR(int, setttyentpath, char *path) {
7761 void *ctx;
7762 COMMON_INTERCEPTOR_ENTER(ctx, setttyentpath, path);
7763 if (path)
7764 COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
7765 return REAL(setttyentpath)(path);
7766 }
7767 #define INIT_TTYENTPATH COMMON_INTERCEPT_FUNCTION(setttyentpath);
7768 #else
7769 #define INIT_TTYENTPATH
7770 #endif
7772 #if SANITIZER_INTERCEPT_PROTOENT
7773 static void write_protoent(void *ctx, struct __sanitizer_protoent *p) {
7774 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p, sizeof(*p));
7776 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p->p_name, internal_strlen(p->p_name) + 1);
7778 SIZE_T pp_size = 1; // One handles the trailing \0
7780 for (char **pp = p->p_aliases; *pp; ++pp, ++pp_size)
7781 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *pp, internal_strlen(*pp) + 1);
7783 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, p->p_aliases, pp_size * sizeof(char *));
7784 }
7786 INTERCEPTOR(struct __sanitizer_protoent *, getprotoent,) {
7787 void *ctx;
7788 COMMON_INTERCEPTOR_ENTER(ctx, getprotoent,);
7789 struct __sanitizer_protoent *p = REAL(getprotoent)();
7790 if (p)
7791 write_protoent(ctx, p);
7792 return p;
7793 }
7795 INTERCEPTOR(struct __sanitizer_protoent *, getprotobyname, const char *name) {
7796 void *ctx;
7797 COMMON_INTERCEPTOR_ENTER(ctx, getprotobyname, name);
7798 if (name)
7799 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
7800 struct __sanitizer_protoent *p = REAL(getprotobyname)(name);
7801 if (p)
7802 write_protoent(ctx, p);
7803 return p;
7804 }
7806 INTERCEPTOR(struct __sanitizer_protoent *, getprotobynumber, int proto) {
7807 void *ctx;
7808 COMMON_INTERCEPTOR_ENTER(ctx, getprotobynumber, proto);
7809 struct __sanitizer_protoent *p = REAL(getprotobynumber)(proto);
7810 if (p)
7811 write_protoent(ctx, p);
7812 return p;
7813 }
7814 #define INIT_PROTOENT \
7815 COMMON_INTERCEPT_FUNCTION(getprotoent); \
7816 COMMON_INTERCEPT_FUNCTION(getprotobyname); \
7817 COMMON_INTERCEPT_FUNCTION(getprotobynumber)
7818 #else
7819 #define INIT_PROTOENT
7820 #endif
7822 #if SANITIZER_INTERCEPT_PROTOENT_R
7823 INTERCEPTOR(int, getprotoent_r, struct __sanitizer_protoent *result_buf,
7824 char *buf, SIZE_T buflen, struct __sanitizer_protoent **result) {
7825 void *ctx;
7826 COMMON_INTERCEPTOR_ENTER(ctx, getprotoent_r, result_buf, buf, buflen,
7827 result);
7828 int res = REAL(getprotoent_r)(result_buf, buf, buflen, result);
7830 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof *result);
7831 if (!res && *result)
7832 write_protoent(ctx, *result);
7833 return res;
7834 }
7836 INTERCEPTOR(int, getprotobyname_r, const char *name,
7837 struct __sanitizer_protoent *result_buf, char *buf, SIZE_T buflen,
7838 struct __sanitizer_protoent **result) {
7839 void *ctx;
7840 COMMON_INTERCEPTOR_ENTER(ctx, getprotobyname_r, name, result_buf, buf,
7841 buflen, result);
7842 if (name)
7843 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
7844 int res = REAL(getprotobyname_r)(name, result_buf, buf, buflen, result);
7846 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof *result);
7847 if (!res && *result)
7848 write_protoent(ctx, *result);
7849 return res;
7850 }
7852 INTERCEPTOR(int, getprotobynumber_r, int num,
7853 struct __sanitizer_protoent *result_buf, char *buf,
7854 SIZE_T buflen, struct __sanitizer_protoent **result) {
7855 void *ctx;
7856 COMMON_INTERCEPTOR_ENTER(ctx, getprotobynumber_r, num, result_buf, buf,
7857 buflen, result);
7858 int res = REAL(getprotobynumber_r)(num, result_buf, buf, buflen, result);
7860 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, result, sizeof *result);
7861 if (!res && *result)
7862 write_protoent(ctx, *result);
7863 return res;
7864 }
7866 #define INIT_PROTOENT_R \
7867 COMMON_INTERCEPT_FUNCTION(getprotoent_r); \
7868 COMMON_INTERCEPT_FUNCTION(getprotobyname_r); \
7869 COMMON_INTERCEPT_FUNCTION(getprotobynumber_r);
7870 #else
7871 #define INIT_PROTOENT_R
7872 #endif
7874 #if SANITIZER_INTERCEPT_NETENT
7875 INTERCEPTOR(struct __sanitizer_netent *, getnetent,) {
7876 void *ctx;
7877 COMMON_INTERCEPTOR_ENTER(ctx, getnetent,);
7878 struct __sanitizer_netent *n = REAL(getnetent)();
7879 if (n) {
7880 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, n, sizeof(*n));
7882 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, n->n_name, internal_strlen(n->n_name) + 1);
7884 SIZE_T nn_size = 1; // One handles the trailing \0
7886 for (char **nn = n->n_aliases; *nn; ++nn, ++nn_size)
7887 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *nn, internal_strlen(*nn) + 1);
7889 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, n->n_aliases, nn_size * sizeof(char *));
7890 }
7891 return n;
7892 }
7894 INTERCEPTOR(struct __sanitizer_netent *, getnetbyname, const char *name) {
7895 void *ctx;
7896 COMMON_INTERCEPTOR_ENTER(ctx, getnetbyname, name);
7897 if (name)
7898 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
7899 struct __sanitizer_netent *n = REAL(getnetbyname)(name);
7900 if (n) {
7901 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, n, sizeof(*n));
7903 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, n->n_name, internal_strlen(n->n_name) + 1);
7905 SIZE_T nn_size = 1; // One handles the trailing \0
7907 for (char **nn = n->n_aliases; *nn; ++nn, ++nn_size)
7908 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *nn, internal_strlen(*nn) + 1);
7910 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, n->n_aliases, nn_size * sizeof(char *));
7911 }
7912 return n;
7913 }
7915 INTERCEPTOR(struct __sanitizer_netent *, getnetbyaddr, u32 net, int type) {
7916 void *ctx;
7917 COMMON_INTERCEPTOR_ENTER(ctx, getnetbyaddr, net, type);
7918 struct __sanitizer_netent *n = REAL(getnetbyaddr)(net, type);
7919 if (n) {
7920 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, n, sizeof(*n));
7922 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, n->n_name, internal_strlen(n->n_name) + 1);
7924 SIZE_T nn_size = 1; // One handles the trailing \0
7926 for (char **nn = n->n_aliases; *nn; ++nn, ++nn_size)
7927 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *nn, internal_strlen(*nn) + 1);
7929 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, n->n_aliases, nn_size * sizeof(char *));
7930 }
7931 return n;
7932 }
7933 #define INIT_NETENT \
7934 COMMON_INTERCEPT_FUNCTION(getnetent); \
7935 COMMON_INTERCEPT_FUNCTION(getnetbyname); \
7936 COMMON_INTERCEPT_FUNCTION(getnetbyaddr)
7937 #else
7938 #define INIT_NETENT
7939 #endif
7941 #if SANITIZER_INTERCEPT_GETMNTINFO
7942 INTERCEPTOR(int, getmntinfo, void **mntbufp, int flags) {
7943 void *ctx;
7944 COMMON_INTERCEPTOR_ENTER(ctx, getmntinfo, mntbufp, flags);
7945 int cnt = REAL(getmntinfo)(mntbufp, flags);
7946 if (cnt > 0 && mntbufp) {
7947 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, mntbufp, sizeof(void *));
7948 if (*mntbufp)
7949 #if SANITIZER_NETBSD
7950 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *mntbufp, cnt * struct_statvfs_sz);
7951 #else
7952 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *mntbufp, cnt * struct_statfs_sz);
7953 #endif
7954 }
7955 return cnt;
7956 }
7957 #define INIT_GETMNTINFO COMMON_INTERCEPT_FUNCTION(getmntinfo)
7958 #else
7959 #define INIT_GETMNTINFO
7960 #endif
7962 #if SANITIZER_INTERCEPT_MI_VECTOR_HASH
7963 INTERCEPTOR(void, mi_vector_hash, const void *key, SIZE_T len, u32 seed,
7964 u32 hashes[3]) {
7965 void *ctx;
7966 COMMON_INTERCEPTOR_ENTER(ctx, mi_vector_hash, key, len, seed, hashes);
7967 if (key)
7968 COMMON_INTERCEPTOR_READ_RANGE(ctx, key, len);
7969 REAL(mi_vector_hash)(key, len, seed, hashes);
7970 if (hashes)
7971 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, hashes, sizeof(hashes[0]) * 3);
7972 }
7973 #define INIT_MI_VECTOR_HASH COMMON_INTERCEPT_FUNCTION(mi_vector_hash)
7974 #else
7975 #define INIT_MI_VECTOR_HASH
7976 #endif
7978 #if SANITIZER_INTERCEPT_SETVBUF
7979 INTERCEPTOR(int, setvbuf, __sanitizer_FILE *stream, char *buf, int mode,
7980 SIZE_T size) {
7981 void *ctx;
7982 COMMON_INTERCEPTOR_ENTER(ctx, setvbuf, stream, buf, mode, size);
7983 int ret = REAL(setvbuf)(stream, buf, mode, size);
7984 if (buf)
7985 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, size);
7986 if (stream)
7987 unpoison_file(stream);
7988 return ret;
7989 }
7991 INTERCEPTOR(void, setbuf, __sanitizer_FILE *stream, char *buf) {
7992 void *ctx;
7993 COMMON_INTERCEPTOR_ENTER(ctx, setbuf, stream, buf);
7994 REAL(setbuf)(stream, buf);
7995 if (buf) {
7996 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, __sanitizer_bufsiz);
7997 }
7998 if (stream)
7999 unpoison_file(stream);
8000 }
8002 INTERCEPTOR(void, setbuffer, __sanitizer_FILE *stream, char *buf, SIZE_T size) {
8003 void *ctx;
8004 COMMON_INTERCEPTOR_ENTER(ctx, setbuffer, stream, buf, size);
8005 REAL(setbuffer)(stream, buf, size);
8006 if (buf) {
8007 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, size);
8008 }
8009 if (stream)
8010 unpoison_file(stream);
8011 }
8013 INTERCEPTOR(void, setlinebuf, __sanitizer_FILE *stream) {
8014 void *ctx;
8015 COMMON_INTERCEPTOR_ENTER(ctx, setlinebuf, stream);
8016 REAL(setlinebuf)(stream);
8017 if (stream)
8018 unpoison_file(stream);
8019 }
8020 #define INIT_SETVBUF COMMON_INTERCEPT_FUNCTION(setvbuf); \
8021 COMMON_INTERCEPT_FUNCTION(setbuf); \
8022 COMMON_INTERCEPT_FUNCTION(setbuffer); \
8023 COMMON_INTERCEPT_FUNCTION(setlinebuf)
8024 #else
8025 #define INIT_SETVBUF
8026 #endif
8028 #if SANITIZER_INTERCEPT_GETVFSSTAT
8029 INTERCEPTOR(int, getvfsstat, void *buf, SIZE_T bufsize, int flags) {
8030 void *ctx;
8031 COMMON_INTERCEPTOR_ENTER(ctx, getvfsstat, buf, bufsize, flags);
8032 int ret = REAL(getvfsstat)(buf, bufsize, flags);
8033 if (buf && ret > 0)
8034 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, ret * struct_statvfs_sz);
8035 return ret;
8036 }
8037 #define INIT_GETVFSSTAT COMMON_INTERCEPT_FUNCTION(getvfsstat)
8038 #else
8039 #define INIT_GETVFSSTAT
8040 #endif
8042 #if SANITIZER_INTERCEPT_REGEX
8043 INTERCEPTOR(int, regcomp, void *preg, const char *pattern, int cflags) {
8044 void *ctx;
8045 COMMON_INTERCEPTOR_ENTER(ctx, regcomp, preg, pattern, cflags);
8046 if (pattern)
8047 COMMON_INTERCEPTOR_READ_RANGE(ctx, pattern, internal_strlen(pattern) + 1);
8048 int res = REAL(regcomp)(preg, pattern, cflags);
8049 if (preg)
8050 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, preg, struct_regex_sz);
8051 return res;
8052 }
8053 INTERCEPTOR(int, regexec, const void *preg, const char *string, SIZE_T nmatch,
8054 struct __sanitizer_regmatch *pmatch[], int eflags) {
8055 void *ctx;
8056 COMMON_INTERCEPTOR_ENTER(ctx, regexec, preg, string, nmatch, pmatch, eflags);
8057 if (preg)
8058 COMMON_INTERCEPTOR_READ_RANGE(ctx, preg, struct_regex_sz);
8059 if (string)
8060 COMMON_INTERCEPTOR_READ_RANGE(ctx, string, internal_strlen(string) + 1);
8061 int res = REAL(regexec)(preg, string, nmatch, pmatch, eflags);
8062 if (!res && pmatch)
8063 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pmatch, nmatch * struct_regmatch_sz);
8064 return res;
8065 }
8066 INTERCEPTOR(SIZE_T, regerror, int errcode, const void *preg, char *errbuf,
8067 SIZE_T errbuf_size) {
8068 void *ctx;
8069 COMMON_INTERCEPTOR_ENTER(ctx, regerror, errcode, preg, errbuf, errbuf_size);
8070 if (preg)
8071 COMMON_INTERCEPTOR_READ_RANGE(ctx, preg, struct_regex_sz);
8072 SIZE_T res = REAL(regerror)(errcode, preg, errbuf, errbuf_size);
8073 if (errbuf)
8074 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, errbuf, internal_strlen(errbuf) + 1);
8075 return res;
8076 }
8077 INTERCEPTOR(void, regfree, const void *preg) {
8078 void *ctx;
8079 COMMON_INTERCEPTOR_ENTER(ctx, regfree, preg);
8080 if (preg)
8081 COMMON_INTERCEPTOR_READ_RANGE(ctx, preg, struct_regex_sz);
8082 REAL(regfree)(preg);
8083 }
8084 #define INIT_REGEX \
8085 COMMON_INTERCEPT_FUNCTION(regcomp); \
8086 COMMON_INTERCEPT_FUNCTION_GLIBC_VER_MIN(regexec, "GLIBC_2.3.4"); \
8087 COMMON_INTERCEPT_FUNCTION(regerror); \
8088 COMMON_INTERCEPT_FUNCTION(regfree);
8089 #else
8090 #define INIT_REGEX
8091 #endif
8093 #if SANITIZER_INTERCEPT_REGEXSUB
8094 INTERCEPTOR(SSIZE_T, regnsub, char *buf, SIZE_T bufsiz, const char *sub,
8095 const struct __sanitizer_regmatch *rm, const char *str) {
8096 void *ctx;
8097 COMMON_INTERCEPTOR_ENTER(ctx, regnsub, buf, bufsiz, sub, rm, str);
8098 if (sub)
8099 COMMON_INTERCEPTOR_READ_RANGE(ctx, sub, internal_strlen(sub) + 1);
8100 // The implementation demands and hardcodes 10 elements
8101 if (rm)
8102 COMMON_INTERCEPTOR_READ_RANGE(ctx, rm, 10 * struct_regmatch_sz);
8103 if (str)
8104 COMMON_INTERCEPTOR_READ_RANGE(ctx, str, internal_strlen(str) + 1);
8105 SSIZE_T res = REAL(regnsub)(buf, bufsiz, sub, rm, str);
8106 if (res > 0 && buf)
8107 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, internal_strlen(buf) + 1);
8108 return res;
8109 }
8110 INTERCEPTOR(SSIZE_T, regasub, char **buf, const char *sub,
8111 const struct __sanitizer_regmatch *rm, const char *sstr) {
8112 void *ctx;
8113 COMMON_INTERCEPTOR_ENTER(ctx, regasub, buf, sub, rm, sstr);
8114 if (sub)
8115 COMMON_INTERCEPTOR_READ_RANGE(ctx, sub, internal_strlen(sub) + 1);
8116 // Hardcode 10 elements as this is hardcoded size
8117 if (rm)
8118 COMMON_INTERCEPTOR_READ_RANGE(ctx, rm, 10 * struct_regmatch_sz);
8119 if (sstr)
8120 COMMON_INTERCEPTOR_READ_RANGE(ctx, sstr, internal_strlen(sstr) + 1);
8121 SSIZE_T res = REAL(regasub)(buf, sub, rm, sstr);
8122 if (res > 0 && buf) {
8123 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, sizeof(char *));
8124 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *buf, internal_strlen(*buf) + 1);
8125 }
8126 return res;
8127 }
8129 #define INIT_REGEXSUB \
8130 COMMON_INTERCEPT_FUNCTION(regnsub); \
8131 COMMON_INTERCEPT_FUNCTION(regasub);
8132 #else
8133 #define INIT_REGEXSUB
8134 #endif
8136 #if SANITIZER_INTERCEPT_FTS
8137 INTERCEPTOR(void *, fts_open, char *const *path_argv, int options,
8138 int (*compar)(void **, void **)) {
8139 void *ctx;
8140 COMMON_INTERCEPTOR_ENTER(ctx, fts_open, path_argv, options, compar);
8141 if (path_argv) {
8142 for (char *const *pa = path_argv; ; ++pa) {
8143 COMMON_INTERCEPTOR_READ_RANGE(ctx, pa, sizeof(char **));
8144 if (!*pa)
8145 break;
8146 COMMON_INTERCEPTOR_READ_RANGE(ctx, *pa, internal_strlen(*pa) + 1);
8147 }
8148 }
8149 // TODO(kamil): handle compar callback
8150 void *fts = REAL(fts_open)(path_argv, options, compar);
8151 if (fts)
8152 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, fts, struct_FTS_sz);
8153 return fts;
8154 }
8156 INTERCEPTOR(void *, fts_read, void *ftsp) {
8157 void *ctx;
8158 COMMON_INTERCEPTOR_ENTER(ctx, fts_read, ftsp);
8159 if (ftsp)
8160 COMMON_INTERCEPTOR_READ_RANGE(ctx, ftsp, struct_FTS_sz);
8161 void *ftsent = REAL(fts_read)(ftsp);
8162 if (ftsent)
8163 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ftsent, struct_FTSENT_sz);
8164 return ftsent;
8165 }
8167 INTERCEPTOR(void *, fts_children, void *ftsp, int options) {
8168 void *ctx;
8169 COMMON_INTERCEPTOR_ENTER(ctx, fts_children, ftsp, options);
8170 if (ftsp)
8171 COMMON_INTERCEPTOR_READ_RANGE(ctx, ftsp, struct_FTS_sz);
8172 void *ftsent = REAL(fts_children)(ftsp, options);
8173 if (ftsent)
8174 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ftsent, struct_FTSENT_sz);
8175 return ftsent;
8176 }
8178 INTERCEPTOR(int, fts_set, void *ftsp, void *f, int options) {
8179 void *ctx;
8180 COMMON_INTERCEPTOR_ENTER(ctx, fts_set, ftsp, f, options);
8181 if (ftsp)
8182 COMMON_INTERCEPTOR_READ_RANGE(ctx, ftsp, struct_FTS_sz);
8183 if (f)
8184 COMMON_INTERCEPTOR_READ_RANGE(ctx, f, struct_FTSENT_sz);
8185 return REAL(fts_set)(ftsp, f, options);
8186 }
8188 INTERCEPTOR(int, fts_close, void *ftsp) {
8189 void *ctx;
8190 COMMON_INTERCEPTOR_ENTER(ctx, fts_close, ftsp);
8191 if (ftsp)
8192 COMMON_INTERCEPTOR_READ_RANGE(ctx, ftsp, struct_FTS_sz);
8193 return REAL(fts_close)(ftsp);
8194 }
8195 #define INIT_FTS \
8196 COMMON_INTERCEPT_FUNCTION(fts_open); \
8197 COMMON_INTERCEPT_FUNCTION(fts_read); \
8198 COMMON_INTERCEPT_FUNCTION(fts_children); \
8199 COMMON_INTERCEPT_FUNCTION(fts_set); \
8200 COMMON_INTERCEPT_FUNCTION(fts_close);
8201 #else
8202 #define INIT_FTS
8203 #endif
8205 #if SANITIZER_INTERCEPT_SYSCTL
8206 INTERCEPTOR(int, sysctl, int *name, unsigned int namelen, void *oldp,
8207 SIZE_T *oldlenp, void *newp, SIZE_T newlen) {
8208 void *ctx;
8209 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
8210 return internal_sysctl(name, namelen, oldp, oldlenp, newp, newlen);
8211 COMMON_INTERCEPTOR_ENTER(ctx, sysctl, name, namelen, oldp, oldlenp, newp,
8212 newlen);
8213 if (name)
8214 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, namelen * sizeof(*name));
8215 if (oldlenp)
8216 COMMON_INTERCEPTOR_READ_RANGE(ctx, oldlenp, sizeof(*oldlenp));
8217 if (newp && newlen)
8218 COMMON_INTERCEPTOR_READ_RANGE(ctx, newp, newlen);
8219 int res = REAL(sysctl)(name, namelen, oldp, oldlenp, newp, newlen);
8220 if (!res) {
8221 if (oldlenp) {
8222 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, oldlenp, sizeof(*oldlenp));
8223 if (oldp)
8224 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, oldp, *oldlenp);
8225 }
8226 }
8227 return res;
8228 }
8230 INTERCEPTOR(int, sysctlbyname, char *sname, void *oldp, SIZE_T *oldlenp,
8231 void *newp, SIZE_T newlen) {
8232 void *ctx;
8233 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
8234 return internal_sysctlbyname(sname, oldp, oldlenp, newp, newlen);
8235 COMMON_INTERCEPTOR_ENTER(ctx, sysctlbyname, sname, oldp, oldlenp, newp,
8236 newlen);
8237 if (sname)
8238 COMMON_INTERCEPTOR_READ_RANGE(ctx, sname, internal_strlen(sname) + 1);
8239 if (oldlenp)
8240 COMMON_INTERCEPTOR_READ_RANGE(ctx, oldlenp, sizeof(*oldlenp));
8241 if (newp && newlen)
8242 COMMON_INTERCEPTOR_READ_RANGE(ctx, newp, newlen);
8243 int res = REAL(sysctlbyname)(sname, oldp, oldlenp, newp, newlen);
8244 if (!res) {
8245 if (oldlenp) {
8246 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, oldlenp, sizeof(*oldlenp));
8247 if (oldp)
8248 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, oldp, *oldlenp);
8249 }
8250 }
8251 return res;
8252 }
8254 INTERCEPTOR(int, sysctlnametomib, const char *sname, int *name,
8255 SIZE_T *namelenp) {
8256 void *ctx;
8257 COMMON_INTERCEPTOR_ENTER(ctx, sysctlnametomib, sname, name, namelenp);
8258 if (sname)
8259 COMMON_INTERCEPTOR_READ_RANGE(ctx, sname, internal_strlen(sname) + 1);
8260 if (namelenp)
8261 COMMON_INTERCEPTOR_READ_RANGE(ctx, namelenp, sizeof(*namelenp));
8262 int res = REAL(sysctlnametomib)(sname, name, namelenp);
8263 if (!res) {
8264 if (namelenp) {
8265 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, namelenp, sizeof(*namelenp));
8266 if (name)
8267 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, name, *namelenp * sizeof(*name));
8268 }
8269 }
8270 return res;
8271 }
8273 #define INIT_SYSCTL \
8274 COMMON_INTERCEPT_FUNCTION(sysctl); \
8275 COMMON_INTERCEPT_FUNCTION(sysctlbyname); \
8276 COMMON_INTERCEPT_FUNCTION(sysctlnametomib);
8277 #else
8278 #define INIT_SYSCTL
8279 #endif
8281 #if SANITIZER_INTERCEPT_ASYSCTL
8282 INTERCEPTOR(void *, asysctl, const int *name, SIZE_T namelen, SIZE_T *len) {
8283 void *ctx;
8284 COMMON_INTERCEPTOR_ENTER(ctx, asysctl, name, namelen, len);
8285 if (name)
8286 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, sizeof(*name) * namelen);
8287 void *res = REAL(asysctl)(name, namelen, len);
8288 if (res && len) {
8289 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, len, sizeof(*len));
8290 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, *len);
8291 }
8292 return res;
8293 }
8295 INTERCEPTOR(void *, asysctlbyname, const char *sname, SIZE_T *len) {
8296 void *ctx;
8297 COMMON_INTERCEPTOR_ENTER(ctx, asysctlbyname, sname, len);
8298 if (sname)
8299 COMMON_INTERCEPTOR_READ_RANGE(ctx, sname, internal_strlen(sname) + 1);
8300 void *res = REAL(asysctlbyname)(sname, len);
8301 if (res && len) {
8302 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, len, sizeof(*len));
8303 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, *len);
8304 }
8305 return res;
8306 }
8307 #define INIT_ASYSCTL \
8308 COMMON_INTERCEPT_FUNCTION(asysctl); \
8309 COMMON_INTERCEPT_FUNCTION(asysctlbyname);
8310 #else
8311 #define INIT_ASYSCTL
8312 #endif
8314 #if SANITIZER_INTERCEPT_SYSCTLGETMIBINFO
8315 INTERCEPTOR(int, sysctlgetmibinfo, char *sname, int *name,
8316 unsigned int *namelenp, char *cname, SIZE_T *csz, void **rnode,
8317 int v) {
8318 void *ctx;
8319 COMMON_INTERCEPTOR_ENTER(ctx, sysctlgetmibinfo, sname, name, namelenp, cname,
8320 csz, rnode, v);
8321 if (sname)
8322 COMMON_INTERCEPTOR_READ_RANGE(ctx, sname, internal_strlen(sname) + 1);
8323 if (namelenp)
8324 COMMON_INTERCEPTOR_READ_RANGE(ctx, namelenp, sizeof(*namelenp));
8325 if (csz)
8326 COMMON_INTERCEPTOR_READ_RANGE(ctx, csz, sizeof(*csz));
8327 // Skip rnode, it's rarely used and not trivial to sanitize
8328 // It's also used mostly internally
8329 int res = REAL(sysctlgetmibinfo)(sname, name, namelenp, cname, csz, rnode, v);
8330 if (!res) {
8331 if (namelenp) {
8332 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, namelenp, sizeof(*namelenp));
8333 if (name)
8334 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, name, *namelenp * sizeof(*name));
8335 }
8336 if (csz) {
8337 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, csz, sizeof(*csz));
8338 if (cname)
8339 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cname, *csz);
8340 }
8341 }
8342 return res;
8343 }
8344 #define INIT_SYSCTLGETMIBINFO \
8345 COMMON_INTERCEPT_FUNCTION(sysctlgetmibinfo);
8346 #else
8347 #define INIT_SYSCTLGETMIBINFO
8348 #endif
8350 #if SANITIZER_INTERCEPT_NL_LANGINFO
8351 INTERCEPTOR(char *, nl_langinfo, long item) {
8352 void *ctx;
8353 COMMON_INTERCEPTOR_ENTER(ctx, nl_langinfo, item);
8354 char *ret = REAL(nl_langinfo)(item);
8355 if (ret)
8356 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, internal_strlen(ret) + 1);
8357 return ret;
8358 }
8359 #define INIT_NL_LANGINFO COMMON_INTERCEPT_FUNCTION(nl_langinfo)
8360 #else
8361 #define INIT_NL_LANGINFO
8362 #endif
8364 #if SANITIZER_INTERCEPT_MODCTL
8365 INTERCEPTOR(int, modctl, int operation, void *argp) {
8366 void *ctx;
8367 int ret;
8368 COMMON_INTERCEPTOR_ENTER(ctx, modctl, operation, argp);
8370 if (operation == modctl_load) {
8371 if (argp) {
8372 __sanitizer_modctl_load_t *ml = (__sanitizer_modctl_load_t *)argp;
8373 COMMON_INTERCEPTOR_READ_RANGE(ctx, ml, sizeof(*ml));
8374 if (ml->ml_filename)
8375 COMMON_INTERCEPTOR_READ_RANGE(ctx, ml->ml_filename,
8376 internal_strlen(ml->ml_filename) + 1);
8377 if (ml->ml_props)
8378 COMMON_INTERCEPTOR_READ_RANGE(ctx, ml->ml_props, ml->ml_propslen);
8379 }
8380 ret = REAL(modctl)(operation, argp);
8381 } else if (operation == modctl_unload) {
8382 if (argp) {
8383 const char *name = (const char *)argp;
8384 COMMON_INTERCEPTOR_READ_RANGE(ctx, name, internal_strlen(name) + 1);
8385 }
8386 ret = REAL(modctl)(operation, argp);
8387 } else if (operation == modctl_stat) {
8388 uptr iov_len;
8389 struct __sanitizer_iovec *iov = (struct __sanitizer_iovec *)argp;
8390 if (iov) {
8391 COMMON_INTERCEPTOR_READ_RANGE(ctx, iov, sizeof(*iov));
8392 iov_len = iov->iov_len;
8393 }
8394 ret = REAL(modctl)(operation, argp);
8395 if (iov)
8396 COMMON_INTERCEPTOR_WRITE_RANGE(
8397 ctx, iov->iov_base, Min(iov_len, iov->iov_len));
8398 } else if (operation == modctl_exists) {
8399 ret = REAL(modctl)(operation, argp);
8400 } else {
8401 ret = REAL(modctl)(operation, argp);
8402 }
8404 return ret;
8405 }
8406 #define INIT_MODCTL COMMON_INTERCEPT_FUNCTION(modctl)
8407 #else
8408 #define INIT_MODCTL
8409 #endif
8411 #if SANITIZER_INTERCEPT_STRTONUM
8412 INTERCEPTOR(long long, strtonum, const char *nptr, long long minval,
8413 long long maxval, const char **errstr) {
8414 void *ctx;
8415 COMMON_INTERCEPTOR_ENTER(ctx, strtonum, nptr, minval, maxval, errstr);
8417 // TODO(kamil): Implement strtoll as a common inteceptor
8418 char *real_endptr;
8419 long long ret = (long long)REAL(strtoimax)(nptr, &real_endptr, 10);
8420 StrtolFixAndCheck(ctx, nptr, nullptr, real_endptr, 10);
8422 ret = REAL(strtonum)(nptr, minval, maxval, errstr);
8423 if (errstr) {
8424 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, errstr, sizeof(const char *));
8425 if (*errstr)
8426 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *errstr, internal_strlen(*errstr) + 1);
8427 }
8428 return ret;
8429 }
8430 #define INIT_STRTONUM COMMON_INTERCEPT_FUNCTION(strtonum)
8431 #else
8432 #define INIT_STRTONUM
8433 #endif
8435 #if SANITIZER_INTERCEPT_FPARSELN
8436 INTERCEPTOR(char *, fparseln, __sanitizer_FILE *stream, SIZE_T *len,
8437 SIZE_T *lineno, const char delim[3], int flags) {
8438 void *ctx;
8439 COMMON_INTERCEPTOR_ENTER(ctx, fparseln, stream, len, lineno, delim, flags);
8440 if (lineno)
8441 COMMON_INTERCEPTOR_READ_RANGE(ctx, lineno, sizeof(*lineno));
8442 if (delim)
8443 COMMON_INTERCEPTOR_READ_RANGE(ctx, delim, sizeof(delim[0]) * 3);
8444 char *ret = REAL(fparseln)(stream, len, lineno, delim, flags);
8445 if (ret) {
8446 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, internal_strlen(ret) + 1);
8447 if (len)
8448 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, len, sizeof(*len));
8449 if (lineno)
8450 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, lineno, sizeof(*lineno));
8451 }
8452 return ret;
8453 }
8454 #define INIT_FPARSELN COMMON_INTERCEPT_FUNCTION(fparseln)
8455 #else
8456 #define INIT_FPARSELN
8457 #endif
8459 #if SANITIZER_INTERCEPT_STATVFS1
8460 INTERCEPTOR(int, statvfs1, const char *path, void *buf, int flags) {
8461 void *ctx;
8462 COMMON_INTERCEPTOR_ENTER(ctx, statvfs1, path, buf, flags);
8463 if (path) COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
8464 int res = REAL(statvfs1)(path, buf, flags);
8465 if (!res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, struct_statvfs_sz);
8466 return res;
8467 }
8468 INTERCEPTOR(int, fstatvfs1, int fd, void *buf, int flags) {
8469 void *ctx;
8470 COMMON_INTERCEPTOR_ENTER(ctx, fstatvfs1, fd, buf, flags);
8471 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
8472 int res = REAL(fstatvfs1)(fd, buf, flags);
8473 if (!res) {
8474 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, struct_statvfs_sz);
8475 if (fd >= 0)
8476 COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
8477 }
8478 return res;
8479 }
8480 #define INIT_STATVFS1 \
8481 COMMON_INTERCEPT_FUNCTION(statvfs1); \
8482 COMMON_INTERCEPT_FUNCTION(fstatvfs1);
8483 #else
8484 #define INIT_STATVFS1
8485 #endif
8487 #if SANITIZER_INTERCEPT_STRTOI
8488 INTERCEPTOR(INTMAX_T, strtoi, const char *nptr, char **endptr, int base,
8489 INTMAX_T low, INTMAX_T high, int *rstatus) {
8490 void *ctx;
8491 COMMON_INTERCEPTOR_ENTER(ctx, strtoi, nptr, endptr, base, low, high, rstatus);
8492 char *real_endptr;
8493 INTMAX_T ret = REAL(strtoi)(nptr, &real_endptr, base, low, high, rstatus);
8494 StrtolFixAndCheck(ctx, nptr, endptr, real_endptr, base);
8495 if (rstatus)
8496 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, rstatus, sizeof(*rstatus));
8497 return ret;
8498 }
8500 INTERCEPTOR(UINTMAX_T, strtou, const char *nptr, char **endptr, int base,
8501 UINTMAX_T low, UINTMAX_T high, int *rstatus) {
8502 void *ctx;
8503 COMMON_INTERCEPTOR_ENTER(ctx, strtou, nptr, endptr, base, low, high, rstatus);
8504 char *real_endptr;
8505 UINTMAX_T ret = REAL(strtou)(nptr, &real_endptr, base, low, high, rstatus);
8506 StrtolFixAndCheck(ctx, nptr, endptr, real_endptr, base);
8507 if (rstatus)
8508 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, rstatus, sizeof(*rstatus));
8509 return ret;
8510 }
8511 #define INIT_STRTOI \
8512 COMMON_INTERCEPT_FUNCTION(strtoi); \
8513 COMMON_INTERCEPT_FUNCTION(strtou)
8514 #else
8515 #define INIT_STRTOI
8516 #endif
8518 #if SANITIZER_INTERCEPT_CAPSICUM
8519 #define CAP_RIGHTS_INIT_INTERCEPTOR(cap_rights_init, rights, ...) \
8520 { \
8521 void *ctx; \
8522 COMMON_INTERCEPTOR_ENTER(ctx, cap_rights_init, rights, ##__VA_ARGS__); \
8523 if (rights) \
8524 COMMON_INTERCEPTOR_READ_RANGE(ctx, rights, sizeof(*rights)); \
8525 __sanitizer_cap_rights_t *ret = \
8526 REAL(cap_rights_init)(rights, ##__VA_ARGS__); \
8527 if (ret) \
8528 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, sizeof(*ret)); \
8529 return ret; \
8530 }
8532 #define CAP_RIGHTS_SET_INTERCEPTOR(cap_rights_set, rights, ...) \
8533 { \
8534 void *ctx; \
8535 COMMON_INTERCEPTOR_ENTER(ctx, cap_rights_set, rights, ##__VA_ARGS__); \
8536 if (rights) \
8537 COMMON_INTERCEPTOR_READ_RANGE(ctx, rights, sizeof(*rights)); \
8538 __sanitizer_cap_rights_t *ret = \
8539 REAL(cap_rights_set)(rights, ##__VA_ARGS__); \
8540 if (ret) \
8541 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, sizeof(*ret)); \
8542 return ret; \
8543 }
8545 #define CAP_RIGHTS_CLEAR_INTERCEPTOR(cap_rights_clear, rights, ...) \
8546 { \
8547 void *ctx; \
8548 COMMON_INTERCEPTOR_ENTER(ctx, cap_rights_clear, rights, ##__VA_ARGS__); \
8549 if (rights) \
8550 COMMON_INTERCEPTOR_READ_RANGE(ctx, rights, sizeof(*rights)); \
8551 __sanitizer_cap_rights_t *ret = \
8552 REAL(cap_rights_clear)(rights, ##__VA_ARGS__); \
8553 if (ret) \
8554 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, sizeof(*ret)); \
8555 return ret; \
8556 }
8558 #define CAP_RIGHTS_IS_SET_INTERCEPTOR(cap_rights_is_set, rights, ...) \
8559 { \
8560 void *ctx; \
8561 COMMON_INTERCEPTOR_ENTER(ctx, cap_rights_is_set, rights, ##__VA_ARGS__); \
8562 if (rights) \
8563 COMMON_INTERCEPTOR_READ_RANGE(ctx, rights, sizeof(*rights)); \
8564 return REAL(cap_rights_is_set)(rights, ##__VA_ARGS__); \
8565 }
8567 INTERCEPTOR(__sanitizer_cap_rights_t *, cap_rights_init,
8568 __sanitizer_cap_rights_t *rights) {
8569 CAP_RIGHTS_INIT_INTERCEPTOR(cap_rights_init, rights);
8570 }
8572 INTERCEPTOR(__sanitizer_cap_rights_t *, cap_rights_set,
8573 __sanitizer_cap_rights_t *rights) {
8574 CAP_RIGHTS_SET_INTERCEPTOR(cap_rights_set, rights);
8575 }
8577 INTERCEPTOR(__sanitizer_cap_rights_t *, cap_rights_clear,
8578 __sanitizer_cap_rights_t *rights) {
8579 CAP_RIGHTS_CLEAR_INTERCEPTOR(cap_rights_clear, rights);
8580 }
8582 INTERCEPTOR(bool, cap_rights_is_set,
8583 __sanitizer_cap_rights_t *rights) {
8584 CAP_RIGHTS_IS_SET_INTERCEPTOR(cap_rights_is_set, rights);
8585 }
8587 INTERCEPTOR(int, cap_rights_limit, int fd,
8588 const __sanitizer_cap_rights_t *rights) {
8589 void *ctx;
8590 COMMON_INTERCEPTOR_ENTER(ctx, cap_rights_limit, fd, rights);
8591 if (rights)
8592 COMMON_INTERCEPTOR_READ_RANGE(ctx, rights, sizeof(*rights));
8594 return REAL(cap_rights_limit)(fd, rights);
8595 }
8597 INTERCEPTOR(int, cap_rights_get, int fd, __sanitizer_cap_rights_t *rights) {
8598 void *ctx;
8599 COMMON_INTERCEPTOR_ENTER(ctx, cap_rights_get, fd, rights);
8600 int ret = REAL(cap_rights_get)(fd, rights);
8601 if (!ret && rights)
8602 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, rights, sizeof(*rights));
8604 return ret;
8605 }
8607 INTERCEPTOR(bool, cap_rights_is_valid, const __sanitizer_cap_rights_t *rights) {
8608 void *ctx;
8609 COMMON_INTERCEPTOR_ENTER(ctx, cap_rights_is_valid, rights);
8610 if (rights)
8611 COMMON_INTERCEPTOR_READ_RANGE(ctx, rights, sizeof(*rights));
8613 return REAL(cap_rights_is_valid(rights));
8614 }
8616 INTERCEPTOR(__sanitizer_cap_rights *, cap_rights_merge,
8617 __sanitizer_cap_rights *dst, const __sanitizer_cap_rights *src) {
8618 void *ctx;
8619 COMMON_INTERCEPTOR_ENTER(ctx, cap_rights_merge, dst, src);
8620 if (src)
8621 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, sizeof(*src));
8623 __sanitizer_cap_rights *ret = REAL(cap_rights_merge)(dst, src);
8624 if (dst)
8625 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, sizeof(*dst));
8627 return ret;
8628 }
8630 INTERCEPTOR(__sanitizer_cap_rights *, cap_rights_remove,
8631 __sanitizer_cap_rights *dst, const __sanitizer_cap_rights *src) {
8632 void *ctx;
8633 COMMON_INTERCEPTOR_ENTER(ctx, cap_rights_remove, dst, src);
8634 if (src)
8635 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, sizeof(*src));
8637 __sanitizer_cap_rights *ret = REAL(cap_rights_remove)(dst, src);
8638 if (dst)
8639 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, sizeof(*dst));
8641 return ret;
8642 }
8644 INTERCEPTOR(bool, cap_rights_contains, const __sanitizer_cap_rights *big,
8645 const __sanitizer_cap_rights *little) {
8646 void *ctx;
8647 COMMON_INTERCEPTOR_ENTER(ctx, cap_rights_contains, big, little);
8648 if (little)
8649 COMMON_INTERCEPTOR_READ_RANGE(ctx, little, sizeof(*little));
8650 if (big)
8651 COMMON_INTERCEPTOR_READ_RANGE(ctx, big, sizeof(*big));
8653 return REAL(cap_rights_contains)(big, little);
8654 }
8656 INTERCEPTOR(int, cap_ioctls_limit, int fd, const uptr *cmds, SIZE_T ncmds) {
8657 void *ctx;
8658 COMMON_INTERCEPTOR_ENTER(ctx, cap_ioctls_limit, fd, cmds, ncmds);
8659 if (cmds)
8660 COMMON_INTERCEPTOR_READ_RANGE(ctx, cmds, sizeof(*cmds) * ncmds);
8662 return REAL(cap_ioctls_limit)(fd, cmds, ncmds);
8663 }
8665 INTERCEPTOR(int, cap_ioctls_get, int fd, uptr *cmds, SIZE_T maxcmds) {
8666 void *ctx;
8667 COMMON_INTERCEPTOR_ENTER(ctx, cap_ioctls_get, fd, cmds, maxcmds);
8668 int ret = REAL(cap_ioctls_get)(fd, cmds, maxcmds);
8669 if (!ret && cmds)
8670 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cmds, sizeof(*cmds) * maxcmds);
8672 return ret;
8673 }
8674 #define INIT_CAPSICUM \
8675 COMMON_INTERCEPT_FUNCTION(cap_rights_init); \
8676 COMMON_INTERCEPT_FUNCTION(cap_rights_set); \
8677 COMMON_INTERCEPT_FUNCTION(cap_rights_clear); \
8678 COMMON_INTERCEPT_FUNCTION(cap_rights_is_set); \
8679 COMMON_INTERCEPT_FUNCTION(cap_rights_get); \
8680 COMMON_INTERCEPT_FUNCTION(cap_rights_limit); \
8681 COMMON_INTERCEPT_FUNCTION(cap_rights_contains); \
8682 COMMON_INTERCEPT_FUNCTION(cap_rights_remove); \
8683 COMMON_INTERCEPT_FUNCTION(cap_rights_merge); \
8684 COMMON_INTERCEPT_FUNCTION(cap_rights_is_valid); \
8685 COMMON_INTERCEPT_FUNCTION(cap_ioctls_get); \
8686 COMMON_INTERCEPT_FUNCTION(cap_ioctls_limit)
8687 #else
8688 #define INIT_CAPSICUM
8689 #endif
8691 #if SANITIZER_INTERCEPT_SHA1
8692 INTERCEPTOR(void, SHA1Init, void *context) {
8693 void *ctx;
8694 COMMON_INTERCEPTOR_ENTER(ctx, SHA1Init, context);
8695 REAL(SHA1Init)(context);
8696 if (context)
8697 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, context, SHA1_CTX_sz);
8698 }
8699 INTERCEPTOR(void, SHA1Update, void *context, const u8 *data, unsigned len) {
8700 void *ctx;
8701 COMMON_INTERCEPTOR_ENTER(ctx, SHA1Update, context, data, len);
8702 if (data && len > 0)
8703 COMMON_INTERCEPTOR_READ_RANGE(ctx, data, len);
8704 if (context)
8705 COMMON_INTERCEPTOR_READ_RANGE(ctx, context, SHA1_CTX_sz);
8706 REAL(SHA1Update)(context, data, len);
8707 if (context)
8708 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, context, SHA1_CTX_sz);
8709 }
8710 INTERCEPTOR(void, SHA1Final, u8 digest[20], void *context) {
8711 void *ctx;
8712 COMMON_INTERCEPTOR_ENTER(ctx, SHA1Final, digest, context);
8713 if (context)
8714 COMMON_INTERCEPTOR_READ_RANGE(ctx, context, SHA1_CTX_sz);
8715 REAL(SHA1Final)(digest, context);
8716 if (digest)
8717 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, digest, sizeof(u8) * 20);
8718 }
8719 INTERCEPTOR(void, SHA1Transform, u32 state[5], u8 buffer[64]) {
8720 void *ctx;
8721 COMMON_INTERCEPTOR_ENTER(ctx, SHA1Transform, state, buffer);
8722 if (state)
8723 COMMON_INTERCEPTOR_READ_RANGE(ctx, state, sizeof(u32) * 5);
8724 if (buffer)
8725 COMMON_INTERCEPTOR_READ_RANGE(ctx, buffer, sizeof(u8) * 64);
8726 REAL(SHA1Transform)(state, buffer);
8727 if (state)
8728 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, state, sizeof(u32) * 5);
8729 }
8730 INTERCEPTOR(char *, SHA1End, void *context, char *buf) {
8731 void *ctx;
8732 COMMON_INTERCEPTOR_ENTER(ctx, SHA1End, context, buf);
8733 if (context)
8734 COMMON_INTERCEPTOR_READ_RANGE(ctx, context, SHA1_CTX_sz);
8735 char *ret = REAL(SHA1End)(context, buf);
8736 if (ret)
8737 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, SHA1_return_length);
8738 return ret;
8739 }
8740 INTERCEPTOR(char *, SHA1File, char *filename, char *buf) {
8741 void *ctx;
8742 COMMON_INTERCEPTOR_ENTER(ctx, SHA1File, filename, buf);
8743 if (filename)
8744 COMMON_INTERCEPTOR_READ_RANGE(ctx, filename, internal_strlen(filename) + 1);
8745 char *ret = REAL(SHA1File)(filename, buf);
8746 if (ret)
8747 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, SHA1_return_length);
8748 return ret;
8749 }
8750 INTERCEPTOR(char *, SHA1FileChunk, char *filename, char *buf, OFF_T offset,
8751 OFF_T length) {
8752 void *ctx;
8753 COMMON_INTERCEPTOR_ENTER(ctx, SHA1FileChunk, filename, buf, offset, length);
8754 if (filename)
8755 COMMON_INTERCEPTOR_READ_RANGE(ctx, filename, internal_strlen(filename) + 1);
8756 char *ret = REAL(SHA1FileChunk)(filename, buf, offset, length);
8757 if (ret)
8758 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, SHA1_return_length);
8759 return ret;
8760 }
8761 INTERCEPTOR(char *, SHA1Data, u8 *data, SIZE_T len, char *buf) {
8762 void *ctx;
8763 COMMON_INTERCEPTOR_ENTER(ctx, SHA1Data, data, len, buf);
8764 if (data)
8765 COMMON_INTERCEPTOR_READ_RANGE(ctx, data, len);
8766 char *ret = REAL(SHA1Data)(data, len, buf);
8767 if (ret)
8768 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, SHA1_return_length);
8769 return ret;
8770 }
8771 #define INIT_SHA1 \
8772 COMMON_INTERCEPT_FUNCTION(SHA1Init); \
8773 COMMON_INTERCEPT_FUNCTION(SHA1Update); \
8774 COMMON_INTERCEPT_FUNCTION(SHA1Final); \
8775 COMMON_INTERCEPT_FUNCTION(SHA1Transform); \
8776 COMMON_INTERCEPT_FUNCTION(SHA1End); \
8777 COMMON_INTERCEPT_FUNCTION(SHA1File); \
8778 COMMON_INTERCEPT_FUNCTION(SHA1FileChunk); \
8779 COMMON_INTERCEPT_FUNCTION(SHA1Data)
8780 #else
8781 #define INIT_SHA1
8782 #endif
8784 #if SANITIZER_INTERCEPT_MD4
8785 INTERCEPTOR(void, MD4Init, void *context) {
8786 void *ctx;
8787 COMMON_INTERCEPTOR_ENTER(ctx, MD4Init, context);
8788 REAL(MD4Init)(context);
8789 if (context)
8790 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, context, MD4_CTX_sz);
8791 }
8793 INTERCEPTOR(void, MD4Update, void *context, const unsigned char *data,
8794 unsigned int len) {
8795 void *ctx;
8796 COMMON_INTERCEPTOR_ENTER(ctx, MD4Update, context, data, len);
8797 if (data && len > 0)
8798 COMMON_INTERCEPTOR_READ_RANGE(ctx, data, len);
8799 if (context)
8800 COMMON_INTERCEPTOR_READ_RANGE(ctx, context, MD4_CTX_sz);
8801 REAL(MD4Update)(context, data, len);
8802 if (context)
8803 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, context, MD4_CTX_sz);
8804 }
8806 INTERCEPTOR(void, MD4Final, unsigned char digest[16], void *context) {
8807 void *ctx;
8808 COMMON_INTERCEPTOR_ENTER(ctx, MD4Final, digest, context);
8809 if (context)
8810 COMMON_INTERCEPTOR_READ_RANGE(ctx, context, MD4_CTX_sz);
8811 REAL(MD4Final)(digest, context);
8812 if (digest)
8813 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, digest, sizeof(unsigned char) * 16);
8814 }
8816 INTERCEPTOR(char *, MD4End, void *context, char *buf) {
8817 void *ctx;
8818 COMMON_INTERCEPTOR_ENTER(ctx, MD4End, context, buf);
8819 if (context)
8820 COMMON_INTERCEPTOR_READ_RANGE(ctx, context, MD4_CTX_sz);
8821 char *ret = REAL(MD4End)(context, buf);
8822 if (ret)
8823 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, MD4_return_length);
8824 return ret;
8825 }
8827 INTERCEPTOR(char *, MD4File, const char *filename, char *buf) {
8828 void *ctx;
8829 COMMON_INTERCEPTOR_ENTER(ctx, MD4File, filename, buf);
8830 if (filename)
8831 COMMON_INTERCEPTOR_READ_RANGE(ctx, filename, internal_strlen(filename) + 1);
8832 char *ret = REAL(MD4File)(filename, buf);
8833 if (ret)
8834 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, MD4_return_length);
8835 return ret;
8836 }
8838 INTERCEPTOR(char *, MD4Data, const unsigned char *data, unsigned int len,
8839 char *buf) {
8840 void *ctx;
8841 COMMON_INTERCEPTOR_ENTER(ctx, MD4Data, data, len, buf);
8842 if (data && len > 0)
8843 COMMON_INTERCEPTOR_READ_RANGE(ctx, data, len);
8844 char *ret = REAL(MD4Data)(data, len, buf);
8845 if (ret)
8846 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, MD4_return_length);
8847 return ret;
8848 }
8850 #define INIT_MD4 \
8851 COMMON_INTERCEPT_FUNCTION(MD4Init); \
8852 COMMON_INTERCEPT_FUNCTION(MD4Update); \
8853 COMMON_INTERCEPT_FUNCTION(MD4Final); \
8854 COMMON_INTERCEPT_FUNCTION(MD4End); \
8855 COMMON_INTERCEPT_FUNCTION(MD4File); \
8856 COMMON_INTERCEPT_FUNCTION(MD4Data)
8857 #else
8858 #define INIT_MD4
8859 #endif
8861 #if SANITIZER_INTERCEPT_RMD160
8862 INTERCEPTOR(void, RMD160Init, void *context) {
8863 void *ctx;
8864 COMMON_INTERCEPTOR_ENTER(ctx, RMD160Init, context);
8865 REAL(RMD160Init)(context);
8866 if (context)
8867 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, context, RMD160_CTX_sz);
8868 }
8869 INTERCEPTOR(void, RMD160Update, void *context, const u8 *data, unsigned len) {
8870 void *ctx;
8871 COMMON_INTERCEPTOR_ENTER(ctx, RMD160Update, context, data, len);
8872 if (data && len > 0)
8873 COMMON_INTERCEPTOR_READ_RANGE(ctx, data, len);
8874 if (context)
8875 COMMON_INTERCEPTOR_READ_RANGE(ctx, context, RMD160_CTX_sz);
8876 REAL(RMD160Update)(context, data, len);
8877 if (context)
8878 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, context, RMD160_CTX_sz);
8879 }
8880 INTERCEPTOR(void, RMD160Final, u8 digest[20], void *context) {
8881 void *ctx;
8882 COMMON_INTERCEPTOR_ENTER(ctx, RMD160Final, digest, context);
8883 if (context)
8884 COMMON_INTERCEPTOR_READ_RANGE(ctx, context, RMD160_CTX_sz);
8885 REAL(RMD160Final)(digest, context);
8886 if (digest)
8887 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, digest, sizeof(u8) * 20);
8888 }
8889 INTERCEPTOR(void, RMD160Transform, u32 state[5], u16 buffer[16]) {
8890 void *ctx;
8891 COMMON_INTERCEPTOR_ENTER(ctx, RMD160Transform, state, buffer);
8892 if (state)
8893 COMMON_INTERCEPTOR_READ_RANGE(ctx, state, sizeof(u32) * 5);
8894 if (buffer)
8895 COMMON_INTERCEPTOR_READ_RANGE(ctx, buffer, sizeof(u32) * 16);
8896 REAL(RMD160Transform)(state, buffer);
8897 if (state)
8898 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, state, sizeof(u32) * 5);
8899 }
8900 INTERCEPTOR(char *, RMD160End, void *context, char *buf) {
8901 void *ctx;
8902 COMMON_INTERCEPTOR_ENTER(ctx, RMD160End, context, buf);
8903 if (context)
8904 COMMON_INTERCEPTOR_READ_RANGE(ctx, context, RMD160_CTX_sz);
8905 char *ret = REAL(RMD160End)(context, buf);
8906 if (ret)
8907 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, RMD160_return_length);
8908 return ret;
8909 }
8910 INTERCEPTOR(char *, RMD160File, char *filename, char *buf) {
8911 void *ctx;
8912 COMMON_INTERCEPTOR_ENTER(ctx, RMD160File, filename, buf);
8913 if (filename)
8914 COMMON_INTERCEPTOR_READ_RANGE(ctx, filename, internal_strlen(filename) + 1);
8915 char *ret = REAL(RMD160File)(filename, buf);
8916 if (ret)
8917 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, RMD160_return_length);
8918 return ret;
8919 }
8920 INTERCEPTOR(char *, RMD160FileChunk, char *filename, char *buf, OFF_T offset,
8921 OFF_T length) {
8922 void *ctx;
8923 COMMON_INTERCEPTOR_ENTER(ctx, RMD160FileChunk, filename, buf, offset, length);
8924 if (filename)
8925 COMMON_INTERCEPTOR_READ_RANGE(ctx, filename, internal_strlen(filename) + 1);
8926 char *ret = REAL(RMD160FileChunk)(filename, buf, offset, length);
8927 if (ret)
8928 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, RMD160_return_length);
8929 return ret;
8930 }
8931 INTERCEPTOR(char *, RMD160Data, u8 *data, SIZE_T len, char *buf) {
8932 void *ctx;
8933 COMMON_INTERCEPTOR_ENTER(ctx, RMD160Data, data, len, buf);
8934 if (data && len > 0)
8935 COMMON_INTERCEPTOR_READ_RANGE(ctx, data, len);
8936 char *ret = REAL(RMD160Data)(data, len, buf);
8937 if (ret)
8938 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, RMD160_return_length);
8939 return ret;
8940 }
8941 #define INIT_RMD160 \
8942 COMMON_INTERCEPT_FUNCTION(RMD160Init); \
8943 COMMON_INTERCEPT_FUNCTION(RMD160Update); \
8944 COMMON_INTERCEPT_FUNCTION(RMD160Final); \
8945 COMMON_INTERCEPT_FUNCTION(RMD160Transform); \
8946 COMMON_INTERCEPT_FUNCTION(RMD160End); \
8947 COMMON_INTERCEPT_FUNCTION(RMD160File); \
8948 COMMON_INTERCEPT_FUNCTION(RMD160FileChunk); \
8949 COMMON_INTERCEPT_FUNCTION(RMD160Data)
8950 #else
8951 #define INIT_RMD160
8952 #endif
8954 #if SANITIZER_INTERCEPT_FSEEK
8955 INTERCEPTOR(int, fseek, __sanitizer_FILE *stream, long int offset, int whence) {
8956 void *ctx;
8957 COMMON_INTERCEPTOR_ENTER(ctx, fseek, stream, offset, whence);
8958 return REAL(fseek)(stream, offset, whence);
8959 }
8960 INTERCEPTOR(int, fseeko, __sanitizer_FILE *stream, OFF_T offset, int whence) {
8961 void *ctx;
8962 COMMON_INTERCEPTOR_ENTER(ctx, fseeko, stream, offset, whence);
8963 return REAL(fseeko)(stream, offset, whence);
8964 }
8965 INTERCEPTOR(long int, ftell, __sanitizer_FILE *stream) {
8966 void *ctx;
8967 COMMON_INTERCEPTOR_ENTER(ctx, ftell, stream);
8968 return REAL(ftell)(stream);
8969 }
8970 INTERCEPTOR(OFF_T, ftello, __sanitizer_FILE *stream) {
8971 void *ctx;
8972 COMMON_INTERCEPTOR_ENTER(ctx, ftello, stream);
8973 return REAL(ftello)(stream);
8974 }
8975 INTERCEPTOR(void, rewind, __sanitizer_FILE *stream) {
8976 void *ctx;
8977 COMMON_INTERCEPTOR_ENTER(ctx, rewind, stream);
8978 return REAL(rewind)(stream);
8979 }
8980 INTERCEPTOR(int, fgetpos, __sanitizer_FILE *stream, void *pos) {
8981 void *ctx;
8982 COMMON_INTERCEPTOR_ENTER(ctx, fgetpos, stream, pos);
8983 int ret = REAL(fgetpos)(stream, pos);
8984 if (pos && !ret)
8985 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, pos, fpos_t_sz);
8986 return ret;
8987 }
8988 INTERCEPTOR(int, fsetpos, __sanitizer_FILE *stream, const void *pos) {
8989 void *ctx;
8990 COMMON_INTERCEPTOR_ENTER(ctx, fsetpos, stream, pos);
8991 if (pos)
8992 COMMON_INTERCEPTOR_READ_RANGE(ctx, pos, fpos_t_sz);
8993 return REAL(fsetpos)(stream, pos);
8994 }
8995 #define INIT_FSEEK \
8996 COMMON_INTERCEPT_FUNCTION(fseek); \
8997 COMMON_INTERCEPT_FUNCTION(fseeko); \
8998 COMMON_INTERCEPT_FUNCTION(ftell); \
8999 COMMON_INTERCEPT_FUNCTION(ftello); \
9000 COMMON_INTERCEPT_FUNCTION(rewind); \
9001 COMMON_INTERCEPT_FUNCTION(fgetpos); \
9002 COMMON_INTERCEPT_FUNCTION(fsetpos)
9003 #else
9004 #define INIT_FSEEK
9005 #endif
9007 #if SANITIZER_INTERCEPT_MD2
9008 INTERCEPTOR(void, MD2Init, void *context) {
9009 void *ctx;
9010 COMMON_INTERCEPTOR_ENTER(ctx, MD2Init, context);
9011 REAL(MD2Init)(context);
9012 if (context)
9013 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, context, MD2_CTX_sz);
9014 }
9016 INTERCEPTOR(void, MD2Update, void *context, const unsigned char *data,
9017 unsigned int len) {
9018 void *ctx;
9019 COMMON_INTERCEPTOR_ENTER(ctx, MD2Update, context, data, len);
9020 if (data && len > 0)
9021 COMMON_INTERCEPTOR_READ_RANGE(ctx, data, len);
9022 if (context)
9023 COMMON_INTERCEPTOR_READ_RANGE(ctx, context, MD2_CTX_sz);
9024 REAL(MD2Update)(context, data, len);
9025 if (context)
9026 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, context, MD2_CTX_sz);
9027 }
9029 INTERCEPTOR(void, MD2Final, unsigned char digest[16], void *context) {
9030 void *ctx;
9031 COMMON_INTERCEPTOR_ENTER(ctx, MD2Final, digest, context);
9032 if (context)
9033 COMMON_INTERCEPTOR_READ_RANGE(ctx, context, MD2_CTX_sz);
9034 REAL(MD2Final)(digest, context);
9035 if (digest)
9036 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, digest, sizeof(unsigned char) * 16);
9037 }
9039 INTERCEPTOR(char *, MD2End, void *context, char *buf) {
9040 void *ctx;
9041 COMMON_INTERCEPTOR_ENTER(ctx, MD2End, context, buf);
9042 if (context)
9043 COMMON_INTERCEPTOR_READ_RANGE(ctx, context, MD2_CTX_sz);
9044 char *ret = REAL(MD2End)(context, buf);
9045 if (ret)
9046 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, MD2_return_length);
9047 return ret;
9048 }
9050 INTERCEPTOR(char *, MD2File, const char *filename, char *buf) {
9051 void *ctx;
9052 COMMON_INTERCEPTOR_ENTER(ctx, MD2File, filename, buf);
9053 if (filename)
9054 COMMON_INTERCEPTOR_READ_RANGE(ctx, filename, internal_strlen(filename) + 1);
9055 char *ret = REAL(MD2File)(filename, buf);
9056 if (ret)
9057 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, MD2_return_length);
9058 return ret;
9059 }
9061 INTERCEPTOR(char *, MD2Data, const unsigned char *data, unsigned int len,
9062 char *buf) {
9063 void *ctx;
9064 COMMON_INTERCEPTOR_ENTER(ctx, MD2Data, data, len, buf);
9065 if (data && len > 0)
9066 COMMON_INTERCEPTOR_READ_RANGE(ctx, data, len);
9067 char *ret = REAL(MD2Data)(data, len, buf);
9068 if (ret)
9069 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, MD2_return_length);
9070 return ret;
9071 }
9073 #define INIT_MD2 \
9074 COMMON_INTERCEPT_FUNCTION(MD2Init); \
9075 COMMON_INTERCEPT_FUNCTION(MD2Update); \
9076 COMMON_INTERCEPT_FUNCTION(MD2Final); \
9077 COMMON_INTERCEPT_FUNCTION(MD2End); \
9078 COMMON_INTERCEPT_FUNCTION(MD2File); \
9079 COMMON_INTERCEPT_FUNCTION(MD2Data)
9080 #else
9081 #define INIT_MD2
9082 #endif
9084 #if SANITIZER_INTERCEPT_VIS
9085 INTERCEPTOR(char *, vis, char *dst, int c, int flag, int nextc) {
9086 void *ctx;
9087 COMMON_INTERCEPTOR_ENTER(ctx, vis, dst, c, flag, nextc);
9088 char *end = REAL(vis)(dst, c, flag, nextc);
9089 // dst is NULL terminated and end points to the NULL char
9090 if (dst && end)
9091 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, end - dst + 1);
9092 return end;
9093 }
9094 INTERCEPTOR(char *, nvis, char *dst, SIZE_T dlen, int c, int flag, int nextc) {
9095 void *ctx;
9096 COMMON_INTERCEPTOR_ENTER(ctx, nvis, dst, dlen, c, flag, nextc);
9097 char *end = REAL(nvis)(dst, dlen, c, flag, nextc);
9098 // nvis cannot make sure the dst is NULL terminated
9099 if (dst && end)
9100 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, end - dst + 1);
9101 return end;
9102 }
9103 INTERCEPTOR(int, strvis, char *dst, const char *src, int flag) {
9104 void *ctx;
9105 COMMON_INTERCEPTOR_ENTER(ctx, strvis, dst, src, flag);
9106 if (src)
9107 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, internal_strlen(src) + 1);
9108 int len = REAL(strvis)(dst, src, flag);
9109 if (dst)
9110 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, len + 1);
9111 return len;
9112 }
9113 INTERCEPTOR(int, stravis, char **dst, const char *src, int flag) {
9114 void *ctx;
9115 COMMON_INTERCEPTOR_ENTER(ctx, stravis, dst, src, flag);
9116 if (src)
9117 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, internal_strlen(src) + 1);
9118 int len = REAL(stravis)(dst, src, flag);
9119 if (dst) {
9120 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, sizeof(char *));
9121 if (*dst)
9122 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *dst, len + 1);
9123 }
9124 return len;
9125 }
9126 INTERCEPTOR(int, strnvis, char *dst, SIZE_T dlen, const char *src, int flag) {
9127 void *ctx;
9128 COMMON_INTERCEPTOR_ENTER(ctx, strnvis, dst, dlen, src, flag);
9129 if (src)
9130 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, internal_strlen(src) + 1);
9131 int len = REAL(strnvis)(dst, dlen, src, flag);
9132 // The interface will be valid even if there is no space for NULL char
9133 if (dst && len > 0)
9134 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, len + 1);
9135 return len;
9136 }
9137 INTERCEPTOR(int, strvisx, char *dst, const char *src, SIZE_T len, int flag) {
9138 void *ctx;
9139 COMMON_INTERCEPTOR_ENTER(ctx, strvisx, dst, src, len, flag);
9140 if (src)
9141 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, len);
9142 int ret = REAL(strvisx)(dst, src, len, flag);
9143 if (dst)
9144 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, ret + 1);
9145 return ret;
9146 }
9147 INTERCEPTOR(int, strnvisx, char *dst, SIZE_T dlen, const char *src, SIZE_T len,
9148 int flag) {
9149 void *ctx;
9150 COMMON_INTERCEPTOR_ENTER(ctx, strnvisx, dst, dlen, src, len, flag);
9151 if (src)
9152 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, len);
9153 int ret = REAL(strnvisx)(dst, dlen, src, len, flag);
9154 if (dst && ret >= 0)
9155 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, ret + 1);
9156 return ret;
9157 }
9158 INTERCEPTOR(int, strenvisx, char *dst, SIZE_T dlen, const char *src, SIZE_T len,
9159 int flag, int *cerr_ptr) {
9160 void *ctx;
9161 COMMON_INTERCEPTOR_ENTER(ctx, strenvisx, dst, dlen, src, len, flag, cerr_ptr);
9162 if (src)
9163 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, len);
9164 // FIXME: only need to be checked when "flag | VIS_NOLOCALE" doesn't hold
9165 // according to the implementation
9166 if (cerr_ptr)
9167 COMMON_INTERCEPTOR_READ_RANGE(ctx, cerr_ptr, sizeof(int));
9168 int ret = REAL(strenvisx)(dst, dlen, src, len, flag, cerr_ptr);
9169 if (dst && ret >= 0)
9170 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, ret + 1);
9171 if (cerr_ptr)
9172 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cerr_ptr, sizeof(int));
9173 return ret;
9174 }
9175 INTERCEPTOR(char *, svis, char *dst, int c, int flag, int nextc,
9176 const char *extra) {
9177 void *ctx;
9178 COMMON_INTERCEPTOR_ENTER(ctx, svis, dst, c, flag, nextc, extra);
9179 if (extra)
9180 COMMON_INTERCEPTOR_READ_RANGE(ctx, extra, internal_strlen(extra) + 1);
9181 char *end = REAL(svis)(dst, c, flag, nextc, extra);
9182 if (dst && end)
9183 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, end - dst + 1);
9184 return end;
9185 }
9186 INTERCEPTOR(char *, snvis, char *dst, SIZE_T dlen, int c, int flag, int nextc,
9187 const char *extra) {
9188 void *ctx;
9189 COMMON_INTERCEPTOR_ENTER(ctx, snvis, dst, dlen, c, flag, nextc, extra);
9190 if (extra)
9191 COMMON_INTERCEPTOR_READ_RANGE(ctx, extra, internal_strlen(extra) + 1);
9192 char *end = REAL(snvis)(dst, dlen, c, flag, nextc, extra);
9193 if (dst && end)
9194 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst,
9195 Min((SIZE_T)(end - dst + 1), dlen));
9196 return end;
9197 }
9198 INTERCEPTOR(int, strsvis, char *dst, const char *src, int flag,
9199 const char *extra) {
9200 void *ctx;
9201 COMMON_INTERCEPTOR_ENTER(ctx, strsvis, dst, src, flag, extra);
9202 if (src)
9203 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, internal_strlen(src) + 1);
9204 if (extra)
9205 COMMON_INTERCEPTOR_READ_RANGE(ctx, extra, internal_strlen(extra) + 1);
9206 int len = REAL(strsvis)(dst, src, flag, extra);
9207 if (dst)
9208 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, len + 1);
9209 return len;
9210 }
9211 INTERCEPTOR(int, strsnvis, char *dst, SIZE_T dlen, const char *src, int flag,
9212 const char *extra) {
9213 void *ctx;
9214 COMMON_INTERCEPTOR_ENTER(ctx, strsnvis, dst, dlen, src, flag, extra);
9215 if (src)
9216 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, internal_strlen(src) + 1);
9217 if (extra)
9218 COMMON_INTERCEPTOR_READ_RANGE(ctx, extra, internal_strlen(extra) + 1);
9219 int len = REAL(strsnvis)(dst, dlen, src, flag, extra);
9220 // The interface will be valid even if there is no space for NULL char
9221 if (dst && len >= 0)
9222 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, len + 1);
9223 return len;
9224 }
9225 INTERCEPTOR(int, strsvisx, char *dst, const char *src, SIZE_T len, int flag,
9226 const char *extra) {
9227 void *ctx;
9228 COMMON_INTERCEPTOR_ENTER(ctx, strsvisx, dst, src, len, flag, extra);
9229 if (src)
9230 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, len);
9231 if (extra)
9232 COMMON_INTERCEPTOR_READ_RANGE(ctx, extra, internal_strlen(extra) + 1);
9233 int ret = REAL(strsvisx)(dst, src, len, flag, extra);
9234 if (dst)
9235 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, ret + 1);
9236 return ret;
9237 }
9238 INTERCEPTOR(int, strsnvisx, char *dst, SIZE_T dlen, const char *src, SIZE_T len,
9239 int flag, const char *extra) {
9240 void *ctx;
9241 COMMON_INTERCEPTOR_ENTER(ctx, strsnvisx, dst, dlen, src, len, flag, extra);
9242 if (src)
9243 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, len);
9244 if (extra)
9245 COMMON_INTERCEPTOR_READ_RANGE(ctx, extra, internal_strlen(extra) + 1);
9246 int ret = REAL(strsnvisx)(dst, dlen, src, len, flag, extra);
9247 if (dst && ret >= 0)
9248 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, ret + 1);
9249 return ret;
9250 }
9251 INTERCEPTOR(int, strsenvisx, char *dst, SIZE_T dlen, const char *src,
9252 SIZE_T len, int flag, const char *extra, int *cerr_ptr) {
9253 void *ctx;
9254 COMMON_INTERCEPTOR_ENTER(ctx, strsenvisx, dst, dlen, src, len, flag, extra,
9255 cerr_ptr);
9256 if (src)
9257 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, len);
9258 if (extra)
9259 COMMON_INTERCEPTOR_READ_RANGE(ctx, extra, internal_strlen(extra) + 1);
9260 // FIXME: only need to be checked when "flag | VIS_NOLOCALE" doesn't hold
9261 // according to the implementation
9262 if (cerr_ptr)
9263 COMMON_INTERCEPTOR_READ_RANGE(ctx, cerr_ptr, sizeof(int));
9264 int ret = REAL(strsenvisx)(dst, dlen, src, len, flag, extra, cerr_ptr);
9265 if (dst && ret >= 0)
9266 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, ret + 1);
9267 if (cerr_ptr)
9268 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cerr_ptr, sizeof(int));
9269 return ret;
9270 }
9271 INTERCEPTOR(int, unvis, char *cp, int c, int *astate, int flag) {
9272 void *ctx;
9273 COMMON_INTERCEPTOR_ENTER(ctx, unvis, cp, c, astate, flag);
9274 if (astate)
9275 COMMON_INTERCEPTOR_READ_RANGE(ctx, astate, sizeof(*astate));
9276 int ret = REAL(unvis)(cp, c, astate, flag);
9277 if (ret == unvis_valid || ret == unvis_validpush) {
9278 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cp, sizeof(*cp));
9279 }
9280 return ret;
9281 }
9282 INTERCEPTOR(int, strunvis, char *dst, const char *src) {
9283 void *ctx;
9284 COMMON_INTERCEPTOR_ENTER(ctx, strunvis, dst, src);
9285 if (src)
9286 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, internal_strlen(src) + 1);
9287 int ret = REAL(strunvis)(dst, src);
9288 if (ret != -1)
9289 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, ret + 1);
9290 return ret;
9291 }
9292 INTERCEPTOR(int, strnunvis, char *dst, SIZE_T dlen, const char *src) {
9293 void *ctx;
9294 COMMON_INTERCEPTOR_ENTER(ctx, strnunvis, dst, dlen, src);
9295 if (src)
9296 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, internal_strlen(src) + 1);
9297 int ret = REAL(strnunvis)(dst, dlen, src);
9298 if (ret != -1)
9299 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, ret + 1);
9300 return ret;
9301 }
9302 INTERCEPTOR(int, strunvisx, char *dst, const char *src, int flag) {
9303 void *ctx;
9304 COMMON_INTERCEPTOR_ENTER(ctx, strunvisx, dst, src, flag);
9305 if (src)
9306 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, internal_strlen(src) + 1);
9307 int ret = REAL(strunvisx)(dst, src, flag);
9308 if (ret != -1)
9309 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, ret + 1);
9310 return ret;
9311 }
9312 INTERCEPTOR(int, strnunvisx, char *dst, SIZE_T dlen, const char *src,
9313 int flag) {
9314 void *ctx;
9315 COMMON_INTERCEPTOR_ENTER(ctx, strnunvisx, dst, dlen, src, flag);
9316 if (src)
9317 COMMON_INTERCEPTOR_READ_RANGE(ctx, src, internal_strlen(src) + 1);
9318 int ret = REAL(strnunvisx)(dst, dlen, src, flag);
9319 if (ret != -1)
9320 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, dst, ret + 1);
9321 return ret;
9322 }
9323 #define INIT_VIS \
9324 COMMON_INTERCEPT_FUNCTION(vis); \
9325 COMMON_INTERCEPT_FUNCTION(nvis); \
9326 COMMON_INTERCEPT_FUNCTION(strvis); \
9327 COMMON_INTERCEPT_FUNCTION(stravis); \
9328 COMMON_INTERCEPT_FUNCTION(strnvis); \
9329 COMMON_INTERCEPT_FUNCTION(strvisx); \
9330 COMMON_INTERCEPT_FUNCTION(strnvisx); \
9331 COMMON_INTERCEPT_FUNCTION(strenvisx); \
9332 COMMON_INTERCEPT_FUNCTION(svis); \
9333 COMMON_INTERCEPT_FUNCTION(snvis); \
9334 COMMON_INTERCEPT_FUNCTION(strsvis); \
9335 COMMON_INTERCEPT_FUNCTION(strsnvis); \
9336 COMMON_INTERCEPT_FUNCTION(strsvisx); \
9337 COMMON_INTERCEPT_FUNCTION(strsnvisx); \
9338 COMMON_INTERCEPT_FUNCTION(strsenvisx); \
9339 COMMON_INTERCEPT_FUNCTION(unvis); \
9340 COMMON_INTERCEPT_FUNCTION(strunvis); \
9341 COMMON_INTERCEPT_FUNCTION(strnunvis); \
9342 COMMON_INTERCEPT_FUNCTION(strunvisx); \
9343 COMMON_INTERCEPT_FUNCTION(strnunvisx)
9344 #else
9345 #define INIT_VIS
9346 #endif
9348 #if SANITIZER_INTERCEPT_CDB
9349 INTERCEPTOR(struct __sanitizer_cdbr *, cdbr_open, const char *path, int flags) {
9350 void *ctx;
9351 COMMON_INTERCEPTOR_ENTER(ctx, cdbr_open, path, flags);
9352 if (path)
9353 COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
9354 struct __sanitizer_cdbr *cdbr = REAL(cdbr_open)(path, flags);
9355 if (cdbr)
9356 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cdbr, sizeof(*cdbr));
9357 return cdbr;
9358 }
9360 INTERCEPTOR(struct __sanitizer_cdbr *, cdbr_open_mem, void *base, SIZE_T size,
9361 int flags, void (*unmap)(void *, void *, SIZE_T), void *cookie) {
9362 void *ctx;
9363 COMMON_INTERCEPTOR_ENTER(ctx, cdbr_open_mem, base, size, flags, unmap,
9364 cookie);
9365 if (base && size)
9366 COMMON_INTERCEPTOR_READ_RANGE(ctx, base, size);
9367 struct __sanitizer_cdbr *cdbr =
9368 REAL(cdbr_open_mem)(base, size, flags, unmap, cookie);
9369 if (cdbr)
9370 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cdbr, sizeof(*cdbr));
9371 return cdbr;
9372 }
9374 INTERCEPTOR(u32, cdbr_entries, struct __sanitizer_cdbr *cdbr) {
9375 void *ctx;
9376 COMMON_INTERCEPTOR_ENTER(ctx, cdbr_entries, cdbr);
9377 if (cdbr)
9378 COMMON_INTERCEPTOR_READ_RANGE(ctx, cdbr, sizeof(*cdbr));
9379 return REAL(cdbr_entries)(cdbr);
9380 }
9382 INTERCEPTOR(int, cdbr_get, struct __sanitizer_cdbr *cdbr, u32 index,
9383 const void **data, SIZE_T *datalen) {
9384 void *ctx;
9385 COMMON_INTERCEPTOR_ENTER(ctx, cdbr_get, cdbr, index, data, datalen);
9386 if (cdbr)
9387 COMMON_INTERCEPTOR_READ_RANGE(ctx, cdbr, sizeof(*cdbr));
9388 int ret = REAL(cdbr_get)(cdbr, index, data, datalen);
9389 if (!ret) {
9390 if (data)
9391 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, data, sizeof(*data));
9392 if (datalen)
9393 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, datalen, sizeof(*datalen));
9394 if (data && datalen)
9395 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *data, *datalen);
9396 }
9397 return ret;
9398 }
9400 INTERCEPTOR(int, cdbr_find, struct __sanitizer_cdbr *cdbr, const void *key,
9401 SIZE_T keylen, const void **data, SIZE_T *datalen) {
9402 void *ctx;
9403 COMMON_INTERCEPTOR_ENTER(ctx, cdbr_find, cdbr, key, keylen, data, datalen);
9404 if (cdbr)
9405 COMMON_INTERCEPTOR_READ_RANGE(ctx, cdbr, sizeof(*cdbr));
9406 if (key)
9407 COMMON_INTERCEPTOR_READ_RANGE(ctx, key, keylen);
9408 int ret = REAL(cdbr_find)(cdbr, key, keylen, data, datalen);
9409 if (!ret) {
9410 if (data)
9411 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, data, sizeof(*data));
9412 if (datalen)
9413 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, datalen, sizeof(*datalen));
9414 if (data && datalen)
9415 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, *data, *datalen);
9416 }
9417 return ret;
9418 }
9420 INTERCEPTOR(void, cdbr_close, struct __sanitizer_cdbr *cdbr) {
9421 void *ctx;
9422 COMMON_INTERCEPTOR_ENTER(ctx, cdbr_close, cdbr);
9423 if (cdbr)
9424 COMMON_INTERCEPTOR_READ_RANGE(ctx, cdbr, sizeof(*cdbr));
9425 REAL(cdbr_close)(cdbr);
9426 }
9428 INTERCEPTOR(struct __sanitizer_cdbw *, cdbw_open) {
9429 void *ctx;
9430 COMMON_INTERCEPTOR_ENTER(ctx, cdbw_open);
9431 struct __sanitizer_cdbw *ret = REAL(cdbw_open)();
9432 if (ret)
9433 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, sizeof(*ret));
9434 return ret;
9435 }
9437 INTERCEPTOR(int, cdbw_put, struct __sanitizer_cdbw *cdbw, const void *key,
9438 SIZE_T keylen, const void *data, SIZE_T datalen) {
9439 void *ctx;
9440 COMMON_INTERCEPTOR_ENTER(ctx, cdbw_put, cdbw, key, keylen, data, datalen);
9441 if (cdbw)
9442 COMMON_INTERCEPTOR_READ_RANGE(ctx, cdbw, sizeof(*cdbw));
9443 if (data && datalen)
9444 COMMON_INTERCEPTOR_READ_RANGE(ctx, data, datalen);
9445 if (key && keylen)
9446 COMMON_INTERCEPTOR_READ_RANGE(ctx, key, keylen);
9447 int ret = REAL(cdbw_put)(cdbw, key, keylen, data, datalen);
9448 if (!ret && cdbw)
9449 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cdbw, sizeof(*cdbw));
9450 return ret;
9451 }
9453 INTERCEPTOR(int, cdbw_put_data, struct __sanitizer_cdbw *cdbw, const void *data,
9454 SIZE_T datalen, u32 *index) {
9455 void *ctx;
9456 COMMON_INTERCEPTOR_ENTER(ctx, cdbw_put_data, cdbw, data, datalen, index);
9457 if (cdbw)
9458 COMMON_INTERCEPTOR_READ_RANGE(ctx, cdbw, sizeof(*cdbw));
9459 if (data && datalen)
9460 COMMON_INTERCEPTOR_READ_RANGE(ctx, data, datalen);
9461 int ret = REAL(cdbw_put_data)(cdbw, data, datalen, index);
9462 if (!ret) {
9463 if (index)
9464 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, index, sizeof(*index));
9465 if (cdbw)
9466 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cdbw, sizeof(*cdbw));
9467 }
9468 return ret;
9469 }
9471 INTERCEPTOR(int, cdbw_put_key, struct __sanitizer_cdbw *cdbw, const void *key,
9472 SIZE_T keylen, u32 index) {
9473 void *ctx;
9474 COMMON_INTERCEPTOR_ENTER(ctx, cdbw_put_key, cdbw, key, keylen, index);
9475 if (cdbw)
9476 COMMON_INTERCEPTOR_READ_RANGE(ctx, cdbw, sizeof(*cdbw));
9477 if (key && keylen)
9478 COMMON_INTERCEPTOR_READ_RANGE(ctx, key, keylen);
9479 int ret = REAL(cdbw_put_key)(cdbw, key, keylen, index);
9480 if (!ret && cdbw)
9481 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cdbw, sizeof(*cdbw));
9482 return ret;
9483 }
9485 INTERCEPTOR(int, cdbw_output, struct __sanitizer_cdbw *cdbw, int output,
9486 const char descr[16], u32 (*seedgen)(void)) {
9487 void *ctx;
9488 COMMON_INTERCEPTOR_ENTER(ctx, cdbw_output, cdbw, output, descr, seedgen);
9489 COMMON_INTERCEPTOR_FD_ACCESS(ctx, output);
9490 if (cdbw)
9491 COMMON_INTERCEPTOR_READ_RANGE(ctx, cdbw, sizeof(*cdbw));
9492 if (descr)
9493 COMMON_INTERCEPTOR_READ_RANGE(ctx, descr, internal_strnlen(descr, 16));
9494 if (seedgen)
9495 COMMON_INTERCEPTOR_READ_RANGE(ctx, (void *)seedgen, sizeof(seedgen));
9496 int ret = REAL(cdbw_output)(cdbw, output, descr, seedgen);
9497 if (!ret) {
9498 if (cdbw)
9499 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, cdbw, sizeof(*cdbw));
9500 if (output >= 0)
9501 COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, output);
9502 }
9503 return ret;
9504 }
9506 INTERCEPTOR(void, cdbw_close, struct __sanitizer_cdbw *cdbw) {
9507 void *ctx;
9508 COMMON_INTERCEPTOR_ENTER(ctx, cdbw_close, cdbw);
9509 if (cdbw)
9510 COMMON_INTERCEPTOR_READ_RANGE(ctx, cdbw, sizeof(*cdbw));
9511 REAL(cdbw_close)(cdbw);
9512 }
9514 #define INIT_CDB \
9515 COMMON_INTERCEPT_FUNCTION(cdbr_open); \
9516 COMMON_INTERCEPT_FUNCTION(cdbr_open_mem); \
9517 COMMON_INTERCEPT_FUNCTION(cdbr_entries); \
9518 COMMON_INTERCEPT_FUNCTION(cdbr_get); \
9519 COMMON_INTERCEPT_FUNCTION(cdbr_find); \
9520 COMMON_INTERCEPT_FUNCTION(cdbr_close); \
9521 COMMON_INTERCEPT_FUNCTION(cdbw_open); \
9522 COMMON_INTERCEPT_FUNCTION(cdbw_put); \
9523 COMMON_INTERCEPT_FUNCTION(cdbw_put_data); \
9524 COMMON_INTERCEPT_FUNCTION(cdbw_put_key); \
9525 COMMON_INTERCEPT_FUNCTION(cdbw_output); \
9526 COMMON_INTERCEPT_FUNCTION(cdbw_close)
9527 #else
9528 #define INIT_CDB
9529 #endif
9531 #if SANITIZER_INTERCEPT_GETFSENT
9532 INTERCEPTOR(void *, getfsent) {
9533 void *ctx;
9534 COMMON_INTERCEPTOR_ENTER(ctx, getfsent);
9535 void *ret = REAL(getfsent)();
9536 if (ret)
9537 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, struct_fstab_sz);
9538 return ret;
9539 }
9541 INTERCEPTOR(void *, getfsspec, const char *spec) {
9542 void *ctx;
9543 COMMON_INTERCEPTOR_ENTER(ctx, getfsspec, spec);
9544 if (spec)
9545 COMMON_INTERCEPTOR_READ_RANGE(ctx, spec, internal_strlen(spec) + 1);
9546 void *ret = REAL(getfsspec)(spec);
9547 if (ret)
9548 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, struct_fstab_sz);
9549 return ret;
9550 }
9552 INTERCEPTOR(void *, getfsfile, const char *file) {
9553 void *ctx;
9554 COMMON_INTERCEPTOR_ENTER(ctx, getfsfile, file);
9555 if (file)
9556 COMMON_INTERCEPTOR_READ_RANGE(ctx, file, internal_strlen(file) + 1);
9557 void *ret = REAL(getfsfile)(file);
9558 if (ret)
9559 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ret, struct_fstab_sz);
9560 return ret;
9561 }
9563 #define INIT_GETFSENT \
9564 COMMON_INTERCEPT_FUNCTION(getfsent); \
9565 COMMON_INTERCEPT_FUNCTION(getfsspec); \
9566 COMMON_INTERCEPT_FUNCTION(getfsfile);
9567 #else
9568 #define INIT_GETFSENT
9569 #endif
9571 #if SANITIZER_INTERCEPT_ARC4RANDOM
9572 INTERCEPTOR(void, arc4random_buf, void *buf, SIZE_T len) {
9573 void *ctx;
9574 COMMON_INTERCEPTOR_ENTER(ctx, arc4random_buf, buf, len);
9575 REAL(arc4random_buf)(buf, len);
9576 if (buf && len)
9577 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, len);
9578 }
9580 INTERCEPTOR(void, arc4random_addrandom, u8 *dat, int datlen) {
9581 void *ctx;
9582 COMMON_INTERCEPTOR_ENTER(ctx, arc4random_addrandom, dat, datlen);
9583 if (dat && datlen)
9584 COMMON_INTERCEPTOR_READ_RANGE(ctx, dat, datlen);
9585 REAL(arc4random_addrandom)(dat, datlen);
9586 }
9588 #define INIT_ARC4RANDOM \
9589 COMMON_INTERCEPT_FUNCTION(arc4random_buf); \
9590 COMMON_INTERCEPT_FUNCTION(arc4random_addrandom);
9591 #else
9592 #define INIT_ARC4RANDOM
9593 #endif
9595 #if SANITIZER_INTERCEPT_POPEN
9596 INTERCEPTOR(__sanitizer_FILE *, popen, const char *command, const char *type) {
9597 void *ctx;
9598 COMMON_INTERCEPTOR_ENTER(ctx, popen, command, type);
9599 if (command)
9600 COMMON_INTERCEPTOR_READ_RANGE(ctx, command, internal_strlen(command) + 1);
9601 if (type)
9602 COMMON_INTERCEPTOR_READ_RANGE(ctx, type, internal_strlen(type) + 1);
9603 __sanitizer_FILE *res = REAL(popen)(command, type);
9604 COMMON_INTERCEPTOR_FILE_OPEN(ctx, res, nullptr);
9605 if (res) unpoison_file(res);
9606 return res;
9607 }
9608 #define INIT_POPEN COMMON_INTERCEPT_FUNCTION(popen)
9609 #else
9610 #define INIT_POPEN
9611 #endif
9613 #if SANITIZER_INTERCEPT_POPENVE
9614 INTERCEPTOR(__sanitizer_FILE *, popenve, const char *path,
9615 char *const *argv, char *const *envp, const char *type) {
9616 void *ctx;
9617 COMMON_INTERCEPTOR_ENTER(ctx, popenve, path, argv, envp, type);
9618 if (path)
9619 COMMON_INTERCEPTOR_READ_RANGE(ctx, path, internal_strlen(path) + 1);
9620 if (argv) {
9621 for (char *const *pa = argv; ; ++pa) {
9622 COMMON_INTERCEPTOR_READ_RANGE(ctx, pa, sizeof(char **));
9623 if (!*pa)
9624 break;
9625 COMMON_INTERCEPTOR_READ_RANGE(ctx, *pa, internal_strlen(*pa) + 1);
9626 }
9627 }
9628 if (envp) {
9629 for (char *const *pa = envp; ; ++pa) {
9630 COMMON_INTERCEPTOR_READ_RANGE(ctx, pa, sizeof(char **));
9631 if (!*pa)
9632 break;
9633 COMMON_INTERCEPTOR_READ_RANGE(ctx, *pa, internal_strlen(*pa) + 1);
9634 }
9635 }
9636 if (type)
9637 COMMON_INTERCEPTOR_READ_RANGE(ctx, type, internal_strlen(type) + 1);
9638 __sanitizer_FILE *res = REAL(popenve)(path, argv, envp, type);
9639 COMMON_INTERCEPTOR_FILE_OPEN(ctx, res, nullptr);
9640 if (res) unpoison_file(res);
9641 return res;
9642 }
9643 #define INIT_POPENVE COMMON_INTERCEPT_FUNCTION(popenve)
9644 #else
9645 #define INIT_POPENVE
9646 #endif
9648 #if SANITIZER_INTERCEPT_PCLOSE
9649 INTERCEPTOR(int, pclose, __sanitizer_FILE *fp) {
9650 void *ctx;
9651 COMMON_INTERCEPTOR_ENTER(ctx, pclose, fp);
9652 COMMON_INTERCEPTOR_FILE_CLOSE(ctx, fp);
9653 const FileMetadata *m = GetInterceptorMetadata(fp);
9654 int res = REAL(pclose)(fp);
9655 if (m) {
9656 COMMON_INTERCEPTOR_INITIALIZE_RANGE(*m->addr, *m->size);
9657 DeleteInterceptorMetadata(fp);
9658 }
9659 return res;
9660 }
9661 #define INIT_PCLOSE COMMON_INTERCEPT_FUNCTION(pclose);
9662 #else
9663 #define INIT_PCLOSE
9664 #endif
9666 #if SANITIZER_INTERCEPT_FUNOPEN
9667 typedef int (*funopen_readfn)(void *cookie, char *buf, int len);
9668 typedef int (*funopen_writefn)(void *cookie, const char *buf, int len);
9669 typedef OFF_T (*funopen_seekfn)(void *cookie, OFF_T offset, int whence);
9670 typedef int (*funopen_closefn)(void *cookie);
9672 struct WrappedFunopenCookie {
9673 void *real_cookie;
9674 funopen_readfn real_read;
9675 funopen_writefn real_write;
9676 funopen_seekfn real_seek;
9677 funopen_closefn real_close;
9678 };
9680 static int wrapped_funopen_read(void *cookie, char *buf, int len) {
9681 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
9682 WrappedFunopenCookie *wrapped_cookie = (WrappedFunopenCookie *)cookie;
9683 funopen_readfn real_read = wrapped_cookie->real_read;
9684 return real_read(wrapped_cookie->real_cookie, buf, len);
9685 }
9687 static int wrapped_funopen_write(void *cookie, const char *buf, int len) {
9688 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
9689 WrappedFunopenCookie *wrapped_cookie = (WrappedFunopenCookie *)cookie;
9690 funopen_writefn real_write = wrapped_cookie->real_write;
9691 return real_write(wrapped_cookie->real_cookie, buf, len);
9692 }
9694 static OFF_T wrapped_funopen_seek(void *cookie, OFF_T offset, int whence) {
9695 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
9696 WrappedFunopenCookie *wrapped_cookie = (WrappedFunopenCookie *)cookie;
9697 funopen_seekfn real_seek = wrapped_cookie->real_seek;
9698 return real_seek(wrapped_cookie->real_cookie, offset, whence);
9699 }
9701 static int wrapped_funopen_close(void *cookie) {
9702 COMMON_INTERCEPTOR_UNPOISON_PARAM(1);
9703 WrappedFunopenCookie *wrapped_cookie = (WrappedFunopenCookie *)cookie;
9704 funopen_closefn real_close = wrapped_cookie->real_close;
9705 int res = real_close(wrapped_cookie->real_cookie);
9706 InternalFree(wrapped_cookie);
9707 return res;
9708 }
9710 INTERCEPTOR(__sanitizer_FILE *, funopen, void *cookie, funopen_readfn readfn,
9711 funopen_writefn writefn, funopen_seekfn seekfn,
9712 funopen_closefn closefn) {
9713 void *ctx;
9714 COMMON_INTERCEPTOR_ENTER(ctx, funopen, cookie, readfn, writefn, seekfn,
9715 closefn);
9717 WrappedFunopenCookie *wrapped_cookie =
9718 (WrappedFunopenCookie *)InternalAlloc(sizeof(WrappedFunopenCookie));
9719 wrapped_cookie->real_cookie = cookie;
9720 wrapped_cookie->real_read = readfn;
9721 wrapped_cookie->real_write = writefn;
9722 wrapped_cookie->real_seek = seekfn;
9723 wrapped_cookie->real_close = closefn;
9725 __sanitizer_FILE *res =
9726 REAL(funopen)(wrapped_cookie,
9727 readfn ? wrapped_funopen_read : nullptr,
9728 writefn ? wrapped_funopen_write : nullptr,
9729 seekfn ? wrapped_funopen_seek : nullptr,
9730 closefn ? wrapped_funopen_close : nullptr);
9731 if (res)
9732 unpoison_file(res);
9733 return res;
9734 }
9735 #define INIT_FUNOPEN COMMON_INTERCEPT_FUNCTION(funopen)
9736 #else
9737 #define INIT_FUNOPEN
9738 #endif
9740 #if SANITIZER_INTERCEPT_FUNOPEN2
9741 typedef SSIZE_T (*funopen2_readfn)(void *cookie, void *buf, SIZE_T len);
9742 typedef SSIZE_T (*funopen2_writefn)(void *cookie, const void *buf, SIZE_T len);
9743 typedef OFF_T (*funopen2_seekfn)(void *cookie, OFF_T offset, int whence);
9744 typedef int (*funopen2_flushfn)(void *cookie);
9745 typedef int (*funopen2_closefn)(void *cookie);
9747 struct WrappedFunopen2Cookie {
9748 void *real_cookie;
9749 funopen2_readfn real_read;
9750 funopen2_writefn real_write;
9751 funopen2_seekfn real_seek;
9752 funopen2_flushfn real_flush;
9753 funopen2_closefn real_close;
9754 };
9756 static SSIZE_T wrapped_funopen2_read(void *cookie, void *buf, SIZE_T len) {
9757 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
9758 WrappedFunopen2Cookie *wrapped_cookie = (WrappedFunopen2Cookie *)cookie;
9759 funopen2_readfn real_read = wrapped_cookie->real_read;
9760 return real_read(wrapped_cookie->real_cookie, buf, len);
9761 }
9763 static SSIZE_T wrapped_funopen2_write(void *cookie, const void *buf,
9764 SIZE_T len) {
9765 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
9766 WrappedFunopen2Cookie *wrapped_cookie = (WrappedFunopen2Cookie *)cookie;
9767 funopen2_writefn real_write = wrapped_cookie->real_write;
9768 return real_write(wrapped_cookie->real_cookie, buf, len);
9769 }
9771 static OFF_T wrapped_funopen2_seek(void *cookie, OFF_T offset, int whence) {
9772 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
9773 WrappedFunopen2Cookie *wrapped_cookie = (WrappedFunopen2Cookie *)cookie;
9774 funopen2_seekfn real_seek = wrapped_cookie->real_seek;
9775 return real_seek(wrapped_cookie->real_cookie, offset, whence);
9776 }
9778 static int wrapped_funopen2_flush(void *cookie) {
9779 COMMON_INTERCEPTOR_UNPOISON_PARAM(1);
9780 WrappedFunopen2Cookie *wrapped_cookie = (WrappedFunopen2Cookie *)cookie;
9781 funopen2_flushfn real_flush = wrapped_cookie->real_flush;
9782 return real_flush(wrapped_cookie->real_cookie);
9783 }
9785 static int wrapped_funopen2_close(void *cookie) {
9786 COMMON_INTERCEPTOR_UNPOISON_PARAM(1);
9787 WrappedFunopen2Cookie *wrapped_cookie = (WrappedFunopen2Cookie *)cookie;
9788 funopen2_closefn real_close = wrapped_cookie->real_close;
9789 int res = real_close(wrapped_cookie->real_cookie);
9790 InternalFree(wrapped_cookie);
9791 return res;
9792 }
9794 INTERCEPTOR(__sanitizer_FILE *, funopen2, void *cookie, funopen2_readfn readfn,
9795 funopen2_writefn writefn, funopen2_seekfn seekfn,
9796 funopen2_flushfn flushfn, funopen2_closefn closefn) {
9797 void *ctx;
9798 COMMON_INTERCEPTOR_ENTER(ctx, funopen2, cookie, readfn, writefn, seekfn,
9799 flushfn, closefn);
9801 WrappedFunopen2Cookie *wrapped_cookie =
9802 (WrappedFunopen2Cookie *)InternalAlloc(sizeof(WrappedFunopen2Cookie));
9803 wrapped_cookie->real_cookie = cookie;
9804 wrapped_cookie->real_read = readfn;
9805 wrapped_cookie->real_write = writefn;
9806 wrapped_cookie->real_seek = seekfn;
9807 wrapped_cookie->real_flush = flushfn;
9808 wrapped_cookie->real_close = closefn;
9810 __sanitizer_FILE *res =
9811 REAL(funopen2)(wrapped_cookie,
9812 readfn ? wrapped_funopen2_read : nullptr,
9813 writefn ? wrapped_funopen2_write : nullptr,
9814 seekfn ? wrapped_funopen2_seek : nullptr,
9815 flushfn ? wrapped_funopen2_flush : nullptr,
9816 closefn ? wrapped_funopen2_close : nullptr);
9817 if (res)
9818 unpoison_file(res);
9819 return res;
9820 }
9821 #define INIT_FUNOPEN2 COMMON_INTERCEPT_FUNCTION(funopen2)
9822 #else
9823 #define INIT_FUNOPEN2
9824 #endif
9826 #if SANITIZER_INTERCEPT_FDEVNAME
9827 INTERCEPTOR(char *, fdevname, int fd) {
9828 void *ctx;
9829 COMMON_INTERCEPTOR_ENTER(ctx, fdevname, fd);
9830 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
9831 char *name = REAL(fdevname)(fd);
9832 if (name) {
9833 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, name, internal_strlen(name) + 1);
9834 if (fd > 0)
9835 COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
9836 }
9837 return name;
9838 }
9840 INTERCEPTOR(char *, fdevname_r, int fd, char *buf, SIZE_T len) {
9841 void *ctx;
9842 COMMON_INTERCEPTOR_ENTER(ctx, fdevname_r, fd, buf, len);
9843 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
9844 char *name = REAL(fdevname_r)(fd, buf, len);
9845 if (name && buf && len > 0) {
9846 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, internal_strlen(buf) + 1);
9847 if (fd > 0)
9848 COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
9849 }
9850 return name;
9851 }
9853 #define INIT_FDEVNAME \
9854 COMMON_INTERCEPT_FUNCTION(fdevname); \
9855 COMMON_INTERCEPT_FUNCTION(fdevname_r);
9856 #else
9857 #define INIT_FDEVNAME
9858 #endif
9860 #if SANITIZER_INTERCEPT_GETUSERSHELL
9861 INTERCEPTOR(char *, getusershell,) {
9862 void *ctx;
9863 COMMON_INTERCEPTOR_ENTER(ctx, getusershell,);
9864 char *res = REAL(getusershell)();
9865 if (res)
9866 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
9867 return res;
9868 }
9870 #define INIT_GETUSERSHELL COMMON_INTERCEPT_FUNCTION(getusershell);
9871 #else
9872 #define INIT_GETUSERSHELL
9873 #endif
9875 #if SANITIZER_INTERCEPT_SL_INIT
9876 INTERCEPTOR(void *, sl_init) {
9877 void *ctx;
9878 COMMON_INTERCEPTOR_ENTER(ctx, sl_init);
9879 void *res = REAL(sl_init)();
9880 if (res)
9881 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, __sanitizer::struct_StringList_sz);
9882 return res;
9883 }
9885 INTERCEPTOR(int, sl_add, void *sl, char *item) {
9886 void *ctx;
9887 COMMON_INTERCEPTOR_ENTER(ctx, sl_add, sl, item);
9888 if (sl)
9889 COMMON_INTERCEPTOR_READ_RANGE(ctx, sl, __sanitizer::struct_StringList_sz);
9890 if (item)
9891 COMMON_INTERCEPTOR_READ_RANGE(ctx, item, internal_strlen(item) + 1);
9892 int res = REAL(sl_add)(sl, item);
9893 if (!res)
9894 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, sl, __sanitizer::struct_StringList_sz);
9895 return res;
9896 }
9898 INTERCEPTOR(char *, sl_find, void *sl, const char *item) {
9899 void *ctx;
9900 COMMON_INTERCEPTOR_ENTER(ctx, sl_find, sl, item);
9901 if (sl)
9902 COMMON_INTERCEPTOR_READ_RANGE(ctx, sl, __sanitizer::struct_StringList_sz);
9903 if (item)
9904 COMMON_INTERCEPTOR_READ_RANGE(ctx, item, internal_strlen(item) + 1);
9905 char *res = REAL(sl_find)(sl, item);
9906 if (res)
9907 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, res, internal_strlen(res) + 1);
9908 return res;
9909 }
9911 INTERCEPTOR(void, sl_free, void *sl, int freeall) {
9912 void *ctx;
9913 COMMON_INTERCEPTOR_ENTER(ctx, sl_free, sl, freeall);
9914 if (sl)
9915 COMMON_INTERCEPTOR_READ_RANGE(ctx, sl, __sanitizer::struct_StringList_sz);
9916 REAL(sl_free)(sl, freeall);
9917 }
9919 #define INIT_SL_INIT \
9920 COMMON_INTERCEPT_FUNCTION(sl_init); \
9921 COMMON_INTERCEPT_FUNCTION(sl_add); \
9922 COMMON_INTERCEPT_FUNCTION(sl_find); \
9923 COMMON_INTERCEPT_FUNCTION(sl_free);
9924 #else
9925 #define INIT_SL_INIT
9926 #endif
9928 #if SANITIZER_INTERCEPT_GETRANDOM
9929 INTERCEPTOR(SSIZE_T, getrandom, void *buf, SIZE_T buflen, unsigned int flags) {
9930 void *ctx;
9931 COMMON_INTERCEPTOR_ENTER(ctx, getrandom, buf, buflen, flags);
9932 // If GRND_NONBLOCK is set in the flags, it is non blocking.
9933 static const int grnd_nonblock = 1;
9934 SSIZE_T n;
9935 if ((flags & grnd_nonblock))
9936 n = REAL(getrandom)(buf, buflen, flags);
9937 else
9938 n = COMMON_INTERCEPTOR_BLOCK_REAL(getrandom)(buf, buflen, flags);
9939 if (n > 0) {
9940 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, n);
9941 }
9942 return n;
9943 }
9944 #define INIT_GETRANDOM COMMON_INTERCEPT_FUNCTION(getrandom)
9945 #else
9946 #define INIT_GETRANDOM
9947 #endif
9949 #if SANITIZER_INTERCEPT_GETENTROPY
9950 INTERCEPTOR(int, getentropy, void *buf, SIZE_T buflen) {
9951 void *ctx;
9952 COMMON_INTERCEPTOR_ENTER(ctx, getentropy, buf, buflen);
9953 int r = REAL(getentropy)(buf, buflen);
9954 if (r == 0) {
9955 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, buflen);
9956 }
9957 return r;
9958 }
9959 #define INIT_GETENTROPY COMMON_INTERCEPT_FUNCTION(getentropy)
9960 #else
9961 #define INIT_GETENTROPY
9962 #endif
9964 #if SANITIZER_INTERCEPT_QSORT_R
9965 typedef int (*qsort_r_compar_f)(const void *, const void *, void *);
9966 struct qsort_r_compar_params {
9967 SIZE_T size;
9968 qsort_r_compar_f compar;
9969 void *arg;
9970 };
9971 static int wrapped_qsort_r_compar(const void *a, const void *b, void *arg) {
9972 qsort_r_compar_params *params = (qsort_r_compar_params *)arg;
9973 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
9974 COMMON_INTERCEPTOR_INITIALIZE_RANGE(a, params->size);
9975 COMMON_INTERCEPTOR_INITIALIZE_RANGE(b, params->size);
9976 return params->compar(a, b, params->arg);
9977 }
9979 INTERCEPTOR(void, qsort_r, void *base, SIZE_T nmemb, SIZE_T size,
9980 qsort_r_compar_f compar, void *arg) {
9981 void *ctx;
9982 COMMON_INTERCEPTOR_ENTER(ctx, qsort_r, base, nmemb, size, compar, arg);
9983 // Run the comparator over all array elements to detect any memory issues.
9984 if (nmemb > 1) {
9985 for (SIZE_T i = 0; i < nmemb - 1; ++i) {
9986 void *p = (void *)((char *)base + i * size);
9987 void *q = (void *)((char *)base + (i + 1) * size);
9988 COMMON_INTERCEPTOR_UNPOISON_PARAM(3);
9989 compar(p, q, arg);
9990 }
9991 }
9992 qsort_r_compar_params params = {size, compar, arg};
9993 REAL(qsort_r)(base, nmemb, size, wrapped_qsort_r_compar, ¶ms);
9994 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, base, nmemb * size);
9995 }
9996 # define INIT_QSORT_R COMMON_INTERCEPT_FUNCTION(qsort_r)
9997 #else
9998 # define INIT_QSORT_R
9999 #endif
10001 #if SANITIZER_INTERCEPT_QSORT && SANITIZER_INTERCEPT_QSORT_R
10002 INTERCEPTOR(void, qsort, void *base, SIZE_T nmemb, SIZE_T size,
10003 qsort_r_compar_f compar) {
10004 void *ctx;
10005 COMMON_INTERCEPTOR_ENTER(ctx, qsort, base, nmemb, size, compar);
10006 WRAP(qsort_r)(base, nmemb, size, compar, nullptr);
10007 }
10008 # define INIT_QSORT COMMON_INTERCEPT_FUNCTION(qsort)
10009 #elif SANITIZER_INTERCEPT_QSORT && !SANITIZER_INTERCEPT_QSORT_R
10010 // Glibc qsort uses a temporary buffer allocated either on stack or on heap.
10011 // Poisoned memory from there may get copied into the comparator arguments,
10012 // where it needs to be dealt with. But even that is not enough - the results of
10013 // the sort may be copied into the input/output array based on the results of
10014 // the comparator calls, but directly from the temp memory, bypassing the
10015 // unpoisoning done in wrapped_qsort_compar. We deal with this by, again,
10016 // unpoisoning the entire array after the sort is done.
10017 //
10018 // We can not check that the entire array is initialized at the beginning. IMHO,
10019 // it's fine for parts of the sorted objects to contain uninitialized memory,
10020 // ex. as padding in structs.
10021 typedef int (*qsort_compar_f)(const void *, const void *);
10022 static THREADLOCAL qsort_compar_f qsort_compar;
10023 static THREADLOCAL SIZE_T qsort_size;
10024 static int wrapped_qsort_compar(const void *a, const void *b) {
10025 COMMON_INTERCEPTOR_UNPOISON_PARAM(2);
10026 COMMON_INTERCEPTOR_INITIALIZE_RANGE(a, qsort_size);
10027 COMMON_INTERCEPTOR_INITIALIZE_RANGE(b, qsort_size);
10028 return qsort_compar(a, b);
10029 }
10031 INTERCEPTOR(void, qsort, void *base, SIZE_T nmemb, SIZE_T size,
10032 qsort_compar_f compar) {
10033 void *ctx;
10034 COMMON_INTERCEPTOR_ENTER(ctx, qsort, base, nmemb, size, compar);
10035 // Run the comparator over all array elements to detect any memory issues.
10036 if (nmemb > 1) {
10037 for (SIZE_T i = 0; i < nmemb - 1; ++i) {
10038 void *p = (void *)((char *)base + i * size);
10039 void *q = (void *)((char *)base + (i + 1) * size);
10040 COMMON_INTERCEPTOR_UNPOISON_PARAM(2);
10041 compar(p, q);
10042 }
10043 }
10044 qsort_compar_f old_compar = qsort_compar;
10045 SIZE_T old_size = qsort_size;
10046 // Handle qsort() implementations that recurse using an
10047 // interposable function call:
10048 bool already_wrapped = compar == wrapped_qsort_compar;
10049 if (already_wrapped) {
10050 // This case should only happen if the qsort() implementation calls itself
10051 // using a preemptible function call (e.g. the FreeBSD libc version).
10052 // Check that the size and comparator arguments are as expected.
10053 CHECK_NE(compar, qsort_compar);
10054 CHECK_EQ(qsort_size, size);
10055 } else {
10056 qsort_compar = compar;
10057 qsort_size = size;
10058 }
10059 REAL(qsort)(base, nmemb, size, wrapped_qsort_compar);
10060 if (!already_wrapped) {
10061 qsort_compar = old_compar;
10062 qsort_size = old_size;
10063 }
10064 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, base, nmemb * size);
10065 }
10066 # define INIT_QSORT COMMON_INTERCEPT_FUNCTION(qsort)
10067 #else
10068 # define INIT_QSORT
10069 #endif
10071 #if SANITIZER_INTERCEPT_BSEARCH
10072 typedef int (*bsearch_compar_f)(const void *, const void *);
10073 struct bsearch_compar_params {
10074 const void *key;
10075 bsearch_compar_f compar;
10076 };
10078 static int wrapped_bsearch_compar(const void *key, const void *b) {
10079 const bsearch_compar_params *params = (const bsearch_compar_params *)key;
10080 COMMON_INTERCEPTOR_UNPOISON_PARAM(2);
10081 return params->compar(params->key, b);
10082 }
10084 INTERCEPTOR(void *, bsearch, const void *key, const void *base, SIZE_T nmemb,
10085 SIZE_T size, bsearch_compar_f compar) {
10086 void *ctx;
10087 COMMON_INTERCEPTOR_ENTER(ctx, bsearch, key, base, nmemb, size, compar);
10088 bsearch_compar_params params = {key, compar};
10089 return REAL(bsearch)(¶ms, base, nmemb, size, wrapped_bsearch_compar);
10090 }
10091 # define INIT_BSEARCH COMMON_INTERCEPT_FUNCTION(bsearch)
10092 #else
10093 # define INIT_BSEARCH
10094 #endif
10096 #if SANITIZER_INTERCEPT_SIGALTSTACK
10097 INTERCEPTOR(int, sigaltstack, void *ss, void *oss) {
10098 void *ctx;
10099 COMMON_INTERCEPTOR_ENTER(ctx, sigaltstack, ss, oss);
10100 int r = REAL(sigaltstack)(ss, oss);
10101 if (r == 0 && oss != nullptr) {
10102 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, oss, struct_stack_t_sz);
10103 }
10104 return r;
10105 }
10106 #define INIT_SIGALTSTACK COMMON_INTERCEPT_FUNCTION(sigaltstack)
10107 #else
10108 #define INIT_SIGALTSTACK
10109 #endif
10111 #if SANITIZER_INTERCEPT_PROCCTL
10112 INTERCEPTOR(int, procctl, int idtype, u64 id, int cmd, uptr data) {
10113 void *ctx;
10114 COMMON_INTERCEPTOR_ENTER(ctx, procctl, idtype, id, cmd, data);
10115 static const int PROC_REAP_ACQUIRE = 2;
10116 static const int PROC_REAP_RELEASE = 3;
10117 static const int PROC_REAP_STATUS = 4;
10118 static const int PROC_REAP_GETPIDS = 5;
10119 static const int PROC_REAP_KILL = 6;
10120 if (cmd < PROC_REAP_ACQUIRE || cmd > PROC_REAP_KILL) {
10121 COMMON_INTERCEPTOR_READ_RANGE(ctx, (void *)data, sizeof(int));
10122 } else {
10123 // reap_acquire/reap_release bears no arguments.
10124 if (cmd > PROC_REAP_RELEASE) {
10125 unsigned int reapsz;
10126 switch (cmd) {
10127 case PROC_REAP_STATUS:
10128 reapsz = struct_procctl_reaper_status_sz;
10129 break;
10130 case PROC_REAP_GETPIDS:
10131 reapsz = struct_procctl_reaper_pids_sz;
10132 break;
10133 case PROC_REAP_KILL:
10134 reapsz = struct_procctl_reaper_kill_sz;
10135 break;
10136 }
10137 COMMON_INTERCEPTOR_READ_RANGE(ctx, (void *)data, reapsz);
10138 }
10139 }
10140 return REAL(procctl)(idtype, id, cmd, data);
10141 }
10142 #define INIT_PROCCTL COMMON_INTERCEPT_FUNCTION(procctl)
10143 #else
10144 #define INIT_PROCCTL
10145 #endif
10147 #if SANITIZER_INTERCEPT_UNAME
10148 INTERCEPTOR(int, uname, struct utsname *utsname) {
10149 #if SANITIZER_LINUX
10150 if (COMMON_INTERCEPTOR_NOTHING_IS_INITIALIZED)
10151 return internal_uname(utsname);
10152 #endif
10153 void *ctx;
10154 COMMON_INTERCEPTOR_ENTER(ctx, uname, utsname);
10155 int res = REAL(uname)(utsname);
10156 if (!res)
10157 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, utsname,
10158 __sanitizer::struct_utsname_sz);
10159 return res;
10160 }
10161 #define INIT_UNAME COMMON_INTERCEPT_FUNCTION(uname)
10162 #else
10163 #define INIT_UNAME
10164 #endif
10166 #if SANITIZER_INTERCEPT___XUNAME
10167 // FreeBSD's <sys/utsname.h> define uname() as
10168 // static __inline int uname(struct utsname *name) {
10169 // return __xuname(SYS_NMLN, (void*)name);
10170 // }
10171 INTERCEPTOR(int, __xuname, int size, void *utsname) {
10172 void *ctx;
10173 COMMON_INTERCEPTOR_ENTER(ctx, __xuname, size, utsname);
10174 int res = REAL(__xuname)(size, utsname);
10175 if (!res)
10176 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, utsname,
10177 __sanitizer::struct_utsname_sz);
10178 return res;
10179 }
10180 #define INIT___XUNAME COMMON_INTERCEPT_FUNCTION(__xuname)
10181 #else
10182 #define INIT___XUNAME
10183 #endif
10185 #if SANITIZER_INTERCEPT_ARGP_PARSE
10186 INTERCEPTOR(int, argp_parse, const struct argp *argp, int argc, char **argv,
10187 unsigned flags, int *arg_index, void *input) {
10188 void *ctx;
10189 COMMON_INTERCEPTOR_ENTER(ctx, argp_parse, argp, argc, argv, flags, arg_index,
10190 input);
10191 for (int i = 0; i < argc; i++)
10192 COMMON_INTERCEPTOR_READ_RANGE(ctx, argv[i], internal_strlen(argv[i]) + 1);
10193 int res = REAL(argp_parse)(argp, argc, argv, flags, arg_index, input);
10194 if (!res && arg_index)
10195 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, arg_index, sizeof(int));
10196 return res;
10197 }
10199 #define INIT_ARGP_PARSE COMMON_INTERCEPT_FUNCTION(argp_parse);
10200 #else
10201 #define INIT_ARGP_PARSE
10202 #endif
10204 #if SANITIZER_INTERCEPT_CPUSET_GETAFFINITY
10205 INTERCEPTOR(int, cpuset_getaffinity, int level, int which, __int64_t id, SIZE_T cpusetsize, __sanitizer_cpuset_t *mask) {
10206 void *ctx;
10207 COMMON_INTERCEPTOR_ENTER(ctx, cpuset_getaffinity, level, which, id, cpusetsize, mask);
10208 int res = REAL(cpuset_getaffinity)(level, which, id, cpusetsize, mask);
10209 if (mask && !res) COMMON_INTERCEPTOR_WRITE_RANGE(ctx, mask, cpusetsize);
10210 return res;
10211 }
10212 #define INIT_CPUSET_GETAFFINITY COMMON_INTERCEPT_FUNCTION(cpuset_getaffinity);
10213 #else
10214 #define INIT_CPUSET_GETAFFINITY
10215 #endif
10217 #if SANITIZER_INTERCEPT_PREADV2
10218 INTERCEPTOR(SSIZE_T, preadv2, int fd, __sanitizer_iovec *iov, int iovcnt,
10219 OFF_T offset, int flags) {
10220 void *ctx;
10221 COMMON_INTERCEPTOR_ENTER(ctx, preadv2, fd, iov, iovcnt, offset, flags);
10222 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
10223 SSIZE_T res = REAL(preadv2)(fd, iov, iovcnt, offset, flags);
10224 if (res > 0) write_iovec(ctx, iov, iovcnt, res);
10225 if (res >= 0 && fd >= 0) COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
10226 return res;
10227 }
10228 #define INIT_PREADV2 COMMON_INTERCEPT_FUNCTION(preadv2)
10229 #else
10230 #define INIT_PREADV2
10231 #endif
10233 #if SANITIZER_INTERCEPT_PWRITEV2
10234 INTERCEPTOR(SSIZE_T, pwritev2, int fd, __sanitizer_iovec *iov, int iovcnt,
10235 OFF_T offset, int flags) {
10236 void *ctx;
10237 COMMON_INTERCEPTOR_ENTER(ctx, pwritev2, fd, iov, iovcnt, offset, flags);
10238 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
10239 if (fd >= 0) COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd);
10240 SSIZE_T res = REAL(pwritev2)(fd, iov, iovcnt, offset, flags);
10241 if (res > 0) read_iovec(ctx, iov, iovcnt, res);
10242 return res;
10243 }
10244 #define INIT_PWRITEV2 COMMON_INTERCEPT_FUNCTION(pwritev2)
10245 #else
10246 #define INIT_PWRITEV2
10247 #endif
10249 #if SANITIZER_INTERCEPT_FREADLINK
10250 INTERCEPTOR(SSIZE_T, freadlink, int fd, char *buf, SIZE_T bufsiz) {
10251 void *ctx;
10252 COMMON_INTERCEPTOR_ENTER(ctx, freadlink, fd, buf, bufsiz);
10253 COMMON_INTERCEPTOR_FD_ACCESS(ctx, fd);
10254 SSIZE_T res = REAL(freadlink)(fd, buf, bufsiz);
10255 if (res > 0)
10256 COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, res);
10257 if (res >= 0 && fd > 0)
10258 COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd);
10259 return res;
10260 }
10261 # define INIT_FREADLINK COMMON_INTERCEPT_FUNCTION(freadlink)
10262 #else
10263 # define INIT_FREADLINK
10264 #endif
10266 #include "sanitizer_common_interceptors_netbsd_compat.inc"
10268 namespace __sanitizer {
10269 void InitializeMemintrinsicInterceptors();
10270 } // namespace __sanitizer
10272 static void InitializeCommonInterceptors() {
10273 #if SI_POSIX
10274 static u64 metadata_mem[sizeof(MetadataHashMap) / sizeof(u64) + 1];
10275 interceptor_metadata_map = new ((void *)&metadata_mem) MetadataHashMap();
10276 #endif
10278 __sanitizer::InitializeMemintrinsicInterceptors();
10280 INIT_MMAP;
10281 INIT_MMAP64;
10282 INIT_TEXTDOMAIN;
10283 INIT_STRLEN;
10284 INIT_STRNLEN;
10285 INIT_STRNDUP;
10286 INIT___STRNDUP;
10287 INIT_STRCMP;
10288 INIT_STRNCMP;
10289 INIT_STRCASECMP;
10290 INIT_STRNCASECMP;
10291 INIT_STRSTR;
10292 INIT_STRCASESTR;
10293 INIT_STRCHR;
10294 INIT_STRCHRNUL;
10295 INIT_STRRCHR;
10296 INIT_STRSPN;
10297 INIT_STRTOK;
10298 INIT_STRPBRK;
10299 INIT_STRXFRM;
10300 INIT___STRXFRM_L;
10301 INIT_MEMCHR;
10302 INIT_MEMCMP;
10303 INIT_BCMP;
10304 INIT_MEMRCHR;
10305 INIT_MEMMEM;
10306 INIT_READ;
10307 INIT_FREAD;
10308 INIT_PREAD;
10309 INIT_PREAD64;
10310 INIT_READV;
10311 INIT_PREADV;
10312 INIT_PREADV64;
10313 INIT_WRITE;
10314 INIT_FWRITE;
10315 INIT_PWRITE;
10316 INIT_PWRITE64;
10317 INIT_WRITEV;
10318 INIT_PWRITEV;
10319 INIT_PWRITEV64;
10320 INIT_FGETS;
10321 INIT_FPUTS;
10322 INIT_PUTS;
10323 INIT_PRCTL;
10324 INIT_LOCALTIME_AND_FRIENDS;
10325 INIT_STRPTIME;
10326 INIT_SCANF;
10327 INIT_ISOC99_SCANF;
10328 INIT_PRINTF;
10329 INIT_PRINTF_L;
10330 INIT_ISOC99_PRINTF;
10331 INIT_FREXP;
10332 INIT_FREXPF_FREXPL;
10333 INIT_GETPWNAM_AND_FRIENDS;
10334 INIT_GETPWNAM_R_AND_FRIENDS;
10335 INIT_GETPWENT;
10336 INIT_FGETPWENT;
10337 INIT_GETPWENT_R;
10338 INIT_FGETPWENT_R;
10339 INIT_FGETGRENT_R;
10340 INIT_SETPWENT;
10341 INIT_CLOCK_GETTIME;
10342 INIT_CLOCK_GETCPUCLOCKID;
10343 INIT_TIMER_CREATE;
10344 INIT_GETITIMER;
10345 INIT_TIME;
10346 INIT_TIMESPEC_GET;
10347 INIT_GLOB;
10348 INIT_GLOB64;
10349 INIT___B64_TO;
10350 INIT_DN_COMP_EXPAND;
10351 INIT_POSIX_SPAWN;
10352 INIT_WAIT;
10353 INIT_WAIT4;
10354 INIT_INET;
10355 INIT_PTHREAD_GETSCHEDPARAM;
10356 INIT_GETADDRINFO;
10357 INIT_GETNAMEINFO;
10358 INIT_GETSOCKNAME;
10359 INIT_GETHOSTBYNAME;
10360 INIT_GETHOSTBYNAME2;
10361 INIT_GETHOSTBYNAME_R;
10362 INIT_GETHOSTBYNAME2_R;
10363 INIT_GETHOSTBYADDR_R;
10364 INIT_GETHOSTENT_R;
10365 INIT_GETSOCKOPT;
10366 INIT_ACCEPT;
10367 INIT_ACCEPT4;
10368 INIT_PACCEPT;
10369 INIT_MODF;
10370 INIT_RECVMSG;
10371 INIT_SENDMSG;
10372 INIT_RECVMMSG;
10373 INIT_SENDMMSG;
10374 INIT_SYSMSG;
10375 INIT_GETPEERNAME;
10376 INIT_IOCTL;
10377 INIT_INET_ATON;
10378 INIT_SYSINFO;
10379 INIT_READDIR;
10380 INIT_READDIR64;
10381 INIT_PTRACE;
10382 INIT_SETLOCALE;
10383 INIT_GETCWD;
10384 INIT_GET_CURRENT_DIR_NAME;
10385 INIT_STRTOIMAX;
10386 INIT_STRTOIMAX_C23;
10387 INIT_MBSTOWCS;
10388 INIT_MBSNRTOWCS;
10389 INIT_WCSTOMBS;
10390 INIT_WCSNRTOMBS;
10391 INIT_WCRTOMB;
10392 INIT_WCTOMB;
10393 INIT_TCGETATTR;
10394 INIT_REALPATH;
10395 INIT_CANONICALIZE_FILE_NAME;
10396 INIT_CONFSTR;
10397 INIT_SCHED_GETAFFINITY;
10398 INIT_SCHED_GETPARAM;
10399 INIT_STRERROR;
10400 INIT_STRERROR_R;
10401 INIT_XPG_STRERROR_R;
10402 INIT_SCANDIR;
10403 INIT_SCANDIR64;
10404 INIT_GETGROUPS;
10405 INIT_POLL;
10406 INIT_PPOLL;
10407 INIT_WORDEXP;
10408 INIT_SIGWAIT;
10409 INIT_SIGWAITINFO;
10410 INIT_SIGTIMEDWAIT;
10411 INIT_SIGSETOPS;
10412 INIT_SIGSET_LOGICOPS;
10413 INIT_SIGPENDING;
10414 INIT_SIGPROCMASK;
10415 INIT_PTHREAD_SIGMASK;
10416 INIT_BACKTRACE;
10417 INIT__EXIT;
10418 INIT___LIBC_THR_SETCANCELSTATE;
10419 INIT_GETMNTENT;
10420 INIT_GETMNTENT_R;
10421 INIT_STATFS;
10422 INIT_STATFS64;
10423 INIT_STATVFS;
10424 INIT_STATVFS64;
10425 INIT_INITGROUPS;
10426 INIT_ETHER_NTOA_ATON;
10427 INIT_ETHER_HOST;
10428 INIT_ETHER_R;
10429 INIT_SHMCTL;
10430 INIT_RANDOM_R;
10431 INIT_PTHREAD_ATTR_GET;
10432 INIT_PTHREAD_ATTR_GET_SCHED;
10433 INIT_PTHREAD_ATTR_GETINHERITSCHED;
10434 INIT_PTHREAD_ATTR_GETAFFINITY_NP;
10435 INIT_PTHREAD_GETAFFINITY_NP;
10436 INIT_PTHREAD_MUTEXATTR_GETPSHARED;
10437 INIT_PTHREAD_MUTEXATTR_GETTYPE;
10438 INIT_PTHREAD_MUTEXATTR_GETPROTOCOL;
10439 INIT_PTHREAD_MUTEXATTR_GETPRIOCEILING;
10440 INIT_PTHREAD_MUTEXATTR_GETROBUST;
10441 INIT_PTHREAD_MUTEXATTR_GETROBUST_NP;
10442 INIT_PTHREAD_RWLOCKATTR_GETPSHARED;
10443 INIT_PTHREAD_RWLOCKATTR_GETKIND_NP;
10444 INIT_PTHREAD_CONDATTR_GETPSHARED;
10445 INIT_PTHREAD_CONDATTR_GETCLOCK;
10446 INIT_PTHREAD_BARRIERATTR_GETPSHARED;
10447 INIT_TMPNAM;
10448 INIT_TMPNAM_R;
10449 INIT_PTSNAME;
10450 INIT_PTSNAME_R;
10451 INIT_TTYNAME;
10452 INIT_TTYNAME_R;
10453 INIT_TEMPNAM;
10454 INIT_PTHREAD_SETNAME_NP;
10455 INIT_PTHREAD_GETNAME_NP;
10456 INIT_SINCOS;
10457 INIT_REMQUO;
10458 INIT_REMQUOL;
10459 INIT_LGAMMA;
10460 INIT_LGAMMAL;
10461 INIT_LGAMMA_R;
10462 INIT_LGAMMAL_R;
10463 INIT_DRAND48_R;
10464 INIT_RAND_R;
10465 INIT_GETLINE;
10466 INIT_ICONV;
10467 INIT_TIMES;
10468 INIT_TLS_GET_ADDR;
10469 INIT_LISTXATTR;
10470 INIT_GETXATTR;
10471 INIT_GETRESID;
10472 INIT_GETIFADDRS;
10473 INIT_IF_INDEXTONAME;
10474 INIT_CAPGET;
10475 INIT_FTIME;
10476 INIT_XDR;
10477 INIT_XDRREC_LINUX;
10478 INIT_TSEARCH;
10479 INIT_LIBIO_INTERNALS;
10480 INIT_FOPEN;
10481 INIT_FOPEN64;
10482 INIT_FLOPEN;
10483 INIT_OPEN_MEMSTREAM;
10484 INIT_OBSTACK;
10485 INIT_FFLUSH;
10486 INIT_FCLOSE;
10487 INIT_DLOPEN_DLCLOSE;
10488 INIT_GETPASS;
10489 INIT_TIMERFD;
10490 INIT_MLOCKX;
10491 INIT_FOPENCOOKIE;
10492 INIT_SEM;
10493 INIT_PTHREAD_SETCANCEL;
10494 INIT_MINCORE;
10495 INIT_PROCESS_VM_READV;
10496 INIT_CTERMID;
10497 INIT_CTERMID_R;
10498 INIT_RECV_RECVFROM;
10499 INIT_SEND_SENDTO;
10500 INIT_STAT;
10501 INIT_STAT64;
10502 INIT_EVENTFD_READ_WRITE;
10503 INIT_LSTAT;
10504 INIT_LSTAT64;
10505 INIT___XSTAT;
10506 INIT___XSTAT64;
10507 INIT___LXSTAT;
10508 INIT___LXSTAT64;
10509 // FIXME: add other *stat interceptors.
10510 INIT_UTMP;
10511 INIT_UTMPX;
10512 INIT_GETLOADAVG;
10513 INIT_WCSLEN;
10514 INIT_WCSCAT;
10515 INIT_WCSDUP;
10516 INIT_WCSXFRM;
10517 INIT___WCSXFRM_L;
10518 INIT_ACCT;
10519 INIT_USER_FROM_UID;
10520 INIT_UID_FROM_USER;
10521 INIT_GROUP_FROM_GID;
10522 INIT_GID_FROM_GROUP;
10523 INIT_ACCESS;
10524 INIT_FACCESSAT;
10525 INIT_GETGROUPLIST;
10526 INIT_GETGROUPMEMBERSHIP;
10527 INIT_READLINK;
10528 INIT_READLINKAT;
10529 INIT_NAME_TO_HANDLE_AT;
10530 INIT_OPEN_BY_HANDLE_AT;
10531 INIT_STRLCPY;
10532 INIT_DEVNAME;
10533 INIT_DEVNAME_R;
10534 INIT_FGETLN;
10535 INIT_STRMODE;
10536 INIT_TTYENT;
10537 INIT_PROTOENT;
10538 INIT_PROTOENT_R;
10539 INIT_NETENT;
10540 INIT_GETMNTINFO;
10541 INIT_MI_VECTOR_HASH;
10542 INIT_SETVBUF;
10543 INIT_GETVFSSTAT;
10544 INIT_REGEX;
10545 INIT_REGEXSUB;
10546 INIT_FTS;
10547 INIT_SYSCTL;
10548 INIT_ASYSCTL;
10549 INIT_SYSCTLGETMIBINFO;
10550 INIT_NL_LANGINFO;
10551 INIT_MODCTL;
10552 INIT_STRTONUM;
10553 INIT_FPARSELN;
10554 INIT_STATVFS1;
10555 INIT_STRTOI;
10556 INIT_CAPSICUM;
10557 INIT_SHA1;
10558 INIT_MD4;
10559 INIT_RMD160;
10560 INIT_FSEEK;
10561 INIT_MD2;
10562 INIT_VIS;
10563 INIT_CDB;
10564 INIT_GETFSENT;
10565 INIT_ARC4RANDOM;
10566 INIT_POPEN;
10567 INIT_POPENVE;
10568 INIT_PCLOSE;
10569 INIT_FUNOPEN;
10570 INIT_FUNOPEN2;
10571 INIT_FDEVNAME;
10572 INIT_GETUSERSHELL;
10573 INIT_SL_INIT;
10574 INIT_GETRANDOM;
10575 INIT_GETENTROPY;
10576 INIT_QSORT;
10577 INIT_QSORT_R;
10578 INIT_BSEARCH;
10579 INIT_SIGALTSTACK;
10580 INIT_PROCCTL
10581 INIT_UNAME;
10582 INIT___XUNAME;
10583 INIT_ARGP_PARSE;
10584 INIT_CPUSET_GETAFFINITY;
10585 INIT_PREADV2;
10586 INIT_PWRITEV2;
10587 INIT_FREADLINK;
10589 INIT___PRINTF_CHK;
10590 }