168d75effSDimitry Andric //=-- lsan_thread.cpp -----------------------------------------------------===// 268d75effSDimitry Andric // 368d75effSDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 468d75effSDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 568d75effSDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 668d75effSDimitry Andric // 768d75effSDimitry Andric //===----------------------------------------------------------------------===// 868d75effSDimitry Andric // 968d75effSDimitry Andric // This file is a part of LeakSanitizer. 1068d75effSDimitry Andric // See lsan_thread.h for details. 1168d75effSDimitry Andric // 1268d75effSDimitry Andric //===----------------------------------------------------------------------===// 1368d75effSDimitry Andric 1468d75effSDimitry Andric #include "lsan_thread.h" 1568d75effSDimitry Andric 165ffd83dbSDimitry Andric #include "lsan.h" 175ffd83dbSDimitry Andric #include "lsan_allocator.h" 185ffd83dbSDimitry Andric #include "lsan_common.h" 1968d75effSDimitry Andric #include "sanitizer_common/sanitizer_common.h" 2068d75effSDimitry Andric #include "sanitizer_common/sanitizer_placement_new.h" 2168d75effSDimitry Andric #include "sanitizer_common/sanitizer_thread_registry.h" 2268d75effSDimitry Andric #include "sanitizer_common/sanitizer_tls_get_addr.h" 2368d75effSDimitry Andric 2468d75effSDimitry Andric namespace __lsan { 2568d75effSDimitry Andric 2668d75effSDimitry Andric static ThreadRegistry *thread_registry; 2768d75effSDimitry Andric 2868d75effSDimitry Andric static ThreadContextBase *CreateThreadContext(u32 tid) { 2968d75effSDimitry Andric void *mem = MmapOrDie(sizeof(ThreadContext), "ThreadContext"); 3068d75effSDimitry Andric return new (mem) ThreadContext(tid); 3168d75effSDimitry Andric } 3268d75effSDimitry Andric 3368d75effSDimitry Andric void InitializeThreadRegistry() { 3468d75effSDimitry Andric static ALIGNED(64) char thread_registry_placeholder[sizeof(ThreadRegistry)]; 35fe6060f1SDimitry Andric thread_registry = 36fe6060f1SDimitry Andric new (thread_registry_placeholder) ThreadRegistry(CreateThreadContext); 3768d75effSDimitry Andric } 3868d75effSDimitry Andric 395ffd83dbSDimitry Andric ThreadContextLsanBase::ThreadContextLsanBase(int tid) 405ffd83dbSDimitry Andric : ThreadContextBase(tid) {} 4168d75effSDimitry Andric 425ffd83dbSDimitry Andric void ThreadContextLsanBase::OnFinished() { 4368d75effSDimitry Andric AllocatorThreadFinish(); 4468d75effSDimitry Andric DTLS_Destroy(); 4568d75effSDimitry Andric } 4668d75effSDimitry Andric 47*349cc55cSDimitry Andric u32 ThreadCreate(u32 parent_tid, bool detached, void *arg) { 48*349cc55cSDimitry Andric return thread_registry->CreateThread(0, detached, parent_tid, arg); 4968d75effSDimitry Andric } 5068d75effSDimitry Andric 515ffd83dbSDimitry Andric void ThreadContextLsanBase::ThreadStart(u32 tid, tid_t os_id, 525ffd83dbSDimitry Andric ThreadType thread_type, void *arg) { 535ffd83dbSDimitry Andric thread_registry->StartThread(tid, os_id, thread_type, arg); 545ffd83dbSDimitry Andric SetCurrentThread(tid); 5568d75effSDimitry Andric } 5668d75effSDimitry Andric 5768d75effSDimitry Andric void ThreadFinish() { 5868d75effSDimitry Andric thread_registry->FinishThread(GetCurrentThread()); 5968d75effSDimitry Andric SetCurrentThread(kInvalidTid); 6068d75effSDimitry Andric } 6168d75effSDimitry Andric 6268d75effSDimitry Andric ThreadContext *CurrentThreadContext() { 635ffd83dbSDimitry Andric if (!thread_registry) 645ffd83dbSDimitry Andric return nullptr; 6568d75effSDimitry Andric if (GetCurrentThread() == kInvalidTid) 6668d75effSDimitry Andric return nullptr; 6768d75effSDimitry Andric // No lock needed when getting current thread. 6868d75effSDimitry Andric return (ThreadContext *)thread_registry->GetThreadLocked(GetCurrentThread()); 6968d75effSDimitry Andric } 7068d75effSDimitry Andric 7168d75effSDimitry Andric void EnsureMainThreadIDIsCorrect() { 72fe6060f1SDimitry Andric if (GetCurrentThread() == kMainTid) 7368d75effSDimitry Andric CurrentThreadContext()->os_id = GetTid(); 7468d75effSDimitry Andric } 7568d75effSDimitry Andric 7668d75effSDimitry Andric ///// Interface to the common LSan module. ///// 7768d75effSDimitry Andric 7868d75effSDimitry Andric void ForEachExtraStackRange(tid_t os_id, RangeIteratorCallback callback, 795ffd83dbSDimitry Andric void *arg) {} 8068d75effSDimitry Andric 815ffd83dbSDimitry Andric void LockThreadRegistry() { thread_registry->Lock(); } 8268d75effSDimitry Andric 835ffd83dbSDimitry Andric void UnlockThreadRegistry() { thread_registry->Unlock(); } 8468d75effSDimitry Andric 8568d75effSDimitry Andric ThreadRegistry *GetThreadRegistryLocked() { 8668d75effSDimitry Andric thread_registry->CheckLocked(); 8768d75effSDimitry Andric return thread_registry; 8868d75effSDimitry Andric } 8968d75effSDimitry Andric 9068d75effSDimitry Andric } // namespace __lsan 91