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