tsan_mman.h 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. //===-- tsan_mman.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-exception
  6. //
  7. //===----------------------------------------------------------------------===//
  8. //
  9. // This file is a part of ThreadSanitizer (TSan), a race detector.
  10. //
  11. //===----------------------------------------------------------------------===//
  12. #ifndef TSAN_MMAN_H
  13. #define TSAN_MMAN_H
  14. #include "tsan_defs.h"
  15. namespace __tsan {
  16. const uptr kDefaultAlignment = 16;
  17. void InitializeAllocator();
  18. void InitializeAllocatorLate();
  19. void ReplaceSystemMalloc();
  20. void AllocatorProcStart(Processor *proc);
  21. void AllocatorProcFinish(Processor *proc);
  22. void AllocatorPrintStats();
  23. // For user allocations.
  24. void *user_alloc_internal(ThreadState *thr, uptr pc, uptr sz,
  25. uptr align = kDefaultAlignment, bool signal = true);
  26. // Does not accept NULL.
  27. void user_free(ThreadState *thr, uptr pc, void *p, bool signal = true);
  28. // Interceptor implementations.
  29. void *user_alloc(ThreadState *thr, uptr pc, uptr sz);
  30. void *user_calloc(ThreadState *thr, uptr pc, uptr sz, uptr n);
  31. void *user_realloc(ThreadState *thr, uptr pc, void *p, uptr sz);
  32. void *user_reallocarray(ThreadState *thr, uptr pc, void *p, uptr sz, uptr n);
  33. void *user_memalign(ThreadState *thr, uptr pc, uptr align, uptr sz);
  34. int user_posix_memalign(ThreadState *thr, uptr pc, void **memptr, uptr align,
  35. uptr sz);
  36. void *user_aligned_alloc(ThreadState *thr, uptr pc, uptr align, uptr sz);
  37. void *user_valloc(ThreadState *thr, uptr pc, uptr sz);
  38. void *user_pvalloc(ThreadState *thr, uptr pc, uptr sz);
  39. uptr user_alloc_usable_size(const void *p);
  40. // Invoking malloc/free hooks that may be installed by the user.
  41. void invoke_malloc_hook(void *ptr, uptr size);
  42. void invoke_free_hook(void *ptr);
  43. // For internal data structures.
  44. void *Alloc(uptr sz);
  45. void FreeImpl(void *p);
  46. template <typename T, typename... Args>
  47. T *New(Args &&...args) {
  48. return new (Alloc(sizeof(T))) T(static_cast<Args &&>(args)...);
  49. }
  50. template <typename T>
  51. void Free(T *&p) {
  52. if (p == nullptr)
  53. return;
  54. FreeImpl(p);
  55. p = nullptr;
  56. }
  57. template <typename T>
  58. void DestroyAndFree(T *&p) {
  59. if (p == nullptr)
  60. return;
  61. p->~T();
  62. Free(p);
  63. }
  64. } // namespace __tsan
  65. #endif // TSAN_MMAN_H