193 lines · cpp
1//===-- sanitizer_wrappers.cpp ----------------------------------*- 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-exception6//7//===----------------------------------------------------------------------===//8//9// Redirect some functions to sanitizer interceptors.10//11//===----------------------------------------------------------------------===//12 13#include <dlfcn.h>14#include <errno.h>15#include <fcntl.h>16#include <inttypes.h>17#include <stdarg.h>18#include <stdio.h>19#include <unistd.h>20 21#include <tuple>22 23namespace __sanitizer {24unsigned long internal_open(const char *filename, int flags);25unsigned long internal_open(const char *filename, int flags, unsigned mode);26unsigned long internal_close(int fd);27unsigned long internal_stat(const char *path, void *buf);28unsigned long internal_lstat(const char *path, void *buf);29unsigned long internal_fstat(int fd, void *buf);30size_t internal_strlen(const char *s);31unsigned long internal_mmap(void *addr, uintptr_t length, int prot, int flags,32 int fd, unsigned long long offset);33void *internal_memcpy(void *dest, const void *src, unsigned long n);34// Used to propagate errno.35bool internal_iserror(uintptr_t retval, int *rverrno = 0);36} // namespace __sanitizer37 38namespace {39 40template <typename T>41struct GetTypes;42 43template <typename R, typename... Args>44struct GetTypes<R(Args...)> {45 using Result = R;46 template <size_t i>47 struct Arg {48 using Type = typename std::tuple_element<i, std::tuple<Args...>>::type;49 };50};51 52#define LLVM_SYMBOLIZER_GET_FUNC(Function) \53 ((__interceptor_##Function) \54 ? (__interceptor_##Function) \55 : reinterpret_cast<decltype(&Function)>(dlsym(RTLD_NEXT, #Function)))56 57#define LLVM_SYMBOLIZER_INTERCEPTOR1(Function, ...) \58 GetTypes<__VA_ARGS__>::Result __interceptor_##Function( \59 GetTypes<__VA_ARGS__>::Arg<0>::Type) __attribute__((weak)); \60 GetTypes<__VA_ARGS__>::Result Function( \61 GetTypes<__VA_ARGS__>::Arg<0>::Type arg0) { \62 return LLVM_SYMBOLIZER_GET_FUNC(Function)(arg0); \63 }64 65#define LLVM_SYMBOLIZER_INTERCEPTOR2(Function, ...) \66 GetTypes<__VA_ARGS__>::Result __interceptor_##Function( \67 GetTypes<__VA_ARGS__>::Arg<0>::Type, \68 GetTypes<__VA_ARGS__>::Arg<1>::Type) __attribute__((weak)); \69 GetTypes<__VA_ARGS__>::Result Function( \70 GetTypes<__VA_ARGS__>::Arg<0>::Type arg0, \71 GetTypes<__VA_ARGS__>::Arg<1>::Type arg1) { \72 return LLVM_SYMBOLIZER_GET_FUNC(Function)(arg0, arg1); \73 }74 75#define LLVM_SYMBOLIZER_INTERCEPTOR3(Function, ...) \76 GetTypes<__VA_ARGS__>::Result __interceptor_##Function( \77 GetTypes<__VA_ARGS__>::Arg<0>::Type, \78 GetTypes<__VA_ARGS__>::Arg<1>::Type, \79 GetTypes<__VA_ARGS__>::Arg<2>::Type) __attribute__((weak)); \80 GetTypes<__VA_ARGS__>::Result Function( \81 GetTypes<__VA_ARGS__>::Arg<0>::Type arg0, \82 GetTypes<__VA_ARGS__>::Arg<1>::Type arg1, \83 GetTypes<__VA_ARGS__>::Arg<2>::Type arg2) { \84 return LLVM_SYMBOLIZER_GET_FUNC(Function)(arg0, arg1, arg2); \85 }86 87#define LLVM_SYMBOLIZER_INTERCEPTOR4(Function, ...) \88 GetTypes<__VA_ARGS__>::Result __interceptor_##Function( \89 GetTypes<__VA_ARGS__>::Arg<0>::Type, \90 GetTypes<__VA_ARGS__>::Arg<1>::Type, \91 GetTypes<__VA_ARGS__>::Arg<2>::Type, \92 GetTypes<__VA_ARGS__>::Arg<3>::Type) __attribute__((weak)); \93 GetTypes<__VA_ARGS__>::Result Function( \94 GetTypes<__VA_ARGS__>::Arg<0>::Type arg0, \95 GetTypes<__VA_ARGS__>::Arg<1>::Type arg1, \96 GetTypes<__VA_ARGS__>::Arg<2>::Type arg2, \97 GetTypes<__VA_ARGS__>::Arg<3>::Type arg3) { \98 return LLVM_SYMBOLIZER_GET_FUNC(Function)(arg0, arg1, arg2, arg3); \99 }100 101} // namespace102 103// C-style interface around internal sanitizer libc functions.104extern "C" {105 106#define RETURN_OR_SET_ERRNO(T, res) \107 int rverrno; \108 if (__sanitizer::internal_iserror(res, &rverrno)) { \109 errno = rverrno; \110 return (T)-1; \111 } \112 return (T)res;113 114int open(const char *filename, int flags, ...) {115 unsigned long res;116 if (flags | O_CREAT) {117 va_list va;118 va_start(va, flags);119 unsigned mode = va_arg(va, unsigned);120 va_end(va);121 res = __sanitizer::internal_open(filename, flags, mode);122 } else {123 res = __sanitizer::internal_open(filename, flags);124 }125 RETURN_OR_SET_ERRNO(int, res);126}127 128int close(int fd) {129 unsigned long res = __sanitizer::internal_close(fd);130 RETURN_OR_SET_ERRNO(int, res);131}132 133#define STAT(func, arg, buf) \134 unsigned long res = __sanitizer::internal_##func(arg, buf); \135 RETURN_OR_SET_ERRNO(int, res);136 137int stat(const char *path, struct stat *buf) { STAT(stat, path, buf); }138 139int lstat(const char *path, struct stat *buf) { STAT(lstat, path, buf); }140 141int fstat(int fd, struct stat *buf) { STAT(fstat, fd, buf); }142 143// Redirect versioned stat functions to the __sanitizer::internal() as well.144int __xstat(int version, const char *path, struct stat *buf) {145 STAT(stat, path, buf);146}147 148int __lxstat(int version, const char *path, struct stat *buf) {149 STAT(lstat, path, buf);150}151 152int __fxstat(int version, int fd, struct stat *buf) { STAT(fstat, fd, buf); }153 154size_t strlen(const char *s) { return __sanitizer::internal_strlen(s); }155 156void *mmap(void *addr, size_t length, int prot, int flags, int fd,157 off_t offset) {158 unsigned long res =159 __sanitizer::internal_mmap(addr, length, prot, flags, fd, offset);160 RETURN_OR_SET_ERRNO(void *, res);161}162 163LLVM_SYMBOLIZER_INTERCEPTOR3(read, ssize_t(int, void *, size_t))164LLVM_SYMBOLIZER_INTERCEPTOR4(pread, ssize_t(int, void *, size_t, off_t))165LLVM_SYMBOLIZER_INTERCEPTOR4(pread64, ssize_t(int, void *, size_t, off64_t))166LLVM_SYMBOLIZER_INTERCEPTOR2(realpath, char *(const char *, char *))167 168LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_cond_broadcast, int(pthread_cond_t *))169LLVM_SYMBOLIZER_INTERCEPTOR2(pthread_cond_wait,170 int(pthread_cond_t *, pthread_mutex_t *))171LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_mutex_lock, int(pthread_mutex_t *))172LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_mutex_unlock, int(pthread_mutex_t *))173LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_mutex_destroy, int(pthread_mutex_t *))174LLVM_SYMBOLIZER_INTERCEPTOR2(pthread_mutex_init,175 int(pthread_mutex_t *,176 const pthread_mutexattr_t *))177LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_mutexattr_destroy,178 int(pthread_mutexattr_t *))179LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_mutexattr_init, int(pthread_mutexattr_t *))180LLVM_SYMBOLIZER_INTERCEPTOR2(pthread_mutexattr_settype,181 int(pthread_mutexattr_t *, int))182LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_getspecific, void *(pthread_key_t))183LLVM_SYMBOLIZER_INTERCEPTOR2(pthread_key_create,184 int(pthread_key_t *, void (*)(void *)))185LLVM_SYMBOLIZER_INTERCEPTOR2(pthread_once,186 int(pthread_once_t *, void (*)(void)))187LLVM_SYMBOLIZER_INTERCEPTOR2(pthread_setspecific,188 int(pthread_key_t, const void *))189LLVM_SYMBOLIZER_INTERCEPTOR3(pthread_sigmask,190 int(int, const sigset_t *, sigset_t *))191 192} // extern "C"193