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__)
69 # if defined(__powerpc64__)
70 const uptr kAllocatorSpace
= 0xa0000000000ULL
;
71 const uptr kAllocatorSize
= 0x20000000000ULL
; // 2T.
73 const uptr kAllocatorSpace
= 0x600000000000ULL
;
74 const uptr kAllocatorSize
= 0x40000000000ULL
; // 4T.
76 template <typename AddressSpaceViewTy
>
77 struct AP64
{ // Allocator64 parameters. Deliberately using a short name.
78 static const uptr kSpaceBeg
= kAllocatorSpace
;
79 static const uptr kSpaceSize
= kAllocatorSize
;
80 static const uptr kMetadataSize
= sizeof(ChunkMetadata
);
81 typedef DefaultSizeClassMap SizeClassMap
;
82 typedef NoOpMapUnmapCallback MapUnmapCallback
;
83 static const uptr kFlags
= 0;
84 using AddressSpaceView
= AddressSpaceViewTy
;
87 template <typename AddressSpaceView
>
88 using PrimaryAllocatorASVT
= SizeClassAllocator64
<AP64
<AddressSpaceView
>>;
89 using PrimaryAllocator
= PrimaryAllocatorASVT
<LocalAddressSpaceView
>;
92 template <typename AddressSpaceView
>
93 using AllocatorASVT
= CombinedAllocator
<PrimaryAllocatorASVT
<AddressSpaceView
>>;
94 using Allocator
= AllocatorASVT
<LocalAddressSpaceView
>;
95 using AllocatorCache
= Allocator::AllocatorCache
;
97 Allocator::AllocatorCache
*GetAllocatorCache();
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
);
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
);
106 void *lsan_reallocarray(void *p
, uptr nmemb
, uptr size
,
107 const StackTrace
&stack
);
108 void *lsan_calloc(uptr nmemb
, uptr size
, const StackTrace
&stack
);
109 void *lsan_valloc(uptr size
, const StackTrace
&stack
);
110 void *lsan_pvalloc(uptr size
, const StackTrace
&stack
);
111 uptr
lsan_mz_size(const void *p
);
113 } // namespace __lsan
115 #endif // LSAN_ALLOCATOR_H