1 //===-- tsd.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 #ifndef SCUDO_TSD_H_ 10 #define SCUDO_TSD_H_ 11 12 #include "atomic_helpers.h" 13 #include "common.h" 14 #include "mutex.h" 15 16 #include <limits.h> // for PTHREAD_DESTRUCTOR_ITERATIONS 17 18 // With some build setups, this might still not be defined. 19 #ifndef PTHREAD_DESTRUCTOR_ITERATIONS 20 #define PTHREAD_DESTRUCTOR_ITERATIONS 4 21 #endif 22 23 namespace scudo { 24 25 template <class Allocator> struct ALIGNED(SCUDO_CACHE_LINE_SIZE) TSD { 26 typename Allocator::CacheT Cache; 27 typename Allocator::QuarantineCacheT QuarantineCache; 28 u8 DestructorIterations; 29 30 void initLinkerInitialized(Allocator *Instance) { 31 Instance->initCache(&Cache); 32 DestructorIterations = PTHREAD_DESTRUCTOR_ITERATIONS; 33 } 34 void init(Allocator *Instance) { 35 memset(this, 0, sizeof(*this)); 36 initLinkerInitialized(Instance); 37 } 38 39 void commitBack(Allocator *Instance) { Instance->commitBack(this); } 40 41 INLINE bool tryLock() { 42 if (Mutex.tryLock()) { 43 atomic_store_relaxed(&Precedence, 0); 44 return true; 45 } 46 if (atomic_load_relaxed(&Precedence) == 0) 47 atomic_store_relaxed( 48 &Precedence, 49 static_cast<uptr>(getMonotonicTime() >> FIRST_32_SECOND_64(16, 0))); 50 return false; 51 } 52 INLINE void lock() { 53 atomic_store_relaxed(&Precedence, 0); 54 Mutex.lock(); 55 } 56 INLINE void unlock() { Mutex.unlock(); } 57 INLINE uptr getPrecedence() { return atomic_load_relaxed(&Precedence); } 58 59 private: 60 HybridMutex Mutex; 61 atomic_uptr Precedence; 62 }; 63 64 } // namespace scudo 65 66 #endif // SCUDO_TSD_H_ 67