asan_internal.h revision adf2b036127d7e887392d9be5b02069b777ee280
11e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//===-- asan_internal.h -----------------------------------------*- C++ -*-===// 21e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 31e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The LLVM Compiler Infrastructure 41e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 51e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This file is distributed under the University of Illinois Open Source 61e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// License. See LICENSE.TXT for details. 71e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 81e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//===----------------------------------------------------------------------===// 91e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This file is a part of AddressSanitizer, an address sanity checker. 111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// ASan-private header which defines various general utilities. 131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//===----------------------------------------------------------------------===// 141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifndef ASAN_INTERNAL_H 151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define ASAN_INTERNAL_H 161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 176f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#if !defined(__linux__) && !defined(__APPLE__) && !defined(_WIN32) 18d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany# error "This operating system is not supported by AddressSanitizer" 19d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany#endif 20d6567c5166412f6acdde851e767c26f332d51d3dKostya Serebryany 216f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#include <stdlib.h> // for size_t, uintptr_t, etc. 226f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko 23858220837f85db7eb5eedd5ea68603803b5eb3c9Kostya Serebryany#if defined(_WIN32) 246f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko// There's no <stdint.h> in Visual Studio 9, so we have to define [u]int*_t. 256f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenkotypedef unsigned __int8 uint8_t; 266f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenkotypedef unsigned __int16 uint16_t; 276f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenkotypedef unsigned __int32 uint32_t; 286f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenkotypedef unsigned __int64 uint64_t; 296f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenkotypedef __int8 int8_t; 306f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenkotypedef __int16 int16_t; 316f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenkotypedef __int32 int32_t; 326f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenkotypedef __int64 int64_t; 33adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov 34adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov# define ALIAS(x) // TODO(timurrrr): do we need this on Windows? 35adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov# define ALIGNED(x) __declspec(align(x)) 36adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov# define NOINLINE __declspec(noinline) 37adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov 38adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov# define ASAN_INTERFACE_ATTRIBUTE // TODO(timurrrr): do we need this on Win? 39adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov#else // defined(_WIN32) 40858220837f85db7eb5eedd5ea68603803b5eb3c9Kostya Serebryany# include <stdint.h> // for __WORDSIZE 41adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov 42adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov# define ALIAS(x) __attribute__((alias(x))) 43adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov# define ALIGNED(x) __attribute__((aligned(x))) 44adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov# define NOINLINE __attribute__((noinline)) 45adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov 46adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov# define ASAN_INTERFACE_ATTRIBUTE __attribute__((visibility("default"))) 47adf2b036127d7e887392d9be5b02069b777ee280Alexey Samsonov#endif // defined(_WIN32) 481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 494616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar// If __WORDSIZE was undefined by the platform, define it in terms of the 50669f543c1400f7a1b9a48313d2f95e54f750bd4bKostya Serebryany// compiler built-ins __LP64__ and _WIN64. 514616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#ifndef __WORDSIZE 52669f543c1400f7a1b9a48313d2f95e54f750bd4bKostya Serebryany#if __LP64__ || defined(_WIN64) 534616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#define __WORDSIZE 64 544616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#else 554616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#define __WORDSIZE 32 564616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#endif 574616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#endif 584616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar 59c8365231004cb1d956aba4164c89ea1398eadd6bAlexander Potapenko#if !defined(__has_feature) 60c8365231004cb1d956aba4164c89ea1398eadd6bAlexander Potapenko#define __has_feature(x) 0 61c8365231004cb1d956aba4164c89ea1398eadd6bAlexander Potapenko#endif 62c8365231004cb1d956aba4164c89ea1398eadd6bAlexander Potapenko 6313ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany#if defined(__has_feature) && __has_feature(address_sanitizer) 641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany# error "The AddressSanitizer run-time should not be" 651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany " instrumented by AddressSanitizer" 661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 68c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany// Build-time configuration options. 69c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany 70c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany// If set, asan will install its own SEGV signal handler. 71c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany#ifndef ASAN_NEEDS_SEGV 72c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany# define ASAN_NEEDS_SEGV 1 73c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany#endif 74c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany 75c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany// If set, asan will intercept C++ exception api call(s). 76c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany#ifndef ASAN_HAS_EXCEPTIONS 77c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany# define ASAN_HAS_EXCEPTIONS 1 78c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany#endif 79c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany 80c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany// If set, asan uses the values of SHADOW_SCALE and SHADOW_OFFSET 81c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany// provided by the instrumented objects. Otherwise constants are used. 82c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany#ifndef ASAN_FLEXIBLE_MAPPING_AND_OFFSET 83c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany# define ASAN_FLEXIBLE_MAPPING_AND_OFFSET 0 84c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany#endif 85c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany 861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// All internal functions in asan reside inside the __asan namespace 871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// to avoid namespace collisions with the user programs. 881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Seperate namespace also makes it simpler to distinguish the asan run-time 891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// functions from the instrumented user code in a profile. 901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanynamespace __asan { 911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyclass AsanThread; 931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystruct AsanStackTrace; 941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 95218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// asan_rtl.cc 961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid CheckFailed(const char *cond, const char *file, int line); 971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid ShowStatsAndAbort(); 981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 99218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// asan_globals.cc 1001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanybool DescribeAddrIfGlobal(uintptr_t addr); 1011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 102218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// asan_malloc_linux.cc / asan_malloc_mac.cc 1031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid ReplaceSystemMalloc(); 1041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 105de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid OutOfMemoryMessageAndDie(const char *mem_type, size_t size); 106de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 107218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// asan_linux.cc / asan_mac.cc 1081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *AsanDoesNotSupportStaticLinkage(); 109de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyint AsanOpenReadonly(const char* filename); 1101e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenkoconst char *AsanGetEnv(const char *name); 111de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 112a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyvoid *AsanMmapFixedNoReserve(uintptr_t fixed_addr, size_t size); 113a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyvoid *AsanMmapFixedReserve(uintptr_t fixed_addr, size_t size); 114a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyvoid *AsanMprotect(uintptr_t fixed_addr, size_t size); 115de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid *AsanMmapSomewhereOrDie(size_t size, const char *where); 116de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid AsanUnmapOrDie(void *ptr, size_t size); 117de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 118ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryanyvoid AsanDisableCoreDumper(); 1199107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryanyvoid GetPcSpBp(void *context, uintptr_t *pc, uintptr_t *sp, uintptr_t *bp); 120ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryany 1210ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanysize_t AsanRead(int fd, void *buf, size_t count); 1220ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanysize_t AsanWrite(int fd, const void *buf, size_t count); 123de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyint AsanClose(int fd); 1241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1254803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryanybool AsanInterceptsSignal(int signum); 126a7e760a53bc43b8e09bfdf5cd6f215267ba99729Kostya Serebryanyvoid InstallSignalHandlers(); 1270ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanyint GetPid(); 128cc4e6862c6a8f8f3ead96bd32b815184a36fadedKostya Serebryanyuintptr_t GetThreadSelf(); 129dde7c33df06a9a0a8056f2357d764a10512206eeKostya Serebryanyint AtomicInc(int *a); 130cc4e6862c6a8f8f3ead96bd32b815184a36fadedKostya Serebryany 131cc4e6862c6a8f8f3ead96bd32b815184a36fadedKostya Serebryany// Wrapper for TLS/TSD. 132cc4e6862c6a8f8f3ead96bd32b815184a36fadedKostya Serebryanyvoid AsanTSDInit(); 133cc4e6862c6a8f8f3ead96bd32b815184a36fadedKostya Serebryanyvoid *AsanTSDGet(); 134cc4e6862c6a8f8f3ead96bd32b815184a36fadedKostya Serebryanyvoid AsanTSDSet(void *tsd); 1354803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany 136df499b44de81fc757a789878f07fcaf19ebb0016Kostya Serebryany// Opens the file 'file_name" and reads up to 'max_len' bytes. 137df499b44de81fc757a789878f07fcaf19ebb0016Kostya Serebryany// The resulting buffer is mmaped and stored in '*buff'. 138df499b44de81fc757a789878f07fcaf19ebb0016Kostya Serebryany// The size of the mmaped region is stored in '*buff_size', 1390ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryany// Returns the number of read bytes or 0 if file can not be opened. 1400ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanysize_t ReadFileToBuffer(const char *file_name, char **buff, 1410ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryany size_t *buff_size, size_t max_len); 142df499b44de81fc757a789878f07fcaf19ebb0016Kostya Serebryany 143218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// asan_printf.cc 1441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid RawWrite(const char *buffer); 1451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint SNPrint(char *buffer, size_t length, const char *format, ...); 1461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid Printf(const char *format, ...); 147df499b44de81fc757a789878f07fcaf19ebb0016Kostya Serebryanyint SScanf(const char *str, const char *format, ...); 1481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid Report(const char *format, ...); 1491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1502d8b3bdb112ebb8ed3f15ee41d4cebcd683b41b0Kostya Serebryany// Don't use std::min and std::max, to minimize dependency on libstdc++. 1512d8b3bdb112ebb8ed3f15ee41d4cebcd683b41b0Kostya Serebryanytemplate<class T> T Min(T a, T b) { return a < b ? a : b; } 1522d8b3bdb112ebb8ed3f15ee41d4cebcd683b41b0Kostya Serebryanytemplate<class T> T Max(T a, T b) { return a > b ? a : b; } 1532d8b3bdb112ebb8ed3f15ee41d4cebcd683b41b0Kostya Serebryany 154218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// asan_poisoning.cc 155218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// Poisons the shadow memory for "size" bytes starting from "addr". 156218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryanyvoid PoisonShadow(uintptr_t addr, size_t size, uint8_t value); 157218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// Poisons the shadow memory for "redzone_size" bytes starting from 158218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// "addr + size". 159218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryanyvoid PoisonShadowPartialRightRedzone(uintptr_t addr, 160218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany uintptr_t size, 161218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany uintptr_t redzone_size, 162218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany uint8_t value); 163218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany 1641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern size_t FLAG_quarantine_size; 1651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int FLAG_demangle; 1661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_symbolize; 1671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int FLAG_v; 1681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern size_t FLAG_redzone; 1691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int FLAG_debug; 1701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_poison_shadow; 1711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int FLAG_report_globals; 1721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern size_t FLAG_malloc_context_size; 1731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_replace_str; 1741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_replace_intrin; 1751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_replace_cfallocator; 1761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_fast_unwind; 1771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_use_fake_stack; 1781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern size_t FLAG_max_malloc_fill_size; 1791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int FLAG_exitcode; 1801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_allow_user_poisoning; 181cb00d134727c322e2f26298912f77e10be46aefeKostya Serebryanyextern int FLAG_sleep_before_dying; 1824803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryanyextern bool FLAG_handle_segv; 1831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int asan_inited; 1851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Used to avoid infinite recursion in __asan_init(). 1861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool asan_init_is_running; 1871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyenum LinkerInitialized { LINKER_INITIALIZED = 0 }; 1891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1900ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanyvoid AsanDie(); 1911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define CHECK(cond) do { if (!(cond)) { \ 1931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany CheckFailed(#cond, __FILE__, __LINE__); \ 1941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}}while(0) 1951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define RAW_CHECK_MSG(expr, msg) do { \ 1971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (!(expr)) { \ 1981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany RawWrite(msg); \ 1990ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryany AsanDie(); \ 2001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } \ 2011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} while (0) 2021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define RAW_CHECK(expr) RAW_CHECK_MSG(expr, #expr) 2041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define UNIMPLEMENTED() CHECK("unimplemented" && 0) 2061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define ASAN_ARRAY_SIZE(a) (sizeof(a)/sizeof((a)[0])) 2081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst size_t kWordSize = __WORDSIZE / 8; 2101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst size_t kWordSizeInBits = 8 * kWordSize; 2111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst size_t kPageSizeBits = 12; 2121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst size_t kPageSize = 1UL << kPageSizeBits; 2131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2146f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#ifndef _WIN32 2151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define GET_CALLER_PC() (uintptr_t)__builtin_return_address(0) 2161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define GET_CURRENT_FRAME() (uintptr_t)__builtin_frame_address(0) 2176f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#else 2186f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko// TODO(timurrrr): implement. 2196f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#define GET_CALLER_PC() (uintptr_t)0 2206f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#define GET_CURRENT_FRAME() (uintptr_t)0 2216f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#endif 2221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define GET_BP_PC_SP \ 2241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uintptr_t bp = GET_CURRENT_FRAME(); \ 2251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uintptr_t pc = GET_CALLER_PC(); \ 2261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uintptr_t local_stack; \ 2271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uintptr_t sp = (uintptr_t)&local_stack; 2281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// These magic values are written to shadow for better error reporting. 2301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanHeapLeftRedzoneMagic = 0xfa; 2311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanHeapRightRedzoneMagic = 0xfb; 2321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanHeapFreeMagic = 0xfd; 2331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanStackLeftRedzoneMagic = 0xf1; 2341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanStackMidRedzoneMagic = 0xf2; 2351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanStackRightRedzoneMagic = 0xf3; 2361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanStackPartialRedzoneMagic = 0xf4; 2371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanStackAfterReturnMagic = 0xf5; 2381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanUserPoisonedMemoryMagic = 0xf7; 2391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanGlobalRedzoneMagic = 0xf9; 2406b30e2cf0d9d471d276f91ef4bb74dbd9876f4d9Kostya Serebryanyconst int kAsanInternalHeapMagic = 0xfe; 2411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic const uintptr_t kCurrentStackFrameMagic = 0x41B58AB3; 2431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic const uintptr_t kRetiredStackFrameMagic = 0x45E0360E; 2441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 245de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany// --------------------------- Bit twiddling ------- {{{1 246de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyinline bool IsPowerOfTwo(size_t x) { 247de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return (x & (x - 1)) == 0; 248de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 249de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 250de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyinline size_t RoundUpTo(size_t size, size_t boundary) { 251de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany CHECK(IsPowerOfTwo(boundary)); 252de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return (size + boundary - 1) & ~(boundary - 1); 253de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 254de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 255b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany// -------------------------- LowLevelAllocator ----- {{{1 256b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany// A simple low-level memory allocator for internal use. 257b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryanyclass LowLevelAllocator { 258b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany public: 259b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany explicit LowLevelAllocator(LinkerInitialized) {} 260b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany // 'size' must be a power of two. 261b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany // Requires an external lock. 262b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany void *Allocate(size_t size); 263b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany private: 264b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany char *allocated_end_; 265b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany char *allocated_current_; 266b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany}; 267b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany 2681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} // namespace __asan 2691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif // ASAN_INTERNAL_H 271