asan_internal.h revision cb00d134727c322e2f26298912f77e10be46aefe
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; 336f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#else 34858220837f85db7eb5eedd5ea68603803b5eb3c9Kostya Serebryany# include <stdint.h> // for __WORDSIZE 356f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#endif // _WIN32 361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 374616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar// If __WORDSIZE was undefined by the platform, define it in terms of the 384616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar// compiler built-in __LP64__. 394616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#ifndef __WORDSIZE 404616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#if __LP64__ 414616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#define __WORDSIZE 64 424616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#else 434616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#define __WORDSIZE 32 444616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#endif 454616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar#endif 464616633bb9e20f66d83bfac70c9afce1922d1eecDaniel Dunbar 47c8365231004cb1d956aba4164c89ea1398eadd6bAlexander Potapenko#if !defined(__has_feature) 48c8365231004cb1d956aba4164c89ea1398eadd6bAlexander Potapenko#define __has_feature(x) 0 49c8365231004cb1d956aba4164c89ea1398eadd6bAlexander Potapenko#endif 50c8365231004cb1d956aba4164c89ea1398eadd6bAlexander Potapenko 5113ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany#if defined(__has_feature) && __has_feature(address_sanitizer) 521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany# error "The AddressSanitizer run-time should not be" 531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany " instrumented by AddressSanitizer" 541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 56c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany// Build-time configuration options. 57c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany 58c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany// If set, asan will install its own SEGV signal handler. 59c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany#ifndef ASAN_NEEDS_SEGV 60c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany# define ASAN_NEEDS_SEGV 1 61c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany#endif 62c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany 63c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany// If set, asan will intercept C++ exception api call(s). 64c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany#ifndef ASAN_HAS_EXCEPTIONS 65c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany# define ASAN_HAS_EXCEPTIONS 1 66c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany#endif 67c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany 68c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany// If set, asan uses the values of SHADOW_SCALE and SHADOW_OFFSET 69c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany// provided by the instrumented objects. Otherwise constants are used. 70c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany#ifndef ASAN_FLEXIBLE_MAPPING_AND_OFFSET 71c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany# define ASAN_FLEXIBLE_MAPPING_AND_OFFSET 0 72c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany#endif 73c6f2223a8772262e5e682403f2d57f0b465a98fcKostya Serebryany 741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// All internal functions in asan reside inside the __asan namespace 751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// to avoid namespace collisions with the user programs. 761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Seperate namespace also makes it simpler to distinguish the asan run-time 771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// functions from the instrumented user code in a profile. 781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanynamespace __asan { 791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyclass AsanThread; 811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystruct AsanStackTrace; 821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 83218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// asan_rtl.cc 841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid CheckFailed(const char *cond, const char *file, int line); 851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid ShowStatsAndAbort(); 861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 87218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// asan_globals.cc 881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanybool DescribeAddrIfGlobal(uintptr_t addr); 891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 90218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// asan_malloc_linux.cc / asan_malloc_mac.cc 911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid ReplaceSystemMalloc(); 921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 93de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid OutOfMemoryMessageAndDie(const char *mem_type, size_t size); 94de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 95218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// asan_linux.cc / asan_mac.cc 961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *AsanDoesNotSupportStaticLinkage(); 97de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyint AsanOpenReadonly(const char* filename); 981e316d7f488a75312539629e9d937e156280eeb6Alexander Potapenkoconst char *AsanGetEnv(const char *name); 99de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 100a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyvoid *AsanMmapFixedNoReserve(uintptr_t fixed_addr, size_t size); 101a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyvoid *AsanMmapFixedReserve(uintptr_t fixed_addr, size_t size); 102a874fe5b5d67152e4e737498d532eec80940bdcdKostya Serebryanyvoid *AsanMprotect(uintptr_t fixed_addr, size_t size); 103de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid *AsanMmapSomewhereOrDie(size_t size, const char *where); 104de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyvoid AsanUnmapOrDie(void *ptr, size_t size); 105de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 106ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryanyvoid AsanDisableCoreDumper(); 1079107c26bd88fc9cf44a2cd7d6967eb830ac63be3Kostya Serebryanyvoid GetPcSpBp(void *context, uintptr_t *pc, uintptr_t *sp, uintptr_t *bp); 108ef14ff6512d7b2e20aa3206dff820b5f90285420Kostya Serebryany 1090ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanysize_t AsanRead(int fd, void *buf, size_t count); 1100ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanysize_t AsanWrite(int fd, const void *buf, size_t count); 111de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyint AsanClose(int fd); 1121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1134803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryanybool AsanInterceptsSignal(int signum); 114a7e760a53bc43b8e09bfdf5cd6f215267ba99729Kostya Serebryanyvoid InstallSignalHandlers(); 1150ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanyint GetPid(); 116cc4e6862c6a8f8f3ead96bd32b815184a36fadedKostya Serebryanyuintptr_t GetThreadSelf(); 117dde7c33df06a9a0a8056f2357d764a10512206eeKostya Serebryanyint AtomicInc(int *a); 118cc4e6862c6a8f8f3ead96bd32b815184a36fadedKostya Serebryany 119cc4e6862c6a8f8f3ead96bd32b815184a36fadedKostya Serebryany// Wrapper for TLS/TSD. 120cc4e6862c6a8f8f3ead96bd32b815184a36fadedKostya Serebryanyvoid AsanTSDInit(); 121cc4e6862c6a8f8f3ead96bd32b815184a36fadedKostya Serebryanyvoid *AsanTSDGet(); 122cc4e6862c6a8f8f3ead96bd32b815184a36fadedKostya Serebryanyvoid AsanTSDSet(void *tsd); 1234803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany 124df499b44de81fc757a789878f07fcaf19ebb0016Kostya Serebryany// Opens the file 'file_name" and reads up to 'max_len' bytes. 125df499b44de81fc757a789878f07fcaf19ebb0016Kostya Serebryany// The resulting buffer is mmaped and stored in '*buff'. 126df499b44de81fc757a789878f07fcaf19ebb0016Kostya Serebryany// The size of the mmaped region is stored in '*buff_size', 1270ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryany// Returns the number of read bytes or 0 if file can not be opened. 1280ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanysize_t ReadFileToBuffer(const char *file_name, char **buff, 1290ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryany size_t *buff_size, size_t max_len); 130df499b44de81fc757a789878f07fcaf19ebb0016Kostya Serebryany 131218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// asan_printf.cc 1321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid RawWrite(const char *buffer); 1331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint SNPrint(char *buffer, size_t length, const char *format, ...); 1341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid Printf(const char *format, ...); 135df499b44de81fc757a789878f07fcaf19ebb0016Kostya Serebryanyint SScanf(const char *str, const char *format, ...); 1361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid Report(const char *format, ...); 1371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1382d8b3bdb112ebb8ed3f15ee41d4cebcd683b41b0Kostya Serebryany// Don't use std::min and std::max, to minimize dependency on libstdc++. 1392d8b3bdb112ebb8ed3f15ee41d4cebcd683b41b0Kostya Serebryanytemplate<class T> T Min(T a, T b) { return a < b ? a : b; } 1402d8b3bdb112ebb8ed3f15ee41d4cebcd683b41b0Kostya Serebryanytemplate<class T> T Max(T a, T b) { return a > b ? a : b; } 1412d8b3bdb112ebb8ed3f15ee41d4cebcd683b41b0Kostya Serebryany 142218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// asan_poisoning.cc 143218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// Poisons the shadow memory for "size" bytes starting from "addr". 144218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryanyvoid PoisonShadow(uintptr_t addr, size_t size, uint8_t value); 145218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// Poisons the shadow memory for "redzone_size" bytes starting from 146218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany// "addr + size". 147218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryanyvoid PoisonShadowPartialRightRedzone(uintptr_t addr, 148218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany uintptr_t size, 149218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany uintptr_t redzone_size, 150218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany uint8_t value); 151218a9b70d7338cf5b727b7dad6b080ad7869c6c2Kostya Serebryany 1521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern size_t FLAG_quarantine_size; 1531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int FLAG_demangle; 1541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_symbolize; 1551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int FLAG_v; 1561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern size_t FLAG_redzone; 1571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int FLAG_debug; 1581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_poison_shadow; 1591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int FLAG_report_globals; 1601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern size_t FLAG_malloc_context_size; 1611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_replace_str; 1621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_replace_intrin; 1631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_replace_cfallocator; 1641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_fast_unwind; 1651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_use_fake_stack; 1661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern size_t FLAG_max_malloc_fill_size; 1671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int FLAG_exitcode; 1681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool FLAG_allow_user_poisoning; 169cb00d134727c322e2f26298912f77e10be46aefeKostya Serebryanyextern int FLAG_sleep_before_dying; 1704803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryanyextern bool FLAG_handle_segv; 1711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int asan_inited; 1731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Used to avoid infinite recursion in __asan_init(). 1741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern bool asan_init_is_running; 1751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyenum LinkerInitialized { LINKER_INITIALIZED = 0 }; 1771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1780ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryanyvoid AsanDie(); 1791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define CHECK(cond) do { if (!(cond)) { \ 1811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany CheckFailed(#cond, __FILE__, __LINE__); \ 1821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}}while(0) 1831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define RAW_CHECK_MSG(expr, msg) do { \ 1851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (!(expr)) { \ 1861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany RawWrite(msg); \ 1870ecf5eb729dd81a43f8585cb438d3cb2a35899edKostya Serebryany AsanDie(); \ 1881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } \ 1891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} while (0) 1901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define RAW_CHECK(expr) RAW_CHECK_MSG(expr, #expr) 1921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define UNIMPLEMENTED() CHECK("unimplemented" && 0) 1941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define ASAN_ARRAY_SIZE(a) (sizeof(a)/sizeof((a)[0])) 1961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst size_t kWordSize = __WORDSIZE / 8; 1981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst size_t kWordSizeInBits = 8 * kWordSize; 1991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst size_t kPageSizeBits = 12; 2001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst size_t kPageSize = 1UL << kPageSizeBits; 2011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2026f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#ifndef _WIN32 2031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define GET_CALLER_PC() (uintptr_t)__builtin_return_address(0) 2041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define GET_CURRENT_FRAME() (uintptr_t)__builtin_frame_address(0) 2056f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#else 2066f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko// TODO(timurrrr): implement. 2076f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#define GET_CALLER_PC() (uintptr_t)0 2086f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#define GET_CURRENT_FRAME() (uintptr_t)0 2096f0452914ec76c786eb865983793bc03b00fc7b6Alexander Potapenko#endif 2101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#define GET_BP_PC_SP \ 2121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uintptr_t bp = GET_CURRENT_FRAME(); \ 2131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uintptr_t pc = GET_CALLER_PC(); \ 2141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uintptr_t local_stack; \ 2151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uintptr_t sp = (uintptr_t)&local_stack; 2161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// These magic values are written to shadow for better error reporting. 2181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanHeapLeftRedzoneMagic = 0xfa; 2191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanHeapRightRedzoneMagic = 0xfb; 2201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanHeapFreeMagic = 0xfd; 2211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanStackLeftRedzoneMagic = 0xf1; 2221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanStackMidRedzoneMagic = 0xf2; 2231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanStackRightRedzoneMagic = 0xf3; 2241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanStackPartialRedzoneMagic = 0xf4; 2251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanStackAfterReturnMagic = 0xf5; 2261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanUserPoisonedMemoryMagic = 0xf7; 2271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst int kAsanGlobalRedzoneMagic = 0xf9; 2286b30e2cf0d9d471d276f91ef4bb74dbd9876f4d9Kostya Serebryanyconst int kAsanInternalHeapMagic = 0xfe; 2291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic const uintptr_t kCurrentStackFrameMagic = 0x41B58AB3; 2311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic const uintptr_t kRetiredStackFrameMagic = 0x45E0360E; 2321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 233de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany// --------------------------- Bit twiddling ------- {{{1 234de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyinline bool IsPowerOfTwo(size_t x) { 235de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return (x & (x - 1)) == 0; 236de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 237de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 238de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryanyinline size_t RoundUpTo(size_t size, size_t boundary) { 239de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany CHECK(IsPowerOfTwo(boundary)); 240de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany return (size + boundary - 1) & ~(boundary - 1); 241de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany} 242de496f451bce322b6cde100456591f1f50ab3477Kostya Serebryany 243b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany// -------------------------- LowLevelAllocator ----- {{{1 244b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany// A simple low-level memory allocator for internal use. 245b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryanyclass LowLevelAllocator { 246b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany public: 247b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany explicit LowLevelAllocator(LinkerInitialized) {} 248b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany // 'size' must be a power of two. 249b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany // Requires an external lock. 250b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany void *Allocate(size_t size); 251b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany private: 252b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany char *allocated_end_; 253b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany char *allocated_current_; 254b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany}; 255b89567ce41bef82cf92a9c741c78b632c07b2781Kostya Serebryany 2561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} // namespace __asan 2571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif // ASAN_INTERNAL_H 259