168d75effSDimitry Andric //===-- dfsan.cpp ---------------------------------------------------------===// 268d75effSDimitry Andric // 368d75effSDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 468d75effSDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 568d75effSDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 668d75effSDimitry Andric // 768d75effSDimitry Andric //===----------------------------------------------------------------------===// 868d75effSDimitry Andric // 968d75effSDimitry Andric // This file is a part of DataFlowSanitizer. 1068d75effSDimitry Andric // 1168d75effSDimitry Andric // This file defines the custom functions listed in done_abilist.txt. 1268d75effSDimitry Andric //===----------------------------------------------------------------------===// 1368d75effSDimitry Andric 1468d75effSDimitry Andric #include <arpa/inet.h> 1568d75effSDimitry Andric #include <assert.h> 1668d75effSDimitry Andric #include <ctype.h> 1768d75effSDimitry Andric #include <dlfcn.h> 1868d75effSDimitry Andric #include <link.h> 1968d75effSDimitry Andric #include <poll.h> 2068d75effSDimitry Andric #include <pthread.h> 2168d75effSDimitry Andric #include <pwd.h> 2268d75effSDimitry Andric #include <sched.h> 2368d75effSDimitry Andric #include <signal.h> 2468d75effSDimitry Andric #include <stdarg.h> 2568d75effSDimitry Andric #include <stdint.h> 2668d75effSDimitry Andric #include <stdio.h> 2768d75effSDimitry Andric #include <stdlib.h> 2868d75effSDimitry Andric #include <string.h> 29*e8d8bef9SDimitry Andric #include <sys/epoll.h> 3068d75effSDimitry Andric #include <sys/resource.h> 3168d75effSDimitry Andric #include <sys/select.h> 32*e8d8bef9SDimitry Andric #include <sys/socket.h> 3368d75effSDimitry Andric #include <sys/stat.h> 3468d75effSDimitry Andric #include <sys/time.h> 3568d75effSDimitry Andric #include <sys/types.h> 3668d75effSDimitry Andric #include <time.h> 3768d75effSDimitry Andric #include <unistd.h> 3868d75effSDimitry Andric 39*e8d8bef9SDimitry Andric #include "dfsan/dfsan.h" 40*e8d8bef9SDimitry Andric #include "sanitizer_common/sanitizer_common.h" 41*e8d8bef9SDimitry Andric #include "sanitizer_common/sanitizer_internal_defs.h" 42*e8d8bef9SDimitry Andric #include "sanitizer_common/sanitizer_linux.h" 43*e8d8bef9SDimitry Andric 4468d75effSDimitry Andric using namespace __dfsan; 4568d75effSDimitry Andric 4668d75effSDimitry Andric #define CALL_WEAK_INTERCEPTOR_HOOK(f, ...) \ 4768d75effSDimitry Andric do { \ 4868d75effSDimitry Andric if (f) \ 4968d75effSDimitry Andric f(__VA_ARGS__); \ 5068d75effSDimitry Andric } while (false) 5168d75effSDimitry Andric #define DECLARE_WEAK_INTERCEPTOR_HOOK(f, ...) \ 5268d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE SANITIZER_WEAK_ATTRIBUTE void f(__VA_ARGS__); 5368d75effSDimitry Andric 5468d75effSDimitry Andric extern "C" { 5568d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int 5668d75effSDimitry Andric __dfsw_stat(const char *path, struct stat *buf, dfsan_label path_label, 5768d75effSDimitry Andric dfsan_label buf_label, dfsan_label *ret_label) { 5868d75effSDimitry Andric int ret = stat(path, buf); 5968d75effSDimitry Andric if (ret == 0) 6068d75effSDimitry Andric dfsan_set_label(0, buf, sizeof(struct stat)); 6168d75effSDimitry Andric *ret_label = 0; 6268d75effSDimitry Andric return ret; 6368d75effSDimitry Andric } 6468d75effSDimitry Andric 6568d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_fstat(int fd, struct stat *buf, 6668d75effSDimitry Andric dfsan_label fd_label, 6768d75effSDimitry Andric dfsan_label buf_label, 6868d75effSDimitry Andric dfsan_label *ret_label) { 6968d75effSDimitry Andric int ret = fstat(fd, buf); 7068d75effSDimitry Andric if (ret == 0) 7168d75effSDimitry Andric dfsan_set_label(0, buf, sizeof(struct stat)); 7268d75effSDimitry Andric *ret_label = 0; 7368d75effSDimitry Andric return ret; 7468d75effSDimitry Andric } 7568d75effSDimitry Andric 7668d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE char *__dfsw_strchr(const char *s, int c, 7768d75effSDimitry Andric dfsan_label s_label, 7868d75effSDimitry Andric dfsan_label c_label, 7968d75effSDimitry Andric dfsan_label *ret_label) { 8068d75effSDimitry Andric for (size_t i = 0;; ++i) { 8168d75effSDimitry Andric if (s[i] == c || s[i] == 0) { 8268d75effSDimitry Andric if (flags().strict_data_dependencies) { 8368d75effSDimitry Andric *ret_label = s_label; 8468d75effSDimitry Andric } else { 8568d75effSDimitry Andric *ret_label = dfsan_union(dfsan_read_label(s, i + 1), 8668d75effSDimitry Andric dfsan_union(s_label, c_label)); 8768d75effSDimitry Andric } 885ffd83dbSDimitry Andric 895ffd83dbSDimitry Andric // If s[i] is the \0 at the end of the string, and \0 is not the 905ffd83dbSDimitry Andric // character we are searching for, then return null. 915ffd83dbSDimitry Andric if (s[i] == 0 && c != 0) { 925ffd83dbSDimitry Andric return nullptr; 935ffd83dbSDimitry Andric } 945ffd83dbSDimitry Andric return const_cast<char *>(s + i); 9568d75effSDimitry Andric } 9668d75effSDimitry Andric } 9768d75effSDimitry Andric } 9868d75effSDimitry Andric 99*e8d8bef9SDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE char *__dfsw_strpbrk(const char *s, 100*e8d8bef9SDimitry Andric const char *accept, 101*e8d8bef9SDimitry Andric dfsan_label s_label, 102*e8d8bef9SDimitry Andric dfsan_label accept_label, 10368d75effSDimitry Andric dfsan_label *ret_label) { 104*e8d8bef9SDimitry Andric const char *ret = strpbrk(s, accept); 105*e8d8bef9SDimitry Andric if (flags().strict_data_dependencies) { 106*e8d8bef9SDimitry Andric *ret_label = ret ? s_label : 0; 107*e8d8bef9SDimitry Andric } else { 108*e8d8bef9SDimitry Andric size_t s_bytes_read = (ret ? ret - s : strlen(s)) + 1; 109*e8d8bef9SDimitry Andric *ret_label = 110*e8d8bef9SDimitry Andric dfsan_union(dfsan_read_label(s, s_bytes_read), 111*e8d8bef9SDimitry Andric dfsan_union(dfsan_read_label(accept, strlen(accept) + 1), 112*e8d8bef9SDimitry Andric dfsan_union(s_label, accept_label))); 113*e8d8bef9SDimitry Andric } 114*e8d8bef9SDimitry Andric return const_cast<char *>(ret); 115*e8d8bef9SDimitry Andric } 116*e8d8bef9SDimitry Andric 117*e8d8bef9SDimitry Andric static int dfsan_memcmp_bcmp(const void *s1, const void *s2, size_t n, 118*e8d8bef9SDimitry Andric dfsan_label s1_label, dfsan_label s2_label, 119*e8d8bef9SDimitry Andric dfsan_label n_label, dfsan_label *ret_label) { 12068d75effSDimitry Andric const char *cs1 = (const char *) s1, *cs2 = (const char *) s2; 12168d75effSDimitry Andric for (size_t i = 0; i != n; ++i) { 12268d75effSDimitry Andric if (cs1[i] != cs2[i]) { 12368d75effSDimitry Andric if (flags().strict_data_dependencies) { 12468d75effSDimitry Andric *ret_label = 0; 12568d75effSDimitry Andric } else { 12668d75effSDimitry Andric *ret_label = dfsan_union(dfsan_read_label(cs1, i + 1), 12768d75effSDimitry Andric dfsan_read_label(cs2, i + 1)); 12868d75effSDimitry Andric } 12968d75effSDimitry Andric return cs1[i] - cs2[i]; 13068d75effSDimitry Andric } 13168d75effSDimitry Andric } 13268d75effSDimitry Andric 13368d75effSDimitry Andric if (flags().strict_data_dependencies) { 13468d75effSDimitry Andric *ret_label = 0; 13568d75effSDimitry Andric } else { 13668d75effSDimitry Andric *ret_label = dfsan_union(dfsan_read_label(cs1, n), 13768d75effSDimitry Andric dfsan_read_label(cs2, n)); 13868d75effSDimitry Andric } 13968d75effSDimitry Andric return 0; 14068d75effSDimitry Andric } 14168d75effSDimitry Andric 142*e8d8bef9SDimitry Andric DECLARE_WEAK_INTERCEPTOR_HOOK(dfsan_weak_hook_memcmp, uptr caller_pc, 143*e8d8bef9SDimitry Andric const void *s1, const void *s2, size_t n, 144*e8d8bef9SDimitry Andric dfsan_label s1_label, dfsan_label s2_label, 145*e8d8bef9SDimitry Andric dfsan_label n_label) 146*e8d8bef9SDimitry Andric 147*e8d8bef9SDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_memcmp(const void *s1, const void *s2, 148*e8d8bef9SDimitry Andric size_t n, dfsan_label s1_label, 149*e8d8bef9SDimitry Andric dfsan_label s2_label, 150*e8d8bef9SDimitry Andric dfsan_label n_label, 151*e8d8bef9SDimitry Andric dfsan_label *ret_label) { 152*e8d8bef9SDimitry Andric CALL_WEAK_INTERCEPTOR_HOOK(dfsan_weak_hook_memcmp, GET_CALLER_PC(), s1, s2, n, 153*e8d8bef9SDimitry Andric s1_label, s2_label, n_label); 154*e8d8bef9SDimitry Andric return dfsan_memcmp_bcmp(s1, s2, n, s1_label, s2_label, n_label, ret_label); 155*e8d8bef9SDimitry Andric } 156*e8d8bef9SDimitry Andric 157*e8d8bef9SDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_bcmp(const void *s1, const void *s2, 158*e8d8bef9SDimitry Andric size_t n, dfsan_label s1_label, 159*e8d8bef9SDimitry Andric dfsan_label s2_label, 160*e8d8bef9SDimitry Andric dfsan_label n_label, 161*e8d8bef9SDimitry Andric dfsan_label *ret_label) { 162*e8d8bef9SDimitry Andric return dfsan_memcmp_bcmp(s1, s2, n, s1_label, s2_label, n_label, ret_label); 163*e8d8bef9SDimitry Andric } 164*e8d8bef9SDimitry Andric 16568d75effSDimitry Andric DECLARE_WEAK_INTERCEPTOR_HOOK(dfsan_weak_hook_strcmp, uptr caller_pc, 16668d75effSDimitry Andric const char *s1, const char *s2, 16768d75effSDimitry Andric dfsan_label s1_label, dfsan_label s2_label) 16868d75effSDimitry Andric 16968d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_strcmp(const char *s1, const char *s2, 17068d75effSDimitry Andric dfsan_label s1_label, 17168d75effSDimitry Andric dfsan_label s2_label, 17268d75effSDimitry Andric dfsan_label *ret_label) { 17368d75effSDimitry Andric CALL_WEAK_INTERCEPTOR_HOOK(dfsan_weak_hook_strcmp, GET_CALLER_PC(), s1, s2, 17468d75effSDimitry Andric s1_label, s2_label); 17568d75effSDimitry Andric for (size_t i = 0;; ++i) { 17668d75effSDimitry Andric if (s1[i] != s2[i] || s1[i] == 0 || s2[i] == 0) { 17768d75effSDimitry Andric if (flags().strict_data_dependencies) { 17868d75effSDimitry Andric *ret_label = 0; 17968d75effSDimitry Andric } else { 18068d75effSDimitry Andric *ret_label = dfsan_union(dfsan_read_label(s1, i + 1), 18168d75effSDimitry Andric dfsan_read_label(s2, i + 1)); 18268d75effSDimitry Andric } 18368d75effSDimitry Andric return s1[i] - s2[i]; 18468d75effSDimitry Andric } 18568d75effSDimitry Andric } 18668d75effSDimitry Andric return 0; 18768d75effSDimitry Andric } 18868d75effSDimitry Andric 18968d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int 19068d75effSDimitry Andric __dfsw_strcasecmp(const char *s1, const char *s2, dfsan_label s1_label, 19168d75effSDimitry Andric dfsan_label s2_label, dfsan_label *ret_label) { 19268d75effSDimitry Andric for (size_t i = 0;; ++i) { 1935ffd83dbSDimitry Andric char s1_lower = tolower(s1[i]); 1945ffd83dbSDimitry Andric char s2_lower = tolower(s2[i]); 1955ffd83dbSDimitry Andric 1965ffd83dbSDimitry Andric if (s1_lower != s2_lower || s1[i] == 0 || s2[i] == 0) { 19768d75effSDimitry Andric if (flags().strict_data_dependencies) { 19868d75effSDimitry Andric *ret_label = 0; 19968d75effSDimitry Andric } else { 20068d75effSDimitry Andric *ret_label = dfsan_union(dfsan_read_label(s1, i + 1), 20168d75effSDimitry Andric dfsan_read_label(s2, i + 1)); 20268d75effSDimitry Andric } 2035ffd83dbSDimitry Andric return s1_lower - s2_lower; 20468d75effSDimitry Andric } 20568d75effSDimitry Andric } 20668d75effSDimitry Andric return 0; 20768d75effSDimitry Andric } 20868d75effSDimitry Andric 20968d75effSDimitry Andric DECLARE_WEAK_INTERCEPTOR_HOOK(dfsan_weak_hook_strncmp, uptr caller_pc, 21068d75effSDimitry Andric const char *s1, const char *s2, size_t n, 21168d75effSDimitry Andric dfsan_label s1_label, dfsan_label s2_label, 21268d75effSDimitry Andric dfsan_label n_label) 21368d75effSDimitry Andric 21468d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_strncmp(const char *s1, const char *s2, 21568d75effSDimitry Andric size_t n, dfsan_label s1_label, 21668d75effSDimitry Andric dfsan_label s2_label, 21768d75effSDimitry Andric dfsan_label n_label, 21868d75effSDimitry Andric dfsan_label *ret_label) { 21968d75effSDimitry Andric if (n == 0) { 22068d75effSDimitry Andric *ret_label = 0; 22168d75effSDimitry Andric return 0; 22268d75effSDimitry Andric } 22368d75effSDimitry Andric 22468d75effSDimitry Andric CALL_WEAK_INTERCEPTOR_HOOK(dfsan_weak_hook_strncmp, GET_CALLER_PC(), s1, s2, 22568d75effSDimitry Andric n, s1_label, s2_label, n_label); 22668d75effSDimitry Andric 22768d75effSDimitry Andric for (size_t i = 0;; ++i) { 22868d75effSDimitry Andric if (s1[i] != s2[i] || s1[i] == 0 || s2[i] == 0 || i == n - 1) { 22968d75effSDimitry Andric if (flags().strict_data_dependencies) { 23068d75effSDimitry Andric *ret_label = 0; 23168d75effSDimitry Andric } else { 23268d75effSDimitry Andric *ret_label = dfsan_union(dfsan_read_label(s1, i + 1), 23368d75effSDimitry Andric dfsan_read_label(s2, i + 1)); 23468d75effSDimitry Andric } 23568d75effSDimitry Andric return s1[i] - s2[i]; 23668d75effSDimitry Andric } 23768d75effSDimitry Andric } 23868d75effSDimitry Andric return 0; 23968d75effSDimitry Andric } 24068d75effSDimitry Andric 24168d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int 24268d75effSDimitry Andric __dfsw_strncasecmp(const char *s1, const char *s2, size_t n, 24368d75effSDimitry Andric dfsan_label s1_label, dfsan_label s2_label, 24468d75effSDimitry Andric dfsan_label n_label, dfsan_label *ret_label) { 24568d75effSDimitry Andric if (n == 0) { 24668d75effSDimitry Andric *ret_label = 0; 24768d75effSDimitry Andric return 0; 24868d75effSDimitry Andric } 24968d75effSDimitry Andric 25068d75effSDimitry Andric for (size_t i = 0;; ++i) { 2515ffd83dbSDimitry Andric char s1_lower = tolower(s1[i]); 2525ffd83dbSDimitry Andric char s2_lower = tolower(s2[i]); 2535ffd83dbSDimitry Andric 2545ffd83dbSDimitry Andric if (s1_lower != s2_lower || s1[i] == 0 || s2[i] == 0 || i == n - 1) { 25568d75effSDimitry Andric if (flags().strict_data_dependencies) { 25668d75effSDimitry Andric *ret_label = 0; 25768d75effSDimitry Andric } else { 25868d75effSDimitry Andric *ret_label = dfsan_union(dfsan_read_label(s1, i + 1), 25968d75effSDimitry Andric dfsan_read_label(s2, i + 1)); 26068d75effSDimitry Andric } 2615ffd83dbSDimitry Andric return s1_lower - s2_lower; 26268d75effSDimitry Andric } 26368d75effSDimitry Andric } 26468d75effSDimitry Andric return 0; 26568d75effSDimitry Andric } 26668d75effSDimitry Andric 26768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE void *__dfsw_calloc(size_t nmemb, size_t size, 26868d75effSDimitry Andric dfsan_label nmemb_label, 26968d75effSDimitry Andric dfsan_label size_label, 27068d75effSDimitry Andric dfsan_label *ret_label) { 27168d75effSDimitry Andric void *p = calloc(nmemb, size); 27268d75effSDimitry Andric dfsan_set_label(0, p, nmemb * size); 27368d75effSDimitry Andric *ret_label = 0; 27468d75effSDimitry Andric return p; 27568d75effSDimitry Andric } 27668d75effSDimitry Andric 27768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE size_t 27868d75effSDimitry Andric __dfsw_strlen(const char *s, dfsan_label s_label, dfsan_label *ret_label) { 27968d75effSDimitry Andric size_t ret = strlen(s); 28068d75effSDimitry Andric if (flags().strict_data_dependencies) { 28168d75effSDimitry Andric *ret_label = 0; 28268d75effSDimitry Andric } else { 28368d75effSDimitry Andric *ret_label = dfsan_read_label(s, ret + 1); 28468d75effSDimitry Andric } 28568d75effSDimitry Andric return ret; 28668d75effSDimitry Andric } 28768d75effSDimitry Andric 28868d75effSDimitry Andric 28968d75effSDimitry Andric static void *dfsan_memcpy(void *dest, const void *src, size_t n) { 29068d75effSDimitry Andric dfsan_label *sdest = shadow_for(dest); 29168d75effSDimitry Andric const dfsan_label *ssrc = shadow_for(src); 29268d75effSDimitry Andric internal_memcpy((void *)sdest, (const void *)ssrc, n * sizeof(dfsan_label)); 29368d75effSDimitry Andric return internal_memcpy(dest, src, n); 29468d75effSDimitry Andric } 29568d75effSDimitry Andric 29668d75effSDimitry Andric static void dfsan_memset(void *s, int c, dfsan_label c_label, size_t n) { 29768d75effSDimitry Andric internal_memset(s, c, n); 29868d75effSDimitry Andric dfsan_set_label(c_label, s, n); 29968d75effSDimitry Andric } 30068d75effSDimitry Andric 30168d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 30268d75effSDimitry Andric void *__dfsw_memcpy(void *dest, const void *src, size_t n, 30368d75effSDimitry Andric dfsan_label dest_label, dfsan_label src_label, 30468d75effSDimitry Andric dfsan_label n_label, dfsan_label *ret_label) { 30568d75effSDimitry Andric *ret_label = dest_label; 30668d75effSDimitry Andric return dfsan_memcpy(dest, src, n); 30768d75effSDimitry Andric } 30868d75effSDimitry Andric 30968d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 31068d75effSDimitry Andric void *__dfsw_memset(void *s, int c, size_t n, 31168d75effSDimitry Andric dfsan_label s_label, dfsan_label c_label, 31268d75effSDimitry Andric dfsan_label n_label, dfsan_label *ret_label) { 31368d75effSDimitry Andric dfsan_memset(s, c, c_label, n); 31468d75effSDimitry Andric *ret_label = s_label; 31568d75effSDimitry Andric return s; 31668d75effSDimitry Andric } 31768d75effSDimitry Andric 31868d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE char * 31968d75effSDimitry Andric __dfsw_strdup(const char *s, dfsan_label s_label, dfsan_label *ret_label) { 32068d75effSDimitry Andric size_t len = strlen(s); 32168d75effSDimitry Andric void *p = malloc(len+1); 32268d75effSDimitry Andric dfsan_memcpy(p, s, len+1); 32368d75effSDimitry Andric *ret_label = 0; 32468d75effSDimitry Andric return static_cast<char *>(p); 32568d75effSDimitry Andric } 32668d75effSDimitry Andric 32768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE char * 32868d75effSDimitry Andric __dfsw_strncpy(char *s1, const char *s2, size_t n, dfsan_label s1_label, 32968d75effSDimitry Andric dfsan_label s2_label, dfsan_label n_label, 33068d75effSDimitry Andric dfsan_label *ret_label) { 33168d75effSDimitry Andric size_t len = strlen(s2); 33268d75effSDimitry Andric if (len < n) { 33368d75effSDimitry Andric dfsan_memcpy(s1, s2, len+1); 33468d75effSDimitry Andric dfsan_memset(s1+len+1, 0, 0, n-len-1); 33568d75effSDimitry Andric } else { 33668d75effSDimitry Andric dfsan_memcpy(s1, s2, n); 33768d75effSDimitry Andric } 33868d75effSDimitry Andric 33968d75effSDimitry Andric *ret_label = s1_label; 34068d75effSDimitry Andric return s1; 34168d75effSDimitry Andric } 34268d75effSDimitry Andric 34368d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE ssize_t 34468d75effSDimitry Andric __dfsw_pread(int fd, void *buf, size_t count, off_t offset, 34568d75effSDimitry Andric dfsan_label fd_label, dfsan_label buf_label, 34668d75effSDimitry Andric dfsan_label count_label, dfsan_label offset_label, 34768d75effSDimitry Andric dfsan_label *ret_label) { 34868d75effSDimitry Andric ssize_t ret = pread(fd, buf, count, offset); 34968d75effSDimitry Andric if (ret > 0) 35068d75effSDimitry Andric dfsan_set_label(0, buf, ret); 35168d75effSDimitry Andric *ret_label = 0; 35268d75effSDimitry Andric return ret; 35368d75effSDimitry Andric } 35468d75effSDimitry Andric 35568d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE ssize_t 35668d75effSDimitry Andric __dfsw_read(int fd, void *buf, size_t count, 35768d75effSDimitry Andric dfsan_label fd_label, dfsan_label buf_label, 35868d75effSDimitry Andric dfsan_label count_label, 35968d75effSDimitry Andric dfsan_label *ret_label) { 36068d75effSDimitry Andric ssize_t ret = read(fd, buf, count); 36168d75effSDimitry Andric if (ret > 0) 36268d75effSDimitry Andric dfsan_set_label(0, buf, ret); 36368d75effSDimitry Andric *ret_label = 0; 36468d75effSDimitry Andric return ret; 36568d75effSDimitry Andric } 36668d75effSDimitry Andric 36768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_clock_gettime(clockid_t clk_id, 36868d75effSDimitry Andric struct timespec *tp, 36968d75effSDimitry Andric dfsan_label clk_id_label, 37068d75effSDimitry Andric dfsan_label tp_label, 37168d75effSDimitry Andric dfsan_label *ret_label) { 37268d75effSDimitry Andric int ret = clock_gettime(clk_id, tp); 37368d75effSDimitry Andric if (ret == 0) 37468d75effSDimitry Andric dfsan_set_label(0, tp, sizeof(struct timespec)); 37568d75effSDimitry Andric *ret_label = 0; 37668d75effSDimitry Andric return ret; 37768d75effSDimitry Andric } 37868d75effSDimitry Andric 37968d75effSDimitry Andric static void unpoison(const void *ptr, uptr size) { 38068d75effSDimitry Andric dfsan_set_label(0, const_cast<void *>(ptr), size); 38168d75effSDimitry Andric } 38268d75effSDimitry Andric 38368d75effSDimitry Andric // dlopen() ultimately calls mmap() down inside the loader, which generally 38468d75effSDimitry Andric // doesn't participate in dynamic symbol resolution. Therefore we won't 38568d75effSDimitry Andric // intercept its calls to mmap, and we have to hook it here. 38668d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE void * 38768d75effSDimitry Andric __dfsw_dlopen(const char *filename, int flag, dfsan_label filename_label, 38868d75effSDimitry Andric dfsan_label flag_label, dfsan_label *ret_label) { 38968d75effSDimitry Andric void *handle = dlopen(filename, flag); 39068d75effSDimitry Andric link_map *map = GET_LINK_MAP_BY_DLOPEN_HANDLE(handle); 39168d75effSDimitry Andric if (map) 39268d75effSDimitry Andric ForEachMappedRegion(map, unpoison); 39368d75effSDimitry Andric *ret_label = 0; 39468d75effSDimitry Andric return handle; 39568d75effSDimitry Andric } 39668d75effSDimitry Andric 39768d75effSDimitry Andric struct pthread_create_info { 39868d75effSDimitry Andric void *(*start_routine_trampoline)(void *, void *, dfsan_label, dfsan_label *); 39968d75effSDimitry Andric void *start_routine; 40068d75effSDimitry Andric void *arg; 40168d75effSDimitry Andric }; 40268d75effSDimitry Andric 40368d75effSDimitry Andric static void *pthread_create_cb(void *p) { 40468d75effSDimitry Andric pthread_create_info pci(*(pthread_create_info *)p); 40568d75effSDimitry Andric free(p); 40668d75effSDimitry Andric dfsan_label ret_label; 40768d75effSDimitry Andric return pci.start_routine_trampoline(pci.start_routine, pci.arg, 0, 40868d75effSDimitry Andric &ret_label); 40968d75effSDimitry Andric } 41068d75effSDimitry Andric 41168d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_pthread_create( 41268d75effSDimitry Andric pthread_t *thread, const pthread_attr_t *attr, 41368d75effSDimitry Andric void *(*start_routine_trampoline)(void *, void *, dfsan_label, 41468d75effSDimitry Andric dfsan_label *), 41568d75effSDimitry Andric void *start_routine, void *arg, dfsan_label thread_label, 41668d75effSDimitry Andric dfsan_label attr_label, dfsan_label start_routine_label, 41768d75effSDimitry Andric dfsan_label arg_label, dfsan_label *ret_label) { 41868d75effSDimitry Andric pthread_create_info *pci = 41968d75effSDimitry Andric (pthread_create_info *)malloc(sizeof(pthread_create_info)); 42068d75effSDimitry Andric pci->start_routine_trampoline = start_routine_trampoline; 42168d75effSDimitry Andric pci->start_routine = start_routine; 42268d75effSDimitry Andric pci->arg = arg; 42368d75effSDimitry Andric int rv = pthread_create(thread, attr, pthread_create_cb, (void *)pci); 42468d75effSDimitry Andric if (rv != 0) 42568d75effSDimitry Andric free(pci); 42668d75effSDimitry Andric *ret_label = 0; 42768d75effSDimitry Andric return rv; 42868d75effSDimitry Andric } 42968d75effSDimitry Andric 430*e8d8bef9SDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_pthread_join(pthread_t thread, 431*e8d8bef9SDimitry Andric void **retval, 432*e8d8bef9SDimitry Andric dfsan_label thread_label, 433*e8d8bef9SDimitry Andric dfsan_label retval_label, 434*e8d8bef9SDimitry Andric dfsan_label *ret_label) { 435*e8d8bef9SDimitry Andric int ret = pthread_join(thread, retval); 436*e8d8bef9SDimitry Andric if (ret == 0 && retval) 437*e8d8bef9SDimitry Andric dfsan_set_label(0, retval, sizeof(*retval)); 438*e8d8bef9SDimitry Andric *ret_label = 0; 439*e8d8bef9SDimitry Andric return ret; 440*e8d8bef9SDimitry Andric } 441*e8d8bef9SDimitry Andric 44268d75effSDimitry Andric struct dl_iterate_phdr_info { 44368d75effSDimitry Andric int (*callback_trampoline)(void *callback, struct dl_phdr_info *info, 44468d75effSDimitry Andric size_t size, void *data, dfsan_label info_label, 44568d75effSDimitry Andric dfsan_label size_label, dfsan_label data_label, 44668d75effSDimitry Andric dfsan_label *ret_label); 44768d75effSDimitry Andric void *callback; 44868d75effSDimitry Andric void *data; 44968d75effSDimitry Andric }; 45068d75effSDimitry Andric 45168d75effSDimitry Andric int dl_iterate_phdr_cb(struct dl_phdr_info *info, size_t size, void *data) { 45268d75effSDimitry Andric dl_iterate_phdr_info *dipi = (dl_iterate_phdr_info *)data; 45368d75effSDimitry Andric dfsan_set_label(0, *info); 45468d75effSDimitry Andric dfsan_set_label(0, const_cast<char *>(info->dlpi_name), 45568d75effSDimitry Andric strlen(info->dlpi_name) + 1); 45668d75effSDimitry Andric dfsan_set_label( 45768d75effSDimitry Andric 0, const_cast<char *>(reinterpret_cast<const char *>(info->dlpi_phdr)), 45868d75effSDimitry Andric sizeof(*info->dlpi_phdr) * info->dlpi_phnum); 45968d75effSDimitry Andric dfsan_label ret_label; 46068d75effSDimitry Andric return dipi->callback_trampoline(dipi->callback, info, size, dipi->data, 0, 0, 46168d75effSDimitry Andric 0, &ret_label); 46268d75effSDimitry Andric } 46368d75effSDimitry Andric 46468d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_dl_iterate_phdr( 46568d75effSDimitry Andric int (*callback_trampoline)(void *callback, struct dl_phdr_info *info, 46668d75effSDimitry Andric size_t size, void *data, dfsan_label info_label, 46768d75effSDimitry Andric dfsan_label size_label, dfsan_label data_label, 46868d75effSDimitry Andric dfsan_label *ret_label), 46968d75effSDimitry Andric void *callback, void *data, dfsan_label callback_label, 47068d75effSDimitry Andric dfsan_label data_label, dfsan_label *ret_label) { 47168d75effSDimitry Andric dl_iterate_phdr_info dipi = { callback_trampoline, callback, data }; 47268d75effSDimitry Andric *ret_label = 0; 47368d75effSDimitry Andric return dl_iterate_phdr(dl_iterate_phdr_cb, &dipi); 47468d75effSDimitry Andric } 47568d75effSDimitry Andric 476*e8d8bef9SDimitry Andric // This function is only available for glibc 2.27 or newer. Mark it weak so 477*e8d8bef9SDimitry Andric // linking succeeds with older glibcs. 478*e8d8bef9SDimitry Andric SANITIZER_WEAK_ATTRIBUTE void _dl_get_tls_static_info(size_t *sizep, 479*e8d8bef9SDimitry Andric size_t *alignp); 480*e8d8bef9SDimitry Andric 481*e8d8bef9SDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE void __dfsw__dl_get_tls_static_info( 482*e8d8bef9SDimitry Andric size_t *sizep, size_t *alignp, dfsan_label sizep_label, 483*e8d8bef9SDimitry Andric dfsan_label alignp_label) { 484*e8d8bef9SDimitry Andric assert(_dl_get_tls_static_info); 485*e8d8bef9SDimitry Andric _dl_get_tls_static_info(sizep, alignp); 486*e8d8bef9SDimitry Andric dfsan_set_label(0, sizep, sizeof(*sizep)); 487*e8d8bef9SDimitry Andric dfsan_set_label(0, alignp, sizeof(*alignp)); 488*e8d8bef9SDimitry Andric } 489*e8d8bef9SDimitry Andric 49068d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 49168d75effSDimitry Andric char *__dfsw_ctime_r(const time_t *timep, char *buf, dfsan_label timep_label, 49268d75effSDimitry Andric dfsan_label buf_label, dfsan_label *ret_label) { 49368d75effSDimitry Andric char *ret = ctime_r(timep, buf); 49468d75effSDimitry Andric if (ret) { 49568d75effSDimitry Andric dfsan_set_label(dfsan_read_label(timep, sizeof(time_t)), buf, 49668d75effSDimitry Andric strlen(buf) + 1); 49768d75effSDimitry Andric *ret_label = buf_label; 49868d75effSDimitry Andric } else { 49968d75effSDimitry Andric *ret_label = 0; 50068d75effSDimitry Andric } 50168d75effSDimitry Andric return ret; 50268d75effSDimitry Andric } 50368d75effSDimitry Andric 50468d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 50568d75effSDimitry Andric char *__dfsw_fgets(char *s, int size, FILE *stream, dfsan_label s_label, 50668d75effSDimitry Andric dfsan_label size_label, dfsan_label stream_label, 50768d75effSDimitry Andric dfsan_label *ret_label) { 50868d75effSDimitry Andric char *ret = fgets(s, size, stream); 50968d75effSDimitry Andric if (ret) { 51068d75effSDimitry Andric dfsan_set_label(0, ret, strlen(ret) + 1); 51168d75effSDimitry Andric *ret_label = s_label; 51268d75effSDimitry Andric } else { 51368d75effSDimitry Andric *ret_label = 0; 51468d75effSDimitry Andric } 51568d75effSDimitry Andric return ret; 51668d75effSDimitry Andric } 51768d75effSDimitry Andric 51868d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 51968d75effSDimitry Andric char *__dfsw_getcwd(char *buf, size_t size, dfsan_label buf_label, 52068d75effSDimitry Andric dfsan_label size_label, dfsan_label *ret_label) { 52168d75effSDimitry Andric char *ret = getcwd(buf, size); 52268d75effSDimitry Andric if (ret) { 52368d75effSDimitry Andric dfsan_set_label(0, ret, strlen(ret) + 1); 52468d75effSDimitry Andric *ret_label = buf_label; 52568d75effSDimitry Andric } else { 52668d75effSDimitry Andric *ret_label = 0; 52768d75effSDimitry Andric } 52868d75effSDimitry Andric return ret; 52968d75effSDimitry Andric } 53068d75effSDimitry Andric 53168d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 53268d75effSDimitry Andric char *__dfsw_get_current_dir_name(dfsan_label *ret_label) { 53368d75effSDimitry Andric char *ret = get_current_dir_name(); 53468d75effSDimitry Andric if (ret) { 53568d75effSDimitry Andric dfsan_set_label(0, ret, strlen(ret) + 1); 53668d75effSDimitry Andric } 53768d75effSDimitry Andric *ret_label = 0; 53868d75effSDimitry Andric return ret; 53968d75effSDimitry Andric } 54068d75effSDimitry Andric 54168d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 54268d75effSDimitry Andric int __dfsw_gethostname(char *name, size_t len, dfsan_label name_label, 54368d75effSDimitry Andric dfsan_label len_label, dfsan_label *ret_label) { 54468d75effSDimitry Andric int ret = gethostname(name, len); 54568d75effSDimitry Andric if (ret == 0) { 54668d75effSDimitry Andric dfsan_set_label(0, name, strlen(name) + 1); 54768d75effSDimitry Andric } 54868d75effSDimitry Andric *ret_label = 0; 54968d75effSDimitry Andric return ret; 55068d75effSDimitry Andric } 55168d75effSDimitry Andric 55268d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 55368d75effSDimitry Andric int __dfsw_getrlimit(int resource, struct rlimit *rlim, 55468d75effSDimitry Andric dfsan_label resource_label, dfsan_label rlim_label, 55568d75effSDimitry Andric dfsan_label *ret_label) { 55668d75effSDimitry Andric int ret = getrlimit(resource, rlim); 55768d75effSDimitry Andric if (ret == 0) { 55868d75effSDimitry Andric dfsan_set_label(0, rlim, sizeof(struct rlimit)); 55968d75effSDimitry Andric } 56068d75effSDimitry Andric *ret_label = 0; 56168d75effSDimitry Andric return ret; 56268d75effSDimitry Andric } 56368d75effSDimitry Andric 56468d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 56568d75effSDimitry Andric int __dfsw_getrusage(int who, struct rusage *usage, dfsan_label who_label, 56668d75effSDimitry Andric dfsan_label usage_label, dfsan_label *ret_label) { 56768d75effSDimitry Andric int ret = getrusage(who, usage); 56868d75effSDimitry Andric if (ret == 0) { 56968d75effSDimitry Andric dfsan_set_label(0, usage, sizeof(struct rusage)); 57068d75effSDimitry Andric } 57168d75effSDimitry Andric *ret_label = 0; 57268d75effSDimitry Andric return ret; 57368d75effSDimitry Andric } 57468d75effSDimitry Andric 57568d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 57668d75effSDimitry Andric char *__dfsw_strcpy(char *dest, const char *src, dfsan_label dst_label, 57768d75effSDimitry Andric dfsan_label src_label, dfsan_label *ret_label) { 57868d75effSDimitry Andric char *ret = strcpy(dest, src); // NOLINT 57968d75effSDimitry Andric if (ret) { 58068d75effSDimitry Andric internal_memcpy(shadow_for(dest), shadow_for(src), 58168d75effSDimitry Andric sizeof(dfsan_label) * (strlen(src) + 1)); 58268d75effSDimitry Andric } 58368d75effSDimitry Andric *ret_label = dst_label; 58468d75effSDimitry Andric return ret; 58568d75effSDimitry Andric } 58668d75effSDimitry Andric 58768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 58868d75effSDimitry Andric long int __dfsw_strtol(const char *nptr, char **endptr, int base, 58968d75effSDimitry Andric dfsan_label nptr_label, dfsan_label endptr_label, 59068d75effSDimitry Andric dfsan_label base_label, dfsan_label *ret_label) { 59168d75effSDimitry Andric char *tmp_endptr; 59268d75effSDimitry Andric long int ret = strtol(nptr, &tmp_endptr, base); 59368d75effSDimitry Andric if (endptr) { 59468d75effSDimitry Andric *endptr = tmp_endptr; 59568d75effSDimitry Andric } 59668d75effSDimitry Andric if (tmp_endptr > nptr) { 59768d75effSDimitry Andric // If *tmp_endptr is '\0' include its label as well. 59868d75effSDimitry Andric *ret_label = dfsan_union( 59968d75effSDimitry Andric base_label, 60068d75effSDimitry Andric dfsan_read_label(nptr, tmp_endptr - nptr + (*tmp_endptr ? 0 : 1))); 60168d75effSDimitry Andric } else { 60268d75effSDimitry Andric *ret_label = 0; 60368d75effSDimitry Andric } 60468d75effSDimitry Andric return ret; 60568d75effSDimitry Andric } 60668d75effSDimitry Andric 60768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 60868d75effSDimitry Andric double __dfsw_strtod(const char *nptr, char **endptr, 60968d75effSDimitry Andric dfsan_label nptr_label, dfsan_label endptr_label, 61068d75effSDimitry Andric dfsan_label *ret_label) { 61168d75effSDimitry Andric char *tmp_endptr; 61268d75effSDimitry Andric double ret = strtod(nptr, &tmp_endptr); 61368d75effSDimitry Andric if (endptr) { 61468d75effSDimitry Andric *endptr = tmp_endptr; 61568d75effSDimitry Andric } 61668d75effSDimitry Andric if (tmp_endptr > nptr) { 61768d75effSDimitry Andric // If *tmp_endptr is '\0' include its label as well. 61868d75effSDimitry Andric *ret_label = dfsan_read_label( 61968d75effSDimitry Andric nptr, 62068d75effSDimitry Andric tmp_endptr - nptr + (*tmp_endptr ? 0 : 1)); 62168d75effSDimitry Andric } else { 62268d75effSDimitry Andric *ret_label = 0; 62368d75effSDimitry Andric } 62468d75effSDimitry Andric return ret; 62568d75effSDimitry Andric } 62668d75effSDimitry Andric 62768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 62868d75effSDimitry Andric long long int __dfsw_strtoll(const char *nptr, char **endptr, int base, 62968d75effSDimitry Andric dfsan_label nptr_label, dfsan_label endptr_label, 63068d75effSDimitry Andric dfsan_label base_label, dfsan_label *ret_label) { 63168d75effSDimitry Andric char *tmp_endptr; 63268d75effSDimitry Andric long long int ret = strtoll(nptr, &tmp_endptr, base); 63368d75effSDimitry Andric if (endptr) { 63468d75effSDimitry Andric *endptr = tmp_endptr; 63568d75effSDimitry Andric } 63668d75effSDimitry Andric if (tmp_endptr > nptr) { 63768d75effSDimitry Andric // If *tmp_endptr is '\0' include its label as well. 63868d75effSDimitry Andric *ret_label = dfsan_union( 63968d75effSDimitry Andric base_label, 64068d75effSDimitry Andric dfsan_read_label(nptr, tmp_endptr - nptr + (*tmp_endptr ? 0 : 1))); 64168d75effSDimitry Andric } else { 64268d75effSDimitry Andric *ret_label = 0; 64368d75effSDimitry Andric } 64468d75effSDimitry Andric return ret; 64568d75effSDimitry Andric } 64668d75effSDimitry Andric 64768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 64868d75effSDimitry Andric unsigned long int __dfsw_strtoul(const char *nptr, char **endptr, int base, 64968d75effSDimitry Andric dfsan_label nptr_label, dfsan_label endptr_label, 65068d75effSDimitry Andric dfsan_label base_label, dfsan_label *ret_label) { 65168d75effSDimitry Andric char *tmp_endptr; 65268d75effSDimitry Andric unsigned long int ret = strtoul(nptr, &tmp_endptr, base); 65368d75effSDimitry Andric if (endptr) { 65468d75effSDimitry Andric *endptr = tmp_endptr; 65568d75effSDimitry Andric } 65668d75effSDimitry Andric if (tmp_endptr > nptr) { 65768d75effSDimitry Andric // If *tmp_endptr is '\0' include its label as well. 65868d75effSDimitry Andric *ret_label = dfsan_union( 65968d75effSDimitry Andric base_label, 66068d75effSDimitry Andric dfsan_read_label(nptr, tmp_endptr - nptr + (*tmp_endptr ? 0 : 1))); 66168d75effSDimitry Andric } else { 66268d75effSDimitry Andric *ret_label = 0; 66368d75effSDimitry Andric } 66468d75effSDimitry Andric return ret; 66568d75effSDimitry Andric } 66668d75effSDimitry Andric 66768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 66868d75effSDimitry Andric long long unsigned int __dfsw_strtoull(const char *nptr, char **endptr, 669*e8d8bef9SDimitry Andric int base, dfsan_label nptr_label, 670*e8d8bef9SDimitry Andric dfsan_label endptr_label, 67168d75effSDimitry Andric dfsan_label base_label, 67268d75effSDimitry Andric dfsan_label *ret_label) { 67368d75effSDimitry Andric char *tmp_endptr; 67468d75effSDimitry Andric long long unsigned int ret = strtoull(nptr, &tmp_endptr, base); 67568d75effSDimitry Andric if (endptr) { 67668d75effSDimitry Andric *endptr = tmp_endptr; 67768d75effSDimitry Andric } 67868d75effSDimitry Andric if (tmp_endptr > nptr) { 67968d75effSDimitry Andric // If *tmp_endptr is '\0' include its label as well. 68068d75effSDimitry Andric *ret_label = dfsan_union( 68168d75effSDimitry Andric base_label, 68268d75effSDimitry Andric dfsan_read_label(nptr, tmp_endptr - nptr + (*tmp_endptr ? 0 : 1))); 68368d75effSDimitry Andric } else { 68468d75effSDimitry Andric *ret_label = 0; 68568d75effSDimitry Andric } 68668d75effSDimitry Andric return ret; 68768d75effSDimitry Andric } 68868d75effSDimitry Andric 68968d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 69068d75effSDimitry Andric time_t __dfsw_time(time_t *t, dfsan_label t_label, dfsan_label *ret_label) { 69168d75effSDimitry Andric time_t ret = time(t); 69268d75effSDimitry Andric if (ret != (time_t) -1 && t) { 69368d75effSDimitry Andric dfsan_set_label(0, t, sizeof(time_t)); 69468d75effSDimitry Andric } 69568d75effSDimitry Andric *ret_label = 0; 69668d75effSDimitry Andric return ret; 69768d75effSDimitry Andric } 69868d75effSDimitry Andric 69968d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 70068d75effSDimitry Andric int __dfsw_inet_pton(int af, const char *src, void *dst, dfsan_label af_label, 70168d75effSDimitry Andric dfsan_label src_label, dfsan_label dst_label, 70268d75effSDimitry Andric dfsan_label *ret_label) { 70368d75effSDimitry Andric int ret = inet_pton(af, src, dst); 70468d75effSDimitry Andric if (ret == 1) { 70568d75effSDimitry Andric dfsan_set_label(dfsan_read_label(src, strlen(src) + 1), dst, 70668d75effSDimitry Andric af == AF_INET ? sizeof(struct in_addr) : sizeof(in6_addr)); 70768d75effSDimitry Andric } 70868d75effSDimitry Andric *ret_label = 0; 70968d75effSDimitry Andric return ret; 71068d75effSDimitry Andric } 71168d75effSDimitry Andric 71268d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 71368d75effSDimitry Andric struct tm *__dfsw_localtime_r(const time_t *timep, struct tm *result, 71468d75effSDimitry Andric dfsan_label timep_label, dfsan_label result_label, 71568d75effSDimitry Andric dfsan_label *ret_label) { 71668d75effSDimitry Andric struct tm *ret = localtime_r(timep, result); 71768d75effSDimitry Andric if (ret) { 71868d75effSDimitry Andric dfsan_set_label(dfsan_read_label(timep, sizeof(time_t)), result, 71968d75effSDimitry Andric sizeof(struct tm)); 72068d75effSDimitry Andric *ret_label = result_label; 72168d75effSDimitry Andric } else { 72268d75effSDimitry Andric *ret_label = 0; 72368d75effSDimitry Andric } 72468d75effSDimitry Andric return ret; 72568d75effSDimitry Andric } 72668d75effSDimitry Andric 72768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 72868d75effSDimitry Andric int __dfsw_getpwuid_r(id_t uid, struct passwd *pwd, 72968d75effSDimitry Andric char *buf, size_t buflen, struct passwd **result, 73068d75effSDimitry Andric dfsan_label uid_label, dfsan_label pwd_label, 73168d75effSDimitry Andric dfsan_label buf_label, dfsan_label buflen_label, 73268d75effSDimitry Andric dfsan_label result_label, dfsan_label *ret_label) { 73368d75effSDimitry Andric // Store the data in pwd, the strings referenced from pwd in buf, and the 73468d75effSDimitry Andric // address of pwd in *result. On failure, NULL is stored in *result. 73568d75effSDimitry Andric int ret = getpwuid_r(uid, pwd, buf, buflen, result); 73668d75effSDimitry Andric if (ret == 0) { 73768d75effSDimitry Andric dfsan_set_label(0, pwd, sizeof(struct passwd)); 73868d75effSDimitry Andric dfsan_set_label(0, buf, strlen(buf) + 1); 73968d75effSDimitry Andric } 74068d75effSDimitry Andric *ret_label = 0; 74168d75effSDimitry Andric dfsan_set_label(0, result, sizeof(struct passwd*)); 74268d75effSDimitry Andric return ret; 74368d75effSDimitry Andric } 74468d75effSDimitry Andric 74568d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 746*e8d8bef9SDimitry Andric int __dfsw_epoll_wait(int epfd, struct epoll_event *events, int maxevents, 747*e8d8bef9SDimitry Andric int timeout, dfsan_label epfd_label, 748*e8d8bef9SDimitry Andric dfsan_label events_label, dfsan_label maxevents_label, 749*e8d8bef9SDimitry Andric dfsan_label timeout_label, dfsan_label *ret_label) { 750*e8d8bef9SDimitry Andric int ret = epoll_wait(epfd, events, maxevents, timeout); 751*e8d8bef9SDimitry Andric if (ret > 0) 752*e8d8bef9SDimitry Andric dfsan_set_label(0, events, ret * sizeof(*events)); 753*e8d8bef9SDimitry Andric *ret_label = 0; 754*e8d8bef9SDimitry Andric return ret; 755*e8d8bef9SDimitry Andric } 756*e8d8bef9SDimitry Andric 757*e8d8bef9SDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 75868d75effSDimitry Andric int __dfsw_poll(struct pollfd *fds, nfds_t nfds, int timeout, 75968d75effSDimitry Andric dfsan_label dfs_label, dfsan_label nfds_label, 76068d75effSDimitry Andric dfsan_label timeout_label, dfsan_label *ret_label) { 76168d75effSDimitry Andric int ret = poll(fds, nfds, timeout); 76268d75effSDimitry Andric if (ret >= 0) { 76368d75effSDimitry Andric for (; nfds > 0; --nfds) { 76468d75effSDimitry Andric dfsan_set_label(0, &fds[nfds - 1].revents, sizeof(fds[nfds - 1].revents)); 76568d75effSDimitry Andric } 76668d75effSDimitry Andric } 76768d75effSDimitry Andric *ret_label = 0; 76868d75effSDimitry Andric return ret; 76968d75effSDimitry Andric } 77068d75effSDimitry Andric 77168d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 77268d75effSDimitry Andric int __dfsw_select(int nfds, fd_set *readfds, fd_set *writefds, 77368d75effSDimitry Andric fd_set *exceptfds, struct timeval *timeout, 77468d75effSDimitry Andric dfsan_label nfds_label, dfsan_label readfds_label, 77568d75effSDimitry Andric dfsan_label writefds_label, dfsan_label exceptfds_label, 77668d75effSDimitry Andric dfsan_label timeout_label, dfsan_label *ret_label) { 77768d75effSDimitry Andric int ret = select(nfds, readfds, writefds, exceptfds, timeout); 77868d75effSDimitry Andric // Clear everything (also on error) since their content is either set or 77968d75effSDimitry Andric // undefined. 78068d75effSDimitry Andric if (readfds) { 78168d75effSDimitry Andric dfsan_set_label(0, readfds, sizeof(fd_set)); 78268d75effSDimitry Andric } 78368d75effSDimitry Andric if (writefds) { 78468d75effSDimitry Andric dfsan_set_label(0, writefds, sizeof(fd_set)); 78568d75effSDimitry Andric } 78668d75effSDimitry Andric if (exceptfds) { 78768d75effSDimitry Andric dfsan_set_label(0, exceptfds, sizeof(fd_set)); 78868d75effSDimitry Andric } 78968d75effSDimitry Andric dfsan_set_label(0, timeout, sizeof(struct timeval)); 79068d75effSDimitry Andric *ret_label = 0; 79168d75effSDimitry Andric return ret; 79268d75effSDimitry Andric } 79368d75effSDimitry Andric 79468d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 79568d75effSDimitry Andric int __dfsw_sched_getaffinity(pid_t pid, size_t cpusetsize, cpu_set_t *mask, 79668d75effSDimitry Andric dfsan_label pid_label, 79768d75effSDimitry Andric dfsan_label cpusetsize_label, 79868d75effSDimitry Andric dfsan_label mask_label, dfsan_label *ret_label) { 79968d75effSDimitry Andric int ret = sched_getaffinity(pid, cpusetsize, mask); 80068d75effSDimitry Andric if (ret == 0) { 80168d75effSDimitry Andric dfsan_set_label(0, mask, cpusetsize); 80268d75effSDimitry Andric } 80368d75effSDimitry Andric *ret_label = 0; 80468d75effSDimitry Andric return ret; 80568d75effSDimitry Andric } 80668d75effSDimitry Andric 80768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 80868d75effSDimitry Andric int __dfsw_sigemptyset(sigset_t *set, dfsan_label set_label, 80968d75effSDimitry Andric dfsan_label *ret_label) { 81068d75effSDimitry Andric int ret = sigemptyset(set); 81168d75effSDimitry Andric dfsan_set_label(0, set, sizeof(sigset_t)); 81268d75effSDimitry Andric return ret; 81368d75effSDimitry Andric } 81468d75effSDimitry Andric 81568d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 81668d75effSDimitry Andric int __dfsw_sigaction(int signum, const struct sigaction *act, 81768d75effSDimitry Andric struct sigaction *oldact, dfsan_label signum_label, 81868d75effSDimitry Andric dfsan_label act_label, dfsan_label oldact_label, 81968d75effSDimitry Andric dfsan_label *ret_label) { 82068d75effSDimitry Andric int ret = sigaction(signum, act, oldact); 82168d75effSDimitry Andric if (oldact) { 82268d75effSDimitry Andric dfsan_set_label(0, oldact, sizeof(struct sigaction)); 82368d75effSDimitry Andric } 82468d75effSDimitry Andric *ret_label = 0; 82568d75effSDimitry Andric return ret; 82668d75effSDimitry Andric } 82768d75effSDimitry Andric 82868d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 829*e8d8bef9SDimitry Andric int __dfsw_sigaltstack(const stack_t *ss, stack_t *old_ss, dfsan_label ss_label, 830*e8d8bef9SDimitry Andric dfsan_label old_ss_label, dfsan_label *ret_label) { 831*e8d8bef9SDimitry Andric int ret = sigaltstack(ss, old_ss); 832*e8d8bef9SDimitry Andric if (ret != -1 && old_ss) 833*e8d8bef9SDimitry Andric dfsan_set_label(0, old_ss, sizeof(*old_ss)); 834*e8d8bef9SDimitry Andric *ret_label = 0; 835*e8d8bef9SDimitry Andric return ret; 836*e8d8bef9SDimitry Andric } 837*e8d8bef9SDimitry Andric 838*e8d8bef9SDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 83968d75effSDimitry Andric int __dfsw_gettimeofday(struct timeval *tv, struct timezone *tz, 84068d75effSDimitry Andric dfsan_label tv_label, dfsan_label tz_label, 84168d75effSDimitry Andric dfsan_label *ret_label) { 84268d75effSDimitry Andric int ret = gettimeofday(tv, tz); 84368d75effSDimitry Andric if (tv) { 84468d75effSDimitry Andric dfsan_set_label(0, tv, sizeof(struct timeval)); 84568d75effSDimitry Andric } 84668d75effSDimitry Andric if (tz) { 84768d75effSDimitry Andric dfsan_set_label(0, tz, sizeof(struct timezone)); 84868d75effSDimitry Andric } 84968d75effSDimitry Andric *ret_label = 0; 85068d75effSDimitry Andric return ret; 85168d75effSDimitry Andric } 85268d75effSDimitry Andric 85368d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE void *__dfsw_memchr(void *s, int c, size_t n, 85468d75effSDimitry Andric dfsan_label s_label, 85568d75effSDimitry Andric dfsan_label c_label, 85668d75effSDimitry Andric dfsan_label n_label, 85768d75effSDimitry Andric dfsan_label *ret_label) { 85868d75effSDimitry Andric void *ret = memchr(s, c, n); 85968d75effSDimitry Andric if (flags().strict_data_dependencies) { 86068d75effSDimitry Andric *ret_label = ret ? s_label : 0; 86168d75effSDimitry Andric } else { 86268d75effSDimitry Andric size_t len = 86368d75effSDimitry Andric ret ? reinterpret_cast<char *>(ret) - reinterpret_cast<char *>(s) + 1 86468d75effSDimitry Andric : n; 86568d75effSDimitry Andric *ret_label = 86668d75effSDimitry Andric dfsan_union(dfsan_read_label(s, len), dfsan_union(s_label, c_label)); 86768d75effSDimitry Andric } 86868d75effSDimitry Andric return ret; 86968d75effSDimitry Andric } 87068d75effSDimitry Andric 87168d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE char *__dfsw_strrchr(char *s, int c, 87268d75effSDimitry Andric dfsan_label s_label, 87368d75effSDimitry Andric dfsan_label c_label, 87468d75effSDimitry Andric dfsan_label *ret_label) { 87568d75effSDimitry Andric char *ret = strrchr(s, c); 87668d75effSDimitry Andric if (flags().strict_data_dependencies) { 87768d75effSDimitry Andric *ret_label = ret ? s_label : 0; 87868d75effSDimitry Andric } else { 87968d75effSDimitry Andric *ret_label = 88068d75effSDimitry Andric dfsan_union(dfsan_read_label(s, strlen(s) + 1), 88168d75effSDimitry Andric dfsan_union(s_label, c_label)); 88268d75effSDimitry Andric } 88368d75effSDimitry Andric 88468d75effSDimitry Andric return ret; 88568d75effSDimitry Andric } 88668d75effSDimitry Andric 88768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE char *__dfsw_strstr(char *haystack, char *needle, 88868d75effSDimitry Andric dfsan_label haystack_label, 88968d75effSDimitry Andric dfsan_label needle_label, 89068d75effSDimitry Andric dfsan_label *ret_label) { 89168d75effSDimitry Andric char *ret = strstr(haystack, needle); 89268d75effSDimitry Andric if (flags().strict_data_dependencies) { 89368d75effSDimitry Andric *ret_label = ret ? haystack_label : 0; 89468d75effSDimitry Andric } else { 89568d75effSDimitry Andric size_t len = ret ? ret + strlen(needle) - haystack : strlen(haystack) + 1; 89668d75effSDimitry Andric *ret_label = 89768d75effSDimitry Andric dfsan_union(dfsan_read_label(haystack, len), 89868d75effSDimitry Andric dfsan_union(dfsan_read_label(needle, strlen(needle) + 1), 89968d75effSDimitry Andric dfsan_union(haystack_label, needle_label))); 90068d75effSDimitry Andric } 90168d75effSDimitry Andric 90268d75effSDimitry Andric return ret; 90368d75effSDimitry Andric } 90468d75effSDimitry Andric 90568d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_nanosleep(const struct timespec *req, 90668d75effSDimitry Andric struct timespec *rem, 90768d75effSDimitry Andric dfsan_label req_label, 90868d75effSDimitry Andric dfsan_label rem_label, 90968d75effSDimitry Andric dfsan_label *ret_label) { 91068d75effSDimitry Andric int ret = nanosleep(req, rem); 91168d75effSDimitry Andric *ret_label = 0; 91268d75effSDimitry Andric if (ret == -1) { 91368d75effSDimitry Andric // Interrupted by a signal, rem is filled with the remaining time. 91468d75effSDimitry Andric dfsan_set_label(0, rem, sizeof(struct timespec)); 91568d75effSDimitry Andric } 91668d75effSDimitry Andric return ret; 91768d75effSDimitry Andric } 91868d75effSDimitry Andric 919*e8d8bef9SDimitry Andric static void clear_msghdr_labels(size_t bytes_written, struct msghdr *msg) { 920*e8d8bef9SDimitry Andric dfsan_set_label(0, msg, sizeof(*msg)); 921*e8d8bef9SDimitry Andric dfsan_set_label(0, msg->msg_name, msg->msg_namelen); 922*e8d8bef9SDimitry Andric dfsan_set_label(0, msg->msg_control, msg->msg_controllen); 923*e8d8bef9SDimitry Andric for (size_t i = 0; bytes_written > 0; ++i) { 924*e8d8bef9SDimitry Andric assert(i < msg->msg_iovlen); 925*e8d8bef9SDimitry Andric struct iovec *iov = &msg->msg_iov[i]; 926*e8d8bef9SDimitry Andric size_t iov_written = 927*e8d8bef9SDimitry Andric bytes_written < iov->iov_len ? bytes_written : iov->iov_len; 928*e8d8bef9SDimitry Andric dfsan_set_label(0, iov->iov_base, iov_written); 929*e8d8bef9SDimitry Andric bytes_written -= iov_written; 930*e8d8bef9SDimitry Andric } 931*e8d8bef9SDimitry Andric } 932*e8d8bef9SDimitry Andric 933*e8d8bef9SDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_recvmmsg( 934*e8d8bef9SDimitry Andric int sockfd, struct mmsghdr *msgvec, unsigned int vlen, int flags, 935*e8d8bef9SDimitry Andric struct timespec *timeout, dfsan_label sockfd_label, 936*e8d8bef9SDimitry Andric dfsan_label msgvec_label, dfsan_label vlen_label, dfsan_label flags_label, 937*e8d8bef9SDimitry Andric dfsan_label timeout_label, dfsan_label *ret_label) { 938*e8d8bef9SDimitry Andric int ret = recvmmsg(sockfd, msgvec, vlen, flags, timeout); 939*e8d8bef9SDimitry Andric for (int i = 0; i < ret; ++i) { 940*e8d8bef9SDimitry Andric dfsan_set_label(0, &msgvec[i].msg_len, sizeof(msgvec[i].msg_len)); 941*e8d8bef9SDimitry Andric clear_msghdr_labels(msgvec[i].msg_len, &msgvec[i].msg_hdr); 942*e8d8bef9SDimitry Andric } 943*e8d8bef9SDimitry Andric *ret_label = 0; 944*e8d8bef9SDimitry Andric return ret; 945*e8d8bef9SDimitry Andric } 946*e8d8bef9SDimitry Andric 947*e8d8bef9SDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE ssize_t __dfsw_recvmsg( 948*e8d8bef9SDimitry Andric int sockfd, struct msghdr *msg, int flags, dfsan_label sockfd_label, 949*e8d8bef9SDimitry Andric dfsan_label msg_label, dfsan_label flags_label, dfsan_label *ret_label) { 950*e8d8bef9SDimitry Andric ssize_t ret = recvmsg(sockfd, msg, flags); 951*e8d8bef9SDimitry Andric if (ret >= 0) 952*e8d8bef9SDimitry Andric clear_msghdr_labels(ret, msg); 953*e8d8bef9SDimitry Andric *ret_label = 0; 954*e8d8bef9SDimitry Andric return ret; 955*e8d8bef9SDimitry Andric } 956*e8d8bef9SDimitry Andric 95768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int 95868d75effSDimitry Andric __dfsw_socketpair(int domain, int type, int protocol, int sv[2], 95968d75effSDimitry Andric dfsan_label domain_label, dfsan_label type_label, 96068d75effSDimitry Andric dfsan_label protocol_label, dfsan_label sv_label, 96168d75effSDimitry Andric dfsan_label *ret_label) { 96268d75effSDimitry Andric int ret = socketpair(domain, type, protocol, sv); 96368d75effSDimitry Andric *ret_label = 0; 96468d75effSDimitry Andric if (ret == 0) { 96568d75effSDimitry Andric dfsan_set_label(0, sv, sizeof(*sv) * 2); 96668d75effSDimitry Andric } 96768d75effSDimitry Andric return ret; 96868d75effSDimitry Andric } 96968d75effSDimitry Andric 970*e8d8bef9SDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_getsockopt( 971*e8d8bef9SDimitry Andric int sockfd, int level, int optname, void *optval, socklen_t *optlen, 972*e8d8bef9SDimitry Andric dfsan_label sockfd_label, dfsan_label level_label, 973*e8d8bef9SDimitry Andric dfsan_label optname_label, dfsan_label optval_label, 974*e8d8bef9SDimitry Andric dfsan_label optlen_label, dfsan_label *ret_label) { 975*e8d8bef9SDimitry Andric int ret = getsockopt(sockfd, level, optname, optval, optlen); 976*e8d8bef9SDimitry Andric if (ret != -1 && optval && optlen) { 977*e8d8bef9SDimitry Andric dfsan_set_label(0, optlen, sizeof(*optlen)); 978*e8d8bef9SDimitry Andric dfsan_set_label(0, optval, *optlen); 979*e8d8bef9SDimitry Andric } 980*e8d8bef9SDimitry Andric *ret_label = 0; 981*e8d8bef9SDimitry Andric return ret; 982*e8d8bef9SDimitry Andric } 983*e8d8bef9SDimitry Andric 984*e8d8bef9SDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_getsockname( 985*e8d8bef9SDimitry Andric int sockfd, struct sockaddr *addr, socklen_t *addrlen, 986*e8d8bef9SDimitry Andric dfsan_label sockfd_label, dfsan_label addr_label, dfsan_label addrlen_label, 987*e8d8bef9SDimitry Andric dfsan_label *ret_label) { 988*e8d8bef9SDimitry Andric socklen_t origlen = addrlen ? *addrlen : 0; 989*e8d8bef9SDimitry Andric int ret = getsockname(sockfd, addr, addrlen); 990*e8d8bef9SDimitry Andric if (ret != -1 && addr && addrlen) { 991*e8d8bef9SDimitry Andric socklen_t written_bytes = origlen < *addrlen ? origlen : *addrlen; 992*e8d8bef9SDimitry Andric dfsan_set_label(0, addrlen, sizeof(*addrlen)); 993*e8d8bef9SDimitry Andric dfsan_set_label(0, addr, written_bytes); 994*e8d8bef9SDimitry Andric } 995*e8d8bef9SDimitry Andric *ret_label = 0; 996*e8d8bef9SDimitry Andric return ret; 997*e8d8bef9SDimitry Andric } 998*e8d8bef9SDimitry Andric 999*e8d8bef9SDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int __dfsw_getpeername( 1000*e8d8bef9SDimitry Andric int sockfd, struct sockaddr *addr, socklen_t *addrlen, 1001*e8d8bef9SDimitry Andric dfsan_label sockfd_label, dfsan_label addr_label, dfsan_label addrlen_label, 1002*e8d8bef9SDimitry Andric dfsan_label *ret_label) { 1003*e8d8bef9SDimitry Andric socklen_t origlen = addrlen ? *addrlen : 0; 1004*e8d8bef9SDimitry Andric int ret = getpeername(sockfd, addr, addrlen); 1005*e8d8bef9SDimitry Andric if (ret != -1 && addr && addrlen) { 1006*e8d8bef9SDimitry Andric socklen_t written_bytes = origlen < *addrlen ? origlen : *addrlen; 1007*e8d8bef9SDimitry Andric dfsan_set_label(0, addrlen, sizeof(*addrlen)); 1008*e8d8bef9SDimitry Andric dfsan_set_label(0, addr, written_bytes); 1009*e8d8bef9SDimitry Andric } 1010*e8d8bef9SDimitry Andric *ret_label = 0; 1011*e8d8bef9SDimitry Andric return ret; 1012*e8d8bef9SDimitry Andric } 1013*e8d8bef9SDimitry Andric 101468d75effSDimitry Andric // Type of the trampoline function passed to the custom version of 101568d75effSDimitry Andric // dfsan_set_write_callback. 101668d75effSDimitry Andric typedef void (*write_trampoline_t)( 101768d75effSDimitry Andric void *callback, 101868d75effSDimitry Andric int fd, const void *buf, ssize_t count, 101968d75effSDimitry Andric dfsan_label fd_label, dfsan_label buf_label, dfsan_label count_label); 102068d75effSDimitry Andric 102168d75effSDimitry Andric // Calls to dfsan_set_write_callback() set the values in this struct. 102268d75effSDimitry Andric // Calls to the custom version of write() read (and invoke) them. 102368d75effSDimitry Andric static struct { 102468d75effSDimitry Andric write_trampoline_t write_callback_trampoline = nullptr; 102568d75effSDimitry Andric void *write_callback = nullptr; 102668d75effSDimitry Andric } write_callback_info; 102768d75effSDimitry Andric 102868d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE void 102968d75effSDimitry Andric __dfsw_dfsan_set_write_callback( 103068d75effSDimitry Andric write_trampoline_t write_callback_trampoline, 103168d75effSDimitry Andric void *write_callback, 103268d75effSDimitry Andric dfsan_label write_callback_label, 103368d75effSDimitry Andric dfsan_label *ret_label) { 103468d75effSDimitry Andric write_callback_info.write_callback_trampoline = write_callback_trampoline; 103568d75effSDimitry Andric write_callback_info.write_callback = write_callback; 103668d75effSDimitry Andric } 103768d75effSDimitry Andric 103868d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE int 103968d75effSDimitry Andric __dfsw_write(int fd, const void *buf, size_t count, 104068d75effSDimitry Andric dfsan_label fd_label, dfsan_label buf_label, 104168d75effSDimitry Andric dfsan_label count_label, dfsan_label *ret_label) { 104268d75effSDimitry Andric if (write_callback_info.write_callback) { 104368d75effSDimitry Andric write_callback_info.write_callback_trampoline( 104468d75effSDimitry Andric write_callback_info.write_callback, 104568d75effSDimitry Andric fd, buf, count, 104668d75effSDimitry Andric fd_label, buf_label, count_label); 104768d75effSDimitry Andric } 104868d75effSDimitry Andric 104968d75effSDimitry Andric *ret_label = 0; 105068d75effSDimitry Andric return write(fd, buf, count); 105168d75effSDimitry Andric } 105268d75effSDimitry Andric } // namespace __dfsan 105368d75effSDimitry Andric 105468d75effSDimitry Andric // Type used to extract a dfsan_label with va_arg() 105568d75effSDimitry Andric typedef int dfsan_label_va; 105668d75effSDimitry Andric 105768d75effSDimitry Andric // Formats a chunk either a constant string or a single format directive (e.g., 105868d75effSDimitry Andric // '%.3f'). 105968d75effSDimitry Andric struct Formatter { 106068d75effSDimitry Andric Formatter(char *str_, const char *fmt_, size_t size_) 106168d75effSDimitry Andric : str(str_), str_off(0), size(size_), fmt_start(fmt_), fmt_cur(fmt_), 106268d75effSDimitry Andric width(-1) {} 106368d75effSDimitry Andric 106468d75effSDimitry Andric int format() { 106568d75effSDimitry Andric char *tmp_fmt = build_format_string(); 106668d75effSDimitry Andric int retval = 106768d75effSDimitry Andric snprintf(str + str_off, str_off < size ? size - str_off : 0, tmp_fmt, 106868d75effSDimitry Andric 0 /* used only to avoid warnings */); 106968d75effSDimitry Andric free(tmp_fmt); 107068d75effSDimitry Andric return retval; 107168d75effSDimitry Andric } 107268d75effSDimitry Andric 107368d75effSDimitry Andric template <typename T> int format(T arg) { 107468d75effSDimitry Andric char *tmp_fmt = build_format_string(); 107568d75effSDimitry Andric int retval; 107668d75effSDimitry Andric if (width >= 0) { 107768d75effSDimitry Andric retval = snprintf(str + str_off, str_off < size ? size - str_off : 0, 107868d75effSDimitry Andric tmp_fmt, width, arg); 107968d75effSDimitry Andric } else { 108068d75effSDimitry Andric retval = snprintf(str + str_off, str_off < size ? size - str_off : 0, 108168d75effSDimitry Andric tmp_fmt, arg); 108268d75effSDimitry Andric } 108368d75effSDimitry Andric free(tmp_fmt); 108468d75effSDimitry Andric return retval; 108568d75effSDimitry Andric } 108668d75effSDimitry Andric 108768d75effSDimitry Andric char *build_format_string() { 108868d75effSDimitry Andric size_t fmt_size = fmt_cur - fmt_start + 1; 108968d75effSDimitry Andric char *new_fmt = (char *)malloc(fmt_size + 1); 109068d75effSDimitry Andric assert(new_fmt); 109168d75effSDimitry Andric internal_memcpy(new_fmt, fmt_start, fmt_size); 109268d75effSDimitry Andric new_fmt[fmt_size] = '\0'; 109368d75effSDimitry Andric return new_fmt; 109468d75effSDimitry Andric } 109568d75effSDimitry Andric 109668d75effSDimitry Andric char *str_cur() { return str + str_off; } 109768d75effSDimitry Andric 109868d75effSDimitry Andric size_t num_written_bytes(int retval) { 109968d75effSDimitry Andric if (retval < 0) { 110068d75effSDimitry Andric return 0; 110168d75effSDimitry Andric } 110268d75effSDimitry Andric 110368d75effSDimitry Andric size_t num_avail = str_off < size ? size - str_off : 0; 110468d75effSDimitry Andric if (num_avail == 0) { 110568d75effSDimitry Andric return 0; 110668d75effSDimitry Andric } 110768d75effSDimitry Andric 110868d75effSDimitry Andric size_t num_written = retval; 110968d75effSDimitry Andric // A return value of {v,}snprintf of size or more means that the output was 111068d75effSDimitry Andric // truncated. 111168d75effSDimitry Andric if (num_written >= num_avail) { 111268d75effSDimitry Andric num_written -= num_avail; 111368d75effSDimitry Andric } 111468d75effSDimitry Andric 111568d75effSDimitry Andric return num_written; 111668d75effSDimitry Andric } 111768d75effSDimitry Andric 111868d75effSDimitry Andric char *str; 111968d75effSDimitry Andric size_t str_off; 112068d75effSDimitry Andric size_t size; 112168d75effSDimitry Andric const char *fmt_start; 112268d75effSDimitry Andric const char *fmt_cur; 112368d75effSDimitry Andric int width; 112468d75effSDimitry Andric }; 112568d75effSDimitry Andric 112668d75effSDimitry Andric // Formats the input and propagates the input labels to the output. The output 112768d75effSDimitry Andric // is stored in 'str'. 'size' bounds the number of output bytes. 'format' and 112868d75effSDimitry Andric // 'ap' are the format string and the list of arguments for formatting. Returns 112968d75effSDimitry Andric // the return value vsnprintf would return. 113068d75effSDimitry Andric // 113168d75effSDimitry Andric // The function tokenizes the format string in chunks representing either a 113268d75effSDimitry Andric // constant string or a single format directive (e.g., '%.3f') and formats each 113368d75effSDimitry Andric // chunk independently into the output string. This approach allows to figure 113468d75effSDimitry Andric // out which bytes of the output string depends on which argument and thus to 113568d75effSDimitry Andric // propagate labels more precisely. 113668d75effSDimitry Andric // 113768d75effSDimitry Andric // WARNING: This implementation does not support conversion specifiers with 113868d75effSDimitry Andric // positional arguments. 113968d75effSDimitry Andric static int format_buffer(char *str, size_t size, const char *fmt, 114068d75effSDimitry Andric dfsan_label *va_labels, dfsan_label *ret_label, 114168d75effSDimitry Andric va_list ap) { 114268d75effSDimitry Andric Formatter formatter(str, fmt, size); 114368d75effSDimitry Andric 114468d75effSDimitry Andric while (*formatter.fmt_cur) { 114568d75effSDimitry Andric formatter.fmt_start = formatter.fmt_cur; 114668d75effSDimitry Andric formatter.width = -1; 114768d75effSDimitry Andric int retval = 0; 114868d75effSDimitry Andric 114968d75effSDimitry Andric if (*formatter.fmt_cur != '%') { 115068d75effSDimitry Andric // Ordinary character. Consume all the characters until a '%' or the end 115168d75effSDimitry Andric // of the string. 115268d75effSDimitry Andric for (; *(formatter.fmt_cur + 1) && *(formatter.fmt_cur + 1) != '%'; 115368d75effSDimitry Andric ++formatter.fmt_cur) {} 115468d75effSDimitry Andric retval = formatter.format(); 115568d75effSDimitry Andric dfsan_set_label(0, formatter.str_cur(), 115668d75effSDimitry Andric formatter.num_written_bytes(retval)); 115768d75effSDimitry Andric } else { 115868d75effSDimitry Andric // Conversion directive. Consume all the characters until a conversion 115968d75effSDimitry Andric // specifier or the end of the string. 116068d75effSDimitry Andric bool end_fmt = false; 116168d75effSDimitry Andric for (; *formatter.fmt_cur && !end_fmt; ) { 116268d75effSDimitry Andric switch (*++formatter.fmt_cur) { 116368d75effSDimitry Andric case 'd': 116468d75effSDimitry Andric case 'i': 116568d75effSDimitry Andric case 'o': 116668d75effSDimitry Andric case 'u': 116768d75effSDimitry Andric case 'x': 116868d75effSDimitry Andric case 'X': 116968d75effSDimitry Andric switch (*(formatter.fmt_cur - 1)) { 117068d75effSDimitry Andric case 'h': 117168d75effSDimitry Andric // Also covers the 'hh' case (since the size of the arg is still 117268d75effSDimitry Andric // an int). 117368d75effSDimitry Andric retval = formatter.format(va_arg(ap, int)); 117468d75effSDimitry Andric break; 117568d75effSDimitry Andric case 'l': 117668d75effSDimitry Andric if (formatter.fmt_cur - formatter.fmt_start >= 2 && 117768d75effSDimitry Andric *(formatter.fmt_cur - 2) == 'l') { 117868d75effSDimitry Andric retval = formatter.format(va_arg(ap, long long int)); 117968d75effSDimitry Andric } else { 118068d75effSDimitry Andric retval = formatter.format(va_arg(ap, long int)); 118168d75effSDimitry Andric } 118268d75effSDimitry Andric break; 118368d75effSDimitry Andric case 'q': 118468d75effSDimitry Andric retval = formatter.format(va_arg(ap, long long int)); 118568d75effSDimitry Andric break; 118668d75effSDimitry Andric case 'j': 118768d75effSDimitry Andric retval = formatter.format(va_arg(ap, intmax_t)); 118868d75effSDimitry Andric break; 118968d75effSDimitry Andric case 'z': 119068d75effSDimitry Andric case 't': 119168d75effSDimitry Andric retval = formatter.format(va_arg(ap, size_t)); 119268d75effSDimitry Andric break; 119368d75effSDimitry Andric default: 119468d75effSDimitry Andric retval = formatter.format(va_arg(ap, int)); 119568d75effSDimitry Andric } 119668d75effSDimitry Andric dfsan_set_label(*va_labels++, formatter.str_cur(), 119768d75effSDimitry Andric formatter.num_written_bytes(retval)); 119868d75effSDimitry Andric end_fmt = true; 119968d75effSDimitry Andric break; 120068d75effSDimitry Andric 120168d75effSDimitry Andric case 'a': 120268d75effSDimitry Andric case 'A': 120368d75effSDimitry Andric case 'e': 120468d75effSDimitry Andric case 'E': 120568d75effSDimitry Andric case 'f': 120668d75effSDimitry Andric case 'F': 120768d75effSDimitry Andric case 'g': 120868d75effSDimitry Andric case 'G': 120968d75effSDimitry Andric if (*(formatter.fmt_cur - 1) == 'L') { 121068d75effSDimitry Andric retval = formatter.format(va_arg(ap, long double)); 121168d75effSDimitry Andric } else { 121268d75effSDimitry Andric retval = formatter.format(va_arg(ap, double)); 121368d75effSDimitry Andric } 121468d75effSDimitry Andric dfsan_set_label(*va_labels++, formatter.str_cur(), 121568d75effSDimitry Andric formatter.num_written_bytes(retval)); 121668d75effSDimitry Andric end_fmt = true; 121768d75effSDimitry Andric break; 121868d75effSDimitry Andric 121968d75effSDimitry Andric case 'c': 122068d75effSDimitry Andric retval = formatter.format(va_arg(ap, int)); 122168d75effSDimitry Andric dfsan_set_label(*va_labels++, formatter.str_cur(), 122268d75effSDimitry Andric formatter.num_written_bytes(retval)); 122368d75effSDimitry Andric end_fmt = true; 122468d75effSDimitry Andric break; 122568d75effSDimitry Andric 122668d75effSDimitry Andric case 's': { 122768d75effSDimitry Andric char *arg = va_arg(ap, char *); 122868d75effSDimitry Andric retval = formatter.format(arg); 122968d75effSDimitry Andric va_labels++; 123068d75effSDimitry Andric internal_memcpy(shadow_for(formatter.str_cur()), shadow_for(arg), 123168d75effSDimitry Andric sizeof(dfsan_label) * 123268d75effSDimitry Andric formatter.num_written_bytes(retval)); 123368d75effSDimitry Andric end_fmt = true; 123468d75effSDimitry Andric break; 123568d75effSDimitry Andric } 123668d75effSDimitry Andric 123768d75effSDimitry Andric case 'p': 123868d75effSDimitry Andric retval = formatter.format(va_arg(ap, void *)); 123968d75effSDimitry Andric dfsan_set_label(*va_labels++, formatter.str_cur(), 124068d75effSDimitry Andric formatter.num_written_bytes(retval)); 124168d75effSDimitry Andric end_fmt = true; 124268d75effSDimitry Andric break; 124368d75effSDimitry Andric 124468d75effSDimitry Andric case 'n': { 124568d75effSDimitry Andric int *ptr = va_arg(ap, int *); 124668d75effSDimitry Andric *ptr = (int)formatter.str_off; 124768d75effSDimitry Andric va_labels++; 124868d75effSDimitry Andric dfsan_set_label(0, ptr, sizeof(ptr)); 124968d75effSDimitry Andric end_fmt = true; 125068d75effSDimitry Andric break; 125168d75effSDimitry Andric } 125268d75effSDimitry Andric 125368d75effSDimitry Andric case '%': 125468d75effSDimitry Andric retval = formatter.format(); 125568d75effSDimitry Andric dfsan_set_label(0, formatter.str_cur(), 125668d75effSDimitry Andric formatter.num_written_bytes(retval)); 125768d75effSDimitry Andric end_fmt = true; 125868d75effSDimitry Andric break; 125968d75effSDimitry Andric 126068d75effSDimitry Andric case '*': 126168d75effSDimitry Andric formatter.width = va_arg(ap, int); 126268d75effSDimitry Andric va_labels++; 126368d75effSDimitry Andric break; 126468d75effSDimitry Andric 126568d75effSDimitry Andric default: 126668d75effSDimitry Andric break; 126768d75effSDimitry Andric } 126868d75effSDimitry Andric } 126968d75effSDimitry Andric } 127068d75effSDimitry Andric 127168d75effSDimitry Andric if (retval < 0) { 127268d75effSDimitry Andric return retval; 127368d75effSDimitry Andric } 127468d75effSDimitry Andric 127568d75effSDimitry Andric formatter.fmt_cur++; 127668d75effSDimitry Andric formatter.str_off += retval; 127768d75effSDimitry Andric } 127868d75effSDimitry Andric 127968d75effSDimitry Andric *ret_label = 0; 128068d75effSDimitry Andric 128168d75effSDimitry Andric // Number of bytes written in total. 128268d75effSDimitry Andric return formatter.str_off; 128368d75effSDimitry Andric } 128468d75effSDimitry Andric 128568d75effSDimitry Andric extern "C" { 128668d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 128768d75effSDimitry Andric int __dfsw_sprintf(char *str, const char *format, dfsan_label str_label, 128868d75effSDimitry Andric dfsan_label format_label, dfsan_label *va_labels, 128968d75effSDimitry Andric dfsan_label *ret_label, ...) { 129068d75effSDimitry Andric va_list ap; 129168d75effSDimitry Andric va_start(ap, ret_label); 129268d75effSDimitry Andric int ret = format_buffer(str, ~0ul, format, va_labels, ret_label, ap); 129368d75effSDimitry Andric va_end(ap); 129468d75effSDimitry Andric return ret; 129568d75effSDimitry Andric } 129668d75effSDimitry Andric 129768d75effSDimitry Andric SANITIZER_INTERFACE_ATTRIBUTE 129868d75effSDimitry Andric int __dfsw_snprintf(char *str, size_t size, const char *format, 129968d75effSDimitry Andric dfsan_label str_label, dfsan_label size_label, 130068d75effSDimitry Andric dfsan_label format_label, dfsan_label *va_labels, 130168d75effSDimitry Andric dfsan_label *ret_label, ...) { 130268d75effSDimitry Andric va_list ap; 130368d75effSDimitry Andric va_start(ap, ret_label); 130468d75effSDimitry Andric int ret = format_buffer(str, size, format, va_labels, ret_label, ap); 130568d75effSDimitry Andric va_end(ap); 130668d75effSDimitry Andric return ret; 130768d75effSDimitry Andric } 130868d75effSDimitry Andric 130968d75effSDimitry Andric // Default empty implementations (weak). Users should redefine them. 131068d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_pc_guard, u32 *) {} 131168d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_pc_guard_init, u32 *, 131268d75effSDimitry Andric u32 *) {} 131368d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_pcs_init, void) {} 131468d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __sanitizer_cov_trace_pc_indir, void) {} 131568d75effSDimitry Andric 131668d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __dfsw___sanitizer_cov_trace_cmp, void) {} 131768d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __dfsw___sanitizer_cov_trace_cmp1, void) {} 131868d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __dfsw___sanitizer_cov_trace_cmp2, void) {} 131968d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __dfsw___sanitizer_cov_trace_cmp4, void) {} 132068d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __dfsw___sanitizer_cov_trace_cmp8, void) {} 132168d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __dfsw___sanitizer_cov_trace_const_cmp1, 132268d75effSDimitry Andric void) {} 132368d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __dfsw___sanitizer_cov_trace_const_cmp2, 132468d75effSDimitry Andric void) {} 132568d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __dfsw___sanitizer_cov_trace_const_cmp4, 132668d75effSDimitry Andric void) {} 132768d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __dfsw___sanitizer_cov_trace_const_cmp8, 132868d75effSDimitry Andric void) {} 132968d75effSDimitry Andric SANITIZER_INTERFACE_WEAK_DEF(void, __dfsw___sanitizer_cov_trace_switch, void) {} 133068d75effSDimitry Andric } // extern "C" 1331