82 lines · c
1//===-- msan_thread.h -------------------------------------------*- C++ -*-===//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-exception6//7//===----------------------------------------------------------------------===//8//9// This file is a part of MemorySanitizer.10//11//===----------------------------------------------------------------------===//12 13#ifndef MSAN_THREAD_H14#define MSAN_THREAD_H15 16#include "msan_allocator.h"17#include "sanitizer_common/sanitizer_common.h"18#include "sanitizer_common/sanitizer_posix.h"19namespace __msan {20 21class MsanThread {22 public:23 static MsanThread *Create(thread_callback_t start_routine, void *arg);24 static void TSDDtor(void *tsd);25 void Destroy();26 27 void Init(); // Should be called from the thread itself.28 thread_return_t ThreadStart();29 30 uptr stack_top();31 uptr stack_bottom();32 uptr tls_begin() { return tls_begin_; }33 uptr tls_end() { return tls_end_; }34 bool IsMainThread() { return start_routine_ == nullptr; }35 36 bool AddrIsInStack(uptr addr);37 38 bool InSignalHandler() { return in_signal_handler_; }39 void EnterSignalHandler() { in_signal_handler_++; }40 void LeaveSignalHandler() { in_signal_handler_--; }41 42 void StartSwitchFiber(uptr bottom, uptr size);43 void FinishSwitchFiber(uptr *bottom_old, uptr *size_old);44 45 MsanThreadLocalMallocStorage &malloc_storage() { return malloc_storage_; }46 47 int destructor_iterations_;48 __sanitizer_sigset_t starting_sigset_;49 50 private:51 // NOTE: There is no MsanThread constructor. It is allocated52 // via mmap() and *must* be valid in zero-initialized state.53 void SetThreadStackAndTls();54 void ClearShadowForThreadStackAndTLS();55 struct StackBounds {56 uptr bottom;57 uptr top;58 };59 StackBounds GetStackBounds() const;60 thread_callback_t start_routine_;61 void *arg_;62 63 bool stack_switching_;64 65 StackBounds stack_;66 StackBounds next_stack_;67 68 uptr tls_begin_;69 uptr tls_end_;70 71 unsigned in_signal_handler_;72 73 MsanThreadLocalMallocStorage malloc_storage_;74};75 76MsanThread *GetCurrentThread();77void SetCurrentThread(MsanThread *t);78 79} // namespace __msan80 81#endif // MSAN_THREAD_H82