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 u32 ThreadCreate(u32 tid, bool detached, void *arg = nullptr);
49 void ThreadFinish();
50 
51 u32 GetCurrentThread();
52 void SetCurrentThread(u32 tid);
53 ThreadContext *CurrentThreadContext();
54 void EnsureMainThreadIDIsCorrect();
55 
56 }  // namespace __lsan
57 
58 #endif  // LSAN_THREAD_H
59