libsanitizer/lsan/lsan_thread.h

   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 OnStarted(void *arg) override;
  25   void OnFinished() override;
  26   uptr stack_begin() { return stack_begin_; }
  27   uptr stack_end() { return stack_end_; }
  28   uptr cache_begin() { return cache_begin_; }
  29   uptr cache_end() { return cache_end_; }
  30
  31   // The argument is passed on to the subclass's OnStarted member function.
  32   static void ThreadStart(u32 tid, tid_t os_id, ThreadType thread_type,
  33                           void *onstarted_arg);
  34
  35  protected:
  36   ~ThreadContextLsanBase() {}
  37   uptr stack_begin_ = 0;
  38   uptr stack_end_ = 0;
  39   uptr cache_begin_ = 0;
  40   uptr cache_end_ = 0;
  41 };
  42
  43 // This subclass of ThreadContextLsanBase is declared in an OS-specific header.
  44 class ThreadContext;
  45
  46 void InitializeThreadRegistry();
  47 void InitializeMainThread();
  48
  49 ThreadRegistry *GetLsanThreadRegistryLocked();
  50
  51 u32 ThreadCreate(u32 tid, bool detached, void *arg = nullptr);
  52 void ThreadFinish();
  53
  54 ThreadContextLsanBase *GetCurrentThread();
  55 inline u32 GetCurrentThreadId() {
  56   ThreadContextLsanBase *ctx = GetCurrentThread();
  57   return ctx ? ctx->tid : kInvalidTid;
  58 }
  59 void SetCurrentThread(ThreadContextLsanBase *tctx);
  60 void EnsureMainThreadIDIsCorrect();
  61
  62 }  // namespace __lsan
  63
  64 #endif  // LSAN_THREAD_H