111
|
1 //=-- lsan_allocator.h ----------------------------------------------------===//
|
|
2 //
|
145
|
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
|
111
|
6 //
|
|
7 //===----------------------------------------------------------------------===//
|
|
8 //
|
|
9 // This file is a part of LeakSanitizer.
|
|
10 // Allocator for standalone LSan.
|
|
11 //
|
|
12 //===----------------------------------------------------------------------===//
|
|
13
|
|
14 #ifndef LSAN_ALLOCATOR_H
|
|
15 #define LSAN_ALLOCATOR_H
|
|
16
|
|
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"
|
|
21
|
|
22 namespace __lsan {
|
|
23
|
|
24 void *Allocate(const StackTrace &stack, uptr size, uptr alignment,
|
|
25 bool cleared);
|
|
26 void Deallocate(void *p);
|
|
27 void *Reallocate(const StackTrace &stack, void *p, uptr new_size,
|
|
28 uptr alignment);
|
|
29 uptr GetMallocUsableSize(const void *p);
|
|
30
|
|
31 template<typename Callable>
|
|
32 void ForEachChunk(const Callable &callback);
|
|
33
|
|
34 void GetAllocatorCacheRange(uptr *begin, uptr *end);
|
|
35 void AllocatorThreadFinish();
|
|
36 void InitializeAllocator();
|
|
37
|
|
38 const bool kAlwaysClearMemory = true;
|
|
39
|
|
40 struct ChunkMetadata {
|
|
41 u8 allocated : 8; // Must be first.
|
|
42 ChunkTag tag : 2;
|
|
43 #if SANITIZER_WORDSIZE == 64
|
|
44 uptr requested_size : 54;
|
|
45 #else
|
|
46 uptr requested_size : 32;
|
|
47 uptr padding : 22;
|
|
48 #endif
|
|
49 u32 stack_trace_id;
|
|
50 };
|
|
51
|
|
52 #if defined(__mips64) || defined(__aarch64__) || defined(__i386__) || \
|
|
53 defined(__arm__)
|
145
|
54 template <typename AddressSpaceViewTy>
|
111
|
55 struct AP32 {
|
|
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;
|
145
|
60 static const uptr kRegionSizeLog = 20;
|
|
61 using AddressSpaceView = AddressSpaceViewTy;
|
111
|
62 typedef NoOpMapUnmapCallback MapUnmapCallback;
|
|
63 static const uptr kFlags = 0;
|
|
64 };
|
145
|
65 template <typename AddressSpaceView>
|
|
66 using PrimaryAllocatorASVT = SizeClassAllocator32<AP32<AddressSpaceView>>;
|
|
67 using PrimaryAllocator = PrimaryAllocatorASVT<LocalAddressSpaceView>;
|
111
|
68 #elif defined(__x86_64__) || defined(__powerpc64__)
|
145
|
69 # if defined(__powerpc64__)
|
|
70 const uptr kAllocatorSpace = 0xa0000000000ULL;
|
|
71 const uptr kAllocatorSize = 0x20000000000ULL; // 2T.
|
|
72 # else
|
|
73 const uptr kAllocatorSpace = 0x600000000000ULL;
|
|
74 const uptr kAllocatorSize = 0x40000000000ULL; // 4T.
|
|
75 # endif
|
|
76 template <typename AddressSpaceViewTy>
|
111
|
77 struct AP64 { // Allocator64 parameters. Deliberately using a short name.
|
145
|
78 static const uptr kSpaceBeg = kAllocatorSpace;
|
|
79 static const uptr kSpaceSize = kAllocatorSize;
|
111
|
80 static const uptr kMetadataSize = sizeof(ChunkMetadata);
|
|
81 typedef DefaultSizeClassMap SizeClassMap;
|
|
82 typedef NoOpMapUnmapCallback MapUnmapCallback;
|
|
83 static const uptr kFlags = 0;
|
145
|
84 using AddressSpaceView = AddressSpaceViewTy;
|
111
|
85 };
|
|
86
|
145
|
87 template <typename AddressSpaceView>
|
|
88 using PrimaryAllocatorASVT = SizeClassAllocator64<AP64<AddressSpaceView>>;
|
|
89 using PrimaryAllocator = PrimaryAllocatorASVT<LocalAddressSpaceView>;
|
111
|
90 #endif
|
145
|
91
|
|
92 template <typename AddressSpaceView>
|
|
93 using AllocatorASVT = CombinedAllocator<PrimaryAllocatorASVT<AddressSpaceView>>;
|
|
94 using Allocator = AllocatorASVT<LocalAddressSpaceView>;
|
|
95 using AllocatorCache = Allocator::AllocatorCache;
|
111
|
96
|
145
|
97 Allocator::AllocatorCache *GetAllocatorCache();
|
111
|
98
|
145
|
99 int lsan_posix_memalign(void **memptr, uptr alignment, uptr size,
|
|
100 const StackTrace &stack);
|
|
101 void *lsan_aligned_alloc(uptr alignment, uptr size, const StackTrace &stack);
|
111
|
102 void *lsan_memalign(uptr alignment, uptr size, const StackTrace &stack);
|
|
103 void *lsan_malloc(uptr size, const StackTrace &stack);
|
|
104 void lsan_free(void *p);
|
|
105 void *lsan_realloc(void *p, uptr size, const StackTrace &stack);
|
145
|
106 void *lsan_reallocarray(void *p, uptr nmemb, uptr size,
|
|
107 const StackTrace &stack);
|
111
|
108 void *lsan_calloc(uptr nmemb, uptr size, const StackTrace &stack);
|
|
109 void *lsan_valloc(uptr size, const StackTrace &stack);
|
145
|
110 void *lsan_pvalloc(uptr size, const StackTrace &stack);
|
111
|
111 uptr lsan_mz_size(const void *p);
|
|
112
|
|
113 } // namespace __lsan
|
|
114
|
|
115 #endif // LSAN_ALLOCATOR_H
|