libsanitizer/sanitizer_common/sanitizer_allocator_internal.h

   1 //===-- sanitizer_allocator_internal.h --------------------------*- C++ -*-===//
   2 //
   3 // This file is distributed under the University of Illinois Open Source
   4 // License. See LICENSE.TXT for details.
   5 //
   6 //===----------------------------------------------------------------------===//
   7 //
   8 // This allocator is used inside run-times.
   9 //
  10 //===----------------------------------------------------------------------===//
  11
  12 #ifndef SANITIZER_ALLOCATOR_INTERNAL_H
  13 #define SANITIZER_ALLOCATOR_INTERNAL_H
  14
  15 #include "sanitizer_allocator.h"
  16 #include "sanitizer_internal_defs.h"
  17
  18 namespace __sanitizer {
  19
  20 // FIXME: Check if we may use even more compact size class map for internal
  21 // purposes.
  22 typedef CompactSizeClassMap InternalSizeClassMap;
  23
  24 static const uptr kInternalAllocatorRegionSizeLog = 20;
  25 static const uptr kInternalAllocatorNumRegions =
  26     SANITIZER_MMAP_RANGE_SIZE >> kInternalAllocatorRegionSizeLog;
  27 #if SANITIZER_WORDSIZE == 32
  28 typedef FlatByteMap<kInternalAllocatorNumRegions> ByteMap;
  29 #else
  30 typedef TwoLevelByteMap<(kInternalAllocatorNumRegions >> 12), 1 << 12> ByteMap;
  31 #endif
  32 struct AP32 {
  33   static const uptr kSpaceBeg = 0;
  34   static const u64 kSpaceSize = SANITIZER_MMAP_RANGE_SIZE;
  35   static const uptr kMetadataSize = 0;
  36   typedef InternalSizeClassMap SizeClassMap;
  37   static const uptr kRegionSizeLog = kInternalAllocatorRegionSizeLog;
  38   typedef __sanitizer::ByteMap ByteMap;
  39   typedef NoOpMapUnmapCallback MapUnmapCallback;
  40   static const uptr kFlags = 0;
  41 };
  42 typedef SizeClassAllocator32<AP32> PrimaryInternalAllocator;
  43
  44 typedef SizeClassAllocatorLocalCache<PrimaryInternalAllocator>
  45     InternalAllocatorCache;
  46
  47 typedef CombinedAllocator<PrimaryInternalAllocator, InternalAllocatorCache,
  48                           LargeMmapAllocator<NoOpMapUnmapCallback, DieOnFailure>
  49                          > InternalAllocator;
  50
  51 void *InternalAlloc(uptr size, InternalAllocatorCache *cache = nullptr,
  52                     uptr alignment = 0);
  53 void *InternalRealloc(void *p, uptr size,
  54                       InternalAllocatorCache *cache = nullptr);
  55 void *InternalCalloc(uptr countr, uptr size,
  56                      InternalAllocatorCache *cache = nullptr);
  57 void InternalFree(void *p, InternalAllocatorCache *cache = nullptr);
  58 InternalAllocator *internal_allocator();
  59
  60 enum InternalAllocEnum {
  61   INTERNAL_ALLOC
  62 };
  63
  64 } // namespace __sanitizer
  65
  66 inline void *operator new(__sanitizer::operator_new_size_type size,
  67                           __sanitizer::InternalAllocEnum) {
  68   return __sanitizer::InternalAlloc(size);
  69 }
  70
  71 #endif // SANITIZER_ALLOCATOR_INTERNAL_H