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; |
Kostya Serebryany | 5b01415 | 2012-06-22 13:00:50 +0000 | [diff] [blame] | 28 | const uptr kCacheLineSize = 64; |
Alexey Samsonov | 40e5128 | 2012-06-15 07:29:14 +0000 | [diff] [blame] | 29 | #ifndef _WIN32 |
| 30 | const uptr kMmapGranularity = kPageSize; |
| 31 | #else |
| 32 | const uptr kMmapGranularity = 1UL << 16; |
| 33 | #endif |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 34 | |
Alexey Samsonov | 4b1f103 | 2012-06-07 07:13:46 +0000 | [diff] [blame] | 35 | // Threads |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 36 | int GetPid(); |
Alexey Samsonov | 70afb91 | 2012-06-15 06:37:34 +0000 | [diff] [blame] | 37 | uptr GetThreadSelf(); |
Alexey Samsonov | cf4d3a0 | 2012-06-07 07:32:00 +0000 | [diff] [blame] | 38 | void GetThreadStackTopAndBottom(bool at_initialization, uptr *stack_top, |
Alexey Samsonov | 4b1f103 | 2012-06-07 07:13:46 +0000 | [diff] [blame] | 39 | uptr *stack_bottom); |
Alexey Samsonov | 40d5b77 | 2012-06-06 16:15:07 +0000 | [diff] [blame] | 40 | |
| 41 | // Memory management |
| 42 | void *MmapOrDie(uptr size, const char *mem_type); |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 43 | void UnmapOrDie(void *addr, uptr size); |
Alexey Samsonov | c70d108 | 2012-06-14 14:42:58 +0000 | [diff] [blame] | 44 | void *MmapFixedNoReserve(uptr fixed_addr, uptr size); |
| 45 | void *Mprotect(uptr fixed_addr, uptr size); |
Alexey Samsonov | 40e5128 | 2012-06-15 07:29:14 +0000 | [diff] [blame] | 46 | // Used to check if we can map shadow memory to a fixed location. |
| 47 | bool MemoryRangeIsAvailable(uptr range_start, uptr range_end); |
Alexey Samsonov | c70d108 | 2012-06-14 14:42:58 +0000 | [diff] [blame] | 48 | |
| 49 | // Internal allocator |
Alexey Samsonov | 3a6ddb8 | 2012-06-07 08:52:56 +0000 | [diff] [blame] | 50 | void *InternalAlloc(uptr size); |
Dmitry Vyukov | c598de9 | 2012-06-25 15:09:24 +0000 | [diff] [blame] | 51 | void InternalFree(void *p); |
| 52 | // Given the pointer p into a valid allocated block, |
| 53 | // returns a pointer to the beginning of the block. |
| 54 | void *InternalAllocBlock(void *p); |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 55 | |
Alexey Samsonov | c70d108 | 2012-06-14 14:42:58 +0000 | [diff] [blame] | 56 | // IO |
Alexey Samsonov | 4b1f103 | 2012-06-07 07:13:46 +0000 | [diff] [blame] | 57 | void RawWrite(const char *buffer); |
Alexey Samsonov | d323f4e | 2012-06-06 13:58:39 +0000 | [diff] [blame] | 58 | void Printf(const char *format, ...); |
Alexey Samsonov | d323f4e | 2012-06-06 13:58:39 +0000 | [diff] [blame] | 59 | void Report(const char *format, ...); |
Alexey Samsonov | 51ae983 | 2012-06-06 13:11:29 +0000 | [diff] [blame] | 60 | |
Alexey Samsonov | fe44fbd | 2012-06-07 05:38:26 +0000 | [diff] [blame] | 61 | // Opens the file 'file_name" and reads up to 'max_len' bytes. |
| 62 | // The resulting buffer is mmaped and stored in '*buff'. |
| 63 | // The size of the mmaped region is stored in '*buff_size', |
| 64 | // Returns the number of read bytes or 0 if file can not be opened. |
| 65 | uptr ReadFileToBuffer(const char *file_name, char **buff, |
| 66 | uptr *buff_size, uptr max_len); |
Alexey Samsonov | 961276a | 2012-07-03 08:24:14 +0000 | [diff] [blame] | 67 | // Maps given file to virtual memory, and returns pointer to it |
| 68 | // (or NULL if the mapping failes). Stores the size of mmaped region |
| 69 | // in '*buff_size'. |
| 70 | void *MapFileToMemory(const char *file_name, uptr *buff_size); |
| 71 | |
Alexey Samsonov | 0c53a38 | 2012-06-14 14:07:21 +0000 | [diff] [blame] | 72 | const char *GetEnv(const char *name); |
Alexey Samsonov | 58a3c58 | 2012-06-18 08:44:30 +0000 | [diff] [blame] | 73 | const char *GetPwd(); |
Alexey Samsonov | fe44fbd | 2012-06-07 05:38:26 +0000 | [diff] [blame] | 74 | |
Alexey Samsonov | ae1e171 | 2012-06-15 06:08:19 +0000 | [diff] [blame] | 75 | // Other |
| 76 | void DisableCoreDumper(); |
| 77 | void DumpProcessMap(); |
Alexey Samsonov | 70afb91 | 2012-06-15 06:37:34 +0000 | [diff] [blame] | 78 | void SleepForSeconds(int seconds); |
Alexey Samsonov | 58a3c58 | 2012-06-18 08:44:30 +0000 | [diff] [blame] | 79 | void SleepForMillis(int millis); |
Alexey Samsonov | 70afb91 | 2012-06-15 06:37:34 +0000 | [diff] [blame] | 80 | void NORETURN Exit(int exitcode); |
| 81 | void NORETURN Abort(); |
| 82 | int Atexit(void (*function)(void)); |
Alexey Samsonov | e4a8898 | 2012-06-15 07:00:31 +0000 | [diff] [blame] | 83 | void SortArray(uptr *array, uptr size); |
Alexey Samsonov | ae1e171 | 2012-06-15 06:08:19 +0000 | [diff] [blame] | 84 | |
Alexey Samsonov | e4a8898 | 2012-06-15 07:00:31 +0000 | [diff] [blame] | 85 | // Math |
Dmitry Vyukov | 03d32ec | 2012-07-05 16:18:28 +0000 | [diff] [blame^] | 86 | INLINE bool IsPowerOfTwo(uptr x) { |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 87 | return (x & (x - 1)) == 0; |
| 88 | } |
Dmitry Vyukov | 03d32ec | 2012-07-05 16:18:28 +0000 | [diff] [blame^] | 89 | INLINE uptr RoundUpTo(uptr size, uptr boundary) { |
Alexey Samsonov | e4a8898 | 2012-06-15 07:00:31 +0000 | [diff] [blame] | 90 | CHECK(IsPowerOfTwo(boundary)); |
Alexey Samsonov | ee07290 | 2012-06-06 09:26:25 +0000 | [diff] [blame] | 91 | return (size + boundary - 1) & ~(boundary - 1); |
| 92 | } |
Alexey Samsonov | e4a8898 | 2012-06-15 07:00:31 +0000 | [diff] [blame] | 93 | // Don't use std::min and std::max, to minimize dependency on libstdc++. |
| 94 | template<class T> T Min(T a, T b) { return a < b ? a : b; } |
| 95 | 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] | 96 | |
Alexey Samsonov | 156958d | 2012-06-15 13:09:52 +0000 | [diff] [blame] | 97 | // Char handling |
Dmitry Vyukov | 03d32ec | 2012-07-05 16:18:28 +0000 | [diff] [blame^] | 98 | INLINE bool IsSpace(int c) { |
Alexey Samsonov | 156958d | 2012-06-15 13:09:52 +0000 | [diff] [blame] | 99 | return (c == ' ') || (c == '\n') || (c == '\t') || |
| 100 | (c == '\f') || (c == '\r') || (c == '\v'); |
| 101 | } |
Dmitry Vyukov | 03d32ec | 2012-07-05 16:18:28 +0000 | [diff] [blame^] | 102 | INLINE bool IsDigit(int c) { |
Alexey Samsonov | 156958d | 2012-06-15 13:09:52 +0000 | [diff] [blame] | 103 | return (c >= '0') && (c <= '9'); |
| 104 | } |
Dmitry Vyukov | 03d32ec | 2012-07-05 16:18:28 +0000 | [diff] [blame^] | 105 | INLINE int ToLower(int c) { |
Alexey Samsonov | 156958d | 2012-06-15 13:09:52 +0000 | [diff] [blame] | 106 | return (c >= 'A' && c <= 'Z') ? (c + 'a' - 'A') : c; |
| 107 | } |
| 108 | |
Kostya Serebryany | c0bbfbf | 2012-06-06 16:33:46 +0000 | [diff] [blame] | 109 | #if __WORDSIZE == 64 |
| 110 | # define FIRST_32_SECOND_64(a, b) (b) |
| 111 | #else |
| 112 | # define FIRST_32_SECOND_64(a, b) (a) |
| 113 | #endif |
| 114 | |
Alexey Samsonov | bc3a7e3 | 2012-06-06 06:47:26 +0000 | [diff] [blame] | 115 | } // namespace __sanitizer |
| 116 | |
| 117 | #endif // SANITIZER_COMMON_H |