13cab2bb3Spatrick //===-- sanitizer_wrappers.cpp ----------------------------------*- C++ -*-===//
23cab2bb3Spatrick //
33cab2bb3Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
43cab2bb3Spatrick // See https://llvm.org/LICENSE.txt for license information.
53cab2bb3Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
63cab2bb3Spatrick //
73cab2bb3Spatrick //===----------------------------------------------------------------------===//
83cab2bb3Spatrick //
93cab2bb3Spatrick // Redirect some functions to sanitizer interceptors.
103cab2bb3Spatrick //
113cab2bb3Spatrick //===----------------------------------------------------------------------===//
123cab2bb3Spatrick
133cab2bb3Spatrick #include <dlfcn.h>
143cab2bb3Spatrick #include <errno.h>
153cab2bb3Spatrick #include <fcntl.h>
16*810390e3Srobert #include <inttypes.h>
173cab2bb3Spatrick #include <stdarg.h>
183cab2bb3Spatrick #include <stdio.h>
193cab2bb3Spatrick #include <unistd.h>
203cab2bb3Spatrick
213cab2bb3Spatrick #include <tuple>
223cab2bb3Spatrick
233cab2bb3Spatrick namespace __sanitizer {
243cab2bb3Spatrick unsigned long internal_open(const char *filename, int flags);
253cab2bb3Spatrick unsigned long internal_open(const char *filename, int flags, unsigned mode);
263cab2bb3Spatrick unsigned long internal_close(int fd);
273cab2bb3Spatrick unsigned long internal_stat(const char *path, void *buf);
283cab2bb3Spatrick unsigned long internal_lstat(const char *path, void *buf);
293cab2bb3Spatrick unsigned long internal_fstat(int fd, void *buf);
303cab2bb3Spatrick size_t internal_strlen(const char *s);
31*810390e3Srobert unsigned long internal_mmap(void *addr, uintptr_t length, int prot, int flags,
32*810390e3Srobert int fd, unsigned long long offset);
333cab2bb3Spatrick void *internal_memcpy(void *dest, const void *src, unsigned long n);
343cab2bb3Spatrick // Used to propagate errno.
35*810390e3Srobert bool internal_iserror(uintptr_t retval, int *rverrno = 0);
363cab2bb3Spatrick } // namespace __sanitizer
373cab2bb3Spatrick
383cab2bb3Spatrick namespace {
393cab2bb3Spatrick
403cab2bb3Spatrick template <typename T>
413cab2bb3Spatrick struct GetTypes;
423cab2bb3Spatrick
433cab2bb3Spatrick template <typename R, typename... Args>
443cab2bb3Spatrick struct GetTypes<R(Args...)> {
453cab2bb3Spatrick using Result = R;
463cab2bb3Spatrick template <size_t i>
473cab2bb3Spatrick struct Arg {
483cab2bb3Spatrick using Type = typename std::tuple_element<i, std::tuple<Args...>>::type;
493cab2bb3Spatrick };
503cab2bb3Spatrick };
513cab2bb3Spatrick
523cab2bb3Spatrick #define LLVM_SYMBOLIZER_GET_FUNC(Function) \
533cab2bb3Spatrick ((__interceptor_##Function) \
543cab2bb3Spatrick ? (__interceptor_##Function) \
553cab2bb3Spatrick : reinterpret_cast<decltype(&Function)>(dlsym(RTLD_NEXT, #Function)))
563cab2bb3Spatrick
573cab2bb3Spatrick #define LLVM_SYMBOLIZER_INTERCEPTOR1(Function, ...) \
583cab2bb3Spatrick GetTypes<__VA_ARGS__>::Result __interceptor_##Function( \
593cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<0>::Type) __attribute__((weak)); \
603cab2bb3Spatrick GetTypes<__VA_ARGS__>::Result Function( \
613cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<0>::Type arg0) { \
623cab2bb3Spatrick return LLVM_SYMBOLIZER_GET_FUNC(Function)(arg0); \
633cab2bb3Spatrick }
643cab2bb3Spatrick
653cab2bb3Spatrick #define LLVM_SYMBOLIZER_INTERCEPTOR2(Function, ...) \
663cab2bb3Spatrick GetTypes<__VA_ARGS__>::Result __interceptor_##Function( \
673cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<0>::Type, \
683cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<1>::Type) __attribute__((weak)); \
693cab2bb3Spatrick GetTypes<__VA_ARGS__>::Result Function( \
703cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<0>::Type arg0, \
713cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<1>::Type arg1) { \
723cab2bb3Spatrick return LLVM_SYMBOLIZER_GET_FUNC(Function)(arg0, arg1); \
733cab2bb3Spatrick }
743cab2bb3Spatrick
753cab2bb3Spatrick #define LLVM_SYMBOLIZER_INTERCEPTOR3(Function, ...) \
763cab2bb3Spatrick GetTypes<__VA_ARGS__>::Result __interceptor_##Function( \
773cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<0>::Type, \
783cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<1>::Type, \
793cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<2>::Type) __attribute__((weak)); \
803cab2bb3Spatrick GetTypes<__VA_ARGS__>::Result Function( \
813cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<0>::Type arg0, \
823cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<1>::Type arg1, \
833cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<2>::Type arg2) { \
843cab2bb3Spatrick return LLVM_SYMBOLIZER_GET_FUNC(Function)(arg0, arg1, arg2); \
853cab2bb3Spatrick }
863cab2bb3Spatrick
873cab2bb3Spatrick #define LLVM_SYMBOLIZER_INTERCEPTOR4(Function, ...) \
883cab2bb3Spatrick GetTypes<__VA_ARGS__>::Result __interceptor_##Function( \
893cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<0>::Type, \
903cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<1>::Type, \
913cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<2>::Type, \
923cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<3>::Type) __attribute__((weak)); \
933cab2bb3Spatrick GetTypes<__VA_ARGS__>::Result Function( \
943cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<0>::Type arg0, \
953cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<1>::Type arg1, \
963cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<2>::Type arg2, \
973cab2bb3Spatrick GetTypes<__VA_ARGS__>::Arg<3>::Type arg3) { \
983cab2bb3Spatrick return LLVM_SYMBOLIZER_GET_FUNC(Function)(arg0, arg1, arg2, arg3); \
993cab2bb3Spatrick }
1003cab2bb3Spatrick
1013cab2bb3Spatrick } // namespace
1023cab2bb3Spatrick
1033cab2bb3Spatrick // C-style interface around internal sanitizer libc functions.
1043cab2bb3Spatrick extern "C" {
1053cab2bb3Spatrick
1063cab2bb3Spatrick #define RETURN_OR_SET_ERRNO(T, res) \
1073cab2bb3Spatrick int rverrno; \
1083cab2bb3Spatrick if (__sanitizer::internal_iserror(res, &rverrno)) { \
1093cab2bb3Spatrick errno = rverrno; \
1103cab2bb3Spatrick return (T)-1; \
1113cab2bb3Spatrick } \
1123cab2bb3Spatrick return (T)res;
1133cab2bb3Spatrick
open(const char * filename,int flags,...)1143cab2bb3Spatrick int open(const char *filename, int flags, ...) {
1153cab2bb3Spatrick unsigned long res;
1163cab2bb3Spatrick if (flags | O_CREAT) {
1173cab2bb3Spatrick va_list va;
1183cab2bb3Spatrick va_start(va, flags);
1193cab2bb3Spatrick unsigned mode = va_arg(va, unsigned);
1203cab2bb3Spatrick va_end(va);
1213cab2bb3Spatrick res = __sanitizer::internal_open(filename, flags, mode);
1223cab2bb3Spatrick } else {
1233cab2bb3Spatrick res = __sanitizer::internal_open(filename, flags);
1243cab2bb3Spatrick }
1253cab2bb3Spatrick RETURN_OR_SET_ERRNO(int, res);
1263cab2bb3Spatrick }
1273cab2bb3Spatrick
close(int fd)1283cab2bb3Spatrick int close(int fd) {
1293cab2bb3Spatrick unsigned long res = __sanitizer::internal_close(fd);
1303cab2bb3Spatrick RETURN_OR_SET_ERRNO(int, res);
1313cab2bb3Spatrick }
1323cab2bb3Spatrick
1333cab2bb3Spatrick #define STAT(func, arg, buf) \
1343cab2bb3Spatrick unsigned long res = __sanitizer::internal_##func(arg, buf); \
1353cab2bb3Spatrick RETURN_OR_SET_ERRNO(int, res);
1363cab2bb3Spatrick
stat(const char * path,struct stat * buf)1373cab2bb3Spatrick int stat(const char *path, struct stat *buf) { STAT(stat, path, buf); }
1383cab2bb3Spatrick
lstat(const char * path,struct stat * buf)1393cab2bb3Spatrick int lstat(const char *path, struct stat *buf) { STAT(lstat, path, buf); }
1403cab2bb3Spatrick
fstat(int fd,struct stat * buf)1413cab2bb3Spatrick int fstat(int fd, struct stat *buf) { STAT(fstat, fd, buf); }
1423cab2bb3Spatrick
1433cab2bb3Spatrick // Redirect versioned stat functions to the __sanitizer::internal() as well.
__xstat(int version,const char * path,struct stat * buf)1443cab2bb3Spatrick int __xstat(int version, const char *path, struct stat *buf) {
1453cab2bb3Spatrick STAT(stat, path, buf);
1463cab2bb3Spatrick }
1473cab2bb3Spatrick
__lxstat(int version,const char * path,struct stat * buf)1483cab2bb3Spatrick int __lxstat(int version, const char *path, struct stat *buf) {
1493cab2bb3Spatrick STAT(lstat, path, buf);
1503cab2bb3Spatrick }
1513cab2bb3Spatrick
__fxstat(int version,int fd,struct stat * buf)1523cab2bb3Spatrick int __fxstat(int version, int fd, struct stat *buf) { STAT(fstat, fd, buf); }
1533cab2bb3Spatrick
strlen(const char * s)1543cab2bb3Spatrick size_t strlen(const char *s) { return __sanitizer::internal_strlen(s); }
1553cab2bb3Spatrick
mmap(void * addr,size_t length,int prot,int flags,int fd,off_t offset)1563cab2bb3Spatrick void *mmap(void *addr, size_t length, int prot, int flags, int fd,
1573cab2bb3Spatrick off_t offset) {
158*810390e3Srobert unsigned long res =
159*810390e3Srobert __sanitizer::internal_mmap(addr, length, prot, flags, fd, offset);
1603cab2bb3Spatrick RETURN_OR_SET_ERRNO(void *, res);
1613cab2bb3Spatrick }
1623cab2bb3Spatrick
1633cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR3(read, ssize_t(int, void *, size_t))
1643cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR4(pread, ssize_t(int, void *, size_t, off_t))
1653cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR4(pread64, ssize_t(int, void *, size_t, off64_t))
1663cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR2(realpath, char *(const char *, char *))
1673cab2bb3Spatrick
1683cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_cond_broadcast, int(pthread_cond_t *))
1693cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR2(pthread_cond_wait,
1703cab2bb3Spatrick int(pthread_cond_t *, pthread_mutex_t *))
1713cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_mutex_lock, int(pthread_mutex_t *))
1723cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_mutex_unlock, int(pthread_mutex_t *))
1733cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_mutex_destroy, int(pthread_mutex_t *))
1743cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR2(pthread_mutex_init,
1753cab2bb3Spatrick int(pthread_mutex_t *,
1763cab2bb3Spatrick const pthread_mutexattr_t *))
1773cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_mutexattr_destroy,
1783cab2bb3Spatrick int(pthread_mutexattr_t *))
1793cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_mutexattr_init, int(pthread_mutexattr_t *))
1803cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR2(pthread_mutexattr_settype,
1813cab2bb3Spatrick int(pthread_mutexattr_t *, int))
1823cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR1(pthread_getspecific, void *(pthread_key_t))
1833cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR2(pthread_key_create,
1843cab2bb3Spatrick int(pthread_key_t *, void (*)(void *)))
1853cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR2(pthread_once,
1863cab2bb3Spatrick int(pthread_once_t *, void (*)(void)))
1873cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR2(pthread_setspecific,
1883cab2bb3Spatrick int(pthread_key_t, const void *))
1893cab2bb3Spatrick LLVM_SYMBOLIZER_INTERCEPTOR3(pthread_sigmask,
1903cab2bb3Spatrick int(int, const sigset_t *, sigset_t *))
1913cab2bb3Spatrick
1923cab2bb3Spatrick } // extern "C"
193