1 //=-- lsan_thread.cpp -----------------------------------------------------===//
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 // See lsan_thread.h for details.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "lsan_thread.h"
15
16 #include "lsan.h"
17 #include "lsan_allocator.h"
18 #include "lsan_common.h"
19 #include "sanitizer_common/sanitizer_common.h"
20 #include "sanitizer_common/sanitizer_placement_new.h"
21 #include "sanitizer_common/sanitizer_thread_registry.h"
22 #include "sanitizer_common/sanitizer_tls_get_addr.h"
23
24 namespace __lsan {
25
26 static ThreadRegistry *thread_registry;
27 static ThreadArgRetval *thread_arg_retval;
28
29 static Mutex mu_for_thread_context;
30 static LowLevelAllocator allocator_for_thread_context;
31
CreateThreadContext(u32 tid)32 static ThreadContextBase *CreateThreadContext(u32 tid) {
33 Lock lock(&mu_for_thread_context);
34 return new (allocator_for_thread_context) ThreadContext(tid);
35 }
36
InitializeThreads()37 void InitializeThreads() {
38 static ALIGNED(alignof(
39 ThreadRegistry)) char thread_registry_placeholder[sizeof(ThreadRegistry)];
40 thread_registry =
41 new (thread_registry_placeholder) ThreadRegistry(CreateThreadContext);
42
43 static ALIGNED(alignof(ThreadArgRetval)) char
44 thread_arg_retval_placeholder[sizeof(ThreadArgRetval)];
45 thread_arg_retval = new (thread_arg_retval_placeholder) ThreadArgRetval();
46 }
47
GetThreadArgRetval()48 ThreadArgRetval &GetThreadArgRetval() { return *thread_arg_retval; }
49
ThreadContextLsanBase(int tid)50 ThreadContextLsanBase::ThreadContextLsanBase(int tid)
51 : ThreadContextBase(tid) {}
52
OnStarted(void * arg)53 void ThreadContextLsanBase::OnStarted(void *arg) {
54 SetCurrentThread(this);
55 AllocatorThreadStart();
56 }
57
OnFinished()58 void ThreadContextLsanBase::OnFinished() {
59 AllocatorThreadFinish();
60 DTLS_Destroy();
61 SetCurrentThread(nullptr);
62 }
63
ThreadCreate(u32 parent_tid,bool detached,void * arg)64 u32 ThreadCreate(u32 parent_tid, bool detached, void *arg) {
65 return thread_registry->CreateThread(0, detached, parent_tid, arg);
66 }
67
ThreadStart(u32 tid,tid_t os_id,ThreadType thread_type,void * arg)68 void ThreadContextLsanBase::ThreadStart(u32 tid, tid_t os_id,
69 ThreadType thread_type, void *arg) {
70 thread_registry->StartThread(tid, os_id, thread_type, arg);
71 }
72
ThreadFinish()73 void ThreadFinish() { thread_registry->FinishThread(GetCurrentThreadId()); }
74
EnsureMainThreadIDIsCorrect()75 void EnsureMainThreadIDIsCorrect() {
76 if (GetCurrentThreadId() == kMainTid)
77 GetCurrentThread()->os_id = GetTid();
78 }
79
80 ///// Interface to the common LSan module. /////
81
GetThreadExtraStackRangesLocked(tid_t os_id,InternalMmapVector<Range> * ranges)82 void GetThreadExtraStackRangesLocked(tid_t os_id,
83 InternalMmapVector<Range> *ranges) {}
GetThreadExtraStackRangesLocked(InternalMmapVector<Range> * ranges)84 void GetThreadExtraStackRangesLocked(InternalMmapVector<Range> *ranges) {}
85
LockThreads()86 void LockThreads() {
87 thread_registry->Lock();
88 thread_arg_retval->Lock();
89 }
90
UnlockThreads()91 void UnlockThreads() {
92 thread_arg_retval->Unlock();
93 thread_registry->Unlock();
94 }
95
GetLsanThreadRegistryLocked()96 ThreadRegistry *GetLsanThreadRegistryLocked() {
97 thread_registry->CheckLocked();
98 return thread_registry;
99 }
100
GetRunningThreadsLocked(InternalMmapVector<tid_t> * threads)101 void GetRunningThreadsLocked(InternalMmapVector<tid_t> *threads) {
102 GetLsanThreadRegistryLocked()->RunCallbackForEachThreadLocked(
103 [](ThreadContextBase *tctx, void *threads) {
104 if (tctx->status == ThreadStatusRunning) {
105 reinterpret_cast<InternalMmapVector<tid_t> *>(threads)->push_back(
106 tctx->os_id);
107 }
108 },
109 threads);
110 }
111
GetAdditionalThreadContextPtrsLocked(InternalMmapVector<uptr> * ptrs)112 void GetAdditionalThreadContextPtrsLocked(InternalMmapVector<uptr> *ptrs) {
113 GetThreadArgRetval().GetAllPtrsLocked(ptrs);
114 }
115
116 } // namespace __lsan
117