1 | //=-- lsan_fuchsia.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 | // Standalone LSan RTL code specific to Fuchsia. |
11 | // |
12 | //===---------------------------------------------------------------------===// |
13 | |
14 | #include "sanitizer_common/sanitizer_platform.h" |
15 | |
16 | #if SANITIZER_FUCHSIA |
17 | #include <zircon/sanitizer.h> |
18 | |
19 | #include "lsan.h" |
20 | #include "lsan_allocator.h" |
21 | |
22 | using namespace __lsan; |
23 | |
24 | namespace __sanitizer { |
25 | // LSan doesn't need to do anything else special in the startup hook. |
26 | void EarlySanitizerInit() {} |
27 | } // namespace __sanitizer |
28 | |
29 | namespace __lsan { |
30 | |
31 | void LsanOnDeadlySignal(int signo, void *siginfo, void *context) {} |
32 | |
33 | ThreadContext::ThreadContext(int tid) : ThreadContextLsanBase(tid) {} |
34 | |
35 | struct OnCreatedArgs { |
36 | uptr stack_begin, stack_end; |
37 | }; |
38 | |
39 | // On Fuchsia, the stack bounds of a new thread are available before |
40 | // the thread itself has started running. |
41 | void ThreadContext::OnCreated(void *arg) { |
42 | // Stack bounds passed through from __sanitizer_before_thread_create_hook |
43 | // or InitializeMainThread. |
44 | auto args = reinterpret_cast<const OnCreatedArgs *>(arg); |
45 | stack_begin_ = args->stack_begin; |
46 | stack_end_ = args->stack_end; |
47 | } |
48 | |
49 | struct OnStartedArgs { |
50 | uptr cache_begin, cache_end; |
51 | }; |
52 | |
53 | void ThreadContext::OnStarted(void *arg) { |
54 | ThreadContextLsanBase::OnStarted(arg); |
55 | auto args = reinterpret_cast<const OnStartedArgs *>(arg); |
56 | cache_begin_ = args->cache_begin; |
57 | cache_end_ = args->cache_end; |
58 | } |
59 | |
60 | void ThreadStart(u32 tid) { |
61 | OnStartedArgs args; |
62 | GetAllocatorCacheRange(&args.cache_begin, &args.cache_end); |
63 | CHECK_EQ(args.cache_end - args.cache_begin, sizeof(AllocatorCache)); |
64 | ThreadContextLsanBase::ThreadStart(tid, GetTid(), ThreadType::Regular, &args); |
65 | } |
66 | |
67 | void InitializeMainThread() { |
68 | OnCreatedArgs args; |
69 | __sanitizer::GetThreadStackTopAndBottom(true, &args.stack_end, |
70 | &args.stack_begin); |
71 | u32 tid = ThreadCreate(kMainTid, true, &args); |
72 | CHECK_EQ(tid, 0); |
73 | ThreadStart(tid); |
74 | } |
75 | |
76 | void GetAllThreadAllocatorCachesLocked(InternalMmapVector<uptr> *caches) { |
77 | GetLsanThreadRegistryLocked()->RunCallbackForEachThreadLocked( |
78 | [](ThreadContextBase *tctx, void *arg) { |
79 | auto ctx = static_cast<ThreadContext *>(tctx); |
80 | static_cast<decltype(caches)>(arg)->push_back(ctx->cache_begin()); |
81 | }, |
82 | caches); |
83 | } |
84 | |
85 | // On Fuchsia, leak detection is done by a special hook after atexit hooks. |
86 | // So this doesn't install any atexit hook like on other platforms. |
87 | void InstallAtExitCheckLeaks() {} |
88 | void InstallAtForkHandler() {} |
89 | |
90 | // ASan defines this to check its `halt_on_error` flag. |
91 | bool UseExitcodeOnLeak() { return true; } |
92 | |
93 | } // namespace __lsan |
94 | |
95 | // These are declared (in extern "C") by <zircon/sanitizer.h>. |
96 | // The system runtime will call our definitions directly. |
97 | |
98 | // This is called before each thread creation is attempted. So, in |
99 | // its first call, the calling thread is the initial and sole thread. |
100 | void *__sanitizer_before_thread_create_hook(thrd_t thread, bool detached, |
101 | const char *name, void *stack_base, |
102 | size_t stack_size) { |
103 | ENSURE_LSAN_INITED; |
104 | EnsureMainThreadIDIsCorrect(); |
105 | OnCreatedArgs args; |
106 | args.stack_begin = reinterpret_cast<uptr>(stack_base); |
107 | args.stack_end = args.stack_begin + stack_size; |
108 | u32 parent_tid = GetCurrentThreadId(); |
109 | u32 tid = ThreadCreate(parent_tid, detached, &args); |
110 | return reinterpret_cast<void *>(static_cast<uptr>(tid)); |
111 | } |
112 | |
113 | // This is called after creating a new thread (in the creating thread), |
114 | // with the pointer returned by __sanitizer_before_thread_create_hook (above). |
115 | void __sanitizer_thread_create_hook(void *hook, thrd_t thread, int error) { |
116 | u32 tid = static_cast<u32>(reinterpret_cast<uptr>(hook)); |
117 | // On success, there is nothing to do here. |
118 | if (error != thrd_success) { |
119 | // Clean up the thread registry for the thread creation that didn't happen. |
120 | GetLsanThreadRegistryLocked()->FinishThread(tid); |
121 | } |
122 | } |
123 | |
124 | // This is called in the newly-created thread before it runs anything else, |
125 | // with the pointer returned by __sanitizer_before_thread_create_hook (above). |
126 | void __sanitizer_thread_start_hook(void *hook, thrd_t self) { |
127 | u32 tid = static_cast<u32>(reinterpret_cast<uptr>(hook)); |
128 | ThreadStart(tid); |
129 | } |
130 | |
131 | // Each thread runs this just before it exits, |
132 | // with the pointer returned by BeforeThreadCreateHook (above). |
133 | // All per-thread destructors have already been called. |
134 | void __sanitizer_thread_exit_hook(void *hook, thrd_t self) { ThreadFinish(); } |
135 | |
136 | #endif // SANITIZER_FUCHSIA |
137 | |