xref: /freebsd-src/contrib/llvm-project/compiler-rt/lib/interception/interception_linux.h (revision 52418fc2be8efa5172b90a3a9e617017173612c4)
10b57cec5SDimitry Andric //===-- interception_linux.h ------------------------------------*- C++ -*-===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric //
90b57cec5SDimitry Andric // This file is a part of AddressSanitizer, an address sanity checker.
100b57cec5SDimitry Andric //
110b57cec5SDimitry Andric // Linux-specific interception methods.
120b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
130b57cec5SDimitry Andric 
140b57cec5SDimitry Andric #if SANITIZER_LINUX || SANITIZER_FREEBSD || SANITIZER_NETBSD || \
15e8d8bef9SDimitry Andric     SANITIZER_SOLARIS
160b57cec5SDimitry Andric 
170b57cec5SDimitry Andric #if !defined(INCLUDED_FROM_INTERCEPTION_LIB)
1806c3fb27SDimitry Andric # error interception_linux.h should be included from interception library only
190b57cec5SDimitry Andric #endif
200b57cec5SDimitry Andric 
210b57cec5SDimitry Andric #ifndef INTERCEPTION_LINUX_H
220b57cec5SDimitry Andric #define INTERCEPTION_LINUX_H
230b57cec5SDimitry Andric 
240b57cec5SDimitry Andric namespace __interception {
250b57cec5SDimitry Andric bool InterceptFunction(const char *name, uptr *ptr_to_real, uptr func,
2606c3fb27SDimitry Andric                        uptr trampoline);
270b57cec5SDimitry Andric bool InterceptFunction(const char *name, const char *ver, uptr *ptr_to_real,
2806c3fb27SDimitry Andric                        uptr func, uptr trampoline);
290b57cec5SDimitry Andric }  // namespace __interception
300b57cec5SDimitry Andric 
31*52418fc2SDimitry Andric // Cast func to type of REAL(func) before casting to uptr in case it is an
32*52418fc2SDimitry Andric // overloaded function, which is the case for some glibc functions when
33*52418fc2SDimitry Andric // _FORTIFY_SOURCE is used. This disambiguates which overload to use.
340b57cec5SDimitry Andric #define INTERCEPT_FUNCTION_LINUX_OR_FREEBSD(func)            \
350b57cec5SDimitry Andric   ::__interception::InterceptFunction(                       \
36*52418fc2SDimitry Andric       #func, (::__interception::uptr *)&REAL(func),          \
37*52418fc2SDimitry Andric       (::__interception::uptr)(decltype(REAL(func)))&(func), \
3806c3fb27SDimitry Andric       (::__interception::uptr) &TRAMPOLINE(func))
390b57cec5SDimitry Andric 
40e8d8bef9SDimitry Andric // dlvsym is a GNU extension supported by some other platforms.
41e8d8bef9SDimitry Andric #if SANITIZER_GLIBC || SANITIZER_FREEBSD || SANITIZER_NETBSD
420b57cec5SDimitry Andric #define INTERCEPT_FUNCTION_VER_LINUX_OR_FREEBSD(func, symver) \
430b57cec5SDimitry Andric   ::__interception::InterceptFunction(                        \
440b57cec5SDimitry Andric       #func, symver,                                          \
450b57cec5SDimitry Andric       (::__interception::uptr *)&REAL(func),                  \
46*52418fc2SDimitry Andric       (::__interception::uptr)(decltype(REAL(func)))&(func),  \
4706c3fb27SDimitry Andric       (::__interception::uptr)&TRAMPOLINE(func))
480b57cec5SDimitry Andric #else
490b57cec5SDimitry Andric #define INTERCEPT_FUNCTION_VER_LINUX_OR_FREEBSD(func, symver) \
500b57cec5SDimitry Andric   INTERCEPT_FUNCTION_LINUX_OR_FREEBSD(func)
51e8d8bef9SDimitry Andric #endif  // SANITIZER_GLIBC || SANITIZER_FREEBSD || SANITIZER_NETBSD
520b57cec5SDimitry Andric 
530b57cec5SDimitry Andric #endif  // INTERCEPTION_LINUX_H
540b57cec5SDimitry Andric #endif  // SANITIZER_LINUX || SANITIZER_FREEBSD || SANITIZER_NETBSD ||
55e8d8bef9SDimitry Andric         // SANITIZER_SOLARIS
56