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