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 kInternalAllocatorSpace = 0;
  25 static const u64 kInternalAllocatorSize = SANITIZER_MMAP_RANGE_SIZE;
  26 static const uptr kInternalAllocatorRegionSizeLog = 20;
  27 #if SANITIZER_WORDSIZE == 32
  28 static const uptr kInternalAllocatorNumRegions =
  29     kInternalAllocatorSize >> kInternalAllocatorRegionSizeLog;
  30 typedef FlatByteMap<kInternalAllocatorNumRegions> ByteMap;
  31 #else
  32 static const uptr kInternalAllocatorNumRegions =
  33     kInternalAllocatorSize >> kInternalAllocatorRegionSizeLog;
  34 typedef TwoLevelByteMap<(kInternalAllocatorNumRegions >> 12), 1 << 12> ByteMap;
  35 #endif
  36 typedef SizeClassAllocator32<
  37     kInternalAllocatorSpace, kInternalAllocatorSize, 0, InternalSizeClassMap,
  38     kInternalAllocatorRegionSizeLog, ByteMap> PrimaryInternalAllocator;
  39
  40 typedef SizeClassAllocatorLocalCache<PrimaryInternalAllocator>
  41     InternalAllocatorCache;
  42
  43 typedef CombinedAllocator<PrimaryInternalAllocator, InternalAllocatorCache,
  44                           LargeMmapAllocator<> > InternalAllocator;
  45
  46 void *InternalAlloc(uptr size, InternalAllocatorCache *cache = nullptr);
  47 void InternalFree(void *p, InternalAllocatorCache *cache = nullptr);
  48 InternalAllocator *internal_allocator();
  49
  50 enum InternalAllocEnum {
  51   INTERNAL_ALLOC
  52 };
  53
  54 } // namespace __sanitizer
  55
  56 inline void *operator new(__sanitizer::operator_new_size_type size,
  57                           InternalAllocEnum) {
  58   return InternalAlloc(size);
  59 }
  60
  61 #endif // SANITIZER_ALLOCATOR_INTERNAL_H