Alexey Samsonov | bc3a7e3 | 2012-06-06 06:47:26 +0000 | [diff] [blame] | 1 | //===-- sanitizer_common.h --------------------------------------*- C++ -*-===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file is shared between AddressSanitizer and ThreadSanitizer |
| 11 | // run-time libraries. |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 12 | // It declares common functions and classes that are used in both runtimes. |
Alexey Samsonov | bc3a7e3 | 2012-06-06 06:47:26 +0000 | [diff] [blame] | 13 | // Implementation of some functions are provided in sanitizer_common, while |
| 14 | // others must be defined by run-time library itself. |
| 15 | //===----------------------------------------------------------------------===// |
| 16 | #ifndef SANITIZER_COMMON_H |
| 17 | #define SANITIZER_COMMON_H |
| 18 | |
| 19 | #include "sanitizer_internal_defs.h" |
| 20 | |
| 21 | namespace __sanitizer { |
| 22 | |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 23 | // Constants. |
| 24 | const uptr kWordSize = __WORDSIZE / 8; |
| 25 | const uptr kWordSizeInBits = 8 * kWordSize; |
| 26 | const uptr kPageSizeBits = 12; |
| 27 | const uptr kPageSize = 1UL << kPageSizeBits; |
Alexey Samsonov | 40e5128 | 2012-06-15 07:29:14 +0000 | [diff] [blame] | 28 | #ifndef _WIN32 |
| 29 | const uptr kMmapGranularity = kPageSize; |
| 30 | #else |
| 31 | const uptr kMmapGranularity = 1UL << 16; |
| 32 | #endif |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 33 | |
Alexey Samsonov | 4b1f103 | 2012-06-07 07:13:46 +0000 | [diff] [blame] | 34 | // Threads |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 35 | int GetPid(); |
Alexey Samsonov | 70afb91 | 2012-06-15 06:37:34 +0000 | [diff] [blame] | 36 | uptr GetThreadSelf(); |
Alexey Samsonov | cf4d3a0 | 2012-06-07 07:32:00 +0000 | [diff] [blame] | 37 | void GetThreadStackTopAndBottom(bool at_initialization, uptr *stack_top, |
Alexey Samsonov | 4b1f103 | 2012-06-07 07:13:46 +0000 | [diff] [blame] | 38 | uptr *stack_bottom); |
Alexey Samsonov | 40d5b77 | 2012-06-06 16:15:07 +0000 | [diff] [blame] | 39 | |
| 40 | // Memory management |
| 41 | void *MmapOrDie(uptr size, const char *mem_type); |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 42 | void UnmapOrDie(void *addr, uptr size); |
Alexey Samsonov | c70d108 | 2012-06-14 14:42:58 +0000 | [diff] [blame] | 43 | void *MmapFixedNoReserve(uptr fixed_addr, uptr size); |
| 44 | void *Mprotect(uptr fixed_addr, uptr size); |
Alexey Samsonov | 40e5128 | 2012-06-15 07:29:14 +0000 | [diff] [blame] | 45 | // Used to check if we can map shadow memory to a fixed location. |
| 46 | bool MemoryRangeIsAvailable(uptr range_start, uptr range_end); |
Alexey Samsonov | c70d108 | 2012-06-14 14:42:58 +0000 | [diff] [blame] | 47 | |
| 48 | // Internal allocator |
Alexey Samsonov | 3a6ddb8 | 2012-06-07 08:52:56 +0000 | [diff] [blame] | 49 | void *InternalAlloc(uptr size); |
| 50 | void InternalFree(void *addr); |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 51 | |
Alexey Samsonov | c70d108 | 2012-06-14 14:42:58 +0000 | [diff] [blame] | 52 | // IO |
Alexey Samsonov | 4b1f103 | 2012-06-07 07:13:46 +0000 | [diff] [blame] | 53 | void RawWrite(const char *buffer); |
Alexey Samsonov | d323f4e | 2012-06-06 13:58:39 +0000 | [diff] [blame] | 54 | void Printf(const char *format, ...); |
| 55 | int SNPrintf(char *buffer, uptr length, const char *format, ...); |
| 56 | void Report(const char *format, ...); |
Alexey Samsonov | 51ae983 | 2012-06-06 13:11:29 +0000 | [diff] [blame] | 57 | |
Alexey Samsonov | fe44fbd | 2012-06-07 05:38:26 +0000 | [diff] [blame] | 58 | // Opens the file 'file_name" and reads up to 'max_len' bytes. |
| 59 | // The resulting buffer is mmaped and stored in '*buff'. |
| 60 | // The size of the mmaped region is stored in '*buff_size', |
| 61 | // Returns the number of read bytes or 0 if file can not be opened. |
| 62 | uptr ReadFileToBuffer(const char *file_name, char **buff, |
| 63 | uptr *buff_size, uptr max_len); |
Alexey Samsonov | 0c53a38 | 2012-06-14 14:07:21 +0000 | [diff] [blame] | 64 | const char *GetEnv(const char *name); |
Alexey Samsonov | fe44fbd | 2012-06-07 05:38:26 +0000 | [diff] [blame] | 65 | |
Alexey Samsonov | ae1e171 | 2012-06-15 06:08:19 +0000 | [diff] [blame] | 66 | // Other |
| 67 | void DisableCoreDumper(); |
| 68 | void DumpProcessMap(); |
Alexey Samsonov | 70afb91 | 2012-06-15 06:37:34 +0000 | [diff] [blame] | 69 | void SleepForSeconds(int seconds); |
| 70 | void NORETURN Exit(int exitcode); |
| 71 | void NORETURN Abort(); |
| 72 | int Atexit(void (*function)(void)); |
Alexey Samsonov | e4a8898 | 2012-06-15 07:00:31 +0000 | [diff] [blame] | 73 | void SortArray(uptr *array, uptr size); |
Alexey Samsonov | ae1e171 | 2012-06-15 06:08:19 +0000 | [diff] [blame] | 74 | |
Alexey Samsonov | e4a8898 | 2012-06-15 07:00:31 +0000 | [diff] [blame] | 75 | // Atomics |
| 76 | int AtomicInc(int *a); |
| 77 | u16 AtomicExchange(u16 *a, u16 new_val); |
| 78 | u8 AtomicExchange(u8 *a, u8 new_val); |
| 79 | |
| 80 | // Math |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 81 | inline bool IsPowerOfTwo(uptr x) { |
| 82 | return (x & (x - 1)) == 0; |
| 83 | } |
| 84 | inline uptr RoundUpTo(uptr size, uptr boundary) { |
Alexey Samsonov | e4a8898 | 2012-06-15 07:00:31 +0000 | [diff] [blame] | 85 | CHECK(IsPowerOfTwo(boundary)); |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 86 | return (size + boundary - 1) & ~(boundary - 1); |
| 87 | } |
Alexey Samsonov | e4a8898 | 2012-06-15 07:00:31 +0000 | [diff] [blame] | 88 | // Don't use std::min and std::max, to minimize dependency on libstdc++. |
| 89 | template<class T> T Min(T a, T b) { return a < b ? a : b; } |
| 90 | template<class T> T Max(T a, T b) { return a > b ? a : b; } |
Alexey Samsonov | bc3a7e3 | 2012-06-06 06:47:26 +0000 | [diff] [blame] | 91 | |
Alexey Samsonov | 156958d | 2012-06-15 13:09:52 +0000 | [diff] [blame^] | 92 | // Char handling |
| 93 | inline bool IsSpace(int c) { |
| 94 | return (c == ' ') || (c == '\n') || (c == '\t') || |
| 95 | (c == '\f') || (c == '\r') || (c == '\v'); |
| 96 | } |
| 97 | inline bool IsDigit(int c) { |
| 98 | return (c >= '0') && (c <= '9'); |
| 99 | } |
| 100 | inline int ToLower(int c) { |
| 101 | return (c >= 'A' && c <= 'Z') ? (c + 'a' - 'A') : c; |
| 102 | } |
| 103 | |
Kostya Serebryany | c0bbfbf | 2012-06-06 16:33:46 +0000 | [diff] [blame] | 104 | #if __WORDSIZE == 64 |
| 105 | # define FIRST_32_SECOND_64(a, b) (b) |
| 106 | #else |
| 107 | # define FIRST_32_SECOND_64(a, b) (a) |
| 108 | #endif |
| 109 | |
Alexey Samsonov | bc3a7e3 | 2012-06-06 06:47:26 +0000 | [diff] [blame] | 110 | } // namespace __sanitizer |
| 111 | |
| 112 | #endif // SANITIZER_COMMON_H |