1 //=-- lsan_thread.h -------------------------------------------------------===// 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 LeakSanitizer. 10 // Thread registry for standalone LSan. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #ifndef LSAN_THREAD_H 15 #define LSAN_THREAD_H 16 17 #include "sanitizer_common/sanitizer_thread_arg_retval.h" 18 #include "sanitizer_common/sanitizer_thread_registry.h" 19 20 namespace __lsan { 21 22 class ThreadContextLsanBase : public ThreadContextBase { 23 public: 24 explicit ThreadContextLsanBase(int tid); 25 void OnStarted(void *arg) override; 26 void OnFinished() override; 27 uptr stack_begin() { return stack_begin_; } 28 uptr stack_end() { return stack_end_; } 29 uptr cache_begin() { return cache_begin_; } 30 uptr cache_end() { return cache_end_; } 31 32 // The argument is passed on to the subclass's OnStarted member function. 33 static void ThreadStart(u32 tid, tid_t os_id, ThreadType thread_type, 34 void *onstarted_arg); 35 36 protected: 37 ~ThreadContextLsanBase() {} 38 uptr stack_begin_ = 0; 39 uptr stack_end_ = 0; 40 uptr cache_begin_ = 0; 41 uptr cache_end_ = 0; 42 }; 43 44 // This subclass of ThreadContextLsanBase is declared in an OS-specific header. 45 class ThreadContext; 46 47 void InitializeThreads(); 48 void InitializeMainThread(); 49 50 ThreadRegistry *GetLsanThreadRegistryLocked(); 51 ThreadArgRetval &GetThreadArgRetval(); 52 53 u32 ThreadCreate(u32 tid, bool detached, void *arg = nullptr); 54 void ThreadFinish(); 55 56 ThreadContextLsanBase *GetCurrentThread(); 57 inline u32 GetCurrentThreadId() { 58 ThreadContextLsanBase *ctx = GetCurrentThread(); 59 return ctx ? ctx->tid : kInvalidTid; 60 } 61 void SetCurrentThread(ThreadContextLsanBase *tctx); 62 void EnsureMainThreadIDIsCorrect(); 63 64 } // namespace __lsan 65 66 #endif // LSAN_THREAD_H 67