contrib/compiler-rt/lib/tsan/rtl/tsan_platform.h

   1 //===-- tsan_platform.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 a part of ThreadSanitizer (TSan), a race detector.
  11 //
  12 // Platform-specific code.
  13 //===----------------------------------------------------------------------===//
  14
  15 #ifndef TSAN_PLATFORM_H
  16 #define TSAN_PLATFORM_H
  17
  18 #if !defined(__LP64__) && !defined(_WIN64)
  19 # error "Only 64-bit is supported"
  20 #endif
  21
  22 #include "tsan_defs.h"
  23 #include "tsan_trace.h"
  24
  25 namespace __tsan {
  26
  27 #if !defined(SANITIZER_GO)
  28
  29 #if defined(__x86_64__)
  30 /*
  31 C/C++ on linux/x86_64 and freebsd/x86_64
  32 0000 0000 1000 - 0100 0000 0000: main binary and/or MAP_32BIT mappings
  33 0100 0000 0000 - 0200 0000 0000: -
  34 0200 0000 0000 - 1000 0000 0000: shadow
  35 1000 0000 0000 - 3000 0000 0000: -
  36 3000 0000 0000 - 4000 0000 0000: metainfo (memory blocks and sync objects)
  37 4000 0000 0000 - 6000 0000 0000: -
  38 6000 0000 0000 - 6200 0000 0000: traces
  39 6200 0000 0000 - 7d00 0000 0000: -
  40 7d00 0000 0000 - 7e00 0000 0000: heap
  41 7e00 0000 0000 - 7e80 0000 0000: -
  42 7e80 0000 0000 - 8000 0000 0000: modules and main thread stack
  43 */
  44 struct Mapping {
  45   static const uptr kMetaShadowBeg = 0x300000000000ull;
  46   static const uptr kMetaShadowEnd = 0x400000000000ull;
  47   static const uptr kTraceMemBeg   = 0x600000000000ull;
  48   static const uptr kTraceMemEnd   = 0x620000000000ull;
  49   static const uptr kShadowBeg     = 0x020000000000ull;
  50   static const uptr kShadowEnd     = 0x100000000000ull;
  51   static const uptr kHeapMemBeg    = 0x7d0000000000ull;
  52   static const uptr kHeapMemEnd    = 0x7e0000000000ull;
  53   static const uptr kLoAppMemBeg   = 0x000000001000ull;
  54   static const uptr kLoAppMemEnd   = 0x010000000000ull;
  55   static const uptr kHiAppMemBeg   = 0x7e8000000000ull;
  56   static const uptr kHiAppMemEnd   = 0x800000000000ull;
  57   static const uptr kAppMemMsk     = 0x7c0000000000ull;
  58   static const uptr kAppMemXor     = 0x020000000000ull;
  59   static const uptr kVdsoBeg       = 0xf000000000000000ull;
  60 };
  61 #elif defined(__mips64)
  62 /*
  63 C/C++ on linux/mips64
  64 0100 0000 00 - 0200 0000 00: main binary
  65 0200 0000 00 - 1400 0000 00: -
  66 1400 0000 00 - 2400 0000 00: shadow
  67 2400 0000 00 - 3000 0000 00: -
  68 3000 0000 00 - 4000 0000 00: metainfo (memory blocks and sync objects)
  69 4000 0000 00 - 6000 0000 00: -
  70 6000 0000 00 - 6200 0000 00: traces
  71 6200 0000 00 - fe00 0000 00: -
  72 fe00 0000 00 - ff00 0000 00: heap
  73 ff00 0000 00 - ff80 0000 00: -
  74 ff80 0000 00 - ffff ffff ff: modules and main thread stack
  75 */
  76 struct Mapping {
  77   static const uptr kMetaShadowBeg = 0x3000000000ull;
  78   static const uptr kMetaShadowEnd = 0x4000000000ull;
  79   static const uptr kTraceMemBeg   = 0x6000000000ull;
  80   static const uptr kTraceMemEnd   = 0x6200000000ull;
  81   static const uptr kShadowBeg     = 0x1400000000ull;
  82   static const uptr kShadowEnd     = 0x2400000000ull;
  83   static const uptr kHeapMemBeg    = 0xfe00000000ull;
  84   static const uptr kHeapMemEnd    = 0xff00000000ull;
  85   static const uptr kLoAppMemBeg   = 0x0100000000ull;
  86   static const uptr kLoAppMemEnd   = 0x0200000000ull;
  87   static const uptr kHiAppMemBeg   = 0xff80000000ull;
  88   static const uptr kHiAppMemEnd   = 0xffffffffffull;
  89   static const uptr kAppMemMsk     = 0xfc00000000ull;
  90   static const uptr kAppMemXor     = 0x0400000000ull;
  91   static const uptr kVdsoBeg       = 0xfffff00000ull;
  92 };
  93 #elif defined(__aarch64__)
  94 // AArch64 supports multiple VMA which leads to multiple address transformation
  95 // functions.  To support these multiple VMAS transformations and mappings TSAN
  96 // runtime for AArch64 uses an external memory read (vmaSize) to select which
  97 // mapping to use.  Although slower, it make a same instrumented binary run on
  98 // multiple kernels.
  99
 100 /*
 101 C/C++ on linux/aarch64 (39-bit VMA)
 102 0000 0010 00 - 0100 0000 00: main binary
 103 0100 0000 00 - 0800 0000 00: -
 104 0800 0000 00 - 2000 0000 00: shadow memory
 105 2000 0000 00 - 3100 0000 00: -
 106 3100 0000 00 - 3400 0000 00: metainfo
 107 3400 0000 00 - 5500 0000 00: -
 108 5500 0000 00 - 5600 0000 00: main binary (PIE)
 109 5600 0000 00 - 6000 0000 00: -
 110 6000 0000 00 - 6200 0000 00: traces
 111 6200 0000 00 - 7d00 0000 00: -
 112 7c00 0000 00 - 7d00 0000 00: heap
 113 7d00 0000 00 - 7fff ffff ff: modules and main thread stack
 114 */
 115 struct Mapping39 {
 116   static const uptr kLoAppMemBeg   = 0x0000001000ull;
 117   static const uptr kLoAppMemEnd   = 0x0100000000ull;
 118   static const uptr kShadowBeg     = 0x0800000000ull;
 119   static const uptr kShadowEnd     = 0x2000000000ull;
 120   static const uptr kMetaShadowBeg = 0x3100000000ull;
 121   static const uptr kMetaShadowEnd = 0x3400000000ull;
 122   static const uptr kMidAppMemBeg  = 0x5500000000ull;
 123   static const uptr kMidAppMemEnd  = 0x5600000000ull;
 124   static const uptr kMidShadowOff  = 0x5000000000ull;
 125   static const uptr kTraceMemBeg   = 0x6000000000ull;
 126   static const uptr kTraceMemEnd   = 0x6200000000ull;
 127   static const uptr kHeapMemBeg    = 0x7c00000000ull;
 128   static const uptr kHeapMemEnd    = 0x7d00000000ull;
 129   static const uptr kHiAppMemBeg   = 0x7e00000000ull;
 130   static const uptr kHiAppMemEnd   = 0x7fffffffffull;
 131   static const uptr kAppMemMsk     = 0x7800000000ull;
 132   static const uptr kAppMemXor     = 0x0200000000ull;
 133   static const uptr kVdsoBeg       = 0x7f00000000ull;
 134 };
 135
 136 /*
 137 C/C++ on linux/aarch64 (42-bit VMA)
 138 00000 0010 00 - 01000 0000 00: main binary
 139 01000 0000 00 - 10000 0000 00: -
 140 10000 0000 00 - 20000 0000 00: shadow memory
 141 20000 0000 00 - 26000 0000 00: -
 142 26000 0000 00 - 28000 0000 00: metainfo
 143 28000 0000 00 - 2aa00 0000 00: -
 144 2aa00 0000 00 - 2ab00 0000 00: main binary (PIE)
 145 2ab00 0000 00 - 36200 0000 00: -
 146 36200 0000 00 - 36240 0000 00: traces
 147 36240 0000 00 - 3e000 0000 00: -
 148 3e000 0000 00 - 3f000 0000 00: heap
 149 3f000 0000 00 - 3ffff ffff ff: modules and main thread stack
 150 */
 151 struct Mapping42 {
 152   static const uptr kLoAppMemBeg   = 0x00000001000ull;
 153   static const uptr kLoAppMemEnd   = 0x01000000000ull;
 154   static const uptr kShadowBeg     = 0x10000000000ull;
 155   static const uptr kShadowEnd     = 0x20000000000ull;
 156   static const uptr kMetaShadowBeg = 0x26000000000ull;
 157   static const uptr kMetaShadowEnd = 0x28000000000ull;
 158   static const uptr kMidAppMemBeg  = 0x2aa00000000ull;
 159   static const uptr kMidAppMemEnd  = 0x2ab00000000ull;
 160   static const uptr kMidShadowOff  = 0x28000000000ull;
 161   static const uptr kTraceMemBeg   = 0x36200000000ull;
 162   static const uptr kTraceMemEnd   = 0x36400000000ull;
 163   static const uptr kHeapMemBeg    = 0x3e000000000ull;
 164   static const uptr kHeapMemEnd    = 0x3f000000000ull;
 165   static const uptr kHiAppMemBeg   = 0x3f000000000ull;
 166   static const uptr kHiAppMemEnd   = 0x3ffffffffffull;
 167   static const uptr kAppMemMsk     = 0x3c000000000ull;
 168   static const uptr kAppMemXor     = 0x04000000000ull;
 169   static const uptr kVdsoBeg       = 0x37f00000000ull;
 170 };
 171
 172 // Indicates the runtime will define the memory regions at runtime.
 173 #define TSAN_RUNTIME_VMA 1
 174 // Indicates that mapping defines a mid range memory segment.
 175 #define TSAN_MID_APP_RANGE 1
 176 #elif defined(__powerpc64__)
 177 // PPC64 supports multiple VMA which leads to multiple address transformation
 178 // functions.  To support these multiple VMAS transformations and mappings TSAN
 179 // runtime for PPC64 uses an external memory read (vmaSize) to select which
 180 // mapping to use.  Although slower, it make a same instrumented binary run on
 181 // multiple kernels.
 182
 183 /*
 184 C/C++ on linux/powerpc64 (44-bit VMA)
 185 0000 0000 0100 - 0001 0000 0000: main binary
 186 0001 0000 0000 - 0001 0000 0000: -
 187 0001 0000 0000 - 0b00 0000 0000: shadow
 188 0b00 0000 0000 - 0b00 0000 0000: -
 189 0b00 0000 0000 - 0d00 0000 0000: metainfo (memory blocks and sync objects)
 190 0d00 0000 0000 - 0d00 0000 0000: -
 191 0d00 0000 0000 - 0f00 0000 0000: traces
 192 0f00 0000 0000 - 0f00 0000 0000: -
 193 0f00 0000 0000 - 0f50 0000 0000: heap
 194 0f50 0000 0000 - 0f60 0000 0000: -
 195 0f60 0000 0000 - 1000 0000 0000: modules and main thread stack
 196 */
 197 struct Mapping44 {
 198   static const uptr kMetaShadowBeg = 0x0b0000000000ull;
 199   static const uptr kMetaShadowEnd = 0x0d0000000000ull;
 200   static const uptr kTraceMemBeg   = 0x0d0000000000ull;
 201   static const uptr kTraceMemEnd   = 0x0f0000000000ull;
 202   static const uptr kShadowBeg     = 0x000100000000ull;
 203   static const uptr kShadowEnd     = 0x0b0000000000ull;
 204   static const uptr kLoAppMemBeg   = 0x000000000100ull;
 205   static const uptr kLoAppMemEnd   = 0x000100000000ull;
 206   static const uptr kHeapMemBeg    = 0x0f0000000000ull;
 207   static const uptr kHeapMemEnd    = 0x0f5000000000ull;
 208   static const uptr kHiAppMemBeg   = 0x0f6000000000ull;
 209   static const uptr kHiAppMemEnd   = 0x100000000000ull; // 44 bits
 210   static const uptr kAppMemMsk     = 0x0f0000000000ull;
 211   static const uptr kAppMemXor     = 0x002100000000ull;
 212   static const uptr kVdsoBeg       = 0x3c0000000000000ull;
 213 };
 214
 215 /*
 216 C/C++ on linux/powerpc64 (46-bit VMA)
 217 0000 0000 1000 - 0100 0000 0000: main binary
 218 0100 0000 0000 - 0200 0000 0000: -
 219 0100 0000 0000 - 1000 0000 0000: shadow
 220 1000 0000 0000 - 1000 0000 0000: -
 221 1000 0000 0000 - 2000 0000 0000: metainfo (memory blocks and sync objects)
 222 2000 0000 0000 - 2000 0000 0000: -
 223 2000 0000 0000 - 2200 0000 0000: traces
 224 2200 0000 0000 - 3d00 0000 0000: -
 225 3d00 0000 0000 - 3e00 0000 0000: heap
 226 3e00 0000 0000 - 3e80 0000 0000: -
 227 3e80 0000 0000 - 4000 0000 0000: modules and main thread stack
 228 */
 229 struct Mapping46 {
 230   static const uptr kMetaShadowBeg = 0x100000000000ull;
 231   static const uptr kMetaShadowEnd = 0x200000000000ull;
 232   static const uptr kTraceMemBeg   = 0x200000000000ull;
 233   static const uptr kTraceMemEnd   = 0x220000000000ull;
 234   static const uptr kShadowBeg     = 0x010000000000ull;
 235   static const uptr kShadowEnd     = 0x100000000000ull;
 236   static const uptr kHeapMemBeg    = 0x3d0000000000ull;
 237   static const uptr kHeapMemEnd    = 0x3e0000000000ull;
 238   static const uptr kLoAppMemBeg   = 0x000000001000ull;
 239   static const uptr kLoAppMemEnd   = 0x010000000000ull;
 240   static const uptr kHiAppMemBeg   = 0x3e8000000000ull;
 241   static const uptr kHiAppMemEnd   = 0x400000000000ull; // 46 bits
 242   static const uptr kAppMemMsk     = 0x3c0000000000ull;
 243   static const uptr kAppMemXor     = 0x020000000000ull;
 244   static const uptr kVdsoBeg       = 0x7800000000000000ull;
 245 };
 246
 247 // Indicates the runtime will define the memory regions at runtime.
 248 #define TSAN_RUNTIME_VMA 1
 249 #endif
 250
 251 #elif defined(SANITIZER_GO) && !SANITIZER_WINDOWS
 252
 253 /* Go on linux, darwin and freebsd
 254 0000 0000 1000 - 0000 1000 0000: executable
 255 0000 1000 0000 - 00c0 0000 0000: -
 256 00c0 0000 0000 - 00e0 0000 0000: heap
 257 00e0 0000 0000 - 2000 0000 0000: -
 258 2000 0000 0000 - 2380 0000 0000: shadow
 259 2380 0000 0000 - 3000 0000 0000: -
 260 3000 0000 0000 - 4000 0000 0000: metainfo (memory blocks and sync objects)
 261 4000 0000 0000 - 6000 0000 0000: -
 262 6000 0000 0000 - 6200 0000 0000: traces
 263 6200 0000 0000 - 8000 0000 0000: -
 264 */
 265
 266 struct Mapping {
 267   static const uptr kMetaShadowBeg = 0x300000000000ull;
 268   static const uptr kMetaShadowEnd = 0x400000000000ull;
 269   static const uptr kTraceMemBeg   = 0x600000000000ull;
 270   static const uptr kTraceMemEnd   = 0x620000000000ull;
 271   static const uptr kShadowBeg     = 0x200000000000ull;
 272   static const uptr kShadowEnd     = 0x238000000000ull;
 273   static const uptr kAppMemBeg     = 0x000000001000ull;
 274   static const uptr kAppMemEnd     = 0x00e000000000ull;
 275 };
 276
 277 #elif defined(SANITIZER_GO) && SANITIZER_WINDOWS
 278
 279 /* Go on windows
 280 0000 0000 1000 - 0000 1000 0000: executable
 281 0000 1000 0000 - 00f8 0000 0000: -
 282 00c0 0000 0000 - 00e0 0000 0000: heap
 283 00e0 0000 0000 - 0100 0000 0000: -
 284 0100 0000 0000 - 0500 0000 0000: shadow
 285 0500 0000 0000 - 0560 0000 0000: -
 286 0560 0000 0000 - 0760 0000 0000: traces
 287 0760 0000 0000 - 07d0 0000 0000: metainfo (memory blocks and sync objects)
 288 07d0 0000 0000 - 8000 0000 0000: -
 289 */
 290
 291 struct Mapping {
 292   static const uptr kMetaShadowBeg = 0x076000000000ull;
 293   static const uptr kMetaShadowEnd = 0x07d000000000ull;
 294   static const uptr kTraceMemBeg   = 0x056000000000ull;
 295   static const uptr kTraceMemEnd   = 0x076000000000ull;
 296   static const uptr kShadowBeg     = 0x010000000000ull;
 297   static const uptr kShadowEnd     = 0x050000000000ull;
 298   static const uptr kAppMemBeg     = 0x000000001000ull;
 299   static const uptr kAppMemEnd     = 0x00e000000000ull;
 300 };
 301
 302 #else
 303 # error "Unknown platform"
 304 #endif
 305
 306
 307 #ifdef TSAN_RUNTIME_VMA
 308 extern uptr vmaSize;
 309 #endif
 310
 311
 312 enum MappingType {
 313   MAPPING_LO_APP_BEG,
 314   MAPPING_LO_APP_END,
 315   MAPPING_HI_APP_BEG,
 316   MAPPING_HI_APP_END,
 317 #ifdef TSAN_MID_APP_RANGE
 318   MAPPING_MID_APP_BEG,
 319   MAPPING_MID_APP_END,
 320 #endif
 321   MAPPING_HEAP_BEG,
 322   MAPPING_HEAP_END,
 323   MAPPING_APP_BEG,
 324   MAPPING_APP_END,
 325   MAPPING_SHADOW_BEG,
 326   MAPPING_SHADOW_END,
 327   MAPPING_META_SHADOW_BEG,
 328   MAPPING_META_SHADOW_END,
 329   MAPPING_TRACE_BEG,
 330   MAPPING_TRACE_END,
 331   MAPPING_VDSO_BEG,
 332 };
 333
 334 template<typename Mapping, int Type>
 335 uptr MappingImpl(void) {
 336   switch (Type) {
 337 #ifndef SANITIZER_GO
 338     case MAPPING_LO_APP_BEG: return Mapping::kLoAppMemBeg;
 339     case MAPPING_LO_APP_END: return Mapping::kLoAppMemEnd;
 340 # ifdef TSAN_MID_APP_RANGE
 341     case MAPPING_MID_APP_BEG: return Mapping::kMidAppMemBeg;
 342     case MAPPING_MID_APP_END: return Mapping::kMidAppMemEnd;
 343 # endif
 344     case MAPPING_HI_APP_BEG: return Mapping::kHiAppMemBeg;
 345     case MAPPING_HI_APP_END: return Mapping::kHiAppMemEnd;
 346     case MAPPING_HEAP_BEG: return Mapping::kHeapMemBeg;
 347     case MAPPING_HEAP_END: return Mapping::kHeapMemEnd;
 348     case MAPPING_VDSO_BEG: return Mapping::kVdsoBeg;
 349 #else
 350     case MAPPING_APP_BEG: return Mapping::kAppMemBeg;
 351     case MAPPING_APP_END: return Mapping::kAppMemEnd;
 352 #endif
 353     case MAPPING_SHADOW_BEG: return Mapping::kShadowBeg;
 354     case MAPPING_SHADOW_END: return Mapping::kShadowEnd;
 355     case MAPPING_META_SHADOW_BEG: return Mapping::kMetaShadowBeg;
 356     case MAPPING_META_SHADOW_END: return Mapping::kMetaShadowEnd;
 357     case MAPPING_TRACE_BEG: return Mapping::kTraceMemBeg;
 358     case MAPPING_TRACE_END: return Mapping::kTraceMemEnd;
 359   }
 360 }
 361
 362 template<int Type>
 363 uptr MappingArchImpl(void) {
 364 #ifdef __aarch64__
 365   if (vmaSize == 39)
 366     return MappingImpl<Mapping39, Type>();
 367   else
 368     return MappingImpl<Mapping42, Type>();
 369   DCHECK(0);
 370 #elif defined(__powerpc64__)
 371   if (vmaSize == 44)
 372     return MappingImpl<Mapping44, Type>();
 373   else
 374     return MappingImpl<Mapping46, Type>();
 375   DCHECK(0);
 376 #else
 377   return MappingImpl<Mapping, Type>();
 378 #endif
 379 }
 380
 381 #ifndef SANITIZER_GO
 382 ALWAYS_INLINE
 383 uptr LoAppMemBeg(void) {
 384   return MappingArchImpl<MAPPING_LO_APP_BEG>();
 385 }
 386 ALWAYS_INLINE
 387 uptr LoAppMemEnd(void) {
 388   return MappingArchImpl<MAPPING_LO_APP_END>();
 389 }
 390
 391 #ifdef TSAN_MID_APP_RANGE
 392 ALWAYS_INLINE
 393 uptr MidAppMemBeg(void) {
 394   return MappingArchImpl<MAPPING_MID_APP_BEG>();
 395 }
 396 ALWAYS_INLINE
 397 uptr MidAppMemEnd(void) {
 398   return MappingArchImpl<MAPPING_MID_APP_END>();
 399 }
 400 #endif
 401
 402 ALWAYS_INLINE
 403 uptr HeapMemBeg(void) {
 404   return MappingArchImpl<MAPPING_HEAP_BEG>();
 405 }
 406 ALWAYS_INLINE
 407 uptr HeapMemEnd(void) {
 408   return MappingArchImpl<MAPPING_HEAP_END>();
 409 }
 410
 411 ALWAYS_INLINE
 412 uptr HiAppMemBeg(void) {
 413   return MappingArchImpl<MAPPING_HI_APP_BEG>();
 414 }
 415 ALWAYS_INLINE
 416 uptr HiAppMemEnd(void) {
 417   return MappingArchImpl<MAPPING_HI_APP_END>();
 418 }
 419
 420 ALWAYS_INLINE
 421 uptr VdsoBeg(void) {
 422   return MappingArchImpl<MAPPING_VDSO_BEG>();
 423 }
 424
 425 #else
 426
 427 ALWAYS_INLINE
 428 uptr AppMemBeg(void) {
 429   return MappingArchImpl<MAPPING_APP_BEG>();
 430 }
 431 ALWAYS_INLINE
 432 uptr AppMemEnd(void) {
 433   return MappingArchImpl<MAPPING_APP_END>();
 434 }
 435
 436 #endif
 437
 438 static inline
 439 bool GetUserRegion(int i, uptr *start, uptr *end) {
 440   switch (i) {
 441   default:
 442     return false;
 443 #ifndef SANITIZER_GO
 444   case 0:
 445     *start = LoAppMemBeg();
 446     *end = LoAppMemEnd();
 447     return true;
 448   case 1:
 449     *start = HiAppMemBeg();
 450     *end = HiAppMemEnd();
 451     return true;
 452   case 2:
 453     *start = HeapMemBeg();
 454     *end = HeapMemEnd();
 455     return true;
 456 # ifdef TSAN_MID_APP_RANGE
 457   case 3:
 458     *start = MidAppMemBeg();
 459     *end = MidAppMemEnd();
 460     return true;
 461 # endif
 462 #else
 463   case 0:
 464     *start = AppMemBeg();
 465     *end = AppMemEnd();
 466     return true;
 467 #endif
 468   }
 469 }
 470
 471 ALWAYS_INLINE
 472 uptr ShadowBeg(void) {
 473   return MappingArchImpl<MAPPING_SHADOW_BEG>();
 474 }
 475 ALWAYS_INLINE
 476 uptr ShadowEnd(void) {
 477   return MappingArchImpl<MAPPING_SHADOW_END>();
 478 }
 479
 480 ALWAYS_INLINE
 481 uptr MetaShadowBeg(void) {
 482   return MappingArchImpl<MAPPING_META_SHADOW_BEG>();
 483 }
 484 ALWAYS_INLINE
 485 uptr MetaShadowEnd(void) {
 486   return MappingArchImpl<MAPPING_META_SHADOW_END>();
 487 }
 488
 489 ALWAYS_INLINE
 490 uptr TraceMemBeg(void) {
 491   return MappingArchImpl<MAPPING_TRACE_BEG>();
 492 }
 493 ALWAYS_INLINE
 494 uptr TraceMemEnd(void) {
 495   return MappingArchImpl<MAPPING_TRACE_END>();
 496 }
 497
 498
 499 template<typename Mapping>
 500 bool IsAppMemImpl(uptr mem) {
 501 #ifndef SANITIZER_GO
 502   return (mem >= Mapping::kHeapMemBeg && mem < Mapping::kHeapMemEnd) ||
 503 # ifdef TSAN_MID_APP_RANGE
 504          (mem >= Mapping::kMidAppMemBeg && mem < Mapping::kMidAppMemEnd) ||
 505 # endif
 506          (mem >= Mapping::kLoAppMemBeg && mem < Mapping::kLoAppMemEnd) ||
 507          (mem >= Mapping::kHiAppMemBeg && mem < Mapping::kHiAppMemEnd);
 508 #else
 509   return mem >= Mapping::kAppMemBeg && mem < Mapping::kAppMemEnd;
 510 #endif
 511 }
 512
 513 ALWAYS_INLINE
 514 bool IsAppMem(uptr mem) {
 515 #ifdef __aarch64__
 516   if (vmaSize == 39)
 517     return IsAppMemImpl<Mapping39>(mem);
 518   else
 519     return IsAppMemImpl<Mapping42>(mem);
 520   DCHECK(0);
 521 #elif defined(__powerpc64__)
 522   if (vmaSize == 44)
 523     return IsAppMemImpl<Mapping44>(mem);
 524   else
 525     return IsAppMemImpl<Mapping46>(mem);
 526   DCHECK(0);
 527 #else
 528   return IsAppMemImpl<Mapping>(mem);
 529 #endif
 530 }
 531
 532
 533 template<typename Mapping>
 534 bool IsShadowMemImpl(uptr mem) {
 535   return mem >= Mapping::kShadowBeg && mem <= Mapping::kShadowEnd;
 536 }
 537
 538 ALWAYS_INLINE
 539 bool IsShadowMem(uptr mem) {
 540 #ifdef __aarch64__
 541   if (vmaSize == 39)
 542     return IsShadowMemImpl<Mapping39>(mem);
 543   else
 544     return IsShadowMemImpl<Mapping42>(mem);
 545   DCHECK(0);
 546 #elif defined(__powerpc64__)
 547   if (vmaSize == 44)
 548     return IsShadowMemImpl<Mapping44>(mem);
 549   else
 550     return IsShadowMemImpl<Mapping46>(mem);
 551   DCHECK(0);
 552 #else
 553   return IsShadowMemImpl<Mapping>(mem);
 554 #endif
 555 }
 556
 557
 558 template<typename Mapping>
 559 bool IsMetaMemImpl(uptr mem) {
 560   return mem >= Mapping::kMetaShadowBeg && mem <= Mapping::kMetaShadowEnd;
 561 }
 562
 563 ALWAYS_INLINE
 564 bool IsMetaMem(uptr mem) {
 565 #ifdef __aarch64__
 566   if (vmaSize == 39)
 567     return IsMetaMemImpl<Mapping39>(mem);
 568   else
 569     return IsMetaMemImpl<Mapping42>(mem);
 570   DCHECK(0);
 571 #elif defined(__powerpc64__)
 572   if (vmaSize == 44)
 573     return IsMetaMemImpl<Mapping44>(mem);
 574   else
 575     return IsMetaMemImpl<Mapping46>(mem);
 576   DCHECK(0);
 577 #else
 578   return IsMetaMemImpl<Mapping>(mem);
 579 #endif
 580 }
 581
 582
 583 template<typename Mapping>
 584 uptr MemToShadowImpl(uptr x) {
 585   DCHECK(IsAppMem(x));
 586 #ifndef SANITIZER_GO
 587   return (((x) & ~(Mapping::kAppMemMsk | (kShadowCell - 1)))
 588       ^ Mapping::kAppMemXor) * kShadowCnt;
 589 #else
 590 # ifndef SANITIZER_WINDOWS
 591   return ((x & ~(kShadowCell - 1)) * kShadowCnt) | Mapping::kShadowBeg;
 592 # else
 593   return ((x & ~(kShadowCell - 1)) * kShadowCnt) + Mapping::kShadowBeg;
 594 # endif
 595 #endif
 596 }
 597
 598 ALWAYS_INLINE
 599 uptr MemToShadow(uptr x) {
 600 #ifdef __aarch64__
 601   if (vmaSize == 39)
 602     return MemToShadowImpl<Mapping39>(x);
 603   else
 604     return MemToShadowImpl<Mapping42>(x);
 605   DCHECK(0);
 606 #elif defined(__powerpc64__)
 607   if (vmaSize == 44)
 608     return MemToShadowImpl<Mapping44>(x);
 609   else
 610     return MemToShadowImpl<Mapping46>(x);
 611   DCHECK(0);
 612 #else
 613   return MemToShadowImpl<Mapping>(x);
 614 #endif
 615 }
 616
 617
 618 template<typename Mapping>
 619 u32 *MemToMetaImpl(uptr x) {
 620   DCHECK(IsAppMem(x));
 621 #ifndef SANITIZER_GO
 622   return (u32*)(((((x) & ~(Mapping::kAppMemMsk | (kMetaShadowCell - 1)))
 623         ^ Mapping::kAppMemXor) / kMetaShadowCell * kMetaShadowSize)
 624           | Mapping::kMetaShadowBeg);
 625 #else
 626   return (u32*)(((x & ~(kMetaShadowCell - 1)) / \
 627       kMetaShadowCell * kMetaShadowSize) | Mapping::kMetaShadowBeg);
 628 #endif
 629 }
 630
 631 ALWAYS_INLINE
 632 u32 *MemToMeta(uptr x) {
 633 #ifdef __aarch64__
 634   if (vmaSize == 39)
 635     return MemToMetaImpl<Mapping39>(x);
 636   else
 637     return MemToMetaImpl<Mapping42>(x);
 638   DCHECK(0);
 639 #elif defined(__powerpc64__)
 640   if (vmaSize == 44)
 641     return MemToMetaImpl<Mapping44>(x);
 642   else
 643     return MemToMetaImpl<Mapping46>(x);
 644   DCHECK(0);
 645 #else
 646   return MemToMetaImpl<Mapping>(x);
 647 #endif
 648 }
 649
 650
 651 template<typename Mapping>
 652 uptr ShadowToMemImpl(uptr s) {
 653   DCHECK(IsShadowMem(s));
 654 #ifndef SANITIZER_GO
 655   if (s >= MemToShadow(Mapping::kLoAppMemBeg)
 656       && s <= MemToShadow(Mapping::kLoAppMemEnd - 1))
 657     return (s / kShadowCnt) ^ Mapping::kAppMemXor;
 658 # ifdef TSAN_MID_APP_RANGE
 659   if (s >= MemToShadow(Mapping::kMidAppMemBeg)
 660       && s <= MemToShadow(Mapping::kMidAppMemEnd - 1))
 661     return ((s / kShadowCnt) ^ Mapping::kAppMemXor) + Mapping::kMidShadowOff;
 662 # endif
 663   else
 664     return ((s / kShadowCnt) ^ Mapping::kAppMemXor) | Mapping::kAppMemMsk;
 665 #else
 666 # ifndef SANITIZER_WINDOWS
 667   return (s & ~Mapping::kShadowBeg) / kShadowCnt;
 668 # else
 669   return (s - Mapping::kShadowBeg) / kShadowCnt;
 670 # endif // SANITIZER_WINDOWS
 671 #endif
 672 }
 673
 674 ALWAYS_INLINE
 675 uptr ShadowToMem(uptr s) {
 676 #ifdef __aarch64__
 677   if (vmaSize == 39)
 678     return ShadowToMemImpl<Mapping39>(s);
 679   else
 680     return ShadowToMemImpl<Mapping42>(s);
 681   DCHECK(0);
 682 #elif defined(__powerpc64__)
 683   if (vmaSize == 44)
 684     return ShadowToMemImpl<Mapping44>(s);
 685   else
 686     return ShadowToMemImpl<Mapping46>(s);
 687   DCHECK(0);
 688 #else
 689   return ShadowToMemImpl<Mapping>(s);
 690 #endif
 691 }
 692
 693
 694
 695 // The additional page is to catch shadow stack overflow as paging fault.
 696 // Windows wants 64K alignment for mmaps.
 697 const uptr kTotalTraceSize = (kTraceSize * sizeof(Event) + sizeof(Trace)
 698     + (64 << 10) + (64 << 10) - 1) & ~((64 << 10) - 1);
 699
 700 template<typename Mapping>
 701 uptr GetThreadTraceImpl(int tid) {
 702   uptr p = Mapping::kTraceMemBeg + (uptr)tid * kTotalTraceSize;
 703   DCHECK_LT(p, Mapping::kTraceMemEnd);
 704   return p;
 705 }
 706
 707 ALWAYS_INLINE
 708 uptr GetThreadTrace(int tid) {
 709 #ifdef __aarch64__
 710   if (vmaSize == 39)
 711     return GetThreadTraceImpl<Mapping39>(tid);
 712   else
 713     return GetThreadTraceImpl<Mapping42>(tid);
 714   DCHECK(0);
 715 #elif defined(__powerpc64__)
 716   if (vmaSize == 44)
 717     return GetThreadTraceImpl<Mapping44>(tid);
 718   else
 719     return GetThreadTraceImpl<Mapping46>(tid);
 720   DCHECK(0);
 721 #else
 722   return GetThreadTraceImpl<Mapping>(tid);
 723 #endif
 724 }
 725
 726
 727 template<typename Mapping>
 728 uptr GetThreadTraceHeaderImpl(int tid) {
 729   uptr p = Mapping::kTraceMemBeg + (uptr)tid * kTotalTraceSize
 730       + kTraceSize * sizeof(Event);
 731   DCHECK_LT(p, Mapping::kTraceMemEnd);
 732   return p;
 733 }
 734
 735 ALWAYS_INLINE
 736 uptr GetThreadTraceHeader(int tid) {
 737 #ifdef __aarch64__
 738   if (vmaSize == 39)
 739     return GetThreadTraceHeaderImpl<Mapping39>(tid);
 740   else
 741     return GetThreadTraceHeaderImpl<Mapping42>(tid);
 742   DCHECK(0);
 743 #elif defined(__powerpc64__)
 744   if (vmaSize == 44)
 745     return GetThreadTraceHeaderImpl<Mapping44>(tid);
 746   else
 747     return GetThreadTraceHeaderImpl<Mapping46>(tid);
 748   DCHECK(0);
 749 #else
 750   return GetThreadTraceHeaderImpl<Mapping>(tid);
 751 #endif
 752 }
 753
 754 void InitializePlatform();
 755 void InitializePlatformEarly();
 756 void CheckAndProtect();
 757 void InitializeShadowMemoryPlatform();
 758 void FlushShadowMemory();
 759 void WriteMemoryProfile(char *buf, uptr buf_size, uptr nthread, uptr nlive);
 760 int ExtractResolvFDs(void *state, int *fds, int nfd);
 761 int ExtractRecvmsgFDs(void *msg, int *fds, int nfd);
 762
 763 int call_pthread_cancel_with_cleanup(int(*fn)(void *c, void *m,
 764     void *abstime), void *c, void *m, void *abstime,
 765     void(*cleanup)(void *arg), void *arg);
 766
 767 void DestroyThreadState();
 768
 769 }  // namespace __tsan
 770
 771 #endif  // TSAN_PLATFORM_H