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_registry.h" 18 19 namespace __lsan { 20 21 class ThreadContextLsanBase : public ThreadContextBase { 22 public: 23 explicit ThreadContextLsanBase(int tid); 24 void OnFinished() override; 25 uptr stack_begin() { return stack_begin_; } 26 uptr stack_end() { return stack_end_; } 27 uptr cache_begin() { return cache_begin_; } 28 uptr cache_end() { return cache_end_; } 29 30 // The argument is passed on to the subclass's OnStarted member function. 31 static void ThreadStart(u32 tid, tid_t os_id, ThreadType thread_type, 32 void *onstarted_arg); 33 34 protected: 35 ~ThreadContextLsanBase() {} 36 uptr stack_begin_ = 0; 37 uptr stack_end_ = 0; 38 uptr cache_begin_ = 0; 39 uptr cache_end_ = 0; 40 }; 41 42 // This subclass of ThreadContextLsanBase is declared in an OS-specific header. 43 class ThreadContext; 44 45 void InitializeThreadRegistry(); 46 void InitializeMainThread(); 47 48 ThreadRegistry *GetLsanThreadRegistryLocked(); 49 50 u32 ThreadCreate(u32 tid, bool detached, void *arg = nullptr); 51 void ThreadFinish(); 52 53 u32 GetCurrentThread(); 54 void SetCurrentThread(u32 tid); 55 ThreadContext *CurrentThreadContext(); 56 void EnsureMainThreadIDIsCorrect(); 57 58 } // namespace __lsan 59 60 #endif // LSAN_THREAD_H 61