compiler-rt/lib/sanitizer_common/sanitizer_atomic.h

   1 //===-- sanitizer_atomic.h --------------------------------------*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file is a part of ThreadSanitizer/AddressSanitizer runtime.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #ifndef SANITIZER_ATOMIC_H
  14 #define SANITIZER_ATOMIC_H
  15
  16 #include "sanitizer_internal_defs.h"
  17
  18 namespace __sanitizer {
  19
  20 enum memory_order {
  21   memory_order_relaxed = 1 << 0,
  22   memory_order_consume = 1 << 1,
  23   memory_order_acquire = 1 << 2,
  24   memory_order_release = 1 << 3,
  25   memory_order_acq_rel = 1 << 4,
  26   memory_order_seq_cst = 1 << 5
  27 };
  28
  29 struct atomic_uint8_t {
  30   typedef u8 Type;
  31   volatile Type val_dont_use;
  32 };
  33
  34 struct atomic_uint16_t {
  35   typedef u16 Type;
  36   volatile Type val_dont_use;
  37 };
  38
  39 struct atomic_sint32_t {
  40   typedef s32 Type;
  41   volatile Type val_dont_use;
  42 };
  43
  44 struct atomic_uint32_t {
  45   typedef u32 Type;
  46   volatile Type val_dont_use;
  47 };
  48
  49 struct atomic_uint64_t {
  50   typedef u64 Type;
  51   // On 32-bit platforms u64 is not necessary aligned on 8 bytes.
  52   volatile ALIGNED(8) Type val_dont_use;
  53 };
  54
  55 struct atomic_uintptr_t {
  56   typedef uptr Type;
  57   volatile Type val_dont_use;
  58 };
  59
  60 }  // namespace __sanitizer
  61
  62 #if defined(__clang__) || defined(__GNUC__)
  63 # include "sanitizer_atomic_clang.h"
  64 #elif defined(_MSC_VER)
  65 # include "sanitizer_atomic_msvc.h"
  66 #else
  67 # error "Unsupported compiler"
  68 #endif
  69
  70 namespace __sanitizer {
  71
  72 // Clutter-reducing helpers.
  73
  74 template<typename T>
  75 inline typename T::Type atomic_load_relaxed(const volatile T *a) {
  76   return atomic_load(a, memory_order_relaxed);
  77 }
  78
  79 template<typename T>
  80 inline void atomic_store_relaxed(volatile T *a, typename T::Type v) {
  81   atomic_store(a, v, memory_order_relaxed);
  82 }
  83
  84 }  // namespace __sanitizer
  85
  86 #endif  // SANITIZER_ATOMIC_H