1 //=-- lsan_allocator.h ----------------------------------------------------===//
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
7 //===----------------------------------------------------------------------===//
9 // This file is a part of LeakSanitizer.
10 // Allocator for standalone LSan.
12 //===----------------------------------------------------------------------===//
14 #ifndef LSAN_ALLOCATOR_H
15 #define LSAN_ALLOCATOR_H
17 #include "sanitizer_common/sanitizer_allocator.h"
18 #include "sanitizer_common/sanitizer_common.h"
19 #include "sanitizer_common/sanitizer_internal_defs.h"
20 #include "lsan_common.h"
24 void *Allocate(const StackTrace &stack, uptr size, uptr alignment,
26 void Deallocate(void *p);
27 void *Reallocate(const StackTrace &stack, void *p, uptr new_size,
29 uptr GetMallocUsableSize(const void *p);
31 template<typename Callable>
32 void ForEachChunk(const Callable &callback);
34 void GetAllocatorCacheRange(uptr *begin, uptr *end);
35 void AllocatorThreadFinish();
36 void InitializeAllocator();
38 const bool kAlwaysClearMemory = true;
40 struct ChunkMetadata {
41 u8 allocated : 8; // Must be first.
43 #if SANITIZER_WORDSIZE == 64
44 uptr requested_size : 54;
46 uptr requested_size : 32;
52 #if defined(__mips64) || defined(__aarch64__) || defined(__i386__) || \
54 template <typename AddressSpaceViewTy>
56 static const uptr kSpaceBeg = 0;
57 static const u64 kSpaceSize = SANITIZER_MMAP_RANGE_SIZE;
58 static const uptr kMetadataSize = sizeof(ChunkMetadata);
59 typedef __sanitizer::CompactSizeClassMap SizeClassMap;
60 static const uptr kRegionSizeLog = 20;
61 using AddressSpaceView = AddressSpaceViewTy;
62 typedef NoOpMapUnmapCallback MapUnmapCallback;
63 static const uptr kFlags = 0;
65 template <typename AddressSpaceView>
66 using PrimaryAllocatorASVT = SizeClassAllocator32<AP32<AddressSpaceView>>;
67 using PrimaryAllocator = PrimaryAllocatorASVT<LocalAddressSpaceView>;
68 #elif defined(__x86_64__) || defined(__powerpc64__) || defined(__s390x__)
69 # if SANITIZER_FUCHSIA
70 const uptr kAllocatorSpace = ~(uptr)0;
71 const uptr kAllocatorSize = 0x40000000000ULL; // 4T.
72 # elif defined(__powerpc64__)
73 const uptr kAllocatorSpace = 0xa0000000000ULL;
74 const uptr kAllocatorSize = 0x20000000000ULL; // 2T.
75 #elif defined(__s390x__)
76 const uptr kAllocatorSpace = 0x40000000000ULL;
77 const uptr kAllocatorSize = 0x40000000000ULL; // 4T.
79 const uptr kAllocatorSpace = 0x600000000000ULL;
80 const uptr kAllocatorSize = 0x40000000000ULL; // 4T.
82 template <typename AddressSpaceViewTy>
83 struct AP64 { // Allocator64 parameters. Deliberately using a short name.
84 static const uptr kSpaceBeg = kAllocatorSpace;
85 static const uptr kSpaceSize = kAllocatorSize;
86 static const uptr kMetadataSize = sizeof(ChunkMetadata);
87 typedef DefaultSizeClassMap SizeClassMap;
88 typedef NoOpMapUnmapCallback MapUnmapCallback;
89 static const uptr kFlags = 0;
90 using AddressSpaceView = AddressSpaceViewTy;
93 template <typename AddressSpaceView>
94 using PrimaryAllocatorASVT = SizeClassAllocator64<AP64<AddressSpaceView>>;
95 using PrimaryAllocator = PrimaryAllocatorASVT<LocalAddressSpaceView>;
98 template <typename AddressSpaceView>
99 using AllocatorASVT = CombinedAllocator<PrimaryAllocatorASVT<AddressSpaceView>>;
100 using Allocator = AllocatorASVT<LocalAddressSpaceView>;
101 using AllocatorCache = Allocator::AllocatorCache;
103 Allocator::AllocatorCache *GetAllocatorCache();
105 int lsan_posix_memalign(void **memptr, uptr alignment, uptr size,
106 const StackTrace &stack);
107 void *lsan_aligned_alloc(uptr alignment, uptr size, const StackTrace &stack);
108 void *lsan_memalign(uptr alignment, uptr size, const StackTrace &stack);
109 void *lsan_malloc(uptr size, const StackTrace &stack);
110 void lsan_free(void *p);
111 void *lsan_realloc(void *p, uptr size, const StackTrace &stack);
112 void *lsan_reallocarray(void *p, uptr nmemb, uptr size,
113 const StackTrace &stack);
114 void *lsan_calloc(uptr nmemb, uptr size, const StackTrace &stack);
115 void *lsan_valloc(uptr size, const StackTrace &stack);
116 void *lsan_pvalloc(uptr size, const StackTrace &stack);
117 uptr lsan_mz_size(const void *p);
119 } // namespace __lsan
121 #endif // LSAN_ALLOCATOR_H