xref: /freebsd/contrib/llvm-project/compiler-rt/lib/lsan/lsan_thread.h (revision dd41de95a84d979615a2ef11df6850622bf6184e)
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   uptr stack_begin_ = 0;
36   uptr stack_end_ = 0;
37   uptr cache_begin_ = 0;
38   uptr cache_end_ = 0;
39 };
40 
41 // This subclass of ThreadContextLsanBase is declared in an OS-specific header.
42 class ThreadContext;
43 
44 void InitializeThreadRegistry();
45 void InitializeMainThread();
46 
47 u32 ThreadCreate(u32 tid, uptr uid, bool detached, void *arg = nullptr);
48 void ThreadFinish();
49 void ThreadJoin(u32 tid);
50 u32 ThreadTid(uptr uid);
51 
52 u32 GetCurrentThread();
53 void SetCurrentThread(u32 tid);
54 ThreadContext *CurrentThreadContext();
55 void EnsureMainThreadIDIsCorrect();
56 
57 }  // namespace __lsan
58 
59 #endif  // LSAN_THREAD_H
60